DataMixingSiStripRawWorker.cc

Go to the documentation of this file.

// File: DataMixingSiStripRawWorker.cc
// Description:  see DataMixingSiStripRawWorker.h
// Author:  Mike Hildreth, University of Notre Dame
//
//--------------------------------------------
 
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include <map>
#include <memory>
//
//
#include "DataMixingSiStripRawWorker.h"
 
using namespace std;
 
namespace edm {
 
  // Virtual constructor
 
  DataMixingSiStripRawWorker::DataMixingSiStripRawWorker() {}
 
  // Constructor
  DataMixingSiStripRawWorker::DataMixingSiStripRawWorker(const edm::ParameterSet &ps, edm::ConsumesCollector &&iC)
      : label_(ps.getParameter<std::string>("Label"))
 
  {
    // get the subdetector names
    //    this->getSubdetectorNames();  //something like this may be useful to
    //    check what we are supposed to do...
 
    // declare the products to produce
 
    Sistripdigi_collectionSig_ = ps.getParameter<edm::InputTag>("SistripdigiCollectionSig");
    SistripLabelSig_ = ps.getParameter<edm::InputTag>("SistripLabelSig");
 
    SiStripPileInputTag_ = ps.getParameter<edm::InputTag>("SiStripPileInputTag");
    SiStripRawInputTag_ = ps.getParameter<edm::InputTag>("SiStripRawInputTag");
 
    SiStripDigiCollectionDM_ = ps.getParameter<std::string>("SiStripDigiCollectionDM");
 
    SiStripRawDigiSource_ = ps.getParameter<std::string>("SiStripRawDigiSource");
 
    // clear local storage for this event
    SiHitStorage_.clear();
 
    edm::InputTag tag = edm::InputTag(Sistripdigi_collectionSig_.label(), SistripLabelSig_.label());
 
    SiStripInputTok_ = iC.consumes<edm::DetSetVector<SiStripDigi>>(tag);
    SiStripRawInputTok_ = iC.consumes<edm::DetSetVector<SiStripRawDigi>>(SiStripRawInputTag_);
  }
 
  // Virtual destructor needed.
  DataMixingSiStripRawWorker::~DataMixingSiStripRawWorker() {}
 
  void DataMixingSiStripRawWorker::addSiStripSignals(const edm::Event &e) {
    edm::Handle<edm::DetSetVector<SiStripDigi>> hSSD;
    edm::Handle<edm::DetSetVector<SiStripRawDigi>> hSSRD;
 
    if (SiStripRawDigiSource_ == "SIGNAL") {
      e.getByToken(SiStripRawInputTok_, hSSRD);
      rawdigicollection_ = hSSRD.product();
    } else if (SiStripRawDigiSource_ == "PILEUP") {
      e.getByToken(SiStripInputTok_, hSSD);
      digicollection_ = hSSD.product();
    }  // else {
       // std::cout << "you shouldn't be here" << std::endl;
       //}
 
  }  // end of addSiStripSignals
 
  void DataMixingSiStripRawWorker::addSiStripPileups(const int bcr,
                                                     const EventPrincipal *ep,
                                                     unsigned int eventNr,
                                                     ModuleCallingContext const *mcc) {
    LogDebug("DataMixingSiStripRawWorker")
        << "\n===============> adding pileups from event  " << ep->id() << " for bunchcrossing " << bcr;
 
    std::shared_ptr<Wrapper<edm::DetSetVector<SiStripDigi>> const> pSSD;
    std::shared_ptr<Wrapper<edm::DetSetVector<SiStripRawDigi>> const> pSSRD;
 
    if (SiStripRawDigiSource_ == "SIGNAL") {
      pSSD = getProductByTag<edm::DetSetVector<SiStripDigi>>(*ep, SiStripPileInputTag_, mcc);
      digicollection_ = const_cast<edm::DetSetVector<SiStripDigi> *>(pSSD->product());
    } else if (SiStripRawDigiSource_ == "PILEUP") {
      pSSRD = getProductByTag<edm::DetSetVector<SiStripRawDigi>>(*ep, SiStripRawInputTag_, mcc);
      rawdigicollection_ = const_cast<edm::DetSetVector<SiStripRawDigi> *>(pSSRD->product());
    } else {
      std::cout << "you shouldn't be here" << std::endl;
    }
 
  }  // end of addSiStripPileups
 
