rarBasePdf.hh

Go to the documentation of this file.

/*****************************************************************************
* Project: BaBar detector at the SLAC PEP-II B-factory
* Package: RooRarFit
 *    File: $Id: rarBasePdf.rdl,v 1.33 2011/06/16 13:18:49 fwilson Exp $
 * Authors: Lei Zhang
 * History:
 *
 * Copyright (C) 2005-2012, University of California, Riverside
 *****************************************************************************/
#ifndef RAR_BASEPDF
#define RAR_BASEPDF

#define NCOLORS 11

#include <string>
using namespace std;

#include "TList.h"
#include "TString.h"
#include "TObject.h"

#include "RooFitCore/RooBinning.hh"
#include "RooFitCore/RooCategory.hh"
#include "RooFitCore/RooConstVar.hh"
#include "RooFitCore/RooExtendPdf.hh"
#include "RooFitCore/RooGenericPdf.hh"
#include "RooFitCore/RooSimultaneous.hh"

#include "RooRarFit/rarConfig.hh"
#include "RooRarFit/rarDatasets.hh"

class rarMLFitter;

class rarBasePdf : public rarConfig {
  
public:
  rarBasePdf();
  rarBasePdf(const char *configFile,const char*configSec,const char*configStr,
             rarDatasets *theDatasets, RooDataSet *theData,
             const char *name, const char *title);
  virtual ~rarBasePdf();
  
  virtual TString getVarSec() {return _paramSec;}
  
  virtual RooArgSet *getPrimaryObs() {return _datasets->getPrimaryObs();}
  
  virtual RooArgSet *getAddOnCols() {return _datasets->getAddOnCols();}
  
  virtual RooDataSet *getData(const char * /*name=0 */) {return _theData;}
  
  virtual rarDatasets *getDatasets() const {return _datasets;}
  
  virtual TString getPdfType() {return _pdfType;}
  
  virtual RooAbsPdf *getPdf() {return _thePdf;}
  virtual RooAbsPdf *getPdfWOvar(RooArgList ignoredObs);
  virtual RooAbsPdf *getDPdfWvar(RooRealVar *theVar);
  virtual RooAbsPdf *getSimPdf(RooSimultaneous *simPdf=0, RooAbsPdf *srcPdf=0);

  virtual void setFitter(rarMLFitter *theFitter) {_theFitter=theFitter;}
  virtual rarMLFitter *getFitter() {return _theFitter;}
  
  virtual void setSimPdf(RooSimultaneous *simPdf) {_theSimPdf=simPdf;}
  virtual void setCondObss(RooArgSet condObsSet);
  virtual void setFitData(RooDataSet *theData=0);
  virtual RooArgSet getParams();
  
  virtual void preAction();
  virtual RooArgSet getArgSet(TString paramNames, Bool_t useRead=kFALSE,
                              RooArgSet *fullSet=0);
  virtual void doPdfFit(TString pdfList="");
  virtual void attachDataSet(const RooAbsData &data);
  virtual Bool_t isNegativeValue();
  virtual RooPlot *doPdfPlot(TList &plotList, TString pdfList="");
  virtual RooArgSet getCorrCoeffs();
  virtual Double_t getCorrCoeff(const TString pn1, const TString pn2);
  virtual RooAbsPdf *getProtGen();
  virtual Bool_t protGenIsDummy() {return _theProtGen==_myDummyPdf;}
  
  virtual void setControlBit(TString controlBitStr, TString bitConfigStr="",
                             TString configSec="");
  virtual void setControlBits(TString controlBitsStr);
  virtual Bool_t getControlBit(TString controlBitStr);
  
  static Int_t getColor(Int_t i);
  
protected:
  virtual void init();
  virtual void addProtVars();
  virtual void addProtVars(TString configName, RooArgSet &protVars);
  virtual void setVarSec(TString paramSec) {_paramSec=paramSec;}
  virtual void addToParams(RooRealVar *theVar);
  virtual void addToObs(RooRealVar *theVar);
  virtual RooArgList *getFormulaArgs(rarStrParser fStrParser);
  virtual Double_t getFormulaVal(TString varStr);
  virtual rarBasePdf *createPdfs(TString Comps="Comps", TList *pdfList=0,
                                 RooAbsCollection *PDFs=0, TString secName="");
  virtual Bool_t matchCatType(RooCatType *catN, RooCatType *catO);
  virtual void saveFracName(TString fracName);
  virtual Bool_t isFracName(TString fracName);
  
  virtual RooBinning *getRange(RooRealVar *theVar, TString rPrefix,
                               Double_t &min, Double_t &max,
                               const Char_t *sec=0, Int_t *nBins=0);
  virtual void saveCorrCoeffs(RooFitResult *fr);
  virtual Bool_t saveCorrCoeff(TString corrCoefName, Double_t corrCoef,
                               Bool_t saveTrivial=kFALSE);
  virtual TString getCorrCoefName(const TString pn1, const TString pn2) const;
  virtual void doXPdfFit(TString pdfList="");
  virtual RooPlot *doXPdfPlot(TList &plotList, TString pdfList="");
  virtual RooPlot *doParamsOnPlot(RooPlot* frame, RooArgSet *params=0,
                                  Int_t sigDigits=2, Option_t *options="NELU",
                                  Double_t xmin=0.65, Double_t xmax=0.99,
                                  Double_t ymax=0.95);
  virtual RooPlot *doChi2OnPlot(RooPlot *frame);
  
  TString _pdfType; 
  
  RooArgSet _obsSet; 
  RooArgSet _fObsSet; 
  
  rarDatasets *_datasets; 
  RooDataSet *_theData; 
  RooArgSet _protDataVars; 
  RooArgSet _conditionalObs; 
  RooArgSet _condObss; 

  RooAbsPdf *_thePdf; 
  RooArgList _subPdfs; 
  RooAbsPdf *_theProtGen; 
  RooArgList _protGenPdfs; 
  RooSimultaneous *_thisSimPdf; 
  RooSimultaneous *_thisSimPdfWOP; 
  RooSimultaneous *_theSimPdf; 
  Int_t _nxPdf; 
  TList _xPdfList; 
  RooAbsPdf *_myDummyPdf; 
  RooArgSet _corrCoeffs; 
  static RooGenericPdf _dummyPdf; 
  static RooConstVar _dummyExpEvt; 
  static RooExtendPdf _dummyExtPdf; 
  static RooCategory _compCat; 
  static Int_t _rarColors[NCOLORS]; 
  static rarMLFitter *_theFitter; 
  static TString _fracNames; 

  TString _paramSec; 
  RooArgSet _params; 
  RooArgList _coeffs; 
  RooArgSet _xParams; 
  string _afterFitSaverStr; 
  
  TString _controlStr; 
  
private:
  rarBasePdf(const rarBasePdf&);
  ClassDef(rarBasePdf, 0) // RooRarFit base Pdf class
    ;
};

#endif

---