TPC/Upgrade/AliToyMCTrack.h

   1 #ifndef TOYMCTRACK_H
   2 #define TOYMCTRACK_H
   3
   4 #include <AliExternalTrackParam.h>
   5 #include <TClonesArray.h>
   6 #include <AliTPCclusterMI.h>
   7 #include <AliCluster.h>
   8
   9 class AliToyMCTrack : public AliExternalTrackParam {
  10
  11  public:
  12   AliToyMCTrack();
  13   AliToyMCTrack(Double_t x, Double_t alpha,
  14              const Double_t param[5],
  15              const Double_t covar[15]);
  16   AliToyMCTrack(Double_t xyz[3],Double_t pxpypz[3],
  17              Double_t cv[21],Short_t sign);
  18   AliToyMCTrack(const AliToyMCTrack &track);
  19   AliToyMCTrack& operator=(const AliToyMCTrack &track);
  20   virtual ~AliToyMCTrack() {}
  21
  22   Int_t GetNumberOfSpacePoints()     const { return fSpacePoints.GetEntriesFast(); }
  23   Int_t GetNumberOfDistSpacePoints() const { return fDistortedSpacePoints.GetEntriesFast(); }
  24   Int_t GetNumberOfITSPoints() const { return fITSPoints.GetEntriesFast(); }
  25   Int_t GetNumberOfTRDPoints() const { return fTRDPoints.GetEntriesFast(); }
  26
  27   const AliTPCclusterMI* GetSpacePoint(Int_t spoint) const { return static_cast<AliTPCclusterMI*> (fSpacePoints.At(spoint)); }
  28   const AliTPCclusterMI* GetDistortedSpacePoint(Int_t dspoint) const { return static_cast<AliTPCclusterMI*> (fDistortedSpacePoints.At(dspoint)); }
  29   const AliCluster* GetITSPoint(Int_t spoint) const { return static_cast<AliCluster*> (fITSPoints.At(spoint)); }
  30   const AliCluster* GetTRDPoint(Int_t spoint) const { return static_cast<AliCluster*> (fTRDPoints.At(spoint)); }
  31
  32   AliTPCclusterMI* AddSpacePoint(const AliTPCclusterMI &spoint);
  33   AliTPCclusterMI* AddDistortedSpacePoint(const AliTPCclusterMI &dspoint);
  34   AliCluster* AddITSPoint(const AliCluster &spoint);
  35   AliCluster* AddTRDPoint(const AliCluster &spoint);
  36  private:
  37
  38   TClonesArray fSpacePoints;
  39   TClonesArray fDistortedSpacePoints;
  40   TClonesArray fITSPoints;
  41   TClonesArray fTRDPoints;
  42
  43   ClassDef(AliToyMCTrack,2)
  44 };
  45
  46 #endif