IsoTrackCalibration.cc

Go to the documentation of this file.

//#define DebugLog

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

// Root objects
#include "TROOT.h"
#include "TSystem.h"
#include "TFile.h"
#include "TProfile.h"
#include "TDirectory.h"
#include "TTree.h"
#include "TH1F.h"
#include "TLorentzVector.h"
#include "TInterpreter.h"

//Tracks
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/HitPattern.h"
#include "DataFormats/TrackReco/interface/TrackBase.h"

// Vertices
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"

//Triggers
#include "DataFormats/Common/interface/TriggerResults.h"
#include "DataFormats/HLTReco/interface/TriggerEvent.h"
#include "DataFormats/HLTReco/interface/TriggerObject.h"
#include "DataFormats/L1GlobalTrigger/interface/L1GlobalTriggerReadoutRecord.h"
#include "DataFormats/Luminosity/interface/LumiDetails.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/Math/interface/deltaPhi.h"

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Common/interface/TriggerNames.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"

#include "Calibration/IsolatedParticles/interface/CaloPropagateTrack.h"
#include "Calibration/IsolatedParticles/interface/ChargeIsolation.h"
#include "Calibration/IsolatedParticles/interface/eCone.h"
#include "Calibration/IsolatedParticles/interface/eECALMatrix.h"
#include "Calibration/IsolatedParticles/interface/TrackSelection.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloEventSetup/interface/CaloTopologyRecord.h"
#include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"
#include "Geometry/CaloTopology/interface/HcalTopology.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h"
#include "CondFormats/DataRecord/interface/EcalChannelStatusRcd.h"
#include "Geometry/CaloTopology/interface/EcalTrigTowerConstituentsMap.h"
#include "RecoLocalCalo/EcalRecAlgos/interface/EcalSeverityLevelAlgo.h"
#include "RecoLocalCalo/EcalRecAlgos/interface/EcalSeverityLevelAlgoRcd.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"
#include "DataFormats/JetReco/interface/PFJet.h"

class IsoTrackCalibration : public edm::one::EDAnalyzer<edm::one::WatchRuns,edm::one::SharedResources> {

public:
  explicit IsoTrackCalibration(const edm::ParameterSet&);
  ~IsoTrackCalibration();
 
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

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

  edm::Service<TFileService>    fs_;
  HLTConfigProvider             hltConfig_;
  std::vector<std::string>      trigNames_, HLTNames_;
  int                           verbosity_;
  spr::trackSelectionParameters selectionParameters_;
  std::string                   theTrackQuality_;
  double                        a_mipR_, a_coneR_, a_charIsoR_;
  bool                          isMC_, isQCD_, isAOD_;
  double                        constTrackPt_, slopeTrackPt_;
  double                        maxEcalEnr_;
  double                        maxNeighborTrackEnr_;
  int                           nRun_;
  edm::InputTag                 triggerEvent_, theTriggerResultsLabel_;
  edm::EDGetTokenT<trigger::TriggerEvent>  tok_trigEvt_;
  edm::EDGetTokenT<edm::TriggerResults>    tok_trigRes_;

  edm::EDGetTokenT<reco::TrackCollection>  tok_genTrack_;
  edm::EDGetTokenT<reco::VertexCollection> tok_recVtx_;
  edm::EDGetTokenT<reco::BeamSpot>         tok_bs_;
  edm::EDGetTokenT<EcalRecHitCollection>   tok_EB_;
  edm::EDGetTokenT<EcalRecHitCollection>   tok_EE_;
  edm::EDGetTokenT<HBHERecHitCollection>   tok_hbhe_;
  edm::EDGetTokenT<GenEventInfoProduct>    tok_ew_; 

