CaloAxis/inc/CaloAxis.h

/**
 * \file CaloAxis.h
 * \author Elena Vannuccini
 */

#ifndef CALOAXIS_H
#define CALOAXIS_H

#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){ Reset(); Set(calo,view,usemechanicalalignement); };
    CaloEvent( CaloLevel1* calo, int view){ Reset(); 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);


};

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