EcalDigiSelector Class Reference

#include <EcalDigiSelector.h>

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
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Private Attributes
edm::EDGetTokenT< reco::SuperClusterCollection >	barrelSuperClusterProducer_
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<>
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 20 of file EcalDigiSelector.h.

Constructor & Destructor Documentation

EcalDigiSelector::EcalDigiSelector

(

const edm::ParameterSet &

ps

)

Definition at line 23 of file EcalDigiSelector.cc.

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

{
  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"));
  
  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_);

}

Member Function Documentation

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

override

Definition at line 53 of file EcalDigiSelector.cc.

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

{
  
  //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.get<CaloTopologyRecord>().get(pTopology);
      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
           std::pair<DetId, float> EDetty = 
               EcalClusterTools::getMaximum(*bc,rechits);
           //get the 3x3 array centered on maximum detid.
           std::vector<DetId> detvec = 
               EcalClusterTools::matrixDetId(topology,EDetty.first, -1, 1, -1, 1);
           //Loop over the 3x3
           for (int ik = 0;ik<int(detvec.size());++ik)
             saveTheseDetIds.push_back(detvec[ik]);
           
         }
         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.get<CaloTopologyRecord>().get(pTopology);
      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
           std::pair<DetId, float> EDetty = EcalClusterTools::getMaximum(*bc,rechits);
           //get the 3x3 array centered on maximum detid.
           std::vector<DetId> detvec = 
               EcalClusterTools::matrixDetId(topology,EDetty.first, -1, 1, -1, 1);
           //Loop over the 3x3
           for (int ik = 0;ik<int(detvec.size());++ik)
             saveTheseDetIds.insert(detvec[ik]);
         }
         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_);
  
}

Member Data Documentation

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

private

Definition at line 35 of file EcalDigiSelector.h.

double EcalDigiSelector::cluster_pt_thresh_

private

Definition at line 47 of file EcalDigiSelector.h.

edm::EDGetTokenT<EBDigiCollection> EcalDigiSelector::EcalEBDigiToken_

private

Definition at line 41 of file EcalDigiSelector.h.

edm::EDGetTokenT<EcalRecHitCollection> EcalDigiSelector::EcalEBRecHitToken_

private

Definition at line 39 of file EcalDigiSelector.h.

edm::EDGetTokenT<EEDigiCollection> EcalDigiSelector::EcalEEDigiToken_

private

Definition at line 42 of file EcalDigiSelector.h.

edm::EDGetTokenT<EcalRecHitCollection> EcalDigiSelector::EcalEERecHitToken_

private

Definition at line 40 of file EcalDigiSelector.h.

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

private

Definition at line 36 of file EcalDigiSelector.h.

int EcalDigiSelector::nclus_sel_

private

Definition at line 49 of file EcalDigiSelector.h.

std::string EcalDigiSelector::selectedEcalEBDigiCollection_

private

Definition at line 32 of file EcalDigiSelector.h.

std::string EcalDigiSelector::selectedEcalEEDigiCollection_

private

Definition at line 33 of file EcalDigiSelector.h.

double EcalDigiSelector::single_cluster_thresh_

private

Definition at line 48 of file EcalDigiSelector.h.