New faeture splitting one event "by track charge". So far we had only "by eta" and...

Robert Grajcarek <grajcarek@physi.uni-heidelberg.de>

diff --git a/ANALYSIS/AliEPSelectionTask.cxx b/ANALYSIS/AliEPSelectionTask.cxx
index 9ba565fcffb95e37b7c1f7df775d60d6cb0ce408..99b7d61e68a5d0a85bf0928d97ef042441ea82f4 100644 (file)
--- a/ANALYSIS/AliEPSelectionTask.cxx
+++ b/ANALYSIS/AliEPSelectionTask.cxx
@@ -531,6 +531,33 @@ void AliEPSelectionTask::GetQsub(TVector2 &Q1, TVector2 &Q2, TObjArray* tracklis
         }
       }
     }
+  } else if (fSplitMethod == AliEPSelectionTask::kCharge) {
+     
+    for (Int_t i = 0; i < nt; i++) {
+      weight = 1;
+      track = dynamic_cast<AliVTrack*> (tracklist->At(i));
+      if (!track) continue;
+      weight = GetWeight(track);
+      Short_t cha = track->Charge();
+      idtemp = track->GetID(); 
+      if ((fAnalysisInput.CompareTo("AOD")==0) && (fAODfilterbit == 128)) idtemp = idtemp*(-1) - 1;
+
+      if (cha > 0) {  
+        mQx1 += (weight*cos(2*track->Phi()));
+        mQy1 += (weight*sin(2*track->Phi()));
+        if (fSaveTrackContribution){
+          EP->GetQContributionXArraysub1()->AddAt(weight*cos(2*track->Phi()),idtemp);
+          EP->GetQContributionYArraysub1()->AddAt(weight*sin(2*track->Phi()),idtemp);
+        }
+      } else if (cha < 0) {
+        mQx2 += (weight*cos(2*track->Phi()));
+        mQy2 += (weight*sin(2*track->Phi()));
+        if (fSaveTrackContribution){
+          EP->GetQContributionXArraysub2()->AddAt(weight*cos(2*track->Phi()),idtemp);
+          EP->GetQContributionYArraysub2()->AddAt(weight*sin(2*track->Phi()),idtemp);
+        }
+      }
+    }
   } else {
     printf("plane resolution determination method not available!\n\n ");
     return;

diff --git a/ANALYSIS/AliEPSelectionTask.h b/ANALYSIS/AliEPSelectionTask.h
index fc9731bfab72e6a0f1d105ec32da31d5ac33ef26..646443277c1b97961259cebc21f782b9e58e10c8 100644 (file)
--- a/ANALYSIS/AliEPSelectionTask.h
+++ b/ANALYSIS/AliEPSelectionTask.h
@@ -30,7 +30,7 @@ class AliEPSelectionTask : public AliAnalysisTaskSE {
 
  public:
   
-  enum ResoMethod{kRandom,kEta};
+  enum ResoMethod{kRandom,kEta,kCharge};
 
   AliEPSelectionTask();
   AliEPSelectionTask(const char *name);