Transition to NewIO
[u/mrichter/AliRoot.git] / MUON / AliMUONTriggerLut.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, 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 /* $Id$ */
17
18 #include "AliMUONTriggerCircuit.h"
19 #include "AliMUONTriggerLut.h"
20 #include "TTree.h"
21 #include "AliRun.h"
22 #include "AliMUON.h"
23 #include "TMath.h"
24 #include "TFile.h"
25 #include "TROOT.h"
26 #include "TH3.h"
27 #include <Riostream.h>
28
29 ClassImp(AliMUONTriggerLut)
30
31 //----------------------------------------------------------------------
32 AliMUONTriggerLut::AliMUONTriggerLut() {
33 // constructor
34   fLptPlus = fLptMinu = fLptUnde = 0;
35   fHptPlus = fHptMinu = fHptUnde = 0;
36   fAptPlus = fAptMinu = fAptUnde = 0;
37 }
38 //----------------------------------------------------------------------
39 AliMUONTriggerLut::~AliMUONTriggerLut() {
40 // destructor
41   delete fLptPlus;  
42   delete fLptMinu;
43   delete fLptUnde;
44   delete fHptPlus;  
45   delete fHptMinu;
46   delete fHptUnde;
47   delete fAptPlus;  
48   delete fAptMinu;
49   delete fAptUnde;
50   fLptPlus = fLptMinu = fLptUnde = 0;
51   fHptPlus = fHptMinu = fHptUnde = 0;  
52   fAptPlus = fAptMinu = fAptUnde = 0;    
53 }
54
55 //----------------------------------------------------------------------
56 AliMUONTriggerLut::AliMUONTriggerLut (const AliMUONTriggerLut& MUONTriggerLut)
57 {
58 // Dummy copy constructor
59 }
60
61 //----------------------------------------------------------------------
62 AliMUONTriggerLut & AliMUONTriggerLut::operator=(const AliMUONTriggerLut& MUONTriggerLut)
63 {
64 // Dummy assignment operator
65     return *this;
66 }
67
68 //----------------------------------------------------------------------
69 void AliMUONTriggerLut::GetLutOutput(Int_t circuit, Int_t xstrip, Int_t idev,
70                                      Int_t ystrip, Int_t lutLpt[2], 
71                                      Int_t lutHpt[2], Int_t lutApt[2]){
72 // return output of LuT for corresponding TH3S  
73
74   static TFile *fileLut;
75   static Bool_t first=kTRUE;  
76   if(first) {
77     printf(" opening MUONTriggerLut.root \n");
78     fileLut = new TFile("$(ALICE_ROOT)/MUON/MUONTriggerLut.root","READ");
79     first=kFALSE;
80   }
81   fileLut->cd();
82
83 // get the pointers to the TH3S objects of the file
84   TH3S *lptPlus = (TH3S*)gROOT->FindObject("LptPlus");  
85   TH3S *lptMinu = (TH3S*)gROOT->FindObject("LptMinu");
86   TH3S *lptUnde = (TH3S*)gROOT->FindObject("LptUnde");
87   TH3S *hptPlus = (TH3S*)gROOT->FindObject("HptPlus");  
88   TH3S *hptMinu = (TH3S*)gROOT->FindObject("HptMinu");
89   TH3S *hptUnde = (TH3S*)gROOT->FindObject("HptUnde");
90   TH3S *aptPlus = (TH3S*)gROOT->FindObject("AptPlus");  
91   TH3S *aptMinu = (TH3S*)gROOT->FindObject("AptMinu");
92   TH3S *aptUnde = (TH3S*)gROOT->FindObject("AptUnde");
93
94   Int_t bin;
95   Short_t binc; 
96   Int_t mask = GetMask(ystrip);        // get ystrip mask
97   
98   // Low pt.............................................. 
99   bin    =          lptPlus->GetBin(circuit,xstrip,idev);
100   binc   = (Short_t)lptPlus->GetBinContent(bin);
101   if ((binc & mask)!=0) lutLpt[1]=1;
102
103   bin    =          lptMinu->GetBin(circuit,xstrip,idev);
104   binc   = (Short_t)lptMinu->GetBinContent(bin);
105   if ((binc & mask)!=0) lutLpt[0]=1;
106   
107   bin    =          lptUnde->GetBin(circuit,xstrip,idev);
108   binc   = (Short_t)lptUnde->GetBinContent(bin);
109   if ((binc & mask)!=0) lutLpt[0]=lutLpt[1]=1;
110
111   // High pt.............................................
112   bin    =          hptPlus->GetBin(circuit,xstrip,idev);
113   binc   = (Short_t)hptPlus->GetBinContent(bin);
114   if ((binc & mask)!=0) lutHpt[1]=1;
115
116   bin    =          hptMinu->GetBin(circuit,xstrip,idev);
117   binc   = (Short_t)hptMinu->GetBinContent(bin);
118   if ((binc & mask)!=0) lutHpt[0]=1;
119
120   bin    =          hptUnde->GetBin(circuit,xstrip,idev);
121   binc   = (Short_t)hptUnde->GetBinContent(bin);
122   if ((binc & mask)!=0) lutHpt[0]=lutHpt[1]=1;
123
124   // All pts.............................................
125   bin    =          aptPlus->GetBin(circuit,xstrip,idev);
126   binc   = (Short_t)aptPlus->GetBinContent(bin);
127   if ((binc & mask)!=0) lutApt[1]=1;
128
129   bin    =          aptMinu->GetBin(circuit,xstrip,idev);
130   binc   = (Short_t)aptMinu->GetBinContent(bin);
131   if ((binc & mask)!=0) lutApt[0]=1;
132
133   bin    =          aptUnde->GetBin(circuit,xstrip,idev);
134   binc   = (Short_t)aptUnde->GetBinContent(bin);
135   if ((binc & mask)!=0) lutApt[0]=lutApt[1]=1;
136
137 // get back to the first file
138 /* no need for this with NewIO
139   TTree *tK = gAlice->TreeK();
140   TFile *file1 = 0;
141   if (tK) file1 = tK->GetCurrentFile();
142   file1->cd();
143 */
144 }
145
146 //----------------------------------------------------------------------
147 Int_t AliMUONTriggerLut::GetMask(Int_t ystrip){
148 // returns the mask corresponding to ystrip
149   Int_t tabMask[16]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
150   Int_t mask=0;
151   tabMask[ystrip]=1;
152   for (Int_t i=0; i<16; i++) {          
153     mask=mask+Int_t(tabMask[i]*TMath::Power(2,i));   
154   }
155   return mask;
156 }
157
158 //----------------------------------------------------------------------
159 void AliMUONTriggerLut::LoadLut(){
160 // !!!!!!! This is dummy version of the LoadLut method !!!!!!!
161 // !!!!!!!         calibration to be done              !!!!!!!
162 // 1) Loop on circuit/Xstrip1/deviation/Ystrip
163 // 2) get corresponding ptCal from AliMUONTriggerCircuit
164 // 3) fill histos with cuts on deviation, ptLow and ptHigh 
165 // 4) store histos in a file
166
167   char fileName[60];
168   sprintf(fileName,"$(ALICE_ROOT)/MUON/MUONTriggerLut.root");
169   printf(" file name is %s\n",fileName);
170
171 // open output file containing histos  
172   TFile *hfile = new TFile(fileName,"RECREATE","Trigger Look Up Table");
173
174   //..........................................circuit/stripX/deviation
175   TH3S *fLptPlus=new TH3S("LptPlus","LptPlus",234,0,234,31,0,31,31,0,31);
176   TH3S *fLptMinu=new TH3S("LptMinu","LptMinu",234,0,234,31,0,31,31,0,31);
177   TH3S *fLptUnde=new TH3S("LptUnde","LptUnde",234,0,234,31,0,31,31,0,31);
178
179   TH3S *fHptPlus=new TH3S("HptPlus","HptPlus",234,0,234,31,0,31,31,0,31);
180   TH3S *fHptMinu=new TH3S("HptMinu","HptMinu",234,0,234,31,0,31,31,0,31);
181   TH3S *fHptUnde=new TH3S("HptUnde","HptUnde",234,0,234,31,0,31,31,0,31);
182
183   TH3S *fAptPlus=new TH3S("AptPlus","AptPlus",234,0,234,31,0,31,31,0,31);
184   TH3S *fAptMinu=new TH3S("AptMinu","AptMinu",234,0,234,31,0,31,31,0,31);
185   TH3S *fAptUnde=new TH3S("AptUnde","AptUnde",234,0,234,31,0,31,31,0,31);
186   
187   Float_t lptTreshold=0.75;
188   Float_t hptTreshold=1.75;
189   
190   AliMUON *pMUON  = (AliMUON*)gAlice->GetModule("MUON");  
191   AliMUONTriggerCircuit* triggerCircuit;
192
193   for (Int_t icirc=0; icirc<234; icirc++) {
194     printf(" Loading LuT for circuit %d \n",icirc);
195     triggerCircuit = &(pMUON->TriggerCircuit(icirc));         
196
197     for (Int_t istripX=0; istripX<31; istripX++) {
198       for (Int_t idev=0; idev<31; idev++) {
199         
200         Short_t iLptPlus, iLptMinu, iLptUnde;
201         Short_t iHptPlus, iHptMinu, iHptUnde;
202         Short_t iAptPlus, iAptMinu, iAptUnde;
203         iLptPlus = iLptMinu = iLptUnde = 0;
204         iHptPlus = iHptMinu = iHptUnde = 0;
205         iAptPlus = iAptMinu = iAptUnde = 0;
206         
207         for (Int_t istripY=0; istripY<16; istripY++) {
208           Float_t pt=triggerCircuit->PtCal(istripX,idev,istripY);
209           
210           if (pt>lptTreshold) {
211             if (idev<15)       iLptMinu=iLptMinu+Int_t(TMath::Power(2,istripY));
212             else if (idev==15) iLptUnde=iLptUnde+Int_t(TMath::Power(2,istripY));
213             else if (idev>15)  iLptPlus=iLptPlus+Int_t(TMath::Power(2,istripY));
214           }
215           if (pt>hptTreshold) {
216             if (idev<15)       iHptMinu=iHptMinu+Int_t(TMath::Power(2,istripY));
217             else if (idev==15) iHptUnde=iHptUnde+Int_t(TMath::Power(2,istripY));
218             else if (idev>15)  iHptPlus=iHptPlus+Int_t(TMath::Power(2,istripY));
219           }
220           if (idev<15)       iAptMinu=iAptMinu+Int_t(TMath::Power(2,istripY));
221           else if (idev==15) iAptUnde=iAptUnde+Int_t(TMath::Power(2,istripY));
222           else if (idev>15)  iAptPlus=iAptPlus+Int_t(TMath::Power(2,istripY));
223
224         } // loop on istripY
225
226         Int_t bin; 
227         
228         bin = fLptMinu->GetBin(icirc,istripX,idev);
229         fLptMinu->SetBinContent(bin,iLptMinu);
230         bin = fLptUnde->GetBin(icirc,istripX,idev);
231         fLptUnde->SetBinContent(bin,iLptUnde);
232         bin = fLptPlus->GetBin(icirc,istripX,idev);
233         fLptPlus->SetBinContent(bin,iLptPlus);
234
235         bin = fHptMinu->GetBin(icirc,istripX,idev);
236         fHptMinu->SetBinContent(bin,iHptMinu);
237         bin = fHptUnde->GetBin(icirc,istripX,idev);
238         fHptUnde->SetBinContent(bin,iHptUnde);
239         bin = fHptPlus->GetBin(icirc,istripX,idev);
240         fHptPlus->SetBinContent(bin,iHptPlus);
241         
242         bin = fAptMinu->GetBin(icirc,istripX,idev);
243         fAptMinu->SetBinContent(bin,iAptMinu);
244         bin = fAptUnde->GetBin(icirc,istripX,idev);
245         fAptUnde->SetBinContent(bin,iAptUnde);
246         bin = fAptPlus->GetBin(icirc,istripX,idev);
247         fAptPlus->SetBinContent(bin,iAptPlus);
248           
249       } // loop on idev
250     } // loop on istripX
251   } // loop on circuit
252
253   hfile->Write();
254   hfile->Close();
255 }
256
257
258
259
260
261
262