PndGemTrackFinderIdeal Class Reference

Ideal track finding algorithm. More...

#include <PndGemTrackFinderIdeal.h>

Inheritance diagram for PndGemTrackFinderIdeal:

Public Member Functions
	PndGemTrackFinderIdeal ()
virtual	~PndGemTrackFinderIdeal ()
virtual void	Init ()
virtual Int_t	DoFind (TClonesArray *hitArray, TClonesArray *trackArray, TClonesArray *trackCandArray)
void	SetVerbose (const Int_t &verbose)
void	SetPrimary (const Int_t &primary)
virtual void	Finish ()
virtual void	SetParContainers ()
	ClassDef (PndGemTrackFinder, 1)

Private Member Functions
	ClassDef (PndGemTrackFinderIdeal, 1)

Private Attributes
TClonesArray *	fMCTrackArray
TClonesArray *	fMCPointArray
Int_t	fVerbose
Int_t	fPrimary
Int_t	fNofEvents
	event counter More...

Detailed Description

Ideal track finding algorithm.

Author

R. Karabowicz r.kar.nosp@m.abow.nosp@m.icz@g.nosp@m.si.d.nosp@m.e

Date

19.03.2009 Track finding procedure operates on mc data. Reads MC truth (MCTracks and MCPoints), creates one PndTrack for each MCTrack and attaches the hits according to the MCTrack of the corresponding MCPoint

Definition at line 28 of file PndGemTrackFinderIdeal.h.

Constructor & Destructor Documentation

PndGemTrackFinderIdeal::PndGemTrackFinderIdeal

(

)

Default constructor

Definition at line 38 of file PndGemTrackFinderIdeal.cxx.

References fMCPointArray, fMCTrackArray, fNofEvents, fPrimary, and fVerbose.

                                               {
  fMCTrackArray = NULL;
  fMCPointArray = NULL;
  fNofEvents    = 0;
  fVerbose = 0;
  fPrimary = 0;
}

PndGemTrackFinderIdeal::~PndGemTrackFinderIdeal ( )

virtual

Destructor

Definition at line 47 of file PndGemTrackFinderIdeal.cxx.

{ }

Member Function Documentation

PndGemTrackFinder::ClassDef ( PndGemTrackFinder  , 1    )

inherited

PndGemTrackFinderIdeal::ClassDef ( PndGemTrackFinderIdeal  , 1    )

private

Int_t PndGemTrackFinderIdeal::DoFind ( TClonesArray *  hitArray, TClonesArray *  trackArray, TClonesArray *  trackCandArray  )

virtual

DoFind method

Parameters

hitArray	Array of Gem hits
trackArray	Array of PndGemTrack

Returns

Number of created tracks

Implements PndGemTrackFinder.

Definition at line 102 of file PndGemTrackFinderIdeal.cxx.

