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	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () 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 submitPVResolutionJobs::desc.

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, relativeConstraints::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(), l1GtPatternGenerator_cfi::bx, caloTruthCellsProducer_cfi::caloParticles, caloParticlesToken_, l1t::HGCalClusterT< C >::centre(), HGCalClusteringDummyImpl::clusterizeDummy(), dummyClustering_, HGCalClusteringDummyImpl::eventSetup(), HGCalShowerShape::eventSetup(), HGCalTriggerTools::eventSetup(), HGCalShowerShape::fillShapes(), relativeConstraints::geometry, get, hcalSimParameters_cfi::he, makeCellsCollection_, makeHitMap(), eostools::move(), l1t::HGCalMulticluster::saveHOverE(), l1t::HGCalClusterT< C >::setDetId(), reco::LeafCandidate::setP4(), singleTopDQM_cfi::setup, showerShape_, jetUpdater_cfi::sort, 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().