RooStudyManager.cxx

Go to the documentation of this file.

/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 * @(#)root/roofitcore:$Id$
 * Authors:                                                                  *
 *   WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu       *
 *   DK, David Kirkby,    UC Irvine,         dkirkby@uci.edu                 *
 *                                                                           *
 * Copyright (c) 2000-2005, Regents of the University of California          *
 *                          and Stanford University. All rights reserved.    *
 *                                                                           *
 * Redistribution and use in source and binary forms,                        *
 * with or without modification, are permitted according to the terms        *
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)             *
 *****************************************************************************/
 
/**
\file RooStudyManager.cxx
\class RooStudyManager
\ingroup Roofitcore
 
Utility class to manage studies that consist of
repeated applications of generate-and-fit operations on a workspace
 
**/
 
 
#include "Riostream.h"
 
#include "RooStudyManager.h"
#include "RooWorkspace.h"
#include "RooAbsStudy.h"
#include "RooDataSet.h"
#include "RooMsgService.h"
#include "RooStudyPackage.h"
#include "TFile.h"
#include "TObjString.h"
#include "TRegexp.h"
#include "TKey.h"
#include <string>
#include "TROOT.h"
#include "TSystem.h"
 
////////////////////////////////////////////////////////////////////////////////
 
RooStudyManager::RooStudyManager(RooWorkspace &w) : _pkg(new RooStudyPackage(w)) {}
 
////////////////////////////////////////////////////////////////////////////////
 
