cbmroot/CbmKresTrainAnn_8h_source.html

#ifndef CBM_KRES_TRAIN_ANN

#define CBM_KRES_TRAIN_ANN


#include "TCanvas.h"

#include "TH2D.h"

#include "TVector3.h"

#include <TClonesArray.h>

#include <vector>


using namespace std;


class CbmKresTrainAnn {


public:

  //***** brief Standard constructor.

  CbmKresTrainAnn();

  //***** brief Standard destructor.

  virtual ~CbmKresTrainAnn();


  void Init();

  void InitHistograms();


  void Exec(int event,

            int IdForANN,

            double InvariantMass,

            double OpeningAngle,

            double PlaneAngle_last,

            double ZPos,

            TVector3 Momentum1,

            TVector3 Momentum2);

  void TrainAndTestAnn();

  void Draw();


private:

  unsigned int fMaxNofTrainSamples;

  double fAnnCut;

  int fNofWrongLikeCorrect;

  int fNofCorrectLikeWrong;


  vector<double> IM_correct;

  vector<double> OA_correct;

  vector<double> Angle_correct;

  vector<double> Z_correct;

  vector<double> Mom1_correct;

  vector<double> Mom2_correct;

  vector<double> IM_wrong;

  vector<double> OA_wrong;

  vector<double> Angle_wrong;

  vector<double> Z_wrong;

  vector<double> Mom1_wrong;

  vector<double> Mom2_wrong;


  vector<TH1*> fHists;

  TH1D* fhAnnOutput_correct;

  TH1D* fhAnnOutput_wrong;

  TH1D* fhCumProb_correct;

  TH1D* fhCumProb_wrong;


  //***** brief Copy constructor.

  CbmKresTrainAnn(const CbmKresTrainAnn&);


  //***** brief Assignment operator.

  CbmKresTrainAnn operator=(const CbmKresTrainAnn&);


  ClassDef(CbmKresTrainAnn, 1)

};


#endif