ToolsOther.cxx

Go to the documentation of this file.

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

 
 
//
// includes
//
 
#include <SampleHandler/ToolsOther.h>
 
#include <RootCoreUtils/Assert.h>
#include <RootCoreUtils/hadd.h>
#include <SampleHandler/MetaNames.h>
#include <SampleHandler/MetaObject.h>
#include <SampleHandler/SampleHandler.h>
#include <SampleHandler/SampleLocal.h>
#include <TFile.h>
#include <TSystem.h>
#include <iostream>
#include <thread>
#include <chrono>
#include <random>
#include <stdexcept>
 
//
// method implementations
//
 
namespace SH
{
  std::unique_ptr<TFile>
  openFile (const std::string& name, const MetaObject& options)
  {
    const unsigned retries
      = options.castInteger (MetaNames::openRetries(),
                 MetaNames::openRetries_default());
    const double wait
      = options.castDouble (MetaNames::openRetriesWait(),
                MetaNames::openRetriesWait_default());
    if (wait < 0)
      throw std::runtime_error ("negative values not allowed for: " + MetaNames::openRetriesWait());
 
    std::unique_ptr<TFile> result;
 
    static std::atomic<unsigned> seed = 0;
    std::default_random_engine re (++seed);
    std::uniform_real_distribution wait_dist (wait/2, wait);
 
    for (unsigned tries = 0; tries <= retries; ++ tries)
    {
      if (tries > 0)
      {
    unsigned mywait = 1000 * wait_dist (re);
    std::cout << "open failed, waiting " << (mywait/1000.) << " seconds: "
          << name << std::endl;
    std::this_thread::sleep_for (std::chrono::milliseconds (mywait));
      }
      result.reset (TFile::Open (name.c_str(), "READ"));
      if (result != nullptr)
    return result;
    }
    throw std::runtime_error ("failed to open file: " + name);
  }
 
 
 
  std::unique_ptr<SampleLocal>
  mergeFiles (const Sample& sample, const std::string& location,
          bool overwrite)
  {
    auto result = std::make_unique<SampleLocal> (sample.name());
    result->add (location);
    *result->meta() = *sample.meta();
    if (overwrite || gSystem->AccessPathName (location.c_str()) != 0)
    {
      std::vector<std::string> files = sample.makeFileList();
      RCU::hadd (location, files);
    }
    return result;
  }
 
 
 
  SampleHandler
  mergeFiles (const SampleHandler& sh, const std::string& location,
          bool overwrite)
  {
    SampleHandler result;
    for (auto sample : sh)
    {
      std::string sublocation = location + sample->name() + ".root";
      result.add (mergeFiles (*sample, sublocation, overwrite));
    }
    return result;
  }
}