MPIAnalyzer/src/MPIntuple.cc

// Original Author:  "Nathaniel Odell"
// Secondary Author: Steven Won
// With contributions from: Andrey Pozdnyakov
//         Created:  Thurs April 22  2010
// $Id: MPIntuple.cc,v 1.35 2011/04/25 20:46:39 naodell Exp $

// system include files
#include <memory>
#include <string>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "DataFormats/GeometryVector/interface/LocalVector.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/Math/interface/deltaPhi.h"

// Libraries for objects
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/JetReco/interface/GenJet.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/METReco/interface/PFMET.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "DataFormats/METReco/interface/PFMETFwd.h"
#include "DataFormats/METReco/interface/MET.h"
#include "DataFormats/METReco/interface/METCollection.h"
#include "DataFormats/METReco/interface/METFwd.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/MuonReco/interface/MuonSelectors.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BTauReco/interface/JetTag.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/Luminosity/interface/LumiSummary.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenRunInfoProduct.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"

//#include "RecoVertex/PrimaryVertexProducer/interface/VertexHigherPtSquared.h"

// JEC
#include "JetMETCorrections/Objects/interface/JetCorrector.h"
#include "JetMETCorrections/Objects/interface/JetCorrectionsRecord.h"
#include "CondFormats/JetMETObjects/interface/JetCorrectionUncertainty.h"
#include "CondFormats/JetMETObjects/interface/JetCorrectorParameters.h"

// ntuple storage classes
#include "TCJet.h"
#include "TCMET.h"
#include "TCPrimaryVtx.h"
#include "TCGenJet.h"
#include "TCMuon.h"
#include "TCElectron.h"

// Need for HLT trigger info:
#include "FWCore/Common/interface/TriggerNames.h"
#include "DataFormats/Common/interface/TriggerResults.h"
#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"

//Root  stuff
#include "TROOT.h"
#include "TH1.h"
#include "TH2.h"
#include "TProfile.h"
#include "TFile.h"
#include "TTree.h"
#include "TString.h"
#include "TObject.h"
#include "TObjArray.h"
#include "TClonesArray.h"
#include "TRefArray.h"
#include "TLorentzVector.h"
#include "TVector3.h"

//MC stuff
//#include "HepMC/GenEvent"

using namespace edm;
using namespace std;
using namespace reco;

//
// class declaration
//

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


        private:
                virtual void beginJob() ;
                virtual void beginRun(const edm::Run&, const edm::EventSetup&) ;
                virtual void analyze(const edm::Event&, const edm::EventSetup&);
                virtual void endLuminosityBlock(const edm::LuminosityBlock&,const edm::EventSetup&);
                virtual void endRun(const edm::Run&, const edm::Event&, const edm::EventSetup&);
                virtual void endJob() ;

                bool genParticleMatch(const HepMC::GenParticle& genParticle, const reco::Candidate& motherParticle);
                bool triggerDecision(edm::Handle<edm::TriggerResults>& hltR, int iTrigger);
                float sumPtSquared(const Vertex& v);

                // ----------member data ---------------------------


                int eventNumber, runNumber, lumiSection, bunchCross;
                float ptHat, qScale, crossSection, evtWeight;
                float lumiDeadCount, lumiLiveFrac, intDelLumi;

                TTree* sTree;
                TFile* ntupleFile;
                edm::InputTag recoJetTag_;
                edm::InputTag recoMETTag_;
                edm::InputTag genJetTag_;
                edm::InputTag muonTag_;
                edm::InputTag electronTag_;
                edm::InputTag primaryVtxTag_;
                edm::InputTag triggerResultsTag_;
                edm::InputTag electronIDMap_;
                bool doGenJets_;
                bool doPFJets_;
                bool triggerHLT_;
                bool isRealData;
                int nJets_;
                string rootfilename;

                //Physics object containers
                TClonesArray* recoJets;
                TClonesArray* recoMET;
                TClonesArray* genJets;
                TClonesArray* primaryVtx;
                TClonesArray* recoMuons;
                TClonesArray* recoElectrons;

                //GenParticles
                TClonesArray* hardPartonP4;
      int partonPdgId[4];

                //Triggers
                string hlTriggerResults_, hltProcess_, triggerName_;
                TriggerNames triggerNames;
                HLTConfigProvider hltConfig_;
                vector<string>  hlNames;
                vector<string>  mpiTriggers_;
                unsigned int triggerStatus;
                unsigned int hltPrescale[32];


                //Histograms
                TH1D * h1_ptHat;

                TH1D * h1_fracAssociatedTracks;
                TH1D * h1_meanJetTrackZ;
                TH1D * h1_trackDxy;
                TH1D * h1_jetVertexZ;
                TH1D * h1_associatedSumPt;
                TH1D * h1_associatedVertexIndex;
                TH2F * h2_nAssTracksVsJetPt;
                TProfile * p1_nVtcs;

                TH1D * h1_partonJetdR;
                TH1D * h1_partonJet2dR;
                TH1D * h1_partonJetdpT;
                TH1D * h1_partonJet2dpT;
                TH1D * h1_partonJetMatch;
                TH1D * h1_partonJetMatch_dR5;
                TH1D * h1_partonJetMatch_dR9;

};

