PndMvdSttGemRiemannTrackFinderTask Class Reference

#include <PndMvdSttGemRiemannTrackFinderTask.h>

Inheritance diagram for PndMvdSttGemRiemannTrackFinderTask:

Public Member Functions
	PndMvdSttGemRiemannTrackFinderTask ()
virtual	~PndMvdSttGemRiemannTrackFinderTask ()
	PndMvdSttGemRiemannTrackFinderTask (const PndMvdSttGemRiemannTrackFinderTask &)=delete
PndMvdSttGemRiemannTrackFinderTask &	operator= (const PndMvdSttGemRiemannTrackFinderTask &)=delete
virtual void	SetParContainers ()
virtual InitStatus	Init ()
virtual InitStatus	ReInit ()
virtual void	Exec (Option_t *opt)
virtual void	FinishEvent ()
void	AddHitBranch (TString branchName)
void	SetMaxSZChi2 (double val)
void	SetMaxSZDist (double val)
void	SetMinPointDist (double val)
void	SetMaxDist (double val)
void	SetCutDistH (TH2F *hist)
void	SetCutChi2H (TH2F *hist)
void	SetB (Double_t B)
void	SetVerbose (Int_t verbose)
void	SetPersistence (Bool_t val)
void	FillHitArray ()
void	InitHitArray (TString branchName)

Private Member Functions
void	Register ()
void	Reset ()
void	ProduceHits ()
	ClassDef (PndMvdSttGemRiemannTrackFinderTask, 1)

Private Attributes
std::vector< TString >	fHitBranch
TString	fTrackBranch
int	fEventNr
double	fMaxSZChi2
double	fMaxSZDist
double	fMinPointDist
double	fMaxDist
double	fB
std::vector< TClonesArray * >	fHitArray
TClonesArray *	fTrackCandArray
TClonesArray *	fRiemannTrackArray
TClonesArray *	fTrackArray
TClonesArray *	fCorrectedSttHitArray
TH2F *	fCutDistH
TH2F *	fCutChi2H
Bool_t	fInitDone
Bool_t	fPersistence
BinaryFunctor *	fStopFunctor
BinaryFunctor *	fTimeGapFunctor
PndGeoSttPar *	fSttParameters
TClonesArray *	fTubeArray

Detailed Description

Definition at line 14 of file PndMvdSttGemRiemannTrackFinderTask.h.

Constructor & Destructor Documentation

PndMvdSttGemRiemannTrackFinderTask::PndMvdSttGemRiemannTrackFinderTask

(

)

Definition at line 25 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References PndGeoHandling::Instance().

                                                                       :
        FairTask("MVD Riemann Track Finder"), 
    fHitBranch(),
    fTrackBranch(""),
    fEventNr(0),
    fMaxSZChi2(1),
    fMaxSZDist(10),
    fMinPointDist(1),
    fMaxDist(1),
    fB(2.), 
    fHitArray(),
          fTrackCandArray(NULL),
          fRiemannTrackArray(NULL),
          fTrackArray(NULL),
          fCorrectedSttHitArray(NULL),
          fCutDistH(NULL),
          fCutChi2H(NULL),
          fInitDone(kFALSE),
          fPersistence(kTRUE),
          fStopFunctor(NULL),
          fTimeGapFunctor(NULL),
          fSttParameters(NULL),
          fTubeArray(NULL)
{
  PndGeoHandling::Instance();
}

PndMvdSttGemRiemannTrackFinderTask::~PndMvdSttGemRiemannTrackFinderTask ( )

virtual

Definition at line 52 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

{
}

PndMvdSttGemRiemannTrackFinderTask::PndMvdSttGemRiemannTrackFinderTask ( const PndMvdSttGemRiemannTrackFinderTask &  )

delete

Member Function Documentation

void PndMvdSttGemRiemannTrackFinderTask::AddHitBranch

(

TString

branchName

)

Definition at line 126 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References fHitBranch, and fInitDone.

{
        if (fInitDone == kFALSE)
                fHitBranch.push_back(branchName);
        else
                std::cout << "-W- AddHitBranch has to be called before the Init() of the task!" << std::endl;
}

PndMvdSttGemRiemannTrackFinderTask::ClassDef ( PndMvdSttGemRiemannTrackFinderTask  , 1    )

private

void PndMvdSttGemRiemannTrackFinderTask::Exec ( Option_t *  opt )

virtual

Virtual method Exec

