The total mult in V0 became float number.

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

diff --git a/STEER/AliMixedEvent.cxx b/STEER/AliMixedEvent.cxx
index 955ae539a6bd8aababb14248aeda791c5202715b..d3800cb3887801d285487f0244b586b8201f41cd 100644 (file)
--- a/STEER/AliMixedEvent.cxx
+++ b/STEER/AliMixedEvent.cxx
@@ -25,28 +25,34 @@
 
 
 #include "AliMixedEvent.h"
+#include "AliExternalTrackParam.h"
+#include "TVector3.h"
+#include "AliVVertex.h"
 #include <TMath.h>
+#include <TMatrix.h>
+#include <TMatrixD.h>
 
 ClassImp(AliMixedEvent)
 
 
 AliMixedEvent::AliMixedEvent() :
     AliVEvent(),
-    fEventList(), 
+    fEventList(),
     fNEvents(0),       
     fNumberOfTracks(0),
-    fNTracksCumul(0)
+    fNTracksCumul(0),
+    fMeanVertex(0)
 {
     // Default constructor
 }
 
-
 AliMixedEvent::AliMixedEvent(const AliMixedEvent& Evnt) :
     AliVEvent(Evnt),
-    fEventList(), 
-    fNEvents(0),       
+    fEventList(),
+    fNEvents(0),
     fNumberOfTracks(0),
-    fNTracksCumul(0)
+    fNTracksCumul(0),
+    fMeanVertex(0)
 { } // Copy constructor
 
 AliMixedEvent& AliMixedEvent::operator=(const AliMixedEvent& vEvnt)
@@ -61,7 +67,7 @@ AliMixedEvent& AliMixedEvent::operator=(const AliMixedEvent& vEvnt)
 void AliMixedEvent::AddEvent(AliVEvent* evt)
 {
     // Add a new event to the list
-    fEventList.Add(evt);
+    fEventList.AddLast(evt);
 }
 
 
@@ -86,12 +92,21 @@ AliVParticle* AliMixedEvent::GetTrack(Int_t i) const
 {
     // Return track # i
     Int_t iEv  = TMath::BinarySearch(fNEvents, fNTracksCumul, i);
+    while((iEv < (fNEvents - 1)) && (fNTracksCumul[iEv] == fNTracksCumul[iEv+1])) {iEv++;}
 
     Int_t irel = i - fNTracksCumul[iEv];
     AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
     return (evt->GetTrack(irel));
 }
 
+const AliVVertex* AliMixedEvent::GetEventVertex(Int_t i) const
+{
+    // Return track # i
+    Int_t iEv  = TMath::BinarySearch(fNEvents, fNTracksCumul, i);
+    while((iEv < (fNEvents - 1)) && (fNTracksCumul[iEv] == fNTracksCumul[iEv+1])) {iEv++;}
+    AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
+    return (evt->GetPrimaryVertex());
+}
 
 void AliMixedEvent::Reset()
 {
@@ -110,3 +125,47 @@ Int_t AliMixedEvent::EventIndex(Int_t itrack)
   // Return the event index for track #itrack
   return  TMath::BinarySearch(fNEvents, fNTracksCumul, itrack);
 }
+
+void AliMixedEvent::ComputeVtx(TObjArray *vertices, Double_t *pos,Double_t *sig){
+//
+// Calculate the mean vertex psoitions from events in the buffer
+ 
+    Int_t nentries = vertices->GetEntriesFast();
+    Double_t sum[3]={0.,0.,0.};
+    Double_t sumsigma[6]={0.,0.,0.,0.,0.,0.};
+    
+    for(Int_t ivtx = 0; ivtx < nentries; ivtx++){
+       AliVVertex *vtx=(AliVVertex*)vertices->UncheckedAt(ivtx);
+       if(!vtx) return;
+       Double_t covariance[6];
+       vtx->GetCovarianceMatrix(covariance);
+       Double_t vtxPos[3];
+       vtx->GetXYZ(vtxPos);
+       if(covariance[0]==0) continue;
+       sum[0]+=vtxPos[0]*(1./covariance[0]);
+       sumsigma[0]+=(1./covariance[0]);
+       if(covariance[2]==0) continue;
+       sum[1]+=vtxPos[1]*(1./covariance[2]);
+       sumsigma[2]+=(1./covariance[2]);
+       if(covariance[5]==0) continue;
+       sum[2]+=vtxPos[2]*(1./covariance[5]);
+       sumsigma[5]+=(1./covariance[5]);
+       if(covariance[1]==0) continue;
+       sumsigma[1]+=(1./covariance[1]);
+       if(covariance[3]==0) continue;
+       sumsigma[3]+=(1./covariance[3]);
+       if(covariance[4]==0) continue;
+       sumsigma[4]+=(1./covariance[4]);
+    }
+    
+    for(Int_t i=0;i<3;i++){
+       if(sumsigma[i]==0) continue;
+       pos[i]=sum[i]/sumsigma[i];
+    }
+    for(Int_t i2=0;i2<6;i2++){
+       if(sumsigma[i2]==0) {sig[i2]=0.; continue;}
+       sig[i2]=1./sumsigma[i2];
+    }
+    return;
+}
+