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.20 2010/11/02 12:13:31 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 "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;
  
  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
  TH1F * h1_nAssociatedTracks;
  TH1F * h1_trackZ;
  TH1F * h1_trackDxy;
  TH1F * h1_allTrackDeltaZ_PV;
  TH1F * h1_allTrackDeltaZ_LJ;
  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","HLT");
}

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 ljVertexZ = -999;
  float 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.et() > 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);

        float scale2 = correctorL2->correction(myJet);
        myJet.scaleEnergy(scale2);
        float 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();

        float sumTrackZ, sumTrackIP;
        float meanTrackZ, meanTrackIP;
        int   nAssociatedTracks = -1;

        sumTrackZ  = sumTrackIP  = 0;
        meanTrackZ = meanTrackIP = -999;

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

          nAssociatedTracks = 0;

          for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){

            const reco::Track    &myJetTrack = **iTrack;
            //const reco::TrackRef &myTrackRef = *iTrack;

            sumTrackZ += myJetTrack.vz();
            sumTrackIP += myJetTrack.dxy(vertexBeamSpot.position());
            ++nAssociatedTracks;

            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;
              
              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();
                                
                  cout<<"Vertex track z = "<<myVertexTrack.vz()<<endl;
                  cout<<"Jet track z = "<<myJetTrack.vz()<<endl;
                
                }
              }       
            }               
          }
          
          meanTrackZ = sumTrackZ/nAssociatedTracks;
          h1_nAssociatedTracks->Fill(nAssociatedTracks);
          h2_nAssTracksVsJetPt->Fill(myJet.pt(), nAssociatedTracks);
          
          for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
            
            const reco::Track &myJetTrack = **iTrack;
            
            if(nAssociatedTracks == 3) h1_trackZ->Fill(meanTrackZ - myJetTrack.vz());
            h1_trackDxy->Fill(myJetTrack.dxy(vertexBeamSpot.position()));
            
          }     
          
          if(pfCount == 0){
            
            ljVertexZ = meanTrackZ;
            
          }else{
            
            h1_allTrackDeltaZ_LJ->Fill(ljVertexZ - meanTrackZ);
            
          }
          
          h1_allTrackDeltaZ_PV->Fill(primaryVertexZ - meanTrackZ);
          
        }
        
        jetCon->SetVtx(0, 0, meanTrackZ);      
        
        ++pfCount;
      }      
    }   
  }
  
  if(!isRealData){
    
    Handle<GenEventInfoProduct> GenInfoHandle;
    iEvent.getByLabel("generator", GenInfoHandle);

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

    ptHat = qScale = -1;

    if(GenInfoHandle.isValid()){

      qScale = GenInfoHandle->qScale();
      ptHat  = (GenInfoHandle->hasBinningValues() ? GenInfoHandle->binningValues()[0] : 0.0);

    }
        
    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());
        
        ++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<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
              triggerStatus |= 0x01 << j;
              
            }
        }
      }
    } 
  }
  //cout<< "total status: "<<hex<<triggerStatus<<endl;


  if((pfCount > 3 || genCount > 3) && 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_trackZ            = new TH1F("h1_trackZ", "#Delta z for associated tracks", 60, -0.3, 0.3);
  h1_nAssociatedTracks = new TH1F("h1_nAssociatedTracks", "Number of tracks associated to jet", 20, -0.5, 19.5);
  h1_trackDxy          = new TH1F("h1_trackDxy", "dxy for all associated tracks", 50, -0.4, 0.4);
  h1_allTrackDeltaZ_PV = new TH1F("h1_allTrackDeltaZ_PV", "#Delta z between jet and primary vertex", 60, -0.3, 0.3);
  h1_allTrackDeltaZ_LJ = new TH1F("h1_allTrackDeltaZ_LJ", "#Delta z between leading jet and other jet vertices", 60, -0.3, 0.3);
  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");

}

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

  ntupleFile->cd();

  h1_trackZ->Write();
  h1_nAssociatedTracks->Write();
  h1_trackDxy->Write();
  h1_allTrackDeltaZ_PV->Write();
  h1_allTrackDeltaZ_LJ->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.21
Committed:	Wed Nov 3 09:53:54 2010 UTC (14 years, 6 months ago) by naodell
Content type:	text/plain
Branch:	MAIN
Changes since 1.20:	+24 -27 lines
Log Message:	Added vertex tracks.
#	User	Rev	Content
1	naodell	1.1	// Original Author: "Nathaniel Odell"
2			// Secondary Author: Steven Won
3	andrey	1.2	// With contributions from: Andrey Pozdnyakov
4	naodell	1.1	// Created: Thurs April 22 2010
5	naodell	1.21	// $Id: MPIntuple.cc,v 1.20 2010/11/02 12:13:31 naodell Exp $
6	naodell	1.1
7			// system include files
8			#include <memory>
9			#include <string>
10
11			// user include files
12			#include "FWCore/Framework/interface/Frameworkfwd.h"
13			#include "FWCore/Framework/interface/EDAnalyzer.h"
14			#include "FWCore/Framework/interface/ESHandle.h"
15			#include "FWCore/Framework/interface/EventSetup.h"
16			#include "FWCore/Framework/interface/Event.h"
17			#include "FWCore/Framework/interface/MakerMacros.h"
18			#include "FWCore/ParameterSet/interface/ParameterSet.h"
19			#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
20			#include "Geometry/Records/interface/CaloGeometryRecord.h"
21	naodell	1.10	#include "DataFormats/Math/interface/deltaPhi.h"
22	naodell	1.1
23	naodell	1.15	// Libraries for objects
24	naodell	1.1	#include "DataFormats/JetReco/interface/CaloJet.h"
25			#include "DataFormats/JetReco/interface/CaloJetCollection.h"
26			#include "DataFormats/JetReco/interface/PFJet.h"
27			#include "DataFormats/JetReco/interface/PFJetCollection.h"
28			#include "DataFormats/JetReco/interface/GenJet.h"
29			#include "DataFormats/JetReco/interface/GenJetCollection.h"
30			#include "DataFormats/JetReco/interface/Jet.h"
31			#include "DataFormats/VertexReco/interface/Vertex.h"
32			#include "DataFormats/VertexReco/interface/VertexFwd.h"
33	naodell	1.7	#include "DataFormats/BTauReco/interface/JetTag.h"
34	naodell	1.15	#include "DataFormats/TrackReco/interface/Track.h"
35	naodell	1.16	#include "DataFormats/BeamSpot/interface/BeamSpot.h"
36	naodell	1.18	#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
37
38	devildog	1.11	//#include "RecoVertex/PrimaryVertexProducer/interface/VertexHigherPtSquared.h"
39	naodell	1.7
40	naodell	1.10	//GenParticles
41	naodell	1.13	//#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
42	naodell	1.1
43	naodell	1.4	// Need to get correctors
44			#include "JetMETCorrections/Objects/interface/JetCorrector.h"
45
46			// ntuple storage classes
47			#include "TCJet.h"
48			#include "TCPrimaryVtx.h"
49
50	andrey	1.2	// Need for HLT trigger info:
51			#include "FWCore/Common/interface/TriggerNames.h"
52			#include "DataFormats/Common/interface/TriggerResults.h"
53			#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"
54
55	naodell	1.4	//Root stuff
56	naodell	1.1	#include "TROOT.h"
57	naodell	1.15	#include "TH1.h"
58	naodell	1.16	#include "TH2.h"
59	naodell	1.15	#include "TProfile.h"
60	naodell	1.1	#include "TFile.h"
61			#include "TTree.h"
62			#include "TString.h"
63			#include "TObject.h"
64			#include "TObjArray.h"
65			#include "TClonesArray.h"
66			#include "TRefArray.h"
67			#include "TLorentzVector.h"
68			#include "TVector3.h"
69
70			using namespace edm;
71			using namespace std;
72			using namespace reco;
73
74			//
75			// class declaration
76			//
77
78			class MPIntuple : public edm::EDAnalyzer {
79			public:
80			explicit MPIntuple(const edm::ParameterSet&);
81			~MPIntuple();
82
83
84			private:
85			virtual void beginJob() ;
86			virtual void analyze(const edm::Event&, const edm::EventSetup&);
87			virtual void endJob() ;
88
89	andrey	1.2	bool triggerDecision(edm::Handle<edm::TriggerResults> &hltR, int iTrigger);
90	devildog	1.11	double sumPtSquared(const Vertex & v);
91	andrey	1.2
92	naodell	1.1	// ----------member data ---------------------------
93
94			edm::InputTag PFJetHandle_;
95			edm::InputTag GenJetHandle_;
96			edm::InputTag PrimaryVtxHandle_;
97	naodell	1.4	edm::InputTag triggerResultsTag_;
98	naodell	1.1	// Counting variables //
99
100	naodell	1.8	int eventNumber, runNumber, lumiSection, bunchCross;
101	naodell	1.18	double ptHat, qScale;
102
103	naodell	1.6	TTree* sTree;
104	naodell	1.1	TFile* ntupleFile;
105
106	naodell	1.4	TClonesArray* jet_ak5PF;
107	naodell	1.1	TClonesArray* jetP4_ak5Gen;
108	naodell	1.5	TClonesArray* primaryVtx;
109	naodell	1.1
110			bool doGenJets_;
111			bool doPFJets_;
112	naodell	1.7	bool triggerHLT_;
113	naodell	1.8	bool isRealData;
114	naodell	1.1	string rootfilename;
115
116	andrey	1.2	//Triggers
117	naodell	1.4	string hlTriggerResults_, hltName_, triggerName_;
118			TriggerNames triggerNames;
119	andrey	1.2	HLTConfigProvider hltConfig_;
120	naodell	1.4	vector<string> hlNames;
121	naodell	1.5	unsigned int triggerStatus;
122	andrey	1.2
123	naodell	1.15	//Histograms
124	naodell	1.16	TH1F * h1_nAssociatedTracks;
125	naodell	1.15	TH1F * h1_trackZ;
126	naodell	1.16	TH1F * h1_trackDxy;
127			TH1F * h1_allTrackDeltaZ_PV;
128			TH1F * h1_allTrackDeltaZ_LJ;
129			TH2F * h2_nAssTracksVsJetPt;
130	naodell	1.15	TProfile * p1_nVtcs;
131
132
133	naodell	1.1	};
134
135	naodell	1.18	MPIntuple::MPIntuple(const edm::ParameterSet& iConfig)
136	naodell	1.1	{
137	naodell	1.18	PFJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PFJetTag");
138			GenJetHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("GenJetTag");
139			PrimaryVtxHandle_ = iConfig.getUntrackedParameter<edm::InputTag>("PrimaryVtxTag");
140			doGenJets_ = iConfig.getUntrackedParameter<bool>("doGenJets");
141			doPFJets_ = iConfig.getUntrackedParameter<bool>("doPFJets");
142			triggerHLT_ = iConfig.getUntrackedParameter<bool>("triggerHLT");
143			rootfilename = iConfig.getUntrackedParameter<string>("rootfilename");
144			hlTriggerResults_ = iConfig.getUntrackedParameter<string>("HLTriggerResults","TriggerResults");
145	naodell	1.20	hltName_ = iConfig.getUntrackedParameter<string>("hltName","HLT");
146	naodell	1.1	}
147
148			MPIntuple::~MPIntuple()
149			{
150
151			}
152
153			//
154			// member functions
155			//
156
157			// ------------ method called to for each event ------------
158			void MPIntuple::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
159			{
160
161	naodell	1.6	eventNumber = iEvent.id().event();
162			runNumber = iEvent.id().run();
163	naodell	1.1	lumiSection = (unsigned int)iEvent.getLuminosityBlock().luminosityBlock();
164	naodell	1.8	bunchCross = (unsigned int)iEvent.bunchCrossing();
165			isRealData = iEvent.isRealData();
166
167	naodell	1.18
168	naodell	1.16	edm::Handle<reco::BeamSpot> beamSpotHandle;
169			iEvent.getByLabel("offlineBeamSpot", beamSpotHandle);
170			reco::BeamSpot vertexBeamSpot = *beamSpotHandle;
171
172			int pfCount = 0;
173			int genCount = 0;
174			int vtxCount = 0;
175			float ljVertexZ = -999;
176			float primaryVertexZ = -999;
177
178			//////////////////////////
179			//Get vertex information//
180			//////////////////////////
181
182			Handle<reco::VertexCollection> primaryVtcs;
183			iEvent.getByLabel(PrimaryVtxHandle_, primaryVtcs);
184
185			for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
186
187			reco::Vertex myVtx = reco::Vertex(*vtx_iter);
188			if(!myVtx.isValid() \|\| myVtx.isFake()) continue;
189			TCPrimaryVtx* vtxCon = new ((*primaryVtx)[vtxCount]) TCPrimaryVtx;
190
191	naodell	1.20	cout<<myVtx.nTracks()<<endl;
192
193	naodell	1.16	vtxCon->SetPosition(myVtx.x(), myVtx.y(), myVtx.z());
194			vtxCon->SetNDof(myVtx.ndof());
195			vtxCon->SetChi2(myVtx.chi2());
196			vtxCon->SetNTrks(myVtx.tracksSize());
197			vtxCon->SetSumPt2Trks(sumPtSquared(myVtx));
198
199			if(vtxCount == 0) primaryVertexZ = myVtx.z();
200
201			++vtxCount;
202
203			}
204
205			p1_nVtcs->Fill(runNumber, vtxCount);
206
207			///////////////////////
208			//get jet information//
209			///////////////////////
210	naodell	1.1
211			if(doPFJets_){
212	naodell	1.4
213	naodell	1.7	edm::Handle<reco::JetTagCollection> bTagHandle1;
214			iEvent.getByLabel("trackCountingHighEffBJetTags", bTagHandle1);
215			const reco::JetTagCollection & bTags1 = *(bTagHandle1.product());
216			reco::JetTagCollection::const_iterator jet_it_1;
217
218			edm::Handle<reco::JetTagCollection> bTagHandle2;
219			iEvent.getByLabel("trackCountingHighPurBJetTags", bTagHandle2);
220			const reco::JetTagCollection & bTags2 = *(bTagHandle2.product());
221			reco::JetTagCollection::const_iterator jet_it_2;
222
223	naodell	1.14
224	naodell	1.4	const JetCorrector* correctorL2 = JetCorrector::getJetCorrector ("ak5PFL2Relative",iSetup);
225			const JetCorrector* correctorL3 = JetCorrector::getJetCorrector ("ak5PFL3Absolute",iSetup);
226
227	naodell	1.1	Handle<reco::PFJetCollection> PFJets;
228			iEvent.getByLabel(PFJetHandle_, PFJets);
229	naodell	1.16
230	naodell	1.1	for(PFJetCollection::const_iterator jet_iter = PFJets->begin(); jet_iter!= PFJets->end(); ++jet_iter){
231
232			reco::PFJet myJet = reco::PFJet(*jet_iter);
233			if(myJet.et() > 5){
234	naodell	1.7
235			for(jet_it_1 = bTags1.begin(); jet_it_1 != bTags1.end(); jet_it_1++) {
236			if( (fabs(jet_it_1->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_1->first->phi(),myJet.phi()) < .005)) break;
237			}
238
239			for(jet_it_2 = bTags2.begin(); jet_it_2 != bTags2.end(); jet_it_2++) {
240			if( (fabs(jet_it_2->first->eta() - myJet.eta()) < .005) && (deltaPhi(jet_it_2->first->phi(),myJet.phi()) < .005)) break;
241			}
242	naodell	1.6
243	naodell	1.16	TCJet* jetCon = new ((*jet_ak5PF)[pfCount]) TCJet;
244	naodell	1.6
245			jetCon->SetP4(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
246	naodell	1.15	jetCon->SetVtx(-999.0, -999.0, -999.0);
247	naodell	1.4
248	naodell	1.6	jetCon->SetChHadFrac(myJet.chargedHadronEnergyFraction());
249			jetCon->SetNeuHadFrac(myJet.neutralHadronEnergyFraction());
250			jetCon->SetChEmFrac(myJet.chargedEmEnergyFraction());
251			jetCon->SetNeuEmFrac(myJet.neutralEmEnergyFraction());
252	naodell	1.4
253	naodell	1.6	jetCon->SetNumConstit(myJet.chargedMultiplicity() + myJet.neutralMultiplicity());
254			jetCon->SetNumChPart(myJet.chargedMultiplicity());
255	naodell	1.4
256	naodell	1.6	jetCon->SetNumChPart(myJet.chargedMultiplicity());
257	naodell	1.4
258	naodell	1.7	if(jet_it_2 != bTags2.end()) jetCon->SetBDiscrTrkCountHiPure(jet_it_2->second);
259			if(jet_it_1 != bTags1.end()) jetCon->SetBDiscrTrkCountHiEff(jet_it_1->second);
260
261			float scale2 = correctorL2->correction(myJet);
262			myJet.scaleEnergy(scale2);
263			float scale3 = correctorL3->correction(myJet);
264			myJet.scaleEnergy(scale3);
265
266			//more corrections?
267
268	naodell	1.6	jetCon->SetJetCorr(2, scale2);
269			jetCon->SetJetCorr(3, scale3);
270	naodell	1.7
271	naodell	1.16	/////////////////////////
272			//get associated tracks//
273			/////////////////////////
274	naodell	1.15
275			const reco::TrackRefVector &tracks = myJet.getTrackRefs();
276
277	naodell	1.16	float sumTrackZ, sumTrackIP;
278			float meanTrackZ, meanTrackIP;
279	naodell	1.17	int nAssociatedTracks = -1;
280	naodell	1.16
281	naodell	1.17	sumTrackZ = sumTrackIP = 0;
282			meanTrackZ = meanTrackIP = -999;
283	naodell	1.15
284	naodell	1.17	if(myJet.pt() > 20 && fabs(myJet.eta()) < 2.4){
285
286			nAssociatedTracks = 0;
287	naodell	1.15
288			for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
289
290	naodell	1.21	const reco::Track &myJetTrack = **iTrack;
291			//const reco::TrackRef &myTrackRef = *iTrack;
292	naodell	1.20
293	naodell	1.21	sumTrackZ += myJetTrack.vz();
294			sumTrackIP += myJetTrack.dxy(vertexBeamSpot.position());
295	naodell	1.15	++nAssociatedTracks;
296	naodell	1.20
297	naodell	1.21	for(VertexCollection::const_iterator vtx_iter = primaryVtcs->begin(); vtx_iter!= primaryVtcs->end(); ++vtx_iter){
298
299			reco::Vertex myVtx = reco::Vertex(*vtx_iter);
300			if(!myVtx.isValid() \|\| myVtx.isFake()) continue;
301
302			for(Vertex::trackRef_iterator iTrackRef = myVtx.tracks_begin(); iTrackRef != myVtx.tracks_end(); ++iTrackRef){
303			const edm::RefToBase<reco::Track> &myTrackRef = *iTrackRef;
304	naodell	1.20
305	naodell	1.21	if(myTrackRef.isAvailable()){
306	naodell	1.20
307	naodell	1.21	const reco::Track &myVertexTrack = *myTrackRef.get();
308
309			cout<<"Vertex track z = "<<myVertexTrack.vz()<<endl;
310			cout<<"Jet track z = "<<myJetTrack.vz()<<endl;
311
312			}
313			}
314			}
315	naodell	1.15	}
316	naodell	1.16
317	naodell	1.15	meanTrackZ = sumTrackZ/nAssociatedTracks;
318	naodell	1.16	h1_nAssociatedTracks->Fill(nAssociatedTracks);
319	naodell	1.17	h2_nAssTracksVsJetPt->Fill(myJet.pt(), nAssociatedTracks);
320	naodell	1.16
321	naodell	1.15	for(TrackRefVector::const_iterator iTrack = tracks.begin(); iTrack != tracks.end(); ++iTrack){
322	naodell	1.16
323	naodell	1.21	const reco::Track &myJetTrack = **iTrack;
324	naodell	1.16
325	naodell	1.21	if(nAssociatedTracks == 3) h1_trackZ->Fill(meanTrackZ - myJetTrack.vz());
326			h1_trackDxy->Fill(myJetTrack.dxy(vertexBeamSpot.position()));
327	naodell	1.16
328			}
329
330			if(pfCount == 0){
331
332			ljVertexZ = meanTrackZ;
333
334			}else{
335
336			h1_allTrackDeltaZ_LJ->Fill(ljVertexZ - meanTrackZ);
337
338	naodell	1.15	}
339	naodell	1.16
340			h1_allTrackDeltaZ_PV->Fill(primaryVertexZ - meanTrackZ);
341
342	naodell	1.15	}
343	naodell	1.16
344			jetCon->SetVtx(0, 0, meanTrackZ);
345
346			++pfCount;
347	naodell	1.4	}
348			}
349	naodell	1.1	}
350
351	naodell	1.10	if(!isRealData){
352	naodell	1.1
353	naodell	1.18	Handle<GenEventInfoProduct> GenInfoHandle;
354			iEvent.getByLabel("generator", GenInfoHandle);
355
356	naodell	1.1	Handle<reco::GenJetCollection> GenJets;
357			iEvent.getByLabel(GenJetHandle_, GenJets);
358	naodell	1.18
359			ptHat = qScale = -1;
360
361			if(GenInfoHandle.isValid()){
362
363			qScale = GenInfoHandle->qScale();
364			ptHat = (GenInfoHandle->hasBinningValues() ? GenInfoHandle->binningValues()[0] : 0.0);
365
366			}
367	naodell	1.16
368	naodell	1.1	for(GenJetCollection::const_iterator jet_iter = GenJets->begin(); jet_iter!= GenJets->end(); ++jet_iter){
369
370			reco::GenJet myJet = reco::GenJet(*jet_iter);
371
372	naodell	1.4	if(myJet.pt() > 5){
373	naodell	1.1
374	naodell	1.16	new ((*jetP4_ak5Gen)[genCount]) TLorentzVector(myJet.px(), myJet.py(), myJet.pz(), myJet.energy());
375	naodell	1.1
376	naodell	1.16	++genCount;
377	naodell	1.1
378			}
379	naodell	1.13	}
380	naodell	1.1	}
381	naodell	1.10
382	naodell	1.16	///////////////////////////
383			//get trigger information//
384			///////////////////////////
385	andrey	1.2
386	naodell	1.7	if(triggerHLT_){
387	andrey	1.2
388	naodell	1.7	edm::Handle<TriggerResults> hltR;
389			triggerResultsTag_ = InputTag(hlTriggerResults_,"",hltName_);
390			iEvent.getByLabel(triggerResultsTag_,hltR);
391
392			const TriggerNames & triggerNames = iEvent.triggerNames(*hltR);
393			hlNames=triggerNames.triggerNames();
394
395	naodell	1.9	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"};
396	naodell	1.7
397			bool triggerPassed = false;
398			triggerStatus = 0x0;
399	naodell	1.4
400	naodell	1.7	for (uint i=0; i<hlNames.size(); ++i) {
401
402			triggerPassed = triggerDecision(hltR, i);
403	naodell	1.4
404	naodell	1.7	if(triggerPassed){
405	andrey	1.3
406	naodell	1.7	for (uint j = 0; j != 32; ++j){
407
408			if (hlNames[i] == MPI_TriggerNames[j])
409			{
410	naodell	1.16	//cout<<"trigger name: "<<hlNames[i]<<"list: "<<dec<<j+1<<endl;
411	naodell	1.7	triggerStatus \|= 0x01 << j;
412
413			}
414			}
415	naodell	1.4	}
416	naodell	1.7	}
417			}
418	naodell	1.16	//cout<< "total status: "<<hex<<triggerStatus<<endl;
419	naodell	1.18
420	naodell	1.14
421
422	naodell	1.16	if((pfCount > 3 \|\| genCount > 3) && vtxCount > 0) sTree -> Fill();
423	naodell	1.15
424	naodell	1.4	jet_ak5PF->Clear();
425	naodell	1.5	primaryVtx->Clear();
426	naodell	1.4	jetP4_ak5Gen->Clear();
427	andrey	1.2
428	naodell	1.1	}
429
430
431			// ------------ method called once each job just before starting event loop ------------
432			void MPIntuple::beginJob()
433			{
434
435			ntupleFile = new TFile(rootfilename.c_str(), "RECREATE");
436	naodell	1.5	sTree = new TTree("mpiTree", "Tree for Jets");
437	naodell	1.15
438	naodell	1.16	h1_trackZ = new TH1F("h1_trackZ", "#Delta z for associated tracks", 60, -0.3, 0.3);
439			h1_nAssociatedTracks = new TH1F("h1_nAssociatedTracks", "Number of tracks associated to jet", 20, -0.5, 19.5);
440			h1_trackDxy = new TH1F("h1_trackDxy", "dxy for all associated tracks", 50, -0.4, 0.4);
441			h1_allTrackDeltaZ_PV = new TH1F("h1_allTrackDeltaZ_PV", "#Delta z between jet and primary vertex", 60, -0.3, 0.3);
442			h1_allTrackDeltaZ_LJ = new TH1F("h1_allTrackDeltaZ_LJ", "#Delta z between leading jet and other jet vertices", 60, -0.3, 0.3);
443	naodell	1.17	h2_nAssTracksVsJetPt = new TH2F("h2_nAssTracksVsJetPt", "nTracks vs Jet p_{T}", 50, 0, 100, 20, -0.5, 19.5);
444	naodell	1.15	p1_nVtcs = new TProfile("p1_nVtcs", "<N> per run", 5200.0, 132300.0, 137500.0, 0.0, 6.0);
445
446	naodell	1.4	jet_ak5PF = new TClonesArray("TCJet");
447	naodell	1.1	jetP4_ak5Gen = new TClonesArray("TLorentzVector");
448	naodell	1.5	primaryVtx = new TClonesArray("TCPrimaryVtx");
449	naodell	1.1
450	naodell	1.15
451	naodell	1.4	sTree->Branch("jet_ak5PF",&jet_ak5PF, 6400, 0);
452	naodell	1.1	sTree->Branch("jetP4_ak5Gen",&jetP4_ak5Gen, 6400, 0);
453	naodell	1.5	sTree->Branch("primaryVtx",&primaryVtx, 6400, 0);
454	naodell	1.1
455			sTree->Branch("eventNumber",&eventNumber, "eventNumber/I");
456			sTree->Branch("runNumber",&runNumber, "runNumber/I");
457			sTree->Branch("lumiSection",&lumiSection, "lumiSection/I");
458	naodell	1.6	sTree->Branch("triggerStatus",&triggerStatus, "triggerStatus/i");
459	naodell	1.8	sTree->Branch("isRealData",&isRealData, "isRealData/i");
460			sTree->Branch("bunchCross",&bunchCross, "bunchCross/i");
461	naodell	1.18	sTree->Branch("ptHat",&ptHat, "ptHat/d");
462	naodell	1.1
463			}
464
465			// ------------ method called once each job just after ending the event loop ------------
466			void MPIntuple::endJob()
467			{
468	naodell	1.15
469	naodell	1.17	ntupleFile->cd();
470
471	naodell	1.15	h1_trackZ->Write();
472	naodell	1.16	h1_nAssociatedTracks->Write();
473			h1_trackDxy->Write();
474			h1_allTrackDeltaZ_PV->Write();
475			h1_allTrackDeltaZ_LJ->Write();
476			h2_nAssTracksVsJetPt->Write();
477	naodell	1.15	p1_nVtcs->Write();
478
479	naodell	1.1	ntupleFile->Write();
480			ntupleFile->Close();
481
482			}
483
484
485	andrey	1.2	bool MPIntuple::triggerDecision(edm::Handle<edm::TriggerResults> &hltR, int iTrigger)
486			{
487			bool triggerPassed = false;
488			if(hltR->wasrun(iTrigger) &&
489			hltR->accept(iTrigger) &&
490			!hltR->error(iTrigger) ){
491			triggerPassed = true;
492			}
493			return triggerPassed;
494	naodell	1.16	}
495
496			double MPIntuple::sumPtSquared(const Vertex & v)
497			{
498			double sum = 0.;
499			double pT;
500			for (Vertex::trackRef_iterator it = v.tracks_begin(); it != v.tracks_end(); it++) {
501			pT = (**it).pt();
502			double epT=(**it).ptError(); pT=pT>epT ? pT-epT : 0;
503
504			sum += pT*pT;
505	andrey	1.2	}
506	naodell	1.16	return sum;
507			}
508	andrey	1.2
509	naodell	1.1	//define this as a plug-in
510			DEFINE_FWK_MODULE(MPIntuple);