HcalIsoTrkAnalyzer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:  HcalCalibAlgos  
// Class:    HcalIsoTrkAnalyzer 

// 
//
// Original Authors: Andrey Pozdnyakov, Sergey Petrushanko,
//                   Grigory Safronov, Olga Kodolova
//         Created:  Thu Jul 12 18:12:19 CEST 2007
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "FWCore/Utilities/interface/Exception.h"

#include "TrackingTools/TrackAssociator/interface/TrackDetMatchInfo.h"
#include "TrackingTools/TrackAssociator/interface/TrackAssociatorParameters.h"
#include "TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h"

#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"
#include "DataFormats/HcalDetId/interface/HcalSubdetector.h" 

#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h" 
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloEventSetup/interface/CaloTopologyRecord.h"
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h"

#include "DataFormats/HcalDetId/interface/HcalDetId.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "TrackPropagation/SteppingHelixPropagator/interface/SteppingHelixPropagator.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "DataFormats/HcalIsolatedTrack/interface/IsolatedPixelTrackCandidate.h"
#include "DataFormats/HcalIsolatedTrack/interface/IsolatedPixelTrackCandidateFwd.h"
#include "Calibration/Tools/interface/MinL3AlgoUniv.h"
#include "Calibration/HcalCalibAlgos/src/TCell.h"
#include "CondFormats/HcalObjects/interface/HcalRespCorrs.h"
#include "CondFormats/DataRecord/interface/HcalRespCorrsRcd.h"

#include "Calibration/HcalCalibAlgos/interface/CommonUsefulStuff.h"


#include "TROOT.h"
#include "TH1.h"
#include "TH2.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 <iostream>
#include <fstream>

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

//
// class decleration
//

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

  //double getDistInPlaneSimple(const GlobalPoint caloPoint, const GlobalPoint rechitPoint);

private:
  virtual void beginJob() override ;
  virtual void analyze(const edm::Event&, const edm::EventSetup&) override;
  virtual void endJob() override ;
  
  // ----------member data ---------------------------
  
 
  TrackDetectorAssociator trackAssociator_;
  TrackAssociatorParameters parameters_;

  const CaloGeometry* geo;
  edm::EDGetTokenT<HBHERecHitCollection> tok_hbhe_;
  edm::EDGetTokenT<HORecHitCollection> tok_ho_;
  edm::EDGetTokenT<EcalRecHitCollection> tok_ecal_;
  edm::EDGetTokenT<reco::IsolatedPixelTrackCandidateCollection> tok_track_;
  edm::EDGetTokenT<reco::TrackCollection> tok_track1_;

  std::string m_inputTrackLabel;
  std::string m_ecalLabel;
  std::string m_ebInstance;
  std::string m_eeInstance;
  std::string m_hcalLabel;
  int m_histoFlag;

  double associationConeSize_, calibrationConeSize_;
  string AxB_;
  bool allowMissingInputs_;
  string outputFileName_;

  double trackEta, trackPhi; 
  double rvert;
  
  
  int nIterations, MinNTrackHitsBarrel,MinNTECHitsEndcap;
  float eventWeight;
  double energyMinIso, energyMaxIso;
  double energyECALmip, maxPNear;
  double hottestHitDistance, EcalCone, EcalConeOuter;

  TFile* rootFile;
  TTree* tree;
  Float_t targetE;
  UInt_t numberOfCells;
  UInt_t cell;
  Float_t cellEnergy; 
  
  TClonesArray* cells;
  TRefArray* cells3x3;
  TRefArray* cellsPF;
  UInt_t  eventNumber;
  UInt_t  runNumber;
  Int_t   iEtaHit;
  UInt_t  iPhiHit;

  Float_t xTrkEcal, yTrkEcal, zTrkEcal;
  Float_t xTrkHcal, yTrkHcal, zTrkHcal;
  Float_t PxTrkHcal, PyTrkHcal, PzTrkHcal;

  Float_t emEnergy, emRingEnergy;

  //    TLorentzVector* exampleP4; 
  TLorentzVector* tagJetP4; // dijet
  TLorentzVector* probeJetP4; // dijet
  Float_t etVetoJet; // dijet
  Float_t tagJetEmFrac; // dijet
  Float_t probeJetEmFrac; // dijet
  
  std::ofstream input_to_L3;
  
};


