DTHitAssociator Class Reference

#include <DTHitAssociator.h>

Classes
class	Config

Public Types
typedef std::map< DTWireId, std::vector< DTDigi > >	DigiMap
typedef std::map< DTWireId, std::vector< DTDigiSimLink > >	LinksMap
typedef std::pair< PSimHit, bool >	PSimHit_withFlag
typedef std::map< DTWireId, std::vector< DTRecHit1DPair > >	RecHitMap
typedef std::pair< uint32_t, EncodedEventId >	SimHitIdpr
typedef std::map< DTWireId, std::vector< PSimHit_withFlag > >	SimHitMap

Public Member Functions
std::vector< SimHitIdpr >	associateDTHitId (const DTRecHit1D *dtrechit) const
std::vector< PSimHit >	associateHit (const TrackingRecHit &hit) const
std::vector< SimHitIdpr >	associateHitId (const TrackingRecHit &hit) const
	DTHitAssociator (const edm::Event &, const edm::EventSetup &, const Config &, bool printRtS)

Public Attributes
DigiMap	mapOfDigi
LinksMap	mapOfLinks
RecHitMap	mapOfRecHit
SimHitMap	mapOfSimHit

Private Member Functions
void	initEvent (const edm::Event &, const edm::EventSetup &)
bool	SimHitOK (const edm::ESHandle< DTGeometry > &, const PSimHit &)

Private Attributes
Config const &	config_
bool	printRtS

Detailed Description

Definition at line 25 of file DTHitAssociator.h.

Member Typedef Documentation

typedef std::map<DTWireId, std::vector<DTDigi> > DTHitAssociator::DigiMap

Definition at line 31 of file DTHitAssociator.h.

typedef std::map<DTWireId, std::vector<DTDigiSimLink> > DTHitAssociator::LinksMap

Definition at line 32 of file DTHitAssociator.h.

typedef std::pair<PSimHit, bool> DTHitAssociator::PSimHit_withFlag

Definition at line 28 of file DTHitAssociator.h.

typedef std::map<DTWireId, std::vector<DTRecHit1DPair> > DTHitAssociator::RecHitMap

Definition at line 30 of file DTHitAssociator.h.

typedef std::pair<uint32_t, EncodedEventId> DTHitAssociator::SimHitIdpr

Definition at line 27 of file DTHitAssociator.h.

typedef std::map<DTWireId, std::vector<PSimHit_withFlag> > DTHitAssociator::SimHitMap

Definition at line 29 of file DTHitAssociator.h.

Constructor & Destructor Documentation

DTHitAssociator::DTHitAssociator	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const Config &	conf,
		bool	printRtS
	)

Definition at line 53 of file DTHitAssociator.cc.

References initEvent().

    :  // input collection labels
      config_(conf),
      printRtS(true)

{
  initEvent(iEvent, iSetup);
}

Member Function Documentation

std::vector< DTHitAssociator::SimHitIdpr > DTHitAssociator::associateDTHitId

(

const DTRecHit1D *

dtrechit

)

const

Definition at line 267 of file DTHitAssociator.cc.

References DTHitAssociator::Config::associatorByWire, config_, DTRecHit1D::digiTime(), PSimHit::eventId(), newFWLiteAna::found, LogTrace, mapOfLinks, mapOfSimHit, PSimHit::trackId(), and DTRecHit1D::wireId().

