Added an ID data member to particles and events, and incremented version accordingly

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnMiniEvent.cxx

diff --git a/PWG2/RESONANCES/AliRsnMiniEvent.cxx b/PWG2/RESONANCES/AliRsnMiniEvent.cxx
index 3de40a132d5eca7ccc83dad7996e65f9f614141a..33344c5655c10a3bf36340444cf7719f816d9172 100644 (file)
--- a/PWG2/RESONANCES/AliRsnMiniEvent.cxx
+++ b/PWG2/RESONANCES/AliRsnMiniEvent.cxx
@@ -24,6 +24,18 @@ void AliRsnMiniEvent::AddParticle(AliRsnMiniParticle copy)
    new (fParticles[n]) AliRsnMiniParticle(copy);
 }
 
+//__________________________________________________________________________________________________
+AliRsnMiniParticle* AliRsnMiniEvent::GetParticle(Int_t i)
+{
+//
+// Return the leading particle
+//
+
+   if (i < 0 || i > fParticles.GetEntriesFast()) return 0x0;
+   
+   return (AliRsnMiniParticle*)fParticles[i];
+}
+
 //__________________________________________________________________________________________________
 AliRsnMiniParticle* AliRsnMiniEvent::LeadingParticle()
 {
@@ -36,3 +48,30 @@ AliRsnMiniParticle* AliRsnMiniEvent::LeadingParticle()
    
    return (AliRsnMiniParticle*)fParticles[fLeading];
 }
+
+//__________________________________________________________________________________________________
+Int_t AliRsnMiniEvent::CountParticles(Char_t charge, Int_t cutID)
+{
+//
+// Counts how many particles have the specified charge and cut bit
+// if charge is not '+', '-' or '0', all charges are considered
+// if cut bit is < 0, it is not checked
+//
+
+   Int_t count = 0;
+   TObjArrayIter next(&fParticles);
+   AliRsnMiniParticle *part = 0x0;
+   
+   while ( (part = (AliRsnMiniParticle*)next()) ) {
+      if (charge == '+' || charge == '-' || charge == '0') {
+         if (part->Charge() != charge) continue;
+      }
+      if (cutID > 0) {
+         if (!part->HasCutBit(cutID)) continue;
+      }
+      count++;
+   }
+   
+   return count;
+}
+