2 #include "HepMC/GenEvent.h"
3 #include "HepMC/GenVertex.h"
4 #include "HepMC/GenParticle.h"
22 using namespace HepPDT;
35 nChargedParticleTracks(0),
103 const std::vector<SimVertex>& simVertices) {
110 unsigned nVtx = simVertices.size();
111 unsigned nTks = simTracks.size();
114 if ( nVtx == 0 )
return;
117 std::vector<int> myVertices(nVtx,-1);
118 std::vector<int> myTracks(nTks,-1);
123 std::map<unsigned, unsigned> geantToIndex;
124 for(
unsigned it=0; it<simTracks.size(); ++it ) {
125 geantToIndex[ simTracks[it].trackId() ] = it;
148 myFilter->setMainVertex(primaryVertex);
153 for(
unsigned trackId=0; trackId<nTks; ++trackId ) {
169 std::map<unsigned, unsigned >::iterator association
170 = geantToIndex.find( motherGeantId );
171 if(association != geantToIndex.end() )
172 motherId = association->second;
174 int originId = motherId == - 1 ? -1 : myTracks[motherId];
188 if ( myVertices[vertexId] == -1 )
196 int motherType = motherId == -1 ? 0 : simTracks[motherId].type();
198 bool notBremInDetector =
199 (
abs(motherType) != 11 &&
abs(motherType) != 13) ||
200 motherType != track.
type() ||
203 if ( notBremInDetector ) {
212 part.setID(track.
type());
218 if ( myTracks[trackId] >= 0 ) {
224 myTracks[trackId] = myTracks[motherId];
225 if ( myTracks[trackId] >= 0 ) {
234 for(
unsigned vertexId=0; vertexId<nVtx; ++vertexId ) {
237 if ( myVertices[vertexId] != -1 )
continue;
248 std::map<unsigned, unsigned >::iterator association
249 = geantToIndex.find( motherGeantId );
250 if(association != geantToIndex.end() )
251 motherId = association->second;
253 int originId = motherId == - 1 ? -1 : myTracks[motherId];
272 for(
int fsimi=0; fsimi < (int)
nTracks() ; ++fsimi) {
289 if ( mom.T() > 0. ) {
315 if ( myPart.
cos2ThetaV()>0.8 || mom.T() < 3. ) {
341 int genEventSize = myGenEvent.particles_size();
342 std::vector<int> myGenVertices(genEventSize, static_cast<int>(0));
345 if ( myGenEvent.particles_empty() )
return;
351 HepMC::GenVertex*
primaryVertex = *(myGenEvent.vertices_begin());
355 primaryVertex->position().y()/10.,
356 primaryVertex->position().z()/10.,
357 primaryVertex->position().t()/10.);
360 myFilter->setMainVertex(primaryVertexPosition);
365 HepMC::GenEvent::particle_const_iterator piter;
366 HepMC::GenEvent::particle_const_iterator pbegin = myGenEvent.particles_begin();
367 HepMC::GenEvent::particle_const_iterator pend = myGenEvent.particles_end();
369 int initialBarcode = 0;
370 if ( pbegin != pend ) initialBarcode = (*pbegin)->barcode();
372 for ( piter = pbegin; piter != pend; ++piter ) {
390 HepMC::GenVertex* productionVertex = p->production_vertex();
391 if ( productionVertex ) {
392 unsigned productionMother = productionVertex->particles_in_size();
393 if ( productionMother ) {
394 unsigned motherId = (*(productionVertex->particles_in_const_begin()))->pdg_id();
395 if (
abs(motherId) < 1000000 )
396 productionVertexPosition =
398 productionVertex->position().y()/10.,
399 productionVertex->position().z()/10.,
400 productionVertex->position().t()/10.);
403 if ( !
myFilter->accept(productionVertexPosition) )
continue;
405 int abspdgId =
abs(p->pdg_id());
406 HepMC::GenVertex* endVertex = p->end_vertex();
410 bool testStable = p->status()%1000==1;
413 if ( p->status() == 2 && abspdgId < 1000000) {
417 endVertex->position().y()/10.,
418 endVertex->position().z()/10.,
419 endVertex->position().t()/10.);
426 bool testDaugh =
false;
430 endVertex->particles_out_size() ) {
431 HepMC::GenVertex::particles_out_const_iterator firstDaughterIt =
432 endVertex->particles_out_const_begin();
433 HepMC::GenVertex::particles_out_const_iterator lastDaughterIt =
434 endVertex->particles_out_const_end();
435 for ( ; firstDaughterIt != lastDaughterIt ; ++firstDaughterIt ) {
437 if ( daugh->status()%1000==1 ) {
439 if (abspdgId == 11 || abspdgId == 13) {
442 endVertex->position().y()/10.,
443 endVertex->position().z()/10.,
444 endVertex->position().t()/10.);
459 if ( !testStable && !testDaugh && p->production_vertex() ) {
461 productionVertexPosition(p->production_vertex()->position().x()/10.,
462 p->production_vertex()->position().y()/10.,
463 p->production_vertex()->position().z()/10.,
464 p->production_vertex()->position().t()/10.);
465 dist = (primaryVertexPosition-productionVertexPosition).Vect().Mag2();
467 bool testDecay = ( dist > 1
e-8 ) ?
true :
false;
470 if ( testStable || testDaugh || testDecay ) {
477 int motherBarcode = p->production_vertex() &&
478 p->production_vertex()->particles_in_const_begin() !=
479 p->production_vertex()->particles_in_const_end() ?
480 (*(p->production_vertex()->particles_in_const_begin()))->barcode() : 0;
483 motherBarcode && myGenVertices[motherBarcode-initialBarcode] ?
484 myGenVertices[motherBarcode-initialBarcode] : mainVertex;
491 part.setID(p->pdg_id());
495 int theTrack = testStable && p->end_vertex() ?
506 ( testStable && p->end_vertex() && !p->end_vertex()->particles_out_size() )
513 p->end_vertex()->position().y()/10.,
514 p->end_vertex()->position().z()/10.,
515 p->end_vertex()->position().t()/10.);
519 if ( theVertex != -1 ) myGenVertices[p->barcode()-initialBarcode] = theVertex;
530 const HepMC::GenVertex*
ev) {
534 if ( !
myFilter->accept(p) && ig >= -1 )
return -1;
545 if ( !
vertex(iv).noParent() ) {
550 if ( motherId < -1 ) ig = motherId;
555 (*theSimTracks)[trackId] = ev ?
558 ev->position().
t()/10.
562 FSimTrack(p,iv,ig,trackId,
this);
572 if ( !
myFilter->accept(v) )
return -1;
586 (*theSimVertices)[vertexId] =
FSimVertex(v,im,vertexId,
this);
597 std::cout <<
"Id Gen Name eta phi pT E Vtx1 "
599 <<
"Moth Vtx2 eta phi R Z Da1 Da2 Ecal?" << std::endl;
601 for ( HepMC::GenEvent::particle_const_iterator
602 piter = myGenEvent.particles_begin();
603 piter != myGenEvent.particles_end();
608 int partId = p->pdg_id();
611 if (
pdt->particle(ParticleID(partId)) !=0 ) {
612 name = (
pdt->particle(ParticleID(partId)))->
name();
624 if ( !p->production_vertex() )
continue;
626 XYZVector vertex1 (p->production_vertex()->position().x()/10.,
627 p->production_vertex()->position().y()/10.,
628 p->production_vertex()->position().z()/10.);
629 vertexId1 = p->production_vertex()->barcode();
631 std::cout.setf(std::ios::fixed, std::ios::floatfield);
632 std::cout.setf(std::ios::right, std::ios::adjustfield);
634 std::cout << std::setw(4) << p->barcode() <<
" "
637 for(
unsigned int k=0;
k<11-name.length() &&
k<12;
k++)
std::cout <<
" ";
639 double eta = momentum1.eta();
640 if ( eta > +10. ) eta = +10.;
641 if ( eta < -10. ) eta = -10.;
642 std::cout << std::setw(6) << std::setprecision(2) << eta <<
" "
643 << std::setw(6) << std::setprecision(2) << momentum1.phi() <<
" "
644 << std::setw(7) << std::setprecision(2) << momentum1.pt() <<
" "
645 << std::setw(7) << std::setprecision(2) << momentum1.e() <<
" "
646 << std::setw(4) << vertexId1 <<
" "
647 << std::setw(6) << std::setprecision(1) << vertex1.x() <<
" "
648 << std::setw(6) << std::setprecision(1) << vertex1.y() <<
" "
649 << std::setw(6) << std::setprecision(1) << vertex1.z() <<
" ";
652 *(p->production_vertex()->particles_in_const_begin());
655 std::cout << std::setw(4) << mother->barcode() <<
" ";
659 if ( p->end_vertex() ) {
661 p->end_vertex()->position().y()/10.,
662 p->end_vertex()->position().z()/10.,
663 p->end_vertex()->position().t()/10.);
664 int vertexId2 = p->end_vertex()->barcode();
666 std::vector<const HepMC::GenParticle*> children;
667 HepMC::GenVertex::particles_out_const_iterator firstDaughterIt =
668 p->end_vertex()->particles_out_const_begin();
669 HepMC::GenVertex::particles_out_const_iterator lastDaughterIt =
670 p->end_vertex()->particles_out_const_end();
671 for ( ; firstDaughterIt != lastDaughterIt ; ++firstDaughterIt ) {
672 children.push_back(*firstDaughterIt);
675 std::cout << std::setw(4) << vertexId2 <<
" "
676 << std::setw(6) << std::setprecision(2) << vertex2.eta() <<
" "
677 << std::setw(6) << std::setprecision(2) << vertex2.phi() <<
" "
678 << std::setw(5) << std::setprecision(1) << vertex2.pt() <<
" "
679 << std::setw(6) << std::setprecision(1) << vertex2.z() <<
" ";
680 for (
unsigned id=0;
id<children.size(); ++id )
681 std::cout << std::setw(4) << children[id]->barcode() <<
" ";
692 std::cout <<
" Id Gen Name eta phi pT E Vtx1 "
694 <<
"Moth Vtx2 eta phi R Z Daughters Ecal?" << std::endl;
const ParticleDataTable * pdt
double lateVertexPosition
const math::XYZVectorD & trackerSurfacePosition() const
void setCharge(float q)
set the MEASURED charge
int addSimVertex(const XYZTLorentzVector &decayVertex, int im=-1, FSimVertexType::VertexType type=FSimVertexType::ANY)
Add a new vertex to the Event and to the various lists.
float charge() const
charge
int addSimTrack(const RawParticle *p, int iv, int ig=-1, const HepMC::GenVertex *ev=0)
Add a new track to the Event and to the various lists.
bool propagateToPreshowerLayer1(bool first=true)
const HepMC::GenParticle * embdGenpart(int i) const
return MC track with a given id
HepPDT::ParticleDataTable ParticleDataTable
std::vector< FSimTrack > * theSimTracks
double PDGmass() const
get the THEORETICAL mass
std::vector< unsigned > * theChargedTracks
FBaseSimEvent(const edm::ParameterSet &kine)
Default constructor.
const XYZTLorentzVector & momentum() const
Temporary (until move of SimTrack to Mathcore) - No! Actually very useful.
unsigned int theTrackSize
unsigned int theVertexSize
KineParticleFilter * myFilter
The particle filter.
void setEcal(const RawParticle &pp, int success)
Set the ecal variables.
int getSuccess() const
Has propagation been performed and was barrel or endcap reached ?
void addParticles(const HepMC::GenEvent &hev)
Add the particles and their vertices to the list.
void setLayer2(const RawParticle &pp, int success)
Set the preshower layer2 variables.
void clear()
clear the FBaseSimEvent content before the next event
void setMagneticField(double b)
Set the magnetic field.
bool notYetToEndVertex(const XYZTLorentzVector &pos) const
Compare the end vertex position with another position.
FSimVertex & vertex(int id) const
Return vertex with given Id.
unsigned int nVertices() const
Number of vertices.
FSimVertexType & vertexType(int id) const
Return vertex with given Id.
bool propagateToVFcalEntrance(bool first=true)
void setLayer1(const RawParticle &pp, int success)
Set the preshower layer1 variables.
const XYZTLorentzVector & momentum() const
the momentum fourvector
double t() const
vertex time
math::XYZVector XYZVector
~FBaseSimEvent()
usual virtual destructor
const math::XYZTLorentzVector & position() const
Temporary (until CMSSW moves to Mathcore) - No ! Actually very useful.
unsigned int nTracks() const
Number of tracks.
bool propagateToHcalExit(bool first=true)
unsigned int nSimVertices
int genpartIndex() const
index of the corresponding Generator particle in the Event container (-1 if no Genpart) ...
double cos2ThetaV() const
A FSimVertexType hold the information on the vertex origine.
Abs< T >::type abs(const T &t)
const math::XYZTLorentzVectorD & position() const
std::vector< FSimVertexType > FSimVertexTypeCollection
collection of FSimVertexType objects
const FSimVertex & vertex() const
Origin vertex.
unsigned int nGenParticles
std::vector< FSimVertex > * theSimVertices
bool propagateToEcalEntrance(bool first=true)
void fill(const HepMC::GenEvent &hev)
fill the FBaseSimEvent from the current HepMC::GenEvent
int vertIndex() const
index of the vertex in the Event container (-1 if no vertex)
const XYZTLorentzVector & vertex() const
the vertex fourvector
unsigned int theChargedSize
const math::XYZTLorentzVectorD & trackerSurfaceMomentum() const
void addChargedTrack(int id)
Add an id in the vector of charged tracks id's.
void setHcalExit(const RawParticle &pp, int success)
Set the hcal exit variables.
unsigned int nChargedParticleTracks
int type() const
particle type (HEP PDT convension)
bool propagateToHcalEntrance(bool first=true)
void initializePdt(const HepPDT::ParticleDataTable *aPdt)
Initialize the particle data table.
void setEndVertex(int endv)
Set the end vertex.
unsigned int nGenParts() const
Number of generator particles.
const math::XYZTLorentzVectorD & momentum() const
static int position[264][3]
void addDaughter(int i)
Add a RecHit for a track on a layer.
void setHO(const RawParticle &pp, int success)
Set the ho variables.
int chargedTrack(int id) const
return "reconstructed" charged tracks index.
void setVFcal(const RawParticle &pp, int success)
Set the hcal variables.
void setHcal(const RawParticle &pp, int success)
Set the hcal variables.
void printMCTruth(const HepMC::GenEvent &hev)
print the original MCTruth event
std::vector< HepMC::GenParticle * > * theGenParticles
bool propagateToHOLayer(bool first=true)
bool propagateToPreshowerLayer2(bool first=true)
math::XYZTLorentzVector XYZTLorentzVector
void print() const
print the FBaseSimEvent in an intelligible way
FSimTrack & track(int id) const
Return track with given Id.
FSimVertexTypeCollection * theFSimVerticesType