[u/mrichter/AliRoot.git] / EVGEN / AliGenCocktailAfterBurner.h

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

/* $Id$ */

// Container class for AliGenerator through recursion.
// (Container is itself an AliGenerator)
// Author: andreas.morsch@cern.ch 
//
#include "AliGenCocktailAfterBurner.h"
#include "AliGenCocktail.h"


class AliGenCocktailEntry;
class AliStack;

class AliGenCocktailAfterBurner : public  AliGenCocktail
{
 public:
    AliGenCocktailAfterBurner();
//    AliGenCocktailAfterBurner(const AliGenCocktailAfterBurner &cocktail){}
     
    virtual ~AliGenCocktailAfterBurner();
    virtual void Init();
    virtual void Generate();
    virtual void SetTracks(Int_t stackno);
    //
    // Add a new generator to the list
    virtual void AddAfterBurner
	(AliGenerator *Generator, char* Name, Float_t RateExp );
    AliGenCocktailAfterBurner & operator=(const AliGenCocktailAfterBurner & rhs);
    
    AliStack* GetStack(Int_t n);
    AliStack* GetActiveStack() {return fActiveStack;}
    
    AliGenerator* GetCurrentGenerator();
    virtual void  SetActiveEventNumber(Int_t actev);
    Int_t GetActiveEventNumber() {return fActiveEvent;}
    static AliMCProcess IntToMCProcess(Int_t no);
 protected:
    Int_t fNAfterBurners; 
    TList  *fAfterBurnerEntries;
    Bool_t fGenerationDone;
    TObjArray *fInternalStacks; // ! List of internal stacks
    Int_t fCurrentEvent; //  Number of current event/stack
    
    
    AliStack* fActiveStack; // ! pointer to the current stack
    Int_t fActiveEvent;
   
    AliGenerator *fCurrentGenerator;
    
    ClassDef(AliGenCocktailAfterBurner,1) // Particle cocktail generator a la SHAKER
};

inline  AliGenerator*  
    AliGenCocktailAfterBurner::GetCurrentGenerator()
{
  return fCurrentGenerator;
}


#endif
Commit	Line	Data
0b359ada	1	#ifndef AliGenCocktailAfterBurner_H
	2	#define AliGenCocktailAfterBurner_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	// Container class for AliGenerator through recursion.
	9	// (Container is itself an AliGenerator)
	10	// Author: andreas.morsch@cern.ch
	11	//
	12	#include "AliGenCocktailAfterBurner.h"
	13	#include "AliGenCocktail.h"
	14
	15
	16	class AliGenCocktailEntry;
	17	class AliStack;
	18
	19	class AliGenCocktailAfterBurner : public AliGenCocktail
	20	{
	21	public:
	22	AliGenCocktailAfterBurner();
	23	// AliGenCocktailAfterBurner(const AliGenCocktailAfterBurner &cocktail){}
	24
	25	virtual ~AliGenCocktailAfterBurner();
	26	virtual void Init();
	27	virtual void Generate();
	28	virtual void SetTracks(Int_t stackno);
	29	//
	30	// Add a new generator to the list
	31	virtual void AddAfterBurner
	32	(AliGenerator Generator, char Name, Float_t RateExp );
	33	AliGenCocktailAfterBurner & operator=(const AliGenCocktailAfterBurner & rhs);
	34
	35	AliStack* GetStack(Int_t n);
	36	AliStack* GetActiveStack() {return fActiveStack;}
	37
	38	AliGenerator* GetCurrentGenerator();
	39	virtual void SetActiveEventNumber(Int_t actev);
	40	Int_t GetActiveEventNumber() {return fActiveEvent;}
	41	static AliMCProcess IntToMCProcess(Int_t no);
	42	protected:
	43	Int_t fNAfterBurners;
	44	TList *fAfterBurnerEntries;
	45	Bool_t fGenerationDone;
	46	TObjArray *fInternalStacks; // ! List of internal stacks
	47	Int_t fCurrentEvent; // Number of current event/stack
	48
	49
	50	AliStack* fActiveStack; // ! pointer to the current stack
	51	Int_t fActiveEvent;
	52
	53	AliGenerator *fCurrentGenerator;
	54
	55	ClassDef(AliGenCocktailAfterBurner,1) // Particle cocktail generator a la SHAKER
	56	};
	57
	58	inline AliGenerator*
	59	AliGenCocktailAfterBurner::GetCurrentGenerator()
	60	{
	61	return fCurrentGenerator;
	62	}
	63
	64
65	#endif
66
67
68
69
70