EcalDigiSelector Class Reference

Inheritance diagram for EcalDigiSelector:

Public Member Functions
	EcalDigiSelector (const edm::ParameterSet &ps)
void	produce (edm::Event &, const edm::EventSetup &) override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Private Attributes
edm::EDGetTokenT< reco::SuperClusterCollection >	barrelSuperClusterProducer_
edm::ESGetToken< CaloTopology, CaloTopologyRecord >	caloTopologyToken_
double	cluster_pt_thresh_
edm::EDGetTokenT< EBDigiCollection >	EcalEBDigiToken_
edm::EDGetTokenT< EcalRecHitCollection >	EcalEBRecHitToken_
edm::EDGetTokenT< EEDigiCollection >	EcalEEDigiToken_
edm::EDGetTokenT< EcalRecHitCollection >	EcalEERecHitToken_
edm::EDGetTokenT< reco::SuperClusterCollection >	endcapSuperClusterProducer_
int	nclus_sel_
std::string	selectedEcalEBDigiCollection_
std::string	selectedEcalEEDigiCollection_
double	single_cluster_thresh_

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

Detailed Description

Definition at line 32 of file EcalDigiSelector.cc.

Constructor & Destructor Documentation

EcalDigiSelector()

EcalDigiSelector::EcalDigiSelector

(

const edm::ParameterSet &

ps

)

Definition at line 61 of file EcalDigiSelector.cc.

                                                            {
  selectedEcalEBDigiCollection_ = ps.getParameter<std::string>("selectedEcalEBDigiCollection");
  selectedEcalEEDigiCollection_ = ps.getParameter<std::string>("selectedEcalEEDigiCollection");
 
  barrelSuperClusterProducer_ =
      consumes<reco::SuperClusterCollection>(ps.getParameter<edm::InputTag>("barrelSuperClusterProducer"));
  endcapSuperClusterProducer_ =
      consumes<reco::SuperClusterCollection>(ps.getParameter<edm::InputTag>("endcapSuperClusterProducer"));
 
  EcalEBRecHitToken_ = consumes<EcalRecHitCollection>(ps.getParameter<edm::InputTag>("EcalEBRecHitTag"));
  EcalEERecHitToken_ = consumes<EcalRecHitCollection>(ps.getParameter<edm::InputTag>("EcalEERecHitTag"));
 
  EcalEBDigiToken_ = consumes<EBDigiCollection>(ps.getParameter<edm::InputTag>("EcalEBDigiTag"));
  EcalEEDigiToken_ = consumes<EEDigiCollection>(ps.getParameter<edm::InputTag>("EcalEEDigiTag"));
 
  caloTopologyToken_ = esConsumes<CaloTopology, CaloTopologyRecord>();
 
  cluster_pt_thresh_ = ps.getParameter<double>("cluster_pt_thresh");
  single_cluster_thresh_ = ps.getParameter<double>("single_cluster_thresh");
 
  nclus_sel_ = ps.getParameter<int>("nclus_sel");
  produces<EBDigiCollection>(selectedEcalEBDigiCollection_);
  produces<EEDigiCollection>(selectedEcalEEDigiCollection_);
}

References edm::ParameterSet::getParameter(), and AlCaHLTBitMon_QueryRunRegistry::string.

Member Function Documentation

produce()

void EcalDigiSelector::produce ( edm::Event &  evt, const edm::EventSetup &  es  )

override

