LauRooFitSlave.hh

Go to the documentation of this file.

/*
Copyright 2017 University of Warwick

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

/*
Laura++ package authors:
John Back
Paul Harrison
Thomas Latham
*/

#ifndef LAU_ROO_FIT_SLAVE
#define LAU_ROO_FIT_SLAVE

#include <utility>
#include <vector>

#include "RooAbsPdf.h"
#include "RooAbsData.h"
#include "RooCategory.h"
#include "RooNLLVar.h"
#include "TMatrixDfwd.h"
#include "TString.h"

#include "LauSimFitSlave.hh"

class LauParameter;


class LauRooFitSlave : public LauSimFitSlave {

        public:
                LauRooFitSlave( RooAbsPdf& model, const Bool_t extended, const RooArgSet& vars, const TString& weightVarName = "" );

                virtual ~LauRooFitSlave();

                virtual void initialise();


                virtual void setParsFromMinuit(Double_t* par, Int_t npar);

                virtual Double_t getTotNegLogLikelihood();

        protected:


                virtual void prepareInitialParArray( TObjArray& array );


                LauParameter* convertToLauParameter( const RooRealVar* rooParameter ) const;


                std::vector< std::pair<RooRealVar*,LauParameter*> > convertToLauParameters( const RooFormulaVar* rooFormula ) const;


                virtual void finaliseExperiment( const LauAbsFitter::FitStatus& fitStat, const TObjArray* parsFromMaster, const TMatrixD* covMat, TObjArray& parsToMaster );


                virtual Bool_t verifyFitData(const TString& dataFileName, const TString& dataTreeName);


                virtual UInt_t readExperimentData();

                virtual void cacheInputFitVars();

        private:
                void cleanData();

                LauRooFitSlave(const LauRooFitSlave& rhs);

                LauRooFitSlave& operator=(const LauRooFitSlave& rhs);

                RooAbsPdf& model_;

                RooArgSet dataVars_;

                TString weightVarName_;

                TFile* dataFile_;

                TTree* dataTree_;

                RooAbsData* exptData_;

                const Bool_t extended_;

                RooCategory iExptCat_;

                RooNLLVar* nllVar_;

                std::vector<RooRealVar*> fitVars_;

                std::vector<LauParameter*> fitPars_;

                ClassDef(LauRooFitSlave,0);
};

#endif