ToolsMeta.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/

 
 
//
// includes
//
 
#include <SampleHandler/ToolsMeta.h>
 
#include <RootCoreUtils/Assert.h>
#include <SampleHandler/MetaFields.h>
#include <SampleHandler/MetaObject.h>
#include <SampleHandler/SampleHandler.h>
#include <SampleHandler/Sample.h>
#include <TSystem.h>
#include <fstream>
#include <sstream>
#include <stdexcept>
 
//
// method implementations
//
 
namespace SH
{
  void readSusyMeta (const SampleHandler& sh, const std::string& inputFile)
  {
    TString myfile = inputFile;
    gSystem->ExpandPathName (myfile);
    std::ifstream file (myfile.Data());
    std::string line;
 
    if (!file)
      throw std::runtime_error ("failed to read file: " + inputFile);
 
    while (std::getline (file, line))
    {
      if (line[0] >= '0' && line[0] <= '9')
      {
    std::istringstream str (line);
    std::string DSID, name;
    float crossSection = 0, kFactor = 0, filterEfficiency = 0,
      xsUncertainty = 0;
 
    if (!(str >> DSID >> name >> crossSection >> kFactor >> filterEfficiency
          >> xsUncertainty))
      throw std::runtime_error ("readSusyMeta: failed to parse line: " + line);
 
    std::string mydsid = "." + DSID + ".";
    for (SampleHandler::iterator sample = sh.begin(),
           end = sh.end(); sample != end; ++ sample)
    {
      if ((*sample)->name().find (mydsid) != std::string::npos)
      {
            double oldCrossSection
              = (*sample)->meta()->castDouble (MetaFields::crossSection);
        (*sample)->meta()->setDouble (MetaFields::crossSection, oldCrossSection + crossSection);
        (*sample)->meta()->setDouble (MetaFields::kfactor, kFactor);
        (*sample)->meta()->setDouble (MetaFields::filterEfficiency, filterEfficiency);
        (*sample)->meta()->setDouble (MetaFields::crossSectionRelUncertainty, xsUncertainty);
      }
    }
      }
    }
  }
 
  void readSusyMetaDir (const SampleHandler& sh, const std::string& inputDir)
  {
    TString mydir = inputDir;
    gSystem->ExpandPathName (mydir);
    void *dirp = 0;
 
    try
    {
      dirp = gSystem->OpenDirectory (mydir.Data());
      const char *file = 0;
      while ((file = gSystem->GetDirEntry (dirp)))
      {
    std::string myfile = inputDir + "/" + file;
    if (myfile.size() > 4 && myfile.substr (myfile.size()-4) == ".txt")
      readSusyMeta (sh, myfile);
      }
      gSystem->FreeDirectory (dirp);
    } catch (...)
    {
      gSystem->FreeDirectory (dirp);
      throw;
    }
  }
}