TrackerLevel2/inc/TrkLevel2.h

/**
 * \file TrkLevel2.h
 * \author Elena Vannuccini
 */
#ifndef trklevel2_h
#define trklevel2_h

#include <TObject.h>
#include <TObjArray.h>
#include <TClonesArray.h>
#include <TRefArray.h>
#include <TRef.h>

#include <TrkParams.h>
#include <TrkLevel1.h>

// z-coordinate of track state-vector reference-plane
#define ZINI 23.5   
// (mechanical) z-coordinate of the tracker planes
#define ZTRK6 -22.22
#define ZTRK5 -13.31
#define ZTRK4 -4.41
#define ZTRK3 4.49
#define ZTRK2 13.39
#define ZTRK1 22.29
// magnet cavity dimensions
#define ZMAGNHIGH 21.83
#define ZMAGNLOW -21.83
#define XMAGNHIGH 8.07 
#define XMAGNLOW -8.07 
#define YMAGNHIGH 6.57 
#define YMAGNLOW -6.57 
// (mechanical) x/y-coordinates of magnet cavity
#define XTRKL -8.1
#define XTRKR  8.1
#define YTRKL -6.6
#define YTRKR  6.6

/**
 * \brief Class to describe, by points, a particle trajectory in the apparatus. 
 *
 * The idea is to create it by integrating the equations of motion, given the 
 * track state vector and the z coordinates where to evaluate track position.
 */
// ==================================================================
class Trajectory : public TObject{
 private:

 public:

    int npoint; ///< number of evaluated points along the trajectory
    float* x;   ///< x coordinates 
    float* y;   ///< y coordinates 
    float* z;   ///< z coordinates 
    float* thx; ///< x projected angle 
    float* thy; ///< y projected angle
    float* tl;  ///< track length

    Trajectory();
    Trajectory(int n);
    Trajectory(int n, float* pz);
    ~Trajectory(){Delete();};
    void Dump();
    void Delete();

    int DoTrack2(float* al);
    float GetLength(){float l=0; for(int i=0; i<npoint;i++)l=l+tl[i]; return l;};
    float GetLength(int,int);

    ClassDef(Trajectory,2);

};
/**
 * \brief Class to describe fitted tracks. 
 *
 * A track is defined by the measured coordinates associated to it, the 
 * track status vector, plus other quantities.
 * A track may have an "image", due to the ambiguity in the y view.
 *
 * Cluster flags: xgood[6], ygood[6]
 * 
 * xgood/ygood = +/- 0lsccccccc
 *                |   |||------- ID (1-7483647) of the included cluster  
 *                |   ||-------- sensor number (1,2   - increasing y)
 *                |   |--------- ladder number (1,2,3 - increasing x)
 *                |------------- does-not/does include bad strips
 */
// ==================================================================
class TrkTrack : public TObject {

private:

    int   seqno;           ///<stored track sequential number
    int   image;           ///<sequential number of track-image
        
public:

    float al[5];           ///<TRACK STATE VECTOR 
    float coval[5][5];     ///<covariance matrix 
    int   xgood[6];        ///<cluster flag for x-view (0 = view not included in the fit) 
    int   ygood[6];        ///<cluster flag for y-view (0 = view not included in the fit) 
    float xm[6];           ///<measured x coordinates
    float ym[6];           ///<measured y coordinates 
    float zm[6];           ///<measured z coordinates 
    float resx[6];         ///<spatial resolution on X view
    float resy[6];         ///<spatial resolution on y view
    float tailx[6];        ///<spatial resolution tail on X view
    float taily[6];        ///<spatial resolution tail on y view
    float chi2;            ///<chi2
    int   nstep;           ///<n. step
    float xv[6];           ///<calculated x coordinates
    float yv[6];           ///<calculated y coordinates
    float zv[6];           ///<calculated z coordinates
    float axv[6];          ///<calculated angles (deg) on x view
    float ayv[6];          ///<calculated angles (deg) on y view
    float dedx_x[6];       ///<dE/dx in MIP (<0 if saturated)
    float dedx_y[6];       ///<dE/dx in MIP (<0 if saturated) 

