PndPidDiscAssociatorTask Class Reference

#include <PndPidDiscAssociatorTask.h>

Inheritance diagram for PndPidDiscAssociatorTask:

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

Protected Member Functions
void	DoPidMatch (PndPidCandidate *pidcand, PndPidProbability *prob)
Double_t	GetPdf (Double_t thetaC, Double_t mom, Double_t mass, Double_t sigma)

Protected Attributes
TClonesArray *	fPidChargedCand
TClonesArray *	fPidChargedProb
	PndPidCandidate TCA for charged particles. More...
TString	fTrackBranchNamePidHypo
	PndPidProbability TCA for charged particles. More...

Detailed Description

Definition at line 20 of file PndPidDiscAssociatorTask.h.

Constructor & Destructor Documentation

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

Definition at line 25 of file PndPidDiscAssociatorTask.cxx.

References fPidChargedProb, fTrackBranchNamePidHypo, and SetTitle().

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

PndPidDiscAssociatorTask::PndPidDiscAssociatorTask

(

)

Definition at line 18 of file PndPidDiscAssociatorTask.cxx.

References fPidChargedProb, and fTrackBranchNamePidHypo.

                                                   {
  //---
  fPidChargedProb = new TClonesArray("PndPidProbability");
  fTrackBranchNamePidHypo="";
}

PndPidDiscAssociatorTask::~PndPidDiscAssociatorTask ( )

virtual

Definition at line 11 of file PndPidDiscAssociatorTask.cxx.

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

Member Function Documentation

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

protected

Definition at line 75 of file PndPidDiscAssociatorTask.cxx.

References Double_t, PndPidCandidate::GetDiscThetaC(), PndPidCandidate::GetMomentum(), GetPdf(), PndPidProbability::SetElectronPdf(), PndPidProbability::SetKaonPdf(), PndPidProbability::SetMuonPdf(), PndPidProbability::SetPionPdf(), PndPidProbability::SetProtonPdf(), and sigma.

Referenced by Exec().

{

  // electron
  {
    Double_t mass   = 0.0005;
    Double_t sigma  = 0.006;
    prob->SetElectronPdf(GetPdf(pidcand->GetDiscThetaC(),pidcand->GetMomentum().Mag(),mass, sigma));
  }

  // muon
  {
    Double_t mass   = 0.106;
    Double_t sigma  = 0.006;
    prob->SetMuonPdf(GetPdf(pidcand->GetDiscThetaC(),pidcand->GetMomentum().Mag(),mass, sigma));
  }
  
  // pion
  {
    Double_t mass   = 0.140;
    Double_t sigma  = 0.006;
    prob->SetPionPdf(GetPdf(pidcand->GetDiscThetaC(),pidcand->GetMomentum().Mag(),mass, sigma));
  }
  
  // kaon
  {
    Double_t mass   = 0.494;
    Double_t sigma  = 0.005;
    prob->SetKaonPdf(GetPdf(pidcand->GetDiscThetaC(),pidcand->GetMomentum().Mag(),mass, sigma));
  }
  
  // proton
  {
    Double_t mass   = 0.938;
    Double_t sigma  = 0.005;
    prob->SetProtonPdf(GetPdf(pidcand->GetDiscThetaC(),pidcand->GetMomentum().Mag(),mass, sigma));
  }
}  

void PndPidDiscAssociatorTask::Exec ( Option_t *  option )

virtual

Definition at line 58 of file PndPidDiscAssociatorTask.cxx.

