HGCalTriggerNtupleHGCDigis.cc

Go to the documentation of this file.

#include "DataFormats/HGCDigi/interface/HGCDigiCollections.h"
#include "DataFormats/ForwardDetId/interface/HGCalDetId.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"
#include "DataFormats/ForwardDetId/interface/ForwardSubdetector.h"
#include "L1Trigger/L1THGCal/interface/HGCalTriggerNtupleBase.h"
#include "L1Trigger/L1THGCal/interface/HGCalTriggerGeometryBase.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"

#include "SimDataFormats/CaloHit/interface/PCaloHit.h"
#include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
#include "SimDataFormats/CaloTest/interface/HGCalTestNumbering.h"
#include "Geometry/HcalCommonData/interface/HcalHitRelabeller.h"
#include "L1Trigger/L1THGCal/interface/HGCalTriggerTools.h"


class HGCalTriggerNtupleHGCDigis : public HGCalTriggerNtupleBase
{

    public:
        HGCalTriggerNtupleHGCDigis(const edm::ParameterSet& conf);
        ~HGCalTriggerNtupleHGCDigis() override{};
        void initialize(TTree&, const edm::ParameterSet&, edm::ConsumesCollector&&) final;
        void fill(const edm::Event& e, const edm::EventSetup& es) final;

    private:
        void simhits(const edm::Event& e, std::unordered_map<uint32_t, double>& simhits_ee, std::unordered_map<uint32_t, double>& simhits_fh, std::unordered_map<uint32_t, double>& simhits_bh);
        void clear() final;

        edm::EDGetToken ee_token_, fh_token_, bh_token_;
        bool is_Simhit_comp_;
        edm::EDGetToken SimHits_inputee_, SimHits_inputfh_, SimHits_inputbh_;

        HGCalTriggerTools triggerTools_;

        int hgcdigi_n_ ;
        std::vector<int> hgcdigi_id_;
        std::vector<int> hgcdigi_subdet_;
        std::vector<int> hgcdigi_side_;
        std::vector<int> hgcdigi_layer_;
        std::vector<int> hgcdigi_wafer_;
        std::vector<int> hgcdigi_wafertype_ ;
        std::vector<int> hgcdigi_cell_;
        std::vector<float> hgcdigi_eta_;
        std::vector<float> hgcdigi_phi_;
        std::vector<float> hgcdigi_z_;
        std::vector<uint32_t> hgcdigi_data_;
        std::vector<int> hgcdigi_isadc_;
        std::vector<float> hgcdigi_simenergy_;

        int bhdigi_n_ ;
        std::vector<int> bhdigi_id_;
        std::vector<int> bhdigi_subdet_;
        std::vector<int> bhdigi_side_;
        std::vector<int> bhdigi_layer_;
        std::vector<int> bhdigi_ieta_;
        std::vector<int> bhdigi_iphi_;
        std::vector<float> bhdigi_eta_;
        std::vector<float> bhdigi_phi_;
        std::vector<float> bhdigi_z_;
        std::vector<uint32_t> bhdigi_data_;
        std::vector<float> bhdigi_simenergy_;

        edm::ESHandle<HGCalTriggerGeometryBase> triggerGeometry_;

};

DEFINE_EDM_PLUGIN(HGCalTriggerNtupleFactory,
        HGCalTriggerNtupleHGCDigis,
        "HGCalTriggerNtupleHGCDigis" );


HGCalTriggerNtupleHGCDigis::
HGCalTriggerNtupleHGCDigis(const edm::ParameterSet& conf):HGCalTriggerNtupleBase(conf)
{
    is_Simhit_comp_ = conf.getParameter<bool>("isSimhitComp");

}

void
HGCalTriggerNtupleHGCDigis::
initialize(TTree& tree, const edm::ParameterSet& conf, edm::ConsumesCollector&& collector)
{

    ee_token_ = collector.consumes<HGCalDigiCollection>(conf.getParameter<edm::InputTag>("HGCDigisEE"));
    fh_token_ = collector.consumes<HGCalDigiCollection>(conf.getParameter<edm::InputTag>("HGCDigisFH"));
    bh_token_ = collector.consumes<HGCalDigiCollection>(conf.getParameter<edm::InputTag>("HGCDigisBH"));
    if (is_Simhit_comp_) {
      SimHits_inputee_ = collector.consumes<edm::PCaloHitContainer>(conf.getParameter<edm::InputTag>("eeSimHits"));
      SimHits_inputfh_ = collector.consumes<edm::PCaloHitContainer>(conf.getParameter<edm::InputTag>("fhSimHits"));
      SimHits_inputbh_ = collector.consumes<edm::PCaloHitContainer>(conf.getParameter<edm::InputTag>("bhSimHits"));
    }
    tree.Branch("hgcdigi_n", &hgcdigi_n_, "hgcdigi_n/I");
    tree.Branch("hgcdigi_id", &hgcdigi_id_);
    tree.Branch("hgcdigi_subdet", &hgcdigi_subdet_);
    tree.Branch("hgcdigi_zside", &hgcdigi_side_);
    tree.Branch("hgcdigi_layer", &hgcdigi_layer_);
    tree.Branch("hgcdigi_wafer", &hgcdigi_wafer_);
    tree.Branch("hgcdigi_wafertype", &hgcdigi_wafertype_);
    tree.Branch("hgcdigi_cell", &hgcdigi_cell_);
    tree.Branch("hgcdigi_eta", &hgcdigi_eta_);
    tree.Branch("hgcdigi_phi", &hgcdigi_phi_);
    tree.Branch("hgcdigi_z", &hgcdigi_z_);
    tree.Branch("hgcdigi_data", &hgcdigi_data_);
    tree.Branch("hgcdigi_isadc", &hgcdigi_isadc_);
    if (is_Simhit_comp_) tree.Branch("hgcdigi_simenergy", &hgcdigi_simenergy_);

    tree.Branch("bhdigi_n", &bhdigi_n_, "bhdigi_n/I");
    tree.Branch("bhdigi_id", &bhdigi_id_);
    tree.Branch("bhdigi_subdet", &bhdigi_subdet_);
    tree.Branch("bhdigi_zside", &bhdigi_side_);
    tree.Branch("bhdigi_layer", &bhdigi_layer_);
    tree.Branch("bhdigi_ieta", &bhdigi_ieta_);
    tree.Branch("bhdigi_iphi", &bhdigi_iphi_);
    tree.Branch("bhdigi_eta", &bhdigi_eta_);
    tree.Branch("bhdigi_phi", &bhdigi_phi_);
    tree.Branch("bhdigi_z", &bhdigi_z_);
    tree.Branch("bhdigi_data", &bhdigi_data_);
    if (is_Simhit_comp_) tree.Branch("bhdigi_simenergy", &bhdigi_simenergy_);
}

