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 <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);
    void Dump();

    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:

//      TRef clx[6];
//      TRef cly[6];
        TRefArray *clx;
        TRefArray *cly;

    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];       ///<signal in MIP (scaled to 300 micrometer)
    float dedx_y[6];       ///<signal in MIP (scaled to 300 micrometer)


    TrkTrack();
    TrkTrack(const TrkTrack&);

        ~TrkTrack(){Delete();};
        
    void Dump();
        void Clear();
        void Delete();
        
    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]==1;};
        Bool_t YGood(int ip){ return ygood[ip]==1;};
        
        TrkCluster *GetClusterX(int ip){TrkCluster *pt = (TrkCluster*)(clx->At(ip)); return pt;};
        TrkCluster *GetClusterY(int ip){TrkCluster *pt = (TrkCluster*)(cly->At(ip)); return pt;};
        
    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&);

    void Dump();
        
    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:
        
//      TRefArray    *PhysicalTrack;  ///< physical tracks (no image) - 
         
 public:

        Int_t         good[12];       ///< event status
//      Int_t good2;
//    Int_t crc[12];

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

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