//
// Calculate the flow in the forward regions using the Q cumulants method
//
#ifndef ALIFORWARDMCFLOWTASKQC_H
#define ALIFORWARDMCFLOWTASKQC_H
/**
 * @file   AliForwardMCFlowTaskQC.h
 * @author Alexander Hansen alexander.hansen@cern.ch
 * 
 * @brief  
 * 
 * 
 * @ingroup pwglf_forward_flow
 */
#include "AliForwardFlowTaskQC.h"
#include "AliForwardFlowWeights.h"
#include <TH2D.h>
class TGraph;

/**
 * Calculate the flow in the forward regions using the Q cumulants method
 *
 * @par Inputs:
 *   - AliAODEvent
 *
 * Outputs:
 *   - AnalysisResults.root
 *
 * @ingroup pwglf_forward_tasks_flow
 * @ingroup pwglf_forward_flow
 *
 */
class AliForwardMCFlowTaskQC : public AliForwardFlowTaskQC
{
public:
  /**
   * Constructor
   */
  AliForwardMCFlowTaskQC();
  /**
   * Constructor
   *
   * @param name Name of task
   */
  AliForwardMCFlowTaskQC(const char* name);
  /**
   * Destructor 
   */
  virtual ~AliForwardMCFlowTaskQC() {}
  /**
   * Check trigger from AODForwardMult object
   * returns true if B trigger is present
   *
   * @param aodfm AliAODForwardMultObject
   * 
   * @return Bool_t 
   */
  virtual Bool_t CheckTrigger(const AliAODForwardMult* aodfm) const;
  /**
   * Check for centrality in AliAODForwardMult object, 
   * if present return true - also sets fCent value
   * can be used to get centrality from impact parameter
   *
   * @param aodfm AliAODForwardMultObject
   * 
   * @return Bool_t 
   */
  virtual Bool_t GetCentrality(const AliAODForwardMult* aodfm);
  /**
   * Set use parametrization from impact parameter for centrality
   *
   * @param use Use impact par
   */
  void SetUseImpactParameter(Bool_t use) { fUseImpactPar = use; }
  /**
   * Set string to add flow to MC truth particles
   *
   * @param type String
   */
  void AddFlow(TString type = "") { fAddFlow = type; }
  /**
   * Set which function fAddFlow should use
   *
   * @param number Type of AddFlow 
   */
  void AddFlowType(Int_t number = 0) { fAddType = number; }
  /**
   * Set which order of flow to add
   *
   * @param order Flow order 
   */
  void AddFlowOrder(Int_t order = 2) { fAddOrder = order; }
 
protected:
  /**
   * Copy constructor
   *
   * @param o Object to copy from
   */
  AliForwardMCFlowTaskQC(const AliForwardMCFlowTaskQC& o);
  /** 
   * Assignment operator 
   * 
   * @param o Object to assing from 
   *
   * @return Reference to this object 
   */
  AliForwardMCFlowTaskQC& operator=(const AliForwardMCFlowTaskQC& o);
  /**
   * Initiate vertex bin objects
   */
  void InitVertexBins();
  /**
   * Initiate diagnostics histograms
   */
  void InitHists();
  /**
   * Analyze event
   *
   * @return true on success 
   */
  Bool_t Analyze();
  /**
   * Finalize analysis
   */
  void Finalize();
  /**
   * Find FMD coverage for this vtx, to make MC histogram match
   */
  void GetFMDLimits();
  /**
   * Loop over AliAODMCParticle branch object and fill d^2N/detadphi histograms
   * add flow if arguments are set
   * 
   * @return true on success
   */
  Bool_t LoopAODMC();
  /**
   * Get centrality form MC impact parameter
   *
   * @return Centrality
   */
  Double_t GetCentFromB() const;
  /**
   * Print the setup of the task
   *
   * @return void
   */
  virtual void PrintFlowSetup() const;
  
  TList         fBinsFMDTR;         //  List with FMDTR VertexBin objects
  TList         fBinsSPDTR;         //  List with SPDTR VertexBin objects
  TList         fBinsMC;            //  List with MC VertexBin objects
  TH2D          fdNdedpMC;          //  d^2N/detadphi MC truth histogram
  AliForwardFlowWeights fWeights;   //  Flow after burner 
  TGraph*       fImpactParToCent;   //  Parametrization of b to centrality
  Bool_t        fUseImpactPar;      //  Flag to use impact parameter for cent
  Double_t      fFMDMinEta;         //  FMD min. eta coverage for this vtx
  Double_t      fFMDMaxEta;         //  FMD max. eta coverage for this vtx
  TString       fAddFlow;           //  Add flow string
  Int_t         fAddType;           //  Add flow type #
  Int_t         fAddOrder;          //  Add flow order

  ClassDef(AliForwardMCFlowTaskQC, 3); // FMD MC analysis task 
};
 
#endif
// Local Variables:
//   mode: C++ 
// End: