PndMvdPidIdealTask Class Reference

#include <PndMvdPidIdealTask.h>

Inheritance diagram for PndMvdPidIdealTask:

Public Member Functions
	PndMvdPidIdealTask (const TString algoName="ideal")
	PndMvdPidIdealTask (PndMvdPidIdealTask &)=delete
PndMvdPidIdealTask &	operator= (PndMvdPidIdealTask &)=delete
virtual void	SetParContainers ()
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)

Private Member Functions
void	Register ()
void	Reset ()
void	ProduceHits ()
	ClassDef (PndMvdPidIdealTask, 1)

Private Attributes
TString	fBranchName
TString	fAlgoName
TClonesArray *	fPointArray
TClonesArray *	fMcArray
TClonesArray *	fTrackOutputArray

Detailed Description

Definition at line 35 of file PndMvdPidIdealTask.h.

Constructor & Destructor Documentation

PndMvdPidIdealTask::PndMvdPidIdealTask

(

const TString

algoName = "ideal"

)

Default constructor

Definition at line 37 of file PndMvdPidIdealTask.cxx.

                                                       :  FairTask("Digitization task for PANDA PndMvd"),
    fBranchName("MVDPoint"),
    fAlgoName(algoName),
    fPointArray(NULL),
    fMcArray(NULL),
    fTrackOutputArray(NULL)
{
}

PndMvdPidIdealTask::PndMvdPidIdealTask ( PndMvdPidIdealTask &  )

delete

Member Function Documentation

PndMvdPidIdealTask::ClassDef ( PndMvdPidIdealTask  , 1    )

private

void PndMvdPidIdealTask::Exec ( Option_t *  opt )

virtual

Virtual method Exec

Definition at line 89 of file PndMvdPidIdealTask.cxx.

References PndMvdIdealPidAlgo::CalcLikelihood(), PndMvdSimplePidAlgo::CalcLikelihood(), PndMvdAdvancedPidAlgo::CalcLikelihood(), dE, dx, fAlgoName, fPointArray, fTrackOutputArray, PndSdsMCPoint::GetPosition(), PndSdsMCPoint::GetPositionOut(), i, and track.

{
  if ( ! fTrackOutputArray ) Fatal("Exec", "No fTrackOutputArray");
  fTrackOutputArray->Delete();

  // the dE/dx information of PndmvdPidCand is obtained by PndMvdMCPoints
  std::map<int, PndMvdPidCand*> pidcand;

  for(int i=0;i<fPointArray->GetEntriesFast();i++) {
    PndSdsMCPoint* mvdpoint=(PndSdsMCPoint*) fPointArray->At(i);
    int track = mvdpoint->GetTrackID();
    if(!pidcand[track])
      pidcand[track]=new PndMvdPidCand();
    double dx = ( mvdpoint->GetPositionOut() - mvdpoint->GetPosition() ).Mag();
    double dE = mvdpoint->GetEnergyLoss();
    TVector3 momentum;
    mvdpoint->Momentum(momentum);
    if(dx>0)
      pidcand[track]->AddMvdHit(dE, dx, momentum.Mag());
  }

  // invoke likelohood calculation, TClonesArray output
  for(std::map<int, PndMvdPidCand*>::iterator it=pidcand.begin();it!=pidcand.end();it++) {
    int track = it->first;
    if (fAlgoName=="ideal")
      PndMvdIdealPidAlgo::CalcLikelihood(pidcand[track]);
    else if (fAlgoName=="simple")
      PndMvdSimplePidAlgo::CalcLikelihood(pidcand[track]);
    else if (fAlgoName=="advanced")
      PndMvdAdvancedPidAlgo::CalcLikelihood(pidcand[track]);

    int size = fTrackOutputArray->GetEntriesFast();
    new ((*fTrackOutputArray)[size]) PndMvdPidCand(*(pidcand[track]));
  }
}

InitStatus PndMvdPidIdealTask::Init ( )

virtual

Definition at line 47 of file PndMvdPidIdealTask.cxx.

References fAlgoName, fBranchName, fMcArray, fPointArray, and fTrackOutputArray.

{
  // Get RootManager
  FairRootManager* ioman = FairRootManager::Instance();
  if ( ! ioman ) {
    std::cout << "-E- PndMvdPidIdealTask::Init: "<< "RootManager not instantiated!" << std::endl;
    return kFATAL;  }

  // Get input array
  fPointArray = (TClonesArray*) ioman->GetObject(fBranchName);
  if ( ! fPointArray ) {
    std::cout << "-W- PndMvdPidIdealTask::Init: "<< "No "<<fBranchName<<" array!" << std::endl;
    return kERROR;  }

  // Get MCTruth collection
  fMcArray=(TClonesArray*) ioman->GetObject("MCTrack");
  if(fMcArray==0)  {
    std::cout << "-W- PndMvdPidIdealTask::Init: "<< "No McTruth array!" << std::endl;
    return kERROR;  }

  // Create and register output array
  fTrackOutputArray = new TClonesArray("PndMvdPidCand");
  ioman->Register("PndMvdPidCand", "PndMvd ideal tracklets", fTrackOutputArray, kTRUE);

  if (fAlgoName!="ideal"&&fAlgoName!="simple"&&fAlgoName!="advanced") {
    std::cout << "-W- PndMvdPidIdealTask::Init: "<< "No pid algorithm named '"<< fAlgoName <<"'! Names are 'ideal', 'simple', 'advanced'" << std::endl;
    return kERROR;
  }
  return kSUCCESS;
}

PndMvdPidIdealTask& PndMvdPidIdealTask::operator= ( PndMvdPidIdealTask &  )

delete

void PndMvdPidIdealTask::ProduceHits ( )

private

void PndMvdPidIdealTask::Register ( )

private

void PndMvdPidIdealTask::Reset ( )

private

void PndMvdPidIdealTask::SetParContainers ( )

virtual

Virtual method Init

Definition at line 78 of file PndMvdPidIdealTask.cxx.

{
  // Get Base Container
//  FairRun* ana = FairRun::Instance();
//  FairRuntimeDb* rtdb=ana->GetRuntimeDb();

}

Member Data Documentation

TString PndMvdPidIdealTask::fAlgoName

private

Definition at line 56 of file PndMvdPidIdealTask.h.

Referenced by Exec(), and Init().

TString PndMvdPidIdealTask::fBranchName

private

Definition at line 55 of file PndMvdPidIdealTask.h.

Referenced by Init().

TClonesArray* PndMvdPidIdealTask::fMcArray

private

Definition at line 60 of file PndMvdPidIdealTask.h.

Referenced by Init().

TClonesArray* PndMvdPidIdealTask::fPointArray

private

Input array of Points

Definition at line 59 of file PndMvdPidIdealTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndMvdPidIdealTask::fTrackOutputArray

private

Output array of Hits

Definition at line 62 of file PndMvdPidIdealTask.h.

Referenced by Exec(), and Init().

---

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

• PndMvdPidIdealTask.h
• PndMvdPidIdealTask.cxx