[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemtoUser / AliFemtoModelCorrFctnSource.h

////////////////////////////////////////////////////////////////////////////////
///                                                                          ///
/// AliFemtoModelCorrFctnSource - the class for correlation function which   ///
/// uses the model framework and weight generation and saves the generated   ///
/// emission source                                                          ///
/// Authors: Adam Kisiel, kisiel@mps.ohio-state.edu                          ///
///                                                                          ///
////////////////////////////////////////////////////////////////////////////////
#ifndef ALIFEMTOMODELCORRFCTNSOURCE_H
#define ALIFEMTOMODELCORRFCTNSOURCE_H

#include "TH2D.h"
#include "AliFemtoCorrFctn.h"
#include "AliFemtoPair.h"
#include "AliFemtoModelManager.h"
#include "AliFemtoModelCorrFctn.h"

class AliFemtoModelCorrFctnSource: public AliFemtoModelCorrFctn {

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

  virtual AliFemtoString Report();

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

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

  virtual AliFemtoModelCorrFctn* Clone();

  void SetUseRPSelection(unsigned short aRPSel);
protected:

  TH1D *fHistROut;     // Distribution of Rout
  TH1D *fHistRSide;    // Distribution of Rside
  TH1D *fHistRLong;    // Distribution of Rlong
  TH1D *fHistRStar;    // Distribution of RStar
  TH1D *fHistdNdR;     // Distribution of RStar weighted by Jacobian 
  TH2D *fHistNumWS;    // Weight spread for numerator
  TH2D *fHistDenWS;    // Weight spread for denominator

private:

  unsigned short fUseRPSelection;  // The pair cut uses RP selection

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

#endif
Commit	Line	Data
	1	////////////////////////////////////////////////////////////////////////////////
	2	/// ///
	3	/// AliFemtoModelCorrFctnSource - the class for correlation function which ///
	4	/// uses the model framework and weight generation and saves the generated ///
	5	/// emission source ///
	6	/// Authors: Adam Kisiel, kisiel@mps.ohio-state.edu ///
	7	/// ///
	8	////////////////////////////////////////////////////////////////////////////////
	9	#ifndef ALIFEMTOMODELCORRFCTNSOURCE_H
	10	#define ALIFEMTOMODELCORRFCTNSOURCE_H
	11
	12	#include "TH2D.h"
	13	#include "AliFemtoCorrFctn.h"
	14	#include "AliFemtoPair.h"
	15	#include "AliFemtoModelManager.h"
	16	#include "AliFemtoModelCorrFctn.h"
	17
	18	class AliFemtoModelCorrFctnSource: public AliFemtoModelCorrFctn {
	19
	20	public:
	21	AliFemtoModelCorrFctnSource();
	22	AliFemtoModelCorrFctnSource(const char *title, Int_t aNbins, Double_t aQinvLo, Double_t aQinvHi);
	23	AliFemtoModelCorrFctnSource(const AliFemtoModelCorrFctnSource& aCorrFctn);
	24	virtual ~AliFemtoModelCorrFctnSource();
	25
	26	AliFemtoModelCorrFctnSource& operator=(const AliFemtoModelCorrFctnSource& aCorrFctn);
	27
	28	virtual AliFemtoString Report();
	29
	30	virtual void AddRealPair(AliFemtoPair* aPair);
	31	virtual void AddMixedPair(AliFemtoPair* aPir);
	32
	33	virtual void Write();
	34	virtual TList* GetOutputList();
	35
	36	virtual AliFemtoModelCorrFctn* Clone();
	37
	38	void SetUseRPSelection(unsigned short aRPSel);
	39	protected:
	40
	41	TH1D *fHistROut; // Distribution of Rout
	42	TH1D *fHistRSide; // Distribution of Rside
	43	TH1D *fHistRLong; // Distribution of Rlong
	44	TH1D *fHistRStar; // Distribution of RStar
	45	TH1D *fHistdNdR; // Distribution of RStar weighted by Jacobian
	46	TH2D *fHistNumWS; // Weight spread for numerator
	47	TH2D *fHistDenWS; // Weight spread for denominator
	48
	49	private:
	50
	51	unsigned short fUseRPSelection; // The pair cut uses RP selection
	52
	53	#ifdef __ROOT__
	54	ClassDef(AliFemtoModelCorrFctnSource, 1)
	55	#endif
	56	};
	57
	58	#endif