CaloAxis/inc/CaloAxis.h

/**
 * \file CaloAxis.h
 * \author Elena Vannuccini
 */
#if !defined(__CINT__) || defined(__MAKECINT__)

#include <TString.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TMath.h>
#include <TLine.h>
#include <TPolyMarker.h>
#include <TSelector.h>
#include <TFile.h>

#include <stdlib.h>
#include <iostream>
using namespace std;

#include <PamLevel2.h>

#endif
//===============================================================================
//
//
//
//
//===============================================================================
/**
 * \brief Class to store the calorimeter hit pattern on the x or y view
 */
class CaloEvent : public TObject{

public:
//  calo event    
    vector<int> ids;  //strip 0-95
    vector<int> idp;  //plane 0-21
    vector<float> q; // signal
    vector<float> zcoord;  
    vector<float> xycoord; 

    
    void Add(int iis, int iip, float fq, float fz, float fxy);
    int  GetN(){ return (int)(q.size()); };

    void Reset();
    void Delete(){ Reset(); };

    ~CaloEvent(){ Delete(); };

    CaloEvent(){ Reset(); };
    CaloEvent( CaloLevel1* calo, int view, Bool_t usemechanicalalignement){ Set(calo,view,usemechanicalalignement); };
    CaloEvent( CaloLevel1* calo, int view){ Set(calo,view); };

    void Set(CaloLevel1* calo, int view, Bool_t usemechanicalalignement);
    void Set(CaloLevel1* calo, int view){ Set(calo,view,0); };

    ClassDef(CaloEvent,1);

};
//===============================================================================
//
//
//
//
//===============================================================================
/**
 * \brief Class to fit an axis through a calorimeter event. 
 *
 * The fit consists in an iterative procedure, where some hits are iterativelly 
 * excluded according to a given tolerance. Two methods are implemented to get the axis:
 * CaloAxis::FitAxis(...)   -- optimized for non-interacting particles
 * CaloAxis::FitShower(...) -- optimized for interacting particles
 *  
 */
class CaloAxis : public TObject {

public:

    CaloEvent *cevent;

//  fitted points
    vector<float> x;///< independent coordinates (z)
    vector<float> y;///< dependente coordinate (x/y)
    vector<float> w;///< fit weight
    vector<float> q;///< signal

    Float_t par[2];///< axis parameters: y = par[0] + x*par[1]
    Float_t covpar[2][2];///< parameter covariance matrix

    TPolyMarker *sel;
    TLine       *axis;
    
    float cog[22]; ///< baricenter (hits associated to the axis only)
    float qpl[22]; ///< total charge in each plane (hits associated to the axis only)
    
    void Init();

    void Set(CaloLevel1* calo, Int_t view){ cevent->Set(calo,view,0); };
    void Set(CaloLevel1* calo, Int_t view, Bool_t usemechanicalalignement){ cevent->Set(calo,view,usemechanicalalignement); };

    CaloAxis(){ cevent = new CaloEvent(); Init(); };
    CaloAxis(CaloLevel1* calo, Int_t view){ cevent = new CaloEvent(); Init(); Set(calo,view); };
    CaloAxis(CaloLevel1* calo, Int_t view, Bool_t usemechanicalalignement){ cevent = new CaloEvent(); Init(); Set(calo,view,usemechanicalalignement); };
    
    void Reset();

    void Delete(){ Reset(); delete cevent; };
    ~CaloAxis(){ Delete(); };

    void Add(float xin, float yin);
    void Add(float xin, float yin, float qin);
    void Add(float xin, float yin, float qin, float win);

    int Fit();

    void Print();

    float GetPar(Int_t i){return par[i];};
    int   GetN(){return (int)(x.size());};
    float* GetX();
    float* GetY();
    float* GetQ();
    float GetChi2();
    float GetQaxis();           ///< signal along the axis
    float GetQaxis(float toll); ///< signal along the axis
    float GetCOG(Int_t ip){ if(ip>=0 && ip<22)return cog[ip]; else return 0;}; ///< COG vs plane
    float GetQ(Int_t ip)  { if(ip>=0 && ip<22)return qpl[ip]; else return 0;}; ///< signal vs plane
    float GetQstrip();          ///< average signal per strip
    float GetYfit(float x){ return par[0]+x*par[1]; }; ///< extrapolated axis coordinate

    float GetXmin();
    float GetXmax();

    void Draw(int col);
    void Draw(){ Draw(5); };

    int FitAxis( CaloLevel1* calo , Int_t view , Float_t toll , Bool_t usemechanicalalignement );    
    int FitAxis( CaloLevel1* calo , Int_t view , Float_t toll ){ return FitAxis(calo,view,toll,0); };

    int FitShower( CaloLevel1* calo , Int_t view , Float_t toll , Bool_t usemechanicalalignement );    
    int FitShower( CaloLevel1* calo , Int_t view , Float_t toll ){ return FitShower(calo,view,toll,0); };

    ClassDef(CaloAxis,1);


};


