PndSttFitTracks Class Reference

#include <PndSttFitTracks.h>

Inheritance diagram for PndSttFitTracks:

Public Member Functions
	PndSttFitTracks ()
	PndSttFitTracks (const char *name, const char *title="FairTask", PndSttTrackFitter *fitter=NULL)
virtual	~PndSttFitTracks ()
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)
virtual void	Finish ()
PndSttTrackFitter *	GetFitter ()
Int_t	GetNofTracks ()
void	UseFitter (PndSttTrackFitter *fitter)
void	AddHitCollectionName (char *hitCollectionName)

Private Member Functions
void	AddAllCollections ()
void	AddHitCollection (char const *collectionName)
void	SetParContainers ()
	ClassDef (PndSttFitTracks, 1)

Private Attributes
PndSttTrackFitter *	fFitter
TClonesArray *	fTrackCandArray
TClonesArray *	fTrackArray
TClonesArray *	fHitArray
Int_t	fNofTracks
std::vector< std::string >	fHitCollectionNames
Bool_t	fCollectionsComplete
TClonesArray *	fTubeArray
PndGeoSttPar *	fSttParameters

Detailed Description

Definition at line 31 of file PndSttFitTracks.h.

Constructor & Destructor Documentation

PndSttFitTracks::PndSttFitTracks

(

)

Default constructor

Definition at line 29 of file PndSttFitTracks.cxx.

References fFitter, and fNofTracks.

{
  fFitter        = NULL;
  fNofTracks     = 0;
}

PndSttFitTracks::PndSttFitTracks	(	const char *	name,
		const char *	title = "FairTask",
		PndSttTrackFitter *	fitter = NULL
	)

Standard constructor

Parameters

name	Name of class
title	Task title
fitter	Pointer to STT track fitter concrete class

Definition at line 39 of file PndSttFitTracks.cxx.

References fCollectionsComplete, fFitter, fNofTracks, and SetTitle().

  : FairTask(name) 
{
  fFitter        = fitter;
  fNofTracks     = 0;
  fCollectionsComplete = kFALSE;
  SetTitle(title);
}

PndSttFitTracks::~PndSttFitTracks ( )

virtual

Destructor

Definition at line 54 of file PndSttFitTracks.cxx.

References fHitCollectionNames.

{ 
  fHitCollectionNames.clear();  
}

Member Function Documentation

void PndSttFitTracks::AddAllCollections ( )

private

Definition at line 194 of file PndSttFitTracks.cxx.

References AddHitCollection(), counter, fCollectionsComplete, and fHitCollectionNames.

Referenced by Exec().

{
    if (!fCollectionsComplete)
    {
        for (size_t counter = 0; counter < fHitCollectionNames.size(); counter++)
        {
            AddHitCollection(fHitCollectionNames[counter].c_str());
        }
        fCollectionsComplete = kTRUE;
    }
}

void PndSttFitTracks::AddHitCollection ( char const *  collectionName )

private

Definition at line 170 of file PndSttFitTracks.cxx.

References PndSttTrackFitter::AddHitCollection(), fFitter, and fHitArray.

Referenced by AddAllCollections().

{
    // Get and check FairRootManager
    FairRootManager
        *ioman = FairRootManager::Instance();
    
    if (!ioman) 
    {
        cout << "-E- PndSttFindTracks::AddHitCollection: "
             << "RootManager not instantised!" << endl;
    }
    
    // Get hit Array
    fHitArray = (TClonesArray*) ioman->GetObject(hitCollectionName);

    if (!fHitArray) 
    {
        cout << "-W- PndSttFindTracks::AddHitCollection: No " << hitCollectionName << " STT hit array!"
             << endl;
    }

    fFitter->AddHitCollection(fHitArray);
}

void PndSttFitTracks::AddHitCollectionName

(

char *

hitCollectionName

)

Definition at line 162 of file PndSttFitTracks.cxx.

References fHitCollectionNames.

Referenced by idealcomplete(), and RecoComplete().

{
    string
        newHitName(hitCollectionName);

    fHitCollectionNames.push_back(newHitName);
}

PndSttFitTracks::ClassDef ( PndSttFitTracks  , 1    )

private

void PndSttFitTracks::Exec ( Option_t *  opt )

virtual

Task execution

Definition at line 115 of file PndSttFitTracks.cxx.

References AddAllCollections(), PndSttTrackFitter::DoFit(), fFitter, fHitArray, fTrackArray, fTrackCandArray, fTubeArray, PndSttTube::GetPosition(), PndSttHit::GetTubeID(), i, and PndSttTrack::SetTrackCandIndex().