HcalIsoTrkAnalyzer::HcalIsoTrkAnalyzer(const edm::ParameterSet& iConfig)
{
  
  m_ecalLabel = iConfig.getUntrackedParameter<std::string> ("ecalRecHitsLabel","ecalRecHit");
  m_ebInstance = iConfig.getUntrackedParameter<std::string> ("ebRecHitsInstance","EcalRecHitsEB");
  m_eeInstance = iConfig.getUntrackedParameter<std::string> ("eeRecHitsInstance","EcalRecHitsEE");
  m_hcalLabel = iConfig.getUntrackedParameter<std::string> ("hcalRecHitsLabel","hbhereco");
 
  tok_hbhe_ = consumes<HBHERecHitCollection>(iConfig.getParameter<edm::InputTag>("hbheInput"));
  tok_ho_ = consumes<HORecHitCollection>(iConfig.getParameter<edm::InputTag>("hoInput"));
  tok_ecal_ = consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("eInput"));
  tok_track_ = consumes<reco::IsolatedPixelTrackCandidateCollection>(iConfig.getParameter<edm::InputTag>("HcalIsolTrackInput"));
  tok_track1_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("trackInput"));
  associationConeSize_=iConfig.getParameter<double>("associationConeSize");
  allowMissingInputs_=iConfig.getUntrackedParameter<bool>("allowMissingInputs",true);
  outputFileName_=iConfig.getParameter<std::string>("outputFileName");

  AxB_=iConfig.getParameter<std::string>("AxB");
  calibrationConeSize_=iConfig.getParameter<double>("calibrationConeSize");

  nIterations = iConfig.getParameter<int>("noOfIterations");
  eventWeight = iConfig.getParameter<double>("eventWeight");
  energyMinIso = iConfig.getParameter<double>("energyMinIso");
  energyMaxIso = iConfig.getParameter<double>("energyMaxIso");
  energyECALmip = iConfig.getParameter<double>("energyECALmip");
  maxPNear = iConfig.getParameter<double>("maxPNear");
  MinNTrackHitsBarrel =  iConfig.getParameter<int>("MinNTrackHitsBarrel");
  MinNTECHitsEndcap =  iConfig.getParameter<int>("MinNTECHitsEndcap");
  hottestHitDistance = iConfig.getParameter<double>("hottestHitDistance");
  EcalCone = iConfig.getParameter<double>("EcalCone");
  EcalConeOuter = iConfig.getParameter<double>("EcalConeOuter");

  edm::ParameterSet parameters = iConfig.getParameter<edm::ParameterSet>("TrackAssociatorParameters");
  edm::ConsumesCollector iC = consumesCollector();
  parameters_.loadParameters( parameters, iC );
  trackAssociator_.useDefaultPropagator();

}

HcalIsoTrkAnalyzer::~HcalIsoTrkAnalyzer()
{
}

// ------------ method called to for each event  ------------
void
HcalIsoTrkAnalyzer::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  using namespace edm;
  using namespace std;

  vector<float> rawEnergyVec;
  vector<int> detiphi;
  vector<int> detieta;
  vector<int> i3i5;
  vector<HcalDetId> detidvec;
  float calEnergy;

  edm::Handle<reco::TrackCollection> isoProdTracks;
  iEvent.getByToken(tok_track1_,isoProdTracks);  

  edm::Handle<reco::IsolatedPixelTrackCandidateCollection> isoPixelTracks;
  iEvent.getByToken(tok_track_,isoPixelTracks);
    
  edm::Handle<EcalRecHitCollection> ecal;
  iEvent.getByToken(tok_ecal_,ecal);
  const EcalRecHitCollection Hitecal = *(ecal.product());
    
  edm::Handle<HBHERecHitCollection> hbhe;
  iEvent.getByToken(tok_hbhe_,hbhe);
  const HBHERecHitCollection Hithbhe = *(hbhe.product());

  edm::ESHandle<CaloGeometry> pG;
  iSetup.get<CaloGeometryRecord>().get(pG);
  geo = pG.product();

  const CaloSubdetectorGeometry* gHcal = geo->getSubdetectorGeometry(DetId::Hcal,HcalBarrel);
  //Note: even though it says HcalBarrel, we actually get the whole Hcal detector geometry!


