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.23 2010/11/10 12:30:58 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/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "DataFormats/Math/interface/deltaPhi.h"

// Libraries for objects
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/JetReco/interface/GenJet.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BTauReco/interface/JetTag.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenRunInfoProduct.h"

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

//GenParticles
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

// Need to get correctors
#include "JetMETCorrections/Objects/interface/JetCorrector.h"

// ntuple storage classes
#include "TCJet.h"
#include "TCPrimaryVtx.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"

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 analyze(const edm::Event&, const edm::EventSetup&);
      virtual void endJob() ;

  bool triggerDecision(edm::Handle<edm::TriggerResults> &hltR, int iTrigger);
  double sumPtSquared(const Vertex & v);

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

  edm::InputTag PFJetHandle_;
  edm::InputTag GenJetHandle_;
  edm::InputTag PrimaryVtxHandle_;
  edm::InputTag triggerResultsTag_;
  //  Counting variables   //

  int eventNumber, runNumber, lumiSection, bunchCross;
  double ptHat, qScale, crossSection;
  
  TTree* sTree;
  TFile* ntupleFile;

  TClonesArray* jet_ak5PF;
  TClonesArray* jetP4_ak5Gen;
  TClonesArray* primaryVtx;

  bool doGenJets_;
  bool doPFJets_;
  bool triggerHLT_;
  bool isRealData;
  string rootfilename;

  //Triggers
  string hlTriggerResults_, hltName_, triggerName_;
  TriggerNames triggerNames;
  HLTConfigProvider hltConfig_;
  vector<string>  hlNames;
  unsigned int triggerStatus;

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


};

