CbmRichElectronIdAnn.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmRichElectronIdAnn.h

#ifndef CBM_RICH_ELECTRONID_ANN
#define CBM_RICH_ELECTRONID_ANN

#include <string>

class CbmRichRing;
class TMultiLayerPerceptron;

using std::string;

class CbmRichElectronIdAnn
{
private:
   string fAnnWeights; // path to the file with weights for ANN
   TMultiLayerPerceptron* fNN; // Pointer to the ANN
public:

   CbmRichElectronIdAnn();

   virtual ~CbmRichElectronIdAnn();

   void Init();

   double DoSelect(
         CbmRichRing* ring,
         double momentum);

   void SetAnnWeights(const string& fileName){fAnnWeights = fileName;}

private:
   CbmRichElectronIdAnn(const CbmRichElectronIdAnn&);

   CbmRichElectronIdAnn& operator=(const CbmRichElectronIdAnn&);
};

#endif