PndAnaPidSelector Class Reference

#include <PndAnaPidSelector.h>

Inheritance diagram for PndAnaPidSelector:

Public Member Functions
	PndAnaPidSelector (const char *name="PndAnaPidSelector", const char *type="", const char *paramid="Default")
virtual	~PndAnaPidSelector ()
virtual Bool_t	Accept (RhoCandidate *b)
virtual Bool_t	Accept (PndPidCandidate *b)
Bool_t	SetCriterion (TString &crit)
Bool_t	SetSelection (TString &crit)
virtual void	SetCriterion (const char *crit)
void	SetCriterion (criterion crit)
void	Select (RhoCandList &l)
void	Select (RhoCandList &in, RhoCandList &out)

Protected Member Functions
TParticlePDG *	CPConjugate (TParticlePDG *aPart)
void	SetTypeAndMass (RhoCandidate *b)

Protected Attributes
PndAnaSelectorPar *	fSelectPar
Double_t	fChargeCrit
	Pointer to the current parameter object. More...
Int_t	fPidSelect
TParticlePDG *	fTypePlus
TParticlePDG *	fTypeMinus
	Particle type to accept. More...
criterion	fCriterion
	Particle type to accept. More...

Detailed Description

Definition at line 21 of file PndAnaPidSelector.h.

Constructor & Destructor Documentation

PndAnaPidSelector::PndAnaPidSelector	(	const char *	name = "PndAnaPidSelector",
		const char *	type = "",
		const char *	paramid = "Default"
	)

Definition at line 40 of file PndAnaPidSelector.cxx.

References fSelectPar, and rtdb.

                                                                                        : // paramid //[R.K.03/2017] unused variable(s)
    RhoParticleSelectorBase ( name,type ) ,
    fChargeCrit ( 0.0 ),
    fPidSelect ( 99 )   // some silly default number here

{
  // reading parameters from RTDB
  FairRun* ana = FairRun::Instance();
  FairRuntimeDb* rtdb=ana->GetRuntimeDb();
  fSelectPar = ( PndAnaSelectorPar* ) ( rtdb->getContainer ( "ANAPidSelections" ) );

  if ( ! fSelectPar ) {
    Fatal ( "PndAnaPidSelector","No ANAPidSelections* parameter found: %s","ANAPidSelections" );
  }

  //      PndAnaFluxPar* fFluxPar = (PndAnaFluxPar*)(rtdb->getContainer("ANAPidSelectionFlux"));
  //      if ( ! fFluxPar ) Fatal("SetParContainers","No ANAPidFlux parameter found: %s",parsetname.Data());
  //    }

  // Caution: The Parameter Set is not filled from the DB IO, yet.
  Int_t runID = ana->GetRunId();

  // actually fill the containers.
  //We might want to do that at another point, because multiple
  //instnces will multiply call the RTDB init. That's bad in a loop!
  rtdb->initContainers ( runID );
}

virtual PndAnaPidSelector::~PndAnaPidSelector ( )

inlinevirtual

Definition at line 28 of file PndAnaPidSelector.h.

{}

Member Function Documentation

Bool_t PndAnaPidSelector::Accept ( RhoCandidate *  b )

virtual

Implements RhoParticleSelectorBase.

Definition at line 180 of file PndAnaPidSelector.cxx.

References all, best, fabs(), fChargeCrit, RhoParticleSelectorBase::fCriterion, fPidSelect, RhoCandidate::GetCharge(), RhoCandidate::GetPidInfo(), loose, RhoParticleSelectorBase::SetTypeAndMass(), tight, variable, veryLoose, and veryTight.