// rof 16.05.2008 start: include the possibility for recalibration (use "recalibrate" label for safety)
/*
  edm::ESHandle <HcalRespCorrs> recalibCorrs;
  iSetup.get<HcalRespCorrsRcd>().get("recalibrate",recalibCorrs);
  const HcalRespCorrs* myRecalib = recalibCorrs.product();
*/
// rof end

  parameters_.useEcal = true;
  parameters_.useHcal = true;
  parameters_.useCalo = false;
  parameters_.useMuon = false;
  parameters_.dREcal = 0.5;
  parameters_.dRHcal = 0.6;  

  if (isoPixelTracks->size()==0) return;

  for (reco::TrackCollection::const_iterator trit=isoProdTracks->begin(); trit!=isoProdTracks->end(); trit++)
    {
      reco::IsolatedPixelTrackCandidateCollection::const_iterator isoMatched=isoPixelTracks->begin();
      bool matched=false;

      //Note: this part needs to be changed: obtain isoPixelTracks from isoProdTracks (should be a way)
      for (reco::IsolatedPixelTrackCandidateCollection::const_iterator it = isoPixelTracks->begin(); it!=isoPixelTracks->end(); it++)
    { 

      if (abs((trit->pt() - it->pt())/it->pt()) < 0.005 && abs(trit->eta() - it->eta()) < 0.01) 
        {
          isoMatched=it;
          matched=true;
          break;
        }
    }
      if (!matched) continue;
      
      if (trit->hitPattern().numberOfValidHits()<MinNTrackHitsBarrel) continue;
      if (fabs(trit->eta())>1.47&&trit->hitPattern().numberOfValidStripTECHits()<MinNTECHitsEndcap) continue;
      

      calEnergy = sqrt(trit->px()*trit->px()+trit->py()*trit->py()+trit->pz()*trit->pz()+0.14*0.14);

      trackEta = trit->eta();
      trackPhi = trit->phi();
      
      double corrHCAL = 1.; //another possibility for correction.  - why?

      //      cout << endl << " ISO TRACK E = "<< calEnergy << " ETA = " << trackEta<< " PHI = " << trackPhi <<  " Correction " <<  corrHCAL<< endl;
      
      rvert = sqrt(trit->vx()*trit->vx()+trit->vy()*trit->vy()+trit->vz()*trit->vz());      
      
      //Associate track with a calorimeter
      TrackDetMatchInfo info = trackAssociator_.associate(iEvent, iSetup,trackAssociator_.getFreeTrajectoryState(iSetup, *trit),parameters_);

        
      xTrkHcal=info.trkGlobPosAtHcal.x();
      yTrkHcal=info.trkGlobPosAtHcal.y();
      zTrkHcal=info.trkGlobPosAtHcal.z();
      
      xTrkEcal=info.trkGlobPosAtEcal.x();
      yTrkEcal=info.trkGlobPosAtEcal.y();
      zTrkEcal=info.trkGlobPosAtEcal.z();
      
      if (xTrkEcal==0 && yTrkEcal==0&& zTrkEcal==0) {cout<<"zero point at Ecal"<<endl; continue;}
      if (xTrkHcal==0 && yTrkHcal==0&& zTrkHcal==0) {cout<<"zero point at Hcal"<<endl; continue;}   
      
      GlobalVector trackMomAtHcal = info.trkMomAtHcal;
    
      PxTrkHcal = trackMomAtHcal.x();
      PyTrkHcal = trackMomAtHcal.y();
      PzTrkHcal = trackMomAtHcal.z();
      
      //PxTrkHcal=0;
      //PyTrkHcal=0;
      //PzTrkHcal=0;

      GlobalPoint gPointEcal(xTrkEcal,yTrkEcal,zTrkEcal);
      GlobalPoint gPointHcal(xTrkHcal,yTrkHcal,zTrkHcal);
      
      //        emEnergy = isoMatched->energyIn();
      emEnergy =  ecalEnergyInCone(gPointEcal, EcalCone, Hitecal, geo);
      
      emRingEnergy = ecalEnergyInCone(gPointEcal, EcalConeOuter, Hitecal, geo) - ecalEnergyInCone(gPointEcal, EcalCone, Hitecal, geo);

      
    int iphitrue = -10;
    int ietatrue = 100;
    
    const HcalDetId tempId = gHcal->getClosestCell(gPointHcal);
    ietatrue = tempId.ieta();
    iphitrue = tempId.iphi();


      //container for used recHits
    std::vector<DetId> usedHits; 
      //      
      //clear usedHits
      usedHits.clear();


      // Find Hcal RecHit with maximum energy and collect other information
      MaxHit_struct MaxHit;
      MaxHit.hitenergy=-100;
      

      for (HBHERecHitCollection::const_iterator hhit=Hithbhe.begin(); hhit!=Hithbhe.end(); hhit++) 
    {
      
      //check that this hit was not considered before and push it into usedHits
      bool hitIsUsed=false;
      DetId hitId=hhit->id();
      for (uint32_t i=0; i<usedHits.size(); i++)
        {
          if (usedHits[i]==hitId) hitIsUsed=true;
        }
      if (hitIsUsed) continue;
      usedHits.push_back(hitId);
      //

      // rof 16.05.2008 start: include the possibility for recalibration
      float recal = 1;
      // rof end

      GlobalPoint pos = geo->getPosition(hhit->detid());

      int iphihitm  = (hhit->id()).iphi();
      int ietahitm  = (hhit->id()).ieta();
      int depthhit = (hhit->id()).depth();
      double enehit = hhit->energy() * recal;
      
      if (depthhit!=1) continue;
       

      double distAtHcal =  getDistInPlaneTrackDir(gPointHcal, trackMomAtHcal, pos);
      //double distAtHcal =  getDistInPlaneSimple(gPointHcal, pos);

      //if(dr<associationConeSize_)      
      if(distAtHcal < associationConeSize_) 
        {
          for (HBHERecHitCollection::const_iterator hhit2=Hithbhe.begin(); hhit2!=Hithbhe.end(); hhit2++) 
        {
          int iphihitm2  = (hhit2->id()).iphi();
          int ietahitm2  = (hhit2->id()).ieta();
          int depthhit2 = (hhit2->id()).depth();
          double enehit2 = hhit2->energy() * recal;
          
          if ( iphihitm==iphihitm2 && ietahitm==ietahitm2  && depthhit!=depthhit2){
            
            enehit = enehit+enehit2;
          }
        }
          
          if(enehit > MaxHit.hitenergy) 
        {
          MaxHit.hitenergy =  enehit;
          MaxHit.ietahitm   = (hhit->id()).ieta();
          MaxHit.iphihitm   = (hhit->id()).iphi();
          MaxHit.depthhit  = (hhit->id()).depth();
          MaxHit.dr   = distAtHcal;

          MaxHit.posMax = geo->getPosition(hhit->detid());
          
        }
        }
    }
      

      Bool_t passCuts = kFALSE;
      if(calEnergy > energyMinIso && calEnergy < energyMaxIso && emEnergy < energyECALmip && 
     isoMatched->maxPtPxl() < maxPNear && abs(MaxHit.ietahitm)<30 && MaxHit.hitenergy > 0. && 
          MaxHit.dr < hottestHitDistance && emRingEnergy<8.){ passCuts = kTRUE; }
      
      
      if(AxB_=="5x5" || AxB_=="3x3" || AxB_=="7x7"|| AxB_=="Cone")
    {

      //clear usedHits
      usedHits.clear();
      //

      for (HBHERecHitCollection::const_iterator hhit=Hithbhe.begin(); hhit!=Hithbhe.end(); hhit++) 
        {

          //check that this hit was not considered before and push it into usedHits
          bool hitIsUsed=false;
          DetId hitId=hhit->id();
          for (uint32_t i=0; i<usedHits.size(); i++)
        {
          if (usedHits[i]==hitId) hitIsUsed=true;
        }
          if (hitIsUsed) 
        {
          //cout<<"Hit is Used!   ieta: "<< (hhit->id()).ieta()<<"  iphi: "<<(hhit->id()).iphi()<<"  depth: "<<(hhit->id()).depth()<<"  hashed_index: "<<hitHashedIndex<<endl;
          continue;}
          
          usedHits.push_back(hitId);
          //

          int DIETA = 100;
          if(MaxHit.ietahitm*(hhit->id()).ieta()>0)
        {
          DIETA = MaxHit.ietahitm - (hhit->id()).ieta();
        }
          if(MaxHit.ietahitm*(hhit->id()).ieta()<0)
        {
          DIETA = MaxHit.ietahitm - (hhit->id()).ieta();
          DIETA = DIETA>0 ? DIETA-1 : DIETA+1; 
        }
          
          int DIPHI = abs(MaxHit.iphihitm - (hhit->id()).iphi());
          DIPHI = DIPHI>36 ? 72-DIPHI : DIPHI;
          /*AP DIPHI = DIPHI<-36 ? 72+DIPHI : DIPHI; */
          
          int numbercell=0;
          if(AxB_=="3x3") numbercell = 1;
          if(AxB_=="5x5") numbercell = 2;
          if(AxB_=="7x7") numbercell = 3;
          if(AxB_=="Cone") numbercell = 1000;
          
          if( abs(DIETA)<=numbercell && (abs(DIPHI)<=numbercell || ( abs(MaxHit.ietahitm)>=20 && abs(DIPHI)<=numbercell+1)) )  {
        
        // rof 16.05.2008 start: include the possibility for recalibration
        float recal = 1;
        
            int iii3i5 = 0;
            
        const GlobalPoint pos2 = geo->getPosition(hhit->detid());       
        
        if(passCuts){
          
          if(AxB_=="5x5" || AxB_=="3x3" || AxB_=="7x7") {
            
            rawEnergyVec.push_back(hhit->energy() * recal * corrHCAL);
            detidvec.push_back(hhit->id());
            detiphi.push_back((hhit->id()).iphi());
            detieta.push_back((hhit->id()).ieta());
            i3i5.push_back(iii3i5);
            
          }
          
          if (AxB_=="Cone" && getDistInPlaneTrackDir(gPointHcal, trackMomAtHcal, pos2) < calibrationConeSize_) {
            //if (AxB_=="Cone" && getDistInPlaneSimple(gPointHcal,pos2) < calibrationConeSize_) {
            
            rawEnergyVec.push_back(hhit->energy() * recal * corrHCAL);
            detidvec.push_back(hhit->id());
            detiphi.push_back((hhit->id()).iphi());
            detieta.push_back((hhit->id()).ieta());
            i3i5.push_back(iii3i5);
            
                  }
          
        }
          }
        }
    }
      
      if(AxB_!="3x3" && AxB_!="5x5" && AxB_!="7x7" && AxB_!="Cone") LogWarning(" AxB ")<<"   Not supported: "<< AxB_;
      
      if(passCuts){
    
    input_to_L3 << rawEnergyVec.size() << "   " << calEnergy;
    
    
    for (unsigned int i=0; i<rawEnergyVec.size(); i++)
      {
        input_to_L3 << "   " << rawEnergyVec.at(i) << "   " << detidvec.at(i).rawId() ;    
        
      }
    input_to_L3 <<endl;
    
        eventNumber = iEvent.id().event();
        runNumber = iEvent.id().run();
        iEtaHit = ietatrue;
        iPhiHit = iphitrue;

    numberOfCells=rawEnergyVec.size();
    targetE = calEnergy;
        
        for (unsigned int ia=0; ia<numberOfCells; ++ia) {
      cellEnergy = rawEnergyVec.at(ia);
      cell = detidvec.at(ia).rawId();
          
      new((*cells)[ia])  TCell(cell, cellEnergy);
      

        } 
    
        tree->Fill();
        
        cells->Clear();   
    
      }
      
      rawEnergyVec.clear();
      detidvec.clear();
      detiphi.clear();
      detieta.clear();
      i3i5.clear();

      try {
    Handle<HORecHitCollection> ho;
    iEvent.getByToken(tok_ho_,ho);
    const HORecHitCollection Hitho = *(ho.product());
    
    //clear usedHits
    usedHits.clear();
    //
    
    for(HORecHitCollection::const_iterator hoItr=Hitho.begin(); hoItr!=Hitho.end(); hoItr++)
      {

        //check that this hit was not considered before and push it into usedHits
        bool hitIsUsed=false;
        DetId hitId=hoItr->id();
        for (uint32_t i=0; i<usedHits.size(); i++)
          {
        if (usedHits[i]==hitId) hitIsUsed=true;
          }
        if (hitIsUsed) continue;

        usedHits.push_back(hitId);

        /*AP
        GlobalPoint pos = geo->getPosition(hoItr->detid());
        double phihit = pos.phi();
        double etahit = pos.eta();
        
        int iphihitm = (hoItr->id()).iphi();
        int ietahitm = (hoItr->id()).ieta();
        int depthhit = (hoItr->id()).depth();
        
        double dphi = fabs(trackPhi - phihit); 
        if(dphi > 4.*atan(1.)) dphi = 8.*atan(1.) - dphi;
        double deta = fabs(trackEta - etahit); 
        double dr = sqrt(dphi*dphi + deta*deta);
        */
        
      }
      } catch (cms::Exception& e) { // can't find it!
    if (!allowMissingInputs_) throw e;
      }
    }
}
    
      

