rarCompBase.cc

Go to the documentation of this file.

/*****************************************************************************
* Project: BaBar detector at the SLAC PEP-II B-factory
* Package: RooRarFit
 *    File: $Id: rarCompBase.cc,v 1.19 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 component base class for RooRarFit
//
// BEGIN_HTML
// This class provides component base 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/RooStringVar.hh"

#include "RooRarFit/rarCompBase.hh"

ClassImp(rarCompBase)
  ;

rarCompBase::rarCompBase()
  : rarBasePdf(),
    _nComp(0)
{
  setControlBits("UseBasePdfFit");
  init();
}

rarCompBase::rarCompBase(const char *configFile, const char *configSec,
                         const char *configStr,
                         rarDatasets *theDatasets, RooDataSet *theData,
                         const char *name, const char *title,
                         Bool_t useBasePdfFit)
  : rarBasePdf(configFile, configSec, configStr,
               theDatasets, theData, name, title),
    _nComp(0)
{
  if (useBasePdfFit) setControlBits("UseBasePdfFit");
  else setControlBits("noUseBasePdfFit");
  init();
}

rarCompBase::~rarCompBase()
{
}

void rarCompBase::init()
{
  cout<<"init of rarCompBase for "<<GetName()<<":"<<endl;
  
  createPdfs("Comps", &_pdfList, &_subPdfs);
  _nComp=_pdfList.GetSize();
  if (_nComp<=0) {
    cout<<"no components defined in RooStringVar \"Comps\" in section \""
        <<_configSec<<"\""<<endl;
    exit(-1);
  }
  
  // add protDataVars
  addProtVars();
  
}

void rarCompBase::setSimPdf(RooSimultaneous *simPdf)
{
  rarBasePdf::setSimPdf(simPdf);
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    thePdf->setSimPdf(simPdf);
  }
}

RooArgSet rarCompBase::getParams()
{
  RooArgSet paramSet(rarBasePdf::getParams());
  // then add params in each of the comps
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    paramSet.add(thePdf->getParams());
  }
  return paramSet;  
}

void rarCompBase::preAction()
{
  cout<<" In rarCompBase preAction for "<<GetName()<<endl;
  // first run the function in base class
  rarBasePdf::preAction();
  // then run the function in each of the comps
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    thePdf->preAction();
  }
}

RooArgSet rarCompBase::getArgSet(TString paramNames, Bool_t useRead,
                                 RooArgSet *fullSet)
{
  // first for its own (base class)
  RooArgSet retVal(rarBasePdf::getArgSet(paramNames, useRead, fullSet));
  // then run the function in each of the comps if #useRead is set
  if (useRead) {
    for (Int_t i=0; i<_nComp; i++) {
      rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
      retVal.add(thePdf->getArgSet(paramNames, useRead, fullSet));
    }
  }
  return retVal;
}

RooArgSet rarCompBase::getCorrCoeffs()
{
  RooArgSet theSet(rarBasePdf::getCorrCoeffs());
  // then add correlation coeffs in each of the comps
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    theSet.add(thePdf->getCorrCoeffs());
  }
  return theSet;
}

void rarCompBase::doPdfFit(TString pdfList)
{
  // first for xtraPfs
  doXPdfFit(pdfList);
  
  cout<<endl<<" In rarCompBase doPdfFit for "<<GetName()<<endl;
  if (!getControlBit("PdfFit")) return;
  if (getControlBit("UseBasePdfFit")) return rarBasePdf::doPdfFit(pdfList);
  // go through all its components
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    thePdf->doPdfFit(pdfList);
  }
  if (getControlBit("UseAlsoBasePdfFit")) return rarBasePdf::doPdfFit(pdfList);
}

RooPlot *rarCompBase::doPdfPlot(TList &plotList, TString pdfList)
{
  RooPlot *frame(0);
  // first for xtraPdfs
  frame=doXPdfPlot(plotList, pdfList);
  
  cout<<endl<<" In rarCompBase doPdfPlot for "<<GetName()<<endl;
  
  //if (!getControlBit("PdfFit")) return frame;
  if (!getControlBit("PdfPlot")) return frame;
  if (getControlBit("PdfPlotDone")) return frame;
  
  // save obs and param
  string obsSaveStr, coeffParamSSaver;
  RooArgSet *theParams(0);
  if (getControlBit("UseBasePdfFit")) {
    writeToStr(_fObsSet, obsSaveStr);
    // restore the fit value for plotting
    theParams=_thePdf->getParameters(_theData);
    writeToStr(*theParams, coeffParamSSaver);
    readFromStr(*theParams, _afterFitSaverStr);
  }
  
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    frame=thePdf->doPdfPlot(plotList, pdfList);
  }
  
  // restore params and obs
  if (getControlBit("UseBasePdfFit")) {// restore params and obs
    readFromStr(*theParams, coeffParamSSaver);
    readFromStr(_fObsSet, obsSaveStr);
  }
  setControlBits("PdfPlotDone");
  
  return frame;
}

RooArgList rarCompBase::getPdfsWOvar(RooArgList ignoredObs)
{
  RooArgList thePdfs;
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    thePdfs.add(*thePdf->getPdfWOvar(ignoredObs));
  }
  //thePdfs.Print("v");
  //thePdfs.Print();
  return thePdfs;
}

RooAbsPdf *rarCompBase::getProtGen()
{
  if (_theProtGen) return _theProtGen;
  if (_subProtGenPdfs.getSize()<1) {
    for (Int_t i=0; i<_nComp; i++) {
      rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
      _subProtGenPdfs.add(*thePdf->getProtGen());
    }
  }
  for (Int_t i=0; i<_nComp; i++)
    if (!((rarBasePdf*)_pdfList.At(i))->protGenIsDummy())return _theProtGen;
  // if we go this far, we should have dummyPdf for all subGenPdfs
  setControlBits("dummySubGenPdfs");
  
  return _theProtGen;
}

void rarCompBase::attachDataSet(const RooAbsData &data)
{
  // first call base class'
  rarBasePdf::attachDataSet(data);
  // then call its sub pdfs
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    thePdf->attachDataSet(data);
  }
}

Bool_t rarCompBase::isNegativeValue()
{

  Bool_t isNegative(kFALSE);
  // first call its base class'
  isNegative=rarBasePdf::isNegativeValue();
  if (isNegative) return isNegative;
  // then call its sub pdfs
  for (Int_t i=0; i<_nComp; i++) {
    rarBasePdf *thePdf=(rarBasePdf*)_pdfList.At(i);
    isNegative=thePdf->isNegativeValue();
    if (isNegative) return isNegative;
  }
  return isNegative;
}

---