CMSSW_4_4_3_patch1/src/RecoLocalTracker/SiStripZeroSuppression/plugins/SiStripMeanCMExtractor.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    SiStripMeanCMExtractor
// Class:      SiStripMeanCMExtractor
// 
//
// Original Author:  Ivan Amos Cali,32 4-A08,+41227673039,
//         Created:  Wed Oct 13 11:50:47 CEST 2010
// $Id: SiStripMeanCMExtractor.cc,v 1.2 2010/11/04 14:12:31 edwenger Exp $
//
//


// system include files
#include <sstream>
#include <memory>
#include <list>
#include <algorithm>
#include <cassert>


// user include files
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

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

#include "DataFormats/Common/interface/DetSet.h"
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/SiStripDigi/interface/SiStripProcessedRawDigi.h"
#include "DataFormats/SiStripDigi/interface/SiStripRawDigi.h"
#include "CondFormats/SiStripObjects/interface/SiStripPedestals.h"
#include "CondFormats/DataRecord/interface/SiStripPedestalsRcd.h"


typedef std::map<uint32_t, std::vector<float> > CMMap;

class SiStripMeanCMExtractor : public edm::EDProducer {
   public:
      explicit SiStripMeanCMExtractor( const edm::ParameterSet&);
      ~SiStripMeanCMExtractor();

   private:
      virtual void beginJob() ;
      virtual void produce(edm::Event&, const edm::EventSetup&);
      virtual void endJob() ;
      
          void init(const edm::EventSetup&);
          
          edm::ESHandle<SiStripPedestals> pedestalHandle_;
      uint32_t pedestal_cache_id_;
  
          void StoreMean(const edm::DetSetVector<SiStripProcessedRawDigi>& );
          void ConvertMeanMapToDetSetVector(std::vector<edm::DetSet<SiStripProcessedRawDigi> >&);
          void CMExtractorFromPedestals(const edm::DetSetVector<SiStripRawDigi>&,std::vector<edm::DetSet<SiStripProcessedRawDigi> >&);
          edm::InputTag _inputTag;
          std::string _Algorithm;
          uint16_t _nEventsToUse;
          uint16_t _actualEvent;
      
          CMMap _CMMap; //it contains the sum of the CM calculated before. The normalization for the number of events it is done at the end when it is written in the DetSetVector.
};


SiStripMeanCMExtractor::SiStripMeanCMExtractor(const edm::ParameterSet& conf) : 
     _inputTag(conf.getParameter<edm::InputTag> ("CMCollection")),
        _Algorithm(conf.getParameter<std::string>("Algorithm")),
        _nEventsToUse(conf.getParameter<uint32_t>("NEvents"))
{

         
    if(_nEventsToUse < 1) _nEventsToUse=1;
        produces< edm::DetSetVector<SiStripProcessedRawDigi> > ("MEANAPVCM"); 
}


SiStripMeanCMExtractor::~SiStripMeanCMExtractor()
{
 
}

void SiStripMeanCMExtractor::init(const edm::EventSetup& es){
     
        uint32_t p_cache_id = es.get<SiStripPedestalsRcd>().cacheIdentifier();
        
        if(p_cache_id != pedestal_cache_id_) {
                es.get<SiStripPedestalsRcd>().get( pedestalHandle_ );
                pedestal_cache_id_ = p_cache_id;
        }
}

// ------------ method called to produce the data  ------------
void
SiStripMeanCMExtractor::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace edm;
   
     
   //if(_actualEvent > _nEventsToUse) return;
   
   std::vector<edm::DetSet<SiStripProcessedRawDigi> > meancm;
   
   if(_Algorithm == "StoredCM"){
        edm::Handle< edm::DetSetVector<SiStripProcessedRawDigi> > inputCM;
        iEvent.getByLabel(_inputTag,inputCM);
        
        this->StoreMean(*inputCM);
        this->ConvertMeanMapToDetSetVector(meancm);
        
   } else if (_Algorithm == "Pedestals"){
     this->init(iSetup);
         
     edm::Handle< edm::DetSetVector<SiStripRawDigi> > input;
     iEvent.getByLabel(_inputTag,input);
         
     this->CMExtractorFromPedestals(*input,meancm);
   }
   
   ++_actualEvent;
   
    
        
        
        std::auto_ptr< edm::DetSetVector<SiStripProcessedRawDigi> > outputMeanCM(new edm::DetSetVector<SiStripProcessedRawDigi>(meancm) );
    iEvent.put( outputMeanCM,"MEANAPVCM");
   
}

