AliBaseMCTrackDensity.h

Go to the documentation of this file.

#ifndef ALIBaseMCTRACKDENSITY_MC
#define ALIBaseMCTRACKDENSITY_MC

#include <TNamed.h>
#include <TVector3.h>
class TList;
class TH1D;
class TH2D;
class TF1;
class TGraph;
class AliMCEvent;
class AliMCParticle;
class AliTrackReference;
class AliStack;
class AliBaseMCWeights;

class AliBaseMCTrackDensity : public TNamed
{
public:
  AliBaseMCTrackDensity();
  AliBaseMCTrackDensity(const char* name);
  AliBaseMCTrackDensity(const AliBaseMCTrackDensity& o);
  AliBaseMCTrackDensity& operator=(const AliBaseMCTrackDensity& o);
  virtual ~AliBaseMCTrackDensity() {}

  void SetUseOnlyPrimary(Bool_t use) { fUseOnlyPrimary = use; }
  void SetUseFlowWeights(Bool_t use);
  void SetTrackGammaToPi0(Bool_t use) { fTrackGammaToPi0 = use; }
  void SetDebug(Bool_t debug=true) { fDebug = debug; }
  virtual void CreateOutputObjects(TList* list);
  virtual void Print(Option_t* option="") const;
  /*
   * Set weights to use 
   * 
   * @param weights Weights object to use 
   */ 
  void SetWeights(AliBaseMCWeights* weights);
  /*
   * Set weights to use 
   * 
   * @param weights Weights object to use 
   */ 
  void SetTruthWeights(AliBaseMCWeights* weights);
protected:
  // temporary remove constness, since event.GetNumberOfPrimaries() was not const
  Bool_t ProcessTracks(const AliMCEvent&   event, 
           const TVector3&     ip, 
           TH2D*               primary);
  Bool_t ProcessTrack(AliMCParticle* particle, 
          const AliMCParticle* mother);
  virtual Int_t GetDetectorId() const = 0;
  virtual AliTrackReference* ProcessRef(AliMCParticle* particle, 
          const AliMCParticle* mother, 
          AliTrackReference* ref) = 0;
  virtual void BeginTrackRefs() {}
  virtual Bool_t CheckTrackRef(AliTrackReference*) const { return true; }
  virtual void EndTrackRefs(Int_t /*nRefs*/) {}
  virtual Double_t StoreParticle(AliMCParticle*       particle, 
         const AliMCParticle* mother,
         AliTrackReference*   ref) const;
  Bool_t GetCollisionParameters(const AliMCEvent& event);
  Double_t GetTrackRefTheta(const AliTrackReference* ref) const;
  const AliMCParticle* GetMother(Int_t iTr, const AliMCEvent& event) const;
  Double_t CalculateWeight(const AliMCParticle* p,
         Bool_t isPrimary) const;
  Double_t CalculateTruthWeight(const AliMCParticle* p) const;
  Bool_t            fUseOnlyPrimary; // Only use primaries 
  TH2D*             fBinFlow;        // eta,phi bin flow 
  TH2D*             fEtaBinFlow;     // dEta vs eta of strip
  TH2D*             fPhiBinFlow;     // dPhi vs phi of strip
  TH1D*             fNRefs;          // Number of track-references per track
  AliBaseMCWeights* fWeights;        // MC weights
  AliBaseMCWeights* fTruthWeights;   // MC truth weights
  TVector3          fIP;             // IP z-coordinate of this event
  Double_t          fB;              // Impact parameter of this event
  Double_t          fPhiR;           // Reaction plane  of this event
  Bool_t            fDebug;          // Debug flag
  Bool_t            fTrackGammaToPi0;// If true, try to track gamma to pi0
  ClassDef(AliBaseMCTrackDensity,6); // Calculate track-ref density
};

#endif
// Local Variables:
//  mode: C++ 
// End: