Possibility to use only ITSTPC or only ITSSA tracks to rereco vertex

diff --git a/PWG1/global/AliAnalysisTaskVertexESD.cxx b/PWG1/global/AliAnalysisTaskVertexESD.cxx
index 6c1196ab063f84ea1a850122e1d8203afb37c843..ffc76f0d9dbe3a872a85f0c99534ce1c3c5b146c 100644 (file)
--- a/PWG1/global/AliAnalysisTaskVertexESD.cxx
+++ b/PWG1/global/AliAnalysisTaskVertexESD.cxx
@@ -66,6 +66,8 @@ fCheckEventType(kTRUE),
 fReadMC(kFALSE),
 fRecoVtxTPC(kFALSE),
 fRecoVtxITSTPC(kFALSE),
+fOnlyITSTPCTracks(kFALSE),
+fOnlyITSSATracks(kFALSE),
 fESD(0), 
 fOutput(0), 
 fNtupleVertexESD(0),
@@ -386,5 +388,22 @@ AliESDVertex* AliAnalysisTaskVertexESD::ReconstructPrimaryVertexITSTPC() const {
   delete initVertex;
   vertexer.SetConstraintOff();
 
+  // use only ITS-TPC or only ITS-SA tracks
+  if(fOnlyITSTPCTracks || fOnlyITSSATracks) {
+    Int_t iskip=0;
+    Int_t *skip = new Int_t[fESD->GetNumberOfTracks()];
+    for(Int_t itr=0;itr<fESD->GetNumberOfTracks(); itr++) {
+      AliESDtrack* track = fESD->GetTrack(itr);
+      if(fOnlyITSTPCTracks && track->GetNcls(1)==0) { // skip ITSSA
+       skip[iskip++]=itr;
+      }
+      if(fOnlyITSSATracks && track->GetNcls(1)>0) { // skip ITSTPC
+       skip[iskip++]=itr;
+      }
+    }
+    vertexer.SetSkipTracks(iskip,skip);
+    delete [] skip; skip=NULL;
+  }
+
   return vertexer.FindPrimaryVertex(fESD);
 }

diff --git a/PWG1/global/AliAnalysisTaskVertexESD.h b/PWG1/global/AliAnalysisTaskVertexESD.h
index 438d4d084ff8c214e394a872003478ebcc4ab56d..c75ea62b8239ba61a18d3e18c9b5c0ddcbaaed88 100644 (file)
--- a/PWG1/global/AliAnalysisTaskVertexESD.h
+++ b/PWG1/global/AliAnalysisTaskVertexESD.h
@@ -40,12 +40,16 @@ class AliAnalysisTaskVertexESD : public AliAnalysisTask
   void           SetReadMC(Bool_t flag=kTRUE) { fReadMC=flag; if(flag) fCheckEventType=kFALSE;}
   void           SetRerecoVertexTPC(Bool_t flag=kTRUE) { fRecoVtxTPC=flag; }
   void           SetRerecoVertexITSTPC(Bool_t flag=kTRUE) { fRecoVtxITSTPC=flag; }
+  void           SetOnlyITSTPCTracks() {fOnlyITSTPCTracks=kTRUE;}
+  void           SetOnlyITSSATracks() {fOnlyITSSATracks=kTRUE;}
   
  protected:
   Bool_t       fCheckEventType; // read only events of type 7
   Bool_t       fReadMC;         // read Monte Carlo
   Bool_t       fRecoVtxTPC;     // reco TPC vertex on the flight
   Bool_t       fRecoVtxITSTPC;  // reco ITS+TPC vertex on the flight
+  Bool_t       fOnlyITSTPCTracks; // only ITS-TPC tracks to redo ITSTPC vertex
+  Bool_t       fOnlyITSSATracks;  // only ITS-SA tracks to redo ITSTPC vertex
   AliESDEvent *fESD;            // ESD object
   TList       *fOutput;         //! list send on output slot 0
   TNtuple     *fNtupleVertexESD;//! output ntuple