Updated macro to calculate space charge fluctuation

[u/mrichter/AliRoot.git] / TPC / Upgrade / AliToyMCTrack.h

diff --git a/TPC/Upgrade/AliToyMCTrack.h b/TPC/Upgrade/AliToyMCTrack.h
index 5022ece0ad8d4ada671c2b76cde610e819dbbc5f..405697483d402b2fdd763795d6768980d6caa9d5 100644 (file)
--- a/TPC/Upgrade/AliToyMCTrack.h
+++ b/TPC/Upgrade/AliToyMCTrack.h
@@ -4,7 +4,9 @@
 #include <AliExternalTrackParam.h>
 #include <TClonesArray.h>
 #include <AliTPCclusterMI.h>
-#include <AliCluster.h>
+// #include <AliCluster.h>
+
+class AliTrackPoint;
 
 class AliToyMCTrack : public AliExternalTrackParam {
  
@@ -15,6 +17,7 @@ class AliToyMCTrack : public AliExternalTrackParam {
             const Double_t covar[15]);
   AliToyMCTrack(Double_t xyz[3],Double_t pxpypz[3],
             Double_t cv[21],Short_t sign);
+  AliToyMCTrack(const AliExternalTrackParam &param);
   AliToyMCTrack(const AliToyMCTrack &track);
   AliToyMCTrack& operator=(const AliToyMCTrack &track);
   virtual ~AliToyMCTrack() {}
@@ -26,13 +29,16 @@ class AliToyMCTrack : public AliExternalTrackParam {
 
   const AliTPCclusterMI* GetSpacePoint(Int_t spoint) const { return static_cast<AliTPCclusterMI*> (fSpacePoints.At(spoint)); }
   const AliTPCclusterMI* GetDistortedSpacePoint(Int_t dspoint) const { return static_cast<AliTPCclusterMI*> (fDistortedSpacePoints.At(dspoint)); }
-  const AliCluster* GetITSPoint(Int_t spoint) const { return static_cast<AliCluster*> (fITSPoints.At(spoint)); }
-  const AliCluster* GetTRDPoint(Int_t spoint) const { return static_cast<AliCluster*> (fTRDPoints.At(spoint)); }
-
   AliTPCclusterMI* AddSpacePoint(const AliTPCclusterMI &spoint);
   AliTPCclusterMI* AddDistortedSpacePoint(const AliTPCclusterMI &dspoint);
-  AliCluster* AddITSPoint(const AliCluster &spoint);
-  AliCluster* AddTRDPoint(const AliCluster &spoint);
+
+  const AliTrackPoint* GetITSPoint(Int_t spoint) const { return static_cast<AliTrackPoint*> (fITSPoints.At(spoint)); }
+  const AliTrackPoint* GetTRDPoint(Int_t spoint) const { return static_cast<AliTrackPoint*> (fTRDPoints.At(spoint)); }
+  
+  AliTrackPoint* AddITSPoint(const AliTrackPoint &spoint);
+  AliTrackPoint* AddTRDPoint(const AliTrackPoint &spoint);
+
+  
  private:
 
   TClonesArray fSpacePoints;