rarAdd.cc

Go to the documentation of this file.

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

// -- CLASS DESCRIPTION [RooRarFit] --
// This class provides AddPdf/AddModel class for RooRarFit
//
// BEGIN_HTML
// This class provides AddPdf/AddModel class for RooRarFit
// END_HTML
//

#include "RooRarFit/rarVersion.hh"

#include "Riostream.h"

#include "RooFitCore/RooArgList.hh"
#include "RooFitCore/RooDataSet.hh"
#include "RooFitCore/RooProdPdf.hh"
#include "RooFitCore/RooRealVar.hh"
#include "RooFitCore/RooStringVar.hh"
#include "RooFitCore/RooGlobalFunc.hh"
using namespace RooFit;

#include "RooFitCore/RooAddPdf.hh"
#include "RooFitCore/RooAddModel.hh"

#include "RooRarFit/rarAdd.hh"

ClassImp(rarAdd)
  ;

rarAdd::rarAdd()
  : rarCompBase(),
    _nCoeff(0)
{
  init();
}

rarAdd::rarAdd(const char *configFile, const char *configSec,
               const char *configStr,
               rarDatasets *theDatasets, RooDataSet *theData,
               const char *name, const char *title,
               Bool_t useBasePdfFit, Bool_t buildAddPdf)
  : rarCompBase(configFile, configSec, configStr,
                theDatasets, theData, name, title, useBasePdfFit),
    _nCoeff(0), _buildAddPdf(buildAddPdf)
{
  init();
}

rarAdd::~rarAdd()
{
}

void rarAdd::init()
{
  // create coeffs
  createAbsVars("Coeffs", 0, &_coeffs);
  _nCoeff=_coeffs.getSize();
  if (_nCoeff>_nComp) {
    cout<<"You should define at most "<<_nComp
        <<" coeffs in config \"Coeffs\" in section \""
        <<getVarSec()<<"\""<<endl;
    exit(-1);
  }
  if (_nCoeff<_nComp-1) {
    cout<<"You should define at least "<<_nComp-1
        <<" coeffs in config \"Coeffs\" in section \""
        <<getVarSec()<<"\""<<endl;
    exit(-1);
  }
  // make sure AddModel is non-extended
  if (("AddModel"==_pdfType) && (_nCoeff==_nComp)) {
    cout<<"You should define "<<_nComp-1
        <<" coeffs in config \"Coeffs\" in section \""
        <<getVarSec()<<"\""<<endl;
    exit(-1);
  }
  _params.Print("v");
  
  if (_buildAddPdf) {
    // build pdf function
    if ("AddModel"==_pdfType) {
      _thePdf=new RooAddModel(Form("the_%s", GetName()),
                              _pdfType+" "+GetTitle(), _subPdfs, _coeffs);
    } else {
      _thePdf=new RooAddPdf(Form("the_%s", GetName()), _pdfType+" "+GetTitle(),
                            _subPdfs, _coeffs);
    }
  }
  // by default do comp plot
  setControlBit("CompsOnPlot", "compsOnPlot");
  
  cout<<"done init of rarAdd for "<<GetName()<<endl<<endl;
}

RooPlot *rarAdd::doPdfPlot(TList &plotList, TString pdfList)
{
  return rarBasePdf::doPdfPlot(plotList, pdfList);
}

RooAbsPdf *rarAdd::getPdfWOvar(RooArgList ignoredObs)
{
  RooAbsPdf *thePdf=_thePdf;
  // check if thePdf depends on theVar
  if (!(thePdf->dependsOn(ignoredObs))) return thePdf;
  // create individual Pdf
  TString theName=Form("the_%s_%s", ignoredObs[0].GetName(), GetName());
  TString theTitle=Form("%s w/o %s", GetTitle(), ignoredObs[0].GetName());
  RooArgList myPdfList=getPdfsWOvar(ignoredObs);
  if ("AddModel"==_pdfType) {
    thePdf=new RooAddModel(theName, theTitle, myPdfList, _coeffs);
  } else {
    thePdf=new RooAddPdf(theName, theTitle, myPdfList, _coeffs);
  }
  //thePdf->Print("v");
  //thePdf->Print();
  if (!(thePdf->dependsOn(ignoredObs))) return thePdf;
  delete thePdf;
  // create a dummy for it
  thePdf=new RooGenericPdf(theName, theTitle, "1",
                           *_thePdf->getParameters(_fullObs));
  //thePdf->Print("v");
  //thePdf->Print();
  return thePdf;
}

RooAbsPdf *rarAdd::getProtGen()
{
  if (_theProtGen) return _theProtGen;
  // call super class' getProtGen
  rarCompBase::getProtGen();
  _theProtGen=_myDummyPdf;
  if (!getControlBit("dummySubGenPdfs")) {
    if ("AddModel"==_pdfType) {
      _theProtGen=new RooAddModel
        (Form("genAdd_%s", GetName()), Form("GenAdd for %s", GetName()),
         _subProtGenPdfs, _coeffs);
    } else {
      _theProtGen=new RooAddPdf
        (Form("genAdd_%s", GetName()), Form("GenAdd for %s", GetName()),
         _subProtGenPdfs, _coeffs);
    }
  }
  if (_protGenPdfs.getSize()>0) {
    RooArgList pdfList(_protGenPdfs);
    pdfList.add(*_theProtGen);
    _theProtGen = new RooProdPdf
      (Form("protGen_%s",GetName()),Form("Prot Gen for %s",GetName()),pdfList);
  }
  
  return _theProtGen;
}

---