Setting the VertexID for the tracks used in the fit of this vertex (A. Dainese)
authorbelikov <belikov@f7af4fe6-9843-0410-8265-dc069ae4e863>
Mon, 6 Apr 2009 10:01:01 +0000 (10:01 +0000)
committerbelikov <belikov@f7af4fe6-9843-0410-8265-dc069ae4e863>
Mon, 6 Apr 2009 10:01:01 +0000 (10:01 +0000)
STEER/AliVertexerTracks.cxx
STEER/AliVertexerTracks.h

index 1c46cca..c3c9270 100644 (file)
@@ -124,7 +124,7 @@ AliVertexerTracks::~AliVertexerTracks()
   if(fIdSel) { delete [] fIdSel; fIdSel=NULL; }
 }
 //----------------------------------------------------------------------------
-AliESDVertex* AliVertexerTracks::FindPrimaryVertex(const AliVEvent *vEvent)
+AliESDVertex* AliVertexerTracks::FindPrimaryVertex(AliVEvent *vEvent)
 {
 //
 // Primary vertex for current ESD or AOD event
@@ -216,6 +216,17 @@ AliESDVertex* AliVertexerTracks::FindPrimaryVertex(const AliVEvent *vEvent)
     olddir->cd();
   }
 
+  // set vertex ID for tracks used in the fit
+  // (only for ESD)
+  if(!inputAOD) {
+    Int_t nIndices = fCurrentVertex->GetNIndices();
+    UShort_t *indices = fCurrentVertex->GetIndices();
+    for(Int_t ind=0; ind<nIndices; ind++) {
+      AliESDtrack *esdt = (AliESDtrack*)vEvent->GetTrack(indices[ind]);
+      esdt->SetVertexID(-1);
+    }
+  }
+
   return fCurrentVertex;
 }
 //----------------------------------------------------------------------------
index 9cc7f84..1445fab 100644 (file)
@@ -39,7 +39,7 @@ class AliVertexerTracks : public TObject {
   AliVertexerTracks(Double_t fieldkG); 
   virtual ~AliVertexerTracks();
 
-  AliESDVertex* FindPrimaryVertex(const AliVEvent *vEvent);
+  AliESDVertex* FindPrimaryVertex(AliVEvent *vEvent);
   AliESDVertex* FindPrimaryVertex(TObjArray *trkArrayOrig,UShort_t *idOrig);
   AliESDVertex* VertexForSelectedTracks(TObjArray *trkArray,UShort_t *id,
                                        Bool_t optUseFitter=kTRUE,