References PndTrackCand::AddHit(), fabs(), fMCPointArray, fMCTrackArray, fNofEvents, fPrimary, fVerbose, PndTrackCandHit::GetHitId(), PndMCTrack::GetMotherID(), PndMCTrack::GetNPoints(), PndMCTrack::GetPdgCode(), PndGemHit::GetPosition(), PndTrackCand::GetSortedHit(), PndMCTrack::GetStartVertex(), kGEM, mom, pos, PndTrack::SetRefIndex(), PndTrackCand::Sort(), CAMath::Sqrt(), and Z.

                                                                   {
  
  // Count events
  fNofEvents++;
  if ( fVerbose ) {
    cout << endl << endl<< endl << endl;
    cout << "=======================================================" << endl;
    cout << "-I-        Event No: " <<  fNofEvents << endl;
    cout << "=======================================================" << endl;
    
    
    cout <<"-I- "<< GetName() <<"::DoFind "<< endl;
    cout << "-------------------------------------------------------" << endl;
    cout << "     ### Start DoFind" << endl;
    cout << "-------------------------------------------------------" << endl;
  }

  // Check pointers
  if( !fMCTrackArray ) {
    cout << "-E- PndGemTrackFinderIdeal::DoFind: "
         << "MCTrack array missing! " << endl;
    return -1;
  }
  
  if( !fMCPointArray ) {
    cout << "-E- "<< GetName() <<"::DoFind: "
         << "MCPoint array missing! " << endl;
    return -1;
  }
  
  if( !hitArray ) {
    cout << "-E- "<< GetName() <<"::DoFind: "
         << "Hit arrays missing! "<< endl;
    return -1;
  }
  
  // Initialise control counters
  Int_t nNoMCTrack   = 0;
  Int_t nNoTrack     = 0;
  Int_t nNoGemPoint  = 0;
  Int_t nNoGemHit    = 0;
  
  // Create pointers to GemHit and GemPoint
  PndGemHit*   gemHit   = NULL;
  FairMCPoint*  mcPoint  = NULL;
  PndMCTrack*  mcTrack  = NULL;
  PndTrack* gemTrack = NULL;
  PndTrackCand* gemTrackCand = NULL;
  PndTrackCandHit tcHit;

  // Declare variables outside the loop
  Int_t ptIndex = 0;       // MC point index
  Int_t mcTrackIndex = 0;  // MC track index
  Int_t trackIndex = 0;    // Gem track index
  Int_t relDetID = -1;//3000;   // 

  // Create STL map from MCTrack index to number of hits
  std::map<Int_t, Int_t> hitMap;
  std::map<Int_t, Double_t> trackPMap;
  std::map<Int_t, Int_t>::iterator itHitMap;
  
  if ( fVerbose ) {
    // Size of fMCTrackArray
    cout <<"# MC Tracks: "<< fMCTrackArray->GetEntriesFast() << endl;
    // Size of fMCTrackArray
    cout <<"# MC Points: "<< fMCPointArray->GetEntriesFast() << endl;
  }

  // Number of Gem hits
  Int_t nGemHits = hitArray->GetEntriesFast();
  if(fVerbose > 2) cout <<"# GemHits: "<< nGemHits << endl;
  
  for(Int_t iHit = 0; iHit < nGemHits; iHit++){
    // Get the pointer to Gem hit
    gemHit = (PndGemHit*) hitArray->At(iHit);
    if(gemHit->GetDetectorID() > relDetID) {
      if(NULL == gemHit ) continue;
      
      // Get point index
      ptIndex = gemHit->GetRefIndex();

      if ( ptIndex == -1 ) continue;

      // Get pointer to MC point
      mcPoint = (FairMCPoint*) fMCPointArray->At(ptIndex);
      if(NULL == mcPoint) continue;
      
      // Get MC track index
      mcTrackIndex = mcPoint->GetTrackID();

      trackPMap[mcTrackIndex] = TMath::Sqrt(mcPoint->GetPx()*mcPoint->GetPx()+
                                            mcPoint->GetPy()*mcPoint->GetPy()+
                                            mcPoint->GetPz()*mcPoint->GetPz());
      
      // Increment the number of hits
      hitMap[mcTrackIndex] += 1;
    }
  }
  

  // Create STL map from MCTrack index to GemTrack index
  map<Int_t, Int_t> trackMap;
  
  // Loop over reconstructable MCTracks and create corresponding GemTrack
  Int_t nMCacc = 0;  // Accepted MC tracks
  Int_t nTracks = 0; // Reconstructable MC tracks
  Int_t nMCTracks = fMCTrackArray->GetEntriesFast();
  
  if(fVerbose > 2) {
    cout <<"# of MC Tracks (nMCTracks): " << nMCTracks << endl;
  }
  
  for(Int_t iMCTrack = 0; iMCTrack < nMCTracks; iMCTrack++) {
    mcTrack = (PndMCTrack*) fMCTrackArray->At(iMCTrack);
    if( !mcTrack ) continue;
    
    
    // Either Primary or All MC tracks are considered
    if(fPrimary) {
      if(mcTrack->GetMotherID() != -1 && fabs((mcTrack->GetStartVertex()).Z())>1.) continue;
      
      if(fVerbose > 2) {
        cout <<"MC track #: "<< iMCTrack
             << "\t MotherID: "<< mcTrack->GetMotherID()
             << "\t StartVertex Z: "<< mcTrack->GetStartVertex().Z() << endl;
      }
    }
    //----------------------------------------------------
    nMCacc++;
    
    if(fVerbose > 2) {
      cout << iMCTrack << ": #GemPoints in MCTrack: "<< mcTrack->GetNPoints(kGEM) << " -> " << hitMap[iMCTrack] << endl;
    }
    if ( !hitMap[iMCTrack] ) continue;

    gemTrackCand = new((*trackCandArray)[nTracks]) PndTrackCand();

    // Loop over hits. Get corresponding MCPoint and MCTrack index
    for(Int_t iHit = 0; iHit < nGemHits; iHit++) {
      gemHit = (PndGemHit*) hitArray->At(iHit);
      if( !gemHit ) {
        nNoGemHit++;
        continue;
      }
    
      ptIndex = gemHit->GetRefIndex();
      if(ptIndex < 0) continue;           // fake or background hit
      mcPoint = (FairMCPoint*) fMCPointArray->At(ptIndex);
    
      if( !mcPoint ) {
        nNoGemPoint++;
        continue;
      }
    
      mcTrackIndex = mcPoint->GetTrackID();
    
      if(mcTrackIndex < 0 || mcTrackIndex > nMCTracks) {
        cout << "-E- "<< GetName() <<"::DoFind: "
             << "MCTrack index out of range. " << mcTrackIndex << " "
             << nMCTracks << endl;
        nNoMCTrack++;
        continue;
      }
      if ( mcTrackIndex != iMCTrack ) continue;

      gemTrackCand->AddHit(FairRootManager::Instance()->GetBranchId("GEMHit"),
                           iHit,gemHit->GetPosition().Mag());

      if(fVerbose > 3) {
        cout << "GEM hit " << iHit << " from GEM point "
             << ptIndex << " (" << mcTrackIndex << ") "
             << "added to GEM track " << trackIndex << endl;
      }
    }

    gemTrackCand->Sort();
  
    //    cout << "gemTrack " << nTracks << " has " << gemTrackCand->GetNHits() << " hits" << endl;
    PndTrackCandHit tch = gemTrackCand->GetSortedHit(0);

    gemHit = (PndGemHit*) hitArray->At(tch.GetHitId());
    
    TVector3 pos;
    TVector3 mom;
    
    ptIndex = gemHit->GetRefIndex();
    mcPoint = (FairMCPoint*) fMCPointArray->At(ptIndex);

    gemHit->Position(pos);
    mcPoint->Momentum(mom);

    mcTrack = (PndMCTrack*) fMCTrackArray->At(mcPoint->GetTrackID());

    int pdg = mcTrack->GetPdgCode();
    double charge = 0.;
    if(pdg<100000000) charge = TDatabasePDG::Instance()->GetParticle(pdg)->Charge()/3.;

//     gemTrackCand->setMcTrackId(iMCTrack);

    FairTrackParP           firstPar(pos,mom,
                                     TVector3(0.5, 0.5, 0.5),
                                     0.1*mom,
                                     (Int_t)charge,
                                     pos,
                                     TVector3(1.,0.,0.),
                                     TVector3(0.,1.,0.));                                        
    FairTrackParP           lastPar (pos,mom,
                                     TVector3(0.5, 0.5, 0.5),
                                     0.1*mom,
                                     (Int_t)charge,
                                     pos,
                                     TVector3(1.,0.,0.),
                                     TVector3(0.,1.,0.));                                        
    
    gemTrack = new((*trackArray)[nTracks]) PndTrack(firstPar, lastPar, *gemTrackCand);
    
    gemTrack->SetRefIndex(iMCTrack);

    trackMap[iMCTrack] = nTracks;
    
    nTracks++;
  }
  
  if(fVerbose) {
    cout << endl;
    cout << "-------------------------------------------------------" << endl;
    cout << "-I-      "<< GetName() <<": Event summary      -I-" << endl;
    cout << "-------------------------------------------------------" << endl;
    cout << "Total Gem hits:  " << nGemHits << endl;
    cout << "MC tracks total: " << nMCTracks << ", accepted: "
         << nMCacc << ", reconstructable: " << nTracks << endl;
    if(nNoGemHit)   cout << "GemHits not found   : " << nNoGemHit   << endl;
    if(nNoGemPoint) cout << "GemPoints not found : " << nNoGemPoint << endl;
    if(nNoMCTrack)  cout << "MCTracks not found  : " << nNoMCTrack  << endl;
    if(nNoTrack)    cout << "GemTracks not found : " << nNoTrack    << endl;
    cout << "------------------------------------------------------" << endl;
    cout << endl;
  } else {
//     cout << "All: "        << nMCTracks
//       << ", Accepted: "      << nMCacc
//       << ", Reconstructed: " << nTracks << endl;
  }
  
  return nTracks;
}