Definition at line 86 of file EcalDigiSelector.cc.

                                                                     {
  //Get BarrelSuperClusters to start.
  edm::Handle<reco::SuperClusterCollection> pBarrelSuperClusters;
 
  evt.getByToken(barrelSuperClusterProducer_, pBarrelSuperClusters);
 
  const reco::SuperClusterCollection& BarrelSuperClusters = *pBarrelSuperClusters;
  //Got BarrelSuperClusters
 
  //Get BarrelSuperClusters to start.
  edm::Handle<reco::SuperClusterCollection> pEndcapSuperClusters;
 
  evt.getByToken(endcapSuperClusterProducer_, pEndcapSuperClusters);
 
  const reco::SuperClusterCollection& EndcapSuperClusters = *pEndcapSuperClusters;
  //Got EndcapSuperClusters
 
  reco::SuperClusterCollection saveBarrelSuperClusters;
  reco::SuperClusterCollection saveEndcapSuperClusters;
  bool meet_single_thresh = false;
  //Loop over barrel superclusters, and apply threshold
  for (int loop = 0; loop < int(BarrelSuperClusters.size()); loop++) {
    SuperCluster clus1 = BarrelSuperClusters[loop];
    float eta1 = clus1.eta();
    float energy1 = clus1.energy();
    float theta1 = 2 * atan(exp(-1. * eta1));
    float cluspt1 = energy1 * sin(theta1);
    if (cluspt1 > cluster_pt_thresh_) {
      saveBarrelSuperClusters.push_back(clus1);
      if (cluspt1 > single_cluster_thresh_)
        meet_single_thresh = true;
    }
  }
 
  //Loop over endcap superclusters, and apply threshold
  for (int loop = 0; loop < int(EndcapSuperClusters.size()); loop++) {
    SuperCluster clus1 = EndcapSuperClusters[loop];
    float eta1 = clus1.eta();
    float energy1 = clus1.energy();
    float theta1 = 2 * atan(exp(-1. * eta1));
    float cluspt1 = energy1 * sin(theta1);
    if (cluspt1 > cluster_pt_thresh_) {
      saveEndcapSuperClusters.push_back(clus1);
      if (cluspt1 > single_cluster_thresh_)
        meet_single_thresh = true;
    }
  }
 
  auto SEBDigiCol = std::make_unique<EBDigiCollection>();
  auto SEEDigiCol = std::make_unique<EEDigiCollection>();
  int TotClus = saveBarrelSuperClusters.size() + saveEndcapSuperClusters.size();
 
  if (TotClus >= nclus_sel_ || meet_single_thresh) {
    if (!saveBarrelSuperClusters.empty()) {
      edm::ESHandle<CaloTopology> pTopology = es.getHandle(caloTopologyToken_);
      const CaloTopology* topology = pTopology.product();
 
      //get barrel digi collection
      edm::Handle<EBDigiCollection> pdigis;
      const EBDigiCollection* digis = nullptr;
      evt.getByToken(EcalEBDigiToken_, pdigis);
      digis = pdigis.product();  // get a ptr to the product
 
      edm::Handle<EcalRecHitCollection> prechits;
      const EcalRecHitCollection* rechits = nullptr;
      evt.getByToken(EcalEBRecHitToken_, prechits);
      rechits = prechits.product();  // get a ptr to the product
 
      if (digis) {
        std::vector<DetId> saveTheseDetIds;
        //pick out the detids for the 3x3 in each of the selected superclusters
        for (int loop = 0; loop < int(saveBarrelSuperClusters.size()); loop++) {
          SuperCluster clus1 = saveBarrelSuperClusters[loop];
          const CaloClusterPtr& bcref = clus1.seed();
          const BasicCluster* bc = bcref.get();
          //Get the maximum detid
          DetId maxDetId = EcalClusterTools::getMaximum(*bc, rechits).first;
          // Loop over the 3x3 array centered on maximum detid
          for (DetId detId : CaloRectangleRange(1, maxDetId, *topology))
            saveTheseDetIds.push_back(detId);
        }
        for (int detloop = 0; detloop < int(saveTheseDetIds.size()); ++detloop) {
          EBDetId detL = EBDetId(saveTheseDetIds[detloop]);
 
          for (EBDigiCollection::const_iterator blah = digis->begin(); blah != digis->end(); blah++) {
            if (detL == blah->id()) {
              EBDataFrame myDigi = (*blah);
              SEBDigiCol->push_back(detL);
 
              EBDataFrame df(SEBDigiCol->back());
              for (int iq = 0; iq < myDigi.size(); ++iq) {
                df.setSample(iq, myDigi.sample(iq).raw());
              }
              //ebcounter++;
            }
          }
          //if (ebcounter >= int(saveTheseDetIds.size())) break;
        }  //loop over dets
      }
 
    }  //If barrel superclusters need saving.
 
    if (!saveEndcapSuperClusters.empty()) {
      edm::ESHandle<CaloTopology> pTopology = es.getHandle(caloTopologyToken_);
      const CaloTopology* topology = pTopology.product();
 
      //Get endcap rec hit collection
      //get endcap digi collection
      edm::Handle<EEDigiCollection> pdigis;
      const EEDigiCollection* digis = nullptr;
      evt.getByToken(EcalEEDigiToken_, pdigis);
      digis = pdigis.product();  // get a ptr to the product
 
      edm::Handle<EcalRecHitCollection> prechits;
      const EcalRecHitCollection* rechits = nullptr;
      evt.getByToken(EcalEERecHitToken_, prechits);
      rechits = prechits.product();  // get a ptr to the product
 
      if (digis) {
        //std::vector<DetId> saveTheseDetIds;
        std::set<DetId> saveTheseDetIds;
        //pick out the digis for the 3x3 in each of the selected superclusters
        for (int loop = 0; loop < int(saveEndcapSuperClusters.size()); loop++) {
          SuperCluster clus1 = saveEndcapSuperClusters[loop];
          const CaloClusterPtr& bcref = clus1.seed();
          const BasicCluster* bc = bcref.get();
          //Get the maximum detid
          DetId maxDetId = EcalClusterTools::getMaximum(*bc, rechits).first;
          // Loop over the 3x3 array centered on maximum detid
          for (DetId detId : CaloRectangleRange(1, maxDetId, *topology))
            saveTheseDetIds.insert(detId);
        }
        int eecounter = 0;
        for (EEDigiCollection::const_iterator blah = digis->begin(); blah != digis->end(); blah++) {
          std::set<DetId>::const_iterator finder = saveTheseDetIds.find(blah->id());
          if (finder != saveTheseDetIds.end()) {
            EEDetId detL = EEDetId(*finder);
 
            if (detL == blah->id()) {
              EEDataFrame myDigi = (*blah);
              SEEDigiCol->push_back(detL);
              EEDataFrame df(SEEDigiCol->back());
              for (int iq = 0; iq < myDigi.size(); ++iq) {
                df.setSample(iq, myDigi.sample(iq).raw());
              }
              eecounter++;
            }
          }
          if (eecounter >= int(saveTheseDetIds.size()))
            break;
        }  //loop over digis
      }
    }  //If endcap superclusters need saving.
 
  }  //If we're actually saving stuff
 
  //Okay, either my collections have been filled with the requisite Digis, or they haven't.
 
  //Empty collection, or full, still put in event.
  SEBDigiCol->sort();
  SEEDigiCol->sort();
  evt.put(std::move(SEBDigiCol), selectedEcalEBDigiCollection_);
  evt.put(std::move(SEEDigiCol), selectedEcalEEDigiCollection_);
}

