Added Gautes changes from Bergen.
[u/mrichter/AliRoot.git] / HLT / trigger / AliD0Trigger.h
index 576d9f9e661f0236b47b8c82e482aa5651c1d9e5..ed5af693db74682cafebedd0b3baec89a8e7e20a 100644 (file)
@@ -2,10 +2,13 @@
 #define AliD0_Trigger
 
 #include "AliL3RootTypes.h"
+#include <math.h>
+#include <TObject.h>
+#include <TObjArray.h>
 
 class AliITStrackV2;
 
-class AliD0Trigger {
+class AliD0Trigger : public TObjArray{
  
  private:
   AliITStrackV2 * posTrack; //!
@@ -20,9 +23,11 @@ class AliD0Trigger {
   AliD0Trigger();
   AliD0Trigger(double c[7],double Bfield,double pv[3]);
   AliD0Trigger(AliITStrackV2 * posT, AliITStrackV2 * negT);
+  AliD0Trigger(double c[7],double Bfield,double pv[3],AliITStrackV2 * posT, AliITStrackV2 * negT);
   virtual ~AliD0Trigger();
 
   void SetTracks(AliITStrackV2 * posT, AliITStrackV2 * negT);
+  void SetV0(double v[3]);
   bool FindInvMass();
   bool FindV0();
   bool FindV0offline(double v[3]);
@@ -38,6 +43,8 @@ class AliD0Trigger {
   double Py(){return (momenta[1]+momenta[4]);}
   double Pz(){return (momenta[2]+momenta[5]);}
   double Energy();
+  //double Eta(){return atanh(cos(atan(Pt()/Pz())));}
+  double Eta(){return 0.5*(log((P()+Pz())/(P()-Pz())));}
   bool pTchild();
 
   ClassDef(AliD0Trigger,1)