[u/mrichter/AliRoot.git] / STEER / AliMCEvent.h

// -*- mode: C++ -*- 
#ifndef ALIMCEVENT_H
#define ALIMCEVENT_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */


/* $Id$ */

//-------------------------------------------------------------------------
//                          Class AliMCEvent
// This class gives access to MC truth during the analysis.
// Monte Carlo truth is containe in the kinematics tree (produced particles) and 
// the tree of reference hits.
//      
// Origin: Andreas Morsch, CERN, andreas.morsch@cern.ch 
//-------------------------------------------------------------------------

#include "AliVirtualEventHandler.h"
class TFile;
class TTree;
class TParticle;
class TClonesArray;
class AliHeader;
class AliStack;


class AliMCEvent : public AliVirtualEventHandler 
{
public:
    AliMCEvent();
    AliMCEvent(const char* name, const char* title);
    virtual ~AliMCEvent();
    virtual void         SetOutputFileName(char* /* fname */) {;}
    virtual char*        GetOutputFileName() {return 0;}
    virtual Bool_t       InitIO(Option_t* opt);
    virtual Bool_t       BeginEvent();
    virtual Bool_t       FinishEvent();
    virtual Bool_t       Terminate();
    virtual Bool_t       TerminateIO();
    //
    AliStack* Stack()  {return fStack;}
    TTree*    TreeTR() {return fTreeTR;}
    Int_t     GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs);
    void      DrawCheck(Int_t i);
	    
private:
    TFile            *fFileE;            //! File with TreeE
    TFile            *fFileK;            //! File with TreeK
    TFile            *fFileTR;           //! File with TreeTR
    TTree            *fTreeE;            //! TreeE  (Event Headers)
    TTree            *fTreeK;            //! TreeK  (kinematics tree)
    TTree            *fTreeTR;           //! TreeTR (track references tree)
    AliStack         *fStack;            //! Current pointer to stack
    AliHeader        *fHeader;           //! Current pointer to header
    TClonesArray     *fTrackReferences;  //! Current list of tarck references
    Int_t             fNEvent;           //! Number of events
    Int_t             fEvent;            //! Current event
    Int_t             fNprimaries;       //! Number of primaries
    Int_t             fNparticles;       //! Number of particles 
    ClassDef(AliMCEvent,1)  //MCEvent class 
};
#endif
Commit	Line	Data
033693d3	1	// -- mode: C++ --
	2	#ifndef ALIMCEVENT_H
	3	#define ALIMCEVENT_H
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7
	8	/* $Id$ */
	9
	10	//-------------------------------------------------------------------------
	11	// Class AliMCEvent
	12	// This class gives access to MC truth during the analysis.
	13	// Monte Carlo truth is containe in the kinematics tree (produced particles) and
	14	// the tree of reference hits.
	15	//
	16	// Origin: Andreas Morsch, CERN, andreas.morsch@cern.ch
	17	//-------------------------------------------------------------------------
	18
	19	#include "AliVirtualEventHandler.h"
	20	class TFile;
	21	class TTree;
	22	class TParticle;
	23	class TClonesArray;
	24	class AliHeader;
	25	class AliStack;
	26
	27
	28	class AliMCEvent : public AliVirtualEventHandler
	29	{
	30	public:
	31	AliMCEvent();
	32	AliMCEvent(const char* name, const char* title);
	33	virtual ~AliMCEvent();
	34	virtual void SetOutputFileName(char* /* fname */) {;}
	35	virtual char* GetOutputFileName() {return 0;}
	36	virtual Bool_t InitIO(Option_t* opt);
	37	virtual Bool_t BeginEvent();
	38	virtual Bool_t FinishEvent();
	39	virtual Bool_t Terminate();
	40	virtual Bool_t TerminateIO();
	41	//
	42	AliStack* Stack() {return fStack;}
	43	TTree* TreeTR() {return fTreeTR;}
	44	Int_t GetParticleAndTR(Int_t i, TParticle& particle, TClonesArray& trefs);
	45	void DrawCheck(Int_t i);
	46
	47	private:
	48	TFile *fFileE; //! File with TreeE
	49	TFile *fFileK; //! File with TreeK
	50	TFile *fFileTR; //! File with TreeTR
	51	TTree *fTreeE; //! TreeE (Event Headers)
	52	TTree *fTreeK; //! TreeK (kinematics tree)
	53	TTree *fTreeTR; //! TreeTR (track references tree)
	54	AliStack *fStack; //! Current pointer to stack
	55	AliHeader *fHeader; //! Current pointer to header
	56	TClonesArray *fTrackReferences; //! Current list of tarck references
	57	Int_t fNEvent; //! Number of events
	58	Int_t fEvent; //! Current event
	59	Int_t fNprimaries; //! Number of primaries
	60	Int_t fNparticles; //! Number of particles
	61	ClassDef(AliMCEvent,1) //MCEvent class
	62	};
	63	#endif
	64