---

MergedTruthProducer Class Reference

#include <SimGeneral/TrackingAnalysis/interface/MergedTruthProducer.h>

Inheritance diagram for MergedTruthProducer:

List of all members.

Public Member Functions
	MergedTruthProducer (const edm::ParameterSet &)
Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &)
Private Attributes
edm::ParameterSet	conf_
std::string	MessageCategory_

---

Detailed Description

Definition at line 10 of file MergedTruthProducer.h.

---

Constructor & Destructor Documentation

MergedTruthProducer::MergedTruthProducer

(

const edm::ParameterSet &

conf

)

[explicit]

Definition at line 15 of file MergedTruthProducer.cc.

References conf_, and MessageCategory_.

                                                                    {
  produces<TrackingVertexCollection>("MergedTrackTruth");
  produces<TrackingParticleCollection>("MergedTrackTruth");

  conf_ = conf;
  MessageCategory_ = "MergedTruthProducer";

}

---

Member Function Documentation

void MergedTruthProducer::produce	(	edm::Event &	event,
		const edm::EventSetup &
	)			[private, virtual]

Implements edm::EDProducer.

Definition at line 24 of file MergedTruthProducer.cc.

References TrackingParticle::addDecayVertex(), edm::RefVector< C, T, F >::begin(), TrackingVertex::clearDaughterTracks(), TrackingParticle::clearDecayVertices(), TrackingVertex::clearParentTracks(), TrackingParticle::clearParentVertex(), edm::RefVector< C, T, F >::end(), edm::Ref< C, T, F >::key(), and TrackingParticle::setParentVertex().

                                                                  {

  edm::Handle<TrackingParticleCollection> elecPH;
  edm::Handle<TrackingParticleCollection> rawPH;
  edm::Handle<TrackingVertexCollection>   rawVH;

  event.getByLabel("trackingtruthprod",rawPH);
  event.getByLabel("trackingtruthprod",rawVH);
  event.getByLabel("electrontruth","ElectronTrackTruth",    elecPH);

//  std::auto_ptr<TrackingParticleCollection>  trackCollection(new TrackingParticleCollection(rawPH.product()));
//  std::auto_ptr<TrackingParticleCollection>   elecCollection(new TrackingParticleCollection(elecPH.product()));
//  std::auto_ptr<TrackingVertexCollection>   vertexCollection(new TrackingVertexCollection(rawVH.product()));

// Create collections of things we will put in event and size appropriately
  auto_ptr<TrackingParticleCollection> tPC(new TrackingParticleCollection);
  auto_ptr<TrackingVertexCollection>   tVC(new TrackingVertexCollection  );
  tPC->reserve(rawPH->size());
  tVC->reserve(rawVH->size());

// Get references before put so we can cross reference
  TrackingParticleRefProd refTPC = event.getRefBeforePut<TrackingParticleCollection>("MergedTrackTruth");
  TrackingVertexRefProd   refTVC = event.getRefBeforePut<TrackingVertexCollection>("MergedTrackTruth");

  std::set<EncodedTruthId> electronGID;  // Keeps track of GeantIDs of electron tracks we've added

// Copy vertices discarding parent & child tracks

  for (TrackingVertexCollection::const_iterator iVertex = rawVH->begin(); iVertex != rawVH->end(); ++iVertex) {
    TrackingVertex newVertex = (*iVertex);
    newVertex.clearDaughterTracks();
    newVertex.clearParentTracks();
    tVC->push_back(newVertex);
  }

  uint eIndex = 0;
  for (TrackingParticleCollection::const_iterator iTrack = elecPH->begin(); iTrack != elecPH->end(); ++iTrack, ++eIndex) {

// Copy references from old vertex, set on new vertex, see comments in next loop

    TrackingVertexRef       sourceV = iTrack->parentVertex();
    TrackingVertexRefVector decayVs = iTrack->decayVertices();
    TrackingParticle newTrack = *iTrack;
    newTrack.clearParentVertex();
    newTrack.clearDecayVertices();
    uint parentIndex = sourceV.key();
    newTrack.setParentVertex(TrackingVertexRef(refTVC,parentIndex));
    (tVC->at(parentIndex)).addDaughterTrack(TrackingParticleRef(refTPC,eIndex));
    for (TrackingVertexRefVector::const_iterator iDecayV = decayVs.begin(); iDecayV != decayVs.end(); ++iDecayV) {
      uint daughterIndex = iDecayV->key();
      newTrack.addDecayVertex(TrackingVertexRef(refTVC,daughterIndex));
      (tVC->at(daughterIndex)).addParentTrack(TrackingParticleRef(refTPC,eIndex));
    }
    tPC->push_back(newTrack);

// Keep track of which tracks we did so we can skip later

    for (TrackingParticle::g4t_iterator g4T = iTrack->g4Track_begin(); g4T !=  iTrack->g4Track_end(); ++g4T) {
      uint GID = g4T->trackId();
      if (GID) {
        electronGID.insert(EncodedTruthId(iTrack->eventId(),GID));
      }
    }
  }

  for (TrackingParticleCollection::const_iterator iTrack = rawPH->begin(); iTrack != rawPH->end(); ++iTrack) {
    bool addTrack = false;
    for (TrackingParticle::g4t_iterator g4T = iTrack->g4Track_begin(); g4T !=  iTrack->g4Track_end(); ++g4T) {
      uint GID = g4T->trackId();
      if (electronGID.count(EncodedTruthId(iTrack->eventId(),GID))) {
        // Do nothing
      } else {
        addTrack = true;
      }
    }

    if (addTrack) { // Skip tracks that were in electron list
      TrackingVertexRef       sourceV = iTrack->parentVertex();
      TrackingVertexRefVector decayVs = iTrack->decayVertices();
      TrackingParticle newTrack = *iTrack;
      newTrack.clearParentVertex();
      newTrack.clearDecayVertices();

      // Set vertex indices for new vertex product and track references in those vertices

      uint parentIndex = sourceV.key(); // Index of parent vertex in vertex container
      uint tIndex      = tPC->size();   // Index of this track in track container
      newTrack.setParentVertex(TrackingVertexRef(refTVC,parentIndex));             // Add vertex to track
      (tVC->at(parentIndex)).addDaughterTrack(TrackingParticleRef(refTPC,tIndex)); // Add track to vertex
      for (TrackingVertexRefVector::const_iterator iDecayV = decayVs.begin(); iDecayV != decayVs.end(); ++iDecayV) {
        uint daughterIndex = iDecayV->key();
        newTrack.addDecayVertex(TrackingVertexRef(refTVC,daughterIndex));            // Add vertex to track
        (tVC->at(daughterIndex)).addParentTrack(TrackingParticleRef(refTPC,tIndex)); // Add track to vertex
      }
      tPC->push_back(newTrack);
    }
  }

// Put TrackingParticles and TrackingVertices in event

  event.put(tPC,"MergedTrackTruth");
  event.put(tVC,"MergedTrackTruth");

//  timers.pop();
//  timers.pop();
}

---

Member Data Documentation

edm::ParameterSet MergedTruthProducer::conf_ [private]

Definition at line 19 of file MergedTruthProducer.h.

Referenced by MergedTruthProducer().

std::string MergedTruthProducer::MessageCategory_ [private]

Definition at line 20 of file MergedTruthProducer.h.

Referenced by MergedTruthProducer().

---

The documentation for this class was generated from the following files:

• SimGeneral/TrackingAnalysis/interface/MergedTruthProducer.h
• SimGeneral/TrackingAnalysis/src/MergedTruthProducer.cc

---