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

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

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

#ifndef AliAnalysisTaskFittingQDistribution_H
#define AliAnalysisTaskFittingQDistribution_H

#include "TString.h"
#include "AliAnalysisTask.h"

class AliESDEvent;
class AliAODEvent;
class AliCFManager;
class AliFittingQDistribution;
class AliFlowEventSimpleMaker;
class TFile;

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

class AliAnalysisTaskFittingQDistribution : public AliAnalysisTask{
 public:
  AliAnalysisTaskFittingQDistribution();
  AliAnalysisTaskFittingQDistribution(const char *name, Bool_t QAon = kFALSE, Bool_t useWeights=kFALSE);
  virtual ~AliAnalysisTaskFittingQDistribution(){}; 
  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);
  
  void           SetAnalysisType(TString type) {this->fAnalysisType = type;}
  TString GetAnalysisType() const {return this->fAnalysisType;}
  
  void SetCFManager1(AliCFManager* cfmgr) {this->fCFManager1 = cfmgr;} 
  AliCFManager* GetCFManager1()           {return this->fCFManager1;}
  void SetCFManager2(AliCFManager* cfmgr) {this->fCFManager2 = cfmgr;} 
  AliCFManager* GetCFManager2()           {return this->fCFManager2;} 
  void          SetQAList1(TList* list)   {this->fQAInt = list; }
  TList*        GetQAList1()              {return this->fQAInt; }
  void          SetQAList2(TList* list)   {this->fQADiff = list; }
  TList*        GetQAList2()              {return this->fQADiff; }
  void          SetQAOn(Bool_t kt)        {this->fQA = kt; }
  Bool_t        GetQAOn()                 {return this->fQA; }
  
  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:
  AliAnalysisTaskFittingQDistribution(const AliAnalysisTaskFittingQDistribution& aatfqd);
  AliAnalysisTaskFittingQDistribution& operator=(const AliAnalysisTaskFittingQDistribution& aatfqd);

  AliESDEvent *fESD;                      // ESD object
  AliAODEvent* fAOD;                      // AOD object
  AliFittingQDistribution* fFQDA;         // Fitting q-distribution Analysis (FQDA) object
  AliFlowEventSimpleMaker* fEventMaker;   // FlowEventSimple maker object
  TString fAnalysisType;                  // string to select which kind of input to analyse (ESD, AOD or MC)
  AliCFManager* fCFManager1;              // correction framework manager
  AliCFManager* fCFManager2;              // correction framework manager
  TList  *fListHistos;                    // collection of output 
     
  TList*        fQAInt;                   // QA histogram list
  TList*        fQADiff;                  // QA histogram list

  Bool_t fQA;                             // flag to set the filling of the QA histograms  
  
  Bool_t       fUseWeights;               // use any weights
  Bool_t       fUsePhiWeights;            // phi weights
  Bool_t       fUsePtWeights;             // v_2(pt) weights
  Bool_t       fUseEtaWeights;            // v_2(eta) weights  
  TList*       fListWeights;              // list with weights                                               
                                                           
  ClassDef(AliAnalysisTaskFittingQDistribution, 1); 
};

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

