[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemto / AliFemtoModelCorrFctn.h

////////////////////////////////////////////////////////////////////////////////
///                                                                          ///
/// AliFemtoModelCorrFctn - the base class for correlation function which    ///
/// uses the model framework and weight generation                           ///
///                                                                          ///
////////////////////////////////////////////////////////////////////////////////
#ifndef ALIFEMTOMODELCORRFCTN_H
#define ALIFEMTOMODELCORRFCTN_H

#include "AliFemtoCorrFctn.h"
class AliFemtoPair;
class AliFemtoModelManager;
class TH1D;
class TH2D;

class AliFemtoModelCorrFctn: public AliFemtoCorrFctn {

public:
  AliFemtoModelCorrFctn();
  AliFemtoModelCorrFctn(const char *title, Int_t aNbins, Double_t aQinvLo, Double_t aQinvHi);
  AliFemtoModelCorrFctn(const AliFemtoModelCorrFctn& aCorrFctn);
  virtual ~AliFemtoModelCorrFctn();

  AliFemtoModelCorrFctn& operator=(const AliFemtoModelCorrFctn& aCorrFctn);

  virtual void ConnectToManager(AliFemtoModelManager *aManager);

  virtual AliFemtoString Report();

  virtual void AddRealPair(AliFemtoPair* aPair);
  virtual void AddMixedPair(AliFemtoPair* aPair);

  virtual void EventBegin(const AliFemtoEvent* aEvent);
  virtual void EventEnd(const AliFemtoEvent* aEvent);
  virtual void Finish();

  virtual TList* GetOutputList();
  virtual void Write();

  virtual AliFemtoModelCorrFctn* Clone();

  Double_t GetQinvTrue(AliFemtoPair*);

protected:
  AliFemtoModelManager *fManager; // Link back to the manager to get the weights

  TH1D *fNumeratorTrue; // Numerator made with pairs from the same event
  TH1D *fNumeratorFake; // Numerator made with pairs from different events (mixed pairs)
  TH1D *fDenominator;   // Denominator made with mixed pairs

  TH1D *fNumeratorTrueIdeal; // Numerator made with pairs (true qinv) from the same event
  TH1D *fNumeratorFakeIdeal; // Numerator made with pairs (true qinv) from different events (mixed pairs)
  TH1D *fDenominatorIdeal;   // Denominator made with mixed pairs (true qinv)

  TH2D *fQgenQrec; // Qinv true (generated) vs. Qinv reconstructed


private:

#ifdef __ROOT__
  ClassDef(AliFemtoModelCorrFctn, 1)
#endif
};

#endif
Commit	Line	Data
76ce4b5b	1	////////////////////////////////////////////////////////////////////////////////
	2	/// ///
	3	/// AliFemtoModelCorrFctn - the base class for correlation function which ///
	4	/// uses the model framework and weight generation ///
	5	/// ///
	6	////////////////////////////////////////////////////////////////////////////////
	7	#ifndef ALIFEMTOMODELCORRFCTN_H
	8	#define ALIFEMTOMODELCORRFCTN_H
	9
	10	#include "AliFemtoCorrFctn.h"
	11	class AliFemtoPair;
	12	class AliFemtoModelManager;
	13	class TH1D;
f02fec5b	14	class TH2D;
76ce4b5b	15
	16	class AliFemtoModelCorrFctn: public AliFemtoCorrFctn {
	17
	18	public:
	19	AliFemtoModelCorrFctn();
	20	AliFemtoModelCorrFctn(const char *title, Int_t aNbins, Double_t aQinvLo, Double_t aQinvHi);
	21	AliFemtoModelCorrFctn(const AliFemtoModelCorrFctn& aCorrFctn);
	22	virtual ~AliFemtoModelCorrFctn();
	23
	24	AliFemtoModelCorrFctn& operator=(const AliFemtoModelCorrFctn& aCorrFctn);
	25
	26	virtual void ConnectToManager(AliFemtoModelManager *aManager);
	27
	28	virtual AliFemtoString Report();
	29
	30	virtual void AddRealPair(AliFemtoPair* aPair);
f02fec5b	31	virtual void AddMixedPair(AliFemtoPair* aPair);
76ce4b5b	32
	33	virtual void EventBegin(const AliFemtoEvent* aEvent);
	34	virtual void EventEnd(const AliFemtoEvent* aEvent);
	35	virtual void Finish();
	36
	37	virtual TList* GetOutputList();
	38	virtual void Write();
	39
	40	virtual AliFemtoModelCorrFctn* Clone();
	41
f02fec5b	42	Double_t GetQinvTrue(AliFemtoPair*);
f02fec5b	43
76ce4b5b	44	protected:
76ce4b5b	45	AliFemtoModelManager *fManager; // Link back to the manager to get the weights
f02fec5b	46
76ce4b5b	47	TH1D *fNumeratorTrue; // Numerator made with pairs from the same event
	48	TH1D *fNumeratorFake; // Numerator made with pairs from different events (mixed pairs)
	49	TH1D *fDenominator; // Denominator made with mixed pairs
	50
f02fec5b	51	TH1D *fNumeratorTrueIdeal; // Numerator made with pairs (true qinv) from the same event
	52	TH1D *fNumeratorFakeIdeal; // Numerator made with pairs (true qinv) from different events (mixed pairs)
	53	TH1D *fDenominatorIdeal; // Denominator made with mixed pairs (true qinv)
	54
	55	TH2D *fQgenQrec; // Qinv true (generated) vs. Qinv reconstructed
	56
	57
76ce4b5b	58	private:
	59
	60	#ifdef __ROOT__
	61	ClassDef(AliFemtoModelCorrFctn, 1)
	62	#endif
	63	};
	64
	65	#endif