  TTree                     *tree;
  int                        t_Run, t_Event, t_nVtx, t_nTrk, t_ieta; 
  double                     t_EventWeight, t_p, t_pt, t_phi;
  double                     t_eHcal, t_eHcal10, t_eHcal30; 
  double                     t_eMipDR, t_hmaxNearP;
  bool                       t_selectTk, t_qltyMissFlag, t_qltyPVFlag;
  std::vector<unsigned int> *t_DetIds, *t_DetIds1, *t_DetIds3;
  std::vector<double>       *t_HitEnergies, *t_HitEnergies1, *t_HitEnergies3;

  TH1F                      *h_nTrk, *h_nVtx;
  TProfile                  *h_RecHit_iEta, *h_RecHit_num;
  TH1I                      *h_iEta, *h_tketa0[5], *h_tketa1[5], *h_tketa2[5];
  TH1I                      *h_tketa3[5], *h_tketa4[5], *h_tketa5[5];
  TH1F                      *h_Rechit_E, *h_jetp;
  TH1F              *h_jetpt[4];
  TH1I                      *h_tketav1[5][6], *h_tketav2[5][6];
};

IsoTrackCalibration::IsoTrackCalibration(const edm::ParameterSet& iConfig) : 
  nRun_(0) {

  usesResource("TFileService");
  //now do whatever initialization is needed
  const double isolationRadius(28.9);
  verbosity_                          = iConfig.getUntrackedParameter<int>("Verbosity",0);
  trigNames_                          = iConfig.getUntrackedParameter<std::vector<std::string> >("Triggers");
  theTrackQuality_                    = iConfig.getUntrackedParameter<std::string>("TrackQuality","highPurity");
  reco::TrackBase::TrackQuality trackQuality_=reco::TrackBase::qualityByName(theTrackQuality_);
  constTrackPt_                       = iConfig.getUntrackedParameter<double>("ConstTrackPt", 10.0);
  slopeTrackPt_                       = iConfig.getUntrackedParameter<double>("SlopeTrackPt", 0.16);
  selectionParameters_.minPt          = constTrackPt_;
  selectionParameters_.minQuality     = trackQuality_;
  selectionParameters_.maxDxyPV       = iConfig.getUntrackedParameter<double>("MaxDxyPV", 0.02);
  selectionParameters_.maxDzPV        = iConfig.getUntrackedParameter<double>("MaxDzPV",  0.02);
  selectionParameters_.maxChi2        = iConfig.getUntrackedParameter<double>("MaxChi2",  5.0);
  selectionParameters_.maxDpOverP     = iConfig.getUntrackedParameter<double>("MaxDpOverP",  0.1);
  selectionParameters_.minOuterHit    = iConfig.getUntrackedParameter<int>("MinOuterHit", 4);
  selectionParameters_.minLayerCrossed= iConfig.getUntrackedParameter<int>("MinLayerCrossed", 8);
  selectionParameters_.maxInMiss      = iConfig.getUntrackedParameter<int>("MaxInMiss", 0);
  selectionParameters_.maxOutMiss     = iConfig.getUntrackedParameter<int>("MaxOutMiss", 0);
  a_coneR_                            = iConfig.getUntrackedParameter<double>("ConeRadius",34.98);
  a_charIsoR_                         = a_coneR_ + isolationRadius;
  a_mipR_                             = iConfig.getUntrackedParameter<double>("ConeRadiusMIP",14.0);
  maxEcalEnr_                         = iConfig.getUntrackedParameter<double>("MaxEcalEnergyInCone",2.5);
  maxNeighborTrackEnr_                = iConfig.getUntrackedParameter<double>("MaxNeighborTrackEnergy",40.0);
  isMC_                               = iConfig.getUntrackedParameter<bool>("IsMC", false);
  isQCD_                              = iConfig.getUntrackedParameter<bool>("IsQCD", false);
  isAOD_                              = iConfig.getUntrackedParameter<bool>("IsAOD", true);
  triggerEvent_                       = edm::InputTag("hltTriggerSummaryAOD","","HLT");
  theTriggerResultsLabel_             = edm::InputTag("TriggerResults","","HLT");

  // define tokens for access
  tok_trigEvt_  = consumes<trigger::TriggerEvent>(triggerEvent_);
  tok_trigRes_  = consumes<edm::TriggerResults>(theTriggerResultsLabel_);
  tok_genTrack_ = consumes<reco::TrackCollection>(edm::InputTag("generalTracks"));
  tok_recVtx_   = consumes<reco::VertexCollection>(edm::InputTag("offlinePrimaryVertices"));
  tok_bs_       = consumes<reco::BeamSpot>(edm::InputTag("offlineBeamSpot"));
  tok_ew_       = consumes<GenEventInfoProduct>(edm::InputTag("generator")); 
 
  if (isAOD_) {
    tok_EB_     = consumes<EcalRecHitCollection>(edm::InputTag("reducedEcalRecHitsEB"));
    tok_EE_     = consumes<EcalRecHitCollection>(edm::InputTag("reducedEcalRecHitsEE"));
    tok_hbhe_   = consumes<HBHERecHitCollection>(edm::InputTag("reducedHcalRecHits", "hbhereco"));
  } else {
    tok_EB_     = consumes<EcalRecHitCollection>(edm::InputTag("ecalRecHit","EcalRecHitsEB"));
    tok_EE_     = consumes<EcalRecHitCollection>(edm::InputTag("ecalRecHit","EcalRecHitsEE"));
    tok_hbhe_   = consumes<HBHERecHitCollection>(edm::InputTag("hbhereco"));
  }

#ifdef DebugLog
  if (verbosity_>=0) {
    std::cout <<"Parameters read from config file \n" 
          <<"\t minPt "           << selectionParameters_.minPt   
              <<"\t theTrackQuality " << theTrackQuality_
          <<"\t minQuality "      << selectionParameters_.minQuality
          <<"\t maxDxyPV "        << selectionParameters_.maxDxyPV          
          <<"\t maxDzPV "         << selectionParameters_.maxDzPV          
          <<"\t maxChi2 "         << selectionParameters_.maxChi2          
          <<"\t maxDpOverP "      << selectionParameters_.maxDpOverP
          <<"\t minOuterHit "     << selectionParameters_.minOuterHit
          <<"\t minLayerCrossed " << selectionParameters_.minLayerCrossed
          <<"\t maxInMiss "       << selectionParameters_.maxInMiss
          <<"\t maxOutMiss "      << selectionParameters_.maxOutMiss
          <<"\t a_coneR "         << a_coneR_      
          <<"\t a_charIsoR "      << a_charIsoR_
          <<"\t a_mipR "          << a_mipR_
          <<"\t isMC "            << isMC_
          <<"\t isQCD "           << isQCD_
          <<"\t isAOD "           << isAOD_
          << std::endl;
    std::cout << trigNames_.size() << " triggers to be studied:";
    for (unsigned int k=0; k<trigNames_.size(); ++k)
      std::cout << " " << trigNames_[k];
    std::cout << std::endl;
  }
#endif
}