void SiStripMeanCMExtractor::CMExtractorFromPedestals(const edm::DetSetVector<SiStripRawDigi>& input, std::vector<edm::DetSet<SiStripProcessedRawDigi> >& meancm){
        meancm.clear();
        meancm.reserve(15000);    
        
         for ( edm::DetSetVector<SiStripRawDigi>::const_iterator 
          rawDigis = input.begin(); rawDigis != input.end(); rawDigis++) {
         SiStripPedestals::Range detPedestalRange = pedestalHandle_->getRange(rawDigis->id);
                 edm::DetSet<SiStripProcessedRawDigi> MeanCMDetSet(rawDigis->id);
                
                for(uint16_t APV = 0; APV < rawDigis->size()/128; ++APV){
                        uint16_t MinPed =0;
                        for(uint16_t strip = APV*128; strip< (APV+1)*128; ++strip){
                          uint16_t ped =  (uint16_t)pedestalHandle_->getPed(strip,detPedestalRange);
                          if(ped < MinPed) MinPed = ped;
                        }
                        if(MinPed>128) MinPed=128;
                        MeanCMDetSet.push_back(MinPed);
                }
                
                meancm.push_back(MeanCMDetSet); 
        }
}

void SiStripMeanCMExtractor::StoreMean(const edm::DetSetVector<SiStripProcessedRawDigi>& Input){
        
        uint32_t detId;
        CMMap::iterator itMap;
        edm::DetSetVector<SiStripProcessedRawDigi>::const_iterator itInput;
        
        for(itInput = Input.begin(); itInput != Input.end(); ++itInput){
                detId = itInput->id;
                itMap = _CMMap.find(detId);
                edm::DetSet<SiStripProcessedRawDigi>::const_iterator itCM;
                std::vector<float> MeanCMNValue;
                MeanCMNValue.clear();
                if(itMap!=_CMMap.end()){   //the detId was already found
                    std::vector< float >& MapContent = itMap->second;
                        std::vector<float>::iterator itMapVector = MapContent.begin();
                        for(itCM = itInput->begin(); itCM != itInput->end(); ++itCM, ++itMapVector){
                                MeanCMNValue.push_back(itCM->adc() + *itMapVector); 
            }
                        _CMMap.erase(itMap);
            _CMMap.insert(itMap, std::pair<uint32_t, std::vector<float> >(detId,MeanCMNValue));                 
                } else {                 //no detId found
                        for(itCM = itInput->begin(); itCM != itInput->end(); ++itCM) MeanCMNValue.push_back(itCM->adc());                       
                        _CMMap.insert(std::pair<uint32_t, std::vector<float> >(detId,MeanCMNValue));
                }
        }
  
}

void 
SiStripMeanCMExtractor::ConvertMeanMapToDetSetVector(std::vector<edm::DetSet<SiStripProcessedRawDigi> >& meancm){
        CMMap::iterator itMap;
        std::vector<float>::const_iterator itMapVector;
        
        meancm.clear();
        meancm.reserve(15000);    
        
    for(itMap = _CMMap.begin(); itMap != _CMMap.end(); ++itMap){
       edm::DetSet<SiStripProcessedRawDigi> MeanCMDetSet(itMap->first);
           for(itMapVector = (itMap->second).begin(); itMapVector != (itMap->second).end(); ++itMapVector) MeanCMDetSet.push_back(*itMapVector/(float)_actualEvent);
           meancm.push_back(MeanCMDetSet);      
        }       
    
}
void 
SiStripMeanCMExtractor::beginJob()
{
        _actualEvent =1;
                
        _CMMap.clear();
        
}

void 
SiStripMeanCMExtractor::endJob() {
    
           
}

DEFINE_FWK_MODULE(SiStripMeanCMExtractor);