    TrkTrack();
    TrkTrack(const TrkTrack&);

    ~TrkTrack(){ Delete(); };
        
    void Dump();
    void Clear();
    void Clear(Option_t *option){Clear();};
    void Delete();
    void Copy(TrkTrack&);
//    void Set();

    Int_t  GetSeqNo(){return seqno;}        ///< Returns the track sequential number
    Int_t  GetImageSeqNo(){return image;}   ///< Returns the track image sequential number
    Bool_t HasImage(){return !(image==-1);} ///< Returns true if the track has an image
    int DoTrack(Trajectory* t);                         ///< Evaluates the trajectory in the apparatus.
    int DoTrack2(Trajectory* t);                        ///< Evaluates the trajectory in the apparatus.
    float BdL(){return 0;};                                     ///< Evaluates the integral of B*dL along the track.
    Int_t GetNX(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=(Int_t)XGood(i); return n;}; 
    Int_t GetNY(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=(Int_t)YGood(i); return n;};
    Int_t GetNtot(){return GetNX()+GetNY();};
    Float_t GetRigidity();
    Float_t GetDeflection();
    Bool_t IsSaturated(int,int);
    Bool_t IsSaturated(int);
    Bool_t IsSaturated();
    Bool_t IsBad(int,int);
    Float_t GetDEDX();
    Float_t GetDEDX(int);
    Float_t GetDEDX(int,int);
    Int_t GetLeverArmX();
    Int_t GetLeverArmY();

    void SetMeasure(double *xmeas, double *ymeas, double *zmeas);
    void SetResolution(double *rx, double *ry);
    void SetTail(double *tx, double *ty, double factor);
    void SetStudentParam(int flag);
    void SetGood(int *xg, int *yg);
    void LoadField(TString s);
    void Fit(double pfixed, int& fail, int iprint, int froml1);
    void Fit(double pfixed, int& fail, int iprint){ Fit(pfixed,fail,iprint,0); };
    void FitReset();
    void SetTrackingMode(int trackmode);
    void SetPrecisionFactor(double fact);
    void SetStepMin(int istepmin);
    Bool_t IsInsideCavity();

    void EvaluateClusterPositions();

    void FillMiniStruct(cMini2track&);
    void SetFromMiniStruct(cMini2track*);
    
    Int_t GetClusterX_ID(int ip);
    Int_t GetClusterY_ID(int ip);
    Int_t GetLadder(int ip);
    Int_t GetSensor(int ip);
    Bool_t XGood(int ip){ return GetClusterX_ID(ip)!=-1; };
    Bool_t YGood(int ip){ return GetClusterY_ID(ip)!=-1; };
    void ResetXGood(int ip){ xgood[ip]=0; };
    void ResetYGood(int ip){ ygood[ip]=0; };
    void SetXGood(int ip, int clid, int is);
    void SetYGood(int ip, int clid, int is);

    Bool_t BadClusterX(int ip){ return IsBad(ip,0); };
    Bool_t BadClusterY(int ip){ return IsBad(ip,1); };

    Bool_t SaturatedClusterX(int ip){ return IsSaturated(ip,0); };
    Bool_t SaturatedClusterY(int ip){ return IsSaturated(ip,1); };

    TrkTrack* GetTrkTrack(){return this;};

    friend class TrkLevel2;

    ClassDef(TrkTrack,3);

};
/**
 * \brief Class to describe single clusters ("singlets"). 
 *
 * Single clusters are clusters not associated to any track.
 */
class TrkSinglet : public TObject {

private:
        

public:
        
    int plane;       ///<plane 
    float coord[2];  ///<coordinate (on sensor 1 and 2)
    float sgnl;      ///<cluster signal in MIP (<0 if saturated)

    TrkSinglet();
    TrkSinglet(const TrkSinglet&);
    ~TrkSinglet(){Delete();};

    void Dump();
    void Clear();
    void Clear(Option_t *option){Clear();};
    void Delete(){Clear();};
    Float_t GetSignal(){return fabs(sgnl);}
    Bool_t IsSaturated(){return (sgnl<0); };
        
