AlCaHcalNoiseProducer.cc

Go to the documentation of this file.

/*
Original author Grigory Safronov

27/03/09 - compilation from :
HLTrigger/special/src/HLTHcalNoiseFilter.cc
Calibration/HcalAlCaRecoProducers/src/AlCaEcalHcalReadoutsProducer.cc
Calibration/HcalIsolatedTrackReco/src/SubdetFEDSelector.cc

*/


#include "Calibration/HcalAlCaRecoProducers/interface/AlCaHcalNoiseProducer.h"

#include "DataFormats/FEDRawData/interface/FEDRawData.h"
#include "DataFormats/FEDRawData/interface/FEDNumbering.h"

#include "EventFilter/RawDataCollector/interface/RawDataFEDSelector.h"

#include "DataFormats/CaloTowers/interface/CaloTower.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/METReco/interface/CaloMET.h"
#include "DataFormats/Math/interface/deltaR.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"


AlCaHcalNoiseProducer::AlCaHcalNoiseProducer(const edm::ParameterSet& iConfig)
{
   tok_jets_ = consumes<reco::CaloJetCollection>(iConfig.getParameter<edm::InputTag>("JetSource"));
   tok_met_ = consumes<reco::CaloMETCollection>(iConfig.getParameter<edm::InputTag>("MetSource"));
   tok_tower_ = consumes<CaloTowerCollection>(iConfig.getParameter<edm::InputTag>("TowerSource"));
   useMet_ = iConfig.getParameter<bool>("UseMET");
   useJet_ = iConfig.getParameter<bool>("UseJet");
   MetCut_ = iConfig.getParameter<double>("MetCut");
   JetMinE_ = iConfig.getParameter<double>("JetMinE");
   JetHCALminEnergyFraction_ = iConfig.getParameter<double>("JetHCALminEnergyFraction");

   tok_ho_ = consumes<HORecHitCollection>(iConfig.getParameter<edm::InputTag>("hoInput"));
   tok_hf_ = consumes<HFRecHitCollection>(iConfig.getParameter<edm::InputTag>("hfInput"));
   tok_hbhe_ = consumes<HBHERecHitCollection>(iConfig.getParameter<edm::InputTag>("hbheInput"));
   ecalLabels_= iConfig.getParameter<std::vector<edm::InputTag> >("ecalInputs");
   tok_ps_ = consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("ecalPSInput"));
   tok_raw_ = consumes<FEDRawDataCollection>(iConfig.getParameter<edm::InputTag>("rawInput"));

   const unsigned nLabels = ecalLabels_.size();
   for( unsigned i=0; i != nLabels; i++ )
      toks_ecal_.push_back( consumes<EcalRecHitCollection>(ecalLabels_[i]) ); 

  
   //register products
   produces<HBHERecHitCollection>("HBHERecHitCollectionFHN");
   produces<HORecHitCollection>("HORecHitCollectionFHN");
   produces<HFRecHitCollection>("HFRecHitCollectionFHN");
   
   produces<EcalRecHitCollection>("EcalRecHitCollectionFHN");
   produces<EcalRecHitCollection>("PSEcalRecHitCollectionFHN");
    
   produces<FEDRawDataCollection>("HcalFEDsFHN");

}


AlCaHcalNoiseProducer::~AlCaHcalNoiseProducer()
{
}