{
  // Accept or reject one candidate based on it's PID hypothesis, the pid
  // pdf values and the selection criterion.
  // This might be not performant, since each time we access an object to
  // read the criterion value. Is a field of values faster?

  if ( b == 0 ) {
    Error ( "PndAnaPidSelector::Accept","Candiate missing." ); return kFALSE;
  }
  //Info("PndAnaPidSelector::Accept","Request: Q=%g pid=%i crit=%i",fChargeCrit,fPidSelect,fCriterion);
  // too stringent on charge with +-1. ??
  if ( fChargeCrit!=0 && fChargeCrit!=b->GetCharge() ) {
    //std::cout<<"PndAnaPidSelector::Accept: charge reject. fChargeCrit="<<fChargeCrit<<" b.GetCharge()"<<b.GetCharge() <<std::endl;
    return kFALSE;
  }

  if ( fPidSelect==99 ) {
    return kTRUE;
  } // no PID requested? Fine!

  // if e, mu, pi, k or p (but no + or -) is requested we reject neutrals
  if ( fPidSelect!=99 && fabs ( b->GetCharge() ) <0.001 ) {
    return kFALSE;
  }

  SetTypeAndMass ( b );

  double Lcheck = b->GetPidInfo ( fPidSelect );
  //std::cout<<"PndAnaPidSelector::Accept: Lcheck="<<Lcheck<<" fPidSelect="<<fPidSelect<<std::endl;

  if ( fCriterion == veryLoose ) {
    if ( Lcheck<fSelectPar->GetVeryLooseCrit ( fPidSelect ) ) {
      return kFALSE;
    }
  } else if ( fCriterion == loose ) {
    if ( Lcheck<fSelectPar->GetLooseCrit ( fPidSelect ) ) {
      return kFALSE;
    }
  } else if ( fCriterion == tight ) {
    if ( Lcheck<fSelectPar->GetTightCrit ( fPidSelect ) ) {
      return kFALSE;
    }
  } else if ( fCriterion == veryTight ) {
    if ( Lcheck<fSelectPar->GetVeryTightCrit ( fPidSelect ) ) {
      return kFALSE;
    }
  } else if ( fCriterion == variable ) {
    if ( Lcheck<fSelectPar->GetVariableCrit ( fPidSelect ) ) {
      return kFALSE;
    }
  } else if ( fCriterion == all ) {
    return kTRUE;
  } else if ( fCriterion == best ) {
    double Le  = b->GetPidInfo ( 0 );
    double Lmu = b->GetPidInfo ( 1 );
    double Lpi = b->GetPidInfo ( 2 );
    double Lk  = b->GetPidInfo ( 3 );
    double Lp  = b->GetPidInfo ( 4 );

    if ( Lcheck<Le || Lcheck<Lmu || Lcheck<Lpi || Lcheck<Lk || Lcheck<Lp ) {
      return kFALSE;
    }
  }

  return kTRUE;
}

Bool_t PndAnaPidSelector::Accept ( PndPidCandidate *  b )

virtual

Implements RhoParticleSelectorBase.

Definition at line 248 of file PndAnaPidSelector.cxx.

{
  Warning ( "PndAnaPidSelector::Accept(VAbsMicroCandidate&)","No implementation for this. Please use PndAnaPidSelector::Accept(RhoCandidate&)" );
  return kFALSE;
}

TParticlePDG * RhoParticleSelectorBase::CPConjugate ( TParticlePDG *  aPart )

protectedinherited

Definition at line 129 of file RhoParticleSelectorBase.cxx.

References name, and TString.

Referenced by RhoParticleSelectorBase::RhoParticleSelectorBase(), and SetSelection().

{
  //    TDatabasePDG *pdt = TDatabasePDG::Instance();
  TDatabasePDG* pdt = TDatabasePDG::Instance();
  Int_t theCode = thePart->PdgCode();
  TString name = thePart->GetName();

  // Is it a charged particle ?
  if ( name.Index ( "-" ) >0 || name.Index ( "+" ) >0 || abs ( theCode ) ==2212 ) { theCode = -theCode; }

  return pdt->GetParticle ( theCode );
}

void RhoParticleSelectorBase::Select ( RhoCandList &  l )

inherited

Definition at line 94 of file RhoParticleSelectorBase.cxx.

