-
Notifications
You must be signed in to change notification settings - Fork 37
/
SBSBigBite.h
269 lines (210 loc) · 9.26 KB
/
SBSBigBite.h
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
#ifndef SBSBigBite_h
#define SBSBigBite_h
#include "THaSpectrometer.h"
class TList;
class THaTrack;
class TH2D;
class SBSBigBite : public THaSpectrometer {
public:
SBSBigBite( const char *name, const char *description );
virtual ~SBSBigBite();
virtual void Clear( Option_t* opt="");
virtual Int_t CoarseReconstruct();
virtual Int_t CoarseTrack();
virtual Int_t Reconstruct();
virtual Int_t Track();
virtual Int_t CalcPID();
virtual Int_t FindVertices( TClonesArray& tracks );
virtual Int_t TrackCalc();
//copied from THaHRS...
Bool_t GetTrSorting() const;
Bool_t SetTrSorting( Bool_t set = false );
Bool_t GetMultiTracks() const;
Bool_t SetMultiTracks( Bool_t set = false );
Bool_t GetUseBeamPosInOptics() const { return fUseBeamPosInOptics; }
void SetUseBeamPosInOptics( bool val=true ){ fUseBeamPosInOptics = val; }
//virtual Int_t Begin( THaRunBase* r=0 );
//virtual Int_t End( THaRunBase* r=0 );
protected:
virtual Int_t ReadDatabase( const TDatime& date );
virtual Int_t ReadRunDatabase( const TDatime& date );
virtual Int_t DefineVariables( EMode mode = kDefine );
virtual void DefinePidParticles();
void CalcOpticsCoords( THaTrack* the_track );//calculate optics coords from "focal plane" coords
void CalcTargetCoords( THaTrack* the_track );//calculate target coords from optics coords
void CalcFpCoords( THaTrack* the_track ); //
void CalcTrackTiming( THaTrack* the_track );
void CalcTrackPID( THaTrack* the_track );
Int_t proba_pssh(Double_t eps_etot_ratio,
Double_t& proba_e, Double_t& proba_pi);
Int_t proba_pcal(Double_t etot_p_ratio,
Double_t& proba_e, Double_t& proba_pi);
Int_t proba_grinch(Int_t npmt, Double_t p,
Double_t& proba_e, Double_t& proba_pi);
void InitOpticsAxes(double, const TVector3 & );
void InitOpticsAxes(double); //version with only bend angle argument
void InitGEMAxes(double, double, const TVector3 & );
void InitGEMAxes(double, double); //version with only angle arguments:
//Version using more general x,y,z axis rotations:
void InitGEMAxes(double, double, double, const TVector3 & );
void InitGEMAxes(double, double, double);
// My current understanding is that fPointingOffset designates the point
// towards the beamline that the "mouth" of the spectrometer is pointing to...
// which we might also need to define.
// Hence, we might want to define an additional set of parameters:
// detector stack pitch, yaw, roll + actual position of first GEM tracker?
// These angles are wrt the "ideal" central ray
// double fDetectorStackPitch;
// double fDetectorStackYaw;
// double fDetectorStackRoll;
// AJRP: reworking these parameters to be more consistent with the zero-field alignment code:
Double_t fMagDist; //We could use "colldist" from THaSpectrometer for this, but that would be needlessly confusing
Double_t fGEMtheta; //polar angle of GEM z axis wrt TARGET transport coordinates
Double_t fGEMphi; //azimuthal angle of GEM z axis wrt TARGET transport coordinates
Double_t fGEMax; //GEM x axis rotation
Double_t fGEMay; //GEM y axis rotation
Double_t fGEMaz; //GEM z axis rotation
TVector3 fGEMorigin; //Absolute position of GEM origin relative to target center, in TARGET transport coordinates
TVector3 fGEMxaxis_global;
TVector3 fGEMyaxis_global;
TVector3 fGEMzaxis_global;
TVector3 fOpticsOrigin; //Give origin of ideal optics system
double fOpticsAngle; //Ideal central bend angle of GEM wrt BigBite
TVector3 fOpticsXaxis_global;
TVector3 fOpticsYaxis_global;
TVector3 fOpticsZaxis_global;
//TRotation fOpt2DetRot;// transformation from optics (ideal) to detector (actual)
//TRotation fDet2OptRot;// transformation from detector (actual) to optics (ideal)
bool fDownBendingMode; //Default = false:
UInt_t fPrecon_flag; //Indicate which momentum reconstruction formalism we are using:
// 0 (default) = expansion of p*thetabend vs fp x, y, x' y'
// 1 = expansion as p*thetabend = pth(firstorder) * ( 1 + delta ), where
// delta is expanded in terms of fp variables and p*thbend(firstorder) = A*(1+(B+C*magdist)*xptarget)
// Parameters of (optional) first-order model for BigBite momentum reconstruction:
// (p * thetabend)_1st_order = A*( 1 + (B+C*magdist)*xptarget);
// (p * thetabend) = (p*thetabend)_1st_order * ( 1 + delta ); here delta is an expansion in terms of focal-plane variables:
Double_t fA_pth1; // default value (from simulation) is 0.28615 * 0.97
Double_t fB_pth1; // default value (from simulation) is 0.1976
Double_t fC_pth1; // default value (from simulation) is 0.4764
//Extra coefficents to remove the beam dependance
Double_t fA_vy;
Double_t fB_vy;
bool fIsMC;
bool fUseBeamPosInOptics; //default false;
int fOpticsOrder;
std::vector<double> fb_xptar;
std::vector<double> fb_yptar;
std::vector<double> fb_ytar;
std::vector<double> fb_pinv;
//AJRP: changed the exponents to integers here for speed:
std::vector<int> f_oi;
std::vector<int> f_oj;
std::vector<int> f_ok;
std::vector<int> f_ol;
std::vector<int> f_om;
//Build out the same infrastructure for single-arm electron TOF (basically path length):
Double_t fETOF_avg;
int fETOF_order;
std::vector<double> fb_ETOF;
std::vector<int> f_oi_ETOF;
std::vector<int> f_oj_ETOF;
std::vector<int> f_ok_ETOF;
std::vector<int> f_ol_ETOF;
std::vector<int> f_om_ETOF;
//Only relevant if downbending optics are defined via the DB:
int fOpticsOrderDownbend;
std::vector<double> fb_xptar_downbend;
std::vector<double> fb_yptar_downbend;
std::vector<double> fb_ytar_downbend;
std::vector<double> fb_pinv_downbend;
//AJRP: changed the exponents to integers here for speed:
std::vector<int> f_oi_downbend;
std::vector<int> f_oj_downbend;
std::vector<int> f_ok_downbend;
std::vector<int> f_ol_downbend;
std::vector<int> f_om_downbend;
bool fUseForwardOptics; //default to false: turning this on will enable forward optics-based track search constraints
int fForwardOpticsOrder;
std::vector<double> fb_xfp;
std::vector<double> fb_yfp;
std::vector<double> fb_xpfp;
std::vector<double> fb_ypfp;
//AJRP: changed the exponents to integers here for speed:
std::vector<int> f_foi;
std::vector<int> f_foj;
std::vector<int> f_fok;
std::vector<int> f_fol;
std::vector<int> f_fom;
Double_t fPtheta_00000;
Double_t fPtheta_10000;
Double_t fPtheta_00100;
//Double_t fXptar_00000;
Double_t fXptar_10000;
Double_t fXptar_00100;
Double_t fYtar_01000;
Double_t fYtar_00010;
//std::vector<double> f_xtg_exp;
Double_t fFrontConstraintWidthX;
Double_t fFrontConstraintWidthY;
Double_t fBackConstraintWidthX;
Double_t fBackConstraintWidthY;
//Let's add some handy-dandy offsets for the front point only,
//so we can center the peaks at zero, tighten up the windows,
//and get tracking to run faster until things are better calibrated:
Double_t fFrontConstraintX0;
Double_t fFrontConstraintY0;
//Might as well add back constraint offsets as well to account for a possible mismatch between
//shower cluster position and track position:
Double_t fBackConstraintX0;
Double_t fBackConstraintY0;
// Angle constraint widths:
Double_t fConstraintWidth_theta;
Double_t fConstraintWidth_phi;
//The following parameters are used to calculate the weights in the constraint point calculation:
Double_t fSigmaX_shower; //default = block size /12
Double_t fSigmaY_shower; //default = block size /12
Double_t fSigmaX_preshower; //default = block size /12
Double_t fSigmaY_preshower; //default = block size /12
Double_t fSigmaX_hodo; //default = bar vertical size / sqrt(12)
Double_t fSigmaY_hodo; //default = hit resolution from left/right time difference
//for output only... Vectors instead?
std::vector<double> fFrontConstraintX;
std::vector<double> fFrontConstraintY;
std::vector<double> fFrontConstraintZ;
std::vector<double> fBackConstraintX;
std::vector<double> fBackConstraintY;
std::vector<double> fBackConstraintZ;
std::vector<double> fEpsEtotRatio;
std::vector<double> fEtot;
std::vector<double> fEtotPratio;
double fTrackGrinchClusCorr_0;
double fTrackGrinchClusCorr_1;
double fTrackGrinchClusCorr_Sigma;
std::vector<double> fEpsEtotRatio_table;
std::vector<double> fProba_e_PSSH_table;
std::vector<double> fProba_pi_PSSH_table;
std::vector<double> fEtotPratio_table;
std::vector<double> fProba_e_PCAL_table;
std::vector<double> fProba_pi_PCAL_table;
std::vector<double> fP_table;
std::vector<double> fNGRINCHPMTs_table;
std::vector<double> fProba_e_GRINCH_table;
std::vector<std::vector<double>> fProba_pi_GRINCH_table;
std::vector<double> fProbaE;
std::vector<double> fProbaPi;
//This is now redundant with fOpticsAngle
// double fTrackerPitchAngle;
TH2D* h1_yVx_bcp;
TH2D* h1_x_fcpVbcp;
TH2D* h1_yVx_fcp;
TH2D* h1_y_fcpVbcp;
TH2D* h1_dyVdx;
double fECaloFudgeFactor;// poor man's solution to apply the calorimeter constraint
enum {
kMultiTracks = BIT(13), // Tracks are to be sorted by chi2
kSortTracks = BIT(14), // Tracks are to be sorted by chi2
kAutoStdDets = BIT(15) // Auto-create standard detectors if no "vdc"
};
ClassDef(SBSBigBite,0) // BigBite spectrometer
};
#endif//ROOT_TreeSearch_SBSBigBite