MpdKalmanHit.cxx 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  1. /// \class MpdKalmanHit
  2. ///
  3. /// Hit object for Kalman tracking in the MPD detector
  4. /// \author Alexander Zinchenko (LHEP, JINR, Dubna)
  5. #include "MpdKalmanHit.h"
  6. #include <TMath.h>
  7. //__________________________________________________________________________
  8. MpdKalmanHit::MpdKalmanHit()
  9. : TObject(),
  10. fDetectorID(0),
  11. fFlag(0),
  12. fLength(0.),
  13. fIndex(0),
  14. fHitType(kFixedP),
  15. fNofDim(2),
  16. fSignal(0.),
  17. fDist(0.),
  18. fEdge(99.)
  19. {
  20. /// Default constructor
  21. for (Int_t i = 0; i < 2; ++i) fMeas[i] = fErr[i] = fCosSin[i] = 0;
  22. fCosSin[0] = 1.;
  23. }
  24. //__________________________________________________________________________
  25. MpdKalmanHit::MpdKalmanHit(Int_t detID, Int_t nDim, HitType hitType, Double_t *meas, Double_t *err,
  26. Double_t *cosSin, Double_t signal, Double_t dist, Int_t index, Double_t edge)
  27. : TObject(),
  28. fDetectorID(detID),
  29. fFlag(0),
  30. fLength(0.),
  31. //fIndex(index),
  32. fHitType(hitType),
  33. fNofDim(nDim),
  34. fSignal(signal),
  35. fDist(dist),
  36. fEdge(edge)
  37. {
  38. /// Constructor
  39. for (Int_t i = 0; i < 2; ++i) {
  40. fMeas[i] = meas[i];
  41. fErr[i] = err[i];
  42. fCosSin[i] = cosSin[i];
  43. }
  44. SetIndex(index);
  45. }
  46. //__________________________________________________________________________
  47. MpdKalmanHit::MpdKalmanHit (const MpdKalmanHit& hit)
  48. : TObject(hit),
  49. fDetectorID(hit.fDetectorID),
  50. fFlag(hit.fFlag),
  51. fLength(hit.fLength),
  52. fIndex(hit.fIndex),
  53. fHitType(hit.fHitType),
  54. fNofDim(hit.fNofDim),
  55. fSignal(hit.fSignal),
  56. fDist(hit.fDist),
  57. fEdge(hit.fEdge)
  58. {
  59. ///copy constructor
  60. for (Int_t i = 0; i < 2; ++i) {
  61. fMeas[i] = hit.fMeas[i];
  62. fErr[i] = hit.fErr[i];
  63. fCosSin[i] = hit.fCosSin[i];
  64. }
  65. }
  66. //__________________________________________________________________________
  67. MpdKalmanHit::~MpdKalmanHit()
  68. {
  69. /// Destructor
  70. }
  71. //__________________________________________________________________________
  72. Int_t MpdKalmanHit::Compare(const TObject* hit) const
  73. {
  74. /// "Compare" function to sort in descending order in fDist
  75. MpdKalmanHit *kHit = (MpdKalmanHit*) hit;
  76. if (kHit->GetType() == GetType()) {
  77. // Check layers
  78. if (GetLayer() < kHit->GetLayer()) return 1;
  79. else if (GetLayer() > kHit->GetLayer()) return -1;
  80. if (GetType() == kFixedP) {
  81. // Sort according to sector number
  82. if (GetDetectorID() % 1000000 < kHit->GetDetectorID() % 1000000) return -1;
  83. else if (GetDetectorID() % 1000000 > kHit->GetDetectorID() % 1000000) return 1;
  84. }
  85. }
  86. if (TMath::Abs(GetDist()) < TMath::Abs(kHit->GetDist())) return 1;
  87. else if (TMath::Abs(GetDist()) > TMath::Abs(kHit->GetDist())) return -1;
  88. return 0;
  89. }
  90. //__________________________________________________________________________
  91. void MpdKalmanHit::Print(Option_t *opt)
  92. {
  93. /// Print hit info
  94. }
  95. //__________________________________________________________________________
  96. Double_t MpdKalmanHit::GetPos() const
  97. {
  98. /// Distance to (0,0,0)
  99. /*
  100. if (fHitType != kFixedP) return fDist;
  101. printf(" !!! Not implemented for kFixedP hits. Exit. \n");
  102. exit(0);
  103. */
  104. return fDist;
  105. }
  106. //__________________________________________________________________________
  107. void MpdKalmanHit::SetIndex(Int_t indx)
  108. {
  109. /// Add point index
  110. Int_t size = fIndex.GetSize();
  111. fIndex.Set (size + 1);
  112. fIndex[size] = indx;
  113. }
  114. ClassImp(MpdKalmanHit)