void
HGCalTriggerNtupleHGCDigis::
fill(const edm::Event& e, const edm::EventSetup& es)
{
    es.get<CaloGeometryRecord>().get(triggerGeometry_);

    edm::Handle<HGCalDigiCollection> ee_digis_h;
    e.getByToken(ee_token_, ee_digis_h);
    const HGCalDigiCollection& ee_digis = *ee_digis_h;
    edm::Handle<HGCalDigiCollection> fh_digis_h;
    e.getByToken(fh_token_, fh_digis_h);
    const HGCalDigiCollection& fh_digis = *fh_digis_h;
    edm::Handle<HGCalDigiCollection> bh_digis_h;
    e.getByToken(bh_token_, bh_digis_h);
    const HGCalDigiCollection& bh_digis = *bh_digis_h;

    triggerTools_.eventSetup(es);

    // sim hit association
    std::unordered_map<uint32_t, double> simhits_ee;
    std::unordered_map<uint32_t, double> simhits_fh;
    std::unordered_map<uint32_t, double> simhits_bh;
    if (is_Simhit_comp_) simhits(e, simhits_ee, simhits_fh, simhits_bh);

    clear();
    hgcdigi_n_ = ee_digis.size() + fh_digis.size();
    hgcdigi_id_.reserve(hgcdigi_n_);
    hgcdigi_subdet_.reserve(hgcdigi_n_);
    hgcdigi_side_.reserve(hgcdigi_n_);
    hgcdigi_layer_.reserve(hgcdigi_n_);
    hgcdigi_wafer_.reserve(hgcdigi_n_);
    hgcdigi_wafertype_.reserve(hgcdigi_n_);
    hgcdigi_cell_.reserve(hgcdigi_n_);
    hgcdigi_eta_.reserve(hgcdigi_n_);
    hgcdigi_phi_.reserve(hgcdigi_n_);
    hgcdigi_z_.reserve(hgcdigi_n_);
    hgcdigi_data_.reserve(hgcdigi_n_);
    hgcdigi_isadc_.reserve(hgcdigi_n_);
    if (is_Simhit_comp_) hgcdigi_simenergy_.reserve(hgcdigi_n_);

    bhdigi_n_ = bh_digis.size();
    bhdigi_id_.reserve(bhdigi_n_);
    bhdigi_subdet_.reserve(bhdigi_n_);
    bhdigi_side_.reserve(bhdigi_n_);
    bhdigi_layer_.reserve(bhdigi_n_);
    bhdigi_ieta_.reserve(bhdigi_n_);
    bhdigi_iphi_.reserve(bhdigi_n_);
    bhdigi_eta_.reserve(bhdigi_n_);
    bhdigi_phi_.reserve(bhdigi_n_);
    bhdigi_z_.reserve(bhdigi_n_);
    if (is_Simhit_comp_) bhdigi_simenergy_.reserve(bhdigi_n_);

    const int kIntimeSample = 2;
    for(const auto& digi : ee_digis)
      {
        const HGCalDetId id(digi.id());
        hgcdigi_id_.emplace_back(id.rawId());
        hgcdigi_subdet_.emplace_back(ForwardSubdetector::HGCEE);
        hgcdigi_side_.emplace_back(id.zside());
        hgcdigi_layer_.emplace_back(triggerTools_.layerWithOffset(id));
        hgcdigi_wafer_.emplace_back(id.wafer());
        hgcdigi_wafertype_.emplace_back(id.waferType());
        hgcdigi_cell_.emplace_back(id.cell());
        GlobalPoint cellpos = triggerGeometry_->eeGeometry()->getPosition(id.rawId());
        hgcdigi_eta_.emplace_back(cellpos.eta());
        hgcdigi_phi_.emplace_back(cellpos.phi());
        hgcdigi_z_.emplace_back(cellpos.z());
        hgcdigi_data_.emplace_back(digi[kIntimeSample].data());
        int is_adc=0;
        if (!(digi[kIntimeSample].mode())) is_adc =1;
        hgcdigi_isadc_.emplace_back(is_adc);
        if (is_Simhit_comp_) {
          double hit_energy=0;
          auto itr = simhits_ee.find(id);
          if(itr!=simhits_ee.end())hit_energy = itr->second;
          hgcdigi_simenergy_.emplace_back(hit_energy);
        }
      }

    for(const auto& digi : fh_digis)
      {
        const HGCalDetId id(digi.id());
        hgcdigi_id_.emplace_back(id.rawId());
        hgcdigi_subdet_.emplace_back(ForwardSubdetector::HGCHEF);
        hgcdigi_side_.emplace_back(id.zside());
        hgcdigi_layer_.emplace_back(triggerTools_.layerWithOffset(id));
        hgcdigi_wafer_.emplace_back(id.wafer());
        hgcdigi_wafertype_.emplace_back(id.waferType());
        hgcdigi_cell_.emplace_back(id.cell());
        GlobalPoint cellpos = triggerGeometry_->fhGeometry()->getPosition(id.rawId());
        hgcdigi_eta_.emplace_back(cellpos.eta());
        hgcdigi_phi_.emplace_back(cellpos.phi());
        hgcdigi_z_.emplace_back(cellpos.z());
        hgcdigi_data_.emplace_back(digi[kIntimeSample].data());
        int is_adc=0;
        if (!(digi[kIntimeSample].mode())) is_adc =1;
        hgcdigi_isadc_.emplace_back(is_adc);
        if (is_Simhit_comp_) {
          double hit_energy=0;
          auto itr = simhits_fh.find(id);
          if(itr!=simhits_fh.end())hit_energy = itr->second;
          hgcdigi_simenergy_.emplace_back(hit_energy);
        }
      }

     for(const auto& digi : bh_digis)
      {
        const HcalDetId id(digi.id());
        bhdigi_id_.emplace_back(id.rawId());
        bhdigi_subdet_.emplace_back(id.subdetId());
        bhdigi_side_.emplace_back(id.zside());
        bhdigi_layer_.emplace_back(triggerTools_.layerWithOffset(id));
        bhdigi_ieta_.emplace_back(id.ieta());
        bhdigi_iphi_.emplace_back(id.iphi());
        GlobalPoint cellpos = triggerGeometry_->bhGeometry()->getPosition(id.rawId());
        bhdigi_eta_.emplace_back(cellpos.eta());
        bhdigi_phi_.emplace_back(cellpos.phi());
        bhdigi_z_.emplace_back(cellpos.z());
        bhdigi_data_.emplace_back(digi[kIntimeSample].data());
        if (is_Simhit_comp_) {
          double hit_energy=0;
          auto itr = simhits_bh.find(id);
          if(itr!=simhits_bh.end())hit_energy = itr->second;
          bhdigi_simenergy_.emplace_back(hit_energy);
        }
      }
}