// ------------ method called once each job just before starting event loop  ------------
void 
HcalIsoTrkAnalyzer::beginJob()
{

  //  MyL3Algo = new MinL3AlgoUniv<HcalDetId>(eventWeight);

    input_to_L3.open("input_to_l3.txt");

    rootFile = new TFile("rootFile.root", "RECREATE");
    tree = new TTree("hcalCalibTree", "Tree for IsoTrack Calibration");  
    cells = new TClonesArray("TCell", 10000);
//    cells3x3 = new TRefArray;
//    cellsPF = new TRefArray;
//    exampleP4 = new TLorentzVector();
    tagJetP4 = new TLorentzVector(); // dijet
    probeJetP4 = new TLorentzVector();  // dijet   
    
    tree->Branch("cells", &cells, 64000, 0);
    tree->Branch("targetE", &targetE, "targetE/F");
    tree->Branch("emEnergy", &emEnergy, "emEnergy/F");

    tree->Branch("PxTrkHcal", &PxTrkHcal, "PxTrkHcal/F");
    tree->Branch("PyTrkHcal", &PyTrkHcal, "PyTrkHcal/F");
    tree->Branch("PzTrkHcal", &PzTrkHcal, "PzTrkHcal/F");


    tree->Branch("xTrkHcal", &xTrkHcal, "xTrkHcal/F");
    tree->Branch("yTrkHcal", &yTrkHcal, "yTrkHcal/F");
    tree->Branch("zTrkHcal", &zTrkHcal, "zTrkHcal/F");

    tree->Branch("iEtaHit", &iEtaHit, "iEtaHit/I");
    tree->Branch("iPhiHit", &iPhiHit, "iPhiHit/i");
    tree->Branch("eventNumber", &eventNumber, "eventNumber/i");
    tree->Branch("runNumber", &runNumber, "runNumber/i");

    tree->Branch("tagJetP4", "TLorentzVector", &tagJetP4); // dijet
    tree->Branch("probeJetP4", "TLorentzVector", &probeJetP4); // dijet
    tree->Branch("etVetoJet", &etVetoJet, "etVetoJet/F"); // dijet
    tree->Branch("tagJetEmFrac", &tagJetEmFrac,"tagJetEmFrac/F"); // dijet
    tree->Branch("probeJetEmFrac", &probeJetEmFrac,"probeJetEmFrac/F"); // dijet

}

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

    input_to_L3.close();

    rootFile->Write();
    rootFile->Close();

    if (cells) delete cells;
//    if (cells3x3) delete cells3x3;
//    if (cellsPF) delete cellsPF;
//    if (exampleP4) delete exampleP4;
    if (tagJetP4) delete tagJetP4; // dijet
    if (probeJetP4) delete probeJetP4; // dijet

}

//define this as a plug-in
DEFINE_FWK_MODULE(HcalIsoTrkAnalyzer);