AliParticleContainer.h

Go to the documentation of this file.

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

#include <iterator>
#include <TArrayI.h>
#include <AliVParticle.h>

class AliVEvent;
class AliTLorentzVector;

#include "AliEmcalContainer.h"

class AliParticleContainer : public AliEmcalContainer {
 public:

  class accept_iterator : public std::iterator<std::bidirectional_iterator_tag,
                                               AliVParticle,std::ptrdiff_t,
                                               AliVParticle **, AliVParticle *&>
  {
  public:
    accept_iterator(const AliParticleContainer *cont, int startpos, bool forward  = true);
    accept_iterator(const accept_iterator &other);
    ~accept_iterator() {}
    accept_iterator &operator=(const accept_iterator &other);
    bool operator!=(const accept_iterator &other) const;

    accept_iterator &operator++();
    accept_iterator operator++(int);
    accept_iterator &operator--();
    accept_iterator operator--(int);

    AliVParticle *operator*() const;

  private:
    accept_iterator();

    const AliParticleContainer      *fkContainer;       
    TArrayI                          fAcceptIndices;    
    int                              fCurrentPos;       
    bool                             fForward;          
  };

  class all_iterator : public std::iterator<std::bidirectional_iterator_tag,
                                            AliVParticle,std::ptrdiff_t,
                                            AliVParticle **, AliVParticle *&>
  {
  public:
    all_iterator(const AliParticleContainer *cont, int startpos, bool forward  = true);
    all_iterator(const all_iterator &other);
    ~all_iterator() {}
    all_iterator &operator=(const all_iterator &other);
    bool operator!=(const all_iterator &other) const;

    all_iterator &operator++();
    all_iterator operator++(int);
    all_iterator &operator--();
    all_iterator operator--(int);

    AliVParticle *operator*() const;

  private:
    all_iterator();

    const AliParticleContainer      *fkContainer;       
    int                              fCurrentPos;       
    bool                             fForward;          
  };

  AliParticleContainer();
  AliParticleContainer(const char *name);
  virtual ~AliParticleContainer(){;}

  virtual Bool_t              ApplyParticleCuts(const AliVParticle* vp, UInt_t &rejectionReason) const;
  virtual Bool_t              ApplyKinematicCuts(const AliTLorentzVector& mom, UInt_t &rejectionReason) const;
  virtual Bool_t              AcceptObject(Int_t i, UInt_t &rejectionReason) const              { return AcceptParticle(i, rejectionReason);}
  virtual Bool_t              AcceptObject(const TObject* obj, UInt_t &rejectionReason) const   { return AcceptParticle(dynamic_cast<const AliVParticle*>(obj), rejectionReason);}
  virtual Bool_t              AcceptParticle(const AliVParticle* vp, UInt_t &rejectionReason) const        ;
  virtual Bool_t              AcceptParticle(Int_t i, UInt_t &rejectionReason) const                       ;
  Double_t                    GetParticlePtCut()                        const   { return GetMinPt()     ; }
  Double_t                    GetParticleEtaMin()                       const   { return GetMinEta()    ; }
  Double_t                    GetParticleEtaMax()                       const   { return GetMaxEta()    ; }
  Double_t                    GetParticlePhiMin()                       const   { return GetMinPhi()    ; }
  Double_t                    GetParticlePhiMax()                       const   { return GetMaxPhi()    ; }
  void                        SetParticlePtCut(Double_t cut)                    { SetMinPt(cut)         ; }
  void                        SetParticleEtaLimits(Double_t min, Double_t max)  { SetEtaLimits(min, max); }
  void                        SetParticlePhiLimits(Double_t min, Double_t max)  { SetPhiLimits(min, max); }
  virtual AliVParticle       *GetLeadingParticle(const char* opt="")         ;
  virtual AliVParticle       *GetParticle(Int_t i=-1)                   const;
  virtual AliVParticle       *GetAcceptParticle(Int_t i=-1)             const;
  virtual AliVParticle       *GetNextAcceptParticle()                        ;
  virtual AliVParticle       *GetNextParticle()                              ;
  virtual Bool_t              GetMomentumFromParticle(TLorentzVector &mom, const AliVParticle* part, Double_t mass) const;
  virtual Bool_t              GetMomentumFromParticle(TLorentzVector &mom, const AliVParticle* part) const;
  virtual Bool_t              GetMomentum(TLorentzVector &mom, Int_t i) const;
  virtual Bool_t              GetAcceptMomentum(TLorentzVector &mom, Int_t i) const;
  virtual Bool_t              GetNextMomentum(TLorentzVector &mom);
  virtual Bool_t              GetNextAcceptMomentum(TLorentzVector &mom);
  Int_t                       GetNParticles()                           const   {return GetNEntries();}
  Int_t                       GetNAcceptedParticles()                   const;
  void                        SetMinDistanceTPCSectorEdge(Double_t min)         { fMinDistanceTPCSectorEdge = min; }
  void                        SetCharge(Short_t c)                              { fCharge = c         ; }
  void                        SelectHIJING(Bool_t s)                            { if (s) fGeneratorIndex = 0; else fGeneratorIndex = -1; }
  void                        SetGeneratorIndex(Short_t i)                      { fGeneratorIndex = i  ; }

  const char*                 GetTitle() const;

  accept_iterator             accept_begin() const;
  accept_iterator             accept_end() const;
  accept_iterator             accept_rbegin() const;
  accept_iterator             accept_rend() const;

  all_iterator                begin() const;
  all_iterator                end() const;
  all_iterator                rbegin() const;
  all_iterator                rend() const;

 protected:

  Double_t                    fMinDistanceTPCSectorEdge;      
  Short_t                     fCharge;                        
  Short_t                     fGeneratorIndex;                

 private:
  AliParticleContainer(const AliParticleContainer& obj); // copy constructor
  AliParticleContainer& operator=(const AliParticleContainer& other); // assignment

  ClassDef(AliParticleContainer,9);

};

int TestParticleContainerIterator(const AliParticleContainer *const cont, int iteratorType = 0, bool verbose = false);


#endif