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"
#include "TTree.h"
#include "TFile.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