CbmRichElectronIdAnn.cxx

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

#include "CbmRichElectronIdAnn.h"
#include "CbmRichRing.h"
#include "TMultiLayerPerceptron.h"
#include "TTree.h"
#include "TMath.h"
#include "TSystem.h"

#include <iostream>

using std::cout;
using std::endl;

CbmRichElectronIdAnn::CbmRichElectronIdAnn():
   fAnnWeights(""),
   fNN(NULL)
{
   fAnnWeights = string(gSystem->Getenv("VMCWORKDIR")) + "/parameters/rich/rich_elid_ann_weights.txt";
}

CbmRichElectronIdAnn::~CbmRichElectronIdAnn()
{

}

void CbmRichElectronIdAnn::Init()
{
   TTree *simu = new TTree ("MonteCarlo","MontecarloData");
   Double_t x[9];
   Double_t xOut;

   simu->Branch("x0", &x[0],"x0/D");
   simu->Branch("x1", &x[1],"x1/D");
   simu->Branch("x2", &x[2],"x2/D");
   simu->Branch("x3", &x[3],"x3/D");
   simu->Branch("x4", &x[4],"x4/D");
   simu->Branch("x5", &x[5],"x5/D");
   simu->Branch("x6", &x[6],"x6/D");
   simu->Branch("x7", &x[7],"x7/D");
   simu->Branch("x8", &x[8],"x8/D");
   simu->Branch("xOut", &xOut,"xOut/D");

   fNN = new TMultiLayerPerceptron("x0,x1,x2,x3,x4,x5,x6,x7,x8:18:xOut",simu);
   cout << "-I- CbmRichElIdAnn: get NeuralNet weight parameters from: " << fAnnWeights << endl;
   fNN->LoadWeights(fAnnWeights.c_str());
}

double CbmRichElectronIdAnn::DoSelect(
      CbmRichRing* ring,
      double momentum)
{
    if (ring->GetAaxis() >= 10. || ring->GetAaxis() <= 0. ||
        ring->GetBaxis() >= 10. || ring->GetBaxis() <= 0. ||
        ring->GetNofHits() <= 5. ||
        ring->GetDistance() <= 0. || ring->GetDistance() >= 999. ||
        ring->GetRadialPosition() <= 0. || ring->GetRadialPosition() >= 999. ||
        ring->GetPhi() <= -6.5 || ring->GetPhi() >= 6.5 ||
        ring->GetRadialAngle() <=-6.5 || ring->GetRadialAngle() >= 6.5 ){

        return -1.;
    }
    double params[9];
    params[0] = ring->GetAaxis() / 10.;
    params[1] = ring->GetBaxis() / 10.;
    params[2] = (ring->GetPhi() + 1.57) / 3.14;
    params[3] = ring->GetRadialAngle() / 6.28;
    params[4] = (ring->GetChi2()/ring->GetNDF()) / 1.2;
    params[5] = ring->GetRadialPosition() / 110.;
    params[6] = ring->GetNofHits() / 45.;
    params[7] = ring->GetDistance() / 5.;
    params[8] =  momentum / 15.;

    for (int k = 0; k < 9; k++){
       if (params[k] < 0.0) params[k] = 0.0;
       if (params[k] > 1.0) params[k] = 1.0;
    }

    double nnEval = fNN->Evaluate(0, params);

    if (TMath::IsNaN(nnEval) == 1) {
        cout << " -W- CbmRichElectronIdAnn: nnEval nan " << endl;
        nnEval = -1.;
    }

    return nnEval;
}