//===============================================================================
//
//
//
//
//===============================================================================
1	mocchiut	1.1	/**
2			* \file CaloAxis.h
3			* \author Elena Vannuccini
4			*/
5			#if !defined(__CINT__) \|\| defined(__MAKECINT__)
6
7			#include <TString.h>
8			#include <TH1F.h>
9			#include <TH2F.h>
10			#include <TMath.h>
11			#include <TLine.h>
12			#include <TPolyMarker.h>
13			#include <TSelector.h>
14			#include <TFile.h>
15
16			#include <stdlib.h>
17			#include <iostream>
18			using namespace std;
19
20			#include <PamLevel2.h>
21
22			#endif
23			//===============================================================================
24			//
25			//
26			//
27			//
28			//===============================================================================
29			/**
30			* \brief Class to store the calorimeter hit pattern on the x or y view
31			*/
32			class CaloEvent : public TObject{
33
34			public:
35			// calo event
36			vector<int> ids; //strip 0-95
37			vector<int> idp; //plane 0-21
38			vector<float> q; // signal
39			vector<float> zcoord;
40			vector<float> xycoord;
41
42
43			void Add(int iis, int iip, float fq, float fz, float fxy);
44			int GetN(){ return (int)(q.size()); };
45
46			void Reset();
47			void Delete(){ Reset(); };
48
49			~CaloEvent(){ Delete(); };
50
51			CaloEvent(){ Reset(); };
52			CaloEvent( CaloLevel1* calo, int view, Bool_t usemechanicalalignement){ Set(calo,view,usemechanicalalignement); };
53			CaloEvent( CaloLevel1* calo, int view){ Set(calo,view); };
54
55			void Set(CaloLevel1* calo, int view, Bool_t usemechanicalalignement);
56			void Set(CaloLevel1* calo, int view){ Set(calo,view,0); };
57
58			ClassDef(CaloEvent,1);
59
60			};
61			//===============================================================================
62			//
63			//
64			//
65			//
66			//===============================================================================
67			/**
68			* \brief Class to fit an axis through a calorimeter event.
69			*
70			* The fit consists in an iterative procedure, where some hits are iterativelly
71			* excluded according to a given tolerance. Two methods are implemented to get the axis:
72			* CaloAxis::FitAxis(...) -- optimized for non-interacting particles
73			* CaloAxis::FitShower(...) -- optimized for interacting particles
74			*
75			*/
76			class CaloAxis : public TObject {
77
78			public:
79
80			CaloEvent *cevent;
81
82			// fitted points
83			vector<float> x;///< independent coordinates (z)
84			vector<float> y;///< dependente coordinate (x/y)
85			vector<float> w;///< fit weight
86			vector<float> q;///< signal
87
88			Float_t par[2];///< axis parameters: y = par[0] + x*par[1]
89			Float_t covpar[2][2];///< parameter covariance matrix
90
91			TPolyMarker *sel;
92			TLine *axis;
93
94			float cog[22]; ///< baricenter (hits associated to the axis only)
95			float qpl[22]; ///< total charge in each plane (hits associated to the axis only)
96
97			void Init();
98
99			void Set(CaloLevel1* calo, Int_t view){ cevent->Set(calo,view,0); };
100			void Set(CaloLevel1* calo, Int_t view, Bool_t usemechanicalalignement){ cevent->Set(calo,view,usemechanicalalignement); };
101
102			CaloAxis(){ cevent = new CaloEvent(); Init(); };
103			CaloAxis(CaloLevel1* calo, Int_t view){ cevent = new CaloEvent(); Init(); Set(calo,view); };
104			CaloAxis(CaloLevel1* calo, Int_t view, Bool_t usemechanicalalignement){ cevent = new CaloEvent(); Init(); Set(calo,view,usemechanicalalignement); };
105
106			void Reset();
107
108			void Delete(){ Reset(); delete cevent; };
109			~CaloAxis(){ Delete(); };
110
111			void Add(float xin, float yin);
112			void Add(float xin, float yin, float qin);
113			void Add(float xin, float yin, float qin, float win);
114
115			int Fit();
116
117			void Print();
118
119			float GetPar(Int_t i){return par[i];};
120			int GetN(){return (int)(x.size());};
121			float* GetX();
122			float* GetY();
123			float* GetQ();
124			float GetChi2();
125			float GetQaxis(); ///< signal along the axis
126			float GetQaxis(float toll); ///< signal along the axis
127			float GetCOG(Int_t ip){ if(ip>=0 && ip<22)return cog[ip]; else return 0;}; ///< COG vs plane
128			float GetQ(Int_t ip) { if(ip>=0 && ip<22)return qpl[ip]; else return 0;}; ///< signal vs plane
129			float GetQstrip(); ///< average signal per strip
130			float GetYfit(float x){ return par[0]+x*par[1]; }; ///< extrapolated axis coordinate
131
132			float GetXmin();
133			float GetXmax();
134
135			void Draw(int col);
136			void Draw(){ Draw(5); };
137
138			int FitAxis( CaloLevel1* calo , Int_t view , Float_t toll , Bool_t usemechanicalalignement );
139			int FitAxis( CaloLevel1* calo , Int_t view , Float_t toll ){ return FitAxis(calo,view,toll,0); };
140
141			int FitShower( CaloLevel1* calo , Int_t view , Float_t toll , Bool_t usemechanicalalignement );
142			int FitShower( CaloLevel1* calo , Int_t view , Float_t toll ){ return FitShower(calo,view,toll,0); };
143
144			ClassDef(CaloAxis,1);
145
146
147			};
148
149
150			//===============================================================================
151			//
152			//
153			//
154			//
155			//===============================================================================