/data/refman/pasoursint/CMSSW_5_3_10_patch1/src/RecoParticleFlow/PFClusterTools/interface/CalibratableTest.h

Go to the documentation of this file.

#ifndef CALIBRATABLETEST_H_
#define CALIBRATABLETEST_H_

// system include files
#include <memory>
#include <vector>
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "DataFormats/ParticleFlowReco/interface/Calibratable.h"

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFSimParticleFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHitFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"

#include <TTree.h>


class CalibratableTest : public edm::EDAnalyzer {
public:
        explicit CalibratableTest(const edm::ParameterSet&);
        ~CalibratableTest();

        /*
         * Returns the collection in the event matching the Handle.
         */
        template<class T> void getCollection(edm::Handle<T>& c,
                        const edm::InputTag& tag, const edm::Event& event) const;

private:
        /* 
         * The usual EDAnalyzer methods
         */
        virtual void beginJob();
        virtual void analyze(const edm::Event&, const edm::EventSetup&);
        virtual void endJob();

        /*
         * Called for each particle in the event to fill the tree and
         * reset calib_.
         */
        virtual void fillTreeAndReset();

        /*
         * Finds the main pions generated by FastSim
         */
        std::vector<unsigned> findPrimarySimParticles(
                        const std::vector<reco::PFSimParticle>& sims);

        /*
         * Finds PFCandidates within some deltaR of the supplied sim particle.
         */
        std::vector<unsigned> findCandidatesInDeltaR(const reco::PFSimParticle& pft,
                        const std::vector<reco::PFCandidate>& cands, const double& deltaR);

        /*
         * Fills calib_ with PFCandidate information.
         */
        void extractCandidate(const reco::PFCandidate& cand);

        /*
         * Computes delta R between two objects.
         */
        double deltaR(const double& eta1, const double& eta2, const double& phi1,
                        const double& phi2);

        //Calibratable tree
        TTree* tree_;

        //This analyzer produces a file with a tree so we need,
        edm::Service<TFileService> fileservice_;

        //Debug?
        int debug_;

        //True if this event was generally successful: 
        //if no sim particles are found, this will be false
        bool thisEventPasses_;
        //True if this particle should be written to the tree
        //If no PFCandidates are found within a delta R cone of the sim particle,
        //this will be false.
        bool thisParticlePasses_;

        //The Calibratable object
        pftools::Calibratable* calib_;

        //Increment nWrites_ for every record written to the tree
        //Incremement nFails_ for every event/particle not written to the tree (thisEventPasses_ == false)
        unsigned nParticleWrites_, nParticleFails_;
        unsigned nEventWrites_, nEventFails_;
        
        //Size of cone around sim particle to look for PFCandidates
        double deltaRCandToSim_;

        //Collection input tags
        edm::InputTag inputTagCandidates_;
        edm::InputTag inputTagSimParticles_;
        edm::InputTag inputTagClustersEcal_;
        edm::InputTag inputTagClustersHcal_;

        //collections
        edm::Handle<reco::PFCandidateCollection>* pfCandidates_;
        edm::Handle<reco::PFSimParticleCollection>* simParticles_;
        edm::Handle<reco::PFClusterCollection>* clustersEcal_;
        edm::Handle<reco::PFClusterCollection>* clustersHcal_;

};

template<class T> void CalibratableTest::getCollection(edm::Handle<T>& c,
                const edm::InputTag& tag, const edm::Event& event) const {

        try {
                event.getByLabel(tag, c);
                if(!c.isValid()) {
                        std::cout << "Warning! Collection for label " << tag << " is not valid!" << std::endl;
                }
        }
        catch (cms::Exception& err) {
                std::cout << "Couldn't get collection\n";
                std::ostringstream err;
                //LogError("Error getting collection!") << err;
        }
}

#endif /*CALIBRATABLETEST_H_*/