PndPidMlAssociatorTask.h

Go to the documentation of this file.

#ifndef PID_MLASSOCIATOR_H
#define PID_MLASSOCIATOR_H

// **************************************************************************
//  Author: Waleed Esmail e-mail: w.esmail@fz-juelich.de
//   
//  Output Probability Distributions from Ml Algorithms
//
// Created: 17-03-18
//
// **************************************************************************

#include "TClonesArray.h"
#include "FairTask.h"

class PndPidCandidate;
class PndPidProbability;

class PndPidMlAssociatorTask : public FairTask {
  
 protected:
  
  TClonesArray* fPidChargedCand;                                        
  TClonesArray *fPidAlgoMl = new TClonesArray("PndPidProbability");     
  TClonesArray* fPidChargedProb;                                        

  TTree *inTree;                                                        
  Int_t i;                                                             
  
 public:
  
  virtual void Exec(Option_t * option);
  virtual InitStatus Init();                        //
  
  void Register();
  void Reset();
  
  PndPidMlAssociatorTask(const char *name, const char *title="Pnd Task");
  PndPidMlAssociatorTask(TFile *inFile=nullptr);
  virtual ~PndPidMlAssociatorTask();
  
  void SetVerbose(Bool_t verb)            { fVerbose = verb;};
        
  virtual void SetParContainers();
  virtual void Finish();

  ClassDef(PndPidMlAssociatorTask,1)   // PndPidMlTask
  
    };

#endif