EIC Software
Reference for
EIC
simulation and reconstruction software on GitHub
Home page
Related Pages
Modules
Namespaces
Classes
Files
External Links
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Groups
Pages
CbmRichTrainAnnSelect.h
Go to the documentation of this file.
Or view
the newest version in sPHENIX GitHub for file CbmRichTrainAnnSelect.h
1
10
#ifndef CBM_RICH_TRAIN_ANN_SELECT
11
#define CBM_RICH_TRAIN_ANN_SELECT
12
13
#include "
FairTask.h
"
14
#include <vector>
15
#include "
CbmRichRingFitterCOP.h
"
16
#include "
CbmRichRingSelectImpl.h
"
17
#include "TH1D.h"
18
using
std::vector
;
19
20
//class TH1D;
21
//class TH1;
22
//class CbmRichRingFitterCOP;
23
//class CbmRichRingSelectImpl;
24
class
CbmRichRing
;
25
34
class
RingSelectParam
35
{
36
public
:
37
Double_t
fNofHits
;
38
Double_t
fAngle
;
39
Double_t
fHitsOnRing
;
40
Double_t
fRadPos
;
41
Double_t
fChi2
;
42
Double_t
fRadius
;
43
};
44
53
class
CbmRichTrainAnnSelect
:
public
FairTask
54
{
55
public
:
56
60
CbmRichTrainAnnSelect
();
61
65
virtual
~CbmRichTrainAnnSelect
();
66
70
virtual
InitStatus
Init
();
71
75
virtual
void
Exec
(
76
Option_t* option);
77
81
virtual
void
FinishTask
();
82
83
private
:
84
TClonesArray*
fRichRings
;
// Array of CbmRichRings
85
TClonesArray*
fMcTracks
;
// Array of CbmMCTracks
86
TClonesArray*
fRichRingMatches
;
// Array of CbmRichRingMatch
87
88
Int_t
fEventNumber
;
89
Double_t
fQuota
;
// Quality quota for reconstructed ring
90
Int_t
fMaxNofTrainSamples
;
// maximum number of train samples for ANN
91
Int_t
fNofFakeLikeTrue
;
92
Int_t
fNofTrueLikeFake
;
93
Double_t
fAnnCut
;
94
95
// Difference Fake and True rings histograms BEGIN
96
// [0] = is true; [1] = is fake
97
vector<TH1D*>
fhNofHits
;
98
vector<TH1D*>
fhAngle
;
99
vector<TH1D*>
fhNofHitsOnRing
;
100
vector<TH1D*>
fhChi2
;
101
vector<TH1D*>
fhRadPos
;
102
vector<TH1D*>
fhRadius
;
103
104
// ANN outputs
105
// [0] = is true; [1] = is fake
106
vector<TH1D*>
fhAnnOutput
;
107
vector<TH1D*>
fhCumProb
;
108
109
// Data for ANN input
110
// [0] = true, [1] = fake
111
vector<vector<RingSelectParam> >
fRSParams
;
112
113
CbmRichRingFitterCOP
*
fFitCOP
;
// Pointer to COP fitter
114
CbmRichRingSelectImpl
*
fSelectImpl
;
// Pointer to ellipse fitter
115
116
vector<TH1*>
fHists
;
// Store pointer for all histograms
117
121
void
SetRecFlag
();
122
126
void
DiffFakeTrueCircle
();
127
131
void
TrainAndTestAnn
();
132
136
void
Draw
();
137
141
CbmRichTrainAnnSelect
(
const
CbmRichTrainAnnSelect
&);
142
146
CbmRichTrainAnnSelect
&
operator=
(
const
CbmRichTrainAnnSelect
&);
147
148
ClassDef
(
CbmRichTrainAnnSelect
,1)
149
};
150
151
#endif
EicRoot
blob
master
rich
CbmRichTrainAnnSelect.h
Built by
Jin Huang
. updated:
Mon Jan 22 2024 12:43:37
using
1.8.2 with
EIC GitHub integration