PndPidTestTask Class Reference

#include <PndPidTestTask.h>

Inheritance diagram for PndPidTestTask:

Public Member Functions
	PndPidTestTask ()
	~PndPidTestTask ()
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)
virtual void	Finish ()
void	SetClassifier (TString val)
void	SetSelector (TString val)
void	PrintConfusionMatrix (bool relative=false)
std::vector< std::vector< int > >	GetConfusionMatrix () const

Private Member Functions
virtual void	SetParContainers ()
	ClassDef (PndPidTestTask, 1)

Private Attributes
int	fEvtCount
std::vector< std::vector< int > >	fConfusionMatrix
std::map< int, int >	fPdgIndex
std::vector< std::string >	fParticleNames
PndAnalysis *	fAnalysis
TString	fClassifier
TString	fSelector
std::vector< RhoCandList * >	fCandLists

Detailed Description

Definition at line 21 of file PndPidTestTask.h.

Constructor & Destructor Documentation

PndPidTestTask::PndPidTestTask

(

)

Definition at line 49 of file PndPidTestTask.cxx.

References fCandLists, fConfusionMatrix, fParticleNames, fPdgIndex, i, and vec.

                               :
  FairTask("Panda Pid Test Task") {

    fPdgIndex[11] = 0;
    fPdgIndex[13] = 1;
    fPdgIndex[211] = 2;
    fPdgIndex[321] = 3;
    fPdgIndex[2212] = 4;
    fPdgIndex[-11] = 5;
    fPdgIndex[-13] = 6;
    fPdgIndex[-211] = 7;
    fPdgIndex[-321] = 8;
    fPdgIndex[-2212] = 9;

    fParticleNames.push_back("Electron");
    fParticleNames.push_back("Muon");
    fParticleNames.push_back("Pion");
    fParticleNames.push_back("Kaon");
    fParticleNames.push_back("Proton");


    std::cout << "fPdgIndex.size()1 = " << fPdgIndex.size() << std::endl;

    for (int i = 0; i < fPdgIndex.size(); i++){
        std::vector<int> vec(fPdgIndex.size(), 0);
        fConfusionMatrix.push_back(vec);
    }

    // *** RhoCandLists for the analysis
    for (int i = 0; i < fPdgIndex.size(); i++){
        fCandLists.push_back(new RhoCandList());
    }
}

PndPidTestTask::~PndPidTestTask

(

)

Definition at line 86 of file PndPidTestTask.cxx.

{ }

Member Function Documentation

PndPidTestTask::ClassDef ( PndPidTestTask  , 1    )

private

void PndPidTestTask::Exec ( Option_t *  opt )

virtual

Definition at line 117 of file PndPidTestTask.cxx.

References fAnalysis, fCandLists, fClassifier, fConfusionMatrix, fEvtCount, PndAnalysis::FillList(), fParticleNames, fPdgIndex, fSelector, fVerbose, PndAnalysis::GetEventInTask(), RhoCandidate::GetMcTruth(), RhoCandidate::PdgCode(), and RhoCandidate::PrintOn().

{
        // *** some variables
        int j=0;
        
        // necessary to read the next event
        fAnalysis->GetEventInTask();
        
        if (!(++fEvtCount%100)) cout << "evt "<<fEvtCount<<endl;
        

        

        fAnalysis->FillList(*fCandLists[0], fParticleNames[0] + fSelector + "Plus", fClassifier);
        fAnalysis->FillList(*fCandLists[1], fParticleNames[1] + fSelector + "Plus", fClassifier);
        fAnalysis->FillList(*fCandLists[2], fParticleNames[2] + fSelector + "Plus", fClassifier);
        fAnalysis->FillList(*fCandLists[3], fParticleNames[3] + fSelector + "Plus", fClassifier);
    fAnalysis->FillList(*fCandLists[4], fParticleNames[4] + fSelector + "Plus", fClassifier);
    fAnalysis->FillList(*fCandLists[5], fParticleNames[0] + fSelector + "Minus", fClassifier);
    fAnalysis->FillList(*fCandLists[6], fParticleNames[1] + fSelector + "Minus", fClassifier);
    fAnalysis->FillList(*fCandLists[7], fParticleNames[2] + fSelector + "Minus", fClassifier);
    fAnalysis->FillList(*fCandLists[8], fParticleNames[3] + fSelector + "Minus", fClassifier);
    fAnalysis->FillList(*fCandLists[9], fParticleNames[4] + fSelector + "Minus", fClassifier);

    int candListIndex = 0;
    for (auto candidates : fCandLists){
        if (fVerbose > 1) std::cout << "CandList " << candListIndex++ << " Entries: " << candidates->GetLength() << std::endl;
        for (int candInd = 0; candInd < candidates->GetLength(); candInd++){
            RhoCandidate* cand = (*candidates)[candInd];
            if (fVerbose > 1) cand->PrintOn(std::cout);
            if (fVerbose > 1) std::cout << std::endl;
            RhoCandidate* mccand = cand->GetMcTruth();
            if (fVerbose > 1) std::cout << candInd << " Cand: " << cand->PdgCode() << " MCCand " << mccand->PdgCode() << std::endl;
            if (fVerbose > 1) std::cout << *(FairMultiLinkedData_Interface*)mccand << std::endl;
            if (fPdgIndex.count(cand->PdgCode()) > 0 && fPdgIndex.count(mccand->PdgCode()) > 0){
                fConfusionMatrix[fPdgIndex[mccand->PdgCode()]][fPdgIndex[cand->PdgCode()]]++;
            } else {
                std::cout << "-W- PndPidTestTask::Exec Wrong PDG Codes: Cand: " << cand->PdgCode() << " MCCand: " << mccand->PdgCode() << std::endl;
            }
        }
    }
        
}

