PndAnaPidCombiner Class Reference

#include <PndAnaPidCombiner.h>

Inheritance diagram for PndAnaPidCombiner:

Public Member Functions
	PndAnaPidCombiner (const char *name="PndAnaPidCombiner", TString tcanames="")
	~PndAnaPidCombiner ()
void	SetTcaNames (TString &names, TString postfix="")
void	SetDefaults ()
void	ClearNames ()
void	Init ()
void	ApplyFlat (RhoCandidate *tc)
Bool_t	Apply (RhoCandidate *tc)
Bool_t	Apply (RhoCandList &tcl)

Private Member Functions
TClonesArray *	ReadTCA (const TString &tcaname)

Private Attributes
FairRootManager *	fRootManager
std::map< TString, TClonesArray * >	fPidArrays
std::vector< TString >	fCurrentPidArrays
	pid data map, full More...
PndPidProbability *	fPidResult
	pid data current request More...
Bool_t	fInitialized

Detailed Description

Definition at line 23 of file PndAnaPidCombiner.h.

Constructor & Destructor Documentation

PndAnaPidCombiner::PndAnaPidCombiner	(	const char *	name = "PndAnaPidCombiner",
		TString	tcanames = ""
	)

Definition at line 30 of file PndAnaPidCombiner.cxx.

                                                                          :
  TNamed ( name,"Panda PID Combiner" ) ,
  fRootManager ( 0 ),
  fPidArrays(),//FIXME: What should the initializing constructor contain here?
  fPidResult ( 0 ),
  fInitialized ( kFALSE )
{
  //  std::cout<< "PndAnaPidCombiner created with tcanames: "<<tcanames.Data()<<std::endl;
  if ( tcanames=="" ) {
    SetDefaults();
  } else {
    SetTcaNames ( tcanames );
  }

  fRootManager=FairRootManager::Instance();

  fPidResult = new PndPidProbability();
}

PndAnaPidCombiner::~PndAnaPidCombiner ( )

inline

Definition at line 30 of file PndAnaPidCombiner.h.

{/*empty*/};

Member Function Documentation

Bool_t PndAnaPidCombiner::Apply

(

RhoCandidate *

tc

)

Definition at line 60 of file PndAnaPidCombiner.cxx.

References ApplyFlat(), Bool_t, fCurrentPidArrays, fPidArrays, fPidResult, PndPidProbability::GetElectronPidProb(), PndPidProbability::GetIndex(), PndPidProbability::GetKaonPidProb(), PndPidProbability::GetMuonPidProb(), PndPidProbability::GetPionPidProb(), PndPidProbability::GetProtonPidProb(), PndPidProbability::GetSumProb(), RhoCandidate::GetTrackNumber(), PndPidProbability::Reset(), PndPidProbability::SetIndex(), and RhoCandidate::SetPidInfo().

Referenced by Apply(), and PndAnalysis::FillList().