References edm::DataFrameContainer::begin(), hgcalPerformanceValidation::df, edm::DataFrameContainer::end(), reco::CaloCluster::energy(), reco::CaloCluster::eta(), HLT_FULL_cff::eta1, JetChargeProducer_cfi::exp, HLT_FULL_cff::finder, edm::Ptr< T >::get(), edm::Event::getByToken(), edm::EventSetup::getHandle(), createfilelist::int, heppy_loop::loop, eostools::move(), edm::Handle< T >::product(), edm::ESHandle< T >::product(), edm::Event::put(), EcalMGPASample::raw(), HI_PhotonSkim_cff::rechits, EcalDataFrame::sample(), reco::SuperCluster::seed(), funct::sin(), and EcalDataFrame::size().

Member Data Documentation

barrelSuperClusterProducer_

edm::EDGetTokenT<reco::SuperClusterCollection> EcalDigiSelector::barrelSuperClusterProducer_

private

Definition at line 42 of file EcalDigiSelector.cc.

caloTopologyToken_

edm::ESGetToken<CaloTopology, CaloTopologyRecord> EcalDigiSelector::caloTopologyToken_

private

Definition at line 50 of file EcalDigiSelector.cc.

cluster_pt_thresh_

double EcalDigiSelector::cluster_pt_thresh_

private

Definition at line 52 of file EcalDigiSelector.cc.

EcalEBDigiToken_

edm::EDGetTokenT<EBDigiCollection> EcalDigiSelector::EcalEBDigiToken_

private

Definition at line 48 of file EcalDigiSelector.cc.

EcalEBRecHitToken_

edm::EDGetTokenT<EcalRecHitCollection> EcalDigiSelector::EcalEBRecHitToken_

private

Definition at line 46 of file EcalDigiSelector.cc.

EcalEEDigiToken_

edm::EDGetTokenT<EEDigiCollection> EcalDigiSelector::EcalEEDigiToken_

private

Definition at line 49 of file EcalDigiSelector.cc.

EcalEERecHitToken_

edm::EDGetTokenT<EcalRecHitCollection> EcalDigiSelector::EcalEERecHitToken_

private

Definition at line 47 of file EcalDigiSelector.cc.

endcapSuperClusterProducer_

edm::EDGetTokenT<reco::SuperClusterCollection> EcalDigiSelector::endcapSuperClusterProducer_

private

Definition at line 43 of file EcalDigiSelector.cc.

nclus_sel_

int EcalDigiSelector::nclus_sel_

private

Definition at line 54 of file EcalDigiSelector.cc.

selectedEcalEBDigiCollection_

std::string EcalDigiSelector::selectedEcalEBDigiCollection_

private

Definition at line 39 of file EcalDigiSelector.cc.

selectedEcalEEDigiCollection_

std::string EcalDigiSelector::selectedEcalEEDigiCollection_

private

Definition at line 40 of file EcalDigiSelector.cc.

single_cluster_thresh_

double EcalDigiSelector::single_cluster_thresh_

private

Definition at line 53 of file EcalDigiSelector.cc.