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 <TrkStruct.h>
#include <TrkParams.h>
#include <TrkLevel1.h>

// z-coordinate of track state-vector reference-plane
#define ZINI 23.5   
// upper and lower (mechanical) z-coordinate of the tracker
//#define ZTRKUP 22.29
//#define ZTRKDW -22.22
// (mechanical) z-coordinate of the tracker planes
#define ZTRK6 -22.23
#define ZTRK5 -13.32
#define ZTRK4 -4.42
#define ZTRK3 4.48
#define ZTRK2 13.38
#define ZTRK1 22.28
// (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.
 */
// ==================================================================
class TrkTrack : public TObject {

private:

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

        
public:

    TRefArray *clx;
    TRefArray *cly;

/*     float al[5];           ///<TRACK STATE VECTOR  */
    float al[5];           ///<TRACK STATE VECTOR 
    float coval[5][5];     ///<covariance matrix 
    int   xgood[6];        ///<mask of included x planes 
    int   ygood[6];        ///<mask of included y planes 
    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 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 
    float dedx_y[6];       ///<dE/dx in MIP 

    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+=XGood(i); return n;}; 
    Int_t GetNY(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=YGood(i); return n;};
    Int_t GetNtot(){return GetNX()+GetNY();};
    Float_t GetRigidity();
    Float_t GetDeflection();
    Float_t GetDEDX();
    Float_t GetDEDX(Int_t ip){if( !(XGood(ip)+YGood(ip)) ) return 0; return (dedx_x[ip]+dedx_y[ip])/(XGood(ip)+YGood(ip));};
    // sono un'imbecille... assegno xm e ym anche quando si tratta di un singolo
    // non posso quindi usare xm e ym per dire se una vista e` inclusa nel fit o no
/*      Bool_t XGood(int ip){ return xm[ip] != -100.;};
        Bool_t YGood(int ip){ return ym[ip] != -100.;};*/
    Bool_t XGood(int ip){ return xgood[ip]> 0;};
    Bool_t YGood(int ip){ return ygood[ip]> 0;};
    

    void SetMeasure(double *xmeas, double *ymeas, double *zmeas);
    void SetResolution(double *rx, double *ry);
    void SetGood(int *xg, int *yg);
    void LoadField(TString s);
    void Fit(double pfixed, int& fail, int iprint);
    void FitReset();
    void SetTrackingMode(int trackmode);
    void SetPrecisionFactor(double fact);
    void SetStepMin(int istepmin);

    void FillMiniStruct(cMini2track&);
    void SetFromMiniStruct(cMini2track*);
    
    TrkCluster *GetClusterX(int ip);
    TrkCluster *GetClusterY(int ip);

    Int_t GetClusterX_ID(int ip){return xgood[ip];};
    Int_t GetClusterY_ID(int ip){return ygood[ip];};
    
    TrkTrack* GetTrkTrack(){return this;};

    friend class TrkLevel2;

    ClassDef(TrkTrack,2);

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

private:
        

public:
        
    TRef cls;

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

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

    void Dump();
    void Clear();
    void Clear(Option_t *option){Clear();};
    void Delete(){Clear();};
        
    TrkCluster *GetCluster(){TrkCluster *pt = (TrkCluster*)cls.GetObject(); return pt;};

    friend class TrkLevel2;

    ClassDef(TrkSinglet,2);

};