{
  // Apply the multiplied pdf's to the RhoCandidate
  // If on of the pdf's is not available, it is skipped
  Bool_t check=kTRUE;

  //TODO: Merge PID info now.
  fPidResult->Reset();
  // combine algorithms
  TClonesArray* aTca=0;
  PndPidProbability* aProb=0;
  Int_t trackIndex = tc->GetTrackNumber();
  //std::cout<<"PidCombiner: Try RhoCandidate uid:"<<tc->Uid()<<" trknr:"<<trackIndex<<std::endl;
  //std::cout<<tc<<std::endl;
  fPidResult->SetIndex(trackIndex);
  if ( fCurrentPidArrays.size()==0 || trackIndex<0 ) {
    ApplyFlat ( tc );
    return kFALSE;
  }

  for ( std::vector<TString>::iterator iter=fCurrentPidArrays.begin();
        iter!=fCurrentPidArrays.end(); iter++ ) {
    aTca=fPidArrays[*iter];
    //Info ( "Apply","try tca %s at %p",*iter.Data(),aTca );

    if ( 0==aTca ) {
      Error ( "Apply", "PID Probability array not found, skip setting pid for candidate %i.",trackIndex );
      check=kFALSE;
      continue;
    }

    if ( 0==aTca->GetEntriesFast() ) {
      Error ( "Apply", "PID Probability array '%s' at %p of size zero, skip setting pid for candidate %i.", aTca->GetName(), aTca, trackIndex );
      continue;
      check=kFALSE;
    }

    if ( trackIndex>=aTca->GetEntriesFast() ) {
      Error ( "Apply", "Index out of '%s' array (%p) bounds, skip setting pid for candidate %i.",aTca->GetName(),aTca,trackIndex );
      continue;
      check=kFALSE;
    }

    aProb= ( PndPidProbability* ) aTca->At ( trackIndex );

    if ( aProb == 0 ) {
      Error ( "Apply", "PID Probability object in array '%s' at %p not found, skip setting pid for candidate %i.",aTca->GetName(),aTca,trackIndex );
      continue;
      check=kFALSE;
    }

    if ( trackIndex!=aProb->GetIndex() ) {
      Error ( "Apply", "PID Probability object index (%i) is not the track index (%i). Is that bad?",aProb->GetIndex(),trackIndex );
      continue;
      check=kFALSE;
    }

    // catch Zeros to avoid NAN from Div/Zero
    if ( aProb->GetSumProb() == 0 ) {
      continue;
    }

    // avoid numerics with low numbers
    //aProb->NormalizeTo(1.);
    //now multiply
    *fPidResult *= *aProb;
  }
  //std::cout<<"PndAnaPidCombiner: ";  fPidResult->Print();
  //TODO renormalizing is done in the Pid object upon request ???
  //fPidResult->NormalizeTo(1.);
  //std::cout<<"PndAnaPisCombiner: scaled = ";  fPidResult->Print();
  // numbering see PndPidListMaker
  // No flux implemented! To come for each Detector!
  tc->SetPidInfo ( 0,fPidResult->GetElectronPidProb() );
  tc->SetPidInfo ( 1,fPidResult->GetMuonPidProb() );
  tc->SetPidInfo ( 2,fPidResult->GetPionPidProb() );
  tc->SetPidInfo ( 3,fPidResult->GetKaonPidProb() );
  tc->SetPidInfo ( 4,fPidResult->GetProtonPidProb() );
  //std::cout<<"PndAnaPidCombiner: "<<tc<<std::endl;

  return check;
}

Bool_t PndAnaPidCombiner::Apply

(

RhoCandList &

tcl

)

Definition at line 49 of file PndAnaPidCombiner.cxx.

References Apply(), Bool_t, and RhoCandList::GetLength().

{
  Bool_t check = kTRUE;
  Bool_t chack = kTRUE;
  for ( int j=0; j<tcl.GetLength(); j++ ) {
    chack = Apply( tcl[j] );
    check = check && chack;
  }
  return check;
}

void PndAnaPidCombiner::ApplyFlat

(

RhoCandidate *

tc

)

Definition at line 143 of file PndAnaPidCombiner.cxx.

References RhoCandidate::SetPidInfo().

Referenced by Apply().

{
  tc->SetPidInfo ( 0,0.2 );
  tc->SetPidInfo ( 1,0.2 );
  tc->SetPidInfo ( 2,0.2 );
  tc->SetPidInfo ( 3,0.2 );
  tc->SetPidInfo ( 4,0.2 );
  return;
}

void PndAnaPidCombiner::ClearNames ( )

inline

Definition at line 34 of file PndAnaPidCombiner.h.

References fPidArrays.

{fPidArrays.clear(); };

void PndAnaPidCombiner::Init

(

)

TClonesArray * PndAnaPidCombiner::ReadTCA ( const TString &  tcaname )

private

Definition at line 220 of file PndAnaPidCombiner.cxx.

References fRootManager.

Referenced by SetTcaNames().

{
  // Fetch a TCLonesArray from the framework by its root name
  if (tcaname == "") {
    Warning ( "PndAnaPidCombiner::ReadTCA()","Empty TCA name." );
    return NULL;
  }
  //std::cout<<" -I- PndAnaPidCombiner::ReadTCA(): Try fetching branch "<<tcaname.Data()<<" rootmanager="<<fRootManager<<std::endl;
  TClonesArray* tca = ( TClonesArray* ) fRootManager->GetObject ( tcaname.Data() );

  if ( ! tca ) { // this information is already provoded by the RootManager
    //Warning ( "PndAnaPidCombiner::ReadTCA()","No \"%s\" array found.",tcaname.Data() );
    return NULL;
  }

  return tca;
}

void PndAnaPidCombiner::SetDefaults

(

)

Definition at line 153 of file PndAnaPidCombiner.cxx.

References SetTcaNames(), and TString.

