Rev 54 | Rev 71 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
25 | f9daq | 1 | #ifndef CVODNIK_H |
2 | #define CVODNIK_H |
||
3 | |||
4 | #include "TROOT.h" |
||
5 | #include "TVector3.h" |
||
6 | #include "TMath.h" |
||
7 | #include "TPolyLine3D.h" |
||
8 | #include "TRandom.h" |
||
9 | #include "TDatime.h" |
||
10 | #include "TH1F.h" |
||
11 | #include "TH2F.h" |
||
12 | #include "TColor.h" |
||
13 | |||
14 | |||
15 | |||
16 | #define MARGIN 1.0e-6 |
||
17 | #define DEGREE 180.0/3.14159265358979312 |
||
54 | f9daq | 18 | #define MAX_REFLECTIONS 19 |
25 | f9daq | 19 | |
54 | f9daq | 20 | |
25 | f9daq | 21 | using namespace TMath; |
22 | |||
23 | const double c_reflectivity = 0.96; |
||
24 | const int REFRACTION = 1; |
||
25 | const int REFLECTION = 2; |
||
26 | const int TOTAL = 3; |
||
54 | f9daq | 27 | const int ABSORBED = -2; |
25 | f9daq | 28 | const TVector3 CENTER(-2.0, 0.0, 0.0); |
29 | const int nch = 50; // number of "channels" used as number of bins in histograms |
||
30 | |||
54 | f9daq | 31 | enum Fate {missed=0, hitExit=1, refracted=-1, enter=2, noreflection=-2, backreflected=-3, rays=3, absorbed=4}; |
25 | f9daq | 32 | |
33 | int dbg = 0; |
||
34 | |||
35 | //================================================================================= |
||
36 | // zarek (premica) |
||
37 | class CRay |
||
38 | { |
||
39 | public: |
||
40 | CRay() : |
||
41 | r(TVector3(0,0,0)), |
||
42 | n(TVector3(1,0,0)), |
||
43 | p(TVector3(0,1,0)), |
||
44 | color(1) |
||
45 | {}; |
||
46 | CRay(TVector3 r0, TVector3 n0) : |
||
47 | r(r0), |
||
48 | n(n0.Unit()), |
||
49 | p(TVector3(0,1,0)), |
||
50 | color(1) |
||
51 | {}; |
||
52 | CRay(double x0, double y0, double z0, double l0, double m0, double n0) : |
||
53 | r(TVector3(x0,y0,z0)), |
||
54 | n(TVector3(l0,m0,n0).Unit()), |
||
55 | p(TVector3(0,1,0)), |
||
56 | color(1) |
||
57 | {}; |
||
58 | |||
59 | void Set(TVector3 r0, TVector3 n0); |
||
60 | //void Set(double x0, double y0, double z0, double l0, double m0, double n0); |
||
61 | void SetColor(int c){color = c;}; |
||
62 | void SetPolarization(TVector3 p0) {p = p0.Unit();}; |
||
63 | |||
64 | //inline CRay & operator = (const CRay &); |
||
65 | |||
66 | TVector3 GetR() const {return r;}; |
||
67 | TVector3 GetN() const {return n;}; |
||
68 | TVector3 GetP() const {return p;}; |
||
69 | |||
70 | void Print(); |
||
71 | void Draw(); |
||
72 | void Draw(double x_from, double x_to); |
||
73 | void DrawS(double x_from, double t); |
||
74 | |||
75 | private: |
||
76 | TVector3 r; |
||
77 | TVector3 n; |
||
78 | TVector3 p; //r = point on line, n = normal, p = polarization |
||
79 | int color; |
||
80 | }; |
||
81 | //================================================================================= |
||
82 | |||
83 | |||
84 | //================================================================================= |
||
85 | // ravnina, definirana in omejena s 4 tockami na njej (stirikotni poligon) |
||
86 | // zanasam se na taksno definicijo: (normala kaze noter) |
||
87 | // r1----->r2 |
||
88 | // ^ | |
||
89 | // | | |
||
90 | // | | |
||
91 | // | � |
||
92 | // r0<-----r3 |
||
93 | class CPlane4 |
||
94 | { |
||
95 | |||
96 | public: |
||
97 | CPlane4(); |
||
98 | CPlane4(TVector3 r1, TVector3 r2, TVector3 r3, TVector3 r4); |
||
99 | CPlane4(TVector3 *vr); |
||
100 | void Set(TVector3 r1, TVector3 r2, TVector3 r3, TVector3 r4); |
||
101 | void Set(TVector3 *vr) {Set(vr[0], vr[1], vr[2], vr[3]);}; |
||
102 | void FlipN(){n = -n;}; |
||
103 | |||
104 | int GetIntersection(TVector3 *vec, CRay ray); |
||
105 | int IsInTri(TVector3 vec, TVector3 e1, TVector3 e2, double); |
||
106 | int IsVectorIn(TVector3 vec); |
||
107 | int TestIntersection(CRay in); // ray in & ??? plane |
||
108 | int TestIntersection(TVector3 *vec, CRay in); // plane defined with normal vec and ray in |
||
109 | |||
110 | TVector3 GetN() {return n;}; |
||
111 | |||
112 | void Print(); |
||
113 | void Draw(int color = 1, int width = 1); |
||
114 | |||
115 | private: |
||
116 | TVector3 r[4], n; |
||
117 | double A, B, C, D; |
||
118 | TVector3 edge[4]; // vektorji stranic |
||
119 | double angle_r[4]; // koti ob posameznem vogalu |
||
120 | }; |
||
121 | //================================================================================= |
||
122 | |||
123 | |||
124 | //================================================================================= |
||
125 | // ravna opticna povrsina: refractor, zrcalo ali povrsina s totalnim odbojem |
||
126 | #define SURF_DUMMY 0 |
||
127 | #define SURF_REFRA 1 |
||
128 | #define SURF_REFLE 2 |
||
129 | #define SURF_TOTAL 3 |
||
130 | //#define SURF_TOTAL 1 |
||
131 | #define SURF_IMPER 4 |
||
132 | //#define SURF_DIFUS 5 |
||
133 | |||
134 | class CSurface : public CPlane4 |
||
135 | { |
||
136 | |||
137 | public: |
||
138 | CSurface(int type0 = 0); |
||
139 | CSurface(int type0, TVector3 r1, TVector3 r2, TVector3 r3, TVector3 r4, |
||
140 | double n10, double n20, double reflectivity); |
||
141 | CSurface(int type0, TVector3 *vr, double n10, double n20, double reflectivity); |
||
142 | |||
143 | void SetV(TVector3 *vr){CPlane4::Set(vr);}; |
||
144 | void SetType(int type0){type = type0;}; |
||
145 | void SetIndex(double n10, double n20); |
||
146 | void SetReflection(double reflectivity){reflection = reflectivity;}; |
||
147 | void SetFresnel(int f = 1) {fresnel = f;}; |
||
148 | |||
149 | void Set(int type0, TVector3 *vr, double n10, double n20, double reflectivity) |
||
150 | {type = type0; CPlane4::Set(vr); SetIndex(n10, n20); reflection = reflectivity;}; |
||
151 | void Set(int type0, TVector3 r1, TVector3 r2, TVector3 r3, TVector3 r4, double n10, double n20, double reflectivity) |
||
152 | {type = type0; CPlane4::Set(r1, r2, r3, r4); SetIndex(n10, n20); reflection = reflectivity;}; |
||
153 | |||
154 | int PropagateRay(CRay in, CRay *out, TVector3 *intersection); |
||
155 | |||
156 | double N1_N2(int sign) { return ((sign > 0) ? n1/n2 : n2/n1);}; |
||
157 | |||
158 | void Print(){printf("Type = %d\n", type); CPlane4::Print();}; |
||
159 | |||
160 | private: |
||
161 | int type; //0 = dummy; 1 = refractor; 2 = reflector; 3 = total reflection |
||
162 | double n1, n2, n1_n2; //index of refraction, n1 @ +normal, n2 @ -normal |
||
163 | double reflection; // odbojnost stranic |
||
164 | TRandom rand; // za racunanje verjetnosti odboja od zrcala |
||
165 | double cosTtotal; //cosinus mejnega kota totalnega odboja za dana n1 in n2 |
||
166 | |||
167 | int fresnel; // ali naj uposteva Fresnelove enacbe |
||
168 | }; |
||
169 | //================================================================================= |
||
170 | |||
171 | |||
172 | //================================================================================= |
||
173 | // +------- n1 |
||
174 | // | ------- |
||
175 | // | n2 ----+ |
||
176 | // | | |
||
177 | // 0| M*SiPM | SiPM |
||
178 | // | | |
||
179 | // | ----+ |
||
180 | // | ------R |
||
181 | // +------- |
||
182 | // <-------- d -------> |
||
183 | // |
||
184 | |||
185 | |||
54 | f9daq | 186 | |
25 | f9daq | 187 | int RCol(int col, int solid); |
188 | |||
189 | |||
190 | class DetectorParameters |
||
191 | { |
||
192 | public: |
||
193 | DetectorParameters(double a, double b, double d, double active, double n1, double n2, double n3, TVector3 gap): |
||
194 | _a(a), |
||
195 | _b(b), |
||
196 | _d(d), |
||
197 | _active(active), |
||
198 | _n1(n1), |
||
199 | _n2(n2), |
||
200 | _n3(n3), |
||
201 | _gap(gap), |
||
70 | f9daq | 202 | _fresnel(1), |
203 | _guideOn(1), |
||
204 | offsetY(b/2.0), |
||
205 | offsetZ(b/2.0), |
||
206 | _plateOn(1), |
||
25 | f9daq | 207 | _plateWidth(1), |
208 | _glassOn(0), |
||
70 | f9daq | 209 | _glassD(0) |
25 | f9daq | 210 | {}; |
211 | DetectorParameters(double a, double b, double d): |
||
212 | _a(a), |
||
213 | _b(b), |
||
214 | _d(d), |
||
215 | _active(a), |
||
216 | _n1(1.0), |
||
70 | f9daq | 217 | _n2(1.53), |
218 | _n3(1.46), |
||
25 | f9daq | 219 | _gap(TVector3(0,0,0)), |
220 | _fresnel(1), |
||
221 | _guideOn(1), |
||
70 | f9daq | 222 | offsetY(b/2.0), |
223 | offsetZ(b/2.0), |
||
54 | f9daq | 224 | _plateOn(1), |
70 | f9daq | 225 | _plateWidth(1), |
226 | _glassOn(0), |
||
227 | _glassD(0) |
||
25 | f9daq | 228 | {}; |
229 | ~DetectorParameters() {}; |
||
230 | |||
70 | f9daq | 231 | void setGuide(double a, double b, double d) { |
25 | f9daq | 232 | _a = a; |
233 | _b = b; |
||
70 | f9daq | 234 | //_M = b/a; |
25 | f9daq | 235 | _d = d; |
236 | }; |
||
237 | void setGap(double x, double y, double z) { _gap = TVector3(x,y,z); }; |
||
238 | void setFresnel(int fresnel) { _fresnel = fresnel; }; |
||
239 | void setGlass(int glassOn, double glassD) { _glassOn = glassOn; _glassD = glassD; }; |
||
240 | void setGuideOn(int guideOn) { _guideOn = guideOn; }; |
||
241 | void setPlate(int plateOn, double plateWidth) { _plateOn = plateOn; _plateWidth = plateWidth; }; |
||
70 | f9daq | 242 | void setIndices(double n1, double n2, double n3) { _n1 = n1; _n2 = n2; _n3 = n3; }; |
25 | f9daq | 243 | |
70 | f9daq | 244 | double getLightYield() { return ( Power(_b,2)/Power(_active,2)); }; |
245 | double getM() { return (_b/_a); }; |
||
25 | f9daq | 246 | double getA() {return _a;}; |
247 | double getB() {return _b;}; |
||
248 | double getD() {return _d;}; |
||
249 | double getN1() {return _n1; }; |
||
250 | double getN2() {return _n2;}; |
||
251 | double getN3() {return _n3;}; |
||
252 | double getActive() {return _active;}; |
||
253 | TVector3 getGap() {return _gap;}; |
||
254 | double getPlateWidth() {return _plateWidth; }; |
||
255 | int getFresnel() { return _fresnel; }; |
||
256 | int getGlassOn() { return _glassOn; }; |
||
257 | double getGlassD() { return _glassD; }; |
||
258 | int getGuideOn() { return _guideOn; }; |
||
259 | int getPlateOn() { return _plateOn; }; |
||
54 | f9daq | 260 | double getOffsetY() { return offsetY; }; |
261 | double getOffsetZ() { return offsetZ; }; |
||
25 | f9daq | 262 | |
263 | private: |
||
264 | double _a; |
||
265 | double _b; |
||
266 | double _d; |
||
267 | double _active; |
||
268 | double _n1; |
||
269 | double _n2; |
||
270 | double _n3; |
||
271 | TVector3 _gap; |
||
70 | f9daq | 272 | |
273 | int _fresnel; |
||
274 | |||
275 | int _guideOn; |
||
54 | f9daq | 276 | double offsetY; |
277 | double offsetZ; |
||
25 | f9daq | 278 | |
70 | f9daq | 279 | int _plateOn; |
280 | double _plateWidth; |
||
281 | |||
25 | f9daq | 282 | int _glassOn; |
283 | double _glassD; |
||
284 | |||
285 | }; |
||
286 | |||
287 | class Guide |
||
288 | { |
||
289 | public: |
||
290 | Guide(TVector3 center0, DetectorParameters& parameters); |
||
291 | ~Guide() { |
||
54 | f9daq | 292 | for (int jk=0; jk<6; jk++) delete s_side[jk]; |
293 | delete hfate; |
||
294 | delete hnodb_all; |
||
295 | delete hnodb_exit; |
||
296 | delete hin; |
||
297 | delete hout; |
||
25 | f9daq | 298 | } |
299 | |||
300 | Fate PropagateRay(CRay in, CRay *out, int *n_points, TVector3 *points); |
||
301 | |||
302 | double getD() { return _d; } |
||
303 | double getN1() { return n1; }; |
||
304 | double getN2() { return n2; }; |
||
305 | double getN3() { return n3; }; |
||
306 | TH1F* GetHFate() const {return hfate;}; |
||
307 | TH1F* GetHNOdb_all() const {return hnodb_all;}; |
||
308 | TH1F* GetHNOdb_exit() const {return hnodb_exit;}; |
||
309 | TH2F* GetHIn() const {return hin;}; |
||
310 | TH2F* GetHOut() const {return hout;}; |
||
311 | int GetExitHits() {return (int)hfate->GetBinContent(5);}; |
||
312 | int GetEnteranceHits() {return (int)hfate->GetBinContent(6);}; |
||
313 | void GetVFate(int *out); |
||
54 | f9daq | 314 | int GetMAXODB() const {return MAX_REFLECTIONS;}; |
25 | f9daq | 315 | |
316 | void Draw(int color = 1, int width = 1); |
||
317 | void DrawSkel(int color = 1, int width = 1); |
||
318 | |||
319 | private: |
||
320 | Fate fate; |
||
321 | double _d; // parameters |
||
322 | double n1, n2, n3; // refractive index: n1 above entry surface, n2 inside, n3 after exit |
||
323 | double _r; |
||
324 | double absorption; |
||
325 | double A; |
||
326 | TRandom rand; // for material absorption |
||
327 | CSurface *s_side[6]; |
||
328 | TVector3 center, vodnik_edge[8]; |
||
329 | |||
330 | TH1F *hfate, *hnodb_all, *hnodb_exit; |
||
331 | TH2F *hin, *hout; |
||
332 | }; |
||
333 | //================================================================================= |
||
334 | |||
335 | //=============================================================================================================================== <<<<<<<< |
||
336 | // ravnina - krog |
||
337 | class CPlaneR |
||
338 | { |
||
339 | |||
340 | public: |
||
341 | CPlaneR(TVector3 c, TVector3 n0, double R0) |
||
342 | {center = c; n = n0; _r = R0;}; |
||
343 | |||
344 | void Set(TVector3 c, TVector3 n0, double R0) |
||
345 | {center = c; n = n0; _r = R0;}; |
||
346 | |||
347 | int TestIntersection(TVector3 *vec, CRay in); |
||
348 | |||
349 | void Draw(int color = 1, int NN = 32); |
||
350 | |||
351 | private: |
||
352 | TVector3 n, center; |
||
353 | double _r; |
||
354 | }; |
||
355 | //================================================================================= |
||
356 | |||
357 | class Plate |
||
358 | { |
||
359 | public: |
||
360 | Plate(DetectorParameters& parameters); |
||
54 | f9daq | 361 | ~Plate() { |
362 | for (int jk=0; jk<6; jk++) delete sides[jk]; // the same, needs solution |
||
25 | f9daq | 363 | }; |
364 | |||
365 | void draw(int color, int width); |
||
366 | void drawSkel(int color, int width); |
||
54 | f9daq | 367 | Fate propagateRay(CRay, CRay*, int*, TVector3*); |
25 | f9daq | 368 | |
369 | private: |
||
370 | TVector3 plate_edge[8]; |
||
371 | CSurface *sides[6]; |
||
372 | }; |
||
373 | |||
374 | // ================================================================================ |
||
375 | |||
376 | class CDetector |
||
377 | { |
||
378 | public: |
||
379 | CDetector(TVector3 center0, DetectorParameters& parameters); |
||
380 | ~CDetector() { |
||
381 | delete glass; |
||
382 | delete glass_circle; |
||
383 | delete hglass; |
||
384 | delete active; |
||
385 | delete hactive; |
||
386 | delete hlaser; |
||
387 | delete detector; |
||
388 | delete hdetector; |
||
389 | delete guide; |
||
390 | delete plate; |
||
391 | delete histoPlate; |
||
392 | //delete window; |
||
393 | //delete window_circle; |
||
394 | //delete hwindow; |
||
395 | } |
||
396 | |||
397 | //void SetLGType(int in = SURF_REFRA, int side = SURF_REFRA, int out = SURF_REFRA) |
||
398 | // {type_in = in; type_side = side; type_out = out;}; |
||
399 | |||
400 | /* void SetLG(double SiPM0, double M0, double d0, double n10, double n20, double n30, double R0) |
||
401 | { SiPM=SiPM0; |
||
402 | M=M0; |
||
403 | d=d0; |
||
404 | n1=n10; |
||
405 | n2=n20; |
||
406 | n3=n30; |
||
407 | R=R0; |
||
408 | }; */ |
||
409 | //void SetR(double R0) {R = R0;}; |
||
410 | //void SetGlass(int glass_on0, double glass_d0) |
||
411 | //{glass_on = glass_on0; glass_d = glass_d0;}; |
||
412 | //void SetGap(double x_gap0, double y_gap0, double z_gap0) |
||
413 | //{x_gap = x_gap0; y_gap = y_gap0; z_gap = z_gap0;}; |
||
414 | void SetRCol(int in, int lg, int out, int gla) |
||
415 | {col_in = in; col_lg = lg; col_out = out; col_rgla = gla;}; |
||
416 | void SetDCol(int LG0, int glass0, int active0) |
||
417 | {col_LG = LG0; col_glass = glass0; col_active = active0;}; |
||
418 | //void SetFresnel(int b) |
||
419 | //{fresnel = b;}; |
||
420 | //void SetAbsorption(int b, double A0) |
||
421 | //{absorption = b; A = A0;}; |
||
422 | //void SetWindow(double wR, double d0) {window_R = wR; window_d = d0;}; |
||
423 | //void SetGuideOn(int b) |
||
424 | //{guide_on = b;}; |
||
425 | |||
426 | //void Init(); |
||
427 | |||
428 | int Propagate(CRay, CRay*, int); |
||
429 | |||
430 | Guide* GetLG() const {return guide;}; |
||
431 | //TH2F* GetHWindow() const {return hwindow;}; |
||
432 | TH2F* GetHGlass() const {return hglass;}; |
||
433 | TH2F* GetHActive() const {return hactive;}; |
||
434 | TH2F* GetHLaser() const {return hlaser;}; |
||
435 | TH2F* GetHDetector() const {return hdetector;}; |
||
70 | f9daq | 436 | TH2F* GetHPlate() const {return histoPlate;}; |
25 | f9daq | 437 | |
438 | //double GetSiPM() const {return SiPM;}; |
||
439 | //double GetM() const {return M;}; |
||
440 | //double GetD() const {return d;}; |
||
441 | //double GetR() const {return R;}; |
||
442 | //TVector3 GetVGap() const {return (TVector3(x_gap, y_gap, z_gap));}; |
||
443 | //int IsGlass() {return glass_on;}; |
||
444 | //double GetGlass() {return glass_d;}; |
||
445 | //double GetWindowR() {return window_R;}; |
||
446 | //double GetWindowD() {return window_d;}; |
||
447 | |||
448 | void Draw(int width = 2); |
||
449 | |||
450 | private: |
||
451 | Fate fate; |
||
452 | TVector3 center; |
||
453 | |||
454 | //int type_in, type_side, type_out; |
||
455 | //double SiPM, M, d, n1, n2, n3, R; |
||
456 | //double detectorActive; |
||
457 | |||
458 | |||
459 | int glass_on; |
||
460 | double glass_d; |
||
461 | CSurface *glass; |
||
462 | CPlaneR *glass_circle; |
||
463 | TH2F *hglass; |
||
464 | |||
465 | //double x_gap, y_gap, z_gap; |
||
466 | CPlane4 *active; |
||
467 | TH2F *hactive, *hlaser; |
||
468 | |||
469 | CPlane4 *detector; |
||
470 | TH2F *hdetector; |
||
471 | |||
472 | int col_in, col_lg, col_out, col_rgla; |
||
473 | int col_LG, col_glass, col_active; |
||
474 | |||
475 | //int fresnel, absorption; |
||
476 | //double A; |
||
477 | |||
478 | int guide_on; |
||
479 | |||
480 | //double window_R, window_d; |
||
481 | //CSurface *window; |
||
482 | //CPlaneR *window_circle; |
||
483 | //TH2F *hwindow; |
||
484 | |||
485 | Guide *guide; |
||
486 | Plate *plate; |
||
487 | |||
488 | double _plateWidth; |
||
489 | int _plateOn; |
||
490 | |||
491 | TH2F *histoPlate; |
||
54 | f9daq | 492 | |
493 | double offsetY; |
||
494 | double offsetZ; |
||
25 | f9daq | 495 | }; |
496 | |||
497 | |||
498 | |||
499 | //================================================================================= |
||
500 | #endif |