///
/// Converts a RooDataSet to a TTree which then can be
/// browsed.
///
TTree* Utils::convertRooDatasetToTTree(RooDataSet *d)
{
// set up the TTree based on the content of the first
// row of the dataset
map<string,float> variables; ///< the proxy variables
TTree* t = new TTree("tree", "tree");
TIterator* it = d->get(0)->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
variables.insert(pair<string,float>(p->GetName(),p->getVal()));
t->Branch(p->GetName(), &variables[p->GetName()], TString(p->GetName())+"/F");
}
delete it;
// loop over the dataset, filling the tree
int nEntries = d->sumEntries();
for ( int i=0; i<nEntries; i++ ){
it = d->get(i)->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
variables[p->GetName()] = p->getVal();
}
delete it;
t->Fill();
}
return t;
}
///
/// Merge two named sets of variables inside a RooWorkspace.
/// Duplicate variables will only be contained once.
///
void Utils::mergeNamedSets(RooWorkspace *w, TString mergedSet, TString set1, TString set2)
{
// 1. fill all variables into a vector
vector<string> varsAll;
TIterator* it = w->set(set1)->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ) varsAll.push_back(p->GetName());
delete it;
it = w->set(set2)->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ) varsAll.push_back(p->GetName());
delete it;
// 2. remove duplicates
sort(varsAll.begin(), varsAll.end());
vector<string> vars;
vars.push_back(varsAll[0]);
string previous = varsAll[0];
for ( int i=1; i<varsAll.size(); i++ ){
if ( previous==varsAll[i] ) continue;
vars.push_back(varsAll[i]);
previous=varsAll[i];
}
// 3. make new, combined set on the workspace
TString varsCommaList = "";
for ( int i=0; i<vars.size(); i++ ){
varsCommaList.Append(vars[i]);
if ( i<vars.size()-1 ) varsCommaList.Append(",");
}
w->defineSet(mergedSet, varsCommaList);
}
void LHCOWriter::AnalyseJets()
{
Jet *element;
Track *track;
Int_t counter;
fItJet->Reset();
while((element = static_cast<Jet*>(fItJet->Next())))
{
if(element->TauTag != 0) continue;
Reset();
counter = 0;
fItTrack->Reset();
while((track = static_cast<Track*>(fItTrack->Next())))
{
if(element->P4().DeltaR(track->P4()) < 0.5) ++counter;
}
fIntParam[1] = 4;
fDblParam[0] = element->Eta;
fDblParam[1] = element->Phi;
fDblParam[2] = element->PT;
fDblParam[3] = element->Mass;
fDblParam[4] = counter;
fDblParam[5] = element->BTag;
fDblParam[6] = element->EhadOverEem;
Write();
}
}
void PDF_GLWADS_DKDpi_K3pi::setObservables(config c)
{
switch(c)
{
case truth:{
setObservablesTruth();
break;
}
case toy:{
setObservablesToy();
break;
}
case lumi1fb:{
obsValSource = "1fb-1, ExpNll/sept2012K3PIResult.root";
TString File = this->dir+"/ExpNll/sept2012K3PIResult.root";
TFile *fr = TFile::Open(File);
RooFitResult *r = (RooFitResult*)fr->Get("fitresult_model_reducedData_binned");
assert(r);
TIterator* it = observables->createIterator();
while ( RooRealVar* pObs = (RooRealVar*)it->Next() )
{
RooRealVar* pRes = (RooRealVar*)r->floatParsFinal().find(obsTmkToMalcolm(pObs->GetName()));
pObs->setVal(pRes->getVal());
}
fr->Close();
delete r;
delete fr;
break;
}
case lumi3fb:{
obsValSource = "3fb-1 ANA v7 unblind"; // https://twiki.cern.ch/twiki/pub/LHCbPhysics/B2D0K/LHCb-ANA-2014-071-v7.pdf (see Vavas email 04/08/15)
// these get transformed over from the new inputs using ExpNll/transportGLWADS_new_to_old.py
// in the case of the DK only (robust) combination some of the observables don't exist
// usemap as the temp store
std::map< TString, double > vals;
vals["rkp_k3pi_obs"] = 0.0793;
vals["afav_dk_k3pi_obs"] = -0.0004;
vals["afav_dpi_k3pi_obs"] = 0.0;
vals["rp_dk_k3pi_obs"] = 0.018369;
vals["rm_dk_k3pi_obs"] = 0.009611;
vals["rp_dpi_k3pi_obs"] = 0.003683;
vals["rm_dpi_k3pi_obs"] = 0.003857;
// now can loop the observables and set the values
TIterator* it = observables->createIterator();
while ( RooRealVar* pObs = (RooRealVar*)it->Next() ){
pObs->setVal(vals[pObs->GetName()]);
}
vals.clear();
break;
}
default:{
cout << "PDF_GLWADS_DKDpi_K3pi::setObservables() : ERROR : config "+ConfigToTString(c)+" not found." << endl;
exit(1);
}
}
}
void printMassFrom2DParameters(RooWorkspace myws, TPad* Pad, bool isPbPb, string pdfName, bool isWeighted)
{
Pad->cd();
TLatex *t = new TLatex(); t->SetNDC(); t->SetTextSize(0.026); float dy = 0.025;
RooArgSet* Parameters = (RooArgSet*)myws.pdf(pdfName.c_str())->getParameters(RooArgSet(*myws.var("invMass"), *myws.var("ctau"), *myws.var("ctauErr")))->selectByAttrib("Constant",kFALSE);
TIterator* parIt = Parameters->createIterator();
for (RooRealVar* it = (RooRealVar*)parIt->Next(); it!=NULL; it = (RooRealVar*)parIt->Next() ) {
stringstream ss(it->GetName()); string s1, s2, s3, label;
getline(ss, s1, '_'); getline(ss, s2, '_'); getline(ss, s3, '_');
// Parse the parameter's labels
if(s1=="invMass" || s1=="ctauErr" || s1=="ctau"){continue;} else if(s1=="MassRatio"){continue;}
else if(s1=="One"){continue;} else if(s1=="mMin"){continue;} else if(s1=="mMax"){continue;}
if(s1=="RFrac2Svs1S"){ s1="R_{#psi(2S)/J/#psi}"; }
else if(s1=="rSigma21"){ s1="(#sigma_{2}/#sigma_{1})"; }
else if(s1.find("sigma")!=std::string::npos || s1.find("lambda")!=std::string::npos || s1.find("alpha")!=std::string::npos){
s1=Form("#%s",s1.c_str());
}
if(s2=="PbPbvsPP") { s2="PbPb/PP"; }
else if(s2=="Jpsi") { s2="J/#psi"; }
else if(s2=="Psi2S") { s2="#psi(2S)"; }
else if(s2=="Bkg") { s2="bkg"; }
else if(s2=="CtauRes") { continue; }
else if(s2=="JpsiNoPR") { continue; }
else if(s2=="JpsiPR") { continue; }
else if(s2=="Psi2SNoPR"){ continue; }
else if(s2=="Psi2SPR") { continue; }
else if(s2=="BkgNoPR") { continue; }
else if(s2=="BkgPR") { continue; }
else if(s2=="Bkg" && (s1=="N" || s1=="b")) { continue; }
else {continue;}
if(s3!=""){
label=Form("%s_{%s}^{%s}", s1.c_str(), s2.c_str(), s3.c_str());
}
else {
label=Form("%s^{%s}", s1.c_str(), s2.c_str());
}
// Print the parameter's results
if(s1=="N"){
t->DrawLatex(0.20, 0.76-dy, Form((isWeighted?"%s = %.6f#pm%.6f ":"%s = %.0f#pm%.0f "), label.c_str(), it->getValV(), it->getError())); dy+=0.045;
}
else if(s1.find("#sigma_{2}/#sigma_{1}")!=std::string::npos){
t->DrawLatex(0.20, 0.76-dy, Form("%s = %.3f#pm%.3f ", label.c_str(), it->getValV(), it->getError())); dy+=0.045;
}
else if(s1.find("sigma")!=std::string::npos){
t->DrawLatex(0.20, 0.76-dy, Form("%s = %.2f#pm%.2f MeV/c^{2}", label.c_str(), it->getValV()*1000., it->getError()*1000.)); dy+=0.045;
}
else if(s1.find("lambda")!=std::string::npos){
t->DrawLatex(0.20, 0.76-dy, Form("%s = %.4f#pm%.4f", label.c_str(), it->getValV(), it->getError())); dy+=0.045;
}
else if(s1.find("m")!=std::string::npos){
t->DrawLatex(0.20, 0.76-dy, Form("%s = %.5f#pm%.5f GeV/c^{2}", label.c_str(), it->getValV(), it->getError())); dy+=0.045;
}
else {
t->DrawLatex(0.20, 0.76-dy, Form("%s = %.4f#pm%.4f", label.c_str(), it->getValV(), it->getError())); dy+=0.045;
}
}
};
int Scheme::FindDOF(QString name, bool& isBase, bool &isExt)
{
if(Root.IsNull())
{
throw NoLabelInicialisationException("Scheme: label do not inicialised.");
}
// printf ("Scheme:10\n");
GetEqvList();
TIterator eqv = EqvList.GetIterator();
// printf ("Scheme:11\n");
int num = 1;
while (eqv.More())
{
bool isOK;
// printf ("Scheme:------12\n");
TElement el (eqv.Value());
TList B (el.GetField ("doflist", isOK));
TInteger isb (el.GetField ("isBase", isOK));
TInteger ise (el.GetField ("isExt", isOK));
// if (isOK) printf ("Scheme:13 OK OK OK\n");
TIterator dofit = B.GetIterator();
// printf ("Scheme:14\n");
while (dofit.More())
{
TString dof (dofit.Value());
if (name == dof.GetValue())
{
isBase = (isb.GetValue()!=0);
isExt = (ise.GetValue()!=0);
return num;
}
dofit.Next();
}
eqv.Next();
num++;
}
return -1;
}
void SetConstant(const RooArgSet * vars, Bool_t value ) {
//
// Set the constant attribute for all vars in the set
//
TIterator * pIter = vars->createIterator(); // we do own this
for(TObject * pObj = pIter->Next(); pObj; pObj = pIter->Next() ) {
((RooRealVar *)pObj)->setConstant(value);
}
delete pIter;
return;
}
void WriteMergeObjects( TFile *target ) {
cout << "Writing the merged data." << endl;
TIterator *nextobj = MergeObjects.MakeIterator();
TObjString *pathname_obj;
while( (pathname_obj = (TObjString *)nextobj->Next()) ) {
TString path,name;
SplitPathName(pathname_obj->String(),&path,&name);
TObject *obj = MergeObjects.GetValue(pathname_obj);
target->cd(path);
obj->Write( name );
delete obj;
}
MergeObjects.Clear();
target->Write();
// Temporarily let multiple root files remain if > 2GB
// Prevent Target_1.root Target_2.root, ... from happening.
// long long max_tree_size = 200000000000LL; // 200 GB
// if(TTree::GetMaxTreeSize() < max_tree_size ) {
// TTree::SetMaxTreeSize(max_tree_size);
// }
nextobj = MergeChains.MakeIterator();
TObjString *pathname_obj;
while( (pathname_obj = (TObjString *)nextobj->Next()) ) {
TString path,name;
SplitPathName(pathname_obj->String(),&path,&name);
TChain *ch = (TChain *)MergeChains.GetValue(pathname_obj);
target->cd(path);
ch->Merge(target,0,"KEEP");
delete ch;
// in case of multiple objects with same pathname, must remove
// this one from the list so we don't get the same (deleted)
// one next time we look up the same name
MergeChains.Remove(pathname_obj);
}
MergeChains.Clear();
InitializedMergeObjects = false;
}
void Utils::floatParameters(const RooAbsCollection* set)
{
TIterator* it = set->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
p->setConstant(false);
}
}
///
/// Float each parameter in the named set "parname" inside workspace "w".
///
void Utils::floatParameters(RooWorkspace* w, TString parname)
{
TIterator* it = w->set(parname)->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
p->setConstant(false);
}
}
af::Model OutValue::GetModelOfValue()
{
if(Root.IsNull())
{
throw NoLabelInicialisationException("OutValue: label do not inicialised.");
}
TIterator TITER;
Model vallr;
TITER.Init(Root, false);
while (TITER.More())
{
Model val = (Model)TITER.Value();
QString name = val.GetName();
if(name == "Model")
{
return val;
}
TITER.Next();
}
TElement e(Root);
throw NoExistException("OutValue: Model Of Value do not exists.", e);
return vallr;
}
///
/// load Parameter limits
/// by default the "free" limit is loaded, can be changed to "phys" by command line argument
///
void MethodDatasetsProbScan::loadParameterLimits() {
TString rangeName = arg->enforcePhysRange ? "phys" : "free";
if ( arg->debug ) cout << "DEBUG in Combiner::loadParameterLimits() : loading parameter ranges: " << rangeName << endl;
TIterator* it = w->set(pdf->getParName())->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ) setLimit(w, p->GetName(), rangeName);
delete it;
}
RooFitResult *breakDownFit(RooSimultaneous *m, RooAbsData *d, RooRealVar *mass, bool precondition = false){
if(precondition){
const char *catsName = m->indexCat().GetName();
TIterator *it = m->indexCat().typeIterator();
while(RooCatType* ci = dynamic_cast<RooCatType*>(it->Next())) {
const Text_t *catLabel = ci->GetName();
RooAbsPdf *pdf = m->getPdf(Form("%s",catLabel));
RooAbsData *reduced = d->reduce(SelectVars(*mass),Cut(Form("%s==%s::%s",catsName, catsName, catLabel)));
RooFitResult *r = pdf->fitTo(*reduced,PrintLevel(-1),Save(),
Minimizer("Minuit2","migrad"),Strategy(0),Hesse(false),Minos(false),Optimize(false)
);
cout << catsName << " " << catLabel << " M2migrad0 " << r->status() << endl;
if(r->status()!=0){
RooFitResult *r = pdf->fitTo(*reduced, PrintLevel(-1), Save());
cout << catsName << " " << catLabel << " Mmigrad1 " << r->status() << endl;
}
}
}
RooFitResult *r = m->fitTo(*d, Save(), PrintLevel(-1),
Strategy(0));
cout << "Global fit Mmigrad0 " << r->status() << endl;
if(r->status()!=0){
RooFitResult *r = m->fitTo(*d, PrintLevel(-1), Save(),
Minimizer("Minuit","minimize"),Strategy(2));
cout << "Global fit Mminimize2 " << r->status() << endl;
return r;
}
return r;
}
af::TList Scheme::GetIncludeList()
{
if(Root.IsNull())
{
throw NoLabelInicialisationException("Scheme: label do not inicialised.");
}
// return IncludeList;
TIterator TITER;
TList valls;
TITER.Init(Root, false);
while (TITER.More())
{
TList val = (TList)TITER.Value();
QString name = val.GetName();
if(name == "IncludeList")
{
return val;
}
TITER.Next();
}
TElement e(Root);
throw NoExistException("Scheme: Include List do not exists.", e);
return valls;
}
QString Include::GetFile()
{
if(Root.IsNull())
{
throw NoLabelInicialisationException("Include: label do not inicialised.");
}
if(Root.IsNull())
{
throw NoLabelInicialisationException("Include: label do not inicialised.");
}
TIterator TITER;
TITER.Init(Root, false);
while (TITER.More())
{
TElement val = TITER.Value();
QString name = val.GetName();
if(name == "File name")
{
TString valint = (TString)val;
return valint.GetValue();
}
TITER.Next();
}
return "";
}
void scaleToRate(const char* collectionName, double rateFactor) {
TRegexp reg(collectionName, kTRUE);
// gDirectory->ls();
TList* list = gDirectory->GetList() ;
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
while (obj = iter->Next()) {
if (! obj->InheritsFrom(TH1::Class())) {
// cout << "bugger" << endl;
continue;
}
TString name = obj->GetName();
cout << "Testing name: " << name << " against " << collectionName << endl;
if (TString(collectionName).MaybeRegexp()) {
cout << "we have a possible match" << endl;
cout << "Trying to match to " << TString(obj->GetName()) << endl;
if (TString(obj->GetName()).Index(reg) < 0 ) {
cout << "failure here. Argument returns " << TString(obj->GetName()).Index(reg) << endl;
continue;
}
}
else if (! name.BeginsWith(collectionName)) continue;
cout << "We're trying to scale" << name << endl;
((TH1*)obj)->Scale(rateFactor);
}
}
// Method by name
TH1F* eff(const char* name1, const char* name2, const char* name="eff"){
// Get a list of object and their iterator
TList* list = gDirectory->GetList() ;
TIterator* iter = list->MakeIterator();
// Loop over objects, set the pointers
TObject* obj;
TH1F* h1=0;
TH1F* h2=0;
TString str1 = Form("%s",name1);
TString str2 = Form("%s",name2);
while(obj=iter->Next()) {
TString objName = obj->GetName();
if (objName == str1) h1 = (TH1F*) obj;
if (objName == str2) h2 = (TH1F*) obj;
}
// quit if not found
if (h1 == 0) {
cout << "Histogram " << name1 << " not found" << endl;
return 0;
}
if (h2 == 0) {
cout << "Histogram " << name2 << " not found" << endl;
return 0;
}
// Call the method by pointer
TH1F* temp = eff(h1, h2, name);
return temp;
}
/**
* @brief Write all TObjects from a given TCollection into a certain directory structure in the file
*
* @param col pointer to a TCollection-based container
* @param dirname name of a directory inside the output file to which the objects should be written
* @param subdirname optional name of a subdirectory inside dirname to which the objects should be written
*
* This method whites all TObject-based objects contained in the TCollection-based container (see ROOT documentation)
* into a directory whose name is given by dirname inside the output file. If dirname does not exist
* in the output file, it will be created. Otherwise, contents of the col collection will be appended to an existing
* directory.
*
* If the optional subdirectory name is specified (subdirname parameter, defaults to empty string) then the
* contents of the collection will be written to "dirname/subdirname". If the "subdirname" directory does not
* exist inside the "dirname" directory, it will be created.
*
*/
void JPetWriter::writeCollection(const TCollection* col, const char* dirname, const char* subdirname)
{
TDirectory* current = fFile->GetDirectory(dirname);
if (!current) {
current = fFile->mkdir(dirname);
}
assert(current);
// use a subdirectory if requested by user
if (!std::string(subdirname).empty()) {
if (current->GetDirectory(subdirname)) {
current = current->GetDirectory(subdirname);
} else {
current = current->mkdir(subdirname);
}
}
assert(current);
current->cd();
TIterator* it = col->MakeIterator();
TObject* obj;
while ((obj = it->Next())) {
obj->Write();
}
fFile->cd();
}
void scalebins(const char* patORpfx, Double_t scale) {
TRegexp reg(patORpfx, kFALSE);
TList* list = gDirectory->GetList() ;
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
while (obj = iter->Next()) {
if (! obj->InheritsFrom(TH1::Class())) continue;
TString name = obj->GetName();
if (TString(patORpfx).MaybeRegexp()) {
if (TString(obj->GetName()).Index(reg) < 0 ) continue;
} else if (! name.BeginsWith(patORpfx)) continue;
Double_t binWidth, binContent, binError, newBinContent, newBinError;
for (Int_t i = 1; i <= ((TH1*)obj)->GetNbinsX(); ++i) {
binWidth = ((TH1*)obj)->GetBinWidth(i);
binContent = ((TH1*)obj)->GetBinContent(i);
binError = ((TH1*)obj)->GetBinError(i);
newBinContent = (binContent*scale)/binWidth;
newBinError = (binError*scale)/binWidth;
((TH1*)obj)->SetBinContent(i, newBinContent);
((TH1*)obj)->SetBinError(i, newBinContent);
// Rename y axis with scale
}
}
}
void yaxis(const char* patORpfx, const char* title) {
TRegexp reg(patORpfx, kFALSE);
TList* list = gDirectory->GetList() ;
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
while (obj = iter->Next()) {
if (! (obj->InheritsFrom(TH1::Class()) || obj->InheritsFrom(THStack::Class()))) continue;
TString name = obj->GetName();
if (TString(patORpfx).MaybeRegexp()) {
if (TString(obj->GetName()).Index(reg) < 0 ) continue;
} else if (! name.BeginsWith(patORpfx)) continue;
if (obj->InheritsFrom(TH1::Class()))
((TH1*)obj)->GetYaxis()->SetTitle(title);
if (obj->InheritsFrom(THStack::Class())) {
((THStack*)obj)->Draw();
((THStack*)obj)->GetYaxis()->SetTitle(title);
}
}
}
TLegend* legend(THStack* stack, Option_t* option = "lp", Bool_t addColor = kFALSE, Int_t token = -1,
Float_t xmin = 0.50, Float_t ymin = 0.51, Float_t xmax = 0.85, Float_t ymax = 0.92) {
if(! stack) return 0;
TLegend* leg = new TLegend(xmin, ymin, xmax, ymax);
TList* list = stack->GetHists();
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
//Hist color iterator
Int_t colorIt = 1;
while (obj = iter->Next()) {
if (! obj->InheritsFrom(TH1::Class())) continue;
if (addColor) {
hist::color(obj->GetName(), colorIt);
++colorIt;
}
if (token == -1)
leg->AddEntry(obj, obj->GetTitle(), option);
else {
TString name(obj->GetName());
TObjArray* a = name.Tokenize("_");
if (a->GetEntries() <= token)
leg->AddEntry(obj, obj->GetName(), option);
else
leg->AddEntry(obj, a->At(token)->GetName(), option);
}
}
return leg;
}
void normalize(const char* patORpfx) {
TRegexp reg(patORpfx, kFALSE);
TList* list = gDirectory->GetList() ;
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
while (obj = iter->Next()) {
if (! obj->InheritsFrom(TH1::Class())) continue;
TString name = obj->GetName();
if (TString(patORpfx).MaybeRegexp()) {
if (TString(obj->GetName()).Index(reg) < 0 ) continue;
} else if (! name.BeginsWith(patORpfx)) continue;
Double_t integral = 0;
if (obj->InheritsFrom(TH2::Class()))
integral = ((TH2*)obj)->Integral();
else
integral = ((TH1*)obj)->Integral();
if (integral) {
((TH1*)obj)->Sumw2();
((TH1*)obj)->Scale(1./integral);
}
}
}
void colors(TCanvas* canvas, Color_t color = 1) {
if(! canvas) return 0;
TList* list = canvas->GetListOfPrimitives();
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
//Hist color iterator
Int_t colorIt = color;
while (obj = iter->Next()) {
if (! obj->InheritsFrom(TH1::Class())) continue;
//yellow
if (colorIt == 5)
++colorIt;
hist::color(obj->GetName(), colorIt);
if (colorIt == 40)
colorIt = 1;
else
++colorIt;
}
}
///
/// Fills vector with floating pars names
///
void Utils::getParameters(const RooFitResult &result, std::vector<TString> &names){
RooArgList pars = result.floatParsFinal();
TIterator * it = pars.createIterator();
while(RooRealVar* p = (RooRealVar*) it->Next()){
names.push_back(TString(p->GetName()));
}
};
///
/// Perform a couple of consistency checks to make it easier
/// to find bugs:
/// - check if all observables end with '_obs'
/// - check if all predicted observables end with '_th'
/// - check if the 'observables' and 'theory' lists are correctly ordered
///
bool PDF_Abs::checkConsistency()
{
if ( m_isCrossCorPdf ) return true;
bool allOk = true;
// check if all observables end with '_obs'
TIterator* it = observables->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
TString pObsName = p->GetName();
pObsName.ReplaceAll(uniqueID,"");
if ( !pObsName.EndsWith("_obs") ){
cout << "PDF_Abs::checkConsistency() : " << name << " : observable " << p->GetName() << " doesn't end with '_obs'" << endl;
allOk = false;
}
}
// check if all predicted observables end with '_th'
delete it; it = theory->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
TString pThName = p->GetName();
pThName.ReplaceAll(uniqueID,"");
if ( !pThName.EndsWith("_th") ){
cout << "PDF_Abs::checkConsistency() : " << name << " : theory " << p->GetName() << " doesn't end with '_th'" << endl;
allOk = false;
}
}
// check if the 'observables' and 'theory' lists are correctly ordered
for ( int i=0; i<nObs; i++ ){
RooAbsArg* pTh = theory->at(i);
TString base = pTh->GetName();
base.ReplaceAll("_th","");
base.ReplaceAll(uniqueID,"");
TString pObsName = observables->at(i)->GetName();
pObsName.ReplaceAll(uniqueID,"");
if ( pObsName != base+"_obs"){
cout << "PDF_Abs::checkConsistency() : " << name << " : " << pTh->GetName() << " doesn't match its observable." << endl;
cout << " Expected '" << base+"_obs" << "'. Found '" << pObsName << "'." << endl;
cout << " Check ordering of the 'theory' and 'observables' lists!" << endl;
allOk = false;
}
}
return allOk;
}
///
/// Load a named parameter range for a list of parameters.
///
/// \param set - The list holding the parameters.
/// \param limitname - Name of the limit to set.
///
void Utils::setLimit(const RooAbsCollection* set, TString limitname)
{
RooMsgService::instance().setGlobalKillBelow(ERROR);
TIterator* it = set->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
p->setRange(p->getMin(limitname), p->getMax(limitname));
}
RooMsgService::instance().setGlobalKillBelow(INFO);
}
void drawsame(const char* canvasName, const char* patORpfx,Option_t* drawOption = "") {
// cout << "testing this method" << endl;
TRegexp reg(patORpfx, kFALSE);
TList* list = gDirectory->GetList() ;
// cout << "this bleeping directory has " << gDirectory->GetNkeys() << "things in it" << endl;
TIterator* iter = list->MakeIterator();
TObject* obj = 0;
TObject* canvas = 0;
Bool_t makeCanvas = false;
canvas = gROOT->GetListOfCanvases()->FindObject(canvasName);
//If canvas does not exist, remember to create it
// cout << "found our canvas" << endl;
if (! canvas) makeCanvas = true;
while (obj = iter->Next()) {
// cout << "We have an object" << endl;
if (! obj->InheritsFrom(TH1::Class())) continue;
TString name = obj->GetName();
// cout << "Testing object of name " << name << endl;
// if (TString(patORpfx).MaybeRegexp()) { THIS BASICALLY ADDS IT IF IT HAS A BLEEPING PULSE
if (TString(name).Contains(patORpfx)) {
// cout << "possible match" << endl;
if (TString(obj->GetName()).Index(reg) < 0 ) {
// cout << "not a match here" << endl;
continue;
}
else if (! name.BeginsWith(patORpfx)) {
// cout << "mismatched beginning" << endl;
continue;
}
if (makeCanvas) {
canvas = new TCanvas(canvasName, canvasName);
makeCanvas = false;
}
((TH1*)obj)->UseCurrentStyle();
((TCanvas*)canvas)->cd();
if (!((TCanvas*)canvas)->GetListOfPrimitives()->GetEntries()) {
// cout << "Drawing with non-same option" << endl;
((TH1*)obj)->Draw(Form("%s", drawOption));
}
else {
// cout << "Drawing with the same option" << endl;
((TH1*)obj)->Draw(Form("SAME%s", drawOption));
}
}
}
hist::colors((TCanvas*)canvas);
}
//
// Randomize all parameters of a set defined in a given
// workspace.
//
void Utils::randomizeParameters(RooWorkspace* w, TString setname)
{
TIterator* it = w->set(setname)->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
if ( p->isConstant() ) continue;
// sample from uniform distribution
p->randomize();
}
}
///
/// Set each parameter in setMe to the value found in values.
/// Do nothing if parameter is not found in values.
///
void Utils::setParameters(const RooAbsCollection* setMe, const RooAbsCollection* values)
{
TIterator* it = setMe->createIterator();
while ( RooRealVar* p = (RooRealVar*)it->Next() ){
RooRealVar *var = (RooRealVar*)values->find(p->GetName());
if ( var ) p->setVal(var->getVal());
}
delete it;
}
///
/// Make an Asimov toy: set all observables set to truth values.
/// The Asimov point needs to be loaded in the combiner before.
/// \param c - combiner which should be set to an asimov toy
///
void GammaComboEngine::setAsimovObservables(Combiner* c)
{
if ( !c->isCombined() ) {
cout << "GammaComboEngine::setAsimovObservables() : ERROR : Can't set an Asimov toy before "
"the combiner is combined. Call combine() first." << endl;
exit(1);
}
// set observables to asimov values in workspace
RooWorkspace* w = c->getWorkspace();
TIterator* itObs = c->getObservables()->createIterator();
while(RooRealVar* pObs = (RooRealVar*) itObs->Next()) {
// get theory name from the observable name
TString pThName = pObs->GetName();
pThName.ReplaceAll("obs","th");
// get the theory relation
RooAbsReal* th = w->function(pThName);
if ( th==0 ) {
cout << "GammaComboEngine::setAsimovObservables() : ERROR : theory relation not found in workspace: " << pThName << endl;
exit(1);
}
// set the observable to what the theory relation predicts
pObs->setVal(th->getVal());
}
delete itObs;
// write back the asimov values to the PDF object so that when
// the PDF is printed, the asimov values show up
for ( int i=0; i<c->getPdfs().size(); i++ ) {
PDF_Abs* pdf = c->getPdfs()[i];
pdf->setObservableSourceString("Asimov");
TIterator* itObs = pdf->getObservables()->createIterator();
while(RooRealVar* pObs = (RooRealVar*) itObs->Next()) {
RooAbsReal* obs = w->var(pObs->GetName());
if ( obs==0 ) {
cout << "GammaComboEngine::setAsimovObservables() : ERROR : observable not found in workspace: " << pObs->GetName() << endl;
exit(1);
}
pdf->setObservable(pObs->GetName(), obs->getVal());
}
delete itObs;
}
}