Referenced by PndAnalysis::FillList().

{
  // Set list of names and weights to the default PANDA
  //TString names = "PidAlgoMvd;PidAlgoStt;PidAlgoEmcBayes;PidAlgoDrc;PidAlgoDisc;PidAlgoMdtHardCuts;";
  //TString names = "PidAlgoIdealCharged";
  //TString names = "PidMvaChargedProbability";
  //SetTcaNames ( names );

  TString dummy("");
  SetTcaNames(dummy,dummy);
  return;
}

void PndAnaPidCombiner::SetTcaNames	(	TString &	names,
		TString	postfix = ""
	)

Definition at line 166 of file PndAnaPidCombiner.cxx.

References fCurrentPidArrays, fPidArrays, fRootManager, full(), ReadTCA(), and TString.

Referenced by PndAnalysis::FillList(), and SetDefaults().

{
  fCurrentPidArrays.clear();
  // Tokenizer, cool thingy!
  TStringToken list ( names,";" );
  //use TString class part (inherited, Tokenizer stores data there)
  while ( list.NextToken() )
  {
    TString branch=(TString)list;
    if (branch == "") continue;
    TString full=branch+postfix;
    if (!fPidArrays[full])
    {
      //std::cout<<" -I- PndAnaPidCombiner::SetTcaNames(): try finding \""<<full.Data()<<"\""<<std::endl;
      TClonesArray * tmpar = ReadTCA(full.Data());
      if(tmpar) {
        fPidArrays[full]=tmpar;
        fRootManager->ReadBranchEvent(full.Data());
        // we have the branch now, lets use it
        fCurrentPidArrays.push_back (full);
        //std::cout<<" -I- PndAnaPidCombiner::SetTcaNames(): \""<<branch.Data()<<"\" + \""<<postfix.Data()<<"\" = \""<<full.Data()<<"\""<<std::endl;
      } else {
        // now there is no branch with the full name, let's try without the PID postfix
        // TODO here we would need to select one other branch, if available, before going to fallback
        // Users may use Multikalman for protons and pions only, but want to reconstruct muons
        if (!fPidArrays[branch])
        {
          TClonesArray * tmpar2 = ReadTCA(branch.Data());
          if(tmpar2)
          {
            fPidArrays[branch]=tmpar;
            fRootManager->ReadBranchEvent(branch.Data());
            // we have the backup branch now, lets use it
            fCurrentPidArrays.push_back (branch);
            std::cout<<" WARNING TO ANALYST: PndAnaPidCombiner::Init() could not find "<< full.Data()<<" we use a backup branch: "<<branch.Data()<<std::endl;
          } else {
            // now we even don't have a backup
            std::cout<<" WARNING TO ANALYST: PndAnaPidCombiner::Init() could not find "<< full.Data()<<" nor a backup branch "<<branch.Data()<<std::endl;
          }
        } else {
          // we have the backup branch already loaded, lets use it
          fCurrentPidArrays.push_back (branch);
        }
      }
    } else {
      //std::cout<<" -I- PndAnaPidCombiner::SetTcaNames(): we have alreasy a cached TClonesArray named \""<<full.Data()<<"\""<<std::endl;
      // we have the branch already loaded, lets use it
      fCurrentPidArrays.push_back(full);
    }
  }

  //std::cout<<"PidCombiner initialized."<<std::endl;
}

Member Data Documentation

std::vector<TString> PndAnaPidCombiner::fCurrentPidArrays

private

pid data map, full

Definition at line 45 of file PndAnaPidCombiner.h.

Referenced by Apply(), and SetTcaNames().

Bool_t PndAnaPidCombiner::fInitialized

private

Definition at line 47 of file PndAnaPidCombiner.h.

std::map<TString,TClonesArray*> PndAnaPidCombiner::fPidArrays

private

Definition at line 44 of file PndAnaPidCombiner.h.

Referenced by Apply(), ClearNames(), and SetTcaNames().

PndPidProbability* PndAnaPidCombiner::fPidResult

private

pid data current request

Definition at line 46 of file PndAnaPidCombiner.h.

Referenced by Apply().

FairRootManager* PndAnaPidCombiner::fRootManager

private

Definition at line 43 of file PndAnaPidCombiner.h.

Referenced by ReadTCA(), and SetTcaNames().

---

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

• PndAnaPidCombiner.h
• PndAnaPidCombiner.cxx