dump_workspace.cxx

Go to the documentation of this file.

#include <iostream>
#include <iomanip>

#include "TFile.h"
#include "TIterator.h"

#include "RooArgSet.h"
#include "RooRealVar.h"

#include "RooStats/ModelConfig.h"

using namespace std;
using namespace RooStats;

int main(int argc, char *argv[]){
  for(int iarg = 1; iarg < argc; ++iarg){
    TFile file(argv[iarg], "read");
    if(!file.IsOpen()) continue;
    RooWorkspace *w = static_cast<RooWorkspace*>(file.Get("w"));
    if(w == nullptr) continue;
    w->Print();

    cout << fixed << setprecision(2);

    cout << "Constants:" << endl;
    const RooArgSet &vars = w->allVars();
    TIterator *iter_ptr = vars.createIterator();
    for(; iter_ptr != nullptr && *(*iter_ptr) != nullptr; iter_ptr->Next()){
      RooAbsArg *var = static_cast<RooAbsArg*>(*(*iter_ptr));
      if(var == nullptr) continue;
      if(!var->isConstant()) continue;
      var->Print();
    }

    cout << endl << "Fundamental variables:" << endl;
    iter_ptr->Reset();
    for(; iter_ptr != nullptr && *(*iter_ptr) != nullptr; iter_ptr->Next()){
      RooAbsArg *var = static_cast<RooAbsArg*>(*(*iter_ptr));
      if(var == nullptr) continue;
      if(var->isConstant()) continue;
      var->Print();
    }

    cout << endl << "Yields: " << endl;
    cout
      << ' ' << setw(64) << "Name"
      << ' ' << setw(12) << "Background"
      << ' ' << setw(12) << "Signal"
      << ' ' << setw(12) << "Total"
      << ' ' << setw(12) << "Observed"
      << endl;

    iter_ptr->Reset();
    for(; iter_ptr != nullptr && *(*iter_ptr) != nullptr; iter_ptr->Next()){
      RooAbsArg *var = static_cast<RooAbsArg*>(*(*iter_ptr));
      if(var == nullptr) continue;
      string nobs_name = var->GetName();
      if(nobs_name.substr(0, 9) != "nobs_BLK_") continue;
      string nbkg_name = nobs_name;
      string nsig_name = nobs_name;
      nbkg_name.replace(1, 3, "bkg");
      nsig_name.replace(1, 3, "sig");
      RooRealVar *nobs_arg = static_cast<RooRealVar*>(w->arg(nobs_name.c_str()));
      RooRealVar *nbkg_arg = static_cast<RooRealVar*>(w->arg(nbkg_name.c_str()));
      RooRealVar *nsig_arg = static_cast<RooRealVar*>(w->arg(nsig_name.c_str()));
      if(nobs_arg==nullptr || nbkg_arg==nullptr || nsig_arg==nullptr) continue;
      double nobs = nobs_arg->getVal();
      double nbkg = nbkg_arg->getVal();
      double nsig = nsig_arg->getVal();
      cout
    << ' ' << setw(64) << nobs_name.substr(5)
    << ' ' << setw(12) << nbkg
    << ' ' << setw(12) << nsig
    << ' ' << setw(12) << nbkg+nsig
    << ' ' << setw(12) << nobs
    << endl;
    }
    
  }
}