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 (auto *sample : sh)
    {
      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());
      // rationale: OpenDirectory returns null for a missing/mistyped
      //   directory, and GetDirEntry/FreeDirectory tolerate a null
      //   handle, so this would otherwise silently read nothing; throw
      //   for consistency with readSusyMeta on a missing file.
      if (dirp == nullptr)
        throw std::runtime_error ("could not open directory: " + inputDir);
      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;
    }
  }
}