MISE/ManagerStringFragmentation.h at master · altsybee/MISE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#ifndef ManagerStringFragmentation_H
#define	ManagerStringFragmentation_H

//#include "Event.h"
#include "TList.h"
#include "TFile.h"
#include "TString.h"

//#include "ManagerStringFragmentationBase.h"

class TH1D;
class TH2D;
class TCanvas;
//class Event;
//class NucleiCollision;
class StringDescr;
//class AnalyserBase;

class ManagerStringFragmentation {
public:
    ManagerStringFragmentation();
    ManagerStringFragmentation(const ManagerStringFragmentation&);
    virtual ~ManagerStringFragmentation();

    // file name with NuclColl data
    void setInputFileNameNucleiCollisions(TString name) { fInputFileName_NucleiCollisions = name; }
    void setInputFileNameStringBoosts(TString name) { fInputFileName_StringBoosts = name; }

    void setOutputDirectoryName(TString strDirName) { fOutputDirName = strDirName; }
    void setOutputFileName(TString name) { fOutputFileName = name; }
    void setDrawHistos(bool flag) { fDrawHistos = flag; }

    void setFillEventTree(bool flag) { fFillEventTree = flag; }

    void setNumberOfCentralityBins(int nCentralityBins) { fNumberOfCentralityBins = nCentralityBins; }
    void setCutMinNumberOfParticles(int n) { fCutMinNumberOfParticles = n; }

    void setWhatToDoWithHardScattering(int what) { whatToDoWithHardScattering = what; }

    //    Event *getEvent() /*const*/
//    {
//        //if (!fEvent)
//        //    fEvent = new Event();
//        return fEvent;
//    }

    void initOutputObjects();
    void applyFragmentationToEvents( StringDescr *strDescr, int nEvents );
    void drawStatHists();

//    void SetFlagGenerateCentralEvent(bool flag) { fFlagGenerateCentralEventByHand = flag; }
//    void SetFlagGenerateSemicentralEvent(bool flag) { fFlagGenerateSemicentralEventByHand = flag; }
//    void SetImpactParameterByHand(float par) { fImpactParameterByHand = par; }

    void cleanup();
    //TList * getProcessorsList() { return &fLRCproc; }
private:
    //TString fStrSpecTitle;

//    NucleiCollision *fPtrNuclStruct;

    bool fPrintInfo;
    bool fDrawHistos;
    bool fFillEventTree;

    int whatToDoWithHardScattering; // 0 - makeTwoJets, 1 - particle pair with random pt from Power law

    TString fInputFileName_NucleiCollisions;
    TString fInputFileName_StringBoosts;

    TString fOutputDirName;
    TString fOutputFileName;
    TString fOutputListName;
//    Event *fEvent;
//    int fNevents;
    Int_t fNumberOfCentralityBins;
    //Distribution *fNsourcesDistr;
    //Distribution *fNparticlesDistr;
    //Distribution *fPtDistr;
    Int_t fCutMinNumberOfParticles;

//    bool fFlagGenerateCentralEventByHand;
//    bool fFlagGenerateSemicentralEventByHand;
//    float fImpactParameterByHand;

    TH1D *fHistSources;
    TH1D *fHistParticlesInSource;
    TH1D *fHistParticlesInSourceIncludingJets;

    TH1D *fHistParticlesInJets;

    TH1D *fHistParticlesInEvent;
    TH1D *fHistParticlesInCutConditionInEvent;
    TH2D *fHistParticlesInCutConditionVsNu;

//    TH1D *fHistParticlesInEventInEta;
//    TH1D *fHistParticlesInCutConditionInEventInEta;

    TH1D* fHistPt;
    TH1D* fHistEta;
    TH1D* fHistEtaInPtCuts[5];
    TH1D* fHistPhi;
    TH1D* fHistPtAfterCuts;
    TH1D* fHistPtAfterCutsPID[6]; // [5];
    TH1D* fHistPtBeforeKick;

    TH1D* fHistNeventsInCentralityClasses;

    TH2D* fHist2DEtaPhi;

    TCanvas *fCanv;
    TFile *fOutputFile;
    //TList fLRCproc;       //  AliLRCProcess objects list
    //int fLrcNum;

//    //window sets
//    TList * fWindowSets[100];
//    int fNumberOfWindowSets;

//    double fPtCutMin;
//    double fPtCutMax;

    //TRandom3 fRand;
};

#endif	/* ManagerStringFragmentation_H */