CbmRoot
Public Member Functions | Private Member Functions | Private Attributes | List of all members
CbmRichTrainAnnElectrons Class Reference

Train ANN for electron identification in RICH. More...

#include <CbmRichTrainAnnElectrons.h>

Inheritance diagram for CbmRichTrainAnnElectrons:
[legend]
Collaboration diagram for CbmRichTrainAnnElectrons:
[legend]

Public Member Functions

 CbmRichTrainAnnElectrons ()
 Default constructor. More...
 
virtual ~CbmRichTrainAnnElectrons ()
 Destructor. More...
 
virtual InitStatus Init ()
 Inherited from FairTask. More...
 
virtual void Exec (Option_t *option)
 Inherited from FairTask. More...
 
virtual void FinishTask ()
 Inherited from FairTask. More...
 

Private Member Functions

void DiffElandPi ()
 Fill input parameters for ANN in array and histograms. More...
 
void TrainAndTestAnn ()
 Train and test ANN. More...
 
void Draw (Option_t *="")
 Draw results. More...
 
TCanvas * CreateCanvas (const string &name, const string &title, int width, int height)
 
void SaveCanvasToImage ()
 
 CbmRichTrainAnnElectrons (const CbmRichTrainAnnElectrons &)
 Copy constructor. More...
 
CbmRichTrainAnnElectrons & operator= (const CbmRichTrainAnnElectrons &)
 Assignment operator. More...
 

Private Attributes

Int_t fEventNum
 
TClonesArray * fRichHits
 
TClonesArray * fRichRings
 
TClonesArray * fRichPoints
 
TClonesArray * fMCTracks
 
TClonesArray * fRichRingMatches
 
TClonesArray * fRichProj
 
TClonesArray * fStsTrackMatches
 
TClonesArray * fGlobalTracks
 
TClonesArray * fStsTracks
 
Int_t fMinNofHitsInRichRing
 
Double_t fQuota
 
UInt_t fMaxNofTrainSamples
 
Int_t fNofPiLikeEl
 
Int_t fNofElLikePi
 
Double_t fAnnCut
 
vector< TH1D * > fhAnnOutput
 
vector< TH1D * > fhCumProb
 
vector< vector< RingElectronParam > > fRElIdParams
 
vector< TCanvas * > fCanvas
 
string fOutputDir
 
vector< TH1D * > fhAaxis
 
vector< TH1D * > fhBaxis
 
vector< TH1D * > fhDistTrueMatch
 
vector< TH1D * > fhDistMisMatch
 
vector< TH1D * > fhNofHits
 
vector< TH1D * > fhChi2
 
vector< TH1D * > fhRadPos
 
vector< TH2D * > fhAaxisVsMom
 
vector< TH2D * > fhBaxisVsMom
 
vector< TH2D * > fhPhiVsRadAng
 
vector< TH1 * > fHists
 

Detailed Description

Train ANN for electron identification in RICH.

Author
Semen Lebedev s.leb.nosp@m.edev.nosp@m.@gsi..nosp@m.de
Date
2011

Definition at line 57 of file CbmRichTrainAnnElectrons.h.

Constructor & Destructor Documentation

◆ CbmRichTrainAnnElectrons() [1/2]

CbmRichTrainAnnElectrons::CbmRichTrainAnnElectrons ( )

Default constructor.

Definition at line 46 of file CbmRichTrainAnnElectrons.cxx.

References fhAaxis, fhAaxisVsMom, fhAnnOutput, fhBaxis, fhBaxisVsMom, fhChi2, fhCumProb, fhDistMisMatch, fhDistTrueMatch, fHists, fhNofHits, fhPhiVsRadAng, fhRadPos, fRElIdParams, and i.

◆ ~CbmRichTrainAnnElectrons()

CbmRichTrainAnnElectrons::~CbmRichTrainAnnElectrons ( )
virtual

Destructor.

Definition at line 192 of file CbmRichTrainAnnElectrons.cxx.

◆ CbmRichTrainAnnElectrons() [2/2]

CbmRichTrainAnnElectrons::CbmRichTrainAnnElectrons ( const CbmRichTrainAnnElectrons &  )
private

Copy constructor.

Member Function Documentation

◆ CreateCanvas()

TCanvas * CbmRichTrainAnnElectrons::CreateCanvas ( const string &  name,
const string &  title,
int  width,
int  height 
)
private

Definition at line 624 of file CbmRichTrainAnnElectrons.cxx.

References fCanvas.

Referenced by Draw().

◆ DiffElandPi()

void CbmRichTrainAnnElectrons::DiffElandPi ( )
private

Fill input parameters for ANN in array and histograms.

Definition at line 252 of file CbmRichTrainAnnElectrons.cxx.

