CbmRoot
CbmTrdSetTracksPidANN.h
Go to the documentation of this file.
1 // -------------------------------------------------------------------------
2 // ----- CbmTrdSetTracksPidANN header file -----
3 // ----- Created 06/03/2007 by Simeon Lebedev -----
4 // -------------------------------------------------------------------------
5 
6 
21 #ifndef CBM_TRD_SET_TRACKS_PID_ANN_H
22 #define CBM_TRD_SET_TRACKS_PID_ANN_H
23 
24 #include "FairTask.h"
25 
26 #include "TString.h"
27 
28 #include <vector>
29 class TMultiLayerPerceptron;
30 class TClonesArray;
31 
32 class CbmTrdSetTracksPidANN : public FairTask {
33 
34 public:
37 
38 
44  CbmTrdSetTracksPidANN(const char* name, const char* title = "FairTask");
45 
46 
48  virtual ~CbmTrdSetTracksPidANN();
49 
50 
52  virtual InitStatus Init();
53 
54 
56  virtual void Exec(Option_t* opt);
57 
58 
60  virtual void Finish();
61 
62 
64  virtual void SetParContainers();
65 
66 
68  Int_t GetNofTracks() const { return fNofTracks; };
69  Double_t GetANNPar1() const { return fANNPar1; };
70  Double_t GetANNPar2() const { return fANNPar2; };
71 
73  void SetANNPar1(Double_t param) { fANNPar1 = param; };
74  void SetANNPar2(Double_t param) { fANNPar2 = param; };
75 
77  // use SetTRDGeometryName
78  void SetInputFile(TString) {};
79 
80  //should be "mb" or "st"
81  //set the geometry, you use "mb" for Munster-Buharest; "st" for standard
82  void SetTRDGeometryType(TString trdGeometryType) {
83  fTRDGeometryType = trdGeometryType;
84  }
85 
86 
87 private:
90 
91  TClonesArray* fTrackArray; // Input array of TRD tracks
92  TClonesArray* fTrdHitArray; // Input array of TRD Hits
93 
94  Int_t fNofTracks; // Number of tracks successfully fitted
95  Double_t fANNPar1; // Parameter1 for Wkn method
96  Double_t fANNPar2; // Parameter2 for Wkn method
97 
98  std::vector<TMultiLayerPerceptron*> fNN; // Neural Network
99 
100  TString fTRDGeometryType; // name of the TRD geometry
101 
102  Bool_t ReadData(); // Read the weights needed for ANN
103 
105 };
106 
107 #endif
CbmTrdSetTracksPidANN::fNofTracks
Int_t fNofTracks
Definition: CbmTrdSetTracksPidANN.h:94
CbmTrdSetTracksPidANN::CbmTrdSetTracksPidANN
CbmTrdSetTracksPidANN()
Definition: CbmTrdSetTracksPidANN.cxx:25
CbmTrdSetTracksPidANN::SetANNPar2
void SetANNPar2(Double_t param)
Definition: CbmTrdSetTracksPidANN.h:74
CbmTrdSetTracksPidANN::ClassDef
ClassDef(CbmTrdSetTracksPidANN, 2)
CbmTrdSetTracksPidANN::GetANNPar1
Double_t GetANNPar1() const
Definition: CbmTrdSetTracksPidANN.h:69
CbmTrdSetTracksPidANN::SetTRDGeometryType
void SetTRDGeometryType(TString trdGeometryType)
Definition: CbmTrdSetTracksPidANN.h:82
CbmTrdSetTracksPidANN::SetInputFile
void SetInputFile(TString)
deprecated method, will be deleted soon
Definition: CbmTrdSetTracksPidANN.h:78
CbmTrdSetTracksPidANN::GetANNPar2
Double_t GetANNPar2() const
Definition: CbmTrdSetTracksPidANN.h:70
CbmTrdSetTracksPidANN::operator=
CbmTrdSetTracksPidANN & operator=(const CbmTrdSetTracksPidANN &)
CbmTrdSetTracksPidANN::~CbmTrdSetTracksPidANN
virtual ~CbmTrdSetTracksPidANN()
Definition: CbmTrdSetTracksPidANN.cxx:39
CbmTrdSetTracksPidANN::fANNPar1
Double_t fANNPar1
Definition: CbmTrdSetTracksPidANN.h:95
CbmTrdSetTracksPidANN::fNN
std::vector< TMultiLayerPerceptron * > fNN
Definition: CbmTrdSetTracksPidANN.h:98
CbmTrdSetTracksPidANN::GetNofTracks
Int_t GetNofTracks() const
Definition: CbmTrdSetTracksPidANN.h:68
CbmTrdSetTracksPidANN::SetANNPar1
void SetANNPar1(Double_t param)
Definition: CbmTrdSetTracksPidANN.h:73
CbmTrdSetTracksPidANN::fTrackArray
TClonesArray * fTrackArray
Definition: CbmTrdSetTracksPidANN.h:91
CbmTrdSetTracksPidANN::ReadData
Bool_t ReadData()
Definition: CbmTrdSetTracksPidANN.cxx:43
CbmTrdSetTracksPidANN::SetParContainers
virtual void SetParContainers()
Definition: CbmTrdSetTracksPidANN.cxx:41
CbmTrdSetTracksPidANN::Init
virtual InitStatus Init()
Definition: CbmTrdSetTracksPidANN.cxx:115
CbmTrdSetTracksPidANN::CbmTrdSetTracksPidANN
CbmTrdSetTracksPidANN(const CbmTrdSetTracksPidANN &)
CbmTrdSetTracksPidANN::fTRDGeometryType
TString fTRDGeometryType
Definition: CbmTrdSetTracksPidANN.h:100
CbmTrdSetTracksPidANN::fTrdHitArray
TClonesArray * fTrdHitArray
Definition: CbmTrdSetTracksPidANN.h:92
CbmTrdSetTracksPidANN
Definition: CbmTrdSetTracksPidANN.h:32
CbmTrdSetTracksPidANN::fANNPar2
Double_t fANNPar2
Definition: CbmTrdSetTracksPidANN.h:96
CbmTrdSetTracksPidANN::Finish
virtual void Finish()
Definition: CbmTrdSetTracksPidANN.cxx:176
CbmTrdSetTracksPidANN::Exec
virtual void Exec(Option_t *opt)
Definition: CbmTrdSetTracksPidANN.cxx:136