    friend class TrkLevel2;

    ClassDef(TrkSinglet,3);

};

/**
 * \brief Class to describe tracker LEVEL2 data.
 *
 * A tracker events is defined by some general variables, plus the collection of all the fitted tracks and all 
 * single clusters on X and Y views. 
 * Tracks and single clusters ("singlets") are described by the classes TrkTrack and TrkSinglet respectivelly.
 * 
 * Each track may have an "image", due to the ambiguity on the Y view, which is stored also. 
 * Thus, the number of stored tracks ( ntrk() ) differs from the number of "physical" tracks ( GetNTracks() ). 
 * Proper methods allow to sort tracks and select the physical ones ( GetTracks() ).
 */
class TrkLevel2 : public TObject {

 private:
 
 public:

    Int_t         good[12];       ///< event status
    UInt_t        VKmask[12];     ///< Viking-chip mask
    UInt_t        VKflag[12];     ///< Viking-chip flag

    TClonesArray *Track;        ///< fitted tracks
    TClonesArray *SingletX;     ///< x singlets
    TClonesArray *SingletY;     ///< y singlets

    TrkLevel2();
//    TrkLevel2(cTrkLevel2 *);
    ~TrkLevel2(){Delete();};
        
    void Clear();
    void Clear(Option_t *option){Clear();};
    void Delete();
    void Set();
    
    int ntrk() {return Track->GetEntries();}    ///< number of stored track
    int nclsx(){return SingletX->GetEntries();} ///< number of x singlets 
    int nclsy(){return SingletY->GetEntries();} ///< number of y singlets 

    void Dump();
    void SetFromLevel2Struct(cTrkLevel2 *, TrkLevel1 *);
    void SetFromLevel2Struct(cTrkLevel2 *s2){ SetFromLevel2Struct(s2, NULL);          };
    void SetFromLevel2Struct(TrkLevel1 *l1) { SetFromLevel2Struct(&level2event_, l1); };    
    void SetFromLevel2Struct()              { SetFromLevel2Struct(&level2event_);     };    
    void GetLevel2Struct(cTrkLevel2 *) const;
    void LoadField(TString);
    float GetBX(float* v){return TrkParams::GetBX(v);};///< Bx (kGauss)
    float GetBY(float* v){return TrkParams::GetBY(v);};///< By (kGauss)
    float GetBZ(float* v){return TrkParams::GetBZ(v);};///< Bz (kGauss)
    Float_t GetZTrk(Int_t);
    Float_t GetXTrkLeft(){return XTRKL;};
    Float_t GetXTrkRight(){return XTRKR;};
    Float_t GetYTrkLeft(){return YTRKL;};
    Float_t GetYTrkRight(){return YTRKR;};
    
    Bool_t IsMaskedVK(int,int);
    Bool_t GetVKMask(int,int);
    Bool_t GetVKFlag(int,int);

    TrkSinglet   *GetSingletX(int);
    TrkSinglet   *GetSingletY(int);
    
    TrkTrack     *GetStoredTrack(int i);
    Int_t         GetSeqNo(Int_t i)  {return (((TrkTrack *)Track->At(i))->seqno);}; ///< Returns track sequential number

    TRefArray *GetTracks_NFitSorted();
    TRefArray *GetTracks(){return this->GetTracks_NFitSorted();};
    
    Int_t     GetNTracks();
    TrkTrack* GetTrack(int i);
    TrkTrack* GetTrackImage(int i);
    
    TrkLevel2*    GetTrkLevel2(){return this;}
    TClonesArray* GetTrackArray(){return Track;};///< returns pointer to the track array
    
    ClassDef(TrkLevel2,3);

};

