[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
#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
 */
class AliFMDMCCorrector : public AliFMDCorrector
{
public:
  /** 
   * Constructor 
   */
  AliFMDMCCorrector()
    : AliFMDCorrector(),
      fFMD1i(0), 
      fFMD2i(0),
      fFMD2o(0),
      fFMD3i(0),
      fFMD3o(0),
      fComps(0)
  {}
  /** 
   * Constructor 
   * 
   * @param name Name of object
   */
  AliFMDMCCorrector(const char* name)
    : AliFMDCorrector(name),
      fFMD1i(0), 
      fFMD2i(0),
      fFMD2o(0),
      fFMD3i(0),
      fFMD3o(0),
      fComps(0)
  {}
  /** 
   * 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)
  {}
  /** 
   * Destructor 
   */
  virtual ~AliFMDMCCorrector();
  /** 
   * Assignement operator
   * 
   * @param o Object to assign from 
   * 
   * @return Reference to this object
   */
  AliFMDMCCorrector& operator=(const AliFMDMCCorrector&);
  /** 
   * 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);
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 
  
  ClassDef(AliFMDMCCorrector,1); // Calculate Nch density 
};

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