From 2ce872f026cad7face24a538dcd34022a0a5dde8 Mon Sep 17 00:00:00 2001 From: cblume Date: Wed, 6 May 2009 11:56:50 +0000 Subject: [PATCH] Remove CreateGraphs() and AliTRDpidRefMaker(const Char_t*,cons tChar_t*) --- TRD/qaRec/AliTRDpidRefMaker.h | 245 +++++++++++++++++----------------- 1 file changed, 122 insertions(+), 123 deletions(-) diff --git a/TRD/qaRec/AliTRDpidRefMaker.h b/TRD/qaRec/AliTRDpidRefMaker.h index 2174c8b2f62..8c247c1ccf6 100644 --- a/TRD/qaRec/AliTRDpidRefMaker.h +++ b/TRD/qaRec/AliTRDpidRefMaker.h @@ -1,123 +1,122 @@ -#ifndef ALITRDPIDREFMAKER_H -#define ALITRDPIDREFMAKER_H - -////////////////////////////////////////////////////// -// -// Task to build PID reference tree for the training -// of neural networs for the TRD PID -// -// Authors: Alex Wilk -// Markus Heide -// -/////////////////////////////////////////////////////// - -#ifndef ALITRDRECOTASK_H -#include "AliTRDrecoTask.h" -#endif - -#ifndef ALIPID_H -#include "AliPID.h" -#endif - -#ifndef ALIPIDCALPID_H -#include "Cal/AliTRDCalPID.h" -#endif - -#ifndef ALITRDGEOMETRY_H -#include "AliTRDgeometry.h" -#endif - -class TTree; -class TEventList; -class TMultiLayerPerceptron; -class AliPID; -class AliTRDtrackV1; -class AliTRDReconstructor; -class AliTRDpidRefMaker : public AliTRDrecoTask -{ - -public: - enum { - k006 = 0 - ,k008 = 1 - ,k010 = 2 - ,k015 = 3 - ,k020 = 4 - ,k030 = 5 - ,k040 = 6 - ,k050 = 7 - ,k060 = 8 - ,k080 = 9 - ,k100 = 10 - ,kAll = 11 - }; - - enum { - kGraphTrain = 0 - ,kGraphTest = 1 - }; - - enum { - kMoniTrain = 50 - }; - - AliTRDpidRefMaker(); - AliTRDpidRefMaker(const Char_t *InFileNN, const Char_t *InFileLQ); - - virtual ~AliTRDpidRefMaker(); - - void CreateOutputObjects(); - void Exec(Option_t *option); - Int_t GetEpochs() {return fEpochs;}; - Int_t GetMinTrain() {return fMinTrain;}; - Int_t GetTrainMomBin() {return fTrainMomBin;}; - - Bool_t PostProcess(); - - void SetEpochs(Int_t epochs) {fEpochs = epochs;}; - void SetMinTrain(Int_t mintrain) {fMinTrain = mintrain;}; - void SetTrainMomBin(Int_t trainmombin) {fTrainMomBin = trainmombin;}; - void SetDate(Int_t date) {fDate = date;}; - void SetDoTraining(Bool_t train) {fDoTraining = train;}; - void SetContinueTraining(Bool_t continTrain) {fContinueTraining = continTrain;}; - void SetTrainPath(Int_t path) {fTrainPath = path;}; - void LoadFiles(const Char_t *InFileNN, const Char_t *InFileLQ); - - void Terminate(Option_t *); - - void MakeTrainingLists(); // build the training and the test list - void MonitorTraining(Int_t mombin); // monitor training process - void LoadContainer(const Char_t *InFileCont); - void CreateGraphs(); - -private: - AliTRDpidRefMaker(const AliTRDpidRefMaker&); // not implemented - AliTRDpidRefMaker& operator=(const AliTRDpidRefMaker&); // not implemented - - void GetV0info(AliTRDtrackV1 *TRDtrack, Float_t *v0pdg); // get the v0 information - void TrainNetworks(Int_t mombin); // train the neural networks for a given momentum bin - void BuildLQRefs(Int_t mombin); // build the 2dim histos for a given momentum bin - - AliTRDReconstructor *fReconstructor; //! reconstructor needed for recalculation the PID - TTree *fNN; // NN data - TTree *fLQ; // LQ data - TEventList *fTrain[AliTRDCalPID::kNMom][AliTRDgeometry::kNlayer]; // Training list for each momentum - TEventList *fTest[AliTRDCalPID::kNMom][AliTRDgeometry::kNlayer]; // Test list for each momentum - TMultiLayerPerceptron *fNet[AliTRDgeometry::kNlayer]; // artificial neural network - - Int_t fLayer; // TRD layer index - Int_t fTrainMomBin; // momentum bin for the training - Int_t fEpochs; // Number of epochs for the training of the NNs - Int_t fMinTrain; // minimum of events needed for training - Int_t fDate; // date stamp for training of the NNs - Float_t fMom; // momentum - Float_t *fdEdx[10]; // dEdx array - Float_t fv0pid[AliPID::kSPECIES]; // pid from v0s - Bool_t fDoTraining; // checks if training will be done - Bool_t fContinueTraining; // checks if training from an older run should be continued - Int_t fTrainPath; // sets the path for continuing the training - - ClassDef(AliTRDpidRefMaker, 1); // TRD reference maker for NN -}; - -#endif +#ifndef ALITRDPIDREFMAKER_H +#define ALITRDPIDREFMAKER_H + +////////////////////////////////////////////////////// +// +// Task to build PID reference tree for the training +// of neural networs for the TRD PID +// +// Authors: Alex Wilk +// Markus Heide +// +/////////////////////////////////////////////////////// + +#ifndef ALITRDRECOTASK_H +#include "AliTRDrecoTask.h" +#endif + +#ifndef ALIPID_H +#include "AliPID.h" +#endif + +#ifndef ALIPIDCALPID_H +#include "Cal/AliTRDCalPID.h" +#endif + +#ifndef ALITRDGEOMETRY_H +#include "AliTRDgeometry.h" +#endif + +class TTree; +class TEventList; +class TMultiLayerPerceptron; +class AliPID; +class AliTRDtrackV1; +class AliTRDReconstructor; +class AliTRDpidRefMaker : public AliTRDrecoTask +{ + +public: + enum { + k006 = 0 + ,k008 = 1 + ,k010 = 2 + ,k015 = 3 + ,k020 = 4 + ,k030 = 5 + ,k040 = 6 + ,k050 = 7 + ,k060 = 8 + ,k080 = 9 + ,k100 = 10 + ,kAll = 11 + }; + + enum { + kGraphTrain = 0 + ,kGraphTest = 1 + }; + + enum { + kMoniTrain = 50 + }; + + AliTRDpidRefMaker(); + + virtual ~AliTRDpidRefMaker(); + + void CreateOutputObjects(); + void Exec(Option_t *option); + Int_t GetEpochs() {return fEpochs;}; + Int_t GetMinTrain() {return fMinTrain;}; + Int_t GetTrainMomBin() {return fTrainMomBin;}; + + Bool_t PostProcess(); + + void SetEpochs(Int_t epochs) {fEpochs = epochs;}; + void SetMinTrain(Int_t mintrain) {fMinTrain = mintrain;}; + void SetTrainMomBin(Int_t trainmombin) {fTrainMomBin = trainmombin;}; + void SetDate(Int_t date) {fDate = date;}; + void SetDoTraining(Bool_t train) {fDoTraining = train;}; + void SetContinueTraining(Bool_t continTrain) {fContinueTraining = continTrain;}; + void SetTrainPath(Int_t path) {fTrainPath = path;}; + void LoadFiles(const Char_t *InFileNN, const Char_t *InFileLQ); + + void Terminate(Option_t *); + + void MakeTrainingLists(); // build the training and the test list + void MonitorTraining(Int_t mombin); // monitor training process + void LoadContainer(const Char_t *InFileCont); + //void CreateGraphs(); + +private: + AliTRDpidRefMaker(const AliTRDpidRefMaker&); // not implemented + AliTRDpidRefMaker& operator=(const AliTRDpidRefMaker&); // not implemented + + void GetV0info(AliTRDtrackV1 *TRDtrack, Float_t *v0pdg); // get the v0 information + void TrainNetworks(Int_t mombin); // train the neural networks for a given momentum bin + void BuildLQRefs(Int_t mombin); // build the 2dim histos for a given momentum bin + + AliTRDReconstructor *fReconstructor; //! reconstructor needed for recalculation the PID + TTree *fNN; // NN data + TTree *fLQ; // LQ data + TEventList *fTrain[AliTRDCalPID::kNMom][AliTRDgeometry::kNlayer]; // Training list for each momentum + TEventList *fTest[AliTRDCalPID::kNMom][AliTRDgeometry::kNlayer]; // Test list for each momentum + TMultiLayerPerceptron *fNet[AliTRDgeometry::kNlayer]; // artificial neural network + + Int_t fLayer; // TRD layer index + Int_t fTrainMomBin; // momentum bin for the training + Int_t fEpochs; // Number of epochs for the training of the NNs + Int_t fMinTrain; // minimum of events needed for training + Int_t fDate; // date stamp for training of the NNs + Float_t fMom; // momentum + Float_t *fdEdx[10]; // dEdx array + Float_t fv0pid[AliPID::kSPECIES]; // pid from v0s + Bool_t fDoTraining; // checks if training will be done + Bool_t fContinueTraining; // checks if training from an older run should be continued + Int_t fTrainPath; // sets the path for continuing the training + + ClassDef(AliTRDpidRefMaker, 1); // TRD reference maker for NN +}; + +#endif -- 2.39.3