CaloTruthCellsProducer Class Reference

Inheritance diagram for CaloTruthCellsProducer:

Public Member Functions
	CaloTruthCellsProducer (edm::ParameterSet const &)
	~CaloTruthCellsProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Types
typedef edm::AssociationMap< edm::OneToMany< CaloParticleCollection, l1t::HGCalTriggerCellBxCollection > >	CaloToCellsMap

Private Member Functions
std::unordered_map< uint32_t, double >	makeHitMap (edm::Event const &, edm::EventSetup const &, HGCalTriggerGeometryBase const &) const
void	produce (edm::Event &, edm::EventSetup const &) override

Private Attributes
edm::EDGetTokenT< CaloParticleCollection >	caloParticlesToken_
HGCalClusteringDummyImpl	dummyClustering_
bool	makeCellsCollection_
HGCalShowerShape	showerShape_
edm::EDGetTokenT< std::vector< PCaloHit > >	simHitsTokenEE_
edm::EDGetTokenT< std::vector< PCaloHit > >	simHitsTokenHEback_
edm::EDGetTokenT< std::vector< PCaloHit > >	simHitsTokenHEfront_
edm::EDGetTokenT< l1t::HGCalTriggerCellBxCollection >	triggerCellsToken_
HGCalTriggerTools	triggerTools_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 26 of file CaloTruthCellsProducer.cc.

Member Typedef Documentation

CaloToCellsMap

typedef edm::AssociationMap<edm::OneToMany<CaloParticleCollection, l1t::HGCalTriggerCellBxCollection> > CaloTruthCellsProducer::CaloToCellsMap

private

Definition at line 40 of file CaloTruthCellsProducer.cc.

Constructor & Destructor Documentation

CaloTruthCellsProducer()

CaloTruthCellsProducer::CaloTruthCellsProducer ( edm::ParameterSet const &  config )

explicit

Definition at line 55 of file CaloTruthCellsProducer.cc.

    : makeCellsCollection_(config.getParameter<bool>("makeCellsCollection")),
      caloParticlesToken_(consumes<CaloParticleCollection>(config.getParameter<edm::InputTag>("caloParticles"))),
      triggerCellsToken_(
          consumes<l1t::HGCalTriggerCellBxCollection>(config.getParameter<edm::InputTag>("triggerCells"))),
      simHitsTokenEE_(consumes<std::vector<PCaloHit>>(config.getParameter<edm::InputTag>("simHitsEE"))),
      simHitsTokenHEfront_(consumes<std::vector<PCaloHit>>(config.getParameter<edm::InputTag>("simHitsHEfront"))),
      simHitsTokenHEback_(consumes<std::vector<PCaloHit>>(config.getParameter<edm::InputTag>("simHitsHEback"))),
      dummyClustering_(config.getParameterSet("dummyClustering")) {
  produces<CaloToCellsMap>();
  produces<l1t::HGCalClusterBxCollection>();
  produces<l1t::HGCalMulticlusterBxCollection>();
  if (makeCellsCollection_)
    produces<l1t::HGCalTriggerCellBxCollection>();
}

References makeCellsCollection_.

~CaloTruthCellsProducer()

CaloTruthCellsProducer::~CaloTruthCellsProducer ( )

override

Definition at line 71 of file CaloTruthCellsProducer.cc.

{}

Member Function Documentation

fillDescriptions()

void CaloTruthCellsProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 271 of file CaloTruthCellsProducer.cc.

                                                                                        {
  //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);
}

References edm::ConfigurationDescriptions::addDefault(), and edm::ParameterSetDescription::setUnknown().

makeHitMap()

std::unordered_map< uint32_t, double > CaloTruthCellsProducer::makeHitMap ( edm::Event const &  event, edm::EventSetup const &  setup, HGCalTriggerGeometryBase const &  geometry  ) const

private

