CMSSW_4_4_3_patch1/src/RecoParticleFlow/PFClusterTools/interface/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_*/