MPIntuple::MPIntuple(const edm::ParameterSet& iConfig)
{
        recoJetTag_       = iConfig.getUntrackedParameter<edm::InputTag>("RecoJetTag");
        recoMETTag_       = iConfig.getUntrackedParameter<edm::InputTag>("RecoMETTag");
        muonTag_          = iConfig.getUntrackedParameter<edm::InputTag>("MuonTag");
        electronTag_      = iConfig.getUntrackedParameter<edm::InputTag>("ElectronTag");
        genJetTag_        = iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag");
        primaryVtxTag_    = iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag");
        electronIDMap_    = iConfig.getParameter<edm::InputTag>("electronIDMap");
        nJets_            = iConfig.getUntrackedParameter<int>("nJets");
        doGenJets_        = iConfig.getUntrackedParameter<bool>("doGenJets");
        doPFJets_         = iConfig.getUntrackedParameter<bool>("doPFJets");
        triggerHLT_       = iConfig.getUntrackedParameter<bool>("triggerHLT");
        hlTriggerResults_ = iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults");
        hltProcess_       = iConfig.getUntrackedParameter<string>("hltName");
        mpiTriggers_      = iConfig.getUntrackedParameter<vector<string> >("triggers");
        rootfilename      = iConfig.getUntrackedParameter<string>("rootfilename");
}

MPIntuple::~MPIntuple()
{

}

//
// member functions
//

