PndPidTestTask.h

Go to the documentation of this file.

#ifndef PndPidTestTask_H
#define PndPidTestTask_H 1


#include "FairTask.h"
#include <map>
#include <string>
#include "TLorentzVector.h"

class TClonesArray;
class TObjectArray;
class TH1F;
class TH2F;

class RhoMassParticleSelector;
class PndAnalysis;
class RhoCandList;
class RhoTuple;


class PndPidTestTask : public FairTask
{

 public:
        
        // ** Default constructor   
        PndPidTestTask();
        
        // ** Destructor 
        ~PndPidTestTask();
        
        // ** Virtual method Init 
        virtual InitStatus Init();
        
        // ** Virtual method Exec 
        virtual void Exec(Option_t* opt);
        
        virtual void Finish();

        void SetClassifier(TString val){
            fClassifier = val;
        }
        
        void SetSelector(TString val){
            fSelector = val;
        }

        void PrintConfusionMatrix(bool relative = false);

        std::vector<std::vector <int> > GetConfusionMatrix() const {return fConfusionMatrix;}

 protected:
        
        
 private: 
        // *** event counter
        int fEvtCount;  
        
        std::vector<std::vector <int> > fConfusionMatrix;
        std::map<int, int> fPdgIndex;
        std::vector<std::string> fParticleNames;
        
        
        
        // *** the PndAnalysis object
        PndAnalysis *fAnalysis;
        

        // *** Get parameter containers
        virtual void SetParContainers();
        
        TString fClassifier;
        TString fSelector;
    std::vector<RhoCandList*> fCandLists;

        
        ClassDef(PndPidTestTask,1);
  
};

#endif