CbmSttFitTracks.cxx 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. // -------------------------------------------------------------------------
  2. // ----- CbmSttFitTracks source file -----
  3. // ----- Created 18/02/05 by V. Friese -----
  4. // -------------------------------------------------------------------------
  5. #include "TClonesArray.h"
  6. #include "FairRootManager.h"
  7. #include "CbmSttFitTracks.h"
  8. #include "CbmSttTrackFitter.h"
  9. #include "CbmSttTrack.h"
  10. #include <iostream>
  11. using namespace std;
  12. // ----- Default constructor -------------------------------------------
  13. CbmSttFitTracks::CbmSttFitTracks()
  14. {
  15. fFitter = NULL;
  16. fTrackArray = NULL;
  17. fNofTracks = 0;
  18. }
  19. // -------------------------------------------------------------------------
  20. // ----- Standard constructor ------------------------------------------
  21. CbmSttFitTracks::CbmSttFitTracks(const char* name,
  22. const char* title,
  23. CbmSttTrackFitter* fitter)
  24. : FairTask(name)
  25. {
  26. fFitter = fitter;
  27. fTrackArray = NULL;
  28. fNofTracks = 0;
  29. fCollectionsComplete = kFALSE;
  30. }
  31. // -------------------------------------------------------------------------
  32. // ----- Destructor ----------------------------------------------------
  33. CbmSttFitTracks::~CbmSttFitTracks()
  34. {
  35. fHitCollectionNames.clear();
  36. }
  37. // -------------------------------------------------------------------------
  38. // ----- Public method Init (abstract in base class) --------------------
  39. InitStatus CbmSttFitTracks::Init()
  40. {
  41. // Check for Track fitter
  42. if (! fFitter)
  43. {
  44. cout << "-E- CbmSttFitTracks: No track fitter selected!" << endl;
  45. return kERROR;
  46. }
  47. // Get and check FairRootManager
  48. FairRootManager
  49. *ioman = FairRootManager::Instance();
  50. if (! ioman)
  51. {
  52. cout << "-E- CbmSttFitTracks::Init: "
  53. << "RootManager not instantised!" << endl;
  54. return kFATAL;
  55. }
  56. // Get SttTrack array
  57. fTrackArray = (TClonesArray*) ioman->GetObject("STTTrack"); //=>SG
  58. if ( ! fTrackArray)
  59. {
  60. cout << "-E- CbmSttFitTracks::Init: No SttTrack array!"
  61. << endl;
  62. return kERROR;
  63. }
  64. // Call the Init method of the track fitter
  65. fFitter->Init();
  66. return kSUCCESS;
  67. }
  68. // -------------------------------------------------------------------------
  69. // ----- Public method Exec --------------------------------------------
  70. void CbmSttFitTracks::Exec(Option_t* opt)
  71. {
  72. AddAllCollections();
  73. if ( !fTrackArray )
  74. return; // =>SG
  75. Int_t nTracks = fTrackArray->GetEntriesFast();
  76. for (Int_t iTrack=0; iTrack<nTracks; iTrack++)
  77. {
  78. CbmSttTrack* pTrack = (CbmSttTrack*)fTrackArray->At(iTrack);
  79. fFitter->DoFit(pTrack);
  80. }
  81. }
  82. // -------------------------------------------------------------------------
  83. // ----- Public method Finish ------------------------------------------
  84. void CbmSttFitTracks::Finish()
  85. {
  86. }
  87. // -------------------------------------------------------------------------
  88. void CbmSttFitTracks::AddHitCollectionName(char *hitCollectionName)
  89. {
  90. string
  91. newHitName(hitCollectionName);
  92. fHitCollectionNames.push_back(newHitName);
  93. }
  94. void CbmSttFitTracks::AddHitCollection(char const *hitCollectionName)
  95. {
  96. // Get and check FairRootManager
  97. FairRootManager
  98. *ioman = FairRootManager::Instance();
  99. if (!ioman)
  100. {
  101. cout << "-E- CbmSttFindTracks::AddHitCollection: "
  102. << "RootManager not instantised!" << endl;
  103. }
  104. // Get hit Array
  105. TClonesArray
  106. *fHitArray = (TClonesArray*) ioman->GetObject(hitCollectionName);
  107. if (!fHitArray)
  108. {
  109. cout << "-W- CbmSttFindTracks::AddHitCollection: No " << hitCollectionName << " STT hit array!"
  110. << endl;
  111. }
  112. fFitter->AddHitCollection(fHitArray);
  113. }
  114. void CbmSttFitTracks::AddAllCollections()
  115. {
  116. if (!fCollectionsComplete)
  117. {
  118. for (int counter = 0; counter < fHitCollectionNames.size(); counter++)
  119. {
  120. AddHitCollection(fHitCollectionNames[counter].c_str());
  121. }
  122. fCollectionsComplete = kTRUE;
  123. }
  124. }
  125. ClassImp(CbmSttFitTracks)