Compilation on Windoiws/Cygwin

[u/mrichter/AliRoot.git] / STEER / AliAODJet.cxx

diff --git a/STEER/AliAODJet.cxx b/STEER/AliAODJet.cxx
index cc3ee4c71e3556dfc65021676edeae7d5174f4c5..7fea86783668de0fc86160dfb3f9af039ccc2738 100644 (file)
--- a/STEER/AliAODJet.cxx
+++ b/STEER/AliAODJet.cxx
@@ -16,33 +16,78 @@
 /* $Id$ */
 
 //-------------------------------------------------------------------------
-//     AOD track base class
-//     Author: Markus Oldenburg, CERN
+//     AOD class for jets
+//     Author: Andreas Morsch, CERN
 //-------------------------------------------------------------------------
 
+#include <TLorentzVector.h>
 #include "AliAODJet.h"
 
 ClassImp(AliAODJet)
 
 
 //______________________________________________________________________________
-AliAODJet::AliAODJet() 
+AliAODJet::AliAODJet() :
+    AliVParticle(),
+    fMomentum(0),
+    fRefTracks(new TRefArray())
 {
   // constructor
+    fBackgEnergy[0]   = 0.;     
+    fBackgEnergy[1]   = 0.;
+    fEffectiveArea[0] = 0.;   
+    fEffectiveArea[1] = 0.;   
 }
 
+AliAODJet::AliAODJet(Double_t px, Double_t py, Double_t pz, Double_t e):
+    AliVParticle(),
+    fMomentum(0),
+    fRefTracks(new TRefArray())
+{
+  // constructor
+    fBackgEnergy[0]   = 0.;     
+    fBackgEnergy[1]   = 0.;
+    fEffectiveArea[0] = 0.;   
+    fEffectiveArea[1] = 0.;   
+    fMomentum = new TLorentzVector(px, py, pz, e);
+}
+
+AliAODJet::AliAODJet(TLorentzVector & p):
+    AliVParticle(),
+    fMomentum(0),
+    fRefTracks(new TRefArray())
+{
+  // constructor
+    fBackgEnergy[0]   = 0.;     
+    fBackgEnergy[1]   = 0.;
+    fEffectiveArea[0] = 0.;   
+    fEffectiveArea[1] = 0.;   
+    fMomentum = new TLorentzVector(p);
+}
+
+
 //______________________________________________________________________________
 AliAODJet::~AliAODJet() 
 {
   // destructor
+    delete fMomentum;
+    delete fRefTracks;
 }
 
 //______________________________________________________________________________
 AliAODJet::AliAODJet(const AliAODJet& jet) :
-  AliVirtualParticle(jet)
+    AliVParticle(jet),
+    fMomentum(0),
+    fRefTracks(0)
 {
   // Copy constructor
+    fBackgEnergy[0]   = jet.fBackgEnergy[0];
+    fBackgEnergy[1]   = jet.fBackgEnergy[1];
+    fEffectiveArea[0] = jet.fEffectiveArea[0];
+    fEffectiveArea[1] = jet.fEffectiveArea[1];
 
+    fMomentum  = new TLorentzVector(*jet.fMomentum);
+    fRefTracks = new TRefArray(*jet.fRefTracks);
 }
 
 //______________________________________________________________________________
@@ -50,8 +95,61 @@ AliAODJet& AliAODJet::operator=(const AliAODJet& jet)
 {
   // Assignment operator
   if(this!=&jet) {
+
+    fBackgEnergy[0]   = jet.fBackgEnergy[0];
+    fBackgEnergy[1]   = jet.fBackgEnergy[1];
+    fEffectiveArea[0] = jet.fEffectiveArea[0];
+    fEffectiveArea[1] = jet.fEffectiveArea[1];
+
+    delete fMomentum;
+    fMomentum  = new TLorentzVector(*jet.fMomentum);
+    delete fRefTracks;
+    fRefTracks = new TRefArray(*jet.fRefTracks);    
   }
 
   return *this;
 }
 
+void AliAODJet::Print(Option_t* /*option*/) const 
+{
+  // Print information of all data members
+  printf("Jet 4-vector:\n");
+  printf("     E  = %13.3f\n", E() );
+  printf("     Px = %13.3f\n", Px());
+  printf("     Py = %13.3f\n", Py());
+  printf("     Pz = %13.3f\n", Pz());
+  printf("Background Energy:\n");
+  printf("Charged:  %13.3f\n", ChargedBgEnergy());
+  printf("Neutral:  %13.3f\n", NeutralBgEnergy());
+  printf("Total:    %13.3f\n", TotalBgEnergy());
+  printf("Effective Area: \n");
+  printf("Charged:  %13.3f\n", EffectiveAreaCharged());
+  printf("Neutral:  %13.3f\n", EffectiveAreaNeutral());
+}
+
+void  AliAODJet::SetPxPyPzE(Double_t px, Double_t py, Double_t pz, Double_t e){
+  // 
+  // Set the four Momentum from outside
+  // MomentumVector()->SetPxPyPzE() cannot be used since pointer can be 0x0
+  //
+
+  if(!fMomentum){
+    fMomentum = new TLorentzVector(px,py,pz,e);
+  }
+  else{
+    fMomentum->SetPxPyPzE(px,py,pz,e);
+  }
+}
+
+Double_t AliAODJet::DeltaR(const AliVParticle* part){
+
+  // Helper function to calculate the distance between two jets
+  // or a jet and particle
+
+  Double_t dPhi = Phi() - part->Phi(); 
+  if(dPhi>TMath::Pi())dPhi = dPhi - 2.*TMath::Pi();
+  if(dPhi<(-1.*TMath::Pi()))dPhi = dPhi + 2.*TMath::Pi();
+  Double_t dEta = Eta() - part->Eta(); 
+  Double_t dR = TMath::Sqrt(dPhi*dPhi+dEta*dEta);
+  return dR;
+}