Coverity fix

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

//
// Mini-Event
// Contains only the useful quantities computed on the event
// which can be used for event mixing, or for direct output
// when doing analysis w.r. to multiplicity or event plane, for example.
//
// Author: A. Pulvirenti
//

#include "AliRsnMiniParticle.h"
#include "AliRsnMiniEvent.h"

ClassImp(AliRsnMiniEvent)

//__________________________________________________________________________________________________
void AliRsnMiniEvent::AddParticle(AliRsnMiniParticle copy)
{
//
// Add a new particle to the list and returns a pointer to it,
// in order to allow to se its parameters.
//

   Int_t n = fParticles.GetEntries();
   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()
{
//
// Return the leading particle
//

   if (fLeading < 0) return 0x0;
   if (fLeading >= fParticles.GetEntriesFast()) return 0x0;
   
   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;
}