References RhoCandList::Select().

{
  l.Select(this);
  //  RhoCandList tmp ( l );
  //  Select ( tmp,l );
}

void RhoParticleSelectorBase::Select ( RhoCandList &  in, RhoCandList &  out  )

inherited

Definition at line 101 of file RhoParticleSelectorBase.cxx.

References RhoParticleSelectorBase::Accept(), c, RhoCandList::Cleanup(), RhoCandList::GetLength(), i, n, and RhoCandList::Put().

{
  out.Cleanup();
  Int_t n = in.GetLength();
  for ( Int_t i=0; i<n; i++ ) {
    RhoCandidate* c = in[i];

    if ( Accept ( c ) ) {
      //printf("selector \"%s\" accepted i=%i\n",fName.Data(),i);
      out.Put ( c );
    } //else printf("selector \"%s\" rejected i=%i\n",fName.Data(),i);
  }
}

Bool_t PndAnaPidSelector::SetCriterion ( TString &  crit )

inline

Definition at line 33 of file PndAnaPidSelector.h.

References SetSelection().

Referenced by PndAnalysis::FillList().

{return SetSelection(crit);};

void RhoParticleSelectorBase::SetCriterion ( const char *  crit )

virtualinherited

Reimplemented in RhoGoodTrackSelector, RhoGoodPhotonSelector, and PndGoodTrackSelector.

Definition at line 61 of file RhoParticleSelectorBase.cxx.

References all, best, loose, tight, TString, variable, veryLoose, and veryTight.

Referenced by PndSimpleAnalysis::Exec(), PndSimpleAnalysis::FillGenericLists(), PndPidListMaker::FillList(), PndAnalysisTaskExample::Init(), PndAnalysisTask::Init(), PndHypSimpleAna::Init(), PndHypFullAna::Init(), PndHypFullIdealAna::Init(), PndGoodTrackSelector::SetCriterion(), RhoGoodPhotonSelector::SetCriterion(), and SetSelection().

{
  TString crit ( c );
  crit.ToLower();

  if ( crit=="best" ) {
    SetCriterion ( best );
  } else if ( crit=="veryloose" ) {
    SetCriterion ( veryLoose );
  } else if ( crit=="loose" ) {
    SetCriterion ( loose );
  } else if ( crit=="tight" ) {
    SetCriterion ( tight );
  } else if ( crit=="verytight" ) {
    SetCriterion ( veryTight );
  } else if ( crit=="variable" ) {
    SetCriterion ( variable );
  } else if ( crit=="all" ) {
    SetCriterion ( all );
  } else {
    cerr << GetName() << ": Unknown criterion " << crit.Data() << endl;
    return;
  }

//  fParms.Set("criteria",crit,"Selector criterion",this);
  //cout << this->ClassName() << "::SetCriterion " << c << " for " << GetName() << endl;
}

void RhoParticleSelectorBase::SetCriterion ( criterion  crit )

inherited

Definition at line 89 of file RhoParticleSelectorBase.cxx.

References RhoParticleSelectorBase::fCriterion.

{
  fCriterion = crit;
}

Bool_t PndAnaPidSelector::SetSelection

(

TString &

crit

)

Definition at line 68 of file PndAnaPidSelector.cxx.

References all, best, RhoParticleSelectorBase::CPConjugate(), fChargeCrit, fPidSelect, RhoParticleSelectorBase::fTypeMinus, RhoParticleSelectorBase::fTypePlus, loose, RhoParticleSelectorBase::SetCriterion(), tight, TString, variable, veryLoose, and veryTight.

Referenced by SetCriterion().