Definition at line 143 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References PndMvdSttGemRiemannTrackFinder::AddHits(), PndTrackCand::CalcTimeStamp(), PndRiemannTrack::correctSttHit(), fB, fCutChi2H, fCutDistH, fEventNr, fHitArray, fHitBranch, FillHitArray(), PndMvdSttGemRiemannTrackFinder::FindTracks(), fMaxDist, fMaxSZChi2, fMaxSZDist, fMinPointDist, fTrackArray, fTrackCandArray, fTubeArray, fVerbose, PndTrackCand::GetNHits(), PndRiemannTrackFinder::GetPndTrack(), PndRiemannTrackFinder::GetTrack(), PndRiemannTrackFinder::GetTrackCand(), i, PndRiemannTrackFinder::NTracks(), PndTrack::Print(), PndMvdSttGemRiemannTrackFinder::SetCutChi2H(), PndMvdSttGemRiemannTrackFinder::SetCutDistH(), PndRiemannTrackFinder::SetMaxPlaneDistance(), PndRiemannTrackFinder::SetMaxSZChi2(), PndRiemannTrackFinder::SetMaxSZDist(), PndRiemannTrackFinder::SetMinNumberOfHits(), PndRiemannTrackFinder::SetMinPointDist(), PndMvdSttGemRiemannTrackFinder::SetSttTubeArray(), PndTrack::SetTrackCand(), PndTrack::SetTrackCandRef(), PndRiemannTrackFinder::SetVerbose(), trackFinder, PndRiemannHit::x(), and PndRiemannHit::z().

{

//      SetVerbose(0);
  // Reset output array
  if ( ! fTrackCandArray )
    Fatal("Exec", "No trackCandArray");

  fTrackCandArray->Delete();
  fTrackArray->Delete();
 // fRiemannTrackArray->Delete();

  PndMvdSttGemRiemannTrackFinder trackFinder;
  trackFinder.SetVerbose(fVerbose);
  trackFinder.SetSttTubeArray(fTubeArray);

  FillHitArray();

  FairRootManager *ioman = FairRootManager::Instance();

  if (fVerbose > 0) {
          std::cout << std::endl;
          std::cout << "------------- event " << fEventNr++ << "----------------" << std::endl;
  }

  for (int i = 0; i < (int)fHitBranch.size(); i++){
          trackFinder.AddHits(fHitArray[i], ioman->GetBranchId(fHitBranch[i]));
          if (fVerbose > 1) std::cout << "TrackFinder.AddHits: " << ioman->GetBranchId(fHitBranch[i]) << std::endl;
  }
  trackFinder.SetMaxSZChi2(fMaxSZChi2);
  trackFinder.SetMinPointDist(fMinPointDist);
  trackFinder.SetMaxPlaneDistance(fMaxDist);
  trackFinder.SetMaxSZDist(fMaxSZDist);
  trackFinder.SetVerbose(fVerbose);
  trackFinder.SetMinNumberOfHits(4);
  //trackFinder.SetGeoH(fGeoH);

  trackFinder.SetCutDistH(fCutDistH);
  trackFinder.SetCutChi2H(fCutChi2H);

  trackFinder.FindTracks();


  if (fVerbose > 0) std::cout << " -I- PndMvdSttGemRiemannTrackFinderTask::Exec : Found Tracks: " << trackFinder.NTracks() << " in event no. " << fEventNr << std::endl;
  if (fVerbose > 0) std::cout << " -I- PndMvdSttGemRiemannTrackFinderTask::Exec : ----------------" << std::endl;


  for (int i = 0; i < trackFinder.NTracks(); i++){
          PndTrackCand* myCand = new ((*fTrackCandArray)[i])PndTrackCand(trackFinder.GetTrackCand(i));
//        if (fVerbose > 1)
//        {
//                std::cout << "Track " << i << std::endl;
//                std::cout << "Links: ";
//                ((FairMultiLinkedData*) myCand)->Print();
//                std::cout << std::endl;
//        }

          myCand->CalcTimeStamp();
//        if (fVerbose > 1)
//                trackFinder.GetTrack(i).Print();
          //PndRiemannTrack myTrack = trackFinder.GetTrack(i);

          PndTrack* myTrack = new ((*fTrackArray)[i])PndTrack(trackFinder.GetPndTrack(i, fB));
          if (myCand->GetTimeStamp() == 0){
                  myTrack->SetTimeStamp(0.0001 * (i+1));
                  myCand->SetTimeStamp(0.0001 * (i+1));

          } else {
                  myTrack->SetTimeStamp(myCand->GetTimeStamp());
                  myTrack->SetTimeStampError(myCand->GetTimeStampError());
          }
          myTrack->SetLink(FairLink("MVDRiemannTrackCand", i));
          myTrack->SetTrackCandRef(myCand);
          myTrack->SetTrackCand(*myCand);

          if (fVerbose > 1) {
                  std::cout << i << ": ";
                  myTrack->Print();
          }

          for (size_t j = 0; j < myCand->GetNHits(); j++){
                  if (myCand->GetLink(j).GetType() == FairRootManager::Instance()->GetBranchId("STTHit")){
                          PndSttHit* mySttHit = (PndSttHit*)FairRootManager::Instance()->GetCloneOfLinkData(myCand->GetLink(j));
                          PndRiemannTrack myRiemannTrack = trackFinder.GetTrack(i);
                          PndRiemannHit correctedSttHit = myRiemannTrack.correctSttHit(mySttHit);
                          FairHit correctedFairHit;
                          correctedFairHit.SetXYZ(correctedSttHit.x().X(), correctedSttHit.x().Y(), correctedSttHit.z());
//                        std::cout << "PndMvdSttGemRiemannTrackFinderTask::Exec Correction: " << mySttHit->GetX() << "/" << mySttHit->GetY() << "/" << mySttHit->GetZ()
//                                          << " CorrectedHit: " << correctedFairHit.GetX() << "/"<< correctedFairHit.GetY() << "/" << correctedFairHit.GetZ()<< std::endl;

//                        Fair1Hit* copyHit = new((*fCorrectedSttHitArray)[fCorrectedSttHitArray->GetEntriesFast()])FairHit(correctedFairHit);

                  }
          }
//        PndRiemannTrack* newTrack = new ((*fRiemannTrackArray)[i])PndRiemannTrack();
//        TVectorD origin = myTrack.orig();
//        newTrack->init(origin[0], origin[1], myTrack.r(), myTrack.dip(), 0);
//        for (int j = 0; j < myTrack.getNumHits(); j++){
//                newTrack->addHit(*myTrack.getHit(j));
//        }
  }
  fTrackCandArray->Sort();
  fTrackArray->Sort();

  for (int i = 0; i < fTrackCandArray->GetEntriesFast(); i++){
          PndTrackCand* myCand =(PndTrackCand*)fTrackCandArray->At(i);
          if (fVerbose > 1)
          {
                  std::cout << "Track " << i << std::endl;
                  std::cout << "Links: ";
                  ((FairMultiLinkedData*) myCand)->Print();
                  std::cout << std::endl;
          }
  }
}