Definition at line 234 of file CaloTruthCellsProducer.cc.

                                                                                                       {
  std::unordered_map<uint32_t, double> hitMap;  // cellId -> sim energy
 
  typedef std::function<DetId(DetId const&)> DetIdMapper;
  typedef std::pair<edm::EDGetTokenT<std::vector<PCaloHit>> const*, DetIdMapper> SimHitSpec;
 
  SimHitSpec specs[3] = {{&simHitsTokenEE_,
                          [this, &geometry](DetId const& simId) -> DetId {
                            return this->triggerTools_.simToReco(simId, geometry.eeTopology());
                          }},
                         {&simHitsTokenHEfront_,
                          [this, &geometry](DetId const& simId) -> DetId {
                            return this->triggerTools_.simToReco(simId, geometry.fhTopology());
                          }},
                         {&simHitsTokenHEback_, nullptr}};
  if (geometry.isV9Geometry())
    specs[2].second = [this, &geometry](DetId const& simId) -> DetId {
      return this->triggerTools_.simToReco(simId, geometry.hscTopology());
    };
  else
    specs[2].second = [this, &geometry](DetId const& simId) -> DetId {
      return this->triggerTools_.simToReco(simId, geometry.bhTopology());
    };
 
  for (auto const& tt : specs) {
    edm::Handle<std::vector<PCaloHit>> handle;
    event.getByToken(*tt.first, handle);
    auto const& simhits(*handle);
 
    for (auto const& simhit : simhits)
      hitMap.emplace(tt.second(simhit.id()), simhit.energy());
  }
 
  return hitMap;
}

References HiBiasedCentrality_cfi::function, geometry, patZpeak::handle, edm::second(), HLTBitAnalyser_cfi::simhits, simHitsTokenEE_, simHitsTokenHEback_, simHitsTokenHEfront_, HGCalTriggerTools::simToReco(), HistogramManager_cfi::specs, triggerTools_, and groupFilesInBlocks::tt.

Referenced by produce().

produce()

void CaloTruthCellsProducer::produce ( edm::Event &  event, edm::EventSetup const &  setup  )

overrideprivate

