FileUtils.h

Go to the documentation of this file.

#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 {
 
std::unique_ptr<TFile> GetOutput (std::filesystem::path output,
                                  const char * name);
 
void StandardInit ();
 
std::vector<std::filesystem::path> GetROOTfiles
        (std::vector<std::filesystem::path> inputs); 
 
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 char * 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");
        // TODO: what if the file had already been opened?? will it be closed prematurely?
        unique_ptr<THX> h(fIn->Get<THX>(name));
        if (!h) {
            namespace DE = Darwin::Exceptions;
            BOOST_THROW_EXCEPTION(DE::BadInput(Form("`%s` cannot be found in (one of) the "
                                                    " file(s).", name), fIn));
        }
        if (sum)
            sum->Add(h.get());
        else {
            sum = move(h);
            sum->SetDirectory(nullptr);
        }
    }
    return sum;
}
 
std::string GetFirstTreeLocation
                (const std::filesystem::path& input); 
 
std::unique_ptr<TTree> NewTree (const std::unique_ptr<TChain>& chain, 
                                const bool makeFriend, 
                                const char * thisFunc); 
 
} // namespace Darwin::Tools
 
#define DT_GetOutput(output) Darwin::Tools::GetOutput(output, __func__)
#define DT_NewTree(input,makeFriend) Darwin::Tools::NewTree(input, makeFriend, __func__)