void PndMvdSttGemRiemannTrackFinderTask::FillHitArray

(

)

Definition at line 276 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References At, Double_t, fHitArray, fStopFunctor, fTimeGapFunctor, and GetEntriesFast().

Referenced by Exec().

{
        Double_t eventTime = -1;
        if (FairRunAna::Instance()->IsTimeStamp()){
                fHitArray[0]->Delete();
                fHitArray[1]->Delete();

                fHitArray[0] = FairRootManager::Instance()->GetData("MVDHitsPixel", fTimeGapFunctor, 10); //FairRootManager::Instance()->GetEventTime() +
                std::cout << "PixelHits: " << fHitArray[0]->GetEntriesFast() << std::endl;
                if (fHitArray[0]->GetEntriesFast() > 1){
                        FairTimeStamp* data = (FairTimeStamp*)fHitArray[0]->At(1);
                        eventTime = data->GetTimeStamp();
                        std::cout << "EventTime: " << eventTime << std::endl;
                        fHitArray[1] = FairRootManager::Instance()->GetData("MVDHitsStrip", fStopFunctor, eventTime - 10, fStopFunctor, eventTime + 10);
                        std::cout << "StripHits: " << fHitArray[1]->GetEntriesFast() << std::endl;
                }
        }
}

void PndMvdSttGemRiemannTrackFinderTask::FinishEvent ( )

virtual

Definition at line 259 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References fCorrectedSttHitArray, fTrackArray, and fTrackCandArray.

{
        fTrackCandArray->Delete();
        fTrackArray->Delete();
        fCorrectedSttHitArray->Delete();
//      fRiemannTrackArray->Delete();
}

InitStatus PndMvdSttGemRiemannTrackFinderTask::Init ( )

virtual

Definition at line 69 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References fCorrectedSttHitArray, fHitBranch, PndSttMapCreator::FillTubeArray(), fInitDone, fPersistence, fStopFunctor, fSttParameters, fTimeGapFunctor, fTrackArray, fTrackCandArray, fTubeArray, i, and InitHitArray().

{
  FairRootManager* ioman = FairRootManager::Instance();

  if ( ! ioman )
    {
      std::cout << "-E- PndMvdSttGemRiemannTrackFinderTask::Init: "
     << "RootManager not instantiated!" << std::endl;
      return kFATAL;
    }

  // Get input array
//  if (FairRunAna::Instance()->IsTimeStamp()){
//        if (fTimeBasedHitBranch.size() == 0){
//                std::cout << "-W- PndMvdSttGemRiemannTrackFinderTask::Init: " << "No Branch Names given with AddHitBranch(TString branchName)! Standard BranchNames taken!" << std::endl;
//                fTimeBasedHitBranch.push_back(FairTSBufferParameters("MVDHitsPixel", new TimeGap(), 10));
//        }
//  }
  if (fHitBranch.size() == 0){
          std::cout << "-W- PndMvdSttGemRiemannTrackFinderTask::Init: " << "No Branch Names given with AddHitBranch(TString branchName)! Standard BranchNames taken!" << std::endl;
          fHitBranch.push_back("MVDHitsPixel");
          fHitBranch.push_back("MVDHitsStrip");
          fHitBranch.push_back("STTHit");
          fHitBranch.push_back("GEMHit");
          fHitBranch.push_back("STTCombinedSkewedHits");
  }

  for (int i = 0; i < (int)fHitBranch.size(); i++){
          InitHitArray(fHitBranch[i]);
  }



  fTrackCandArray = new TClonesArray("PndTrackCand");
  ioman->Register("MVDRiemannTrackCand", "MVD", fTrackCandArray, fPersistence);

  fTrackArray = new TClonesArray("PndTrack");
  ioman->Register("MVDTrack", "MVD", fTrackArray, fPersistence);

  fCorrectedSttHitArray = new TClonesArray("FairHit");
  ioman->Register("CorrectedSttHit", "Stt", fCorrectedSttHitArray, kTRUE);

//  fRiemannTrackArray = new TClonesArray("PndRiemannTrack");
//  ioman->Register("MVDRiemannTrack", "MVD", fRiemannTrackArray, kTRUE);


  fStopFunctor= new StopTime();
  fTimeGapFunctor = new TimeGap();

  PndSttMapCreator *mapper = new PndSttMapCreator(fSttParameters);
  fTubeArray = mapper->FillTubeArray();

  std::cout << "-I- PndMvdSttGemRiemannTrackFinderTask: Initialisation successfull" << std::endl;
  fInitDone = kTRUE;
  return kSUCCESS;
}

void PndMvdSttGemRiemannTrackFinderTask::InitHitArray

(

TString

branchName

)

Definition at line 267 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References fHitArray.

Referenced by Init().

{
          TClonesArray* tempArray = (TClonesArray*) FairRootManager::Instance()->GetObject(branchName);
          if (tempArray == 0){
                  std::cout << "-W- PndMvdSttGemRiemannTrackFinderTask::Init: " << "No hitArray for BranchName " << branchName.Data() << std::endl;
          }
          fHitArray.push_back(tempArray);
}

PndMvdSttGemRiemannTrackFinderTask& PndMvdSttGemRiemannTrackFinderTask::operator= ( const PndMvdSttGemRiemannTrackFinderTask &  )

delete

void PndMvdSttGemRiemannTrackFinderTask::ProduceHits ( )

private

void PndMvdSttGemRiemannTrackFinderTask::Register ( )

private

InitStatus PndMvdSttGemRiemannTrackFinderTask::ReInit ( )

virtual

Definition at line 62 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

{
  InitStatus stat=kSUCCESS;
  return stat;
}

void PndMvdSttGemRiemannTrackFinderTask::Reset ( )

private

void PndMvdSttGemRiemannTrackFinderTask::SetB ( Double_t  B )

inline

Definition at line 41 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fB.

{fB = B;}

void PndMvdSttGemRiemannTrackFinderTask::SetCutChi2H ( TH2F *  hist )

inline

Definition at line 39 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fCutChi2H, and hist.

{fCutChi2H=hist;}

void PndMvdSttGemRiemannTrackFinderTask::SetCutDistH ( TH2F *  hist )

inline

Definition at line 38 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fCutDistH, and hist.

{fCutDistH=hist;}

void PndMvdSttGemRiemannTrackFinderTask::SetMaxDist ( double  val )

inline

Definition at line 36 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fMaxDist, and val.

{fMaxDist = val;}

void PndMvdSttGemRiemannTrackFinderTask::SetMaxSZChi2 ( double  val )

inline

Definition at line 33 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fMaxSZChi2, and val.

{fMaxSZChi2 = val;}

void PndMvdSttGemRiemannTrackFinderTask::SetMaxSZDist ( double  val )

inline

Definition at line 34 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fMaxSZDist, and val.

{fMaxSZDist = val;}

void PndMvdSttGemRiemannTrackFinderTask::SetMinPointDist ( double  val )

inline

Definition at line 35 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fMinPointDist, and val.

{fMinPointDist = val;}

void PndMvdSttGemRiemannTrackFinderTask::SetParContainers ( )

virtual

Virtual method Init

Definition at line 56 of file PndMvdSttGemRiemannTrackFinderTask.cxx.

References fSttParameters, and rtdb.

{
  FairRuntimeDb* rtdb = FairRunAna::Instance()->GetRuntimeDb();
  fSttParameters = (PndGeoSttPar*) rtdb->getContainer("PndGeoSttPar");
}

void PndMvdSttGemRiemannTrackFinderTask::SetPersistence ( Bool_t  val )

inline

Definition at line 45 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fPersistence, and val.

{ fPersistence = val;};

void PndMvdSttGemRiemannTrackFinderTask::SetVerbose ( Int_t  verbose )

inline

Definition at line 44 of file PndMvdSttGemRiemannTrackFinderTask.h.

References fVerbose, and verbose.

{ fVerbose = verbose;};

Member Data Documentation

double PndMvdSttGemRiemannTrackFinderTask::fB

private

Definition at line 62 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetB().

TClonesArray* PndMvdSttGemRiemannTrackFinderTask::fCorrectedSttHitArray

private

Definition at line 68 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by FinishEvent(), and Init().

TH2F* PndMvdSttGemRiemannTrackFinderTask::fCutChi2H

private

Definition at line 71 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetCutChi2H().

TH2F* PndMvdSttGemRiemannTrackFinderTask::fCutDistH

private

Definition at line 70 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetCutDistH().

int PndMvdSttGemRiemannTrackFinderTask::fEventNr

private

Definition at line 55 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec().

std::vector<TClonesArray*> PndMvdSttGemRiemannTrackFinderTask::fHitArray

private

Definition at line 64 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), FillHitArray(), and InitHitArray().

std::vector<TString> PndMvdSttGemRiemannTrackFinderTask::fHitBranch

private

Definition at line 52 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by AddHitBranch(), Exec(), and Init().

Bool_t PndMvdSttGemRiemannTrackFinderTask::fInitDone

private

Definition at line 73 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by AddHitBranch(), and Init().

double PndMvdSttGemRiemannTrackFinderTask::fMaxDist

private

Definition at line 60 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetMaxDist().

double PndMvdSttGemRiemannTrackFinderTask::fMaxSZChi2

private

Definition at line 57 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetMaxSZChi2().

double PndMvdSttGemRiemannTrackFinderTask::fMaxSZDist

private

Definition at line 58 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetMaxSZDist().

double PndMvdSttGemRiemannTrackFinderTask::fMinPointDist

private

Definition at line 59 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and SetMinPointDist().

Bool_t PndMvdSttGemRiemannTrackFinderTask::fPersistence

private

Definition at line 74 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Init(), and SetPersistence().

TClonesArray* PndMvdSttGemRiemannTrackFinderTask::fRiemannTrackArray

private

Definition at line 66 of file PndMvdSttGemRiemannTrackFinderTask.h.

BinaryFunctor* PndMvdSttGemRiemannTrackFinderTask::fStopFunctor

private

Definition at line 76 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by FillHitArray(), and Init().

PndGeoSttPar* PndMvdSttGemRiemannTrackFinderTask::fSttParameters

private

Definition at line 79 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Init(), and SetParContainers().

BinaryFunctor* PndMvdSttGemRiemannTrackFinderTask::fTimeGapFunctor

private

Definition at line 77 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by FillHitArray(), and Init().

TClonesArray* PndMvdSttGemRiemannTrackFinderTask::fTrackArray

private

Definition at line 67 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), FinishEvent(), and Init().

TString PndMvdSttGemRiemannTrackFinderTask::fTrackBranch

private

Definition at line 53 of file PndMvdSttGemRiemannTrackFinderTask.h.

TClonesArray* PndMvdSttGemRiemannTrackFinderTask::fTrackCandArray

private

Definition at line 65 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), FinishEvent(), and Init().

TClonesArray* PndMvdSttGemRiemannTrackFinderTask::fTubeArray

private

Definition at line 80 of file PndMvdSttGemRiemannTrackFinderTask.h.

Referenced by Exec(), and Init().

---

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

• PndMvdSttGemRiemannTrackFinderTask.h
• PndMvdSttGemRiemannTrackFinderTask.cxx