MPIntuple::MPIntuple(const edm::ParameterSet& iConfig)
{
  PFJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag");
  GenJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag");
  PrimaryVtxHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag");
  doGenJets_ = iConfig.getUntrackedParameter<bool>("doGenJets");
  doPFJets_ = iConfig.getUntrackedParameter<bool>("doPFJets");
  triggerHLT_ = iConfig.getUntrackedParameter<bool>("triggerHLT");
  rootfilename = iConfig.getUntrackedParameter<string>("rootfilename");
  hlTriggerResults_ = iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults");
  hltName_ = iConfig.getUntrackedParameter<string>("hltName");
}

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;
  double primaryVertexZ = -999;

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

  Handle<reco::VertexCollection> primaryVtcs;
  iEvent.getByLabel(PrimaryVtxHandle_, 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;
      
    //cout<<myVtx.nTracks()<<endl;

    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());
    reco::JetTagCollection::const_iterator jet_it_1;

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


    const JetCorrector* correctorL2 = JetCorrector::getJetCorrector ("ak5PFL2Relative",iSetup);
    const JetCorrector* correctorL3 = JetCorrector::getJetCorrector ("ak5PFL3Absolute",iSetup);

    Handle<reco::PFJetCollection> PFJets;
    iEvent.getByLabel(PFJetHandle_, PFJets);
        
    for(PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter){
      
      reco::PFJet myJet = reco::PFJet(*jet_iter);
      if(myJet.pt() > 5){

        for(jet_it_1 = bTags1.begin(); jet_it_1 != bTags1.end(); jet_it_1++) {
           if( (fabs(jet_it_1->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_1->first->phi(),myJet.phi()) < .005)) break;
        }

        for(jet_it_2 = bTags2.begin(); jet_it_2 != bTags2.end(); jet_it_2++) {
           if( (fabs(jet_it_2->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_2->first->phi(),myJet.phi()) < .005)) break;
        }
                
        TCJet* jetCon = new ((*jet_ak5PF)[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());

        jetCon->SetNumChPart(myJet.chargedMultiplicity());

        if(jet_it_2 != bTags2.end()) jetCon->SetBDiscrTrkCountHiPure(jet_it_2->second);
        if(jet_it_1 != bTags1.end()) jetCon->SetBDiscrTrkCountHiEff(jet_it_1->second);

        double scale2 = correctorL2->correction(myJet);
        myJet.scaleEnergy(scale2);
        double scale3 = correctorL3->correction(myJet);
        myJet.scaleEnergy(scale3);

        //more corrections?

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

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

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

        vector<TVector3> vtxPositionCollection;
        vector<double>  associatedTrackSumPt;
        vector<unsigned int> jetTrackAddresses;
        double sumTrackX, sumTrackY, sumTrackZ, sumTrackIP, sumTrackPt;
        //double meanTrackX, meanTrackY, meanTrackZ, meanTrackIP;
        int   nJetTracks, nVertexTracks, nAssociatedTracks, vertexIndex;
        int   vCount = 0;

        nJetTracks = nVertexTracks = nAssociatedTracks = 0;
        sumTrackX = sumTrackY = sumTrackZ  = sumTrackIP  = sumTrackPt = 0;
        //meanTrackZ = meanTrackIP = -999;

        if(myJet.pt() > 10 && fabs(myJet.eta()) < 2.4){

          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);

          }
          /*meanTrackX = sumTrackX/nJetTracks;
          meanTrackY = sumTrackY/nJetTracks;      
          meanTrackZ = sumTrackZ/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;  
            }
                    
            double maxSumPtFraction = 0;
            vCount = vertexIndex = 0;

            for(vector<double>::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->SetVtxSumPt(maxSumPtFraction);
            jetCon->SetVtxIndex(vertexIndex);

            h1_fracAssociatedTracks->Fill((double)nAssociatedTracks/(double)nJetTracks);
            h1_associatedSumPt->Fill(maxSumPtFraction);
            h1_associatedVertexIndex->Fill(vertexIndex);
            h2_nAssTracksVsJetPt->Fill(myJet.pt(), nAssociatedTracks);
          }
        }        
        ++pfCount;
      }      
    }   
  }
  
  if(!isRealData){
    
    Handle<GenEventInfoProduct> GenEventInfoHandle;
    iEvent.getByLabel("generator", GenEventInfoHandle);

    Handle<GenRunInfoProduct> GenRunInfoHandle;
    iEvent.getByLabel("generator", GenRunInfoHandle);

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

    ptHat = qScale = -1; crossSection = 0;

    if(GenEventInfoHandle.isValid())
    {
      qScale       = GenEventInfoHandle->qScale();
      ptHat        = (GenEventInfoHandle->hasBinningValues() ? GenEventInfoHandle->binningValues()[0] : 0.0);
    }
    if(GenRunInfoHandle.isValid())
    {
      crossSection = GenRunInfoHandle->crossSection();
    }
        
    for(GenJetCollection::const_iterator jet_iter = GenJets->begin(); jet_iter!= GenJets->end(); ++jet_iter){
      
      reco::GenJet myJet = reco::GenJet(*jet_iter);
      
      if(myJet.pt() > 5){
        
        new ((*jetP4_ak5Gen)[genCount]) TLorentzVector(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());

        const reco::GenParticle *myGenParticle = myJet.getGenConstituent(0);
        const reco::Candidate   *myMother      = myGenParticle->mother();
        int   nGrandMas = 1;

        for(int iter = 0; nGrandMas != 0; ++iter)
        {
          myMother  = myMother->mother();
          nGrandMas = myMother->mother()->numberOfMothers(); 
        }
        //cout<<myJet.print()<<endl;
        cout<<"Jet flavor: "<< myMother->pdgId()<<endl;
        ++genCount;
        
      }
    }    
  }
  
  ///////////////////////////  
  //get trigger information//
  ///////////////////////////

  if(triggerHLT_){

    edm::Handle<TriggerResults> hltR;
    triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltName_);
    iEvent.getByLabel(triggerResultsTag_,hltR);
    
    const TriggerNames & triggerNames = iEvent.triggerNames(*hltR);
    hlNames=triggerNames.triggerNames();
    
    string MPI_TriggerNames[] = {"HLT_PixelTracks_Multiplicity70", 
                                 "HLT_MinBiasBSC_NoBPTX", 
                                 "HLT_PixelTracks_Multiplicity40",
                                 "HLT_L1Tech_HCAL_HF", 
                                 "HLT_IsoTrackHB_8E29", 
                                 "HLT_IsoTrackHE_8E29", 
                                 "HLT_L1Tech_RPC_TTU_RBst1_collisions", 
                                 "HLT_L1_BscMinBiasOR_BptxPlusORMinus", 
                                 "HLT_L1Tech_BSC_halo_forPhysicsBackground", 
                                 "HLT_L1Tech_BSC_HighMultiplicity", 
                                 "HLT_MinBiasPixel_DoubleIsoTrack5", 
                                 "HLT_MinBiasPixel_DoubleTrack", 
                                 "HLT_MinBiasPixel_SingleTrack", 
                                 "HLT_ZeroBiasPixel_SingleTrack", 
                                 "HLT_MinBiasBSC", 
                                 "HLT_StoppedHSCP_8E29", 
                                 "HLT_Jet15U_HcalNoiseFiltered", 
                                 "HLT_QuadJet15U", 
                                 "HLT_DiJetAve30U_8E29", 
                                 "HLT_DiJetAve15U_8E29", 
                                 "HLT_FwdJet20U", 
                                 "HLT_Jet50U", 
                                 "HLT_Jet30U", 
                                 "HLT_Jet15U", 
                                 "HLT_BTagMu_Jet10U", 
                                 "HLT_DoubleJet15U_ForwardBackward", 
                                 "HLT_BTagIP_Jet50U", 
                                 "HLT_DoubleLooseIsoTau15", 
                                 "HLT_SingleLooseIsoTau20", 
                                 "HLT_HT100U", 
                                 "HLT_MET100", 
                                 "HLT_MET45"};
    
    bool triggerPassed = false;
    triggerStatus = 0x0;
    
    for (uint i=0; i<hlNames.size(); ++i) {
      
      triggerPassed = triggerDecision(hltR, i);
      
      if(triggerPassed){
        
        for (uint j = 0; j != 32; ++j){

          if (hlNames[i] == MPI_TriggerNames[j])
          {
            //cout<<"HLTrigger name: "<<hlNames[i]<<" bit: "<<dec<<j+1<<endl;
            triggerStatus |= 0x01 << j;       
          }
        }
      }
    } 
  }
  //cout<< "total status: "<<hex<<triggerStatus<<endl;


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

  jet_ak5PF->Clear();
  primaryVtx->Clear();
  jetP4_ak5Gen->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_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);
  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);

  jet_ak5PF                = new TClonesArray("TCJet");
  jetP4_ak5Gen             = new TClonesArray("TLorentzVector");
  primaryVtx               = new TClonesArray("TCPrimaryVtx");


  sTree->Branch("jet_ak5PF",&jet_ak5PF, 6400, 0);
  sTree->Branch("jetP4_ak5Gen",&jetP4_ak5Gen, 6400, 0);
  sTree->Branch("primaryVtx",&primaryVtx, 6400, 0);

  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("isRealData",&isRealData, "isRealData/i");
  sTree->Branch("bunchCross",&bunchCross, "bunchCross/i");
  sTree->Branch("ptHat",&ptHat, "ptHat/d");
  sTree->Branch("qScale", &qScale, "qScale/d");
  sTree->Branch("crossSection", &crossSection, "crossSection/d");
}

// ------------ method called once each job just after ending the event loop  ------------
void MPIntuple::endJob() 
{

  ntupleFile->cd();

  h1_fracAssociatedTracks->Write();
  h1_meanJetTrackZ->Write();
  h1_trackDxy->Write();
  h1_jetVertexZ->Write();
  h1_associatedSumPt->Write();
  h1_associatedVertexIndex->Write();
  h2_nAssTracksVsJetPt->Write();
  p1_nVtcs->Write();

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

}


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;
}

double MPIntuple::sumPtSquared(const Vertex & v)
{
  double sum = 0.;
  double pT;
  for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
    pT = (**it).pt();
    double 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.24
Committed:	Mon Nov 29 13:09:04 2010 UTC (14 years, 5 months ago) by naodell
Content type:	text/plain
Branch:	MAIN
Changes since 1.23:	+173 -76 lines
Log Message:	Include genJet flaovor; next update, add genJet class