// ------------ method called to for each event  ------------
void MPIntuple::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{

        eventNumber  = iEvent.id().event(); 
        runNumber    = iEvent.id().run();
        lumiSection  = (unsigned int)iEvent.getLuminosityBlock().luminosityBlock();
        bunchCross   = (unsigned int)iEvent.bunchCrossing();
        isRealData   = iEvent.isRealData();

        edm::Handle<reco::BeamSpot> beamSpotHandle;
        iEvent.getByLabel("offlineBeamSpot", beamSpotHandle);
        reco::BeamSpot vertexBeamSpot = *beamSpotHandle;

        int pfCount   = 0;
        int genCount  = 0;
        int vtxCount  = 0;
        float primaryVertexZ = -999;

        //////////////////////////
        //Get vertex information//
        //////////////////////////

        Handle<reco::VertexCollection> primaryVtcs;
        iEvent.getByLabel(primaryVtxTag_, primaryVtcs);

        for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
                reco::Vertex myVtx = reco::Vertex(*vtx_iter);
                if(!myVtx.isValid() || myVtx.isFake()) continue;
                TCPrimaryVtx* vtxCon = new ((*primaryVtx)[vtxCount]) TCPrimaryVtx;
                vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
                vtxCon->SetNDof(myVtx.ndof());
                vtxCon->SetChi2(myVtx.chi2());
                vtxCon->SetNTrks(myVtx.tracksSize()); 
                vtxCon->SetSumPt2Trks(sumPtSquared(myVtx));
                if(vtxCount == 0) primaryVertexZ = myVtx.z();
                ++vtxCount;
        }

        p1_nVtcs->Fill(runNumber, vtxCount);

        ///////////////////////
        //get jet information//
        ///////////////////////

        if(doPFJets_){

                edm::Handle<reco::JetTagCollection> bTagHandle1;
                iEvent.getByLabel("trackCountingHighEffBJetTags", bTagHandle1);
                const reco::JetTagCollection & bTags1 = *(bTagHandle1.product());

                edm::Handle<reco::JetTagCollection> bTagHandle2;
                iEvent.getByLabel("trackCountingHighPurBJetTags", bTagHandle2);
                const reco::JetTagCollection & bTags2 = *(bTagHandle2.product());
                reco::JetTagCollection::const_iterator jet_it_1, jet_it_2;

                edm::ESHandle<JetCorrectorParametersCollection> JetCorParColl;
                iSetup.get<JetCorrectionsRecord>().get("AK5PF",JetCorParColl);
                JetCorrectorParameters const & JetCorPar = (*JetCorParColl)["Uncertainty"];
                JetCorrectionUncertainty *jecUnc = new JetCorrectionUncertainty(JetCorPar);

                //const JetCorrector* correctorL1  = JetCorrector::getJetCorrector("ak5PFL1Offset",iSetup);
                const JetCorrector* correctorL2  = JetCorrector::getJetCorrector("ak5PFL2Relative",iSetup);
                const JetCorrector* correctorL3  = JetCorrector::getJetCorrector("ak5PFL3Absolute",iSetup);
                const JetCorrector* correctorRes = JetCorrector::getJetCorrector("ak5PFResidual", iSetup);

                Handle<reco::PFJetCollection> PFJets;
                iEvent.getByLabel(recoJetTag_, PFJets);


                for (PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter) {

                        reco::PFJet myJet  = reco::PFJet(*jet_iter);
                        reco::PFJet corJet = reco::PFJet(*jet_iter);

                        float scale2 = correctorL2->correction(corJet);
                        corJet.scaleEnergy(scale2);
                        float scale3 = correctorL3->correction(corJet);
                        corJet.scaleEnergy(scale3);

                        if (corJet.pt() > 5) {

                                TCJet* jetCon = new ((*recoJets)[pfCount]) TCJet;

                                jetCon->SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
                                jetCon->SetVtx(-999.0, -999.0, -999.0);
                                jetCon->SetChHadFrac(myJet.chargedHadronEnergyFraction());
                                jetCon->SetNeuHadFrac(myJet.neutralHadronEnergyFraction());
                                jetCon->SetChEmFrac(myJet.chargedEmEnergyFraction());
                                jetCon->SetNeuEmFrac(myJet.neutralEmEnergyFraction());
                                jetCon->SetNumConstit(myJet.chargedMultiplicity() + myJet.neutralMultiplicity());
                                jetCon->SetNumChPart(myJet.chargedMultiplicity());

                                // Get b-tag information
                                for (jet_it_1 = bTags1.begin(); jet_it_1 != bTags1.end(); jet_it_1++) {
                                        if (sqrt(pow(jet_it_1->first->eta() - corJet.eta(), 2) + pow(deltaPhi(jet_it_1->first->phi(),corJet.phi()), 2)) == 0.) {
                                                jetCon->SetBDiscrTrkCountHiEff(jet_it_1->second);
                                                break;
                                        }
                                }

                                for (jet_it_2 = bTags2.begin(); jet_it_2 != bTags2.end(); jet_it_2++) {
                                        if (sqrt(pow(jet_it_2->first->eta() - corJet.eta(), 2) + pow(deltaPhi(jet_it_2->first->phi(),corJet.phi()), 2)) == 0.) {
                                                jetCon->SetBDiscrTrkCountHiPure(jet_it_2->second);
                                                break;
                                        }
                                }


                                //add more corrections

                                jetCon->SetJetCorr(2, scale2);
                                jetCon->SetJetCorr(3, scale3);

                                if (isRealData) {
                                        float scaleRes = correctorRes->correction(corJet);
                                        jetCon->SetJetCorr(4, scaleRes);

                                        jecUnc->setJetEta(corJet.eta());
                                        jecUnc->setJetPt(corJet.pt());
                                        jetCon->SetUncertaintyJES(jecUnc->getUncertainty(true)); 
                                }

                                /////////////////////////
                                //get associated tracks//
                                /////////////////////////

                                const reco::TrackRefVector &tracks = myJet.getTrackRefs();

                                vector<TVector3> vtxPositionCollection;
                                vector<float>  associatedTrackSumPt;
                                vector<unsigned int> jetTrackAddresses;
                                float sumTrackX, sumTrackY, sumTrackZ, sumTrackIP, sumTrackPt;
                                int   nJetTracks, nVertexTracks, nAssociatedTracks, vertexIndex;
                                int   vCount = 0;

                                nJetTracks = nVertexTracks = nAssociatedTracks = 0;
                                sumTrackX = sumTrackY = sumTrackZ  = sumTrackIP  = sumTrackPt = 0;

                                if(fabs(myJet.eta()) < 2.5){

                                        for (TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack) {
                                                const reco::Track &myJetTrack = **iTrack;

                                                sumTrackPt += myJetTrack.pt();
                                                sumTrackX  += myJetTrack.vx();
                                                sumTrackY  += myJetTrack.vy();            
                                                sumTrackZ  += myJetTrack.vz();
                                                sumTrackIP += myJetTrack.dxy(vertexBeamSpot.position());
                                                jetTrackAddresses.push_back((unsigned int)&myJetTrack);
                                                ++nJetTracks;
                                        }

                                        if (nJetTracks > 0) {
                                                jetCon->SetVtx(sumTrackX/nJetTracks, sumTrackY/nJetTracks, sumTrackZ/nJetTracks);       
                                                h1_meanJetTrackZ->Fill(sumTrackZ/nJetTracks);
                                                h1_trackDxy->Fill(sumTrackIP/nJetTracks);
                                        }
                                        if(jetTrackAddresses.size() > 0){

                                                for (VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter) {           
                                                        reco::Vertex myVtx = reco::Vertex(*vtx_iter); 
                                                        if(!myVtx.isValid() || myVtx.isFake()) continue;
                                                        TVector3 *iVtxPosition = new TVector3(myVtx.x(), myVtx.y(), myVtx.z());
                                                        vtxPositionCollection.push_back(*iVtxPosition);
                                                        associatedTrackSumPt.push_back(0);            

                                                        for(Vertex::trackRef_iterator iTrackRef = myVtx.tracks_begin(); iTrackRef != myVtx.tracks_end(); ++iTrackRef){
                                                                const edm::RefToBase<reco::Track> &myTrackRef = *iTrackRef; 
                                                                if(myTrackRef.isAvailable()){
                                                                        const reco::Track &myVertexTrack = *myTrackRef.get();           

                                                                        for(vector<unsigned int>::const_iterator iTrackAddress = jetTrackAddresses.begin(); iTrackAddress != jetTrackAddresses.end(); ++iTrackAddress){
                                                                                if (*iTrackAddress == (unsigned int)&myVertexTrack) {
                                                                                        associatedTrackSumPt.at(vCount) += myVertexTrack.pt()/sumTrackPt; 
                                                                                        ++nAssociatedTracks;
                                                                                }
                                                                        }
                                                                }
                                                        }
                                                        ++vCount;  
                                                }

                                                float maxSumPtFraction = 0;
                                                vCount = vertexIndex = 0;

                                                for (vector<float>::const_iterator iTrackSumPt = associatedTrackSumPt.begin(); iTrackSumPt != associatedTrackSumPt.end(); ++iTrackSumPt) {
                                                        if (*iTrackSumPt > maxSumPtFraction) {
                                                                maxSumPtFraction = *iTrackSumPt;   
                                                                vertexIndex      = vCount + 1;
                                                        }
                                                        ++vCount;
                                                }
                                                if (vertexIndex > 0) {
                                                        h1_jetVertexZ->Fill(vtxPositionCollection[vertexIndex-1].z());
                                                }

                                                jetCon->SetVtxSumPtFrac(maxSumPtFraction);
                                                jetCon->SetVtxSumPt(sumTrackPt);
                                                jetCon->SetVtxTrackFrac((float)nAssociatedTracks/(float)nJetTracks);
                                                jetCon->SetVtxNTracks(nJetTracks);
                                                jetCon->SetVtxIndex(vertexIndex);
                                        }
                                }        
                                ++pfCount;
                        }      
                }   
        }


        //////////////////
        // Get MET info //
        //////////////////


        Handle<PFMETCollection> MET;
        iEvent.getByLabel(recoMETTag_, MET);

        int metCount = 0;
        for (PFMETCollection::const_iterator iMET = MET->begin(); iMET != MET->end(); ++iMET) {
                TCMET* metCon = new ((*recoMET)[metCount]) TCMET;
                metCon->SetSumEt(iMET->sumEt());
                metCon->SetMet(iMET->et());
                metCon->SetPhi(iMET->phi());
                metCon->SetPhotonEtFraction(iMET->photonEtFraction());
                metCon->SetElectronEtFraction(iMET->electronEtFraction());
                metCon->SetMuonEtFraction(iMET->muonEtFraction());
                metCon->SetNeutralHadronEtFraction(iMET->neutralHadronEtFraction());
                metCon->SetChargedHadronEtFraction(iMET->chargedHadronEtFraction());
                metCon->SetHFHadronEtFraction(iMET->HFHadronEtFraction());
                metCon->SetHFEMEtFraction(iMET->HFEMEtFraction());
        }

        ///////////////////
        // Get muon info //
        ///////////////////


        Handle<MuonCollection> muons;
        iEvent.getByLabel(muonTag_, muons);

        int muCount = 0;
        for (MuonCollection::const_iterator mu = muons->begin(); mu != muons->end(); ++mu) {
                if (mu->isGlobalMuon() && mu->pt() > 8){
                        TCMuon* lepCon = new ((*recoMuons)[muCount]) TCMuon;
                        lepCon->Setp4(mu->px(), mu->py(), mu->pz(), mu->p());
                        lepCon->SetVtx(mu->globalTrack()->vx(),mu->globalTrack()->vy(),mu->globalTrack()->vz());
                        lepCon->SetEta(mu->eta());
                        lepCon->SetPhi(mu->phi());
                        lepCon->SetCharge(mu->charge());
                        lepCon->SetisGLB(mu->isGlobalMuon());
                        lepCon->SetisTRK(mu->isTrackerMuon());
                        lepCon->Setdxy(mu->globalTrack()->dxy(vertexBeamSpot.position()));
                        lepCon->SetnPXLHits(mu->globalTrack()->hitPattern().numberOfValidPixelHits());
                        lepCon->SetnTRKHits(mu->globalTrack()->hitPattern().numberOfValidTrackerHits());
                        lepCon->SetnValidMuHits(mu->globalTrack()->hitPattern().numberOfValidMuonHits());
                        lepCon->SetnMatchSeg(mu->numberOfMatches());
                        lepCon->SetNormChi2(mu->globalTrack()->normalizedChi2());
                        lepCon->SetCaloComp(mu->caloCompatibility());
                        lepCon->SetSegComp(muon::segmentCompatibility(*mu));
                        lepCon->SetEMIso(mu->isolationR03().emEt);
                        lepCon->SetHADIso(mu->isolationR03().hadEt);
                        lepCon->SetTRKIso(mu->isolationR03().sumPt);
                        muCount++;
                }
        }


        ///////////////////////
        // Get electron info //
        ///////////////////////


        Handle<edm::ValueMap<float> > eIDValueMap;
        iEvent.getByLabel( electronIDMap_ , eIDValueMap );
        const edm::ValueMap<float> & eIDmap = * eIDValueMap ;

        Handle<GsfElectronCollection> electrons;
        iEvent.getByLabel(electronTag_, electrons);

        int elCount = 0;
        for (unsigned int i = 0; i < electrons->size(); i++){
                edm::Ref<reco::GsfElectronCollection> electronRef(electrons,i);
                if ( eIDmap[electronRef] == 7 && electronRef->pt() > 15){
                        TCElectron* lepCon = new ((*recoElectrons)[elCount]) TCElectron;
                        lepCon->Setp4(electronRef->px(),electronRef->py(),electronRef->pz(),electronRef->p());
                        lepCon->SetVtx(electronRef->gsfTrack()->vx(),electronRef->gsfTrack()->vy(),electronRef->gsfTrack()->vz());
                        lepCon->SetCharge(electronRef->charge());
                        lepCon->SetEta(electronRef->eta());
                        lepCon->SetPhi(electronRef->phi());
                        lepCon->Setdxy(electronRef->gsfTrack()->dxy(vertexBeamSpot.position()));
                        lepCon->SetNormChi2(electronRef->gsfTrack()->normalizedChi2());
                        lepCon->SetEMIso(electronRef->dr03EcalRecHitSumEt());
                        lepCon->SetHADIso(electronRef->dr03HcalTowerSumEt());
                        lepCon->SetTRKIso(electronRef->dr03TkSumPt());
                        elCount++;
                }
        }


        ////////////////////////
        // Get gen-level info //
        ////////////////////////


        if (!isRealData) {

                Handle<HepMCProduct > genEvtHandle;
                iEvent.getByLabel( "generator", genEvtHandle) ;
                const HepMC::GenEvent* Evt = genEvtHandle->GetEvent();

                Handle<GenEventInfoProduct> GenEventInfoHandle;
                iEvent.getByLabel("generator", GenEventInfoHandle);

                Handle<reco::GenJetCollection> GenJets;
                iEvent.getByLabel(genJetTag_, GenJets);

                ptHat = qScale = -1; crossSection = 0;

                if (GenEventInfoHandle.isValid()) {
                        qScale       = GenEventInfoHandle->qScale();
                        ptHat        = (GenEventInfoHandle->hasBinningValues() ? GenEventInfoHandle->binningValues()[0] : 0.0);
                        evtWeight    = GenEventInfoHandle->weight();

         if (evtWeight != 1) cout<<"Event weight: "<<evtWeight<<endl;
                        //if (qScale != ptHat) cout<<"q scale: "<<qScale<<endl;
                        h1_ptHat->Fill(ptHat);
                }
                vector<HepMC::GenParticle*> genPartons;
                int hardCount = 0;

                //cout<<"Event: "<<Evt->signal_process_id()<<endl;

                for (HepMC::GenEvent::particle_const_iterator iGenParticle = Evt->particles_begin(); iGenParticle != Evt->particles_end(); ++iGenParticle) {
                        HepMC::GenParticle *myGenPart = *iGenParticle;
                        if (myGenPart->status() == 23) {
                                new ((*hardPartonP4)[hardCount]) TLorentzVector(myGenPart->momentum().px(), myGenPart->momentum().py(), myGenPart->momentum().pz(), myGenPart->momentum().e());
                                partonPdgId[hardCount] = myGenPart->pdg_id();

                                //genPartons.push_back(myGenPart);
                                //float genPt = sqrt(pow(myGenPart->momentum().px(), 2) + pow(myGenPart->momentum().py(), 2));
                                ////cout<<"Hard scattering parton "<<hardCount<<" : ("<<genPt<<", "<<myGenPart->momentum().phi()<<", "<<myGenPart->momentum().eta()<<"), vtx : "<<myGenPart->production_vertex()->barcode()<<", pdgId : "<<myGenPart->pdg_id()<<endl;
                                //
                                //for(HepMC::GenVertex::particle_iterator iGenDescend = myGenPart->end_vertex()->particles_begin(HepMC::descendants); 
                                //              iGenDescend != myGenPart->end_vertex()->particles_end(HepMC::descendants); ++iGenDescend) {
                                //      HepMC::GenParticle *myGenDescend = *iGenDescend;
                                //      if (myGenDescend->status() == 71){
                                //              float descPt = sqrt(pow(myGenDescend->momentum().px(), 2) + pow(myGenDescend->momentum().py(), 2));
                                //              //cout<<"\t End partons"<<" : ("<<descPt<<", "<<myGenDescend->momentum().phi()<<", "<<myGenDescend->momentum().eta()<<"), vtx : "<<", pdgId : "<<myGenDescend->pdg_id()<<endl;
                                //      }
                                //}
                                //cout<<hardCount<<endl;
                                ++hardCount;
                        }
                }

                for (GenJetCollection::const_iterator jet_iter = GenJets->begin(); jet_iter!= GenJets->end(); ++jet_iter) {
                        reco::GenJet myJet = reco::GenJet(*jet_iter);      
                        if (myJet.pt() > 15 && fabs(myJet.eta()) < 2.5) { 

                                TCGenJet* jetCon = new ((*genJets)[genCount]) TCGenJet;
                                jetCon->SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
                                jetCon->SetHadEnergy(myJet.hadEnergy());
                                jetCon->SetEmEnergy(myJet.emEnergy());
                                jetCon->SetInvEnergy(myJet.invisibleEnergy());
                                jetCon->SetAuxEnergy(myJet.auxiliaryEnergy());
                                jetCon->SetNumConstit(myJet.getGenConstituents().size());

                                //cout<<"~~JET "<<genCount+1<<"~~"<<endl;
            //cout<<"(pt, phi, eta) = ("<<myJet.pt()<<", "<<myJet.phi()<<", "<<myJet.eta()<<")"<<endl;

                                int pCount, nearestIndex, dR5Index, dR9Index;
                                float nearestDR, nearest2DR, nearestDR5, nearestDR9;
                                float nearestDPT, nearest2DPT, nearestDPT5, nearestDPT9;
                                pCount = nearestIndex = dR5Index = dR9Index = 0;
                                nearestDR = nearest2DR = nearestDR5 = nearestDR9 = 10.;
                                nearestDPT = nearest2DPT = nearestDPT5 = nearestDPT9 = 15.;
                                
                                for (vector<HepMC::GenParticle*>::const_iterator parton_iter = genPartons.begin(); parton_iter != genPartons.end(); ++parton_iter) {
                                        HepMC::GenParticle *myParton = *parton_iter;
                                        float partonPt = sqrt(pow(myParton->momentum().px(), 2) + pow(myParton->momentum().py(), 2));
                                        float dPhi = myJet.phi() - myParton->momentum().phi();
                                        if (dPhi > TMath::Pi()) dPhi = 2*TMath::Pi() - dPhi;
                                        float dR = sqrt(pow(myJet.eta() - myParton->momentum().eta(), 2) + pow(dPhi, 2));

                                        if (dR < nearestDR) {
                                                nearest2DR   = nearestDR;
                                                nearestDR    = dR;
                                                nearest2DPT  = nearestDPT;
                                                nearestDPT   = fabs(partonPt -myJet.pt())/myJet.pt();
                                                nearestIndex = pCount + 1;
                                        }
                                        if (dR < 0.5) {
                                                nearestDR5 = dR;
                                                nearestDPT5 = fabs(partonPt -myJet.pt())/myJet.pt();
                                                if (dR5Index == 0) {
                                                        dR5Index = pCount + 1;
                                                } else { 
                                                        dR5Index = 5; 
                                                }
                                        }
                                        if (dR < 0.9) {
                                                nearestDR9 = dR;
                                                nearestDPT9 = fabs(partonPt -myJet.pt())/myJet.pt();
                                                if (dR9Index == 0) {
                                                        dR9Index = pCount + 1;
                                                } else { 
                                                        dR9Index = 5; 
                                                }
                                        }
                                        //cout<<"dR jet-parton: "<<dR<<" | dpT: "<<partonPt - myJet.pt()<<endl;
                                        ++pCount;
                                }

                                h1_partonJetMatch->Fill(nearestIndex);    
                                h1_partonJetMatch_dR5->Fill(dR5Index);    
                                h1_partonJetMatch_dR9->Fill(dR9Index);    
                                h1_partonJetdR->Fill(nearestDR);       
                                h1_partonJet2dR->Fill(nearest2DR);       
                                h1_partonJetdpT->Fill(nearestDPT);                                      
                                h1_partonJet2dpT->Fill(nearest2DPT);                                    
                                        //h1_partonJetMatch_dR5->Fill();
                                        //h1_partonJetMatch_dR9->Fill();

                                ++genCount;     
                        }
                }
                //cout<<"\nEND EVENT\n"<<endl;
        }
  
  ///////////////////////////  
  //get trigger information//
  ///////////////////////////

        if (triggerHLT_) {

    edm::Handle<TriggerResults> hltR;
    triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltProcess_);
    iEvent.getByLabel(triggerResultsTag_,hltR);

    const TriggerNames & triggerNames = iEvent.triggerNames(*hltR);
    hlNames=triggerNames.triggerNames();   

    triggerStatus = 0x0;    

    //for(int i = 0; i < (int)mpiTriggers_.size(); ++i) hltPrescale[i] = hltConfig_.prescaleValue(iEvent, iSetup, mpiTriggers_[i]);

         for (int i=0; i < (int)hlNames.size(); ++i) {      
                 if (!triggerDecision(hltR, i)) continue;       
                 for (int j = 0; j < (int)mpiTriggers_.size(); ++j){
                         if (hlNames[i].compare(0, mpiTriggers_[j].length(),mpiTriggers_[j]) == 0) {
                                 triggerStatus |= 0x01 << j;
                         }
                 }
         } 
  }

  if((pfCount >= nJets_ || genCount >= nJets_) && vtxCount > 0) sTree -> Fill();

  recoJets->Clear();
  recoMET->Clear();
  recoElectrons->Clear();
  recoMuons->Clear();
  primaryVtx->Clear();
  genJets->Clear();
  hardPartonP4->Clear();
}

