LauRooFitSlave.hh

Go to the documentation of this file.

// Copyright University of Warwick 2016 - 2017.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// Authors:
// Thomas Latham
// John Back
// Paul Harrison

#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