// ------------ method called to produce the data  ------------
void
AlCaHcalNoiseProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   bool acceptEvent=false;

  // filtering basing on HLTrigger/special/src/HLTHcalNoiseFilter.cc:

   bool isAnomalous_BasedOnMET = false;
   bool isAnomalous_BasedOnEnergyFraction=false; 

   if (useMet_)
     {
       edm::Handle <reco::CaloMETCollection> metHandle;
       iEvent.getByToken(tok_met_, metHandle);
       const reco::CaloMETCollection *metCol = metHandle.product();
       const reco::CaloMET met = metCol->front();
    
       if(met.pt() > MetCut_) isAnomalous_BasedOnMET=true;
     }
       
   if (useJet_)
     {
       edm::Handle<reco::CaloJetCollection> calojetHandle;
       iEvent.getByToken(tok_jets_,calojetHandle);
       
       edm::Handle<CaloTowerCollection> towerHandle;
       iEvent.getByToken(tok_tower_, towerHandle);

       std::vector<CaloTower> TowerContainer;
       std::vector<reco::CaloJet> JetContainer;
       TowerContainer.clear();
       JetContainer.clear();
       CaloTower seedTower;
       nEvents++;
       for(reco::CaloJetCollection::const_iterator calojetIter = calojetHandle->begin();calojetIter != calojetHandle->end();++calojetIter) {
     if( ((calojetIter->et())*cosh(calojetIter->eta()) > JetMinE_) && (calojetIter->energyFractionHadronic() > JetHCALminEnergyFraction_) ) {
       JetContainer.push_back(*calojetIter);
       double maxTowerE = 0.0;
       for(CaloTowerCollection::const_iterator kal = towerHandle->begin(); kal != towerHandle->end(); kal++) {
         double dR = deltaR((*calojetIter).eta(),(*calojetIter).phi(),(*kal).eta(),(*kal).phi());
         if( (dR < 0.50) && (kal->p() > maxTowerE) ) {
           maxTowerE = kal->p();
           seedTower = *kal;
         }
       }
       TowerContainer.push_back(seedTower);
     }
     
       }
       if(JetContainer.size() > 0) {
     nAnomalousEvents++;
     isAnomalous_BasedOnEnergyFraction = true;
       }
     }
   
   acceptEvent=((useMet_&&isAnomalous_BasedOnMET)||(useJet_&&isAnomalous_BasedOnEnergyFraction));


    //Create empty output collections
   
  std::auto_ptr<HBHERecHitCollection> miniHBHERecHitCollection(new HBHERecHitCollection);
  std::auto_ptr<HORecHitCollection> miniHORecHitCollection(new HORecHitCollection);
  std::auto_ptr<HFRecHitCollection> miniHFRecHitCollection(new HFRecHitCollection);

  std::auto_ptr<EcalRecHitCollection> outputEColl(new EcalRecHitCollection);
  std::auto_ptr<EcalRecHitCollection> outputESColl(new EcalRecHitCollection);

  std::auto_ptr<FEDRawDataCollection> outputFEDs(new FEDRawDataCollection);


  // if good event get and save all colletions
  if (acceptEvent)
    {
      edm::Handle<HBHERecHitCollection> hbhe;
      edm::Handle<HORecHitCollection> ho;
      edm::Handle<HFRecHitCollection> hf;
      
      iEvent.getByToken(tok_hbhe_,hbhe);
      iEvent.getByToken(tok_ho_,ho);
      iEvent.getByToken(tok_hf_,hf);

      edm::Handle<EcalRecHitCollection> pRecHits;
      iEvent.getByToken(tok_ps_,pRecHits);

      // temporary collection of EB+EE recHits

      std::auto_ptr<EcalRecHitCollection> tmpEcalRecHitCollection(new EcalRecHitCollection);
      
      std::vector<edm::EDGetTokenT<EcalRecHitCollection> >::const_iterator i;
      for (i=toks_ecal_.begin(); i!=toks_ecal_.end(); i++) 
    {
      edm::Handle<EcalRecHitCollection> ec;
      iEvent.getByToken(*i,ec);
      for(EcalRecHitCollection::const_iterator recHit = (*ec).begin(); recHit != (*ec).end(); ++recHit)
        {
          tmpEcalRecHitCollection->push_back(*recHit);
        }
    }    


      const HBHERecHitCollection Hithbhe = *(hbhe.product());
      for(HBHERecHitCollection::const_iterator hbheItr=Hithbhe.begin(); hbheItr!=Hithbhe.end(); hbheItr++)
    {
      miniHBHERecHitCollection->push_back(*hbheItr);
    }
      const HORecHitCollection Hitho = *(ho.product());
      for(HORecHitCollection::const_iterator hoItr=Hitho.begin(); hoItr!=Hitho.end(); hoItr++)
    {
      miniHORecHitCollection->push_back(*hoItr);
    }
      
      const HFRecHitCollection Hithf = *(hf.product());
      for(HFRecHitCollection::const_iterator hfItr=Hithf.begin(); hfItr!=Hithf.end(); hfItr++)
    {
      miniHFRecHitCollection->push_back(*hfItr);
    }

      for (std::vector<EcalRecHit>::const_iterator ehit=tmpEcalRecHitCollection->begin(); ehit!=tmpEcalRecHitCollection->end(); ehit++) 
    {
      outputEColl->push_back(*ehit);
    }

      // write PS
      const EcalRecHitCollection& psrechits = *(pRecHits.product());
      
      for(EcalRecHitCollection::const_iterator i=psrechits.begin(); i!=psrechits.end(); i++) 
    {
      outputESColl->push_back( *i );
    }
      

      // get HCAL FEDs
      edm::Handle<FEDRawDataCollection> rawIn;
      iEvent.getByToken(tok_raw_,rawIn);
      
      std::vector<int> selFEDs;
      for (int i=FEDNumbering::MINHCALFEDID; i<=FEDNumbering::MAXHCALFEDID; i++)
    {
      selFEDs.push_back(i);
    }

      // Copying FEDs :
      const FEDRawDataCollection *rdc=rawIn.product();
      
      //   if ( ( rawData[i].provenance()->processName() != e.processHistory().rbegin()->processName() ) )
      //       continue ; // skip all raw collections not produced by the current process
      
      for ( int j=0; j< FEDNumbering::MAXFEDID; ++j ) 
    {
      bool rightFED=false;
      for (uint32_t k=0; k<selFEDs.size(); k++)
        {
          if (j==selFEDs[k])
        {
          rightFED=true;
        }
        }
      if (!rightFED) continue;
      const FEDRawData & fedData = rdc->FEDData(j);
      size_t size=fedData.size();
      
      if ( size > 0 ) 
        {
          // this fed has data -- lets copy it
          FEDRawData & fedDataProd = outputFEDs->FEDData(j);
          if ( fedDataProd.size() != 0 ) {
        //     std::cout << " More than one FEDRawDataCollection with data in FED ";
        //     std::cout << j << " Skipping the 2nd\n";
        continue;
          }
          fedDataProd.resize(size);
          unsigned char *dataProd=fedDataProd.data();
          const unsigned char *data=fedData.data();
          for ( unsigned int k=0; k<size; ++k ) {
        dataProd[k]=data[k];
          }
        }
    }
    }

  //Put selected information in the event
  iEvent.put( miniHBHERecHitCollection, "HBHERecHitCollectionFHN");
  iEvent.put( miniHORecHitCollection, "HORecHitCollectionFHN");
  iEvent.put( miniHFRecHitCollection, "HFRecHitCollectionFHN");
  iEvent.put( outputEColl, "EcalRecHitCollectionFHN"); 
  iEvent.put( outputESColl, "PSEcalRecHitCollectionFHN");
  iEvent.put( outputFEDs, "HcalFEDsFHN"); 
}