PndPidMlAssociatorTask Class Reference

#include <PndPidMlAssociatorTask.h>

Inheritance diagram for PndPidMlAssociatorTask:

Public Member Functions
virtual void	Exec (Option_t *option)
	Tree Index. More...
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)
virtual void	SetParContainers ()
virtual void	Finish ()

Protected Attributes
TClonesArray *	fPidChargedCand
TClonesArray *	fPidAlgoMl = new TClonesArray("PndPidProbability")
	PndPidCandidate TCA for charged particles. More...
TClonesArray *	fPidChargedProb
	PndPidProbability TCA for charged particles. More...
TTree *	inTree
	PndPidProbability TCA for charged particles. More...
Int_t	i
	Input Tree for Ml outputs. More...

Detailed Description

Definition at line 19 of file PndPidMlAssociatorTask.h.

Constructor & Destructor Documentation

PndPidMlAssociatorTask::PndPidMlAssociatorTask	(	const char *	name,
		const char *	title = "Pnd Task"
	)

Definition at line 35 of file PndPidMlAssociatorTask.cxx.

References fPidChargedProb, and SetTitle().

                                                                                 :FairTask(name) 
{
  //---
  fPidChargedProb = new TClonesArray("PndPidProbability");
  SetTitle(title);
}

PndPidMlAssociatorTask::PndPidMlAssociatorTask

(

TFile *

inFile = nullptr

)

Definition at line 19 of file PndPidMlAssociatorTask.cxx.

References exit(), fPidAlgoMl, fPidChargedProb, i, and inTree.

                                                            {
  //---
  if (inFile->IsZombie() || inFile==nullptr){
          std::cout<<" Error opening Probability file. "<<std::endl;
          exit(-1);
  }
  else{
          fPidChargedProb = new TClonesArray("PndPidProbability");
          inTree = (TTree*)inFile->Get("cbmsim");
      inTree->GetBranch("PidAlgoMl")->SetAutoDelete(kFALSE);
      inTree->SetBranchAddress("PidAlgoMl",&fPidAlgoMl);
      i = 0;
  }
}

PndPidMlAssociatorTask::~PndPidMlAssociatorTask ( )

virtual

Definition at line 12 of file PndPidMlAssociatorTask.cxx.

                                                {
  //
  FairRootManager *fManager =FairRootManager::Instance();
  fManager->Write();
}

Member Function Documentation

void PndPidMlAssociatorTask::Exec ( Option_t *  option )

virtual

Tree Index.

Definition at line 69 of file PndPidMlAssociatorTask.cxx.

References fPidAlgoMl, fPidChargedCand, fPidChargedProb, fVerbose, PndPidProbability::GetElectronPdf(), PndPidProbability::GetKaonPdf(), PndPidProbability::GetMuonPdf(), PndPidProbability::GetPionPdf(), PndPidProbability::GetProtonPdf(), i, inTree, PndPidProbability::SetElectronPdf(), PndPidProbability::SetIndex(), PndPidProbability::SetKaonPdf(), PndPidProbability::SetMuonPdf(), PndPidProbability::SetPionPdf(), and PndPidProbability::SetProtonPdf().

                                            {
          if (fPidChargedProb->GetEntriesFast() != 0) fPidChargedProb->Clear();
  if(fVerbose>1) std::cout << "-I- Start PndPidMlAssociatorTask. "<<std::endl;
  
  inTree->GetEntry(i);
  i++;
      
  // Get the Candidates
  for(Int_t j=0; j<fPidChargedCand->GetEntriesFast(); j++)
    {
                PndPidProbability* procand = (PndPidProbability*)fPidAlgoMl->At(j);
        TClonesArray& pidRef = *fPidChargedProb;
        PndPidProbability* prob = new(pidRef[j]) PndPidProbability();// initializes with zeros
        prob->SetIndex(j);
        prob->SetElectronPdf(procand->GetElectronPdf());
        prob->SetMuonPdf(procand->GetMuonPdf());
        prob->SetPionPdf(procand->GetPionPdf());
        prob->SetKaonPdf(procand->GetKaonPdf());
        prob->SetProtonPdf(procand->GetProtonPdf());
    }
}

void PndPidMlAssociatorTask::Finish ( )

virtual

Definition at line 99 of file PndPidMlAssociatorTask.cxx.

                                    {
}

InitStatus PndPidMlAssociatorTask::Init ( )

virtual

Definition at line 43 of file PndPidMlAssociatorTask.cxx.

References fPidChargedCand, and Register().

                                        {
  
  std::cout << "InitStatus PndPidMlAssociatorTask::Init()" << std::endl;
  

  
  FairRootManager *fManager =FairRootManager::Instance();       

  fPidChargedCand = (TClonesArray *)fManager->GetObject("PidChargedCand");
  if ( ! fPidChargedCand) {
    std::cout << "-I- PndPidMlAssociatorTask::Init: No PndPidCandidate array PidChargedCand there!" << std::endl;
    return kERROR;
  }
     
  Register();
  
  std::cout << "-I- PndPidMlAssociatorTask::Init: Success!" << std::endl;
  
  return kSUCCESS;
}

void PndPidMlAssociatorTask::Register

(

)

Definition at line 92 of file PndPidMlAssociatorTask.cxx.

References fPidChargedProb.

Referenced by Init().

                                      {
  //---
  FairRootManager::Instance()->
    Register("PidAlgoMl","Pid", fPidChargedProb, kTRUE); 
}

void PndPidMlAssociatorTask::Reset

(

)

Definition at line 103 of file PndPidMlAssociatorTask.cxx.

                                   {
  //---
}

void PndPidMlAssociatorTask::SetParContainers ( )

virtual

Get parameter containers

Definition at line 65 of file PndPidMlAssociatorTask.cxx.

                                              {
  //--
}

void PndPidMlAssociatorTask::SetVerbose ( Bool_t  verb )

inline

Definition at line 42 of file PndPidMlAssociatorTask.h.

References fVerbose.

{ fVerbose = verb;};

Member Data Documentation

TClonesArray* PndPidMlAssociatorTask::fPidAlgoMl = new TClonesArray("PndPidProbability")

protected

PndPidCandidate TCA for charged particles.

Definition at line 24 of file PndPidMlAssociatorTask.h.

Referenced by Exec(), and PndPidMlAssociatorTask().

TClonesArray* PndPidMlAssociatorTask::fPidChargedCand

protected

Definition at line 23 of file PndPidMlAssociatorTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndPidMlAssociatorTask::fPidChargedProb

protected

PndPidProbability TCA for charged particles.

Definition at line 25 of file PndPidMlAssociatorTask.h.

Referenced by Exec(), PndPidMlAssociatorTask(), and Register().

Int_t PndPidMlAssociatorTask::i

protected

Input Tree for Ml outputs.

Definition at line 28 of file PndPidMlAssociatorTask.h.

Referenced by Exec(), and PndPidMlAssociatorTask().

TTree* PndPidMlAssociatorTask::inTree

protected

PndPidProbability TCA for charged particles.

Definition at line 27 of file PndPidMlAssociatorTask.h.

Referenced by Exec(), and PndPidMlAssociatorTask().

---

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

• PndPidMlAssociatorTask.h
• PndPidMlAssociatorTask.cxx