virtual void PndGemTrackFinder::Finish ( )

inlinevirtualinherited

Reimplemented in PndGemTrackFinderOnHitsTB.

Definition at line 42 of file PndGemTrackFinder.h.

Referenced by PndGemFindTracks::Finish().

{};

void PndGemTrackFinderIdeal::Init ( )

virtual

Initialisation

Reimplemented from PndGemTrackFinder.

Definition at line 50 of file PndGemTrackFinderIdeal.cxx.

References fMCPointArray, fMCTrackArray, fVerbose, and rtdb.

                                  {
  
  // Get and check FairRootManager
  FairRootManager* ioman = FairRootManager::Instance();
  if( !ioman ) {
    cout << "-E- "<< GetName() <<"::Init: "
         << "RootManager not instantised!" << endl;
    return;
  }
  
  // Get the pointer to the singleton FairRunAna object
  FairRunAna* ana = FairRunAna::Instance();
  if(NULL == ana) {
    cout << "-E- "<< GetName() <<"::Init :"
         <<" no FairRunAna object!" << endl;
    return;
  }
  // Get the pointer to run-time data base
  FairRuntimeDb* rtdb = ana->GetRuntimeDb();
  if(NULL == rtdb) {
    cout << "-E- "<< GetName() <<"::Init :"
         <<" no runtime database!" << endl;
    return;
  }
  
  // Get MCTrack array
  fMCTrackArray  = (TClonesArray*) ioman->GetObject("MCTrack");
  if( !fMCTrackArray ) {
    cout << "-E- "<< GetName() <<"::Init: No MCTrack array!"
         << endl;
    return;
  }
  
  // Get PndGemPoint (MCPoint) array
  fMCPointArray  = (TClonesArray*) ioman->GetObject("GEMPoint");
  if( !fMCPointArray ) {
    cout << "-E- "<< GetName() <<"::Init: No MCPoint array!"
         << endl;
    return;
  }
  if ( fVerbose )
    cout <<"fMCPointArray #: "<< fMCPointArray->GetEntriesFast() << endl;
  
  // Geometry loading
  //TFile *infile = ioman->GetInFile();
  //TGeoManager *geoMan = (TGeoManager*) infile->Get("FAIRGeom");
  //fGemStructure = PndGemStructure::Instance(geoMan);
  
  std::cout << "-I- "<< GetName() <<": Intialization successfull" << std::endl;
}

