AlcaPCCProducer.cc

Go to the documentation of this file.

// C++ standard
#include <string>
// CMS
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "DataFormats/Luminosity/interface/PixelClusterCounts.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"

#include "TMath.h"
//The class 
class AlcaPCCProducer : public edm::one::EDProducer<edm::EndLuminosityBlockProducer,edm::one::WatchLuminosityBlocks>{
  public:
    explicit AlcaPCCProducer(const edm::ParameterSet&);
    ~AlcaPCCProducer() override;

  private:
    void beginLuminosityBlock     (edm::LuminosityBlock const& lumiSeg, const edm::EventSetup& iSetup) override ;
    void endLuminosityBlock       (edm::LuminosityBlock const& lumiSeg, const edm::EventSetup& iSetup) override ;
    void endLuminosityBlockProduce(edm::LuminosityBlock& lumiSeg, const edm::EventSetup& iSetup) override ;
    void produce                  (edm::Event& iEvent, const edm::EventSetup& iSetup) override ;
 
    edm::EDGetTokenT<edmNew::DetSetVector<SiPixelCluster> >  pixelToken;
    edm::InputTag   fPixelClusterLabel;

    std::string trigstring_;      //specifies the trigger Rand or ZeroBias 
    int countEvt_;       //counter
    int countLumi_;      //counter

    std::unique_ptr<reco::PixelClusterCounts> thePCCob;

};

//--------------------------------------------------------------------------------------------------
AlcaPCCProducer::AlcaPCCProducer(const edm::ParameterSet& iConfig)
{
    fPixelClusterLabel = iConfig.getParameter<edm::ParameterSet>("AlcaPCCProducerParameters").getParameter<edm::InputTag>("pixelClusterLabel");
    trigstring_ = iConfig.getParameter<edm::ParameterSet>("AlcaPCCProducerParameters").getUntrackedParameter<std::string>("trigstring","alcaPCC");

    countLumi_ = 0;

    produces<reco::PixelClusterCounts, edm::Transition::EndLuminosityBlock>(trigstring_);
    pixelToken=consumes<edmNew::DetSetVector<SiPixelCluster> >(fPixelClusterLabel);
}

//--------------------------------------------------------------------------------------------------
AlcaPCCProducer::~AlcaPCCProducer(){
}

//--------------------------------------------------------------------------------------------------
void AlcaPCCProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup){
    countEvt_++;

    unsigned int bx=iEvent.bunchCrossing();
    //std::cout<<"The Bunch Crossing"<<bx<<std::endl;
    thePCCob->eventCounter(bx);

    //Looping over the clusters and adding the counts up  
    edm::Handle< edmNew::DetSetVector<SiPixelCluster> > hClusterColl;
    iEvent.getByToken(pixelToken,hClusterColl);

    const edmNew::DetSetVector<SiPixelCluster>& clustColl = *(hClusterColl.product()); 
    // ----------------------------------------------------------------------
    // -- Clusters without tracks
    for (auto const & mod: clustColl) {
        if(mod.empty()) { continue; }
        DetId detId = mod.id();

        //--The following will be used when we make a theshold for the clusters.
        //--Keeping this for features that may be implemented later.
        // -- clusters on this det
        //edmNew::DetSet<SiPixelCluster>::const_iterator  di;
        //int nClusterCount=0;
        //for (di = mod.begin(); di != mod.end(); ++di) {
        //    nClusterCount++;
        //}
        int nCluster = mod.size();
        thePCCob->increment(detId(), bx, nCluster);
    }
}

//--------------------------------------------------------------------------------------------------
void AlcaPCCProducer::beginLuminosityBlock(edm::LuminosityBlock const& lumiSeg, const edm::EventSetup& iSetup){
    //New PCC object at the beginning of each lumi section
    thePCCob = std::make_unique<reco::PixelClusterCounts>(); 
    countLumi_++;

}

//--------------------------------------------------------------------------------------------------
void AlcaPCCProducer::endLuminosityBlock(edm::LuminosityBlock const& lumiSeg, const edm::EventSetup& iSetup){
}

//--------------------------------------------------------------------------------------------------
void AlcaPCCProducer::endLuminosityBlockProduce(edm::LuminosityBlock& lumiSeg, const edm::EventSetup& iSetup){

    //Saving the PCC object
    lumiSeg.put(std::move(thePCCob), std::string(trigstring_)); 

}

DEFINE_FWK_MODULE(AlcaPCCProducer);