Bug fix

[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / AliForwardFlowUtil.h

//
// Class used to handle the input from AODs and put it into histograms
// the Forward Flow tasks can run on
//
#ifndef ALIFORWARDFLOWUTIL_H
#define ALIFORWARDFLOWUTIL_H
/**
 * @file   AliForwardFlowUtil.h
 * @author Alexander Hansen alexander.hansen@cern.ch
 * @date   Fri Mar 25 13:15:40 2011
 * 
 * @brief  
 * 
 * 
 * @ingroup pwglf_forward_flow
 */
#include "TNamed.h"
class AliAODForwardMult;
class AliAODEvent;
class TList;
class TGraph;

/**
 * 
 * Class used to handle the input from AODs and put it into histograms
 * the Forward Flow tasks can run on.
 *
 * @ingroup pwglf_forward_tasks_flow
 * @ingroup pwglf_forward_flow
 */
class AliForwardFlowUtil : public TNamed
{
public:
  /**
   * Constructor
   */
  AliForwardFlowUtil();
  /*
   * Constructor
   *
   * @param l list of histograms for flow analysis
   */
  AliForwardFlowUtil(TList* l);
  /**
   * Check that AOD event meet trigger requirements
   * 
   * @param aodfm Forward multplicity AOD event structure 
   * 
   * @return true on success 
   */
  Bool_t AODCheck(const AliAODForwardMult* aodfm);
  /**
   * Loop over AliAODForwardMult object and fill flow histograms
   * 
   * @param AODevent AOD event structure 
   * 
   * @return true on success
   */
  Bool_t LoopAODFMD(const AliAODEvent* AODevent);
  /*
   * Loop over AliAODCentralMult object and fill flow histograms
   * 
   * @param AODevent AOD event structure 
   * 
   * @return true on success
   */
  Bool_t LoopAODSPD(const AliAODEvent* AODevent) const;
  /**
   * Loop over AliAODForwardMult object and fill flow histograms from
   * track refs
   * 
   * @param AODevent AOD event structure 
   * 
   * @return true on success
   */
  Bool_t LoopAODFMDtrrefHits(const AliAODEvent* AODevent) const;
  /**
   * Loop over AliAODCentralMult object and fill flow histograms from
   * track refs
   * 
   * @param AODevent AOD event structure 
   * 
   * @return true on success
   */
  Bool_t LoopAODSPDtrrefHits(const AliAODEvent* AODevent) const;
  /**
   * Loop over AliAODMCParticle branch object and fill flow histograms
   * add flow if arguments are set
   * 
   * @param AODevent AOD event structure 
   * @param addFlow  What to add flow to 
   * @param type     Type of flow 
   * @param order    Order of added flow 
   * 
   * @return true on success
   */
  Bool_t LoopAODmc(const AliAODEvent* AODevent, TString addFlow, 
                  Int_t type, Int_t order) const;
 /**
   * Get centrality from newest processed event
   */
  Double_t GetCentrality() const { return fCent; }
 /**
   * Get z vertex coordinate from newest processed event
   */
  Float_t GetVertex() const { return fVertex; }
 /**
   * Parametrize ALICE data points
   */
   
protected:
  /*
   * Copy constructor
   *
   * @param o Object to copy from
   */
  AliForwardFlowUtil(const AliForwardFlowUtil& o) : TNamed(),
						    fList(o.fList),
						    fCent(o.fCent),
						    fVertex(o.fVertex),
                                                    fAliceCent4th(o.fAliceCent4th),
                                                    fAlicePt2nd4050(o.fAlicePt2nd4050),
                                                    fAlicePt4th3040(o.fAlicePt4th3040),
                                                    fAlicePt4th4050(o.fAlicePt4th4050),
                                                    fImpactParToCent(o.fImpactParToCent)
                                                    {}
  /** 
   * Assignment operator 
   * 
   * @return Reference to this object 
   */
  AliForwardFlowUtil& operator=(const AliForwardFlowUtil&) { return *this; }
  /**
   * Add pt dependent flow factor
   *
   * @param Pt   @f$ p_T@f$
   * @param type Type of flow 
   */
  Double_t AddptFlow(Double_t Pt, Int_t type) const;
  /**
   * Add pid dependent flow factor
   *
   * @param ID   Particle ID 
   * @param type Type of flow
   */
  Double_t AddpidFlow(Int_t ID, Int_t type) const;
  /**
   * Add eta dependent flow factor
   * 
   * @param Eta  @f$\eta@f$ 
   * @param type Type of flow 
   */
  Double_t AddetaFlow(Double_t Eta, Int_t type) const;
  /**
   * Get centrality form MC impact parameter
   * 
   * @param AODevent AOD event structure 
   */
  Double_t GetCentFromMC(const AliAODEvent* AODevent) const;

  TList* 	fList;   // List of flow histograms
  Double_t	fCent;   // centrality
  Float_t       fVertex; // z vertex coordinate
  TGraph*       fAliceCent4th;
  TGraph*       fAlicePt2nd4050;
  TGraph*       fAlicePt4th3040;
  TGraph*       fAlicePt4th4050;
  TGraph*       fImpactParToCent;

  ClassDef(AliForwardFlowUtil, 1); 
};
 
#endif
// Local Variables:
//   mode: C++ 
// End: