From dd91b595c303ed0bda80f9b32c96295d82e27400 Mon Sep 17 00:00:00 2001
From: snelling <snelling@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Mon, 14 Jun 2010 18:36:51 +0000
Subject: [PATCH] method to mark RP and POI's

---
 .../FLOW/AliFlowCommon/AliFlowEventSimple.cxx | 24 +++++++++++++++++++
 PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.h  |  2 ++
 .../FLOW/AliFlowCommon/AliFlowTrackSimple.cxx |  9 ++++++-
 3 files changed, 34 insertions(+), 1 deletion(-)

diff --git a/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.cxx b/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.cxx
index 05cdfea8b42..015a8940884 100644
--- a/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.cxx
+++ b/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.cxx
@@ -592,3 +592,27 @@ void AliFlowEventSimple::AddFlow(Double_t v1, Double_t v2, Double_t v4)
   }
 }
 
+//_____________________________________________________________________________
+void AliFlowEventSimple::TagRP(AliFlowTrackSimpleCuts* cuts)
+{
+  //tag tracks as reference particles (RPs)
+  for (Int_t i=0; i<fNumberOfTracks; i++)
+  {
+    AliFlowTrackSimple* track = static_cast<AliFlowTrackSimple*>(fTrackCollection->At(i));
+    if (!track) continue;
+    if (cuts->PassesCuts(track)) track->SetForRPSelection();
+  }
+}
+
+//_____________________________________________________________________________
+void AliFlowEventSimple::TagPOI(AliFlowTrackSimpleCuts* cuts)
+{
+  //tag tracks as particles of interest (POIs)
+  for (Int_t i=0; i<fNumberOfTracks; i++)
+  {
+    AliFlowTrackSimple* track = static_cast<AliFlowTrackSimple*>(fTrackCollection->At(i));
+    if (!track) continue;
+    if (cuts->PassesCuts(track)) track->SetForPOISelection();
+  }
+}
+
diff --git a/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.h b/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.h
index 9d970f56295..d7121bd95eb 100644
--- a/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.h
+++ b/PWG2/FLOW/AliFlowCommon/AliFlowEventSimple.h
@@ -64,6 +64,8 @@ class AliFlowEventSimple: public TObject {
 
   void ResolutionPt(Double_t res);
   void TagSubeventsInEta(Double_t etaMinA, Double_t etaMaxA, Double_t etaMinB, Double_t etaMaxB );
+  void TagRP(AliFlowTrackSimpleCuts* cuts );
+  void TagPOI(AliFlowTrackSimpleCuts* cuts );
   void CloneTracks(Int_t n);
   void AddV1( Double_t v1 );
   void AddV2( Double_t v2 );
diff --git a/PWG2/FLOW/AliFlowCommon/AliFlowTrackSimple.cxx b/PWG2/FLOW/AliFlowCommon/AliFlowTrackSimple.cxx
index d7f767a65c4..246eade18fc 100644
--- a/PWG2/FLOW/AliFlowCommon/AliFlowTrackSimple.cxx
+++ b/PWG2/FLOW/AliFlowCommon/AliFlowTrackSimple.cxx
@@ -206,5 +206,12 @@ void AliFlowTrackSimple::AddFlow( Double_t v1,
 void AliFlowTrackSimple::Print( Option_t* option ) const
 {
   //print stuff
-  printf("Phi: %.3f, Eta: %.3f, Pt: %.3f\n",fPhi,fEta,fPt);
+  printf("Phi: %.3f, Eta: %+.3f, Pt: %.3f",fPhi,fEta,fPt);
+  if (InRPSelection()) printf(", RP");
+  if (InPOISelection()) printf(", POI");
+  for (Int_t i=0; i<2; i++)
+  {
+    if (InSubevent(i)) printf(", subevent %i",i);
+  }
+  printf("\n");
 }
-- 
2.39.3