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

#ifndef ALIGENCOCKTAIL_H
#define ALIGENCOCKTAIL_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 "AliGenerator.h"
#include <TArrayF.h>
#include <TList.h>

class AliGenCocktailEntry;
class AliGenCocktailEventHeader;
class TArrayF;
class TFormula;

class AliGenCocktail : public AliGenerator
{
 public:
    AliGenCocktail();
     
    virtual ~AliGenCocktail();
    virtual void Init();
    virtual void FinishRun();
    virtual void Generate();
    virtual void SetVertexSmear(VertexSmear_t smear);
    virtual void SetRandomise(Bool_t flag) {fRandom = flag;}
    virtual void UsePerEventRates() {fUsePerEventRate  = kTRUE;}
	    
    //
    // Add a new generator to the list
    virtual void AddGenerator
      (AliGenerator *Generator, const char* Name, Float_t RateExp, TFormula* formula = 0);
    virtual TList* Entries() {return fEntries;}
    // Iterators
    AliGenCocktailEntry*  FirstGenerator();
    AliGenCocktailEntry*  NextGenerator();
    void FirstGeneratorPair(AliGenCocktailEntry*&e1, AliGenCocktailEntry*&e2);
    void NextGeneratorPair (AliGenCocktailEntry*&e1, AliGenCocktailEntry*&e2);
    virtual void AddHeader(AliGenEventHeader* header);
	    
 protected:
    Int_t fNGenerators;                 // Number of generators booked
    Float_t fTotalRate;                 // Total rate of the generators booked
    Bool_t fRandom;                     // Flag to select random generator from list
    Bool_t fUsePerEventRate;            // Flag to generate the events according to the rate per event    
    TArrayF  fProb;                     // Probability of an event (if fRandom == kTRUE)
    TList  *fEntries;                   // List of Generators
    TObjLink *flnk1;                    // ! Iterator for first generator
    TObjLink *flnk2;                    // ! Iterator for second generator
    AliGenCocktailEventHeader* fHeader; // !Header container  
			   
//
 private:
    AliGenCocktail(const AliGenCocktail &cocktail);
    AliGenCocktail & operator=(const AliGenCocktail & rhs);

    ClassDef(AliGenCocktail,1) // Particle cocktail generator a la SHAKER
};

#endif
Commit	Line	Data
	1	#ifndef ALIGENCOCKTAIL_H
	2	#define ALIGENCOCKTAIL_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
	13	#include "AliGenerator.h"
	14	#include <TArrayF.h>
	15	#include <TList.h>
	16
	17	class AliGenCocktailEntry;
	18	class AliGenCocktailEventHeader;
	19	class TArrayF;
	20	class TFormula;
	21
	22	class AliGenCocktail : public AliGenerator
	23	{
	24	public:
	25	AliGenCocktail();
	26
	27	virtual ~AliGenCocktail();
	28	virtual void Init();
	29	virtual void FinishRun();
	30	virtual void Generate();
	31	virtual void SetVertexSmear(VertexSmear_t smear);
	32	virtual void SetRandomise(Bool_t flag) {fRandom = flag;}
	33	virtual void UsePerEventRates() {fUsePerEventRate = kTRUE;}
	34
	35	//
	36	// Add a new generator to the list
	37	virtual void AddGenerator
	38	(AliGenerator Generator, const char Name, Float_t RateExp, TFormula* formula = 0);
	39	virtual TList* Entries() {return fEntries;}
	40	// Iterators
	41	AliGenCocktailEntry* FirstGenerator();
	42	AliGenCocktailEntry* NextGenerator();
	43	void FirstGeneratorPair(AliGenCocktailEntry&e1, AliGenCocktailEntry&e2);
	44	void NextGeneratorPair (AliGenCocktailEntry&e1, AliGenCocktailEntry&e2);
	45	virtual void AddHeader(AliGenEventHeader* header);
	46
	47	protected:
	48	Int_t fNGenerators; // Number of generators booked
	49	Float_t fTotalRate; // Total rate of the generators booked
	50	Bool_t fRandom; // Flag to select random generator from list
	51	Bool_t fUsePerEventRate; // Flag to generate the events according to the rate per event
	52	TArrayF fProb; // Probability of an event (if fRandom == kTRUE)
	53	TList *fEntries; // List of Generators
	54	TObjLink *flnk1; // ! Iterator for first generator
	55	TObjLink *flnk2; // ! Iterator for second generator
	56	AliGenCocktailEventHeader* fHeader; // !Header container
	57
	58	//
	59	private:
	60	AliGenCocktail(const AliGenCocktail &cocktail);
	61	AliGenCocktail & operator=(const AliGenCocktail & rhs);
	62
	63	ClassDef(AliGenCocktail,1) // Particle cocktail generator a la SHAKER
	64	};
	65
	66	#endif
	67
	68
	69
	70
	71