#endif
1	/**
2	* \file TrkLevel2.h
3	* \author Elena Vannuccini
4	*/
5	#ifndef trklevel2_h
6	#define trklevel2_h
7
8	#include <TObject.h>
9	#include <TObjArray.h>
10	#include <TClonesArray.h>
11	#include <TRefArray.h>
12	#include <TRef.h>
13
14	#include <TrkParams.h>
15	#include <TrkLevel1.h>
16
17	// z-coordinate of track state-vector reference-plane
18	#define ZINI 23.5
19	// (mechanical) z-coordinate of the tracker planes
20	#define ZTRK6 -22.22
21	#define ZTRK5 -13.31
22	#define ZTRK4 -4.41
23	#define ZTRK3 4.49
24	#define ZTRK2 13.39
25	#define ZTRK1 22.29
26	// magnet cavity dimensions
27	#define ZMAGNHIGH 21.83
28	#define ZMAGNLOW -21.83
29	#define XMAGNHIGH 8.07
30	#define XMAGNLOW -8.07
31	#define YMAGNHIGH 6.57
32	#define YMAGNLOW -6.57
33	// (mechanical) x/y-coordinates of magnet cavity
34	#define XTRKL -8.1
35	#define XTRKR 8.1
36	#define YTRKL -6.6
37	#define YTRKR 6.6
38
39	/**
40	* \brief Class to describe, by points, a particle trajectory in the apparatus.
41	*
42	* The idea is to create it by integrating the equations of motion, given the
43	* track state vector and the z coordinates where to evaluate track position.
44	*/
45	// ==================================================================
46	class Trajectory : public TObject{
47	private:
48
49	public:
50
51	int npoint; ///< number of evaluated points along the trajectory
52	float* x; ///< x coordinates
53	float* y; ///< y coordinates
54	float* z; ///< z coordinates
55	float* thx; ///< x projected angle
56	float* thy; ///< y projected angle
57	float* tl; ///< track length
58
59	Trajectory();
60	Trajectory(int n);
61	Trajectory(int n, float* pz);
62	~Trajectory(){Delete();};
63	void Dump();
64	void Delete();
65
66	int DoTrack2(float* al);
67	float GetLength(){float l=0; for(int i=0; i<npoint;i++)l=l+tl[i]; return l;};
68	float GetLength(int,int);
69
70	ClassDef(Trajectory,2);
71
72	};
73	/**
74	* \brief Class to describe fitted tracks.
75	*
76	* A track is defined by the measured coordinates associated to it, the
77	* track status vector, plus other quantities.
78	* A track may have an "image", due to the ambiguity in the y view.
79	*
80	* Cluster flags: xgood[6], ygood[6]
81	*
82	* xgood/ygood = +/- 0lsccccccc
83	* \| \|\|\|------- ID (1-7483647) of the included cluster
84	* \| \|\|-------- sensor number (1,2 - increasing y)
85	* \| \|--------- ladder number (1,2,3 - increasing x)
86	* \|------------- does-not/does include bad strips
87	*/
88	// ==================================================================
89	class TrkTrack : public TObject {
90
91	private:
92
93	int seqno; ///<stored track sequential number
94	int image; ///<sequential number of track-image
95
96	public:
97
98	float al[5]; ///<TRACK STATE VECTOR
99	float coval[5][5]; ///<covariance matrix
100	int xgood[6]; ///<cluster flag for x-view (0 = view not included in the fit)
101	int ygood[6]; ///<cluster flag for y-view (0 = view not included in the fit)
102	float xm[6]; ///<measured x coordinates
103	float ym[6]; ///<measured y coordinates
104	float zm[6]; ///<measured z coordinates
105	float resx[6]; ///<spatial resolution on X view
106	float resy[6]; ///<spatial resolution on y view
107	float tailx[6]; ///<spatial resolution tail on X view
108	float taily[6]; ///<spatial resolution tail on y view
109	float chi2; ///<chi2
110	int nstep; ///<n. step
111	float xv[6]; ///<calculated x coordinates
112	float yv[6]; ///<calculated y coordinates
113	float zv[6]; ///<calculated z coordinates
114	float axv[6]; ///<calculated angles (deg) on x view
115	float ayv[6]; ///<calculated angles (deg) on y view
116	float dedx_x[6]; ///<dE/dx in MIP (<0 if saturated)
117	float dedx_y[6]; ///<dE/dx in MIP (<0 if saturated)
118
119	TrkTrack();
120	TrkTrack(const TrkTrack&);
121
122	~TrkTrack(){ Delete(); };
123
124	void Dump();
125	void Clear();
126	void Clear(Option_t *option){Clear();};
127	void Delete();
128	void Copy(TrkTrack&);
129	// void Set();
130
131	Int_t GetSeqNo(){return seqno;} ///< Returns the track sequential number
132	Int_t GetImageSeqNo(){return image;} ///< Returns the track image sequential number
133	Bool_t HasImage(){return !(image==-1);} ///< Returns true if the track has an image
134	int DoTrack(Trajectory* t); ///< Evaluates the trajectory in the apparatus.
135	int DoTrack2(Trajectory* t); ///< Evaluates the trajectory in the apparatus.
136	float BdL(){return 0;}; ///< Evaluates the integral of B*dL along the track.
137	Int_t GetNX(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=(Int_t)XGood(i); return n;};
138	Int_t GetNY(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=(Int_t)YGood(i); return n;};
139	Int_t GetNtot(){return GetNX()+GetNY();};
140	Float_t GetRigidity();
141	Float_t GetDeflection();
142	Bool_t IsSaturated(int,int);
143	Bool_t IsSaturated(int);
144	Bool_t IsSaturated();
145	Bool_t IsBad(int,int);
146	Float_t GetDEDX();
147	Float_t GetDEDX(int);
148	Float_t GetDEDX(int,int);
149	Int_t GetLeverArmX();
150	Int_t GetLeverArmY();
151
152	void SetMeasure(double xmeas, double ymeas, double *zmeas);
153	void SetResolution(double rx, double ry);
154	void SetTail(double tx, double ty, double factor);
155	void SetStudentParam(int flag);
156	void SetGood(int xg, int yg);
157	void LoadField(TString s);
158	void Fit(double pfixed, int& fail, int iprint, int froml1);
159	void Fit(double pfixed, int& fail, int iprint){ Fit(pfixed,fail,iprint,0); };
160	void FitReset();
161	void SetTrackingMode(int trackmode);
162	void SetPrecisionFactor(double fact);
163	void SetStepMin(int istepmin);
164	Bool_t IsInsideCavity();
165
166	void EvaluateClusterPositions();
167
168	void FillMiniStruct(cMini2track&);
169	void SetFromMiniStruct(cMini2track*);
170
171	Int_t GetClusterX_ID(int ip);
172	Int_t GetClusterY_ID(int ip);
173	Int_t GetLadder(int ip);
174	Int_t GetSensor(int ip);
175	Bool_t XGood(int ip){ return GetClusterX_ID(ip)!=-1; };
176	Bool_t YGood(int ip){ return GetClusterY_ID(ip)!=-1; };
177	void ResetXGood(int ip){ xgood[ip]=0; };
178	void ResetYGood(int ip){ ygood[ip]=0; };
179	void SetXGood(int ip, int clid, int is);
180	void SetYGood(int ip, int clid, int is);
181
182	Bool_t BadClusterX(int ip){ return IsBad(ip,0); };
183	Bool_t BadClusterY(int ip){ return IsBad(ip,1); };
184
185	Bool_t SaturatedClusterX(int ip){ return IsSaturated(ip,0); };
186	Bool_t SaturatedClusterY(int ip){ return IsSaturated(ip,1); };
187
188	TrkTrack* GetTrkTrack(){return this;};
189
190	friend class TrkLevel2;
191
192	ClassDef(TrkTrack,3);
193
194	};
195	/**
196	* \brief Class to describe single clusters ("singlets").
197	*
198	* Single clusters are clusters not associated to any track.
199	*/
200	class TrkSinglet : public TObject {
201
202	private:
203
204
205	public:
206
207	int plane; ///<plane
208	float coord[2]; ///<coordinate (on sensor 1 and 2)
209	float sgnl; ///<cluster signal in MIP (<0 if saturated)
210
211	TrkSinglet();
212	TrkSinglet(const TrkSinglet&);
213	~TrkSinglet(){Delete();};
214
215	void Dump();
216	void Clear();
217	void Clear(Option_t *option){Clear();};
218	void Delete(){Clear();};
219	Float_t GetSignal(){return fabs(sgnl);}
220	Bool_t IsSaturated(){return (sgnl<0); };
221
222	friend class TrkLevel2;
223
224	ClassDef(TrkSinglet,3);
225
226	};
227
228	/**
229	* \brief Class to describe tracker LEVEL2 data.
230	*
231	* A tracker events is defined by some general variables, plus the collection of all the fitted tracks and all
232	* single clusters on X and Y views.
233	* Tracks and single clusters ("singlets") are described by the classes TrkTrack and TrkSinglet respectivelly.
234	*
235	* Each track may have an "image", due to the ambiguity on the Y view, which is stored also.
236	* Thus, the number of stored tracks ( ntrk() ) differs from the number of "physical" tracks ( GetNTracks() ).
237	* Proper methods allow to sort tracks and select the physical ones ( GetTracks() ).
238	*/
239	class TrkLevel2 : public TObject {
240
241	private:
242
243	public:
244
245	Int_t good[12]; ///< event status
246	UInt_t VKmask[12]; ///< Viking-chip mask
247	UInt_t VKflag[12]; ///< Viking-chip flag
248
249	TClonesArray *Track; ///< fitted tracks
250	TClonesArray *SingletX; ///< x singlets
251	TClonesArray *SingletY; ///< y singlets
252
253	TrkLevel2();
254	// TrkLevel2(cTrkLevel2 *);
255	~TrkLevel2(){Delete();};
256
257	void Clear();
258	void Clear(Option_t *option){Clear();};
259	void Delete();
260	void Set();
261
262	int ntrk() {return Track->GetEntries();} ///< number of stored track
263	int nclsx(){return SingletX->GetEntries();} ///< number of x singlets
264	int nclsy(){return SingletY->GetEntries();} ///< number of y singlets
265
266	void Dump();
267	void SetFromLevel2Struct(cTrkLevel2 , TrkLevel1 );
268	void SetFromLevel2Struct(cTrkLevel2 *s2){ SetFromLevel2Struct(s2, NULL); };
269	void SetFromLevel2Struct(TrkLevel1 *l1) { SetFromLevel2Struct(&level2event_, l1); };
270	void SetFromLevel2Struct() { SetFromLevel2Struct(&level2event_); };
271	void GetLevel2Struct(cTrkLevel2 *) const;
272	void LoadField(TString);
273	float GetBX(float* v){return TrkParams::GetBX(v);};///< Bx (kGauss)
274	float GetBY(float* v){return TrkParams::GetBY(v);};///< By (kGauss)
275	float GetBZ(float* v){return TrkParams::GetBZ(v);};///< Bz (kGauss)
276	Float_t GetZTrk(Int_t);
277	Float_t GetXTrkLeft(){return XTRKL;};
278	Float_t GetXTrkRight(){return XTRKR;};
279	Float_t GetYTrkLeft(){return YTRKL;};
280	Float_t GetYTrkRight(){return YTRKR;};
281
282	Bool_t IsMaskedVK(int,int);
283	Bool_t GetVKMask(int,int);
284	Bool_t GetVKFlag(int,int);
285
286	TrkSinglet *GetSingletX(int);
287	TrkSinglet *GetSingletY(int);
288
289	TrkTrack *GetStoredTrack(int i);
290	Int_t GetSeqNo(Int_t i) {return (((TrkTrack *)Track->At(i))->seqno);}; ///< Returns track sequential number
291
292	TRefArray *GetTracks_NFitSorted();
293	TRefArray *GetTracks(){return this->GetTracks_NFitSorted();};
294
295	Int_t GetNTracks();
296	TrkTrack* GetTrack(int i);
297	TrkTrack* GetTrackImage(int i);
298
299	TrkLevel2* GetTrkLevel2(){return this;}
300	TClonesArray* GetTrackArray(){return Track;};///< returns pointer to the track array
301
302	ClassDef(TrkLevel2,3);
303
304	};
305
306	#endif