CbmRichTrainAnnSelect.h

Go to the documentation of this file.

 
#ifndef CBM_RICH_TRAIN_ANN_SELECT
#define CBM_RICH_TRAIN_ANN_SELECT
 
#include "CbmRichRingFitterCOP.h"
#include "CbmRichRingSelectImpl.h"
#include "FairTask.h"
#include "TH1D.h"
#include <vector>
using std::vector;
 
//class TH1D;
//class TH1;
//class CbmRichRingFitterCOP;
//class CbmRichRingSelectImpl;
class CbmRichRing;
 
class RingSelectParam {
public:
  Double_t fNofHits;
  Double_t fAngle;
  Double_t fHitsOnRing;
  Double_t fRadPos;
  Double_t fChi2;
  Double_t fRadius;
};
 
class CbmRichTrainAnnSelect : public FairTask {
public:
  CbmRichTrainAnnSelect();
 
  virtual ~CbmRichTrainAnnSelect();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* option);
 
  virtual void FinishTask();
 
private:
  TClonesArray* fRichRings;        // Array of CbmRichRings
  TClonesArray* fMcTracks;         // Array of CbmMCTracks
  TClonesArray* fRichRingMatches;  // Array of CbmRichRingMatch
 
  Int_t fEventNumber;
  Double_t fQuota;             // Quality quota for reconstructed ring
  UInt_t fMaxNofTrainSamples;  // maximum number of train samples for ANN
  Int_t fNofFakeLikeTrue;
  Int_t fNofTrueLikeFake;
  Double_t fAnnCut;
 
  // Difference Fake and True rings histograms BEGIN
  // [0] = is true; [1] = is fake
  vector<TH1D*> fhNofHits;
  vector<TH1D*> fhAngle;
  vector<TH1D*> fhNofHitsOnRing;
  vector<TH1D*> fhChi2;
  vector<TH1D*> fhRadPos;
  vector<TH1D*> fhRadius;
 
  // ANN outputs
  // [0] = is true; [1] = is fake
  vector<TH1D*> fhAnnOutput;
  vector<TH1D*> fhCumProb;
 
  // Data for ANN input
  // [0] = true, [1] = fake
  vector<vector<RingSelectParam>> fRSParams;
 
  CbmRichRingFitterCOP* fFitCOP;       // Pointer to COP fitter
  CbmRichRingSelectImpl* fSelectImpl;  // Pointer to ellipse fitter
 
  vector<TH1*> fHists;  // Store pointer for all histograms
 
  void SetRecFlag();
 
  void DiffFakeTrueCircle();
 
  void TrainAndTestAnn();
 
  void Draw(Option_t* = "");
 
  CbmRichTrainAnnSelect(const CbmRichTrainAnnSelect&);
 
  CbmRichTrainAnnSelect& operator=(const CbmRichTrainAnnSelect&);
 
  ClassDef(CbmRichTrainAnnSelect, 1)
};
 
#endif