IsoTrackCalibration::~IsoTrackCalibration() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

void IsoTrackCalibration::analyze(const edm::Event& iEvent, 
                  const edm::EventSetup& iSetup) {

  t_Run   = iEvent.id().run();
  t_Event = iEvent.id().event();
#ifdef DebugLog
  if (verbosity_%10 > 0) 
    std::cout << "Run " << t_Run << " Event " << t_Event << " Luminosity " 
          << iEvent.luminosityBlock() << " Bunch " << iEvent.bunchCrossing()
          << " starts ==========" << std::endl;
#endif
  //Get magnetic field and ECAL channel status
  edm::ESHandle<MagneticField> bFieldH;
  iSetup.get<IdealMagneticFieldRecord>().get(bFieldH);
  const MagneticField *bField = bFieldH.product();
  
  edm::ESHandle<EcalSeverityLevelAlgo> sevlv;
  iSetup.get<EcalSeverityLevelAlgoRcd>().get(sevlv);

  // get handles to calogeometry and calotopology
  edm::ESHandle<CaloGeometry> pG;
  iSetup.get<CaloGeometryRecord>().get(pG);
  const CaloGeometry* geo = pG.product();
  
  //Get track collection
  edm::Handle<reco::TrackCollection> trkCollection;
  iEvent.getByToken(tok_genTrack_, trkCollection);
  reco::TrackCollection::const_iterator trkItr;
 
  //event weight for FLAT sample and PU information
  t_EventWeight = 1.0;
  edm::Handle<GenEventInfoProduct> genEventInfo;
  iEvent.getByToken(tok_ew_, genEventInfo);
  if (genEventInfo.isValid()) t_EventWeight = genEventInfo->weight();  

  //Define the best vertex and the beamspot
  edm::Handle<reco::VertexCollection> recVtxs;
  iEvent.getByToken(tok_recVtx_, recVtxs);  
  edm::Handle<reco::BeamSpot> beamSpotH;
  iEvent.getByToken(tok_bs_, beamSpotH);
  math::XYZPoint leadPV(0,0,0);

  t_nVtx = recVtxs->size();
  h_nVtx->Fill(t_nVtx);

  if (recVtxs->size()>0 && !((*recVtxs)[0].isFake())) {
    leadPV = math::XYZPoint( (*recVtxs)[0].x(),(*recVtxs)[0].y(), (*recVtxs)[0].z() );
  } else if (beamSpotH.isValid()) {
    leadPV = beamSpotH->position();
  }
#ifdef DebugLog
  if ((verbosity_/100)%10>0) {
    std::cout << "Primary Vertex " << leadPV;
    if (beamSpotH.isValid()) std::cout << " Beam Spot " 
                       << beamSpotH->position();
    std::cout << std::endl;
  }
#endif  

  // RecHits
  edm::Handle<EcalRecHitCollection> barrelRecHitsHandle;
  edm::Handle<EcalRecHitCollection> endcapRecHitsHandle;
  iEvent.getByToken(tok_EB_, barrelRecHitsHandle);
  iEvent.getByToken(tok_EE_, endcapRecHitsHandle);
  edm::Handle<HBHERecHitCollection> hbhe;
  iEvent.getByToken(tok_hbhe_, hbhe);

  //Trigger
  bool triggerOK = false;
  if (isMC_ && !isQCD_) {
    triggerOK = true; // ignore HLT for single pion MC
  } else {
    trigger::TriggerEvent triggerEvent;
    edm::Handle<trigger::TriggerEvent> triggerEventHandle;
    iEvent.getByToken(tok_trigEvt_, triggerEventHandle);
    if (!triggerEventHandle.isValid()) {
#ifdef DebugLog
      std::cout << "Error! Can't get the product "<< triggerEvent_.label() 
        << std::endl;
#endif
    } else {
      edm::Handle<edm::TriggerResults> triggerResults;
      iEvent.getByToken(tok_trigRes_, triggerResults);
      if (triggerResults.isValid()) {
    const edm::TriggerNames & triggerNames = iEvent.triggerNames(*triggerResults);
    const std::vector<std::string> & triggerNames_ = triggerNames.triggerNames();
    for (unsigned int iHLT=0; iHLT<triggerResults->size(); iHLT++) {
      int hlt    = triggerResults->accept(iHLT);
      if (hlt > 0) {
        for (unsigned int i=0; i<trigNames_.size(); ++i) {
          if (triggerNames_[iHLT].find(trigNames_[i].c_str())!=std::string::npos) {
        triggerOK = true;
#ifdef DebugLog
        if (verbosity_%10 > 0)
          std::cout << "This is the trigger we are looking for "
                << triggerNames_[iHLT] << " Flag " << hlt 
                << std::endl;
#endif
          }
        }
      }
    }
      }
    }
  }

  if (triggerOK) {
    //Propagate tracks to calorimeter surface)
    std::vector<spr::propagatedTrackDirection> trkCaloDirections;
    spr::propagateCALO(trkCollection, geo, bField, theTrackQuality_,
               trkCaloDirections, ((verbosity_/100)%10>2));
    //Loop over tracks
    std::vector<spr::propagatedTrackDirection>::const_iterator trkDetItr;
    unsigned int nTracks(0), nselTracks(0);
    t_nTrk = trkCaloDirections.size();

    for (trkDetItr = trkCaloDirections.begin(),nTracks=0; 
     trkDetItr != trkCaloDirections.end(); trkDetItr++, nTracks++) {

      const reco::Track* pTrack = &(*(trkDetItr->trkItr));
#ifdef DebugLog
      if (verbosity_%10> 0) 
    std::cout << "This track : " << nTracks << " (pt/eta/phi/p) :" 
          << pTrack->pt() << "/" << pTrack->eta() << "/" 
          << pTrack->phi() << "/" << pTrack->p() << std::endl;
#endif

      t_ieta = 0;
      if (trkDetItr->okHCAL) {
    HcalDetId detId = (HcalDetId)(trkDetItr->detIdHCAL);
    t_ieta = detId.ieta();
      }
      // ---------- eta-dependent restriction on Pt ----------------------------
      selectionParameters_.minPt = constTrackPt_ - std::abs((double)t_ieta)*slopeTrackPt_;
      // -----------------------------------------------------------------------
        
      //Selection of good track
      t_selectTk = spr::goodTrack(pTrack,leadPV,selectionParameters_,((verbosity_/100)%10>2));
      spr::trackSelectionParameters oneCutParameters = selectionParameters_;
      oneCutParameters.maxDxyPV  = 10;
      oneCutParameters.maxDzPV   = 100;
      oneCutParameters.maxInMiss = 2;
      oneCutParameters.maxOutMiss= 2;
      bool qltyFlag  = spr::goodTrack(pTrack,leadPV,oneCutParameters,((verbosity_/100)%10>2));
      oneCutParameters           = selectionParameters_;
      oneCutParameters.maxDxyPV  = 10;
      oneCutParameters.maxDzPV   = 100;
      t_qltyMissFlag = spr::goodTrack(pTrack,leadPV,oneCutParameters,((verbosity_/100)%10>2));
      oneCutParameters           = selectionParameters_;
      oneCutParameters.maxInMiss = 2;
      oneCutParameters.maxOutMiss= 2;
      t_qltyPVFlag   = spr::goodTrack(pTrack,leadPV,oneCutParameters,((verbosity_/100)%10>2));
        
#ifdef DebugLog
      if (verbosity_%10 > 0) 
    std::cout << "qltyFlag|okECAL|okHCAL : " << qltyFlag << "|" 
          << trkDetItr->okECAL << "/" << trkDetItr->okHCAL 
          << std::endl;
#endif
      if (qltyFlag && trkDetItr->okECAL && trkDetItr->okHCAL) {
    nselTracks++;
    int nRH_eMipDR(0), nNearTRKs(0);
    //------ ecal energy around track -------------------------------
    t_eMipDR = spr::eCone_ecal(geo, barrelRecHitsHandle, 
                   endcapRecHitsHandle,
                   trkDetItr->pointHCAL,
                   trkDetItr->pointECAL,
                   a_mipR_, trkDetItr->directionECAL, 
                   nRH_eMipDR);
    //---- isolation criteria ----------------------------------------------
    t_hmaxNearP = spr::chargeIsolationCone(nTracks,trkCaloDirections,
                           a_charIsoR_, nNearTRKs, 
                           ((verbosity_/100)%10>2));

    if ( t_eMipDR < maxEcalEnr_ && t_hmaxNearP < maxNeighborTrackEnr_ ) {
      //------------ HCAL --------------------------------------------------
      //------ initialize arrays of DetID and hit energies -----------------
      t_DetIds->clear();
      t_DetIds1->clear();
      t_DetIds3->clear();
      t_HitEnergies->clear();
      t_HitEnergies1->clear();
      t_HitEnergies3->clear();
      int nRecHits(-999), nRecHits1(-999), nRecHits3(-999);       
      std::vector<DetId> ids, ids1, ids3;
      //------ hcal energy in the main cone -------------------------------
      t_eHcal = spr::eCone_hcal(geo, hbhe, trkDetItr->pointHCAL, 
                    trkDetItr->pointECAL, a_coneR_, 
                    trkDetItr->directionHCAL,nRecHits, 
                    ids, *t_HitEnergies);
      t_DetIds->reserve(ids.size());
      for (unsigned int k = 0; k < ids.size(); ++k) {
        t_DetIds->push_back(ids[k].rawId());
      }
      //----- hcal energy in the extended cone 1 (a_coneR+10) --------------
      t_eHcal10 = spr::eCone_hcal(geo, hbhe, trkDetItr->pointHCAL,
                      trkDetItr->pointECAL, a_coneR_+10,
                      trkDetItr->directionHCAL,nRecHits1,
                      ids1, *t_HitEnergies1);
      t_DetIds1->reserve(ids1.size());
      for (unsigned int k = 0; k < ids1.size(); ++k) {
        t_DetIds1->push_back(ids1[k].rawId());
      }
      //----- hcal energy in the extended cone 3 (a_coneR+30) --------------
      t_eHcal30 = spr::eCone_hcal(geo, hbhe, trkDetItr->pointHCAL,
                      trkDetItr->pointECAL, a_coneR_+30,
                      trkDetItr->directionHCAL,nRecHits3,
                      ids3, *t_HitEnergies3);
      t_DetIds3->reserve(ids3.size());
      for (unsigned int k = 0; k < ids3.size(); ++k) {
        t_DetIds3->push_back(ids3[k].rawId());
      }

      t_p = pTrack->p();
      t_pt = pTrack->pt();
      t_phi = pTrack->phi();
#ifdef DebugLog
      if (verbosity_%10 > 0) {
        std::cout << "This track : " << nTracks << " (pt/eta/phi/p) :" 
              << pTrack->pt() << "/" << pTrack->eta() << "/" 
              << pTrack->phi() << "/" << t_p << std::endl;
        std::cout << "e_MIP " << t_eMipDR << " Chg Isolation "
              << t_hmaxNearP << " eHcal" << t_eHcal << " ieta " 
              << t_ieta << " Quality " << t_qltyMissFlag
              << ":" << t_qltyPVFlag << ":" << t_selectTk 
              << std::endl;
        for (unsigned int lll=0;lll<t_DetIds->size();lll++) {
          std::cout << "det id is = " << t_DetIds->at(lll)  << "  " 
            << " hit enery is  = "  << t_HitEnergies->at(lll) 
            << std::endl;         
        }
      }
#endif
      tree->Fill();
    } // end of conditions on t_eMipDR and t_hmaxNearP
      } // end of loose check of track quality
    } // end of loop over tracks

    h_nTrk->Fill(nTracks);
  } // end of triggerOK
}

