Add option to run on some local files with test mode using the alien plugin

[u/mrichter/AliRoot.git] / MFT / AliMuonForwardTrackPair.h

diff --git a/MFT/AliMuonForwardTrackPair.h b/MFT/AliMuonForwardTrackPair.h
index 34e85b6eb20304dfdc3f743d789de2bec733b15a..c5659a22279f8f3760687220b355f7fe1ce5727c 100644 (file)
--- a/MFT/AliMuonForwardTrackPair.h
+++ b/MFT/AliMuonForwardTrackPair.h
@@ -32,20 +32,51 @@ public:
 
   AliMuonForwardTrackPair(const AliMuonForwardTrackPair&);
   AliMuonForwardTrackPair &operator=(const AliMuonForwardTrackPair&);
-  
-  virtual ~AliMuonForwardTrackPair() {}
+  virtual void  Clear(const Option_t* /*opt*/) { fMuonForwardTracks->Delete(); delete fMuonForwardTracks; fMuonForwardTracks = 0x0; }
 
-  void SetTrack(Int_t iTrack, AliMuonForwardTrack *track);
-  AliMuonForwardTrack* GetTrack(Int_t iTrack) { if (iTrack==0 || iTrack==1) return (AliMuonForwardTrack*) fMuonForwardTracks->At(iTrack); else return NULL; }
+  virtual ~AliMuonForwardTrackPair() { fMuonForwardTracks->Delete(); delete fMuonForwardTracks; }
+
+  AliMuonForwardTrack* GetTrack(Int_t iTrack) { 
+    if (iTrack==0 || iTrack==1) return (AliMuonForwardTrack*) fMuonForwardTracks->At(iTrack); 
+    else return NULL; 
+  }
+
+  Int_t GetCharge() { return GetTrack(0)->GetCharge() + GetTrack(1)->GetCharge(); }
+
+  void SetKinemMC();
+  void SetKinem(Double_t z, Int_t nClusters=-1);
+  Bool_t IsKinemSet() { return fIsKinemSet; }
+
+  void SetPointOfClosestApproach();
+  void GetPointOfClosestApproach(Double_t *xyz) { 
+    xyz[0] = fXPointOfClosestApproach; 
+    xyz[1] = fYPointOfClosestApproach; 
+    xyz[2] = fZPointOfClosestApproach; 
+  }
 
   Double_t GetWeightedOffset(Double_t x, Double_t y, Double_t z);
-  Double_t GetMass(Double_t z, Int_t nClusters=-1);
+  Double_t GetWeightedOffsetAtPCA();
+  Double_t GetPCAQuality();
   Double_t GetMassWithoutMFT(Double_t x, Double_t y, Double_t z, Int_t nClusters=-1);
-  Double_t GetMassMC();
+  Double_t GetMassMC()     { return fKinemMC.M(); }
+  Double_t GetRapidityMC() { return fKinemMC.Rapidity(); }
+  Double_t GetPtMC()       { return fKinemMC.Pt(); }
+  Double_t GetMass()     { return fKinem.M(); }
+  Double_t GetRapidity() { return fKinem.Rapidity(); }
+  Double_t GetPx()       { return fKinem.Px(); }
+  Double_t GetPy()       { return fKinem.Py(); }
+  Double_t GetPz()       { return fKinem.Pz(); }
+  Double_t GetPt()       { return fKinem.Pt(); }
+
+  Bool_t IsResonance();
 
 protected:
 
   TClonesArray *fMuonForwardTracks;
+  TLorentzVector fKinemMC, fKinem;
+  Bool_t fIsKinemSet;
+
+  Double_t fXPointOfClosestApproach, fYPointOfClosestApproach, fZPointOfClosestApproach;
 
   ClassDef(AliMuonForwardTrackPair,1)