void
HGCalTriggerNtupleHGCDigis::
simhits(const edm::Event& e, std::unordered_map<uint32_t, double>& simhits_ee, std::unordered_map<uint32_t, double>& simhits_fh, std::unordered_map<uint32_t, double>& simhits_bh)
{

      edm::Handle<edm::PCaloHitContainer> ee_simhits_h;
      e.getByToken(SimHits_inputee_,ee_simhits_h);
      const edm::PCaloHitContainer& ee_simhits = *ee_simhits_h;
      edm::Handle<edm::PCaloHitContainer> fh_simhits_h;
      e.getByToken(SimHits_inputfh_,fh_simhits_h);
      const edm::PCaloHitContainer& fh_simhits = *fh_simhits_h;
      edm::Handle<edm::PCaloHitContainer> bh_simhits_h;
      e.getByToken(SimHits_inputbh_,bh_simhits_h);
      const edm::PCaloHitContainer& bh_simhits = *bh_simhits_h;

      //EE
      int layer=0,cell=0, sec=0, subsec=0, zp=0,subdet=0;
      ForwardSubdetector mysubdet;

      for( const auto& simhit : ee_simhits ) {
        HGCalTestNumbering::unpackHexagonIndex(simhit.id(), subdet, zp, layer, sec, subsec, cell);
        mysubdet = (ForwardSubdetector)(subdet);
        std::pair<int,int> recoLayerCell = triggerGeometry_->eeTopology().dddConstants().simToReco(cell,layer,sec,triggerGeometry_->eeTopology().detectorType());
        cell  = recoLayerCell.first;
        layer = recoLayerCell.second;
        if (layer<0 || cell<0) {
          continue;
        }
        auto itr_insert = simhits_ee.emplace(HGCalDetId(mysubdet,zp,layer,subsec,sec,cell), 0.);
        itr_insert.first->second += simhit.energy();
      }

      //  FH
      layer=0; cell=0; sec=0; subsec=0; zp=0; subdet=0;

      for( const auto& simhit : fh_simhits ) {
        HGCalTestNumbering::unpackHexagonIndex(simhit.id(), subdet, zp, layer, sec, subsec, cell);
        mysubdet = (ForwardSubdetector)(subdet);
        std::pair<int,int> recoLayerCell = triggerGeometry_->fhTopology().dddConstants().simToReco(cell,layer,sec,triggerGeometry_->fhTopology().detectorType());
        cell  = recoLayerCell.first;
        layer = recoLayerCell.second;
        if (layer<0 || cell<0) {
          continue;
        }
        auto itr_insert = simhits_fh.emplace(HGCalDetId(mysubdet,zp,layer,subsec,sec,cell), 0.);
        itr_insert.first->second += simhit.energy();
      }
      //  BH
      for( const auto& simhit : bh_simhits ) {
        HcalDetId id = HcalHitRelabeller::relabel(simhit.id(), triggerGeometry_->bhTopology().dddConstants());
        if (id.subdetId()!=HcalEndcap) continue;
        auto itr_insert = simhits_bh.emplace(id, 0.);
        itr_insert.first->second += simhit.energy();
      }
}


void
HGCalTriggerNtupleHGCDigis::
clear()
{
    hgcdigi_n_ = 0;
    hgcdigi_id_.clear();
    hgcdigi_subdet_.clear();
    hgcdigi_side_.clear();
    hgcdigi_layer_.clear();
    hgcdigi_wafer_.clear();
    hgcdigi_wafertype_.clear();
    hgcdigi_cell_.clear();
    hgcdigi_eta_.clear();
    hgcdigi_phi_.clear();
    hgcdigi_z_.clear();
    hgcdigi_data_.clear();
    hgcdigi_isadc_.clear();
    if  (is_Simhit_comp_) hgcdigi_simenergy_.clear();

    bhdigi_n_ = 0;
    bhdigi_id_.clear();
    bhdigi_subdet_.clear();
    bhdigi_side_.clear();
    bhdigi_layer_.clear();
    bhdigi_ieta_.clear();
    bhdigi_iphi_.clear();
    bhdigi_eta_.clear();
    bhdigi_phi_.clear();
    bhdigi_z_.clear();
    bhdigi_data_.clear();
    if  (is_Simhit_comp_) bhdigi_simenergy_.clear();
}