toolbox.h

Go to the documentation of this file.

#pragma once
 
#include <cassert>
#include <vector>
 
#include <TDirectory.h>
#include <TKey.h>
#include <TString.h>
 
#include "Math/VectorUtil.h"
 
#include "Core/Objects/interface/PhysicsObject.h"
 
#include <boost/exception/all.hpp>
 
namespace DAS {
 
inline bool pt_sort (const PhysicsObject& j1, const PhysicsObject& j2)
{
    return j1.CorrPt() > j2.CorrPt();
}
 
template<typename TTreePtr> bool branchExists (const TTreePtr& tree, TString brName)
{
    auto brList = tree->GetListOfBranches();
    bool brFound = false;
    for (auto it = brList->begin(); it != brList->end(); ++it) {
        TString name = (*it)->GetName();
        if (name == brName) brFound = true;
    }
    return brFound;
}
 
template<typename T> std::vector<T*> GetObjects (TDirectory * dir)
{
    if (!dir)
        BOOST_THROW_EXCEPTION( std::runtime_error("Trying to access inexistent `TDirectory`") );
    std::vector<T*> objects;
    for (const auto&& element: *(dir->GetListOfKeys())) {
        auto key = dynamic_cast<TKey*>(element);
        if (auto obj = dynamic_cast<T*>(key->ReadObj()); obj)
            objects.push_back(obj);
    }
    return objects;
}
 
inline TDirectory * GetDirectory (TDirectory * dir, const std::vector<const char *>& names)
{
    for (const char * name: names) {
        TDirectory * subdir = dir->GetDirectory(name);
        if (!subdir)
            BOOST_THROW_EXCEPTION( std::invalid_argument(
                    Form("`%s` could not be found in `%s`", name, dir->GetName())) );
        dir = subdir;
    }
    return dir;
}
 
using ROOT::Math::VectorUtil::DeltaPhi;
using ROOT::Math::VectorUtil::DeltaR;
 
} // end of DAS namespace