[u/mrichter/AliRoot.git] / PWGCF / FLOW / Documentation / examples / manual / ttree / objects / AliAnalysisTaskTTreeFilter.h

#ifndef ALIANALYSISTTREEFILTER_H
#define ALIANALYSISTTREEFILTER_H

// AliRoot includes
#include "AliAnalysisTaskSE.h"

// forward declarations
class AliVEvent;
class AliVTrack;
class TTree;
class TClonesArray;
class AliFlowTTreeEvent;

class AliAnalysisTaskTTreeFilter : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskTTreeFilter();
  AliAnalysisTaskTTreeFilter(const char *name);

  virtual ~AliAnalysisTaskTTreeFilter();

  virtual void  UserCreateOutputObjects();
  virtual void  UserExec(Option_t *option);
  virtual void  Terminate(Option_t *); 

 private:
  Bool_t        ParseEvent(AliVEvent* event);
  void          ParseTracks(AliVEvent* event);
  void          PushToTTree();
  Bool_t        PassesCuts(AliVEvent* event);
  Bool_t        PassesCuts(AliVTrack* track);

  // Output objects
  TTree*                fTree;              //! output data
  AliFlowTTreeEvent*    fEvent;             //! custom event
  TClonesArray*         fTrackArray;        //! custom tracks

  ClassDef(AliAnalysisTaskTTreeFilter, 1);
};

#endif
Commit	Line	Data
0cb187ff	1	#ifndef ALIANALYSISTTREEFILTER_H
	2	#define ALIANALYSISTTREEFILTER_H
	3
	4	// AliRoot includes
	5	#include "AliAnalysisTaskSE.h"
	6
	7	// forward declarations
	8	class AliVEvent;
	9	class AliVTrack;
	10	class TTree;
	11	class TClonesArray;
	12	class AliFlowTTreeEvent;
	13
	14	class AliAnalysisTaskTTreeFilter : public AliAnalysisTaskSE {
	15	public:
	16	AliAnalysisTaskTTreeFilter();
	17	AliAnalysisTaskTTreeFilter(const char *name);
	18
	19	virtual ~AliAnalysisTaskTTreeFilter();
	20
	21	virtual void UserCreateOutputObjects();
	22	virtual void UserExec(Option_t *option);
	23	virtual void Terminate(Option_t *);
	24
	25	private:
	26	Bool_t ParseEvent(AliVEvent* event);
	27	void ParseTracks(AliVEvent* event);
	28	void PushToTTree();
	29	Bool_t PassesCuts(AliVEvent* event);
	30	Bool_t PassesCuts(AliVTrack* track);
	31
	32	// Output objects
	33	TTree* fTree; //! output data
	34	AliFlowTTreeEvent* fEvent; //! custom event
	35	TClonesArray* fTrackArray; //! custom tracks
	36
	37	ClassDef(AliAnalysisTaskTTreeFilter, 1);
	38	};
	39
	40	#endif