Small mods by Andrei

[u/mrichter/AliRoot.git] / ANALYSIS / AliAODParticle.h

diff --git a/ANALYSIS/AliAODParticle.h b/ANALYSIS/AliAODParticle.h
index e45972e6c4c7bb2a53ce55574a033a2f5969d096..7802056310ce74cd6eea10155cee95d4bce53836 100644 (file)
--- a/ANALYSIS/AliAODParticle.h
+++ b/ANALYSIS/AliAODParticle.h
@@ -17,8 +17,8 @@
 /////////////////////////////////////////////////////////////
 
 #include "AliVAODParticle.h"
-//#include <TLorentzVector.h>
-//#include <TMath.h>
+#include <TLorentzVector.h>
+#include <TMath.h>
 #include <TDatabasePDG.h>
 
 
@@ -45,9 +45,9 @@ public:
   virtual ~AliAODParticle();
   
   AliAODParticle& operator=(const AliAODParticle& in); 
+  AliVAODParticle& operator=(const AliVAODParticle& in); 
   
-  TLorentzVector FourMomentum() const {TLorentzVector v(fPx,fPy,fPz,fE);return v;}
-  TVector3       ProductionVertex() const {TVector3 v(fVx,fVy,fVz);return v;}
+  void           Clear(Option_t* /*option*/ ="");//Must be implemented in order to store this object in Clones Array
   
   void           SetPIDprobability(Int_t pdg, Float_t prob = 1.0);
   Float_t        GetPIDprobability(Int_t pdg) const;
@@ -57,7 +57,7 @@ public:
   
   Int_t          GetPdgCode      () const { return (fPids)?fPids[fPdgIdx]:0;}
 
-  Double_t        GetPidProb      () const { return (fPidProb)?fPidProb[fPdgIdx]:0;}
+  Double_t       GetPidProb      () const { return (fPidProb)?fPidProb[fPdgIdx]:0;}
   
   Int_t          GetUID          () const { return fIdxInEvent;}
   Int_t          GetNumberOfPids () const { return fNPids;}
@@ -93,7 +93,7 @@ public:
     
   Double_t         Pt    () const  //transverse momentum
     { return TMath::Sqrt(fPx*fPx+fPy*fPy); }
-  Double_t         Energy() const { return fE; }
+  Double_t         E() const { return fE; }
   
                                    //Pseudo Rapidity
   Double_t         Eta   () const { if (P() != fPz) return 0.5*TMath::Log((P()+fPz)/(P()-fPz)); 
@@ -123,7 +123,7 @@ public:
   void            SetUID(Int_t id){fIdxInEvent = id;}
   
   const Char_t*   GetName() const; 
-  void            Print() const;
+  void            Print(const Option_t * opt = "") const;
   
   void            SetTPCTrackPoints(AliTrackPoints* tpts){fTPCTrackPoints = tpts;}
   AliTrackPoints* GetTPCTrackPoints() const {return fTPCTrackPoints;}