void IsoTrackCalibration::beginJob() {

  h_nVtx = fs_->make<TH1F>("h_nVtx","h_nVtx",100,0,100);
  h_nTrk = fs_->make<TH1F>("h_nTrk","h_nTrk",100,0,2000);

  tree = fs_->make<TTree>("CalibTree", "CalibTree");
   
  tree->Branch("t_Run",         &t_Run,         "t_Run/I");
  tree->Branch("t_Event",       &t_Event,       "t_Event/I");
  tree->Branch("t_nVtx",        &t_nVtx,        "t_nVtx/I");
  tree->Branch("t_nTrk",        &t_nTrk,        "t_nTrk/I");
  tree->Branch("t_EventWeight", &t_EventWeight, "t_EventWeight/D");
  tree->Branch("t_p",           &t_p,           "t_p/D");
  tree->Branch("t_pt",          &t_pt,          "t_pt/D");
  tree->Branch("t_ieta",        &t_ieta,        "t_ieta/I");
  tree->Branch("t_phi",         &t_phi,         "t_phi/D");
  tree->Branch("t_eMipDR",      &t_eMipDR,      "t_eMipDR/D");
  tree->Branch("t_eHcal",       &t_eHcal,       "t_eHcal/D");
  tree->Branch("t_eHcal10",     &t_eHcal10,     "t_eHcal10/D");
  tree->Branch("t_eHcal30",     &t_eHcal30,     "t_eHcal30/D");
  tree->Branch("t_hmaxNearP",   &t_hmaxNearP,   "t_hmaxNearP/D");
  tree->Branch("t_selectTk",    &t_selectTk,    "t_selectTk/O");
  tree->Branch("t_qltyMissFlag",&t_qltyMissFlag,"t_qltyMissFlag/O");
  tree->Branch("t_qltyPVFlag",  &t_qltyPVFlag,  "t_qltyPVFlag/O)");

  t_DetIds       = new std::vector<unsigned int>();
  t_DetIds1      = new std::vector<unsigned int>();
  t_DetIds3      = new std::vector<unsigned int>();
  t_HitEnergies  = new std::vector<double>();
  t_HitEnergies1 = new std::vector<double>();
  t_HitEnergies3 = new std::vector<double>();

  tree->Branch("t_DetIds",       "std::vector<unsigned int>", &t_DetIds);
  tree->Branch("t_DetIds1",      "std::vector<unsigned int>", &t_DetIds1);
  tree->Branch("t_DetIds3",      "std::vector<unsigned int>", &t_DetIds3);
  tree->Branch("t_HitEnergies",  "std::vector<double>",       &t_HitEnergies);
  tree->Branch("t_HitEnergies1", "std::vector<double>",       &t_HitEnergies1);
  tree->Branch("t_HitEnergies3", "std::vector<double>",       &t_HitEnergies3);
}

