PndPidIdealAssociatorTask Class Reference

#include <PndPidIdealAssociatorTask.h>

Inheritance diagram for PndPidIdealAssociatorTask:

Public Member Functions
virtual void	Exec (Option_t *option)
virtual InitStatus	Init ()
void	Register ()
void	Reset ()
	PndPidIdealAssociatorTask (const char *name, const char *title="Pnd Task")
	PndPidIdealAssociatorTask ()
virtual	~PndPidIdealAssociatorTask ()
void	SetToOnlyOne ()
void	SetVerbose (Bool_t verb)
virtual void	SetParContainers ()
virtual void	Finish ()
void	SetOutputBranch (TString branch)

Protected Member Functions
void	DoPidMatch (PndPidCandidate *pidcand, PndPidProbability *prob)
	Monte-Carlo Truth track TCA. More...

Protected Attributes
TString	fNeutralBranchName
TString	fChargedBranchName
TString	fTrackBranchNamePidHypo
TClonesArray *	fPidChargedCand
TClonesArray *	fPidNeutralCand
	PndPidCandidate TCA for charged particles. More...
TClonesArray *	fPidChargedProb
	PndPidCandidate TCA for neutral particles. More...
TClonesArray *	fPidNeutralProb
	PndPidProbability TCA for charged particles. More...
TClonesArray *	fMCTrack
	PndPidProbability TCA for neutral particles. More...

Detailed Description

Definition at line 21 of file PndPidIdealAssociatorTask.h.

Constructor & Destructor Documentation

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

Definition at line 33 of file PndPidIdealAssociatorTask.cxx.

References fChargedBranchName, fNeutralBranchName, fPidChargedProb, fPidNeutralProb, fTrackBranchNamePidHypo, and SetTitle().

:FairTask(name) {
  //---
  fNeutralBranchName="PidAlgoIdealNeutral";
  fChargedBranchName="PidAlgoIdealCharged";
  fTrackBranchNamePidHypo="";
  fPidChargedProb = new TClonesArray("PndPidProbability");
  fPidNeutralProb = new TClonesArray("PndPidProbability");
  SetTitle(title);
}

PndPidIdealAssociatorTask::PndPidIdealAssociatorTask

(

)

Definition at line 23 of file PndPidIdealAssociatorTask.cxx.

References fChargedBranchName, fNeutralBranchName, fPidChargedProb, fPidNeutralProb, and fTrackBranchNamePidHypo.

                                                     {
  //---
  fNeutralBranchName="PidAlgoIdealNeutral";
  fChargedBranchName="PidAlgoIdealCharged";
  fTrackBranchNamePidHypo="";
  fPidChargedProb = new TClonesArray("PndPidProbability");
  fPidNeutralProb = new TClonesArray("PndPidProbability");
}

PndPidIdealAssociatorTask::~PndPidIdealAssociatorTask ( )

virtual

Definition at line 16 of file PndPidIdealAssociatorTask.cxx.

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

Member Function Documentation

void PndPidIdealAssociatorTask::DoPidMatch ( PndPidCandidate *  pidcand, PndPidProbability *  prob  )

protected

Monte-Carlo Truth track TCA.

Definition at line 111 of file PndPidIdealAssociatorTask.cxx.

References fMCTrack, PndPidCandidate::GetMcIndex(), PndMCTrack::GetPdgCode(), mcpdg, mctrack, PndPidProbability::SetElectronPdf(), PndPidProbability::SetKaonPdf(), PndPidProbability::SetMuonPdf(), PndPidProbability::SetPionPdf(), and PndPidProbability::SetProtonPdf().

Referenced by Exec().