Referenced by associateHit(), associateHitId(), and MuonAssociatorByHitsHelper::getMatchedIds().

                                                                                                       {
  std::vector<SimHitIdpr> matched;

  DTWireId wireid = dtrechit->wireId();

  if (config_.associatorByWire) {
    // matching based on DTWireId : take only "valid" SimHits on that wire

    auto found = mapOfSimHit.find(wireid);
    if (found != mapOfSimHit.end()) {
      for (vector<PSimHit_withFlag>::const_iterator hitIT = found->second.begin(); hitIT != found->second.end();
           ++hitIT) {
        bool valid_hit = hitIT->second;
        PSimHit this_hit = hitIT->first;

        if (valid_hit) {
          SimHitIdpr currentId(this_hit.trackId(), this_hit.eventId());
          matched.push_back(currentId);
        }
      }
    }
  }

  else {
    // matching based on DTDigiSimLink

    float theTime = dtrechit->digiTime();
    int theNumber(-1);

    auto found = mapOfLinks.find(wireid);

    if (found != mapOfLinks.end()) {
      // DTDigiSimLink::time() is set equal to DTDigi::time() only for the
      // DTDigiSimLink corresponding to the digitizied PSimHit other
      // DTDigiSimLinks associated to the same DTDigi are identified by having
      // the same DTDigiSimLink::number()

      // first find the associated digi Number
      for (vector<DTDigiSimLink>::const_iterator linkIT = found->second.begin(); linkIT != found->second.end();
           ++linkIT) {
        float digitime = linkIT->time();
        if (fabs(digitime - theTime) < 0.1) {
          theNumber = linkIT->number();
        }
      }

      // then get all the DTDigiSimLinks with that digi Number (corresponding to
      // valid SimHits
      //  within a time window of the order of the time resolution, specified in
      //  the DTDigitizer)
      for (vector<DTDigiSimLink>::const_iterator linkIT = found->second.begin(); linkIT != found->second.end();
           ++linkIT) {
        int digiNr = linkIT->number();
        if (digiNr == theNumber) {
          SimHitIdpr currentId(linkIT->SimTrackId(), linkIT->eventId());
          matched.push_back(currentId);
        }
      }

    } else {
      LogTrace("DTHitAssociator") << "*** ERROR in DTHitAssociator::associateDTHitId - DTRecHit1D: " << *dtrechit
                                  << " has no associated DTDigiSimLink !" << endl;
      return matched;
    }
  }

  return matched;
}

std::vector< PSimHit > DTHitAssociator::associateHit

(

const TrackingRecHit &

hit

)

const

Definition at line 336 of file DTHitAssociator.cc.

References associateDTHitId(), DTHitAssociator::Config::associatorByWire, config_, newFWLiteAna::found, LogTrace, mapOfSimHit, and DTRecHit1D::wireId().

Referenced by MuonAssociatorByHitsHelper::getMatchedIds().

                                                                                {
  std::vector<PSimHit> simhits;
  std::vector<SimHitIdpr> simtrackids;

  const TrackingRecHit *hitp = &hit;
  const DTRecHit1D *dtrechit = dynamic_cast<const DTRecHit1D *>(hitp);

  if (dtrechit) {
    DTWireId wireid = dtrechit->wireId();

    if (config_.associatorByWire) {
      // matching based on DTWireId : take only "valid" SimHits on that wire

      auto found = mapOfSimHit.find(wireid);
      if (found != mapOfSimHit.end()) {
        for (vector<PSimHit_withFlag>::const_iterator hitIT = found->second.begin(); hitIT != found->second.end();
             ++hitIT) {
          bool valid_hit = hitIT->second;
          if (valid_hit)
            simhits.push_back(hitIT->first);
        }
      }
    } else {
      // matching based on DTDigiSimLink

      simtrackids = associateDTHitId(dtrechit);

      for (vector<SimHitIdpr>::const_iterator idIT = simtrackids.begin(); idIT != simtrackids.end(); ++idIT) {
        uint32_t trId = idIT->first;
        EncodedEventId evId = idIT->second;

        auto found = mapOfSimHit.find(wireid);
        if (found != mapOfSimHit.end()) {
          for (vector<PSimHit_withFlag>::const_iterator hitIT = found->second.begin(); hitIT != found->second.end();
               ++hitIT) {
            if (hitIT->first.trackId() == trId && hitIT->first.eventId() == evId)
              simhits.push_back(hitIT->first);
          }
        }
      }
    }

  } else {
    LogTrace("DTHitAssociator") << "*** WARNING in DTHitAssociator::associateHit, null dynamic_cast !";
  }
  return simhits;
}