void PndPidTestTask::Finish ( )

virtual

Definition at line 181 of file PndPidTestTask.cxx.

References PrintConfusionMatrix().

{
    PrintConfusionMatrix(false);
    PrintConfusionMatrix(true);

}

std::vector<std::vector <int> > PndPidTestTask::GetConfusionMatrix ( ) const

inline

Definition at line 50 of file PndPidTestTask.h.

References fConfusionMatrix.

{return fConfusionMatrix;}

InitStatus PndPidTestTask::Init ( )

virtual

Definition at line 93 of file PndPidTestTask.cxx.

References fAnalysis, and fEvtCount.

{               
        // initialize analysis object
        fAnalysis = new PndAnalysis();
        
        // reset the event counter
        fEvtCount = 0;
        
        return kSUCCESS;
}

void PndPidTestTask::PrintConfusionMatrix

(

bool

relative = false

)

Definition at line 161 of file PndPidTestTask.cxx.

References fConfusionMatrix, fParticleNames, fPdgIndex, and i.

Referenced by Finish().

                                                      {
    for (int i = 0; i < fPdgIndex.size(); i++){
        std::cout << i << " : " << std::setw(15) << fParticleNames[i%fParticleNames.size()] + ((i < fParticleNames.size()) ? "Plus" : "Minus") << " : ";
        for (int j = 0; j < fPdgIndex.size(); j++){
            if (relative == false)
                std::cout << std::setw(5) << fConfusionMatrix[i][j] << " ";
            else {
                std::cout << std::setw(5);
                int entryCount = std::accumulate(fConfusionMatrix[i].begin(), fConfusionMatrix[i].end(), 0);
                if (entryCount > 0)
                    std::cout << std::fixed << std::setprecision(2) << fConfusionMatrix[i][j] / (double)entryCount << " ";
                else
                    std::cout << std::fixed << std::setprecision(2) << 0.0 << " ";
            }
        }
        std::cout << std::endl;
    }
}

void PndPidTestTask::SetClassifier ( TString  val )

inline

Definition at line 40 of file PndPidTestTask.h.

References fClassifier, and val.

Referenced by pidTest_complete().

                                       {
            fClassifier = val;
        }

void PndPidTestTask::SetParContainers ( )

privatevirtual

Definition at line 106 of file PndPidTestTask.cxx.

References run.

{
  // Get run and runtime database
  FairRun* run = FairRun::Instance();
  if ( ! run ) Fatal("SetParContainers", "No analysis run");
}

void PndPidTestTask::SetSelector ( TString  val )

inline

Definition at line 44 of file PndPidTestTask.h.

References fSelector, and val.

Referenced by pidTest_complete().

                                     {
            fSelector = val;
        }

Member Data Documentation

PndAnalysis* PndPidTestTask::fAnalysis

private

Definition at line 66 of file PndPidTestTask.h.

Referenced by Exec(), and Init().

std::vector<RhoCandList*> PndPidTestTask::fCandLists

private

Definition at line 74 of file PndPidTestTask.h.

Referenced by Exec(), and PndPidTestTask().

TString PndPidTestTask::fClassifier

private

Definition at line 72 of file PndPidTestTask.h.

Referenced by Exec(), and SetClassifier().

std::vector<std::vector <int> > PndPidTestTask::fConfusionMatrix

private

Definition at line 59 of file PndPidTestTask.h.

Referenced by Exec(), GetConfusionMatrix(), PndPidTestTask(), and PrintConfusionMatrix().

int PndPidTestTask::fEvtCount

private

Definition at line 57 of file PndPidTestTask.h.

Referenced by Exec(), and Init().

std::vector<std::string> PndPidTestTask::fParticleNames

private

Definition at line 61 of file PndPidTestTask.h.

Referenced by Exec(), PndPidTestTask(), and PrintConfusionMatrix().

std::map<int, int> PndPidTestTask::fPdgIndex

private

Definition at line 60 of file PndPidTestTask.h.

Referenced by Exec(), PndPidTestTask(), and PrintConfusionMatrix().

TString PndPidTestTask::fSelector

private

Definition at line 73 of file PndPidTestTask.h.

Referenced by Exec(), and SetSelector().

---

The documentation for this class was generated from the following files:

• PndPidTestTask.h
• PndPidTestTask.cxx