RooStudyManager::RooStudyManager(RooWorkspace &w, RooAbsStudy &study) : _pkg(new RooStudyPackage(w))
{
 
  _pkg->addStudy(study) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
 
RooStudyManager::RooStudyManager(const char* studyPackFileName)
{
   std::string pwd = gDirectory->GetName();
   std::unique_ptr<TFile> f{TFile::Open(studyPackFileName, "READ")};
   _pkg = dynamic_cast<RooStudyPackage *>(f->Get("studypack"));
   gDirectory->cd((pwd + ":").c_str());
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooStudyManager::addStudy(RooAbsStudy& study)
{
  _pkg->addStudy(study) ;
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooStudyManager::run(Int_t nExperiments)
{
  _pkg->driver(nExperiments) ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooStudyManager::prepareBatchInput(const char* studyName, Int_t nExpPerJob, bool unifiedInput=false)
{
  TFile f(Form("study_data_%s.root",studyName),"RECREATE") ;
  _pkg->Write("studypack") ;
  f.Close() ;
 
  if (unifiedInput) {
 
    // Write header of driver script
    std::ofstream bdr(Form("study_driver_%s.sh", studyName));
    bdr << "#!/bin/sh" << std::endl
        << Form("if [ ! -f study_data_%s.root ] ; then",studyName) << std::endl
        << "uudecode <<EOR" << std::endl ;
    bdr.close() ;
 
    // Write uuencoded ROOT file (base64) in driver script
    gSystem->Exec(Form("cat study_data_%s.root | uuencode -m study_data_%s.root >> study_driver_%s.sh",studyName,studyName,studyName)) ;
 
    // Write remainder of driver script
    std::ofstream bdr2(Form("study_driver_%s.sh", studyName), std::ios::app);
    bdr2 << "EOR" << std::endl
    << "fi" << std::endl
    << "root -l -b <<EOR" << std::endl
    << Form("RooStudyPackage::processFile(\"%s\",%d) ;",studyName,nExpPerJob) << std::endl
    << ".q" << std::endl
    << "EOR" << std::endl ;
    // Remove binary input file
    gSystem->Unlink(Form("study_data_%s.root",studyName)) ;
 
    coutI(DataHandling) << "RooStudyManager::prepareBatchInput batch driver file is '" << Form("study_driver_%s.sh",studyName) << "," << std::endl
         << "     input data files is embedded in driver script" << std::endl ;
 
  } else {
 
     std::ofstream bdr(Form("study_driver_%s.sh", studyName));
     bdr << "#!/bin/sh" << std::endl
         << "root -l -b <<EOR" << std::endl
         << Form("RooStudyPackage::processFile(\"%s\",%d) ;", studyName, nExpPerJob) << std::endl
         << ".q" << std::endl
         << "EOR" << std::endl;
 
     coutI(DataHandling) << "RooStudyManager::prepareBatchInput batch driver file is '"
                         << Form("study_driver_%s.sh", studyName) << "," << std::endl
                         << "     input data file is " << Form("study_data_%s.root", studyName) << std::endl;
 
  }
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooStudyManager::processBatchOutput(const char* filePat)
{
   std::list<std::string> flist;
   expandWildCardSpec(filePat, flist);
 
   TList olist;
 
   for (auto iter = flist.begin(); iter != flist.end(); ++iter) {
      coutP(DataHandling) << "RooStudyManager::processBatchOutput() now reading file " << *iter << std::endl;
      TFile f(iter->c_str());
 
      for (auto *key : static_range_cast<TKey *>(*f.GetListOfKeys())) {
         TObject *obj = f.Get(key->GetName());
         olist.Add(obj->Clone(obj->GetName()));
      }
   }
  aggregateData(&olist) ;
  olist.Delete() ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooStudyManager::aggregateData(TList *olist)
{
   for (auto study : _pkg->studies()) {
      study->aggregateSummaryOutput(olist);
   }
}
 
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// case with one single file
 
void RooStudyManager::expandWildCardSpec(const char *name, std::list<std::string> &result)
{
  if (!TString(name).MaybeWildcard()) {
    result.push_back(name) ;
    return ;
  }
 
   // wildcarding used in name
   TString basename(name);
 
   Int_t dotslashpos = -1;
   {
      Int_t next_dot = basename.Index(".root");
      while(next_dot>=0) {
         dotslashpos = next_dot;
         next_dot = basename.Index(".root",dotslashpos+1);
      }
      if (basename[dotslashpos+5]!='/') {
         // We found the 'last' .root in the name and it is not followed by
         // a '/', so the tree name is _not_ specified in the name.
         dotslashpos = -1;
      }
   }
   //Int_t dotslashpos = basename.Index(".root/");
   TString behind_dot_root;
   if (dotslashpos>=0) {
      // Copy the tree name specification
      behind_dot_root = basename(dotslashpos+6,basename.Length()-dotslashpos+6);
      // and remove it from basename
      basename.Remove(dotslashpos+5);
   }
 
   Int_t slashpos = basename.Last('/');
   TString directory;
   if (slashpos>=0) {
      directory = basename(0,slashpos); // Copy the directory name
      basename.Remove(0,slashpos+1);      // and remove it from basename
   } else {
      directory = gSystem->UnixPathName(gSystem->WorkingDirectory());
   }
 
   TString expand_directory = directory;
   gSystem->ExpandPathName(expand_directory);
   void *dir = gSystem->OpenDirectory(expand_directory.Data());
 
   if (dir) {
      //create a TList to store the file names (not yet sorted)
      TList l;
      TRegexp re(basename,true);
      const char *file;
      while ((file = gSystem->GetDirEntry(dir))) {
         if (!strcmp(file,".") || !strcmp(file,"..")) continue;
         TString s = file;
         if ( (basename!=file) && s.Index(re) == kNPOS) continue;
         l.Add(new TObjString(file));
      }
      gSystem->FreeDirectory(dir);
      //sort the files in alphanumeric order
      l.Sort();
      TIter next(&l);
      TObjString *obj;
      while ((obj = static_cast<TObjString*>(next()))) {
         file = obj->GetName();
         if (behind_dot_root.Length() != 0) {
            result.push_back(Form("%s/%s/%s",directory.Data(),file,behind_dot_root.Data())) ;
         } else {
            result.push_back(Form("%s/%s", directory.Data(), file));
         }
      }
      l.Delete();
   }
}