]> git.uio.no Git - u/mrichter/AliRoot.git/blob - ANALYSIS/AliTRDTriggerAnalysis.cxx
From Jochen Klein: patch which extends the functionality of
[u/mrichter/AliRoot.git] / ANALYSIS / AliTRDTriggerAnalysis.cxx
1 /**************************************************************************
2  * Copyright(c) 2013, ALICE Experiment at CERN, All rights reserved.      *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 // evaluate TRD trigger conditions,
17 // potentially with hardened conditions to remove
18 // triggers caused by conversions of low-pt photons
19 // at large radii
20 //
21 // Author: Jochen Klein <jochen.klein@cern.ch>
22
23 #include "AliLog.h"
24 #include "AliVTrack.h"
25 #include "AliVEvent.h"
26 #include "AliESDEvent.h"
27 #include "AliAODEvent.h"
28 #include "AliVTrdTrack.h"
29 #include "AliESDTrdTrigger.h"
30
31 #include "AliTRDTriggerAnalysis.h"
32
33 AliTRDTriggerAnalysis::AliTRDTriggerAnalysis() :
34   TObject(),
35   fTriggerFlags(),
36   fTriggerInputs(0),
37   fTriggerClasses(0),
38   fVerbosity(0),
39   fRequireMatch(kFALSE),
40   fRequireMatchElectron(kFALSE),
41   fRequireInTime(kTRUE),
42   fTRDlayerMaskEl(0x1),
43   fTRDnTrackletsEl(5),
44   fTRDptHSE(3.),
45   fTRDpidHSE(144),
46   fTRDptHQU(2.),
47   fTRDpidHQU(164),
48   fTRDptHEE(3.),
49   fTRDpidHEE(144),
50   fTRDminSectorHEE(6),
51   fTRDmaxSectorHEE(8),
52   fTRDptHJT(3.),
53   fTRDnHJT(3)
54 {
55   // ctor
56
57   memset(fTriggerFlags, 0, sizeof(fTriggerFlags));
58 }
59
60 AliTRDTriggerAnalysis::~AliTRDTriggerAnalysis()
61 {
62   // dtor
63 }
64
65 void AliTRDTriggerAnalysis::ResetTriggers()
66 {
67   // reset internal cache of trigger status
68
69   memset(fTriggerFlags, 0, sizeof(fTriggerFlags));
70   fTriggerInputs = fTriggerClasses = 0;
71 }
72
73 Bool_t AliTRDTriggerAnalysis::CalcTriggers(const AliVEvent *event)
74 {
75   // evaluate the TRD trigger conditions,
76   // so far HCO, HSE, HQU, HJT, HEE
77
78   ResetTriggers();
79
80   if (!event) {
81     AliErrorClass("event pointer is null");
82     return kFALSE;
83   }
84
85   // GTU information
86   UInt_t header = 0x0;
87
88   if (fVerbosity > 0)
89     printf("******************************************************************\n");
90   const AliESDEvent *esdEvent = dynamic_cast<const AliESDEvent*> (event);
91   if (esdEvent) {
92     AliESDTrdTrigger* trdTriggerInfo = esdEvent->GetTrdTrigger();
93
94     for (Int_t iSector = 0; iSector < 18; ++iSector) {
95       UInt_t  trgFlags = trdTriggerInfo->GetFlags(iSector);
96
97       UInt_t trgContribs = fTriggerContribs[iSector] = trgFlags & 0xfff;
98       header |= trgContribs;
99       if (fVerbosity > 0)
100         printf("sector %2i: %5s %5s %5s %5s %5s %5s %5s %5s (0x%03x)\n", iSector,
101                trgContribs & (1 << 7) ? "TO" : "--",
102                trgContribs & (1 << 6) ? "E2" : "--",
103                trgContribs & (1 << 5) ? "E1" : "--",
104                trgContribs & (1 << 4) ? "J1" : "--",
105                trgContribs & (1 << 3) ? "H2" : "--",
106                trgContribs & (1 << 2) ? "H1" : "--",
107                trgContribs & (1 << 1) ? "E3" : "--",
108                trgContribs & (1 << 0) ? "M1" : "--",
109                trgContribs);
110
111       // trackingDoneTimeSMU = ((trgFlags >> 12) & 0x3ff) * 1./120.;
112
113       // for (Int_t iStack = 0; iStack < 5; ++iStack) {
114       //   trackingDoneTMU = ((trgFlags >> 27) & (1 << iStack)) ? kTRUE : kFALSE;  // TMU-level tracking done flag
115       //   trackingDoneSMUStack = ((trgFlags >> 22) & (1 << iStack)) ? kTRUE : kFALSE;  // SMU-level stack-related tracking done flag
116
117       //   lmeFlags = trdTriggerInfo->GetLME(iStack) & 0xffffff;
118       //   crcErrorFlags = (~(trdTriggerInfo->GetLME(iStack) >> 24)) & 0x3;
119       // }
120     }
121     if (fVerbosity > 0) {
122       printf("------------------------------------------------------------------\n");
123       printf("total    : %5s %5s %5s %5s %5s %5s %5s %5s (0x%03x)\n",
124              header & (1 << 7) ? "TO" : "--",
125              header & (1 << 6) ? "E2" : "--",
126              header & (1 << 5) ? "E1" : "--",
127              header & (1 << 4) ? "J1" : "--",
128              header & (1 << 3) ? "H2" : "--",
129              header & (1 << 2) ? "H1" : "--",
130              header & (1 << 1) ? "E3" : "--",
131              header & (1 << 0) ? "M1" : "--",
132              header);
133     }
134   }
135
136   // evaluate trigger classes
137   TString trgClasses = event->GetFiredTriggerClasses();
138   if (trgClasses.Contains("TRDCO2"))
139     MarkClass(kHCO);
140   if (trgClasses.Contains("WUHJT"))
141     MarkClass(kHJT);
142   if (trgClasses.Contains("WUHSE"))
143     MarkClass(kHSE);
144   if (trgClasses.Contains("WUHQU"))
145     MarkClass(kHQU);
146   if (trgClasses.Contains("WUHEE"))
147     MarkClass(kHEE);
148
149   // evaluate trigger inputs
150   UInt_t  trgInputs = 0;
151   if (esdEvent)
152     trgInputs  = esdEvent->GetHeader()->GetL1TriggerInputs();
153   else if (const AliAODEvent *aodEvent = dynamic_cast<const AliAODEvent*> (event))
154     trgInputs  = aodEvent->GetHeader()->GetL1TriggerInputs();
155   else
156     AliError("failed to retrieve L1 trigger inputs");
157
158   if (trgInputs & (1 <<  8))
159     MarkInput(kHCO);
160   if (trgInputs & (1 <<  9))
161     MarkInput(kHJT);
162   if (trgInputs & (1 << 10))
163     MarkInput(kHSE);
164   if (trgInputs & (1 << 12))
165     MarkInput(kHQU);
166   if (trgInputs & (1 << 13))
167     MarkInput(kHEE);
168
169   // evaluate TRD GTU tracks
170   Int_t nTracks[90]      = { 0 }; // stack-wise counted number of tracks above pt threshold
171
172   Int_t nTrdTracks = event->GetNumberOfTrdTracks();
173
174   for (Int_t iTrack = 0; iTrack < nTrdTracks; ++iTrack) {
175     AliVTrdTrack *trdTrack = event->GetTrdTrack(iTrack);
176     if (!trdTrack) {
177       AliError(Form("Failed to get track %i", iTrack));
178       continue;
179     }
180
181     Int_t globalStack = 5*trdTrack->GetSector() + trdTrack->GetStack();
182
183     MarkCondition(kHCO, globalStack);
184
185     for (Int_t iLayer = 0; iLayer < 6; ++iLayer) {
186       if (trdTrack->GetLayerMask() & (1 << iLayer)) {
187         AliVTrdTracklet *trkl = trdTrack->GetTracklet(iLayer);
188         if (!trkl) {
189           AliError(Form("no tracklet in layer %i where one should be for track %i",
190                    iLayer, iTrack));
191         }
192       }
193     }
194
195     // ignore the track if it was not in time
196     // (if required)
197     if (fRequireInTime && !trdTrack->GetTrackInTime())
198       continue;
199
200     AliVTrack *match = trdTrack->GetTrackMatch();
201     AliDebug(2, Form("GTU track %2i with pt = %5.2f has match: %p (pt = %5.2f)",
202                      iTrack, trdTrack->Pt(), match, match ? match->Pt() : 0));
203
204     // ignore the track if it does not have a matched global track
205     // (if required)
206     if (fRequireMatch && !match)
207       continue;
208
209     // stack-wise counting of tracks above pt threshold for jet trigger
210     if (TMath::Abs(trdTrack->Pt()) >= fTRDptHJT) {
211       ++nTracks[globalStack];
212     }
213
214     // ignore the track for the electron triggers
215     // if it does not have a matched global track
216     // (if required)
217     if (fRequireMatchElectron && !match)
218       continue;
219
220     // ignore the track for the electron triggers
221     // if it does not fulfill the tracklet requirement
222     if (trdTrack->GetNTracklets() < fTRDnTrackletsEl)
223       continue;
224     if ((trdTrack->GetLayerMask() & fTRDlayerMaskEl) != fTRDlayerMaskEl)
225       continue;
226
227     if ((TMath::Abs(trdTrack->Pt()) >= fTRDptHQU) && (trdTrack->GetPID() >= fTRDpidHQU))
228       MarkCondition(kHQU, globalStack);
229
230     if ((TMath::Abs(trdTrack->Pt()) >= fTRDptHSE) && (trdTrack->GetPID() >= fTRDpidHSE))
231       MarkCondition(kHSE, globalStack);
232
233     if ((trdTrack->GetSector() >= fTRDminSectorHEE) && (trdTrack->GetSector() <= fTRDmaxSectorHEE) &&
234         (TMath::Abs(trdTrack->Pt()) >= fTRDptHEE) && (trdTrack->GetPID() >= fTRDpidHEE))
235       MarkCondition(kHEE, globalStack);
236   }
237
238   // check if HJT condition is fulfilled in any stack
239   for (Int_t iStack = 0; iStack < 90; ++iStack) {
240     if (nTracks[iStack] >= fTRDnHJT) {
241       MarkCondition(kHJT, iStack);
242       break;
243     }
244   }
245
246   return kTRUE;
247 }