// ------------ method called once each job just before starting event loop  ------------
void  MPIntuple::beginJob()
{  
  ntupleFile               = new TFile(rootfilename.c_str(), "RECREATE");
  sTree                    = new TTree("mpiTree", "Tree for Jets");

  h1_ptHat                 = new TH1D("h1_ptHat", "ptHat", 15, 10.0, 160.0);
  h1_fracAssociatedTracks  = new TH1D("h1_fracAssociatedTracks", "Fraction of associated jet tracks", 20, 0.0, 1.0);
  h1_trackDxy              = new TH1D("h1_trackDxy", "dxy for all associated tracks", 50, -0.4, 0.4);
  h1_meanJetTrackZ         = new TH1D("h1_meanJetTrackZ", "<z> from jet tracks", 50, -25.0, 25.0);
  h1_jetVertexZ            = new TH1D("h1_jetVertexZ", "z for vertex associated to jet", 50, -25.0, 25.0);
  h1_associatedSumPt       = new TH1D("h1_associatedSumPt", "ratio of sum p_{T} between associatedTracks and jetTracks", 20, 0.0, 1.0);
  h1_associatedVertexIndex = new TH1D("h1_associatedVertex", "number of jets associated to primary vertex", 5, -0.5, 4.5);
  h1_partonJetMatch        = new TH1D("h1_partonJetMatch", "jet association to hardest scattering", 4, 0.5, 4.5);
  h1_partonJetMatch_dR5    = new TH1D("h1_partonJetMatch_dR5", "jet association to hardest scattering (dR < 0.5)", 6, -0.5, 5.5);
  h1_partonJetMatch_dR9    = new TH1D("h1_partonJetMatch_dR9", "jet association to hardest scattering (dR < 0.9)", 6, -0.5, 5.5);
  h1_partonJetdR           = new TH1D("h1_partonJetdR", "jet-nearest parton dR", 50, 0., 5.);
  h1_partonJet2dR          = new TH1D("h1_partonJet2dR", "jet-2nd nearest parton dR", 50, 0., 5.);
  h1_partonJetdpT          = new TH1D("h1_partonJetDpT", "jet-nearest parton dpT", 20, 0., 1.);
  h1_partonJet2dpT         = new TH1D("h1_partonJet2DpT", "jet-2nd nearest parton dpT", 20, 0., 1.);
  h2_nAssTracksVsJetPt     = new TH2F("h2_nAssTracksVsJetPt", "nTracks vs Jet p_{T}", 50, 0, 100, 20, -0.5, 19.5);
  p1_nVtcs                 = new TProfile("p1_nVtcs", "<N> per run", 5200.0, 132300.0, 137500.0, 0.0, 6.0);

  recoJets                 = new TClonesArray("TCJet");
  recoMET                  = new TClonesArray("TCMET");
  recoElectrons            = new TClonesArray("TCElectron");
  recoMuons                = new TClonesArray("TCMuon");
  genJets                  = new TClonesArray("TCGenJet");
  primaryVtx               = new TClonesArray("TCPrimaryVtx");
  hardPartonP4             = new TClonesArray("TLorentzVector");

  sTree->Branch("recoJets",&recoJets, 6400, 0);
  sTree->Branch("recoElectrons",&recoElectrons, 6400, 0);
  sTree->Branch("recoMuons",&recoMuons, 6400, 0);
  sTree->Branch("genJets",&genJets, 6400, 0);
  sTree->Branch("primaryVtx",&primaryVtx, 6400, 0);
  sTree->Branch("recoMET",&recoMET, 6400, 0);
  sTree->Branch("hardPartonP4",&hardPartonP4, 6400, 0);
  sTree->Branch("partonPdgId",partonPdgId, "partonPdgId[32]/i");
  sTree->Branch("eventNumber",&eventNumber, "eventNumber/I");
  sTree->Branch("runNumber",&runNumber, "runNumber/I");
  sTree->Branch("lumiSection",&lumiSection, "lumiSection/I");
  sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/i");
  sTree->Branch("hltPrescale",hltPrescale, "hltPrescale[32]/i");
  sTree->Branch("isRealData",&isRealData, "isRealData/i");
  sTree->Branch("bunchCross",&bunchCross, "bunchCross/i");
  sTree->Branch("lumiDeadCount",&lumiDeadCount, "lumiDeadCount/f");
  sTree->Branch("lumiLiveFrac",&lumiLiveFrac, "lumiLiveFrac/f");
  sTree->Branch("intDelLumi",&intDelLumi, "intDelLumi/f");
  sTree->Branch("ptHat",&ptHat, "ptHat/f");
  sTree->Branch("qScale", &qScale, "qScale/f");
  sTree->Branch("evtWeight", &evtWeight, "evtWeight/f");
  sTree->Branch("crossSection", &crossSection, "crossSection/f");
}

