LauRooFitTask.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_TASK
#define LAU_ROO_FIT_TASK
 
#include "LauSimFitTask.hh"
 
#include "RooArgSet.h"
#include "TMatrixDfwd.h"
#include "TString.h"
 
#include <set>
#include <utility>
#include <vector>
 
class LauParameter;
class RooAbsData;
class RooAbsPdf;
class RooFormulaVar;
class RooNLLVar;
class RooRealVar;
class TFile;
class TTree;
 
class LauRooFitTask : public LauSimFitTask {
 
  public:
    LauRooFitTask( RooAbsPdf& model,
                   const Bool_t extended,
                   const RooArgSet& vars,
                   const TString& weightVarName = "" );
 
    virtual ~LauRooFitTask();
 
    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* parsFromCoordinator,
                                     const TMatrixD* covMat,
                                     TObjArray& parsToCoordinator );
 
 
    virtual Bool_t verifyFitData( const TString& dataFileName, const TString& dataTreeName );
 
 
    virtual UInt_t readExperimentData();
 
    virtual void cacheInputFitVars();
 
  private:
    void cleanData();
 
    LauRooFitTask( const LauRooFitTask& rhs );
 
    LauRooFitTask& operator=( const LauRooFitTask& rhs );
 
    RooAbsPdf& model_;
 
    RooArgSet dataVars_;
 
    TString weightVarName_;
 
    TFile* dataFile_;
 
    TTree* dataTree_;
 
    RooAbsData* exptData_;
 
    const Bool_t extended_;
 
    std::set<UInt_t> iExptSet_;
 
    RooNLLVar* nllVar_;
 
    std::vector<RooRealVar*> fitVars_;
 
    std::vector<LauParameter*> fitPars_;
 
    ClassDef( LauRooFitTask, 0 );
};
 
#endif