CbmRichElectronIdAnn.h

Go to the documentation of this file.

/* Copyright (C) 2008-2020 UGiessen/JINR-LIT, Giessen/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev [committer], Andrey Lebedev */
 
#ifndef CBM_RICH_ELECTRONID_ANN
#define CBM_RICH_ELECTRONID_ANN
 
#include "TClonesArray.h"  // for ROOTCLING
 
#include <string>
 
class CbmRichRing;
class TMultiLayerPerceptron;
 
using std::string;
 
class CbmRichElectronIdAnn {
private:
  CbmRichElectronIdAnn();
 
public:
  static CbmRichElectronIdAnn& GetInstance()
  {
    static CbmRichElectronIdAnn fInstance;
    return fInstance;
  }
 
 
  virtual ~CbmRichElectronIdAnn();
 
  double CalculateAnnValue(int globalTrackIndex, double momentum);
 
  //void SetAnnWeights(const string& fileName){fAnnWeights = fileName;}
 
private:
  string fAnnWeights;          // path to the file with weights for ANN
  TMultiLayerPerceptron* fNN;  // Pointer to the ANN
 
  TClonesArray* fGlobalTracks;
  TClonesArray* fRichRings;
 
  void Init();
 
private:
  CbmRichElectronIdAnn(const CbmRichElectronIdAnn&);
 
  CbmRichElectronIdAnn& operator=(const CbmRichElectronIdAnn&);
};
 
#endif