void MPIntuple::beginRun(const edm::Run& iRun, const edm::EventSetup& iEvent)
{
  bool changed = true; 
  hltConfig_.init(iRun, iEvent, hltProcess_, changed);
}

void MPIntuple::endLuminosityBlock(const edm::LuminosityBlock& iLumi, const edm::EventSetup& iEvent)
{
  edm::Handle<LumiSummary> lumiSummary;
  iLumi.getByLabel("lumiProducer", lumiSummary);

  lumiDeadCount  = lumiSummary->deadcount();
  lumiLiveFrac   = lumiSummary->liveFrac();
  intDelLumi     = lumiSummary->avgInsDelLumi()*93.244;

  //cout<<iLumi.id().luminosityBlock()<<endl;
  //cout<<"\t Dead Count = "<<lumiSummary->deadcount()<<endl;
  //cout<<"\t Fraction of dead time = "<<1 - lumiSummary->liveFrac()<<endl;
  //cout<<"\t Integrated luminosity = "<<lumiSummary->avgInsDelLumi()*93.244<<endl;
  //cout<<"\t Dead time corrected luminosity = "<<lumiSummary->avgInsDelLumi()*lumiSummary->liveFrac()*93.244<<endl;
}

void MPIntuple::endRun(const edm::Run& iRun, const edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  Handle<GenRunInfoProduct> GenRunInfoHandle;
  iEvent.getByLabel("generator", GenRunInfoHandle);

  if (GenRunInfoHandle.isValid()) {
    crossSection = GenRunInfoHandle->crossSection();
  }

  for(int i = 0; i < (int)mpiTriggers_.size(); ++i) hltPrescale[i] = hltConfig_.prescaleValue(iEvent, iSetup, mpiTriggers_[i]);
  //for (int i = 0; i < (int)mpiTriggers_.size(); ++i) cout << mpiTriggers_[i] << " prescale = " << hltPrescale[i] <<endl; 
}
// ------------ method called once each job just after ending the event loop  ------------
void MPIntuple::endJob() 
{
  ntupleFile->cd();

  h1_ptHat->Write();
  h1_fracAssociatedTracks->Write();
  h1_meanJetTrackZ->Write();
  h1_trackDxy->Write();
  h1_jetVertexZ->Write();
  h1_associatedSumPt->Write();
  h1_associatedVertexIndex->Write();
  h1_partonJetMatch->Write();
  h1_partonJetdR->Write();
  h1_partonJet2dR->Write();
  h1_partonJetdpT->Write();
  h1_partonJet2dpT->Write();
  h1_partonJetMatch_dR5->Write();
  h1_partonJetMatch_dR9->Write();
  h2_nAssTracksVsJetPt->Write();
  p1_nVtcs->Write();

  ntupleFile->Write();
  ntupleFile->Close();
}