std::vector< DTHitAssociator::SimHitIdpr > DTHitAssociator::associateHitId

(

const TrackingRecHit &

hit

)

const

Definition at line 253 of file DTHitAssociator.cc.

References associateDTHitId(), and LogTrace.

                                                                                                    {
  std::vector<SimHitIdpr> simtrackids;
  const TrackingRecHit *hitp = &hit;
  const DTRecHit1D *dtrechit = dynamic_cast<const DTRecHit1D *>(hitp);

  if (dtrechit) {
    simtrackids = associateDTHitId(dtrechit);
  } else {
    LogTrace("DTHitAssociator") << "*** WARNING in DTHitAssociator::associateHitId, null dynamic_cast "
                                   "!";
  }
  return simtrackids;
}

void DTHitAssociator::initEvent ( const edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

private

Definition at line 65 of file DTHitAssociator.cc.

References DTHitAssociator::Config::associatorByWire, SplitLinear::begin, MuonDigiCollection< IndexType, DigiType >::begin(), EncodedEventId::bunchCrossing(), config_, DTHitAssociator::Config::crossingframe, DTHitAssociator::Config::DTdigisimlinkTag, DTHitAssociator::Config::DTdigisimlinkToken, DTHitAssociator::Config::DTdigiTag, DTHitAssociator::Config::DTdigiToken, DTHitAssociator::Config::DTrechitTag, DTHitAssociator::Config::DTrechitToken, DTHitAssociator::Config::DTsimhitsTag, DTHitAssociator::Config::DTsimhitsToken, DTHitAssociator::Config::DTsimhitsXFTag, DTHitAssociator::Config::DTsimhitsXFToken, DTHitAssociator::Config::dumpDT, MuonDigiCollection< IndexType, DigiType >::end(), EncodedEventId::event(), PSimHit::eventId(), DTHitAssociator::Config::geomToken, edm::Event::get(), edm::EventSetup::getHandle(), edm::Event::getHandle(), edm::HandleBase::isValid(), edm::InputTag::label(), MainPageGenerator::link, DTHitAssociator::Config::links_exist, LogTrace, mapOfDigi, mapOfLinks, mapOfRecHit, mapOfSimHit, PSimHit::particleType(), printRtS, PSimHit::processType(), sistrip::SpyUtilities::range(), SimHitOK(), and PSimHit::trackId().

Referenced by DTHitAssociator().

                                                                                   {
  LogTrace("DTHitAssociator") << "DTHitAssociator constructor: dumpDT = " << config_.dumpDT
                              << ", crossingframe = " << config_.crossingframe
                              << ", links_exist = " << config_.links_exist
                              << ", associatorByWire = " << config_.associatorByWire;

  // need DT Geometry to discard hits for which the drift time parametrisation
  // is not applicable
  edm::ESHandle<DTGeometry> muonGeom = iSetup.getHandle(config_.geomToken);

  // Get the DT SimHits from the event and map PSimHit by DTWireId
  mapOfSimHit.clear();
  bool takeHit(true);

  if (config_.crossingframe) {
    LogTrace("DTHitAssociator") << "getting CrossingFrame<PSimHit> collection - " << config_.DTsimhitsXFTag;
    CrossingFrame<PSimHit> const &xFrame = iEvent.get(config_.DTsimhitsXFToken);
    unique_ptr<MixCollection<PSimHit>> DTsimhits(new MixCollection<PSimHit>(&xFrame));
    LogTrace("DTHitAssociator") << "... size = " << DTsimhits->size();
    MixCollection<PSimHit>::MixItr isimhit;
    for (isimhit = DTsimhits->begin(); isimhit != DTsimhits->end(); isimhit++) {
      DTWireId wireid((*isimhit).detUnitId());
      takeHit = SimHitOK(muonGeom, *isimhit);
      mapOfSimHit[wireid].push_back(make_pair(*isimhit, takeHit));
    }
  } else if (!config_.DTsimhitsTag.label().empty()) {
    LogTrace("DTHitAssociator") << "getting PSimHit collection - " << config_.DTsimhitsTag;
    edm::PSimHitContainer const &DTsimhits = iEvent.get(config_.DTsimhitsToken);
    LogTrace("DTHitAssociator") << "... size = " << DTsimhits.size();
    edm::PSimHitContainer::const_iterator isimhit;
    for (isimhit = DTsimhits.begin(); isimhit != DTsimhits.end(); isimhit++) {
      DTWireId wireid((*isimhit).detUnitId());
      takeHit = SimHitOK(muonGeom, *isimhit);
      mapOfSimHit[wireid].push_back(make_pair(*isimhit, takeHit));
    }
  }

  // Get the DT Digi collection from the event
  mapOfDigi.clear();
  LogTrace("DTHitAssociator") << "getting DTDigi collection - " << config_.DTdigiTag;
  edm::Handle<DTDigiCollection> digis = iEvent.getHandle(config_.DTdigiToken);

  if (digis.isValid()) {
    // Map DTDigi by DTWireId
    for (DTDigiCollection::DigiRangeIterator detUnit = digis->begin(); detUnit != digis->end(); ++detUnit) {
      const DTLayerId &layerid = (*detUnit).first;
      const DTDigiCollection::Range &range = (*detUnit).second;

      DTDigiCollection::const_iterator digi;
      for (digi = range.first; digi != range.second; ++digi) {
        DTWireId wireid(layerid, (*digi).wire());
        mapOfDigi[wireid].push_back(*digi);
      }
    }
  } else {
    LogTrace("DTHitAssociator") << "... NO DTDigi collection found !";
  }

  mapOfLinks.clear();
  if (config_.links_exist) {
    // Get the DT DigiSimLink collection from the event and map DTDigiSimLink by
    // DTWireId
    LogTrace("DTHitAssociator") << "getting DTDigiSimLink collection - " << config_.DTdigisimlinkTag;
    DTDigiSimLinkCollection const &digisimlinks = iEvent.get(config_.DTdigisimlinkToken);

    for (DTDigiSimLinkCollection::DigiRangeIterator detUnit = digisimlinks.begin(); detUnit != digisimlinks.end();
         ++detUnit) {
      const DTLayerId &layerid = (*detUnit).first;
      const DTDigiSimLinkCollection::Range &range = (*detUnit).second;

      DTDigiSimLinkCollection::const_iterator link;
      for (link = range.first; link != range.second; ++link) {
        DTWireId wireid(layerid, (*link).wire());
        mapOfLinks[wireid].push_back(*link);
      }
    }
  }

  if (config_.dumpDT && printRtS) {
    // Get the DT rechits from the event
    LogTrace("DTHitAssociator") << "getting DTRecHit1DPair collection - " << config_.DTrechitTag;
    DTRecHitCollection const &DTrechits = iEvent.get(config_.DTrechitToken);
    LogTrace("DTHitAssociator") << "... size = " << DTrechits.size();

    // map DTRecHit1DPair by DTWireId
    mapOfRecHit.clear();
    DTRecHitCollection::const_iterator rechit;
    for (rechit = DTrechits.begin(); rechit != DTrechits.end(); ++rechit) {
      DTWireId wireid = (*rechit).wireId();
      mapOfRecHit[wireid].push_back(*rechit);
    }

    if (mapOfSimHit.end() != mapOfSimHit.begin()) {
      LogTrace("DTHitAssociator") << "\n *** Dump DT PSimHit's ***";

      int jwire = 0;
      int ihit = 0;

      for (SimHitMap::const_iterator mapIT = mapOfSimHit.begin(); mapIT != mapOfSimHit.end(); ++mapIT, jwire++) {
        DTWireId wireid = (*mapIT).first;
        for (vector<PSimHit_withFlag>::const_iterator hitIT = mapOfSimHit[wireid].begin();
             hitIT != mapOfSimHit[wireid].end();
             hitIT++, ihit++) {
          PSimHit hit = hitIT->first;
          LogTrace("DTHitAssociator") << "PSimHit " << ihit << ", wire " << wireid  //<<", detID = "<<hit.detUnitId()
                                      << ", SimTrack Id:" << hit.trackId() << "/Evt:(" << hit.eventId().event() << ","
                                      << hit.eventId().bunchCrossing() << ") "
                                      << ", pdg = " << hit.particleType() << ", procs = " << hit.processType();
        }
      }
    } else {
      LogTrace("DTHitAssociator") << "\n *** There are NO DT PSimHit's ***";
    }

    if (mapOfDigi.end() != mapOfDigi.begin()) {
      int jwire = 0;
      int ihit = 0;

      for (DigiMap::const_iterator mapIT = mapOfDigi.begin(); mapIT != mapOfDigi.end(); ++mapIT, jwire++) {
        LogTrace("DTHitAssociator") << "\n *** Dump DT digis ***";

        DTWireId wireid = (*mapIT).first;
        for (vector<DTDigi>::const_iterator hitIT = mapOfDigi[wireid].begin(); hitIT != mapOfDigi[wireid].end();
             hitIT++, ihit++) {
          LogTrace("DTHitAssociator") << "DTDigi " << ihit << ", wire " << wireid << ", number = " << hitIT->number()
                                      << ", TDC counts = " << hitIT->countsTDC();
        }
      }
    } else {
      LogTrace("DTHitAssociator") << "\n *** There are NO DTDigi's ***";
    }

    if (mapOfRecHit.end() != mapOfRecHit.begin())
      LogTrace("DTHitAssociator") << "\n *** Analyze DTRecHitCollection by DTWireId ***";

    int iwire = 0;
    for (RecHitMap::const_iterator mapIT = mapOfRecHit.begin(); mapIT != mapOfRecHit.end(); ++mapIT, iwire++) {
      DTWireId wireid = (*mapIT).first;
      LogTrace("DTHitAssociator") << "\n======================================="
                                     "============================";
      LogTrace("DTHitAssociator") << "wire index = " << iwire << "  *** DTWireId = "
                                  << " (" << wireid << ")";

      if (mapOfSimHit.find(wireid) != mapOfSimHit.end()) {
        LogTrace("DTHitAssociator") << "\n" << mapOfSimHit[wireid].size() << " SimHits (PSimHit):";

        for (vector<PSimHit_withFlag>::const_iterator hitIT = mapOfSimHit[wireid].begin();
             hitIT != mapOfSimHit[wireid].end();
             ++hitIT) {
          stringstream tId;
          tId << (hitIT->first).trackId();
          string simhitlog = "\t SimTrack Id = " + tId.str();
          if (hitIT->second)
            simhitlog = simhitlog + "\t -VALID HIT-";
          else
            simhitlog = simhitlog + "\t -not valid hit-";
          LogTrace("DTHitAssociator") << simhitlog;
        }
      }

      if (mapOfLinks.find(wireid) != mapOfLinks.end()) {
        LogTrace("DTHitAssociator") << "\n" << mapOfLinks[wireid].size() << " Links (DTDigiSimLink):";

        for (vector<DTDigiSimLink>::const_iterator hitIT = mapOfLinks[wireid].begin();
             hitIT != mapOfLinks[wireid].end();
             ++hitIT) {
          LogTrace("DTHitAssociator") << "\t digi number = " << hitIT->number() << ", time = " << hitIT->time()
                                      << ", SimTrackId = " << hitIT->SimTrackId();
        }
      }

      if (mapOfDigi.find(wireid) != mapOfDigi.end()) {
        LogTrace("DTHitAssociator") << "\n" << mapOfDigi[wireid].size() << " Digis (DTDigi):";
        for (vector<DTDigi>::const_iterator hitIT = mapOfDigi[wireid].begin(); hitIT != mapOfDigi[wireid].end();
             ++hitIT) {
          LogTrace("DTHitAssociator") << "\t digi number = " << hitIT->number() << ", time = " << hitIT->time();
        }
      }

      LogTrace("DTHitAssociator") << "\n" << (*mapIT).second.size() << " RecHits (DTRecHit1DPair):";
      for (vector<DTRecHit1DPair>::const_iterator vIT = (*mapIT).second.begin(); vIT != (*mapIT).second.end(); ++vIT) {
        LogTrace("DTHitAssociator") << "\t digi time = " << vIT->digiTime();
      }
    }
  }
}

bool DTHitAssociator::SimHitOK ( const edm::ESHandle< DTGeometry > &  muongeom, const PSimHit &  simhit  )

private

Definition at line 384 of file DTHitAssociator.cc.

References PSimHit::detUnitId(), PSimHit::entryPoint(), PSimHit::exitPoint(), PSimHit::localPosition(), DTTopology::none, DTTopology::onWhichBorder(), mps_fire::result, DTTopology::wirePosition(), x, PV3DBase< T, PVType, FrameType >::x(), DTTopology::xMax, DTTopology::xMin, PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by initEvent().

                                                                                             {
  bool result = true;

  DTWireId wireid(simhit.detUnitId());
  const DTLayer *dtlayer = muongeom->layer(wireid.layerId());
  LocalPoint entryP = simhit.entryPoint();
  LocalPoint exitP = simhit.exitPoint();
  const DTTopology &topo = dtlayer->specificTopology();
  float xwire = topo.wirePosition(wireid.wire());
  float xEntry = entryP.x() - xwire;
  float xExit = exitP.x() - xwire;
  DTTopology::Side entrySide = topo.onWhichBorder(xEntry, entryP.y(), entryP.z());
  DTTopology::Side exitSide = topo.onWhichBorder(xExit, exitP.y(), exitP.z());

  bool noParametrisation =
      ((entrySide == DTTopology::none || exitSide == DTTopology::none) || (entrySide == exitSide) ||
       ((entrySide == DTTopology::xMin && exitSide == DTTopology::xMax) ||
        (entrySide == DTTopology::xMax && exitSide == DTTopology::xMin)));

  // discard hits where parametrization can not be used
  if (noParametrisation) {
    result = false;
    return result;
  }

  float x;
  LocalPoint hitPos = simhit.localPosition();

  if (fabs(hitPos.z()) < 0.002) {
    // hit center within 20 um from z=0, no need to extrapolate.
    x = hitPos.x() - xwire;
  } else {
    x = xEntry - (entryP.z() * (xExit - xEntry)) / (exitP.z() - entryP.z());
  }

  // discard hits where x is out of range of the parametrization (|x|>2.1 cm)
  x *= 10.;  // cm -> mm
  if (fabs(x) > 21.)
    result = false;

  return result;
}

Member Data Documentation

Config const& DTHitAssociator::config_

private

Definition at line 77 of file DTHitAssociator.h.

Referenced by associateDTHitId(), associateHit(), and initEvent().

DigiMap DTHitAssociator::mapOfDigi

Definition at line 70 of file DTHitAssociator.h.

Referenced by initEvent().

LinksMap DTHitAssociator::mapOfLinks

Definition at line 71 of file DTHitAssociator.h.

Referenced by associateDTHitId(), and initEvent().

RecHitMap DTHitAssociator::mapOfRecHit

Definition at line 69 of file DTHitAssociator.h.

Referenced by initEvent().

SimHitMap DTHitAssociator::mapOfSimHit

Definition at line 68 of file DTHitAssociator.h.

Referenced by associateDTHitId(), associateHit(), and initEvent().

bool DTHitAssociator::printRtS

private

Definition at line 78 of file DTHitAssociator.h.

Referenced by initEvent().