[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemto / AliFemtoEventReaderAOD.h

////////////////////////////////////////////////////////////////////////////////
//                                                                            //
// AliFemtoEventReaderAOD - the reader class for the Alice AOD                //
// Reads in AOD information and converts it into internal AliFemtoEvent       //
// Authors: Adam Kisiel kisiel@mps.ohio-state.edu                             //
//                                                                            //
////////////////////////////////////////////////////////////////////////////////

#ifndef ALIFEMTOEVENTREADERAOD_H
#define ALIFEMTOEVENTREADERAOD_H
#include "AliFemtoEventReader.h"
#include "AliFemtoEnumeration.h"

#include <string>
#include <vector>
#include "TTree.h"
#include "TChain.h"
#include "TBits.h"
#include "AliAODEvent.h"
#include <list>
#include "AliPWG2AODTrack.h"

class AliFemtoEvent;
class AliFemtoTrack;

class AliFemtoEventReaderAOD : public AliFemtoEventReader 
{
 public:
  AliFemtoEventReaderAOD();
  AliFemtoEventReaderAOD(const AliFemtoEventReaderAOD &aReader);
  virtual ~AliFemtoEventReaderAOD();

  AliFemtoEventReaderAOD& operator=(const AliFemtoEventReaderAOD& aReader);

  virtual AliFemtoEvent* ReturnHbtEvent();
  AliFemtoString Report();
  void SetInputFile(const char* inputFile);
  void SetFilterBit(UInt_t ibit);

 protected:
  virtual void CopyAODtoFemtoEvent(AliFemtoEvent *tEvent);
  virtual void CopyAODtoFemtoTrack(const AliAODTrack *tAodTrack, 
				   AliFemtoTrack *tFemtoTrack, 
				   AliPWG2AODTrack *tPWG2AODTrack);

  int            fNumberofEvent;    // number of Events in AOD file
  int            fCurEvent;         // number of current event
  AliAODEvent*   fEvent;            // AOD event
  TBits          fAllTrue;          // Bit set with all true bits
  TBits          fAllFalse;         // Bit set with all false bits
  UInt_t         fFilterBit;        // Bitmap bit for AOD filters
  TClonesArray*  fPWG2AODTracks;    // Link to PWG2 specific AOD information (if it exists)

 private:

  string         fInputFile;        // name of input file with AOD filenames
  string         fFileName;         // name of current AOD file
  TChain*        fTree;             // AOD tree
  TFile*         fAodFile;          // AOD file 

#ifdef __ROOT__
  ClassDef(AliFemtoEventReaderAOD, 11)
#endif

    };
  
#endif
Commit	Line	Data
de568db1	1	////////////////////////////////////////////////////////////////////////////////
	2	// //
	3	// AliFemtoEventReaderAOD - the reader class for the Alice AOD //
	4	// Reads in AOD information and converts it into internal AliFemtoEvent //
	5	// Authors: Adam Kisiel kisiel@mps.ohio-state.edu //
	6	// //
	7	////////////////////////////////////////////////////////////////////////////////
	8
	9	#ifndef ALIFEMTOEVENTREADERAOD_H
	10	#define ALIFEMTOEVENTREADERAOD_H
	11	#include "AliFemtoEventReader.h"
	12	#include "AliFemtoEnumeration.h"
	13
	14	#include <string>
	15	#include <vector>
	16	#include "TTree.h"
	17	#include "TChain.h"
	18	#include "TBits.h"
	19	#include "AliAODEvent.h"
	20	#include <list>
	21	#include "AliPWG2AODTrack.h"
	22
	23	class AliFemtoEvent;
	24	class AliFemtoTrack;
	25
	26	class AliFemtoEventReaderAOD : public AliFemtoEventReader
	27	{
	28	public:
	29	AliFemtoEventReaderAOD();
	30	AliFemtoEventReaderAOD(const AliFemtoEventReaderAOD &aReader);
	31	virtual ~AliFemtoEventReaderAOD();
	32
	33	AliFemtoEventReaderAOD& operator=(const AliFemtoEventReaderAOD& aReader);
	34
	35	virtual AliFemtoEvent* ReturnHbtEvent();
	36	AliFemtoString Report();
	37	void SetInputFile(const char* inputFile);
	38	void SetFilterBit(UInt_t ibit);
	39
	40	protected:
	41	virtual void CopyAODtoFemtoEvent(AliFemtoEvent *tEvent);
	42	virtual void CopyAODtoFemtoTrack(const AliAODTrack *tAodTrack,
	43	AliFemtoTrack *tFemtoTrack,
	44	AliPWG2AODTrack *tPWG2AODTrack);
	45
	46	int fNumberofEvent; // number of Events in AOD file
	47	int fCurEvent; // number of current event
	48	AliAODEvent* fEvent; // AOD event
	49	TBits fAllTrue; // Bit set with all true bits
	50	TBits fAllFalse; // Bit set with all false bits
	51	UInt_t fFilterBit; // Bitmap bit for AOD filters
	52	TClonesArray* fPWG2AODTracks; // Link to PWG2 specific AOD information (if it exists)
	53
	54	private:
	55
	56	string fInputFile; // name of input file with AOD filenames
	57	string fFileName; // name of current AOD file
	58	TChain* fTree; // AOD tree
	59	TFile* fAodFile; // AOD file
	60
	61	#ifdef __ROOT__
	62	ClassDef(AliFemtoEventReaderAOD, 11)
	63	#endif
	64
65	};
66
67	#endif
68
69