References RingElectronParam::fAaxis, RingElectronParam::fBaxis, RingElectronParam::fChi2, RingElectronParam::fDistance, fGlobalTracks, fhAaxis, fhAaxisVsMom, fhBaxis, fhBaxisVsMom, fhChi2, fhDistMisMatch, fhDistTrueMatch, fhNofHits, fhPhiVsRadAng, fhRadPos, fMCTracks, RingElectronParam::fMomentum, RingElectronParam::fNofHits, RingElectronParam::fPhi, fQuota, RingElectronParam::fRadAngle, RingElectronParam::fRadPos, fRElIdParams, fRichRingMatches, fRichRings, fStsTrackMatches, fStsTracks, CbmRichRing::GetAaxis(), CbmRichRing::GetBaxis(), CbmRichRing::GetChi2(), CbmLink::GetIndex(), CbmMatch::GetMatchedLink(), CbmMCTrack::GetMotherId(), CbmRichRing::GetNDF(), CbmRichRing::GetNofHits(), CbmTrackMatchNew::GetNofTrueHits(), CbmTrackMatchNew::GetNofWrongHits(), CbmMCTrack::GetP(), CbmMCTrack::GetPdgCode(), CbmRichRing::GetPhi(), CbmRichRing::GetRadialAngle(), CbmRichRing::GetRadialPosition(), CbmGlobalTrack::GetRichRingIndex(), CbmRichUtil::GetRingTrackDistance(), and CbmGlobalTrack::GetStsTrackIndex().

Referenced by Exec().

◆ Draw()

void CbmRichTrainAnnElectrons::Draw ( Option_t *  = "")
private

Draw results.

Definition at line 434 of file CbmRichTrainAnnElectrons.cxx.

References CreateCanvas(), DrawH1(), DrawH2(), fAnnCut, fhAaxis, fhAaxisVsMom, fhAnnOutput, fhBaxis, fhBaxisVsMom, fhChi2, fhCumProb, fhDistMisMatch, fhDistTrueMatch, fhNofHits, fhPhiVsRadAng, fhRadPos, fNofElLikePi, fNofPiLikeEl, fRElIdParams, i, kLinear, kLog, and SetDefaultDrawStyle().

Referenced by FinishTask().

◆ Exec()

void CbmRichTrainAnnElectrons::Exec ( Option_t *  option)
virtual

Inherited from FairTask.

Definition at line 244 of file CbmRichTrainAnnElectrons.cxx.

References DiffElandPi(), fEventNum, and fRElIdParams.

◆ FinishTask()

void CbmRichTrainAnnElectrons::FinishTask ( )
virtual

Inherited from FairTask.

Definition at line 612 of file CbmRichTrainAnnElectrons.cxx.

References Draw(), fHists, i, SaveCanvasToImage(), and TrainAndTestAnn().

◆ Init()

InitStatus CbmRichTrainAnnElectrons::Init ( )
virtual

Inherited from FairTask.

Definition at line 194 of file CbmRichTrainAnnElectrons.cxx.

References fGlobalTracks, fMCTracks, fRichRingMatches, fRichRings, fStsTrackMatches, and fStsTracks.

◆ operator=()

CbmRichTrainAnnElectrons& CbmRichTrainAnnElectrons::operator= ( const CbmRichTrainAnnElectrons &  )
private

Assignment operator.

◆ SaveCanvasToImage()

void CbmRichTrainAnnElectrons::SaveCanvasToImage ( )
private

Definition at line 633 of file CbmRichTrainAnnElectrons.cxx.

References fCanvas, fOutputDir, i, and Cbm::SaveCanvasAsImage().

Referenced by FinishTask().

◆ TrainAndTestAnn()

void CbmRichTrainAnnElectrons::TrainAndTestAnn ( )
private

Train and test ANN.

Definition at line 353 of file CbmRichTrainAnnElectrons.cxx.

References fAnnCut, fhAnnOutput, fMaxNofTrainSamples, fNofElLikePi, fNofPiLikeEl, fRElIdParams, i, and x.

Referenced by FinishTask().

Member Data Documentation

◆ fAnnCut

Double_t CbmRichTrainAnnElectrons::fAnnCut
private

Definition at line 103 of file CbmRichTrainAnnElectrons.h.

Referenced by Draw(), and TrainAndTestAnn().

◆ fCanvas

vector<TCanvas*> CbmRichTrainAnnElectrons::fCanvas
private

Definition at line 113 of file CbmRichTrainAnnElectrons.h.

Referenced by CreateCanvas(), and SaveCanvasToImage().

◆ fEventNum

Int_t CbmRichTrainAnnElectrons::fEventNum
private

Definition at line 86 of file CbmRichTrainAnnElectrons.h.

Referenced by Exec().

◆ fGlobalTracks

TClonesArray* CbmRichTrainAnnElectrons::fGlobalTracks
private

Definition at line 94 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi(), and Init().

◆ fhAaxis