{
    AddAllCollections();

  if ( !fTrackCandArray ) 
    return; // =>SG
  fTrackArray->Clear();

  Int_t nTracks = fTrackCandArray->GetEntriesFast();

  for (Int_t iTrack=0; iTrack<nTracks; iTrack++) 
    {
        PndTrackCand* pTrackCand = (PndTrackCand*) fTrackCandArray->At(iTrack);
        if(!pTrackCand) continue; 

        Int_t size = fTrackArray->GetEntriesFast(); 
        new((*fTrackArray)[size]) PndSttTrack(); 
        PndSttTrack* pTrack = (PndSttTrack*) fTrackArray->At(size); 
        pTrack->SetTrackCandIndex(iTrack);
        pTrack->SetLink(FairLink("SttTrackCand", iTrack));

        fFitter->DoFit(pTrackCand, pTrack);

    }

  // reset hit coordinates
  for(int i = 0; i < fHitArray->GetEntriesFast(); i++) {
    PndSttHit *currenthit = (PndSttHit*) fHitArray->At(i);
    if(!currenthit) continue;
    // tubeID  CHECK added
    Int_t tubeID = currenthit->GetTubeID();
    PndSttTube *tube = (PndSttTube*) fTubeArray->At(tubeID);
    currenthit->SetX(tube->GetPosition().X());
    currenthit->SetY(tube->GetPosition().Y());
    currenthit->SetZ(tube->GetPosition().Z());
  }
}

void PndSttFitTracks::Finish ( )

virtual

Finish at the end of each event

Definition at line 157 of file PndSttFitTracks.cxx.

{ 
}

PndSttTrackFitter* PndSttFitTracks::GetFitter ( )

inline

Accessors

Definition at line 67 of file PndSttFitTracks.h.

References fFitter.

{ return fFitter; };

Int_t PndSttFitTracks::GetNofTracks ( )

inline

Definition at line 68 of file PndSttFitTracks.h.

References fNofTracks.

{ return fNofTracks; };

InitStatus PndSttFitTracks::Init ( )

virtual

Initialisation at beginning of each event

Definition at line 63 of file PndSttFitTracks.cxx.

References fFitter, PndSttMapCreator::FillTubeArray(), fSttParameters, fTrackArray, fTrackCandArray, fTubeArray, PndSttTrackFitter::Init(), and PndSttTrackFitter::SetTubeArray().

{
  // Check for Track fitter
  if (! fFitter) 
    {
      cout << "-E- PndSttFitTracks: No track fitter selected!" << endl;
      return kERROR;
    }

  // Get and check FairRootManager
  FairRootManager
    *ioman = FairRootManager::Instance();

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

  // Get SttTrack array CHECK add
  fTrackCandArray  = (TClonesArray*) ioman->GetObject("STTTrackCand"); //=>SG
  if ( ! fTrackCandArray) 
    {
      cout << "-E- PndSttFitTracks::Init: No SttTrackCand array!"
           << endl;
      return kERROR;
    }

  // Create and register SttTrack array
  fTrackArray = new TClonesArray("PndSttTrack",100); 
  ioman->Register("STTTrack", "STT", fTrackArray, kTRUE); // fPersistence); // CHECK
  
  // CHECK added 
  PndSttMapCreator mapper(fSttParameters);
  fTubeArray = mapper.FillTubeArray();
  
  // Call the Init method of the track fitter
  fFitter->Init();
  fFitter->SetTubeArray(fTubeArray);
  
  return kSUCCESS;
}

void PndSttFitTracks::SetParContainers ( )

private

Definition at line 109 of file PndSttFitTracks.cxx.

References fSttParameters, and rtdb.

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

void PndSttFitTracks::UseFitter ( PndSttTrackFitter *  fitter )

inline

Set concrete track finder

Definition at line 72 of file PndSttFitTracks.h.

References fFitter.

{ fFitter = fitter; };

Member Data Documentation

Bool_t PndSttFitTracks::fCollectionsComplete

private

Definition at line 89 of file PndSttFitTracks.h.

Referenced by AddAllCollections(), and PndSttFitTracks().

PndSttTrackFitter* PndSttFitTracks::fFitter

private

Definition at line 83 of file PndSttFitTracks.h.

Referenced by AddHitCollection(), Exec(), GetFitter(), Init(), PndSttFitTracks(), and UseFitter().

TClonesArray* PndSttFitTracks::fHitArray

private

Definition at line 86 of file PndSttFitTracks.h.

Referenced by AddHitCollection(), and Exec().

std::vector<std::string> PndSttFitTracks::fHitCollectionNames

private

Definition at line 88 of file PndSttFitTracks.h.

Referenced by AddAllCollections(), AddHitCollectionName(), and ~PndSttFitTracks().

Int_t PndSttFitTracks::fNofTracks

private

Definition at line 87 of file PndSttFitTracks.h.

Referenced by GetNofTracks(), and PndSttFitTracks().

PndGeoSttPar* PndSttFitTracks::fSttParameters

private

Definition at line 94 of file PndSttFitTracks.h.

Referenced by Init(), and SetParContainers().

TClonesArray* PndSttFitTracks::fTrackArray

private

Definition at line 85 of file PndSttFitTracks.h.

Referenced by Exec(), and Init().

TClonesArray* PndSttFitTracks::fTrackCandArray

private

Definition at line 84 of file PndSttFitTracks.h.

Referenced by Exec(), and Init().

TClonesArray* PndSttFitTracks::fTubeArray

private

from parameters array of PndSttTube

Definition at line 92 of file PndSttFitTracks.h.

Referenced by Exec(), and Init().

---

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

• PndSttFitTracks.h
• PndSttFitTracks.cxx