References DoPidMatch(), fPidChargedCand, fPidChargedProb, fVerbose, PndPidCandidate::GetDiscIndex(), i, and PndPidProbability::SetIndex().

                                              {
  if (fPidChargedProb->GetEntriesFast() != 0) fPidChargedProb->Clear();
  if(fVerbose>1) std::cout << "-I- Start PndPidDiscAssociatorTask. "<<std::endl;

  // Get the Candidates
  for(Int_t i=0; i<fPidChargedCand->GetEntriesFast(); i++)
    {
      PndPidCandidate* pidcand = (PndPidCandidate*)fPidChargedCand->At(i);
      TClonesArray& pidRef = *fPidChargedProb;
      PndPidProbability* prob = new(pidRef[i]) PndPidProbability();// initializes with zeros
      prob->SetIndex(i);
      if (pidcand->GetDiscIndex()==-1) continue;
      DoPidMatch(pidcand,prob);
    }
 
}

void PndPidDiscAssociatorTask::Finish ( )

virtual

Definition at line 144 of file PndPidDiscAssociatorTask.cxx.

                                      {
}

Double_t PndPidDiscAssociatorTask::GetPdf ( Double_t  thetaC, Double_t  mom, Double_t  mass, Double_t  sigma  )

protected

Definition at line 114 of file PndPidDiscAssociatorTask.cxx.

References CAMath::Cos(), Double_t, CAMath::Sqrt(), and val.

Referenced by DoPidMatch().

{
  
  Double_t beta = mom / TMath::Sqrt(mom*mom + mass*mass);
  if ( (beta>0.) && ((1./1.47/beta)<1.) )
    {
      Double_t center = 1./1.47/beta;
      TF1 *gausPdf = new TF1("gausPdf","gausn",0,1);
      gausPdf->SetParameter(0,1);
      gausPdf->SetParameter(1,center);
      gausPdf->SetParameter(2,sigma);
      Double_t val = gausPdf->Eval(TMath::Cos(thetaC));
      delete gausPdf;
      return val;
    }
  else
    {
      // FIXME: Don't write zeros to that pdf!
      return 0.;
    }
}

InitStatus PndPidDiscAssociatorTask::Init ( )

virtual

Definition at line 34 of file PndPidDiscAssociatorTask.cxx.

References fPidChargedCand, fTrackBranchNamePidHypo, and Register().

                                          {
  
  std::cout << "InitStatus PndPidDiscAssociatorTask::Init()" << std::endl;
  
  FairRootManager *fManager =FairRootManager::Instance();       

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

void PndPidDiscAssociatorTask::Register

(

)

Definition at line 137 of file PndPidDiscAssociatorTask.cxx.

References fPidChargedProb, and fTrackBranchNamePidHypo.

Referenced by Init().

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

void PndPidDiscAssociatorTask::Reset

(

)

Definition at line 148 of file PndPidDiscAssociatorTask.cxx.

                                     {
  //---
}

void PndPidDiscAssociatorTask::SetOutputBranch ( TString  branch )

inline

Definition at line 51 of file PndPidDiscAssociatorTask.h.

Referenced by PndMasterMultiPidTask::PndMasterMultiPidTask().

{ fTrackBranchNamePidHypo = branch; };

void PndPidDiscAssociatorTask::SetParContainers ( )

virtual

Get parameter containers

Definition at line 54 of file PndPidDiscAssociatorTask.cxx.

                                                {
  //--
}

void PndPidDiscAssociatorTask::SetVerbose ( Bool_t  verb )

inline

Definition at line 45 of file PndPidDiscAssociatorTask.h.

References fVerbose.

{ fVerbose = verb  ;};

Member Data Documentation

TClonesArray* PndPidDiscAssociatorTask::fPidChargedCand

protected

Definition at line 24 of file PndPidDiscAssociatorTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndPidDiscAssociatorTask::fPidChargedProb

protected

PndPidCandidate TCA for charged particles.

Definition at line 25 of file PndPidDiscAssociatorTask.h.

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

TString PndPidDiscAssociatorTask::fTrackBranchNamePidHypo

protected

PndPidProbability TCA for charged particles.

Definition at line 27 of file PndPidDiscAssociatorTask.h.

Referenced by Init(), PndPidDiscAssociatorTask(), and Register().

---

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

• PndPidDiscAssociatorTask.h
• PndPidDiscAssociatorTask.cxx