{
  // Cheating for each Pid Candidate.

  Int_t mcid = pidcand->GetMcIndex();
  if(mcid<0) return; // no specified MC id... do nothing
  PndMCTrack *mctrack = (PndMCTrack*)fMCTrack->At(mcid);
  if( 0==mctrack) return; // better do nothing on a null pointer
  Int_t mcpdg = mctrack->GetPdgCode();

  // TLorentzVector lv = pidcand->GetLorentzVector();
  // Double_t mass = TDatabasePDG::Instance()->GetParticle(mcpdg)->Mass();
  // lv.SetXYZM(lv.X(), lv.Y(), lv.Z(), mass);
  //   pidcand->SetLorentzVector(lv);

  // chaet around with the probabilities...
  switch(mcpdg) {
    case -11: //positron
    case 11: //electron
      prob->SetProtonPdf(0.);
      prob->SetKaonPdf(0.);
      prob->SetPionPdf(0.);
      prob->SetMuonPdf(0.);
      prob->SetElectronPdf(1.);
      break;

    case -13: //muon+
    case 13: //muon-
      prob->SetProtonPdf(0.);
      prob->SetKaonPdf(0.);
      prob->SetPionPdf(0.);
      prob->SetMuonPdf(1.);
      prob->SetElectronPdf(0.);
      break;

    case 211: //pion+
    case -211: //pion-
      prob->SetProtonPdf(0.);
      prob->SetKaonPdf(0.);
      prob->SetPionPdf(1.);
      prob->SetMuonPdf(0.);
      prob->SetElectronPdf(0.);
      break;

    case -321: //Kaon+
    case 321: //Kaon-
      prob->SetProtonPdf(0.);
      prob->SetKaonPdf(1.);
      prob->SetPionPdf(0.);
      prob->SetMuonPdf(0.);
      prob->SetElectronPdf(0.);
      break;

    case -2212: //antiproton
    case 2212: //proton
      prob->SetProtonPdf(1.);
      prob->SetKaonPdf(0.);
      prob->SetPionPdf(0.);
      prob->SetMuonPdf(0.);
      prob->SetElectronPdf(0.);
      break;

    default:
      prob->SetProtonPdf(0.2);
      prob->SetKaonPdf(0.2);
      prob->SetPionPdf(0.2);
      prob->SetMuonPdf(0.2);
      prob->SetElectronPdf(0.2);
      break;
  }

}

void PndPidIdealAssociatorTask::Exec ( Option_t *  option )

virtual

Definition at line 89 of file PndPidIdealAssociatorTask.cxx.

References DoPidMatch(), fPidChargedCand, fPidChargedProb, fPidNeutralCand, fPidNeutralProb, fVerbose, PndPidCandidate::GetLorentzVector(), and i.

                                               {
  // CAUTION We use Monte-Carlo info here!

  if(fVerbose>1) std::cout << "-I- Start PndPidIdealAssociatorTask. "<<std::endl;
  if (fPidChargedProb->GetEntriesFast() != 0) fPidChargedProb->Clear();
  if (fPidNeutralProb->GetEntriesFast() != 0) fPidNeutralProb->Clear();
  // Get the Candidates
  for(Int_t i=0; i<fPidChargedCand->GetEntriesFast(); i++){
    PndPidCandidate* pidcand = (PndPidCandidate*)fPidChargedCand->At(i);
    PndPidProbability* prob = new((*fPidChargedProb)[i]) PndPidProbability(0.2,0.2,0.2,0.2,0.2,i);// initializes with equal probability
    if(fVerbose>1) std::cout<<"-I- PndPidIdealAssociatorTask Charged BEFORE  "<< pidcand->GetLorentzVector().M()<<std::endl;;
    DoPidMatch(pidcand,prob);
    if(fVerbose>1) std::cout<<"-I- PndPidIdealAssociatorTask Charged AFTER   "<< pidcand->GetLorentzVector().M()<<std::endl;;
  }
  for(Int_t i=0; i<fPidNeutralCand->GetEntriesFast(); i++){
    //PndPidCandidate* pidcand = (PndPidCandidate*)fPidNeutralCand->At(i); //[R.K. 01/2017] unused variable
    new((*fPidNeutralProb)[i]) PndPidProbability(0.2,0.2,0.2,0.2,0.2,i);// initializes flat //PndPidProbability* prob =  //[R.K.03/2017] unused variable
    //DoPidMatch(pidcand,prob); //TODO match idealy neutral cands, esp. when we have pi0s
  }

}

void PndPidIdealAssociatorTask::Finish ( )

virtual

Definition at line 197 of file PndPidIdealAssociatorTask.cxx.

                                       {
  //---
  //  FairRootManager* ioman = FairRootManager::Instance();
  //  ioman->W
}

InitStatus PndPidIdealAssociatorTask::Init ( )

virtual

Definition at line 52 of file PndPidIdealAssociatorTask.cxx.