{
  //Default setting for criterion
  RhoParticleSelectorBase::SetCriterion ( all );
  TString critcopy(crit);

  // parse criterion string to criteria flags...

  if ( crit.Contains ( "VeryLoose" ) ) {
    RhoParticleSelectorBase::SetCriterion ( veryLoose );
    critcopy.ReplaceAll("VeryLoose","");
  } else if ( crit.Contains ( "Loose" ) ) {
    RhoParticleSelectorBase::SetCriterion ( loose );
    critcopy.ReplaceAll("Loose","");
  } else if ( crit.Contains ( "VeryTight" ) ) {
    RhoParticleSelectorBase::SetCriterion ( veryTight );
    critcopy.ReplaceAll("VeryTight","");
  } else if ( crit.Contains ( "Tight" ) ) {
    RhoParticleSelectorBase::SetCriterion ( tight );
    critcopy.ReplaceAll("Tight","");
  } else if ( crit.Contains ( "Variable" ) ) {
    RhoParticleSelectorBase::SetCriterion ( variable );
    critcopy.ReplaceAll("Variable","");
  } else if ( crit.Contains ( "Best" ) ) {
    RhoParticleSelectorBase::SetCriterion ( best );
    critcopy.ReplaceAll("Best","");
  } else if ( crit.Contains ( "All" ) ) {
    RhoParticleSelectorBase::SetCriterion ( all );
    critcopy.ReplaceAll("All","");
  } // well, that's default anyway

  //TODO: make a 2d array of the criterion numbers for faster access

  fTypePlus=0;

  fTypeMinus=0;

  // Name convention for TDatabsePDG found at $ROOTSYS/etc/pdg_table.txt
  fPidSelect=99;// some silly number here

  if ( crit.Contains ( "Proton" ) ) {
    fTypePlus=TDatabasePDG::Instance()->GetParticle ( "p+" );
    if ( 0==fTypePlus ) {
      fTypePlus=TDatabasePDG::Instance()->GetParticle ( "proton" );
    }
    if ( 0==fTypePlus ) {
      fTypePlus=TDatabasePDG::Instance()->GetParticle (2212);
    }
    fPidSelect=4;
    critcopy.ReplaceAll("Proton","");
  } else if ( crit.Contains ( "Kaon" ) ) {
    fTypePlus=TDatabasePDG::Instance()->GetParticle ( "K+" );
    if ( 0==fTypePlus ) {
      fTypePlus=TDatabasePDG::Instance()->GetParticle (321);
    }
    fPidSelect=3;
    critcopy.ReplaceAll("Kaon","");
  } else if ( crit.Contains ( "Pion" ) ) {
    fTypePlus=TDatabasePDG::Instance()->GetParticle ( "pi+" );
    if ( 0==fTypePlus ) {
      fTypePlus=TDatabasePDG::Instance()->GetParticle (211);
    }
    fPidSelect=2;
    critcopy.ReplaceAll("Pion","");
  } else if ( crit.Contains ( "Muon" ) ) {
    fTypePlus=TDatabasePDG::Instance()->GetParticle ( "mu+" );
    if ( 0==fTypePlus ) {
      fTypePlus=TDatabasePDG::Instance()->GetParticle (-13);
    }
    fPidSelect=1;
    critcopy.ReplaceAll("Muon","");
  } else if ( crit.Contains ( "Electron" ) ) {
    fTypePlus=TDatabasePDG::Instance()->GetParticle ( "e+" );
    if ( 0==fTypePlus ) {
      fTypePlus=TDatabasePDG::Instance()->GetParticle (-11);
    }
    fPidSelect=0;
    critcopy.ReplaceAll("Electron","");
  }

//std::cout<<"PndAnaPidSelector: fPidSelect="<<fPidSelect<<std::endl;

  if ( fTypePlus!=0 ) {
    fTypeMinus = CPConjugate ( fTypePlus );
  }

  if ( crit.Contains ( "Plus" ) ) {
    fChargeCrit=1.;
    critcopy.ReplaceAll("Plus","");
  } else if ( crit.Contains ( "Minus" ) ) {
    fChargeCrit=-1.;
    critcopy.ReplaceAll("Minus","");
  } else {
    fChargeCrit=0;
  }

  if(critcopy.Length()>0){
    std::cout<<"Analysis PID selection criteria setting failed. "<<std::endl;
    std::cout<<"Selection String: \""<<crit.Data()<<"\""<<std::endl;
    std::cout<<"Invalid piece:    \""<<critcopy.Data()<<"\""<<std::endl;
    std::cout<<"Viable optios are:"<<std::endl;
    std::cout<<"\tProton,Kaon,Pion,Muon,Electron"<<std::endl;
    std::cout<<"\tVeryLoose,Loose,Tight,VeryTight,Variable,Best,All"<<std::endl;
    std::cout<<"\tPlus,Minus,Charged,Neutral,(All)"<<std::endl;
    Error("SetSelection()","Invalid selection string.");
    return kFALSE;
  }

  return kTRUE;
}

