[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / AliFMDMCCorrector.h

// 
// This class calculates the exclusive charged particle density
// in each for the 5 FMD rings. 
//
#ifndef ALIFMDMCCORRECTOR_H
#define ALIFMDMCCORRECTOR_H
/**
 * @file   AliFMDMCCorrector.h
 * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
 * @date   Wed Mar 23 14:03:15 2011
 * 
 * @brief  
 * 
 * 
 * @ingroup pwg2_forward_aod
 */
#include "AliFMDCorrector.h"
#include <TList.h>
class TProfile2D;
class TH2;

/** 
 * @defgroup pwg2_forward_mc  Monte-carlo code
 * 
 * @ingroup pwg2_forward
 */
/** 
 * This class calculates the exclusive charged particle density
 * in each for the 5 FMD rings. 
 *
 * @par Input:
 *   - 5 RingHistos objects - each with a number of vertex dependent 
 *     2D histograms of the inclusive charge particle density 
 *
 * @par Output:
 *   - 5 RingHistos objects - each with a number of vertex dependent 
 *     2D histograms of the exclusive charge particle density 
 * 
 * @par Corrections used: 
 *   - AliFMDCorrSecondaryMap;
 *   - AliFMDCorrVertexBias
 *   - AliFMDCorrMergingEfficiency
 *
 * @ingroup pwg2_forward_algo
 * @ingroup pwg2_forward_mc
 * @ingroup pwg2_forward_aod
 */
class AliFMDMCCorrector : public AliFMDCorrector
{
public:
  /** 
   * Constructor 
   */
  AliFMDMCCorrector()
    : AliFMDCorrector(),
      fFMD1i(0), 
      fFMD2i(0),
      fFMD2o(0),
      fFMD3i(0),
      fFMD3o(0),
      fComps(0),
      fSecondaryForMC(true)
  {}
  /** 
   * Constructor 
   * 
   * @param name Name of object
   */
  AliFMDMCCorrector(const char* name)
    : AliFMDCorrector(name),
      fFMD1i(0), 
      fFMD2i(0),
      fFMD2o(0),
      fFMD3i(0),
      fFMD3o(0),
      fComps(0),
      fSecondaryForMC(true)
  {}
  /** 
   * Copy constructor 
   * 
   * @param o Object to copy from 
   */
  AliFMDMCCorrector(const AliFMDMCCorrector& o)
    : AliFMDCorrector(o),
      fFMD1i(o.fFMD1i), 
      fFMD2i(o.fFMD2i),
      fFMD2o(o.fFMD2o),
      fFMD3i(o.fFMD3i),
      fFMD3o(o.fFMD3o),
      fComps(0),
      fSecondaryForMC(o.fSecondaryForMC)
  {}
  /** 
   * Destructor 
   */
  virtual ~AliFMDMCCorrector();
  /** 
   * Assignement operator
   * 
   * @param o Object to assign from 
   * 
   * @return Reference to this object
   */
  AliFMDMCCorrector& operator=(const AliFMDMCCorrector&);
  /** 
   * If set, then do not do the secondary correction for MC data
   * 
   * @param use 
   */
  void SetSecondaryForMC(Bool_t use) { fSecondaryForMC = use; }
  /** 
   * Initialize this object 
   * 
   * @param etaAxis Eta axis to use 
   */
  void Init(const TAxis& etaAxis);
  /** 
   * Do the calculations 
   * 
   * @param hists    Cache of histograms 
   * @param vtxBin   Vertex bin 
   * 
   * @return true on successs 
   */
  virtual Bool_t CorrectMC(AliForwardUtil::Histos& hists, UShort_t vtxBin);
  /** 
   * Compare the result of analysing the ESD for 
   * the inclusive charged particle density to analysing 
   * MC truth 
   * 
   * @param esd 
   * @param mc 
   * 
   * @return 
   */
  virtual Bool_t CompareResults(AliForwardUtil::Histos& esd, 
				AliForwardUtil::Histos& mc);
  /** 
   * Output diagnostic histograms to directory 
   * 
   * @param dir List to write in
   */  
  void DefineOutput(TList* dir);

  /**
   * Print information
   * 
   * @param option Not used 
   */
  void Print(Option_t* option="") const;
protected:
  /** 
   * MAke comparison profiles
   * 
   * @param d     Detector 
   * @param r     Ring 
   * @param axis  Eta axis 
   * 
   * @return Newly allocated profile object
   */
  TProfile2D* Make(UShort_t d, Char_t r, const TAxis& axis) const;
  /** 
   * Fill comparison profiles
   * 
   * @param d    Detector 
   * @param r    Ring 
   * @param esd  ESD histogram
   * @param mc   MC histogram
   */
  void Fill(UShort_t d, Char_t r, TH2* esd, TH2* mc);

  TProfile2D* fFMD1i; // Comparison
  TProfile2D* fFMD2i; // Comparison
  TProfile2D* fFMD2o; // Comparison
  TProfile2D* fFMD3i; // Comparison
  TProfile2D* fFMD3o; // Comparison
  TList*      fComps; // List of comparisons 
  Bool_t      fSecondaryForMC;  // Whether to correct MC data 

  ClassDef(AliFMDMCCorrector,1); // Calculate Nch density 
};

#endif
// Local Variables:
//   mode: C++
// End:
Commit	Line	Data
	1	//
	2	// This class calculates the exclusive charged particle density
	3	// in each for the 5 FMD rings.
	4	//
	5	#ifndef ALIFMDMCCORRECTOR_H
	6	#define ALIFMDMCCORRECTOR_H
	7	/**
	8	* @file AliFMDMCCorrector.h
	9	* @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
	10	* @date Wed Mar 23 14:03:15 2011
	11	*
	12	* @brief
	13	*
	14	*
	15	* @ingroup pwg2_forward_aod
	16	*/
	17	#include "AliFMDCorrector.h"
	18	#include <TList.h>
	19	class TProfile2D;
	20	class TH2;
	21
	22	/**
	23	* @defgroup pwg2_forward_mc Monte-carlo code
	24	*
	25	* @ingroup pwg2_forward
	26	*/
	27	/**
	28	* This class calculates the exclusive charged particle density
	29	* in each for the 5 FMD rings.
	30	*
	31	* @par Input:
	32	* - 5 RingHistos objects - each with a number of vertex dependent
	33	* 2D histograms of the inclusive charge particle density
	34	*
	35	* @par Output:
	36	* - 5 RingHistos objects - each with a number of vertex dependent
	37	* 2D histograms of the exclusive charge particle density
	38	*
	39	* @par Corrections used:
	40	* - AliFMDCorrSecondaryMap;
	41	* - AliFMDCorrVertexBias
	42	* - AliFMDCorrMergingEfficiency
	43	*
	44	* @ingroup pwg2_forward_algo
	45	* @ingroup pwg2_forward_mc
	46	* @ingroup pwg2_forward_aod
	47	*/
	48	class AliFMDMCCorrector : public AliFMDCorrector
	49	{
	50	public:
	51	/**
	52	* Constructor
	53	*/
	54	AliFMDMCCorrector()
	55	: AliFMDCorrector(),
	56	fFMD1i(0),
	57	fFMD2i(0),
	58	fFMD2o(0),
	59	fFMD3i(0),
	60	fFMD3o(0),
	61	fComps(0),
	62	fSecondaryForMC(true)
	63	{}
	64	/**
	65	* Constructor
	66	*
	67	* @param name Name of object
	68	*/
	69	AliFMDMCCorrector(const char* name)
	70	: AliFMDCorrector(name),
	71	fFMD1i(0),
	72	fFMD2i(0),
	73	fFMD2o(0),
	74	fFMD3i(0),
	75	fFMD3o(0),
	76	fComps(0),
	77	fSecondaryForMC(true)
	78	{}
	79	/**
	80	* Copy constructor
	81	*
	82	* @param o Object to copy from
	83	*/
	84	AliFMDMCCorrector(const AliFMDMCCorrector& o)
	85	: AliFMDCorrector(o),
	86	fFMD1i(o.fFMD1i),
	87	fFMD2i(o.fFMD2i),
	88	fFMD2o(o.fFMD2o),
	89	fFMD3i(o.fFMD3i),
	90	fFMD3o(o.fFMD3o),
	91	fComps(0),
	92	fSecondaryForMC(o.fSecondaryForMC)
	93	{}
	94	/**
	95	* Destructor
	96	*/
	97	virtual ~AliFMDMCCorrector();
	98	/**
	99	* Assignement operator
	100	*
	101	* @param o Object to assign from
	102	*
	103	* @return Reference to this object
	104	*/
	105	AliFMDMCCorrector& operator=(const AliFMDMCCorrector&);
	106	/**
	107	* If set, then do not do the secondary correction for MC data
	108	*
	109	* @param use
	110	*/
	111	void SetSecondaryForMC(Bool_t use) { fSecondaryForMC = use; }
	112	/**
	113	* Initialize this object
	114	*
	115	* @param etaAxis Eta axis to use
	116	*/
	117	void Init(const TAxis& etaAxis);
	118	/**
	119	* Do the calculations
	120	*
	121	* @param hists Cache of histograms
	122	* @param vtxBin Vertex bin
	123	*
	124	* @return true on successs
	125	*/
	126	virtual Bool_t CorrectMC(AliForwardUtil::Histos& hists, UShort_t vtxBin);
	127	/**
	128	* Compare the result of analysing the ESD for
	129	* the inclusive charged particle density to analysing
	130	* MC truth
	131	*
	132	* @param esd
	133	* @param mc
	134	*
	135	* @return
	136	*/
	137	virtual Bool_t CompareResults(AliForwardUtil::Histos& esd,
	138	AliForwardUtil::Histos& mc);
	139	/**
	140	* Output diagnostic histograms to directory
	141	*
	142	* @param dir List to write in
	143	*/
	144	void DefineOutput(TList* dir);
	145
	146	/**
	147	* Print information
	148	*
	149	* @param option Not used
	150	*/
	151	void Print(Option_t* option="") const;
	152	protected:
	153	/**
	154	* MAke comparison profiles
	155	*
	156	* @param d Detector
	157	* @param r Ring
	158	* @param axis Eta axis
	159	*
	160	* @return Newly allocated profile object
	161	*/
	162	TProfile2D* Make(UShort_t d, Char_t r, const TAxis& axis) const;
	163	/**
	164	* Fill comparison profiles
	165	*
	166	* @param d Detector
	167	* @param r Ring
	168	* @param esd ESD histogram
	169	* @param mc MC histogram
	170	*/
	171	void Fill(UShort_t d, Char_t r, TH2* esd, TH2* mc);
	172
	173	TProfile2D* fFMD1i; // Comparison
	174	TProfile2D* fFMD2i; // Comparison
	175	TProfile2D* fFMD2o; // Comparison
	176	TProfile2D* fFMD3i; // Comparison
	177	TProfile2D* fFMD3o; // Comparison
	178	TList* fComps; // List of comparisons
	179	Bool_t fSecondaryForMC; // Whether to correct MC data
	180
	181	ClassDef(AliFMDMCCorrector,1); // Calculate Nch density
	182	};
	183
	184	#endif
	185	// Local Variables:
	186	// mode: C++
	187	// End:
	188