AliMCParticleContainer.cxx

Go to the documentation of this file.

//
// Container with name, TClonesArray and cuts for particles
//
// Author: M. Verweij, S. Aiola

#include <iostream>

#include <TClonesArray.h>

#include "AliVEvent.h"
#include "AliLog.h"

#include "AliTLorentzVector.h"
#include "AliMCParticleContainer.h"

ClassImp(AliMCParticleContainer);

AliMCParticleContainer::AliMCParticleContainer():
  AliParticleContainer(),
  fMCFlag(AliAODMCParticle::kPhysicalPrim)
{
  fBaseClassName = "AliAODMCParticle";
  SetClassName("AliAODMCParticle");
}

AliMCParticleContainer::AliMCParticleContainer(const char *name):
  AliParticleContainer(name),
  fMCFlag(AliAODMCParticle::kPhysicalPrim)
{
  fBaseClassName = "AliAODMCParticle";
  SetClassName("AliAODMCParticle");
}

AliAODMCParticle* AliMCParticleContainer::GetMCParticleWithLabel(Int_t lab) const
{
  Int_t i = GetIndexFromLabel(lab);
  if (i >= 0) {
    return GetMCParticle(i);
  }
  else {
    return nullptr;
  }
}

AliAODMCParticle* AliMCParticleContainer::GetAcceptMCParticleWithLabel(Int_t lab)
{
  Int_t i = GetIndexFromLabel(lab);
  if (i >= 0) {
    return GetAcceptMCParticle(i);
  }
  else {
    return nullptr;
  }
}

AliAODMCParticle* AliMCParticleContainer::GetMCParticle(Int_t i) const
{
  if (i == -1) i = fCurrentID;
  if (i < 0 || i >= fClArray->GetEntriesFast()) return 0;
  AliAODMCParticle *vp = static_cast<AliAODMCParticle*>(fClArray->At(i));
  return vp;
}

AliAODMCParticle* AliMCParticleContainer::GetAcceptMCParticle(Int_t i) const
{
  //return pointer to particle if particle is accepted

  UInt_t rejectionReason = 0;
  if (i == -1) i = fCurrentID;
  if (AcceptMCParticle(i, rejectionReason)) {
      return GetMCParticle(i);
  }
  else {
    AliDebugStream(2) << "Particle " << i << " not accepted." << std::endl;
    return nullptr;
  }
}

AliAODMCParticle* AliMCParticleContainer::GetNextAcceptMCParticle()
{
  const Int_t n = GetNEntries();
  AliAODMCParticle *p = 0;
  do {
    fCurrentID++;
    if (fCurrentID >= n) break;
    p = GetAcceptMCParticle(fCurrentID);
  } while (!p);

  return p;
}

AliAODMCParticle* AliMCParticleContainer::GetNextMCParticle()
{
  //Get next particle

  const Int_t n = GetNEntries();
  AliAODMCParticle *p = 0;
  do {
    fCurrentID++;
    if (fCurrentID >= n) break;
    p = GetMCParticle(fCurrentID);
  } while (!p);

  return p;
}

Bool_t AliMCParticleContainer::AcceptMCParticle(const AliAODMCParticle *vp, UInt_t &rejectionReason) const
{
  // Return true if vp is accepted.
  Bool_t r = ApplyMCParticleCuts(vp, rejectionReason);
  if (!r) return kFALSE;

  AliTLorentzVector mom;
  if(!GetMomentumFromParticle(mom, vp)) return false;

  return ApplyKinematicCuts(mom, rejectionReason);
}

Bool_t AliMCParticleContainer::AcceptMCParticle(Int_t i, UInt_t &rejectionReason) const
{
  // Return true if vp is accepted.

  Bool_t r = ApplyMCParticleCuts(GetMCParticle(i), rejectionReason);
  if (!r) return kFALSE;

  AliTLorentzVector mom;
  if (!GetMomentum(mom, i)) return kFALSE;

  return ApplyKinematicCuts(mom, rejectionReason);
}

Bool_t AliMCParticleContainer::ApplyMCParticleCuts(const AliAODMCParticle* vp, UInt_t &rejectionReason) const
{
  // Return true if i^th particle is accepted.
  // Cuts on the particle properties

  if ((vp->GetFlag() & fMCFlag) != fMCFlag) {
    rejectionReason |= kMCFlag;
    return kFALSE;
  }

  return ApplyParticleCuts(vp, rejectionReason);
}

const AliMCParticleIterableContainer AliMCParticleContainer::all() const {
  return AliMCParticleIterableContainer(this, false);
}

const AliMCParticleIterableContainer AliMCParticleContainer::accepted() const {
  return AliMCParticleIterableContainer(this, true);
}

const AliMCParticleIterableMomentumContainer AliMCParticleContainer::all_momentum() const {
  return AliMCParticleIterableMomentumContainer(this, false);
}

const AliMCParticleIterableMomentumContainer AliMCParticleContainer::accepted_momentum() const {
  return AliMCParticleIterableMomentumContainer(this, true);
}

const char* AliMCParticleContainer::GetTitle() const
{
  static TString trackString;
  trackString = TString::Format("%s_pT%04d", GetArrayName().Data(), static_cast<int>(GetMinPt()*1000.0));
  return trackString.Data();
}