Fix coding convention violation
[u/mrichter/AliRoot.git] / ALIFAST / AliFast.h
1 #ifndef AliFast_H
2 #define AliFast_H
3
4 //////////////////////////////////////////////////////////////////////////
5 //                                                                      //
6 // AliFast                                                              //
7 //                                                                      //
8 // Main class to control the AliFast program.                           //
9 //                                                                      //
10 // This class :                                                         //
11 //   - Initialises the run default parameters                           //
12 //   - Provides API to Set/Get run parameters                           //
13 //   - Creates the support lists (TClonesArrays) for the Event structure//
14 //   - Creates the physics objects makers                               //
15 //                                                                      //
16 //////////////////////////////////////////////////////////////////////////
17
18 #include "AliFBigBang.h"
19 #include "AliFHistBrowser.h"
20 #include "AliRun.h"
21
22 class TBrowser;
23 class TChain;
24 class TTree;
25
26 class AliFDet;
27 class AliFMCMaker;
28 class AliFMaker;
29 class AliFTrackMaker;
30 class AliFVirtualDisplay;
31
32 class AliFast : public AliRun {
33
34 public:
35                       AliFast();
36                       AliFast(const AliFast &afast);
37                       AliFast(const char *name, const char *title="The ALIAS fast MonteCarlo");
38    virtual           ~AliFast();
39    AliFast& operator = (const AliFast &afast) 
40      {afast.Copy(*this); return (*this);}
41
42    virtual void       Browse(TBrowser *b);
43    virtual void       Draw(Option_t *option="") const;  // *MENU*
44    Int_t              GetVersion() const {return fVersion;}
45    Int_t              GetVersionDate() const {return fVersionDate;}
46    virtual void       Clear(Option_t *option="");
47    AliFVirtualDisplay *FDisplay() {return fFDisplay;}
48    virtual void       FillClone();
49    virtual void       Finish();
50    virtual void       GetTreeEvent(Int_t event);  // *MENU*
51    virtual void       Init();
52    Bool_t             IsFolder() const {return kTRUE;}
53    virtual void       Make(Int_t i=0);
54    virtual void       Paint(Option_t *option="");
55    virtual void       PrintInfo();
56    virtual void       SetDefaultParameters();
57
58    TList             *Makers()    const {return fMakers;}
59    AliFMaker         *Maker(const char *name) const {return (AliFMaker*)fMakers->FindObject(name);}
60    AliFMCMaker       *MCMaker()     const {return fMCMaker;}
61    AliFTrackMaker    *TrackMaker()  const {return fTrackMaker;}
62    AliFDet           *Detector()    const {return fDet;}
63    TTree             *Tree()        const {return fTree;}
64
65    Int_t             Run()          const {return fRun;}
66    Int_t             Event()        const {return fEvent;}
67    Int_t             Mode()         const {return fMode;}
68    Int_t             TestTrack()    const {return fTestTrack;}
69
70 //    Getters for flags and switches
71    Int_t             Luminosity()   const {return fLuminosity;}
72    Bool_t            Bfield()       const {return fBfield;}
73    Bool_t            Smearing()     const {return fSmearing;} 
74    Int_t             SUSYcodeLSP()  const {return fSUSYcodeLSP;}
75    Bool_t            TrackFinding() const {return fTrackFinding;}
76
77
78 //    Setter for Event Display
79    void           SetDisplay(AliFVirtualDisplay *display) {fFDisplay=display;}
80 //    Setters for flags and switches
81    void           SetLuminosity(Int_t lumi=1)   {fLuminosity=lumi;}
82    void           SetBfield(Bool_t field=1)     {fBfield=field;}
83    void           SetSmearing(Bool_t val=1)     {fSmearing=val;} 
84    void           SetSUSYcodeLSP(Int_t val=66)  {fSUSYcodeLSP=val;}
85    void           SetTrackFinding(Bool_t val=0) {fTrackFinding=val;}
86
87    virtual void   SetRun(Int_t run=1)     {fRun=run;}
88    virtual void   SetEvent(Int_t event=1) {fEvent=event;}
89    virtual void   SetMode(Int_t mode=0)   {fMode=mode;}
90    virtual void   SetTestTrack(Int_t test=0)   {fTestTrack=test;}
91
92    void           FillTree();
93    void           InitChain(TChain *chain);
94    void           MakeTree(const char* name="T", const char*title="AliFast tree");
95
96    void           SortDown(Int_t n, Float_t *a, Int_t *index, Bool_t down=kTRUE) const;
97
98 private:
99
100    void Copy(TObject &afast) const;
101
102    Int_t               fVersion;           //AliFast version number
103    Int_t               fVersionDate;       //AliFast version date
104    Int_t               fMode;              //Run mode
105    Int_t               fTestTrack;         //Test mode for TrackMaker
106    TTree              *fTree;              //!Pointer to the Root tree
107    TList              *fMakers;            //List of Makers
108 //    pointers to standard Makers
109    AliFMCMaker        *fMCMaker;           //!Pointer to MCMaker
110    AliFTrackMaker     *fTrackMaker;        //!Pointer to TrackMaker
111    AliFDet            *fDet;               //!Pointer to Detector
112 //    flags and switches
113    Int_t               fLuminosity;        //Luminosity option (low=1, high=2)
114    Bool_t              fBfield;            //B-field (on=1,off=0)
115    Bool_t              fSmearing;          //Smearing on=1, off=0
116    Int_t               fSUSYcodeLSP;       //Code for SUSY LSP particle
117    Int_t               fTrackFinding;      //Track/finding on=1,off=0
118    AliFHistBrowser     fHistBrowser;       //Object to Browse Maker histograms
119    AliFBigBang         fBigBang;           //!Object to Browse generated particles
120    AliFVirtualDisplay *fFDisplay;           //!Pointer to Event display object
121
122    ClassDef(AliFast, 1)   //AliFast control class
123 };
124
125 R__EXTERN AliFast *gAliFast;
126
127 #endif
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144