void RhoParticleSelectorBase::SetTypeAndMass ( RhoCandidate *  b )

protectedinherited

Definition at line 143 of file RhoParticleSelectorBase.cxx.

References RhoParticleSelectorBase::fTypeMinus, RhoParticleSelectorBase::fTypePlus, RhoCandidate::GetCharge(), RhoCandidate::SetMass(), and RhoCandidate::SetType().

Referenced by Accept(), RhoGoodPhotonSelector::Accept(), RhoEnergyParticleSelector::Accept(), RhoSimpleMuonSelector::Accept(), RhoSimpleKaonSelector::Accept(), RhoSimpleElectronSelector::Accept(), RhoPlusParticleSelector::Accept(), RhoNeutralParticleSelector::Accept(), RhoMomentumParticleSelector::Accept(), RhoMinusParticleSelector::Accept(), RhoMassParticleSelector::Accept(), RhoSimpleProtonSelector::Accept(), and RhoSimplePionSelector::Accept().

{
  if ( b == 0 ) { return; }
  // Set the particle type
  if ( b->GetCharge() >0.1 ) {
    if ( fTypePlus == 0 ) { return; }
    b->SetType ( fTypePlus );
    b->SetMass ( fTypePlus->Mass() );
  } else {
    if ( fTypeMinus == 0 ) { return; }
    b->SetType ( fTypeMinus );
    b->SetMass ( fTypeMinus->Mass() );
  }
}

Member Data Documentation

Double_t PndAnaPidSelector::fChargeCrit

protected

Pointer to the current parameter object.

Definition at line 42 of file PndAnaPidSelector.h.

Referenced by Accept(), and SetSelection().

criterion RhoParticleSelectorBase::fCriterion

protectedinherited

Particle type to accept.

Definition at line 53 of file RhoParticleSelectorBase.h.

Referenced by Accept(), RhoSimpleProtonSelector::Accept(), RhoSimplePionSelector::Accept(), RhoSimpleMuonSelector::Accept(), RhoSimpleKaonSelector::Accept(), RhoSimpleElectronSelector::Accept(), and RhoParticleSelectorBase::SetCriterion().

Int_t PndAnaPidSelector::fPidSelect

protected

Definition at line 44 of file PndAnaPidSelector.h.

Referenced by Accept(), and SetSelection().

PndAnaSelectorPar* PndAnaPidSelector::fSelectPar

protected

Definition at line 38 of file PndAnaPidSelector.h.

Referenced by PndAnaPidSelector().

TParticlePDG* RhoParticleSelectorBase::fTypeMinus

protectedinherited

Particle type to accept.

Definition at line 52 of file RhoParticleSelectorBase.h.

Referenced by RhoParticleSelectorBase::RhoParticleSelectorBase(), SetSelection(), and RhoParticleSelectorBase::SetTypeAndMass().

TParticlePDG* RhoParticleSelectorBase::fTypePlus

protectedinherited

Definition at line 51 of file RhoParticleSelectorBase.h.

Referenced by RhoParticleSelectorBase::RhoParticleSelectorBase(), SetSelection(), and RhoParticleSelectorBase::SetTypeAndMass().

---

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

• PndAnaPidSelector.h
• PndAnaPidSelector.cxx