]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ANALYSIS/AliAODParticle.cxx
Commit new class - for external track comparison
[u/mrichter/AliRoot.git] / ANALYSIS / AliAODParticle.cxx
index a8a44695fc8a50a4a36c435ebb3ec4a858dae7b4..ea6f06e713aa8acd5cdf19d0a02a3b043a87e8ad 100644 (file)
@@ -1,4 +1,20 @@
-#include "AliAODParticle.h"
+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ *                                                                        *
+ * Author: The ALICE Off-line Project.                                    *
+ * Contributors are mentioned in the code where appropriate.              *
+ *                                                                        *
+ * Permission to use, copy, modify and distribute this software and its   *
+ * documentation strictly for non-commercial purposes is hereby granted   *
+ * without fee, provided that the above copyright notice appears in all   *
+ * copies and that both the copyright notice and this permission notice   *
+ * appear in the supporting documentation. The authors make no claims     *
+ * about the suitability of this software for any purpose. It is          *
+ * provided "as is" without express or implied warranty.                  *
+ **************************************************************************/
+
+/* $Id$ */
+
 //___________________________________________________________
 /////////////////////////////////////////////////////////////
 //
 //
 /////////////////////////////////////////////////////////////
 #include <TParticle.h>
-#include "AliTrackPoints.h"
+
+#include "AliAODParticle.h"
 #include "AliClusterMap.h"
+#include "AliLog.h"
+#include "AliTrackPoints.h"
 
 ClassImp(AliAODParticle)
 
@@ -76,11 +95,12 @@ AliAODParticle::AliAODParticle(const AliAODParticle& in):
    fPdgIdx(in.fPdgIdx), fIdxInEvent(in.fIdxInEvent),
    fNPids(in.fNPids),fPids(new Int_t[fNPids]),fPidProb(new Float_t[fNPids]),
    fCalcMass(in.GetCalcMass()),
