/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/SimMuon/GEMDigitizer/src/GEMSim.cc

Go to the documentation of this file.

#include "GEMSim.h"
#include "SimMuon/GEMDigitizer/src/GEMSimSetUp.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"


void GEMSim::fillDigis(int rollDetId, GEMDigiCollection& digis)
{
  for (auto d: strips_)
  {
    if (d.second != -999)
    {
      GEMDigi digi(d.first, d.second); // (strip, bx)
      digis.insertDigi(GEMDetId(rollDetId), digi);
      addLinks(d.first, d.second);
    }
  }
  strips_.clear();
}


void GEMSim::addLinks(unsigned int strip, int bx)
{
  std::pair<unsigned int, int> digi(strip, bx);
  auto channelHitItr = detectorHitMap_.equal_range(digi);

  // find the fraction contribution for each SimTrack
  std::map<int, float> simTrackChargeMap;
  std::map<int, EncodedEventId> eventIdMap;
  float totalCharge = 0;
  for(auto hitItr = channelHitItr.first; hitItr != channelHitItr.second; ++hitItr)
  {
    const PSimHit * hit = hitItr->second;
    // might be zero for unit tests and such
    if(hit == nullptr) continue;
    
    int simTrackId = hit->trackId();
    //float charge = hit->getCharge();
    const float charge = 1.f;
    auto chargeItr = simTrackChargeMap.find(simTrackId);
    if( chargeItr == simTrackChargeMap.end() )
    {
      simTrackChargeMap[simTrackId] = charge;
      eventIdMap[simTrackId] = hit->eventId();
    }
    else 
    {
      chargeItr->second += charge;
    }
    totalCharge += charge;
  }

  for(auto &charge: simTrackChargeMap)
  {
    int simTrackId = charge.first;
    stripDigiSimLinks_.push_back( 
      StripDigiSimLink(strip, simTrackId, eventIdMap[simTrackId], charge.second/totalCharge ));
  }
}