AliParticleMap.h

Go to the documentation of this file.

#ifndef ALIPARTICLEMAP_H
#define ALIPARTICLEMAP_H
/* Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <cstdlib>
#include <vector>
#include <map>

class AliVParticle;
class AliVTrack;

namespace HighPtTracks {

class AliParticleList{
public:
  AliParticleList():
    fParticles()
  {}

  ~AliParticleList(){}

  void AddParticle(AliVTrack *track) { fParticles.push_back(track); }

  AliVTrack *GetParticle(int itrack) const { return fParticles[itrack]; }

  int GetNumberOfParticles() const { return fParticles.size(); }

private:

  std::vector<AliVTrack *>      fParticles;       
};

class AliReconstructedParticlePair{
public:
  AliReconstructedParticlePair():
    fTrueParticle(NULL),
    fRecParticles()
  {}

  AliReconstructedParticlePair(const AliReconstructedParticlePair &ref):
    fTrueParticle(ref.fTrueParticle),
    fRecParticles(ref.fRecParticles)
  {}

  AliReconstructedParticlePair &operator=(const AliReconstructedParticlePair &ref){
    if(this != &ref){
      fTrueParticle = ref.fTrueParticle;
      fRecParticles = ref.fRecParticles;
    }
    return *this;
  }

  ~AliReconstructedParticlePair(){}

  AliVParticle *GetMCTrueParticle() const { return fTrueParticle; }

  const AliParticleList &GetRecTracks() const { return fRecParticles; }

  void SetMCTrueParticle(AliVParticle *const part) { fTrueParticle = part; }

  void SetRecParticles(const AliParticleList &tracks) { fRecParticles = tracks; }

private:

  AliVParticle             *fTrueParticle;            
  AliParticleList           fRecParticles;            
};

class AliParticleMap{
public:
  AliParticleMap():
    fParticles()
  {}
  ~AliParticleMap();

  void AddParticle(AliVTrack *track);
  AliParticleList *GetParticles(int label) const;

  int GetNumberOfParticles() const { return fParticles.size(); }

  void Print() const;

private:
  std::map<int, AliParticleList *>        fParticles;         
};

} /* namespace HighPtTracks */

#endif /* ALIPARTICLEMAP_H */