CbmTrdSetTracksPidANN.h

Go to the documentation of this file.

/* Copyright (C) 2007-2015 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev, Florian Uhlig [committer] */
 
// -------------------------------------------------------------------------
// -----                    CbmTrdSetTracksPidANN header file                -----
// -----                  Created 06/03/2007  by Simeon Lebedev              -----
// -------------------------------------------------------------------------
 
 
#ifndef CBM_TRD_SET_TRACKS_PID_ANN_H
#define CBM_TRD_SET_TRACKS_PID_ANN_H
 
#include "FairTask.h"
#include "TString.h"
 
#include <vector>
class TMultiLayerPerceptron;
class TClonesArray;
 
class CbmTrdSetTracksPidANN : public FairTask {
 
 public:
  CbmTrdSetTracksPidANN();
 
 
  CbmTrdSetTracksPidANN(const char* name, const char* title = "FairTask");
 
 
  virtual ~CbmTrdSetTracksPidANN();
 
 
  virtual InitStatus Init();
 
 
  virtual void Exec(Option_t* opt);
 
 
  virtual void Finish();
 
 
  virtual void SetParContainers();
 
 
  Int_t GetNofTracks() const { return fNofTracks; };
  Double_t GetANNPar1() const { return fANNPar1; };
  Double_t GetANNPar2() const { return fANNPar2; };
 
  void SetANNPar1(Double_t param) { fANNPar1 = param; };
  void SetANNPar2(Double_t param) { fANNPar2 = param; };
 
  // use SetTRDGeometryName
  void SetInputFile(TString){};
 
  //should be "mb" or "st"
  //set the geometry, you use "mb" for Munster-Buharest; "st" for standard
  void SetTRDGeometryType(TString trdGeometryType) { fTRDGeometryType = trdGeometryType; }
 
 
 private:
  CbmTrdSetTracksPidANN& operator=(const CbmTrdSetTracksPidANN&);
  CbmTrdSetTracksPidANN(const CbmTrdSetTracksPidANN&);
 
  TClonesArray* fTrackArray;   // Input array of TRD tracks
  TClonesArray* fTrdHitArray;  // Input array of TRD Hits
 
  Int_t fNofTracks;   // Number of tracks successfully fitted
  Double_t fANNPar1;  // Parameter1 for Wkn method
  Double_t fANNPar2;  // Parameter2 for Wkn method
 
  std::vector<TMultiLayerPerceptron*> fNN;  // Neural Network
 
  TString fTRDGeometryType;  // name of the TRD geometry
 
  Bool_t ReadData();  // Read the weights needed for ANN
 
  ClassDef(CbmTrdSetTracksPidANN, 2);
};
 
#endif