Definition at line 73 of file CaloTruthCellsProducer.cc.

                                                                                {
  edm::Handle<CaloParticleCollection> caloParticlesHandle;
  event.getByToken(caloParticlesToken_, caloParticlesHandle);
  auto const& caloParticles(*caloParticlesHandle);
 
  edm::Handle<l1t::HGCalTriggerCellBxCollection> triggerCellsHandle;
  event.getByToken(triggerCellsToken_, triggerCellsHandle);
  auto const& triggerCells(*triggerCellsHandle);
 
  dummyClustering_.eventSetup(setup);
  showerShape_.eventSetup(setup);
  triggerTools_.eventSetup(setup);
 
  edm::ESHandle<HGCalTriggerGeometryBase> geometryHandle;
  setup.get<CaloGeometryRecord>().get(geometryHandle);
  auto const& geometry(*geometryHandle);
 
  std::unordered_map<uint32_t, CaloParticleRef> tcToCalo;
 
  std::unordered_map<uint32_t, double> hitToEnergy(makeHitMap(event, setup, geometry));  // cellId -> sim energy
  std::unordered_map<uint32_t, std::pair<double, double>>
      tcToEnergies;  // tcId -> {total sim energy, fractioned sim energy}
 
  for (auto const& he : hitToEnergy) {
    DetId hitId(he.first);
    // this line will throw if (for whatever reason) hitId is not mapped to a trigger cell id
    uint32_t tcId(geometry.getTriggerCellFromCell(hitId));
    tcToEnergies[tcId].first += he.second;
  }
 
  // used later to order multiclusters
  std::map<int, CaloParticleRef> orderedCaloRefs;
 
  for (unsigned iP(0); iP != caloParticles.size(); ++iP) {
    auto const& caloParticle(caloParticles.at(iP));
    if (caloParticle.g4Tracks().at(0).eventId().event() != 0)  // pileup
      continue;
 
    CaloParticleRef ref(caloParticlesHandle, iP);
 
    SimClusterRefVector const& simClusters(caloParticle.simClusters());
    for (auto const& simCluster : simClusters) {
      for (auto const& hAndF : simCluster->hits_and_fractions()) {
        DetId hitId(hAndF.first);
        uint32_t tcId;
        try {
          tcId = geometry.getTriggerCellFromCell(hitId);
        } catch (cms::Exception const& ex) {
          edm::LogError("CaloTruthCellsProducer") << ex.what();
          continue;
        }
 
        tcToCalo.emplace(tcId, ref);
        tcToEnergies[tcId].second += hitToEnergy[hAndF.first] * hAndF.second;
      }
    }
 
    // ordered by the gen particle index
    int genIndex(caloParticle.g4Tracks().at(0).genpartIndex() - 1);
    orderedCaloRefs[genIndex] = ref;
  }
 
  auto outMap(std::make_unique<CaloToCellsMap>(caloParticlesHandle, triggerCellsHandle));
  std::unique_ptr<l1t::HGCalTriggerCellBxCollection> outCollection;
  if (makeCellsCollection_)
    outCollection = std::make_unique<l1t::HGCalTriggerCellBxCollection>();
 
  typedef edm::Ptr<l1t::HGCalTriggerCell> TriggerCellPtr;
  typedef edm::Ptr<l1t::HGCalCluster> ClusterPtr;
 
  // ClusteringDummyImpl only considers BX 0, so we dump all cells to one vector
  std::vector<TriggerCellPtr> triggerCellPtrs;
 
  // loop through all bunch crossings
  for (int bx(triggerCells.getFirstBX()); bx <= triggerCells.getLastBX(); ++bx) {
    for (auto cItr(triggerCells.begin(bx)); cItr != triggerCells.end(bx); ++cItr) {
      auto const& cell(*cItr);
 
      auto mapElem(tcToCalo.find(cell.detId()));
      if (mapElem == tcToCalo.end())
        continue;
 
      outMap->insert(mapElem->second,
                     edm::Ref<l1t::HGCalTriggerCellBxCollection>(triggerCellsHandle, triggerCells.key(cItr)));
 
      if (makeCellsCollection_) {
        auto const& simEnergies(tcToEnergies.at(cell.detId()));
        if (simEnergies.first > 0.) {
          double eFraction(simEnergies.second / simEnergies.first);
 
          outCollection->push_back(bx, cell);
          auto& newCell((*outCollection)[outCollection->size() - 1]);
 
          newCell.setMipPt(cell.mipPt() * eFraction);
          newCell.setP4(cell.p4() * eFraction);
        }
      }
 
      triggerCellPtrs.emplace_back(triggerCellsHandle, triggerCells.key(cItr));
    }
  }
 
  event.put(std::move(outMap));
  if (makeCellsCollection_)
    event.put(std::move(outCollection));
 
  auto outClusters(std::make_unique<l1t::HGCalClusterBxCollection>());
 
  auto sortCellPtrs(
      [](TriggerCellPtr const& lhs, TriggerCellPtr const& rhs) -> bool { return lhs->mipPt() > rhs->mipPt(); });
 
  std::sort(triggerCellPtrs.begin(), triggerCellPtrs.end(), sortCellPtrs);
  dummyClustering_.clusterizeDummy(triggerCellPtrs, *outClusters);
 
  std::unordered_map<unsigned, std::vector<unsigned>> caloToClusterIndices;
  for (unsigned iC(0); iC != outClusters->size(); ++iC) {
    auto const& cluster((*outClusters)[iC]);
    // cluster detId and cell detId are identical
    auto caloRef(tcToCalo.at(cluster.detId()));
    caloToClusterIndices[caloRef.key()].push_back(iC);
  }
 
  auto clustersHandle(event.put(std::move(outClusters)));
 
  auto outMulticlusters(std::make_unique<l1t::HGCalMulticlusterBxCollection>());
 
  for (auto const& ocr : orderedCaloRefs) {
    auto const& ref(ocr.second);
 
    if (ref.isNull())  // shouldn't happen
      continue;
 
    auto const& caloParticle(*ref);
 
    l1t::HGCalMulticluster multicluster;
 
    for (unsigned iC : caloToClusterIndices[ref.key()]) {
      ClusterPtr clPtr(clustersHandle, iC);
      multicluster.addConstituent(clPtr, true, 1.);
    }
 
    // Set the gen particle index as the DetId
    multicluster.setDetId(caloParticle.g4Tracks().at(0).genpartIndex() - 1);
 
    auto const& centre(multicluster.centre());
    math::PtEtaPhiMLorentzVector multiclusterP4(multicluster.sumPt(), centre.eta(), centre.phi(), 0.);
    multicluster.setP4(multiclusterP4);
 
    showerShape_.fillShapes(multicluster, geometry);
 
    // not setting the quality flag
    // multicluster.setHwQual(id_->decision(multicluster));
    // fill H/E
    multicluster.saveHOverE();
 
    outMulticlusters->push_back(0, multicluster);
  }
 
  event.put(std::move(outMulticlusters));
}

