FileUtils.h

Go to the documentation of this file.

// SPDX-License-Identifier: GPLv3-or-later
//
// SPDX-FileCopyrightText: Patrick L.S. Connor <patrick.connor@desy.de>
// SPDX-FileCopyrightText: Louis Moureaux <louis.moureaux@cern.ch>
 
#pragma once
 
#include "exceptions.h"
 
#include <TChain.h>
#include <TFile.h>
#include <TH1.h>
#include <TTree.h>
 
#include <filesystem>
#include <string>
#include <memory>
#include <vector>
 
namespace Darwin::Tools {
 
[[deprecated("Use Darwin::Tools::Flow.")]]
std::unique_ptr<TFile> GetOutput (std::filesystem::path output,
                                  const std::string& name);
 
void StandardInit ();
 
std::vector<std::filesystem::path> GetROOTfiles
        (std::vector<std::filesystem::path> inputs); 
 
[[deprecated("Use Darwin::Tools::Flow::GetInputTree().")]]
std::unique_ptr<TChain> GetChain
                (std::vector<std::filesystem::path> inputs, 
                 const char * name = "events"); 
 
template<typename THX = TH1>
inline std::unique_ptr<THX> GetHist
                (std::vector<std::filesystem::path> inputs, 
                 const std::string& name = "h") 
{
    using namespace std;
    namespace fs = filesystem;
 
    unique_ptr<THX> sum;
    inputs = GetROOTfiles(inputs);
    for (auto const& input: inputs) {
        auto fIn = make_unique<TFile>(input.c_str(), "READ");
        unique_ptr<THX> h(fIn->Get<THX>(name.c_str()));
        if (!h) {
            namespace DE = Darwin::Exceptions;
            BOOST_THROW_EXCEPTION(DE::BadInput(Form("`%s` cannot be found in (one of) the "
                                                    " file(s).", name.c_str()), fIn));
        }
        if (sum)
            sum->Add(h.get());
        else {
            sum = std::move(h);
            sum->SetDirectory(nullptr);
        }
    }
    return sum;
}
 
std::string GetFirstTreeLocation
                (const std::filesystem::path& input); 
 
std::shared_ptr<TFile> GetOutputFile (const std::filesystem::path&, 
      const std::source_location = std::source_location::current()); 
 
[[deprecated("Use Darwin::Tools::Flow::GetOutputTree().")]]
std::unique_ptr<TTree> NewTree (const std::unique_ptr<TChain>& chain, 
                                const std::string& thisFunc); 
 
} // namespace Darwin::Tools
 
#define DT_GetOutput(output) Darwin::Tools::GetOutput(output, __func__)
#define DT_NewTree(input,makeFriend) Darwin::Tools::NewTree(input, __func__)