  void DataMixingSiStripRawWorker::putSiStrip(edm::Event &e) {
    //------------------
    //  (1) Fill a map from the Digi collection
    //
 
    // fill in maps of SiStripDigis
    OneDetectorMap LocalMap;
 
    // loop on all detsets (detectorIDs) inside the input collection
    edm::DetSetVector<SiStripDigi>::const_iterator DSViter = digicollection_->begin();
    for (; DSViter != digicollection_->end(); DSViter++) {
#ifdef DEBUG
      LogDebug("DataMixingSiStripRawWorker") << "Processing DetID " << DSViter->id;
#endif
 
      LocalMap.clear();
      LocalMap.reserve((DSViter->data).size());
      LocalMap.insert(LocalMap.end(), (DSViter->data).begin(), (DSViter->data).end());
 
      SiHitStorage_.insert(SiGlobalIndex::value_type(DSViter->id, LocalMap));
    }
 
    //------------------
    //  (2) Loop over the input RawDigi collection and add the Digis from the map
    //
 
    // collection of RawDigis to put back in the event
    std::vector<edm::DetSet<SiStripRawDigi>> vSiStripRawDigi;
 
    // loop on all detsets (detectorIDs) inside the SiStripRawDigis collection
    edm::DetSetVector<SiStripRawDigi>::const_iterator rawDSViter = rawdigicollection_->begin();
    for (; rawDSViter != rawdigicollection_->end(); rawDSViter++) {
      // Make empty collection with this detID
      edm::DetSet<SiStripRawDigi> SSRD(rawDSViter->id);
 
      // find local map (if it exists) for this detector ID
      SiGlobalIndex::const_iterator itest;
      itest = SiHitStorage_.find(rawDSViter->id);
 
      // if detID already has digis in existing map, add them to rawdigis
      if (itest != SiHitStorage_.end()) {
#ifdef DEBUG
        LogDebug("DataMixingSiStripRawWorker") << "Pileups: Processing DetID " << rawDSViter->id;
#endif
 
        // get the map from storage
        LocalMap = itest->second;
        OneDetectorMap::const_iterator iLocal = LocalMap.begin();
 
        // loop on all strips in rawdigi detset
        int currentstrip = 0;
        edm::DetSet<SiStripRawDigi>::const_iterator iRawDigi = rawDSViter->begin();
        while (iRawDigi != rawDSViter->end()) {
          int ADCSum = iRawDigi->adc();
 
          // if current strip exists in map, add ADC values
          if (iLocal->strip() == currentstrip) {
            ADCSum += iLocal->adc();
            iLocal++;
          }
 
          // put ADC sum in DetSet and go to next strip
          SSRD.push_back(SiStripRawDigi(ADCSum));
          iRawDigi++;
          currentstrip++;
        }
 
        // copy combined digi+rawdigi into rawdigi DetSetVector
        vSiStripRawDigi.push_back(SSRD);
 
        // otherwise, just copy the rawdigis from the background event to the
        // output
      } else {
        vSiStripRawDigi.push_back(*rawDSViter);
      }
    }
 
    //------------------
    //  (3) Put the new RawDigi collection back into the event
    //
 
    // make new raw digi collection
    std::unique_ptr<edm::DetSetVector<SiStripRawDigi>> MySiStripRawDigis(
        new edm::DetSetVector<SiStripRawDigi>(vSiStripRawDigi));
 
    // put collection
    e.put(std::move(MySiStripRawDigis), SiStripDigiCollectionDM_);
 
    // clear local storage for this event
    SiHitStorage_.clear();
  }
 
}  // namespace edm