bool MPIntuple::genParticleMatch(const HepMC::GenParticle& genParticle, const reco::Candidate& motherParticle)
{
        float genPhi, momPhi, genEta, momEta;
        genPhi = genParticle.momentum().phi();
        genEta = genParticle.momentum().eta();
        momPhi = motherParticle.phi();
   momEta = motherParticle.eta();

        float dPhi = momPhi - genPhi;
        if (dPhi > TMath::Pi()) dPhi = 2*TMath::Pi() - dPhi;
        float dR = sqrt(pow((genPhi - momPhi), 2) + pow((genEta - momEta), 2));
        if (dR < 0.00001) return true;
        else return false;
}

bool MPIntuple::triggerDecision(edm::Handle<edm::TriggerResults> &hltR, int iTrigger)
{
    bool triggerPassed = false;
    if(hltR->wasrun(iTrigger) &&
       hltR->accept(iTrigger) &&
       !hltR->error(iTrigger) ){
      triggerPassed = true;
    }
    return triggerPassed;
}

float MPIntuple::sumPtSquared(const Vertex & v)
{
  float sum = 0.;
  float pT;
  for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
    pT = (**it).pt();
    float epT=(**it).ptError(); pT=pT>epT ? pT-epT : 0;

    sum += pT*pT;
  }
  return sum;
}

//define this as a plug-in
DEFINE_FWK_MODULE(MPIntuple);
Revision:	1.36
Committed:	Wed Apr 27 15:54:46 2011 UTC (14 years ago) by naodell
Content type:	text/plain
Branch:	MAIN
Changes since 1.35:	+2 -2 lines
Log Message:	Start transition for 4XX compatibility