rarDatasetDef.cc

Go to the documentation of this file.

/*****************************************************************************
* Project: BaBar detector at the SLAC PEP-II B-factory
* Package: RooRarFit
 *    File: $Id: rarDatasetDef.cc,v 1.13 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 dataset definition class for RooRarFit
//
// BEGIN_HTML
// This class provides dataset definition class for RooRarFit
// END_HTML
//

#include "RooRarFit/rarVersion.hh"

#include "Riostream.h"

#include "RooFitCore/RooArgList.hh"
#include "RooFitCore/RooArgSet.hh"
#include "RooFitCore/RooCategory.hh"
#include "RooFitCore/RooRealVar.hh"
#include "RooFitCore/RooStringVar.hh"

#include "RooRarFit/rarDatasetDef.hh"

ClassImp(rarDatasetDef)
  ;

rarDatasetDef::rarDatasetDef()
  : rarConfig(),
    _primaryObs(0), _addonCols(0)
{
  init();
}

rarDatasetDef::rarDatasetDef(const char *configFile, const char *configSec)
  : rarConfig(configFile, configSec, "null", "DatasetDef", "Dataset Def"),
    _primaryObs(0), _addonCols(0)
{
  init();
}

rarDatasetDef::~rarDatasetDef()
{
}

void rarDatasetDef::init()
{
  cout<<"init of rarDatasetDef:"<<endl;
  // first set #_createFundamental and #_fullNameSchema
  _createFundamental=kTRUE;
  _fullNameSchema="self";
  
  // create the full, primary, and addon obs sets.
  _fullObs=new RooArgSet("theFullObs");
  _primaryObs=new RooArgSet("thePrimaryObs");
  _addonCols=new RooArgSet("theAddOnCols");
  
  // read in field info
  createAbsVars("Fields", _fullObs, _primaryObs);
  // check number of fields
  Int_t nField=_primaryObs->getSize();
  if (nField<=0) {
    cout<<"no fields defined in RooStringVar \"Fields\" in section \""
        <<_configSec<<"\""<<endl;
    exit(-1);
  }
  // now for addon cols
  createAbsVars("AddOns", _fullObs, _addonCols);
  // print cout full obs
  _fullObs->Print("v");
  cout<<endl;
}

RooArgList *rarDatasetDef::getFormulaArgs(rarStrParser fStrParser)
{
  RooArgList *depList=new RooArgList;
  Int_t nArgs=fStrParser.nArgs();
  if (nArgs<=0) return depList;
  for (Int_t i=0; i<nArgs; i++) {
    if (isNumber(fStrParser[i])) break;
    RooAbsArg *theDep=createAbsReal(fStrParser[i], fStrParser[i]); // create it
    depList->add(*theDep);
  }
  
  return depList;
}

void rarDatasetDef::setVal(TString var, Double_t val)
{
  RooAbsArg *theArg=_primaryObs->find(var);
  { // RooRealVar
    RooRealVar *theVar=dynamic_cast<RooRealVar*>(theArg);
    if (theVar) {
      theVar->setVal(val);
      return;
    }
  }
  { // RooCategory
    RooCategory *theVar=dynamic_cast<RooCategory*>(theArg);
    if (theVar) {
      theVar->setIndex((Int_t)val);
      return;
    }
  }
  { // RooStringVar
    RooStringVar *theVar=dynamic_cast<RooStringVar*>(theArg);
    if (theVar) {
      theVar->setVal(Form("%8x", (UInt_t) val));
      return;
    }
  }
  
  static Int_t counter(0);
  counter++;
  if (counter<100)
    cout<<"Can not find "<<var<<endl;
}

void rarDatasetDef::setVal(TString var, TString val)
{
  RooAbsArg *theArg=_primaryObs->find(var);
  { // RooStringVar
    RooStringVar *theVar=dynamic_cast<RooStringVar*>(theArg);
    if (theVar) {
      theVar->setVal(val);
      return;
    }
  }
  { // RooCategory
    RooCategory *theVar=dynamic_cast<RooCategory*>(theArg);
    if (theVar) {
      theVar->setLabel(val);
      return;
    }
  }
  
  static Int_t counter(0);
  counter++;
  if (counter<100)
    cout<<"Can not find "<<var<<endl;
}

---