1 |
/** |
2 |
* \file CaloFranzini.cpp |
3 |
* \author Emiliano Mocchiutti (2007/07/18) |
4 |
*/ |
5 |
// |
6 |
// headers |
7 |
// |
8 |
#include <CaloFranzini.h> |
9 |
//-------------------------------------- |
10 |
/** |
11 |
* Default constructor |
12 |
*/ |
13 |
CaloFranzini::CaloFranzini(){ |
14 |
Clear(); |
15 |
} |
16 |
|
17 |
CaloFranzini::CaloFranzini(PamLevel2 *l2p){ |
18 |
// |
19 |
file = NULL; |
20 |
brig = NULL; |
21 |
nbin = 0; |
22 |
// |
23 |
L2 = l2p; |
24 |
// |
25 |
if ( !L2->IsORB() ) printf(" WARNING: OrbitalInfo Tree is needed, the plugin could not work properly without it \n"); |
26 |
// |
27 |
// Default variables |
28 |
// |
29 |
debug = false; |
30 |
dolong = true; |
31 |
dofull = false; |
32 |
// |
33 |
Clear(); |
34 |
// |
35 |
} |
36 |
|
37 |
void CaloFranzini::Clear(){ |
38 |
// |
39 |
OBT = 0; |
40 |
PKT = 0; |
41 |
atime = 0; |
42 |
longtzeta = 0.; |
43 |
fulltzeta = 0.; |
44 |
memset(estrip, 0, 2*22*96*sizeof(Float_t)); |
45 |
memset(qplane, 0, 2*22*sizeof(Float_t)); |
46 |
// |
47 |
} |
48 |
|
49 |
void CaloFranzini::Print(){ |
50 |
// |
51 |
Process(); |
52 |
// |
53 |
printf("========================================================================\n"); |
54 |
printf(" OBT: %u PKT: %u ATIME: %u \n",OBT,PKT,atime); |
55 |
printf(" debug [debug flag]:.. %i\n",debug); |
56 |
printf(" longtzeta :.. %f\n",longtzeta); |
57 |
printf(" fulltzeta :.. %f\n",fulltzeta); |
58 |
printf("========================================================================\n"); |
59 |
// |
60 |
} |
61 |
|
62 |
void CaloFranzini::Delete(){ |
63 |
if ( file ) file->Close(); |
64 |
Clear(); |
65 |
printf(" ERROR: CaloFranzini variables _NOT_ filled \n"); |
66 |
}; |
67 |
// |
68 |
if ( debug ) this->Print(); |
69 |
if ( debug ) printf(" exit \n"); |
70 |
// |
71 |
} |
72 |
|
73 |
Bool_t CaloFranzini::CreateMatrixFile(TString matrixfile){ |
74 |
// |
75 |
file = new TFile(matrixfile.Data(),"READ"); |
76 |
// |
77 |
if ( !file || file->IsZombie() ){ |
78 |
file = new TFile(matrixfile.Data(),"RECREATE"); |
79 |
} else { |
80 |
printf(" ERROR: file %s already existing!\n Choose another name or delete the old file\n",matrixfile.Data()); |
81 |
return(false); |
82 |
}; |
83 |
// |
84 |
return(true); |
85 |
// |
86 |
} |
87 |
|
88 |
void CaloFranzini::WriteNumBin(Int_t numbin){ |
89 |
TArrayI nbi(1, &numbin); |
90 |
file->WriteObject(&nbi, "nbinenergy"); |
91 |
} |
92 |
|
93 |
void CaloFranzini::WriteRigBin(TArrayF *rigbin){ |
94 |
file->WriteObject(&rigbin, "binrig"); |
95 |
} |
96 |
|
97 |
void CaloFranzini::WriteLongMean(TArrayF *qpl, Int_t bin){ |
98 |
TString name = Form("qplmeann%i",bin); |
99 |
file->WriteObject(&qpl,name.Data()); |
100 |
} |
101 |
|
102 |
void CaloFranzini::WriteLongMatrix(TMatrixD *mat, Int_t bin){ |
103 |
TString name = Form("matrixn%i",bin); |
104 |
file->WriteObject(&(*mat),name.Data()); |
105 |
} |
106 |
|
107 |
void CaloFranzini::CloseMatrixFile(){ |
108 |
file->Close(); |
109 |
} |
110 |
|
111 |
|
112 |
Bool_t CaloFranzini::Open(TString matrixfile){ |
113 |
// |
114 |
// find matrix file |
115 |
// |
116 |
if ( !strcmp(matrixfile.Data(),"") ){ |
117 |
if (dolong) matrixfile = (TString)gSystem->ExpandPathName("$PAM_CALIB")+"/cal-param/covmatrix_longel.root"; |
118 |
if (dofull) matrixfile = (TString)gSystem->ExpandPathName("$PAM_CALIB")+"/cal-param/covmatrix_fullel.root"; |
119 |
}; |
120 |
// |
121 |
file = new TFile(matrixfile.Data(),"READ"); |
122 |
// |
123 |
if ( !file || file->IsZombie() ){ |
124 |
printf(" ERROR: cannot open file %s \n",matrixfile.Data()); |
125 |
return(false); |
126 |
}; |
127 |
// |
128 |
TArrayI *numbin = (TArrayI*)file->Get("nbinenergy"); |
129 |
if ( !numbin ){ |
130 |
printf(" ERROR: cannot read number of bins from file %s \n",matrixfile.Data()); |
131 |
return(false); |
132 |
}; |
133 |
nbin = (Int_t)numbin->At(0); |
134 |
if ( nbin <= 0 ){ |
135 |
printf(" ERROR: cannot work with 0 energy bins (from file %s) \n",matrixfile.Data()); |
136 |
return(false); |
137 |
}; |
138 |
// |
139 |
brig = (TArrayF*)file->Get("binrig"); |
140 |
if ( !brig ){ |
141 |
printf(" ERROR: cannot read rigidity binning from file %s \n",matrixfile.Data()); |
142 |
return(false); |
143 |
}; |
144 |
// |
145 |
return(true); |
146 |
// |
147 |
} |
148 |
|
149 |
TMatrixD *CaloFranzini::LoadCovarianceMatrix(Float_t rig){ |
150 |
// |
151 |
TString name; |
152 |
for (Int_t i = 0; i<nbin-1; i++){ |
153 |
if ( rig>=brig->At(i) && rig < brig->At(i+1) ){ |
154 |
name = Form("matrixn%i",i); |
155 |
break; |
156 |
}; |
157 |
}; |
158 |
if ( rig < brig->At(0) ){ |
159 |
printf(" WARNING: Event with rigidity lower than the first covariance matrix bin (rig = %f, lower limit = %f)\n",rig,brig->At(0)); |
160 |
name = "matrixn0"; |
161 |
printf(" Using matrix %s \n",name.Data()); |
162 |
}; |
163 |
if ( rig >= brig->At(nbin) ){ |
164 |
printf(" WARNING: Event with rigidity higher than the last covariance matrix bin (rig = %f, upper limit = %f)\n",rig,brig->At(nbin)); |
165 |
name = Form("matrixn%i",nbin-1); |
166 |
printf(" Using matrix %s \n",name.Data()); |
167 |
}; |
168 |
// |
169 |
TMatrixD *matrix = (TMatrixD*)file->Get(name.Data()); |
170 |
// |
171 |
return(matrix); |
172 |
// |
173 |
} |
174 |
|
175 |
|
176 |
TArrayF *CaloFranzini::LoadLongAverage(Float_t rig){ |
177 |
// |
178 |
TString name; |
179 |
for (Int_t i = 0; i<nbin-1; i++){ |
180 |
if ( rig>=brig->At(i) && rig < brig->At(i+1) ){ |
181 |
name = Form("qplmeann%i",i); |
182 |
break; |
183 |
}; |
184 |
}; |
185 |
if ( rig < brig->At(0) ){ |
186 |
printf(" WARNING: Event with rigidity lower than the first qplmean bin (rig = %f, lower limit = %f)\n",rig,brig->At(0)); |
187 |
name = "qplmeann0"; |
188 |
printf(" Using qplmean %s \n",name.Data()); |
189 |
}; |
190 |
if ( rig >= brig->At(nbin) ){ |
191 |
printf(" WARNING: Event with rigidity higher than the last qplmean bin (rig = %f, upper limit = %f)\n",rig,brig->At(nbin)); |
192 |
name = Form("qplmeann%i",nbin-1); |
193 |
printf(" Using qplmean %s \n",name.Data()); |
194 |
}; |
195 |
// |
196 |
TArrayF *qplmean = (TArrayF*)file->Get(name.Data()); |
197 |
// |
198 |
return(qplmean); |
199 |
// |
200 |
} |