#endif
Commit	Line	Data
9bed2723	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 fitting *
	9	* q-distribution *
	10	* *
	11	* authors: Naomi van der Kolk *
	12	* (kolk@nikhef.nl) *
	13	* Raimond Snellings *
	14	* (snelling@nikhef.nl) *
	15	* Ante Bilandzic *
	16	* (anteb@nikhef.nl) *
	17	* ***********************************/
	18
	19	#ifndef AliAnalysisTaskFittingQDistribution_H
	20	#define AliAnalysisTaskFittingQDistribution_H
	21
	22	#include "TString.h"
	23	#include "AliAnalysisTask.h"
	24
	25	class AliESDEvent;
	26	class AliAODEvent;
	27	class AliCFManager;
	28	class AliFittingQDistribution;
	29	class AliFlowEventSimpleMaker;
	30	class TFile;
	31
	32	//================================================================================================================
	33
	34	class AliAnalysisTaskFittingQDistribution : public AliAnalysisTask{
	35	public:
	36	AliAnalysisTaskFittingQDistribution();
14db9c04	37	AliAnalysisTaskFittingQDistribution(const char *name, Bool_t QAon = kFALSE, Bool_t useWeights=kFALSE);
9bed2723	38	virtual ~AliAnalysisTaskFittingQDistribution(){};
	39
	40	virtual void ConnectInputData(Option_t *);
	41	virtual void CreateOutputObjects();
	42	virtual void Exec(Option_t *option);
	43	virtual void Terminate(Option_t *);
	44
	45	void SetAnalysisType(TString type) {this->fAnalysisType = type;}
	46	TString GetAnalysisType() const {return this->fAnalysisType;}
	47
	48	void SetCFManager1(AliCFManager* cfmgr) {this->fCFManager1 = cfmgr;}
	49	AliCFManager* GetCFManager1() {return this->fCFManager1;}
	50	void SetCFManager2(AliCFManager* cfmgr) {this->fCFManager2 = cfmgr;}
	51	AliCFManager* GetCFManager2() {return this->fCFManager2;}
	52	void SetQAList1(TList* list) {this->fQAInt = list; }
	53	TList* GetQAList1() {return this->fQAInt; }
	54	void SetQAList2(TList* list) {this->fQADiff = list; }
	55	TList* GetQAList2() {return this->fQADiff; }
	56	void SetQAOn(Bool_t kt) {this->fQA = kt; }
	57	Bool_t GetQAOn() {return this->fQA; }
14db9c04	58
	59	void SetUsePhiWeights(Bool_t const uPhiW) {this->fUsePhiWeights = uPhiW;};
	60	Bool_t GetUsePhiWeights() const {return this->fUsePhiWeights;};
	61	void SetUsePtWeights(Bool_t const uPtW) {this->fUsePtWeights = uPtW;};
	62	Bool_t GetUsePtWeights() const {return this->fUsePtWeights;};
	63	void SetUseEtaWeights(Bool_t const uEtaW) {this->fUseEtaWeights = uEtaW;};
	64	Bool_t GetUseEtaWeights() const {return this->fUseEtaWeights;};
9bed2723	65
	66	private:
	67	AliAnalysisTaskFittingQDistribution(const AliAnalysisTaskFittingQDistribution& aatfqd);
	68	AliAnalysisTaskFittingQDistribution& operator=(const AliAnalysisTaskFittingQDistribution& aatfqd);
	69
	70	AliESDEvent *fESD; // ESD object
	71	AliAODEvent* fAOD; // AOD object
	72	AliFittingQDistribution* fFQDA; // Fitting q-distribution Analysis (FQDA) object
	73	AliFlowEventSimpleMaker* fEventMaker; // FlowEventSimple maker object
	74	TString fAnalysisType; // string to select which kind of input to analyse (ESD, AOD or MC)
	75	AliCFManager* fCFManager1; // correction framework manager
	76	AliCFManager* fCFManager2; // correction framework manager
	77	TList *fListHistos; // collection of output
	78
	79	TList* fQAInt; // QA histogram list
	80	TList* fQADiff; // QA histogram list
	81
14db9c04	82	Bool_t fQA; // flag to set the filling of the QA histograms
	83
	84	Bool_t fUseWeights; // use any weights
	85	Bool_t fUsePhiWeights; // phi weights
	86	Bool_t fUsePtWeights; // v_2(pt) weights
	87	Bool_t fUseEtaWeights; // v_2(eta) weights
	88	TList* fListWeights; // list with weights
9bed2723	89
	90	ClassDef(AliAnalysisTaskFittingQDistribution, 1);
	91	};
	92
	93	//================================================================================================================
	94
	95	#endif
	96
	97
	98
	99
	100
	101
	102
	103
	104
	105
	106