References fMCTrack, fPidChargedCand, fPidNeutralCand, fTrackBranchNamePidHypo, and Register().

                                           {

  //  cout << "InitStatus PndPidIdealAssociatorTask::Init()" << endl;

  FairRootManager *fManager =FairRootManager::Instance();

  // TODO: Am I allowed to write in these Arrays?
  fPidChargedCand = (TClonesArray *)fManager->GetObject("PidChargedCand"+fTrackBranchNamePidHypo);
  if ( ! fPidChargedCand) {
    std::cout << "-I- PndPidIdealAssociatorTask::Init: No PndPidCandidate array PidChargedCand there!" << std::endl;
    return kERROR;
  }

  fPidNeutralCand = (TClonesArray *)fManager->GetObject("PidNeutralCand"+fTrackBranchNamePidHypo);
  if ( ! fPidNeutralCand) {
    std::cout << "-I- PndPidIdealAssociatorTask::Init: No PndPidCandidate array PidNeutralCand there!" << std::endl;
    return kERROR;
  }

  fMCTrack = (TClonesArray*) fManager->GetObject("MCTrack");
  if ( ! fMCTrack) {
    std::cout << "-I- PndPidIdealAssociatorTask::Init: No MC Track array there!" << std::endl;
    return kERROR;
  }

  Register();

  std::cout << "-I- PndPidIdealAssociatorTask::Init: Success!" << std::endl;

  return kSUCCESS;
}

void PndPidIdealAssociatorTask::Register

(

)

Definition at line 188 of file PndPidIdealAssociatorTask.cxx.

References fChargedBranchName, fNeutralBranchName, fPidChargedProb, fPidNeutralProb, and fTrackBranchNamePidHypo.

Referenced by Init().

                                         {
  //---
  FairRootManager::Instance()->
  Register(fChargedBranchName+fTrackBranchNamePidHypo,"Pid", fPidChargedProb, kTRUE);
  FairRootManager::Instance()->
  Register(fNeutralBranchName+fTrackBranchNamePidHypo,"Pid", fPidNeutralProb, kTRUE);
}

void PndPidIdealAssociatorTask::Reset

(

)

Definition at line 203 of file PndPidIdealAssociatorTask.cxx.

                                      {
  //---
}

void PndPidIdealAssociatorTask::SetOutputBranch ( TString  branch )

inline

Definition at line 56 of file PndPidIdealAssociatorTask.h.

References fTrackBranchNamePidHypo.

Referenced by PndMasterMultiPidTask::PndMasterMultiPidTask().

{ fTrackBranchNamePidHypo = branch; };

void PndPidIdealAssociatorTask::SetParContainers ( )

virtual

Get parameter containers

Definition at line 85 of file PndPidIdealAssociatorTask.cxx.

                                                 {
  //--
}

void PndPidIdealAssociatorTask::SetToOnlyOne

(

)

Definition at line 44 of file PndPidIdealAssociatorTask.cxx.

References fChargedBranchName, and fNeutralBranchName.

{
  fNeutralBranchName="PidNeutralProbability";
  fChargedBranchName="PidChargedProbability";
}

void PndPidIdealAssociatorTask::SetVerbose ( Bool_t  verb )

inline

Definition at line 50 of file PndPidIdealAssociatorTask.h.

References fVerbose.

{ fVerbose = verb  ;};

Member Data Documentation

TString PndPidIdealAssociatorTask::fChargedBranchName

protected

Definition at line 25 of file PndPidIdealAssociatorTask.h.

Referenced by PndPidIdealAssociatorTask(), Register(), and SetToOnlyOne().

TClonesArray* PndPidIdealAssociatorTask::fMCTrack

protected

PndPidProbability TCA for neutral particles.

Definition at line 33 of file PndPidIdealAssociatorTask.h.

Referenced by DoPidMatch(), and Init().

TString PndPidIdealAssociatorTask::fNeutralBranchName

protected

Definition at line 24 of file PndPidIdealAssociatorTask.h.

Referenced by PndPidIdealAssociatorTask(), Register(), and SetToOnlyOne().

TClonesArray* PndPidIdealAssociatorTask::fPidChargedCand

protected

Definition at line 29 of file PndPidIdealAssociatorTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndPidIdealAssociatorTask::fPidChargedProb

protected

PndPidCandidate TCA for neutral particles.

Definition at line 31 of file PndPidIdealAssociatorTask.h.

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

TClonesArray* PndPidIdealAssociatorTask::fPidNeutralCand

protected

PndPidCandidate TCA for charged particles.

Definition at line 30 of file PndPidIdealAssociatorTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndPidIdealAssociatorTask::fPidNeutralProb

protected

PndPidProbability TCA for charged particles.

Definition at line 32 of file PndPidIdealAssociatorTask.h.

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

TString PndPidIdealAssociatorTask::fTrackBranchNamePidHypo

protected

Definition at line 27 of file PndPidIdealAssociatorTask.h.

Referenced by Init(), PndPidIdealAssociatorTask(), Register(), and SetOutputBranch().

---

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

• PndPidIdealAssociatorTask.h
• PndPidIdealAssociatorTask.cxx