Restored functionality

[u/mrichter/AliRoot.git] / STEER / AliAODPhoton.h

diff --git a/STEER/AliAODPhoton.h b/STEER/AliAODPhoton.h
index eff8649c21f2dd77630563de0bff8e696738d860..3b9488635ee90473d6352c6cdc75470419d0ab89 100644 (file)
--- a/STEER/AliAODPhoton.h
+++ b/STEER/AliAODPhoton.h
@@ -11,11 +11,11 @@
 //-------------------------------------------------------------------------
 
 #include <TLorentzVector.h>
-#include "AliVirtualParticle.h"
+#include "AliVParticle.h"
 #include "AliAODVertex.h"
 
 
-class AliAODPhoton : public AliVirtualParticle {
+class AliAODPhoton : public AliVParticle {
 
  public:
     AliAODPhoton();
@@ -24,14 +24,14 @@ class AliAODPhoton : public AliVirtualParticle {
     virtual ~AliAODPhoton();
     AliAODPhoton(const AliAODPhoton& photon); 
     AliAODPhoton& operator=(const AliAODPhoton& photon);
-// AliVirtualParticle methods
+// AliVParticle methods
     virtual Double_t Px()         const { return fMomentum->Px();      }
     virtual Double_t Py()         const { return fMomentum->Py();      }
     virtual Double_t Pz()         const { return fMomentum->Pz();      }
     virtual Double_t Pt()         const { return fMomentum->Pt();      }
     virtual Double_t P()          const { return fMomentum->P();       }
     virtual Double_t OneOverPt()  const { return 1. / fMomentum->Pt(); }
-    virtual Double_t Phi()        const { return fMomentum->Phi();     }
+    virtual Double_t Phi()        const;
     virtual Double_t Theta()      const { return fMomentum->Theta();   }
     virtual Double_t E()          const { return fMomentum->E();       }
     virtual Double_t M()          const { return fMomentum->M();       }
@@ -52,4 +52,12 @@ class AliAODPhoton : public AliVirtualParticle {
     ClassDef(AliAODPhoton,1);
 };
 
+inline Double_t AliAODPhoton::Phi() const
+{
+    // Return phi
+    Double_t phi = fMomentum->Phi();
+    if (phi < 0.) phi += 2. * TMath::Pi();
+    return phi;
+}
+
 #endif