[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowTasks / AliAnalysisTaskMixedHarmonics.h

/* 
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. 
 * See cxx source for full Copyright notice 
 * $Id$ 
 */

/**************************************
 * analysis task for mixed harmomics  * 
 *                                    * 
 * authors: Naomi van der Kolk        *
 *           (kolk@nikhef.nl)         *  
 *          Raimond Snellings         *
 *           (snelling@nikhef.nl)     * 
 *          Ante Bilandzic            *
 *           (anteb@nikhef.nl)        * 
 * ***********************************/

#ifndef ALIANALYSISTASKMIXEDHARMONICS_H
#define ALIANALYSISTASKMIXEDHARMONICS_H

#include "TString.h"
#include "AliAnalysisTaskSE.h"

class TList;
class AliFlowEventSimple;
class AliFlowAnalysisWithMixedHarmonics;

//================================================================================================================

class AliAnalysisTaskMixedHarmonics : public AliAnalysisTaskSE{
 public:
  AliAnalysisTaskMixedHarmonics();
  AliAnalysisTaskMixedHarmonics(const char *name, Bool_t useParticleWeights=kFALSE);
  virtual ~AliAnalysisTaskMixedHarmonics(){}; 
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  
  // common:
  void SetHarmonic(Int_t const h) {this->fHarmonic = h;};
  Int_t GetHarmonic() const {return this->fHarmonic;}; 
  void SetNoOfMultipicityBins(Int_t const nomb) {this->fNoOfMultipicityBins = nomb;};
  Int_t GetNoOfMultipicityBins() const {return this->fNoOfMultipicityBins;};   
  void SetMultipicityBinWidth(Double_t const mbw) {this->fMultipicityBinWidth = mbw;};
  Double_t GetMultipicityBinWidth() const {return this->fMultipicityBinWidth;};   
  void SetMinMultiplicity(Double_t const mm) {this->fMinMultiplicity = mm;};
  Double_t GetMinMultiplicity() const {return this->fMinMultiplicity;}; 
  void SetOppositeChargesPOI(Bool_t const ocp) {this->fOppositeChargesPOI = ocp;};
  Bool_t GetOppositeChargesPOI() const {return this->fOppositeChargesPOI;}; 
  void SetEvaluateDifferential3pCorrelator(Bool_t const ed3pc) {this->fEvaluateDifferential3pCorrelator = ed3pc;};
  Bool_t GetEvaluateDifferential3pCorrelator() const {return this->fEvaluateDifferential3pCorrelator;};       
  void SetCorrectForDetectorEffects(Bool_t const cfde) {this->fCorrectForDetectorEffects = cfde;};
  Bool_t GetCorrectForDetectorEffects() const {return this->fCorrectForDetectorEffects;}; 
  void SetPrintOnTheScreen(Bool_t const pots) {this->fPrintOnTheScreen = pots;};
  Bool_t GetPrintOnTheScreen() const {return this->fPrintOnTheScreen;};  
  void SetCalculateVsM(Bool_t const cvm) {this->fCalculateVsM = cvm;};
  Bool_t GetCalculateVsM() const {return this->fCalculateVsM;};  
  void SetShowBinLabelsVsM(Bool_t const sblvm) {this->fShowBinLabelsVsM = sblvm;};
  Bool_t GetShowBinLabelsVsM() const {return this->fShowBinLabelsVsM;};     
  // particle weights:
  void SetUsePhiWeights(Bool_t const uPhiW) {this->fUsePhiWeights = uPhiW;};
  Bool_t GetUsePhiWeights() const {return this->fUsePhiWeights;};
  void SetUsePtWeights(Bool_t const uPtW) {this->fUsePtWeights = uPtW;};
  Bool_t GetUsePtWeights() const {return this->fUsePtWeights;};
  void SetUseEtaWeights(Bool_t const uEtaW) {this->fUseEtaWeights = uEtaW;};
  Bool_t GetUseEtaWeights() const {return this->fUseEtaWeights;};
 
 private:
  AliAnalysisTaskMixedHarmonics(const AliAnalysisTaskMixedHarmonics& aatmh);
  AliAnalysisTaskMixedHarmonics& operator=(const AliAnalysisTaskMixedHarmonics& aatmh);
  
  AliFlowEventSimple *fEvent; // the input event
  AliFlowAnalysisWithMixedHarmonics *fMH; // mixed harmonics object
  TList *fListHistos; // collection of output 
  // common:  
  Int_t fHarmonic; // integer n in cos[n(phi1+phi2-2phi3)]
  Int_t fNoOfMultipicityBins; // number of multiplicity bins
  Double_t fMultipicityBinWidth; // width of multiplicity bin
  Double_t fMinMultiplicity; // minimal multiplicity
  Bool_t fOppositeChargesPOI; // two POIs, psi1 and psi2, in correlator <<cos[psi1+psi2-2phi3)]>> will be taken with opposite charges 
  Bool_t fEvaluateDifferential3pCorrelator; // evaluate <<cos[psi1+psi2-2phi3)]>>, where psi1 and psi2 are two POIs      
  Bool_t fCorrectForDetectorEffects; // correct 3-p correlator for detector effects  
  Bool_t fPrintOnTheScreen; // print or not the final results on the screen
  Bool_t fCalculateVsM; // calculate correlators vs multiplicity
  Bool_t fShowBinLabelsVsM; // in histograms holding results vs multiplicity show bin labels in the format M_lowEdge \leq M < M_upperEdge  
  // particle weights:
  Bool_t fUseParticleWeights; // use any particle weights
  Bool_t fUsePhiWeights; // use phi weights
  Bool_t fUsePtWeights; // use pt weights
  Bool_t fUseEtaWeights; // use eta weights  
  TList *fWeightsList; // list with weights
  
  ClassDef(AliAnalysisTaskMixedHarmonics, 1); 
};

//================================================================================================================

#endif
Commit	Line	Data
67a6b178	1	/*
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved.
	3	* See cxx source for full Copyright notice
	4	* $Id$
	5	*/
	6
	7	/**************************************
	8	* analysis task for mixed harmomics *
	9	* *
	10	* authors: Naomi van der Kolk *
	11	* (kolk@nikhef.nl) *
	12	* Raimond Snellings *
	13	* (snelling@nikhef.nl) *
	14	* Ante Bilandzic *
	15	* (anteb@nikhef.nl) *
	16	* ***********************************/
	17
	18	#ifndef ALIANALYSISTASKMIXEDHARMONICS_H
	19	#define ALIANALYSISTASKMIXEDHARMONICS_H
	20
	21	#include "TString.h"
	22	#include "AliAnalysisTaskSE.h"
	23
	24	class TList;
	25	class AliFlowEventSimple;
	26	class AliFlowAnalysisWithMixedHarmonics;
	27
	28	//================================================================================================================
	29
	30	class AliAnalysisTaskMixedHarmonics : public AliAnalysisTaskSE{
	31	public:
	32	AliAnalysisTaskMixedHarmonics();
	33	AliAnalysisTaskMixedHarmonics(const char *name, Bool_t useParticleWeights=kFALSE);
	34	virtual ~AliAnalysisTaskMixedHarmonics(){};
	35
	36	virtual void UserCreateOutputObjects();
	37	virtual void UserExec(Option_t *option);
	38	virtual void Terminate(Option_t *);
	39
	40	// common:
e999459d	41	void SetHarmonic(Int_t const h) {this->fHarmonic = h;};
e999459d	42	Int_t GetHarmonic() const {return this->fHarmonic;};
67a6b178	43	void SetNoOfMultipicityBins(Int_t const nomb) {this->fNoOfMultipicityBins = nomb;};
	44	Int_t GetNoOfMultipicityBins() const {return this->fNoOfMultipicityBins;};
	45	void SetMultipicityBinWidth(Double_t const mbw) {this->fMultipicityBinWidth = mbw;};
	46	Double_t GetMultipicityBinWidth() const {return this->fMultipicityBinWidth;};
	47	void SetMinMultiplicity(Double_t const mm) {this->fMinMultiplicity = mm;};
	48	Double_t GetMinMultiplicity() const {return this->fMinMultiplicity;};
e999459d	49	void SetOppositeChargesPOI(Bool_t const ocp) {this->fOppositeChargesPOI = ocp;};
	50	Bool_t GetOppositeChargesPOI() const {return this->fOppositeChargesPOI;};
	51	void SetEvaluateDifferential3pCorrelator(Bool_t const ed3pc) {this->fEvaluateDifferential3pCorrelator = ed3pc;};
	52	Bool_t GetEvaluateDifferential3pCorrelator() const {return this->fEvaluateDifferential3pCorrelator;};
67a6b178	53	void SetCorrectForDetectorEffects(Bool_t const cfde) {this->fCorrectForDetectorEffects = cfde;};
67a6b178	54	Bool_t GetCorrectForDetectorEffects() const {return this->fCorrectForDetectorEffects;};
b71a354b	55	void SetPrintOnTheScreen(Bool_t const pots) {this->fPrintOnTheScreen = pots;};
	56	Bool_t GetPrintOnTheScreen() const {return this->fPrintOnTheScreen;};
	57	void SetCalculateVsM(Bool_t const cvm) {this->fCalculateVsM = cvm;};
	58	Bool_t GetCalculateVsM() const {return this->fCalculateVsM;};
	59	void SetShowBinLabelsVsM(Bool_t const sblvm) {this->fShowBinLabelsVsM = sblvm;};
	60	Bool_t GetShowBinLabelsVsM() const {return this->fShowBinLabelsVsM;};
67a6b178	61	// particle weights:
	62	void SetUsePhiWeights(Bool_t const uPhiW) {this->fUsePhiWeights = uPhiW;};
	63	Bool_t GetUsePhiWeights() const {return this->fUsePhiWeights;};
	64	void SetUsePtWeights(Bool_t const uPtW) {this->fUsePtWeights = uPtW;};
	65	Bool_t GetUsePtWeights() const {return this->fUsePtWeights;};
	66	void SetUseEtaWeights(Bool_t const uEtaW) {this->fUseEtaWeights = uEtaW;};
	67	Bool_t GetUseEtaWeights() const {return this->fUseEtaWeights;};
	68
	69	private:
	70	AliAnalysisTaskMixedHarmonics(const AliAnalysisTaskMixedHarmonics& aatmh);
	71	AliAnalysisTaskMixedHarmonics& operator=(const AliAnalysisTaskMixedHarmonics& aatmh);
	72
	73	AliFlowEventSimple *fEvent; // the input event
	74	AliFlowAnalysisWithMixedHarmonics *fMH; // mixed harmonics object
	75	TList *fListHistos; // collection of output
	76	// common:
e999459d	77	Int_t fHarmonic; // integer n in cos[n(phi1+phi2-2phi3)]
67a6b178	78	Int_t fNoOfMultipicityBins; // number of multiplicity bins
	79	Double_t fMultipicityBinWidth; // width of multiplicity bin
	80	Double_t fMinMultiplicity; // minimal multiplicity
e999459d	81	Bool_t fOppositeChargesPOI; // two POIs, psi1 and psi2, in correlator <<cos[psi1+psi2-2phi3)]>> will be taken with opposite charges
e999459d	82	Bool_t fEvaluateDifferential3pCorrelator; // evaluate <<cos[psi1+psi2-2phi3)]>>, where psi1 and psi2 are two POIs
b71a354b	83	Bool_t fCorrectForDetectorEffects; // correct 3-p correlator for detector effects
	84	Bool_t fPrintOnTheScreen; // print or not the final results on the screen
	85	Bool_t fCalculateVsM; // calculate correlators vs multiplicity
	86	Bool_t fShowBinLabelsVsM; // in histograms holding results vs multiplicity show bin labels in the format M_lowEdge \leq M < M_upperEdge
67a6b178	87	// particle weights:
	88	Bool_t fUseParticleWeights; // use any particle weights
	89	Bool_t fUsePhiWeights; // use phi weights
	90	Bool_t fUsePtWeights; // use pt weights
	91	Bool_t fUseEtaWeights; // use eta weights
	92	TList *fWeightsList; // list with weights
	93
5c078806	94	ClassDef(AliAnalysisTaskMixedHarmonics, 1);
67a6b178	95	};
	96
	97	//================================================================================================================
	98
	99	#endif
	100
	101
	102
	103
	104
	105
	106
	107
	108
	109
	110