hist2workspace.cxx

Go to the documentation of this file.

// @(#)root/roostats:$Id$
// Author: Kyle Cranmer, George Lewis
/*************************************************************************
 * Copyright (C) 1995-2008, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#include <string>
#include <exception>
#include <vector>
 
#include <TROOT.h>
 
#include "RooStats/HistFactory/ConfigParser.h"
#include "RooStats/HistFactory/MakeModelAndMeasurementsFast.h"
#include "HFMsgService.h"
 
// When updating the docs for the hist2workspace executable, please recreate
// the Python file with the get_argparse() function (code below), and run the
// following argparse2help.py invocations:
//
// 1. To generate the string in the kCommandLineOptionsHelp variable
//
//      python cmake/scripts/argparse2help.py hist2workspace-argparse.py hist2workspaceCommandLineOptionsHelp.h
//
// 2. Recreate the man file that should be put in roofit/man/man1/:
//
//      python cmake/scripts/argparse2help.py hist2workspace-argparse.py hist2workspace.1
//
// The content of hist2workspace-argparse.py is:
//
// ```Python
// import argparse
//
//
// def get_argparse():
//     DESCRIPTION = "hist2workspace is a utility to create RooFit/RooStats workspace from histograms"
//     parser = argparse.ArgumentParser(prog="hist2workspace", description=DESCRIPTION)
//     parser.add_argument("-v", help="switch HistFactory message stream to INFO level", action="store_true")
//     parser.add_argument("-vv", help="switch HistFactory message stream to DEBUG level", action="store_true")
//     parser.add_argument(
//         "-disable_binned_fit_optimization",
//         help="disable the binned fit optimization used in HistFactory since ROOT 6.28",
//         action="store_true",
//     )
//     return parser
// ```
 
constexpr static const char kCommandLineOptionsHelp[] = R"RAW(
usage: hist2workspace [-h] [-v] [-vv] [-disable_binned_fit_optimization]
 
hist2workspace is a utility to create RooFit/RooStats workspace from histograms
 
OPTIONS:
  -h, --help                                  show this help message and exit
  -v                                          switch HistFactory message stream to INFO level
  -vv                                         switch HistFactory message stream to DEBUG level
  -disable_binned_fit_optimization            disable the binned fit optimization used in HistFactory since ROOT 6.28
)RAW";
 
 
namespace RooStats {
  namespace HistFactory {
    void fastDriver(std::string const& input, HistoToWorkspaceFactoryFast::Configuration const& cfg) {
 
      // Create the initial list of measurements and channels
      std::vector< HistFactory::Measurement > measurement_list;
      // std::vector< HistFactory::Channel >     channel_list;
 
      // Fill them using the XML parser
      HistFactory::ConfigParser xmlParser;
      measurement_list = xmlParser.GetMeasurementsFromXML( input );
 
      // At this point, we have all the information we need
      // from the xml files.
 
      // We will make the measurements 1-by-1
      for(unsigned int i = 0; i < measurement_list.size(); ++i) {
   HistFactory::Measurement measurement = measurement_list.at(i);
   measurement.CollectHistograms();
   MakeModelAndMeasurementFast(measurement, cfg);
      }
 
      return;
 
    }
  } // namespace RooStats
} // namespace HistFactory
 
/**
 * \ingroup HistFactory
 * main function of the hist2workspace executable.
 * It creates RooFit models from an xml config and files with histograms.
 * See MakeModelAndMeasurementFast(), for further instructions.
 * \param[in] argc number of CLI arguments
 * \param[in] argv pointer to arguments
 *
 * -h Help
 * -v Switch HistFactory message stream to INFO level.
 * -vv Switch HistFactory message stream to DEBUG level.
 * -disable_binned_fit_optimization Disable the binned fit optimization used in HistFactory since ROOT 6.28.
 */
int main(int argc, char** argv) {
 
  if( !(argc > 1) ) {
    std::cerr << "need input file" << std::endl;
    exit(1);
  }
 
  //Switch off ROOT histogram memory management
  gROOT->SetMustClean(false);
  TDirectory::AddDirectory(false);
  cxcoutIHF << "hist2workspace is less verbose now. Use -v and -vv for more details." << std::endl;
  RooMsgService::instance().getStream(1).minLevel = RooFit::PROGRESS;
  RooMsgService::instance().getStream(2).minLevel = RooFit::PROGRESS;
  RooMsgService::instance().getStream(2).addTopic(RooFit::HistFactory);
 
  std::string driverArg;
  RooStats::HistFactory::HistoToWorkspaceFactoryFast::Configuration cfg{};
 
  for (int i=1; i < argc; ++i) {
    std::string input = argv[i];
 
    if (input == "-h" || input == "--help"){
      fprintf(stderr, kCommandLineOptionsHelp);
      return 0;
    }
 
    if (input == "-v") {
      RooMsgService::instance().getStream(1).minLevel = RooFit::INFO;
      RooMsgService::instance().getStream(2).minLevel = RooFit::INFO;
      continue;
    }
 
    if (input == "-vv") {
      RooMsgService::instance().getStream(1).minLevel = RooFit::INFO;
      RooMsgService::instance().getStream(2).minLevel = RooFit::DEBUG;
      continue;
    }
 
    if(input == "-disable_binned_fit_optimization") {
      cfg.binnedFitOptimization = false;
      continue;
    }
 
    driverArg = argv[i];
  }
 
  RooStats::HistFactory::fastDriver(driverArg, cfg);
 
  return 0;
}