TreeUtility.h

Go to the documentation of this file.

#ifndef TREEUTILITY_HH_
#define TREEUTILITY_HH_
#include "RecoParticleFlow/PFClusterTools/interface/DetectorElement.h"
#include "RecoParticleFlow/PFClusterTools/interface/ParticleDeposit.h"
#include "DataFormats/ParticleFlowReco/interface/Calibratable.h"
#include "DataFormats/ParticleFlowReco/interface/CalibrationProvenance.h"

#include <boost/shared_ptr.hpp>
#include <TFile.h>
#include <TChain.h>
#include <vector>
#include <string>
#include <map>

namespace pftools {
typedef boost::shared_ptr<Calibratable> CalibratablePtr;
class TreeUtility {
public:

    TreeUtility();
    virtual ~TreeUtility();

    unsigned getParticleDepositsDirectly(TChain& sourceChain,
            std::vector<ParticleDepositPtr>& toBeFilled,
            CalibrationTarget target, DetectorElementPtr offset,
            DetectorElementPtr ecal, DetectorElementPtr hcal, bool includeOffset = false);

    unsigned getCalibratablesFromRootFile(TChain& tree,
            std::vector<Calibratable>& toBeFilled);

    unsigned convertCalibratablesToParticleDeposits(
            const std::vector<Calibratable>& input,
            std::vector<ParticleDepositPtr>& toBeFilled,
            CalibrationTarget target, DetectorElementPtr offset,
            DetectorElementPtr ecal, DetectorElementPtr hcal, bool includeOffset = false);

    void dumpCaloDataToCSV(TChain& chain, std::string csvFilename, double range, bool gaus = false);


private:
    std::map<std::string, unsigned> vetos_;

};
}
#endif /*TREEUTILITY_HH_*/