// ------------ method called when starting to processes a run  ------------
void IsoTrackCalibration::beginRun(edm::Run const& iRun, edm::EventSetup const& iSetup) {

  bool changed_(true);
  bool flag =  hltConfig_.init(iRun,iSetup,"HLT",changed_);
  edm::LogInfo("HcalIsoTrack") << "Run[" << nRun_ << "] " << iRun.run() 
                               << " process HLT init flag "
                               << flag << " change flag " << changed_;

  // check if trigger names in (new) config                       
  if (changed_) {
    changed_ = false;
    edm::LogInfo("HcalIsoTrack") << "New trigger menu found !!!";
    const unsigned int n(hltConfig_.size());
    for (unsigned itrig=0; itrig<trigNames_.size(); itrig++) {
      unsigned int triggerindx = hltConfig_.triggerIndex(trigNames_[itrig]);
      if (triggerindx >= n) {
        edm::LogWarning("HcalIsoTrack") << trigNames_[itrig] << " " 
                                        << triggerindx << " does not exist in "
                                        << "the current menu";
      } else {
        edm::LogInfo("HcalIsoTrack") << trigNames_[itrig] << " " 
                                     << triggerindx << " exists";
      }
    }
  }
}

// ------------ method called when ending the processing of a run  ------------
void IsoTrackCalibration::endRun(edm::Run const& iRun, edm::EventSetup const&) {
  ++nRun_;
  edm::LogWarning("HcalIsoTrack") << "endRun[" << nRun_ << "] " << iRun.run();
}

// ------------ method called when starting to processes a luminosity block  ------------
void IsoTrackCalibration::beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) {}
// ------------ method called when ending the processing of a luminosity block  ------------
void IsoTrackCalibration::endLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) {}
// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void IsoTrackCalibration::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.setUnknown();
  descriptions.addDefault(desc);
}

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