-   fPx(in.Px()),fPy(in.Py()),fPz(in.Pz()),fE(in.Energy()), 
+   fPx(in.Px()),fPy(in.Py()),fPz(in.Pz()),fE(in.E()), 
    fVx(in.Vx()),fVy(in.Vy()),fVz(in.Vz()),fVt(in.T()),
    fTPCTrackPoints(0x0),fITSTrackPoints(0x0),fClusterMap(0x0)
 {
  //Copy constructor
+// Info("AliAODParticle(const AliAODParticle& in)","");
  for(Int_t i = 0; i<fNPids; i++)
   {
     fPids[i] =  in.fPids[i];
@@ -106,7 +126,7 @@ AliAODParticle::AliAODParticle(const AliVAODParticle& in):
    fTPCTrackPoints(0x0),fITSTrackPoints(0x0),fClusterMap(0x0)
 {
  //Copy constructor
+// Info("AliAODParticle(const AliVAODParticle& in)","");
  for(Int_t i = 0; i<in.GetNumberOfPids(); i++)
   {
     SetPIDprobability(in.GetNthPid(i),in.GetNthPidProb(i));
@@ -148,48 +168,68 @@ AliAODParticle::~AliAODParticle()
 }
 //______________________________________________________________________________
 
-AliAODParticle& AliAODParticle::operator=(const AliAODParticle& in)
+void AliAODParticle::Clear(Option_t*)
 {
-//assigment operator
-  
-  fNPids = in.fNPids;
+//Must be implemented in order to store this object in Clones Array
   delete [] fPids;
   delete [] fPidProb;
-  Int_t* fPids = new Int_t[fNPids];
-  Float_t* fPidProb = new Float_t[fNPids];
-  for (Int_t i = 0; i < fNPids;i++)
-   {
-     fPids[i]    = in.fPids[i];
-     fPidProb[i] = in.fPidProb[i];
-   }
-   
-  fPdgIdx = in.fPdgIdx; 
-  fIdxInEvent = in.fIdxInEvent;  
-  fCalcMass = in.GetCalcMass();
-  fPx = in.Px();
-  fPy = in.Py();
-  fPz = in.Pz();
-  fE = in.Energy(); 
-  fVx = in.Vx();
-  fVy = in.Vy();
-  fVz = in.Vz();
-  fVt = in.T();
-  
   delete fTPCTrackPoints;
-  fTPCTrackPoints = (in.fTPCTrackPoints)?(AliTrackPoints*)fTPCTrackPoints->Clone():0x0;
-
   delete fITSTrackPoints;
-  fITSTrackPoints = (in.fITSTrackPoints)?(AliTrackPoints*)fITSTrackPoints->Clone():0x0;
-  
   delete fClusterMap;
-  fClusterMap =  (in.fClusterMap)?(AliClusterMap*)in.fClusterMap->Clone():0x0;
   
+  fPids = 0x0;
+  fPidProb = 0x0;
+  fTPCTrackPoints = 0x0;
+  fITSTrackPoints = 0x0;
+  fClusterMap = 0x0;
+}
+//______________________________________________________________________________
+
+AliAODParticle& AliAODParticle::operator=(const AliAODParticle& in)
+{
+//assigment operator
+//  Info("operator=(const AliAODParticle& in)","AliAODParticle");
+  if (&in != this) {
+    AliVAODParticle::operator=(in);
+    fNPids = in.fNPids;
+    delete [] fPids;
+    delete [] fPidProb;
+    fPids = new Int_t[fNPids];
+    fPidProb = new Float_t[fNPids];
+    for (Int_t i = 0; i < fNPids;i++)
+      {
+       fPids[i]    = in.fPids[i];
+       fPidProb[i] = in.fPidProb[i];
+      }
+    
+    fPdgIdx = in.fPdgIdx; 
+    fIdxInEvent = in.fIdxInEvent;  
+    fCalcMass = in.GetCalcMass();
+    fPx = in.Px();
+    fPy = in.Py();
+    fPz = in.Pz();
+    fE = in.E(); 
+    fVx = in.Vx();
+    fVy = in.Vy();
+    fVz = in.Vz();
+    fVt = in.T();
+    
+    delete fTPCTrackPoints;
+    fTPCTrackPoints = (in.fTPCTrackPoints)?(AliTrackPoints*)in.fTPCTrackPoints->Clone():0x0;
+    
+    delete fITSTrackPoints;
+    fITSTrackPoints = (in.fITSTrackPoints)?(AliTrackPoints*)in.fITSTrackPoints->Clone():0x0;
+    
+    delete fClusterMap;
+    fClusterMap =  (in.fClusterMap)?(AliClusterMap*)in.fClusterMap->Clone():0x0;
+  }
   return *this;
 }
 //______________________________________________________________________________
 
 void AliAODParticle::SetPdgCode(Int_t pdg,Float_t prob)
 {
+//Set PDG Code
   SetPIDprobability(pdg,prob);
   fPdgIdx = GetPidSlot(pdg);
 }
@@ -199,13 +239,14 @@ void AliAODParticle::SetPIDprobability(Int_t pdg, Float_t prob)
 {
 //Sets another pdg code and corresponding probabilty
 //Ids are set in decreasing order
-//Check if total prbaility is not ivercoming unity is performed
+//Check if total probability is not overcoming unity is performed
 //in case, warning is printed
-  if (GetDebug() > 9) Info("SetPIDprobability","Setting PID %d prob %f",pdg,prob);
+  AliDebug(9,Form("SetPIDprobability","Setting PID %d prob %f",pdg,prob));
 
   Float_t totprob = 0.0;//sums up probabilities
   Int_t idx = GetPidSlot(pdg);
   Int_t i;
+
   if (idx > -1) 
    {
      fPidProb[idx] = prob;
@@ -214,10 +255,7 @@ void AliAODParticle::SetPIDprobability(Int_t pdg, Float_t prob)
        {
          Warning("SetPIDprobability","Total probability greater than unity (%f)",totprob);
        }
-     if (GetDebug() > 9) 
-      {
-        Info("SetPIDprobability","Current Total probability: %f",totprob);
-      }
+     AliDebug(9,Form("Current Total probability: %f",totprob));
      return;
    }
     
@@ -230,7 +268,7 @@ void AliAODParticle::SetPIDprobability(Int_t pdg, Float_t prob)
    {
      if ( fPidProb[i] > prob)
       {
-        if (GetDebug()>9) Info("SetPID","Copying entry %d",i);
+        AliDebug(9,Form("Copying entry %d",i));
         aPidProbNew[i] = fPidProb[i];
         aPidsNew[i] = fPids[i];
         totprob+=fPidProb[i];
@@ -238,15 +276,15 @@ void AliAODParticle::SetPIDprobability(Int_t pdg, Float_t prob)
      else break;
    }
 
-  if (GetDebug() > 9) Info("SetPID","Setting new PID on entry %d",i);
+  AliDebug(9,Form("SetPID","Setting new PID on entry %d",i));
   aPidProbNew[i] = prob;
   aPidsNew[i] = pdg;
   totprob+=prob;
   
 
-  for (Int_t j = fNPids-1; j > i ;j--)//copy rest of old araays 
+  for (Int_t j = fNPids-1; j > i ;j--)//copy rest of old arays 
    {
-     if (GetDebug() > 9) Info("SetPID","Copying from old entry %d to new entry %d",j-1,j);
+     AliDebug(9,Form("SetPID","Copying from old entry %d to new entry %d",j-1,j));
      aPidProbNew[j] = fPidProb[j-1];
      aPidsNew[j] = fPids[j-1];
      totprob+=fPidProb[j-1];
@@ -324,7 +362,7 @@ Float_t AliAODParticle::GetNthPidProb(Int_t idx) const
 }
 //______________________________________________________________________________
 
-void AliAODParticle::Print() const
+void AliAODParticle::Print(const Option_t * /*opt*/) const
 {
 //prints information about particle
   printf("____________________________________________________\n");
@@ -340,7 +378,7 @@ void AliAODParticle::Print() const
    }
   
   printf("Px: %+f Py: %+f Pz: %+f E: %+f Calculated Mass: %f\nVx: %+f Vy: %+f Vz: %+f T: %+f\n",
-          Px(),Py(),Pz(),Energy(),GetCalcMass(),Vx(),Vy(),Vz(),T());
+          Px(),Py(),Pz(),E(),GetCalcMass(),Vx(),Vy(),Vz(),T());
 
   for (Int_t i = 0; i < fNPids; i++)
    {
@@ -355,6 +393,10 @@ void AliAODParticle::Print() const
         printf("Not known\n");
       }
    }
+  
+  if (fITSTrackPoints) fITSTrackPoints->Print();
+  if (fTPCTrackPoints) fTPCTrackPoints->Print();
+   
 }
 
 //______________________________________________________________________________