vector<TH1D*> CbmRichTrainAnnElectrons::fhAaxis
private

Definition at line 118 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhAaxisVsMom

vector<TH2D*> CbmRichTrainAnnElectrons::fhAaxisVsMom
private

Definition at line 130 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhAnnOutput

vector<TH1D*> CbmRichTrainAnnElectrons::fhAnnOutput
private

Definition at line 106 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), Draw(), and TrainAndTestAnn().

◆ fhBaxis

vector<TH1D*> CbmRichTrainAnnElectrons::fhBaxis
private

Definition at line 119 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhBaxisVsMom

vector<TH2D*> CbmRichTrainAnnElectrons::fhBaxisVsMom
private

Definition at line 131 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhChi2

vector<TH1D*> CbmRichTrainAnnElectrons::fhChi2
private

Definition at line 127 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhCumProb

vector<TH1D*> CbmRichTrainAnnElectrons::fhCumProb
private

Definition at line 107 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), and Draw().

◆ fhDistMisMatch

vector<TH1D*> CbmRichTrainAnnElectrons::fhDistMisMatch
private

Definition at line 125 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhDistTrueMatch

vector<TH1D*> CbmRichTrainAnnElectrons::fhDistTrueMatch
private

Definition at line 123 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fHists

vector<TH1*> CbmRichTrainAnnElectrons::fHists
private

Definition at line 134 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), and FinishTask().

◆ fhNofHits

vector<TH1D*> CbmRichTrainAnnElectrons::fhNofHits
private

Definition at line 126 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhPhiVsRadAng

vector<TH2D*> CbmRichTrainAnnElectrons::fhPhiVsRadAng
private

Definition at line 132 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fhRadPos

vector<TH1D*> CbmRichTrainAnnElectrons::fhRadPos
private

Definition at line 129 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), and Draw().

◆ fMaxNofTrainSamples

UInt_t CbmRichTrainAnnElectrons::fMaxNofTrainSamples
private

Definition at line 99 of file CbmRichTrainAnnElectrons.h.

Referenced by TrainAndTestAnn().

◆ fMCTracks

TClonesArray* CbmRichTrainAnnElectrons::fMCTracks
private

Definition at line 90 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi(), and Init().

◆ fMinNofHitsInRichRing

Int_t CbmRichTrainAnnElectrons::fMinNofHitsInRichRing
private

Definition at line 97 of file CbmRichTrainAnnElectrons.h.

◆ fNofElLikePi

Int_t CbmRichTrainAnnElectrons::fNofElLikePi
private

Definition at line 102 of file CbmRichTrainAnnElectrons.h.

Referenced by Draw(), and TrainAndTestAnn().

◆ fNofPiLikeEl

Int_t CbmRichTrainAnnElectrons::fNofPiLikeEl
private

Definition at line 101 of file CbmRichTrainAnnElectrons.h.

Referenced by Draw(), and TrainAndTestAnn().

◆ fOutputDir

string CbmRichTrainAnnElectrons::fOutputDir
private

Definition at line 114 of file CbmRichTrainAnnElectrons.h.

Referenced by SaveCanvasToImage().

◆ fQuota

Double_t CbmRichTrainAnnElectrons::fQuota
private

Definition at line 98 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi().

◆ fRElIdParams

vector<vector<RingElectronParam> > CbmRichTrainAnnElectrons::fRElIdParams
private

Definition at line 111 of file CbmRichTrainAnnElectrons.h.

Referenced by CbmRichTrainAnnElectrons(), DiffElandPi(), Draw(), Exec(), and TrainAndTestAnn().

◆ fRichHits

TClonesArray* CbmRichTrainAnnElectrons::fRichHits
private

Definition at line 87 of file CbmRichTrainAnnElectrons.h.

◆ fRichPoints

TClonesArray* CbmRichTrainAnnElectrons::fRichPoints
private

Definition at line 89 of file CbmRichTrainAnnElectrons.h.

◆ fRichProj

TClonesArray* CbmRichTrainAnnElectrons::fRichProj
private

Definition at line 92 of file CbmRichTrainAnnElectrons.h.

◆ fRichRingMatches

TClonesArray* CbmRichTrainAnnElectrons::fRichRingMatches
private

Definition at line 91 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi(), and Init().

◆ fRichRings

TClonesArray* CbmRichTrainAnnElectrons::fRichRings
private

Definition at line 88 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi(), and Init().

◆ fStsTrackMatches

TClonesArray* CbmRichTrainAnnElectrons::fStsTrackMatches
private

Definition at line 93 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi(), and Init().

◆ fStsTracks

TClonesArray* CbmRichTrainAnnElectrons::fStsTracks
private

Definition at line 95 of file CbmRichTrainAnnElectrons.h.

Referenced by DiffElandPi(), and Init().

The documentation for this class was generated from the following files: