]>
Commit | Line | Data |
---|---|---|
3bb122c7 | 1 | |
2 | #include <TROOT.h> | |
3 | #include <TSystem.h> | |
4 | #include <TInterpreter.h> | |
5 | #include <TChain.h> | |
6 | #include <TFile.h> | |
7 | #include <TList.h> | |
8 | #include <iostream> | |
9 | ||
10 | #include "AliFMDAnalysisTaskSharing.h" | |
11 | #include "AliAnalysisManager.h" | |
12 | #include "AliESDFMD.h" | |
3bb122c7 | 13 | #include "AliMCEventHandler.h" |
14 | #include "AliStack.h" | |
15 | #include "AliESDVertex.h" | |
16 | #include "AliFMDAnaParameters.h" | |
17 | ||
18 | ClassImp(AliFMDAnalysisTaskSharing) | |
19 | ||
20 | //_____________________________________________________________________ | |
21 | AliFMDAnalysisTaskSharing::AliFMDAnalysisTaskSharing() | |
22 | : fDebug(0), | |
23 | fESD(0x0), | |
8dc7c4c2 | 24 | fOutputESD(), |
25 | foutputESDFMD(), | |
c78bc12b | 26 | fSharedThis(kFALSE), |
8dc823cc | 27 | fSharedPrev(kFALSE) |
3bb122c7 | 28 | { |
29 | // Default constructor | |
30 | DefineInput (0, AliESDEvent::Class()); | |
31 | DefineOutput(0, AliESDEvent::Class()); | |
32 | } | |
33 | //_____________________________________________________________________ | |
34 | AliFMDAnalysisTaskSharing::AliFMDAnalysisTaskSharing(const char* name): | |
35 | AliAnalysisTask(name, "AnalysisTaskFMD"), | |
36 | fDebug(0), | |
37 | fESD(0x0), | |
8dc7c4c2 | 38 | fOutputESD(), |
39 | foutputESDFMD(), | |
c78bc12b | 40 | fSharedThis(kFALSE), |
8dc823cc | 41 | fSharedPrev(kFALSE) |
42 | ||
3bb122c7 | 43 | { |
44 | DefineInput (0, AliESDEvent::Class()); | |
45 | DefineOutput(0, AliESDEvent::Class()); | |
46 | } | |
47 | //_____________________________________________________________________ | |
48 | void AliFMDAnalysisTaskSharing::CreateOutputObjects() | |
49 | { | |
8dc7c4c2 | 50 | fOutputESD.CreateStdContent(); |
3bb122c7 | 51 | } |
52 | //_____________________________________________________________________ | |
53 | void AliFMDAnalysisTaskSharing::ConnectInputData(Option_t */*option*/) | |
54 | { | |
55 | fESD = (AliESDEvent*)GetInputData(0); | |
56 | } | |
57 | //_____________________________________________________________________ | |
58 | void AliFMDAnalysisTaskSharing::Exec(Option_t */*option*/) | |
59 | { | |
3bb122c7 | 60 | AliESD* old = fESD->GetAliESDOld(); |
61 | if (old) { | |
62 | fESD->CopyFromOldESD(); | |
63 | } | |
64 | ||
8dc7c4c2 | 65 | foutputESDFMD.Clear(); |
3bb122c7 | 66 | |
8dc7c4c2 | 67 | fOutputESD.SetPrimaryVertexSPD(fESD->GetPrimaryVertexSPD()); |
3bb122c7 | 68 | |
69 | AliESDFMD* fmd = fESD->GetFMDData(); | |
70 | ||
71 | if (!fmd) return; | |
72 | ||
73 | for(UShort_t det=1;det<=3;det++) { | |
74 | Int_t nRings = (det==1 ? 1 : 2); | |
75 | for (UShort_t ir = 0; ir < nRings; ir++) { | |
76 | Char_t ring = (ir == 0 ? 'I' : 'O'); | |
77 | UShort_t nsec = (ir == 0 ? 20 : 40); | |
78 | UShort_t nstr = (ir == 0 ? 512 : 256); | |
8dc7c4c2 | 79 | for(UShort_t sec =0; sec < nsec; sec++) { |
80 | fSharedThis = kFALSE; | |
81 | fSharedPrev = kFALSE; | |
3bb122c7 | 82 | for(UShort_t strip = 0; strip < nstr; strip++) { |
8dc7c4c2 | 83 | foutputESDFMD.SetMultiplicity(det,ring,sec,strip,0.); |
3bb122c7 | 84 | Float_t mult = fmd->Multiplicity(det,ring,sec,strip); |
8dc7c4c2 | 85 | if(mult == AliESDFMD::kInvalidMult || mult == 0) continue; |
c78bc12b | 86 | |
8dc823cc | 87 | Float_t Eprev = 0; |
88 | Float_t Enext = 0; | |
89 | if(strip != 0) | |
90 | if(fmd->Multiplicity(det,ring,sec,strip-1) != AliESDFMD::kInvalidMult) | |
91 | Eprev = fmd->Multiplicity(det,ring,sec,strip-1); | |
92 | if(strip != nstr - 1) | |
93 | if(fmd->Multiplicity(det,ring,sec,strip+1) != AliESDFMD::kInvalidMult) | |
94 | Enext = fmd->Multiplicity(det,ring,sec,strip+1); | |
95 | ||
96 | Float_t nParticles = GetMultiplicityOfStrip(mult,Eprev,Enext,det,ring); | |
8dc7c4c2 | 97 | foutputESDFMD.SetMultiplicity(det,ring,sec,strip,nParticles); |
98 | foutputESDFMD.SetEta(det,ring,sec,strip,fmd->Eta(det,ring,sec,strip)); | |
634ec4d1 | 99 | |
3bb122c7 | 100 | } |
101 | } | |
102 | } | |
103 | } | |
8dc7c4c2 | 104 | fOutputESD.SetFMDData(&foutputESDFMD); |
3bb122c7 | 105 | |
8dc7c4c2 | 106 | PostData(0, &fOutputESD); |
3bb122c7 | 107 | |
108 | } | |
8dc823cc | 109 | //_____________________________________________________________________ |
110 | Float_t AliFMDAnalysisTaskSharing::GetMultiplicityOfStrip(Float_t mult, | |
111 | Float_t Eprev, | |
112 | Float_t Enext, | |
113 | Int_t det, | |
114 | Char_t ring) { | |
115 | AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance(); | |
116 | Float_t nParticles = 0; | |
8dc7c4c2 | 117 | Float_t cutLow = 0.2; |
118 | Float_t cutHigh = pars->GetMPV(det,ring) - 2*pars->GetSigma(det,ring); | |
8dc823cc | 119 | Float_t Etotal = mult; |
8dc7c4c2 | 120 | /* |
121 | if(mult > 3*pars->GetMPV(det,ring) && | |
122 | (Enext > 3*pars->GetMPV(det,ring) || (Enext > 3*pars->GetMPV(det,ring)))) | |
123 | return 0; | |
8dc823cc | 124 | |
8dc7c4c2 | 125 | if(mult > 5*pars->GetMPV(det,ring)) |
126 | return 0; | |
127 | */ | |
c78bc12b | 128 | if(fSharedThis) { |
129 | fSharedThis = kFALSE; | |
130 | fSharedPrev = kTRUE; | |
8dc823cc | 131 | return 0.; |
132 | } | |
133 | ||
134 | if(Etotal < 0.33*pars->GetMPV(det,ring)) { | |
c78bc12b | 135 | fSharedThis = kFALSE; |
136 | fSharedPrev = kFALSE; | |
8dc823cc | 137 | return 0.; |
138 | } | |
139 | ||
140 | if(Eprev > cutLow && Eprev < cutHigh && !fSharedPrev ) { | |
141 | Etotal += Eprev; | |
142 | } | |
143 | ||
144 | if(Enext > cutLow && Enext < cutHigh ) { | |
145 | Etotal += Enext; | |
c78bc12b | 146 | fSharedThis = kTRUE; |
8dc823cc | 147 | } |
148 | ||
8dc7c4c2 | 149 | if(Etotal > cutHigh ) { |
8dc823cc | 150 | nParticles = 1; |
c78bc12b | 151 | fSharedPrev = kTRUE; |
152 | } | |
153 | else { | |
154 | fSharedThis = kFALSE; | |
155 | fSharedPrev = kFALSE; | |
156 | } | |
8dc823cc | 157 | |
158 | return nParticles; | |
159 | } | |
160 | ||
3bb122c7 | 161 | //_____________________________________________________________________ |
162 | // | |
163 | // EOF | |
164 | // |