[u/mrichter/AliRoot.git] / PWG / DevNanoAOD / AliAnalysisTaskNanoAODFilter.h

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

/* $Id$ */
/* AliAnalysisTaskNanoAODFilter.h
 *
 * Template task producing a P_t spectrum and pseudorapidity distribution.
 * Includes explanations of physics and primary track selections
 *
 * Based on tutorial example from offline pages
 * Edited by Arvinder Palaha
 */
#ifndef ALIANALYSISTASKNANOAODESEFILTER_H
#define ALIANALYSISTASKNANOAODESEFILTER_H

class TH1F;
class TList;
class AliESDtrackCuts;
class AliAnalysisCuts;
class AliNanoAODReplicator;
class AliNanoAODCustomSetter;

#ifndef ALIANALYSISTASKSE_H
#include "AliAnalysisTaskSE.h"
#endif

class AliAnalysisTaskNanoAODFilter : public AliAnalysisTaskSE {
public:
  AliAnalysisTaskNanoAODFilter();
  AliAnalysisTaskNanoAODFilter(const char *name, Bool_t saveCutsFlag);
  virtual ~AliAnalysisTaskNanoAODFilter();
  
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *option);
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void Terminate(Option_t *);
  virtual void FinishTaskOutput() ;

  Int_t GetMCMode() { return fMCMode; }
  void  SetMCMode (Int_t var) { fMCMode = var;}
  void AddFilteredAOD(const char* aodfilename, const char* title);

  AliAnalysisCuts *           GetEvtCuts() { return fEvtCuts; }
  AliAnalysisCuts *           GetTrkCuts() { return fTrkCuts; }
  AliNanoAODCustomSetter *    GetSetter()  { return fSetter; }
  TString                     GetVarList() { return fVarList; }
  TString                     GetVarListHead() { return fVarListHead; }
  Bool_t                      GetSaveCutsFlag() { return fSaveCutsFlag; }

  void  SetEvtCuts     (AliAnalysisCuts * var           ) { fEvtCuts = var;}
  void  SetTrkCuts     (AliAnalysisCuts * var           ) { fTrkCuts = var;}
  void  SetSetter      (AliNanoAODCustomSetter * var    ) { fSetter = var;}
  void  SetVarList     (TString var                     ) { fVarList = var;}
  void  SetVarListHead (TString var                     ) { fVarListHead = var;}
    
private:
  Int_t fMCMode; // true if processing monte carlo. if > 1 not all MC particles are filtered
  AliNanoAODReplicator* fTrkrep       ; // ! replicator
  TString                  fVarList      ; // List of variables to be added to the special track
  TString fVarListHead; // List of variables to be added to the special header
  AliAnalysisCuts * fEvtCuts; // Event cuts
  AliAnalysisCuts * fTrkCuts; // Track cuts

  AliNanoAODCustomSetter * fSetter; // setter for custom variables
  
  Bool_t fSaveCutsFlag; // If true, the event and track cuts are saved to disk. Can only be set in the constructor.

  
  AliAnalysisTaskNanoAODFilter(const AliAnalysisTaskNanoAODFilter&); // not implemented
  AliAnalysisTaskNanoAODFilter& operator=(const AliAnalysisTaskNanoAODFilter&); // not implemented
    
  ClassDef(AliAnalysisTaskNanoAODFilter, 1); // example of analysis
};

#endif
Commit	Line	Data
778636d4	1	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	2	* See cxx source for full Copyright notice */
	3
	4	/* $Id$ */
	5	/* AliAnalysisTaskNanoAODFilter.h
	6	*
	7	* Template task producing a P_t spectrum and pseudorapidity distribution.
	8	* Includes explanations of physics and primary track selections
	9	*
	10	* Based on tutorial example from offline pages
	11	* Edited by Arvinder Palaha
	12	*/
	13	#ifndef ALIANALYSISTASKNANOAODESEFILTER_H
	14	#define ALIANALYSISTASKNANOAODESEFILTER_H
	15
	16	class TH1F;
	17	class TList;
	18	class AliESDtrackCuts;
	19	class AliAnalysisCuts;
	20	class AliNanoAODReplicator;
	21	class AliNanoAODCustomSetter;
	22
	23	#ifndef ALIANALYSISTASKSE_H
	24	#include "AliAnalysisTaskSE.h"
	25	#endif
	26
	27	class AliAnalysisTaskNanoAODFilter : public AliAnalysisTaskSE {
	28	public:
	29	AliAnalysisTaskNanoAODFilter();
	30	AliAnalysisTaskNanoAODFilter(const char *name, Bool_t saveCutsFlag);
	31	virtual ~AliAnalysisTaskNanoAODFilter();
	32
	33	virtual void UserCreateOutputObjects();
	34	virtual void UserExec(Option_t *option);
	35	virtual void Init();
	36	virtual void LocalInit() {Init();}
	37	virtual void Terminate(Option_t *);
	38	virtual void FinishTaskOutput() ;
	39
	40	Int_t GetMCMode() { return fMCMode; }
	41	void SetMCMode (Int_t var) { fMCMode = var;}
	42	void AddFilteredAOD(const char* aodfilename, const char* title);
	43
	44	AliAnalysisCuts * GetEvtCuts() { return fEvtCuts; }
	45	AliAnalysisCuts * GetTrkCuts() { return fTrkCuts; }
	46	AliNanoAODCustomSetter * GetSetter() { return fSetter; }
	47	TString GetVarList() { return fVarList; }
	48	TString GetVarListHead() { return fVarListHead; }
	49	Bool_t GetSaveCutsFlag() { return fSaveCutsFlag; }
	50
	51	void SetEvtCuts (AliAnalysisCuts * var ) { fEvtCuts = var;}
	52	void SetTrkCuts (AliAnalysisCuts * var ) { fTrkCuts = var;}
	53	void SetSetter (AliNanoAODCustomSetter * var ) { fSetter = var;}
	54	void SetVarList (TString var ) { fVarList = var;}
	55	void SetVarListHead (TString var ) { fVarListHead = var;}
	56
	57	private:
	58	Int_t fMCMode; // true if processing monte carlo. if > 1 not all MC particles are filtered
	59	AliNanoAODReplicator* fTrkrep ; // ! replicator
	60	TString fVarList ; // List of variables to be added to the special track
	61	TString fVarListHead; // List of variables to be added to the special header
	62	AliAnalysisCuts * fEvtCuts; // Event cuts
	63	AliAnalysisCuts * fTrkCuts; // Track cuts
	64
65	AliNanoAODCustomSetter * fSetter; // setter for custom variables
66
67	Bool_t fSaveCutsFlag; // If true, the event and track cuts are saved to disk. Can only be set in the constructor.
68
69
70	AliAnalysisTaskNanoAODFilter(const AliAnalysisTaskNanoAODFilter&); // not implemented
71	AliAnalysisTaskNanoAODFilter& operator=(const AliAnalysisTaskNanoAODFilter&); // not implemented
72
73	ClassDef(AliAnalysisTaskNanoAODFilter, 1); // example of analysis
74	};
75
76	#endif
77