References l1t::HGCalClusterT< C >::addConstituent(), simKBmtfDigis_cfi::bx, caloTruthCellsProducer_cfi::caloParticles, caloParticlesToken_, l1t::HGCalClusterT< C >::centre(), HGCalClusteringDummyImpl::clusterizeDummy(), dummyClustering_, HGCalClusteringDummyImpl::eventSetup(), HGCalShowerShape::eventSetup(), HGCalTriggerTools::eventSetup(), HGCalShowerShape::fillShapes(), geometry, get, es_hardcode_cfi::he, makeCellsCollection_, makeHitMap(), eostools::move(), l1t::HGCalMulticluster::saveHOverE(), l1t::HGCalClusterT< C >::setDetId(), reco::LeafCandidate::setP4(), singleTopDQM_cfi::setup, showerShape_, l1t::HGCalClusterT< C >::sumPt(), caloTruthCellsProducer_cfi::triggerCells, triggerCellsToken_, triggerTools_, and cms::Exception::what().

Member Data Documentation

caloParticlesToken_

edm::EDGetTokenT<CaloParticleCollection> CaloTruthCellsProducer::caloParticlesToken_

private

Definition at line 43 of file CaloTruthCellsProducer.cc.

Referenced by produce().

dummyClustering_

HGCalClusteringDummyImpl CaloTruthCellsProducer::dummyClustering_

private

Definition at line 49 of file CaloTruthCellsProducer.cc.

Referenced by produce().

makeCellsCollection_

bool CaloTruthCellsProducer::makeCellsCollection_

private

Definition at line 42 of file CaloTruthCellsProducer.cc.

Referenced by CaloTruthCellsProducer(), and produce().

showerShape_

HGCalShowerShape CaloTruthCellsProducer::showerShape_

private

Definition at line 50 of file CaloTruthCellsProducer.cc.

Referenced by produce().

simHitsTokenEE_

edm::EDGetTokenT<std::vector<PCaloHit> > CaloTruthCellsProducer::simHitsTokenEE_

private

Definition at line 45 of file CaloTruthCellsProducer.cc.

Referenced by makeHitMap().

simHitsTokenHEback_

edm::EDGetTokenT<std::vector<PCaloHit> > CaloTruthCellsProducer::simHitsTokenHEback_

private

Definition at line 47 of file CaloTruthCellsProducer.cc.

Referenced by makeHitMap().

simHitsTokenHEfront_

edm::EDGetTokenT<std::vector<PCaloHit> > CaloTruthCellsProducer::simHitsTokenHEfront_

private

Definition at line 46 of file CaloTruthCellsProducer.cc.

Referenced by makeHitMap().

triggerCellsToken_

edm::EDGetTokenT<l1t::HGCalTriggerCellBxCollection> CaloTruthCellsProducer::triggerCellsToken_

private

Definition at line 44 of file CaloTruthCellsProducer.cc.

Referenced by produce().

triggerTools_

HGCalTriggerTools CaloTruthCellsProducer::triggerTools_

private

Definition at line 52 of file CaloTruthCellsProducer.cc.

Referenced by makeHitMap(), and produce().