virtual void PndGemTrackFinder::SetParContainers ( )

inlinevirtualinherited

Virtual method SetParContainers. If needed, to be implemented in the concrete class.

Reimplemented in PndGemTrackFinderOnHitsTB, and PndGemTrackFinderOnHits.

Definition at line 47 of file PndGemTrackFinder.h.

{ };

void PndGemTrackFinderIdeal::SetPrimary ( const Int_t &  primary )

inline

Definition at line 53 of file PndGemTrackFinderIdeal.h.

References fPrimary.

Referenced by gem_idealTracks(), locT_all(), and locT_theta().

{ fPrimary = primary; };

void PndGemTrackFinderIdeal::SetVerbose ( const Int_t &  verbose )

inline

Public modifiers

Definition at line 52 of file PndGemTrackFinderIdeal.h.

References fVerbose, and verbose.

Referenced by gem_idealTracks(), locT_all(), and locT_theta().

{ fVerbose = verbose; };

Member Data Documentation

TClonesArray* PndGemTrackFinderIdeal::fMCPointArray

private

Definition at line 60 of file PndGemTrackFinderIdeal.h.

Referenced by DoFind(), Init(), and PndGemTrackFinderIdeal().

TClonesArray* PndGemTrackFinderIdeal::fMCTrackArray

private

Arrays of MC information

Definition at line 53 of file PndGemTrackFinderIdeal.h.

Referenced by DoFind(), Init(), and PndGemTrackFinderIdeal().

Int_t PndGemTrackFinderIdeal::fNofEvents

private

event counter

Event counter

Definition at line 80 of file PndGemTrackFinderIdeal.h.

Referenced by DoFind(), and PndGemTrackFinderIdeal().

Int_t PndGemTrackFinderIdeal::fPrimary

private

Primaries level 0 - All MC tracks are considered 1 - Only primary MC tracks

Definition at line 77 of file PndGemTrackFinderIdeal.h.

Referenced by DoFind(), PndGemTrackFinderIdeal(), and SetPrimary().

Int_t PndGemTrackFinderIdeal::fVerbose

private

fGemStructure Verbosity level. 0 - quit 1 - event level 2 - track level 3 - debug (maximal output)

Definition at line 71 of file PndGemTrackFinderIdeal.h.

Referenced by DoFind(), Init(), PndGemTrackFinderIdeal(), and SetVerbose().

---

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

• PndGemTrackFinderIdeal.h
• PndGemTrackFinderIdeal.cxx