/**
 * \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

    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*);
    float GetBY(float*);
    float GetBZ(float*);
    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;};
    
    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
//    TClonesArray *GetTracks_Chi2Sorted();
//    TClonesArray *GetTracks_NFitSorted();
//    TClonesArray *GetTracks();
    TRefArray *GetTracks_NFitSorted();
    TRefArray *GetTracks(){return this->GetTracks_NFitSorted();};
    
//    int       GetNTracks(){return this->GetTracks()->GetEntries();}
    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,2);

};

#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 <TrkStruct.h>
15	#include <TrkParams.h>
16	#include <TrkLevel1.h>
17
18	// z-coordinate of track state-vector reference-plane
19	#define ZINI 23.5
20	// upper and lower (mechanical) z-coordinate of the tracker
21	//#define ZTRKUP 22.29
22	//#define ZTRKDW -22.22
23	// (mechanical) z-coordinate of the tracker planes
24	#define ZTRK6 -22.23
25	#define ZTRK5 -13.32
26	#define ZTRK4 -4.42
27	#define ZTRK3 4.48
28	#define ZTRK2 13.38
29	#define ZTRK1 22.28
30	// (mechanical) x/y-coordinates of magnet cavity
31	#define XTRKL -8.1
32	#define XTRKR 8.1
33	#define YTRKL -6.6
34	#define YTRKR 6.6
35
36	/**
37	* \brief Class to describe, by points, a particle trajectory in the apparatus.
38	*
39	* The idea is to create it by integrating the equations of motion, given the
40	* track state vector and the z coordinates where to evaluate track position.
41	*/
42	// ==================================================================
43	class Trajectory : public TObject{
44	private:
45
46	public:
47
48	int npoint; ///< number of evaluated points along the trajectory
49	float* x; ///< x coordinates
50	float* y; ///< y coordinates
51	float* z; ///< z coordinates
52	float* thx; ///< x projected angle
53	float* thy; ///< y projected angle
54	float* tl; ///< track length
55
56	Trajectory();
57	Trajectory(int n);
58	Trajectory(int n, float* pz);
59	~Trajectory(){Delete();};
60	void Dump();
61	void Delete();
62
63	int DoTrack2(float* al);
64	float GetLength(){float l=0; for(int i=0; i<npoint;i++)l=l+tl[i]; return l;};
65	float GetLength(int,int);
66
67	ClassDef(Trajectory,2);
68
69	};
70	/**
71	* \brief Class to describe fitted tracks.
72	*
73	* A track is defined by the measured coordinates associated to it, the
74	* track status vector, plus other quantities.
75	* A track may have an "image", due to the ambiguity in the y view.
76	*/
77	// ==================================================================
78	class TrkTrack : public TObject {
79
80	private:
81
82	int seqno; ///<stored track sequential number
83	int image; ///<sequential number of track-image
84
85
86	public:
87
88	TRefArray *clx;
89	TRefArray *cly;
90
91	/* float al[5]; ///<TRACK STATE VECTOR */
92	float al[5]; ///<TRACK STATE VECTOR
93	float coval[5][5]; ///<covariance matrix
94	int xgood[6]; ///<mask of included x planes
95	int ygood[6]; ///<mask of included y planes
96	float xm[6]; ///<measured x coordinates
97	float ym[6]; ///<measured y coordinates
98	float zm[6]; ///<measured z coordinates
99	float resx[6]; ///<spatial resolution on X view
100	float resy[6]; ///<spatial resolution on y view
101	float chi2; ///<chi2
102	int nstep; ///<n. step
103	float xv[6]; ///<calculated x coordinates
104	float yv[6]; ///<calculated y coordinates
105	float zv[6]; ///<calculated z coordinates
106	float axv[6]; ///<calculated angles (deg) on x view
107	float ayv[6]; ///<calculated angles (deg) on y view
108	float dedx_x[6]; ///<dE/dx in MIP
109	float dedx_y[6]; ///<dE/dx in MIP
110
111	TrkTrack();
112	TrkTrack(const TrkTrack&);
113
114	~TrkTrack(){ Delete(); };
115
116	void Dump();
117	void Clear();
118	void Clear(Option_t *option){Clear();};
119	void Delete();
120	void Copy(TrkTrack&);
121	// void Set();
122
123	Int_t GetSeqNo(){return seqno;} ///< Returns the track sequential number
124	Int_t GetImageSeqNo(){return image;} ///< Returns the track image sequential number
125	Bool_t HasImage(){return !(image==-1);} ///< Returns true if the track has an image
126	int DoTrack(Trajectory* t); ///< Evaluates the trajectory in the apparatus.
127	int DoTrack2(Trajectory* t); ///< Evaluates the trajectory in the apparatus.
128	float BdL(){return 0;}; ///< Evaluates the integral of B*dL along the track.
129	Int_t GetNX(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=XGood(i); return n;};
130	Int_t GetNY(){Int_t n=0; for(Int_t i=0; i<6; i++)n+=YGood(i); return n;};
131	Int_t GetNtot(){return GetNX()+GetNY();};
132	Float_t GetRigidity();
133	Float_t GetDeflection();
134	Float_t GetDEDX();
135	Float_t GetDEDX(Int_t ip){if( !(XGood(ip)+YGood(ip)) ) return 0; return (dedx_x[ip]+dedx_y[ip])/(XGood(ip)+YGood(ip));};
136	// sono un'imbecille... assegno xm e ym anche quando si tratta di un singolo
137	// non posso quindi usare xm e ym per dire se una vista e` inclusa nel fit o no
138	/* Bool_t XGood(int ip){ return xm[ip] != -100.;};
139	Bool_t YGood(int ip){ return ym[ip] != -100.;};*/
140	Bool_t XGood(int ip){ return xgood[ip]> 0;};
141	Bool_t YGood(int ip){ return ygood[ip]> 0;};
142
143
144	void SetMeasure(double xmeas, double ymeas, double *zmeas);
145	void SetResolution(double rx, double ry);
146	void SetGood(int xg, int yg);
147	void LoadField(TString s);
148	void Fit(double pfixed, int& fail, int iprint);
149	void FitReset();
150	void SetTrackingMode(int trackmode);
151	void SetPrecisionFactor(double fact);
152	void SetStepMin(int istepmin);
153
154	void FillMiniStruct(cMini2track&);
155	void SetFromMiniStruct(cMini2track*);
156
157	TrkCluster *GetClusterX(int ip);
158	TrkCluster *GetClusterY(int ip);
159
160	Int_t GetClusterX_ID(int ip){return xgood[ip];};
161	Int_t GetClusterY_ID(int ip){return ygood[ip];};
162
163	TrkTrack* GetTrkTrack(){return this;};
164
165	friend class TrkLevel2;
166
167	ClassDef(TrkTrack,2);
168
169	};
170	/**
171	* \brief Class to describe single clusters ("singlets").
172	*
173	* Single clusters are clusters not associated to any track.
174	*/
175	class TrkSinglet : public TObject {
176
177	private:
178
179
180	public:
181
182	TRef cls;
183
184	int plane; ///<plane
185	float coord[2]; ///<coordinate (on sensor 1 and 2)
186	float sgnl; ///<cluster signal in MIP
187
188	TrkSinglet();
189	TrkSinglet(const TrkSinglet&);
190	~TrkSinglet(){Delete();};
191
192	void Dump();
193	void Clear();
194	void Clear(Option_t *option){Clear();};
195	void Delete(){Clear();};
196
197	TrkCluster GetCluster(){TrkCluster pt = (TrkCluster*)cls.GetObject(); return pt;};
198
199	friend class TrkLevel2;
200
201	ClassDef(TrkSinglet,2);
202
203	};
204
205	/**
206	* \brief Class to describe tracker LEVEL2 data.
207	*
208	* A tracker events is defined by some general variables, plus the collection of all the fitted tracks and all
209	* single clusters on X and Y views.
210	* Tracks and single clusters ("singlets") are described by the classes TrkTrack and TrkSinglet respectivelly.
211	*
212	* Each track may have an "image", due to the ambiguity on the Y view, which is stored also.
213	* Thus, the number of stored tracks ( ntrk() ) differs from the number of "physical" tracks ( GetNTracks() ).
214	* Proper methods allow to sort tracks and select the physical ones ( GetTracks() ).
215	*/
216	class TrkLevel2 : public TObject {
217
218	private:
219
220	public:
221
222	Int_t good[12]; ///< event status
223
224	TClonesArray *Track; ///< fitted tracks
225	TClonesArray *SingletX; ///< x singlets
226	TClonesArray *SingletY; ///< y singlets
227
228	TrkLevel2();
229	// TrkLevel2(cTrkLevel2 *);
230	~TrkLevel2(){Delete();};
231
232	void Clear();
233	void Clear(Option_t *option){Clear();};
234	void Delete();
235	void Set();
236
237	int ntrk() {return Track->GetEntries();} ///< number of stored track
238	int nclsx(){return SingletX->GetEntries();} ///< number of x singlets
239	int nclsy(){return SingletY->GetEntries();} ///< number of y singlets
240
241	void Dump();
242	void SetFromLevel2Struct(cTrkLevel2 , TrkLevel1 );
243	void SetFromLevel2Struct(cTrkLevel2 *s2){ SetFromLevel2Struct(s2, NULL); };
244	void SetFromLevel2Struct(TrkLevel1 *l1) { SetFromLevel2Struct(&level2event_, l1); };
245	void SetFromLevel2Struct() { SetFromLevel2Struct(&level2event_); };
246	void GetLevel2Struct(cTrkLevel2 *) const;
247	void LoadField(TString);
248	float GetBX(float*);
249	float GetBY(float*);
250	float GetBZ(float*);
251	Float_t GetZTrk(Int_t);
252	Float_t GetXTrkLeft(){return XTRKL;};
253	Float_t GetXTrkRight(){return XTRKR;};
254	Float_t GetYTrkLeft(){return YTRKL;};
255	Float_t GetYTrkRight(){return YTRKR;};
256
257	TrkSinglet *GetSingletX(int);
258	TrkSinglet *GetSingletY(int);
259
260	TrkTrack *GetStoredTrack(int i);
261	Int_t GetSeqNo(Int_t i) {return (((TrkTrack *)Track->At(i))->seqno);}; ///< Returns track sequential number
262	// TClonesArray *GetTracks_Chi2Sorted();
263	// TClonesArray *GetTracks_NFitSorted();
264	// TClonesArray *GetTracks();
265	TRefArray *GetTracks_NFitSorted();
266	TRefArray *GetTracks(){return this->GetTracks_NFitSorted();};
267
268	// int GetNTracks(){return this->GetTracks()->GetEntries();}
269	Int_t GetNTracks();
270	TrkTrack* GetTrack(int i);
271	TrkTrack* GetTrackImage(int i);
272
273	TrkLevel2* GetTrkLevel2(){return this;}
274	TClonesArray* GetTrackArray(){return Track;};///< returns pointer to the track array
275
276	ClassDef(TrkLevel2,2);
277
278	};
279
280	#endif