PndEvtFilter Class Reference

#include <PndEvtFilter.h>

Inheritance diagram for PndEvtFilter:

Public Types
enum	ChargeState {   kNeutral =0, kPlus, kMinus, kCharged,   kAll, kChargeLastElement }
enum	MomState { kMomTotal =0, kMomTrans, kMomZ, kMomLastElement }
enum	GeomState {   kTheta =0, kPhi, kVertexZ, kVertexRho,   kVertexRadius, kGeomLastElement }
enum	LogicOp { kAnd =0, kOr, kLogicOpLastElement }

Public Member Functions
	PndEvtFilter ()
	PndEvtFilter (const char *name, const char *title="PndEvtFilter")
virtual	~PndEvtFilter ()
Bool_t	Init ()
Bool_t	FillList (RhoCandList &rhoOutList, Int_t inPdgCode, Double_t pdgCodeCharge=kNoChargeSpecified)
virtual Bool_t	EventMatches (Int_t evtNr)=0
virtual Bool_t	FilterActive ()=0
Bool_t	SetListOfParticles (TClonesArray *ParticleList)
void	PrintAllTParticleInEvent ()
void	SetVerbose (Int_t verbose=12)
void	ShowAcceptedEventNumbers ()
void	ShowEvtNrsToAccept ()
void	SetTestMode (Int_t *arrayPtr, Int_t nEntries)
Bool_t	TestPassed ()
Bool_t	GetCharge (Int_t inPdgCode, Double_t *pdgCodeCharge)

Protected Attributes
TDatabasePDG *	fdbPdg
TClonesArray *	fParticleList
Int_t	fVerbose
Bool_t	fTestMode
std::set< Int_t >	fAcceptedEventNumbers
std::set< Int_t >	fEvtNrsToAccept
Int_t	fEventNr

Static Protected Attributes
static const Double_t	kNoChargeSpecified = -999.9

Private Member Functions
	PndEvtFilter (const PndEvtFilter &G)
PndEvtFilter &	operator= (const PndEvtFilter &)
	ClassDef (PndEvtFilter, 1)

Detailed Description

Definition at line 36 of file PndEvtFilter.h.

Member Enumeration Documentation

enum FairEvtFilter::ChargeState

inherited

Enumerator
kNeutral
kPlus
kMinus
kCharged
kAll
kChargeLastElement

Definition at line 39 of file FairEvtFilter.h.

{ kNeutral=0, kPlus, kMinus, kCharged, kAll, kChargeLastElement };

enum FairEvtFilter::GeomState

inherited

Enumerator
kTheta
kPhi
kVertexZ
kVertexRho
kVertexRadius
kGeomLastElement

Definition at line 41 of file FairEvtFilter.h.

{kTheta=0, kPhi, kVertexZ, kVertexRho, kVertexRadius, kGeomLastElement};

enum FairEvtFilter::LogicOp

inherited

Enumerator
kAnd
kOr
kLogicOpLastElement

Definition at line 42 of file FairEvtFilter.h.

{ kAnd=0, kOr, kLogicOpLastElement };

enum FairEvtFilter::MomState

inherited

Enumerator
kMomTotal
kMomTrans
kMomZ
kMomLastElement

Definition at line 40 of file FairEvtFilter.h.

{ kMomTotal=0, kMomTrans, kMomZ, kMomLastElement};

Constructor & Destructor Documentation

PndEvtFilter::PndEvtFilter

(

)

Default constructor.

Definition at line 10 of file PndEvtFilter.cxx.

                          : FairEvtFilter(){
}

PndEvtFilter::PndEvtFilter	(	const char *	name,
		const char *	title = "PndEvtFilter"
	)

Constructor with name and title

Definition at line 17 of file PndEvtFilter.cxx.

: FairEvtFilter(name, title){
}

PndEvtFilter::~PndEvtFilter ( )

virtual

Destructor.

Definition at line 25 of file PndEvtFilter.cxx.

{ }

PndEvtFilter::PndEvtFilter ( const PndEvtFilter &  G )

private

Member Function Documentation

PndEvtFilter::ClassDef ( PndEvtFilter  , 1    )

private

virtual Bool_t PndEvtFilter::EventMatches ( Int_t  evtNr )

pure virtual

Implements FairEvtFilter.

Implemented in PndEvtFilterOnInvMassCounts.

Bool_t PndEvtFilter::FillList	(	RhoCandList &	rhoOutList,
		Int_t	inPdgCode,
		Double_t	pdgCodeCharge = kNoChargeSpecified
	)

Definition at line 31 of file PndEvtFilter.cxx.

References RhoCandList::Add(), RhoCandList::Cleanup(), Double_t, FairEvtFilter::fdbPdg, FairEvtFilter::fParticleList, FairEvtFilter::GetCharge(), FairEvtFilter::kNoChargeSpecified, particle, RhoCandidate::SetMcTruth(), RhoCandidate::SetPos(), and RhoCandidate::SetType().

Referenced by PndEvtFilterOnInvMassCounts::EventMatches().

{
        rhoOutList.Cleanup();


        // Get charge corresponding to pdgCode if it has not been passed as an argument by the user
        if ( kNoChargeSpecified == pdgCodeCharge ){
                if ( kFALSE == GetCharge( inPdgCode, &pdgCodeCharge ) ) {
                        return kFALSE;
                }
        }



        // find all TParticle on stack with a charge(particle) == charge(pdgCode)
        for (Int_t iPart=0; iPart<fParticleList->GetEntries(); ++iPart) {
                TParticle *particle = (TParticle*)fParticleList->At(iPart);

                // get charge for particle
                TParticlePDG*  pdt = particle->GetPDG();
                if (0==pdt) continue; // unknown particle type (KG, 10/2015)

                Double_t pCharge = pdt->Charge()/3.; // TParticlePDG contains charge in units of |e|/3

                if ( pdgCodeCharge != pCharge ){ continue; } // skip all particles with different charge


                //              // mark stable particles
                //              Bool_t isStable = kFALSE;
                //              switch(abs(particle->GetPdgCode())) {
                //              case 22:   isStable = true; break;
                //              case 11:   isStable = true; break;
                //              case 13:   isStable = true; break;
                //              case 211:  isStable = true; break;
                //              case 321:  isStable = true; break;
                //              case 2212: isStable = true; break;
                //              }
                //
                //              if (kFALSE == isStable) { continue; } // skip all particles which are not considered stable


                // store TParticle as RhoCandidate in output TCA and modify 4 momentum to match particle hypothesis
                TVector3 pVertex(particle->Vx(),particle->Vy(),particle->Vz());
                TLorentzVector p4(particle->Px(),particle->Py(),particle->Pz(),particle->Energy());
                // assign mass hypothesis according to pdgCode
                p4.SetVectM(p4.Vect(),fdbPdg->GetParticle(inPdgCode)->Mass());
                // construct RhoCandidate
                RhoCandidate rhoCand(p4,pCharge);
                rhoCand.SetMcTruth(&rhoCand);
                rhoCand.SetPos(pVertex);
                rhoCand.SetType(inPdgCode);
                // save to output
                rhoOutList.Add(&rhoCand);
        }//trackloop


        return kTRUE;
}

virtual Bool_t PndEvtFilter::FilterActive ( )

pure virtual

Implements FairEvtFilter.

Implemented in PndEvtFilterOnInvMassCounts.

Bool_t FairEvtFilter::GetCharge ( Int_t  inPdgCode, Double_t *  pdgCodeCharge  )

inherited

Definition at line 62 of file FairEvtFilter.cxx.

References FairEvtFilter::fdbPdg, FairEvtFilter::fVerbose, and FairEvtFilter::kNoChargeSpecified.

Referenced by FairEvtFilterOnSingleParticleCounts::CountCharge(), FillList(), and PndEvtFilterOnInvMassCounts::SetPdgCodesToCombine().

{
        // Try to find the pdg code
        TParticlePDG *ptrToPdg = fdbPdg->GetParticle(inPdgCode);
        if ( 0 == ptrToPdg) {
                // ignore particles with unknown charges
                std::cout << "WARNING from FairEvtFilter::GetCharge Charge of pdgCode " << inPdgCode << " is unknown and will be ignored!\n";
                *pdgCodeCharge = kNoChargeSpecified;
                return kFALSE;
        }
        *pdgCodeCharge = ptrToPdg->Charge()/3.; // TParticlePDG contains charge in units of |e|/3
        if ( fVerbose > 1 ) std::cout << "Found pdgCodeCharge = " << *pdgCodeCharge << " for inPdgCode " << inPdgCode << '\n';
        return kTRUE;
}

Bool_t PndEvtFilter::Init ( )

inline

Definition at line 54 of file PndEvtFilter.h.

{ return kTRUE;}

PndEvtFilter& PndEvtFilter::operator= ( const PndEvtFilter &  )

inlineprivate

Definition at line 74 of file PndEvtFilter.h.

{return *this;}

void FairEvtFilter::PrintAllTParticleInEvent ( )

inherited

Definition at line 52 of file FairEvtFilter.cxx.

References FairEvtFilter::fParticleList, and particle.

Referenced by PndEvtFilterOnInvMassCounts::EventMatches(), and FairEvtFilterOnSingleParticleCounts::EventMatches().

                                            {
        // sanity checks
        if (0==fParticleList){ return; }
        for (Int_t iPart=0; iPart<fParticleList->GetEntries(); ++iPart) {
                TParticle *particle = (TParticle*)fParticleList->At(iPart);
                particle->Print();
        }
}

Bool_t FairEvtFilter::SetListOfParticles ( TClonesArray *  ParticleList )

inlineinherited

Definition at line 57 of file FairEvtFilter.h.

References FairEvtFilter::fParticleList.

Referenced by FairFilteredPrimaryGenerator::GenerateEvent().

{fParticleList=ParticleList; return kTRUE;};

void FairEvtFilter::SetTestMode ( Int_t *  arrayPtr, Int_t  nEntries  )

inlineinherited

Definition at line 91 of file FairEvtFilter.h.

References FairEvtFilter::fEvtNrsToAccept, and FairEvtFilter::fTestMode.

                                                         {
                //turns on the test mode with the declared fEvtNrsToAccept
                fTestMode=kTRUE;
                std::set<Int_t> evtNrsToAccept (arrayPtr,arrayPtr+nEntries);
                fEvtNrsToAccept=evtNrsToAccept;
        }

void FairEvtFilter::SetVerbose ( Int_t  verbose = 12 )

inlineinherited

Definition at line 67 of file FairEvtFilter.h.

References FairEvtFilter::fVerbose, and verbose.

Referenced by sim_filter_inv_mass().

                                         {
                if ( verbose >= 0 ){
                        fVerbose = verbose;
                        std::cout << "FairEvtFilter: fVerbose is now set to " << verbose << "\n";
                } else {
                        std::cout << "\n\n\n  -WARNING from FairEvtFilter: verbose must be a positive number! Check your SetVerbose call!\n\n\n";
                }
        }

void FairEvtFilter::ShowAcceptedEventNumbers ( )

inlineinherited

Definition at line 77 of file FairEvtFilter.h.

References FairEvtFilter::fAcceptedEventNumbers.

                                       {
                // for QA
                // shows fAcceptedEventNumbers that is filled after running a simulation
                std::cout << "\n fAcceptedEventNumbers" << " = " << fAcceptedEventNumbers;
        }

void FairEvtFilter::ShowEvtNrsToAccept ( )

inlineinherited

Definition at line 84 of file FairEvtFilter.h.

References FairEvtFilter::fEvtNrsToAccept.

                                 {
                // for QA
                //shows fEvtNrsToAccept that has to be set if you want to run your simulation in test mode
                std::cout << "\n fEvtNrsToAccept" << " = " << fEvtNrsToAccept;
        }

Bool_t FairEvtFilter::TestPassed ( )

inlineinherited

Definition at line 99 of file FairEvtFilter.h.

References FairEvtFilter::fAcceptedEventNumbers, FairEvtFilter::fEvtNrsToAccept, and FairEvtFilter::fTestMode.

                           {
                if(kFALSE==fTestMode){
                        std::cout << "\n\n\n WARNING from FairEvtFilter: Test mode not set.\n\n\n";
                        return kFALSE;
                }
                if(fAcceptedEventNumbers==fEvtNrsToAccept){
                        //std::cout << "\n\n\n FairEvtFilter: Test passed.\n\n\n";
                        return kTRUE;
                }else{
                        //std::cout << "\n\n\n FairEvtFilter: Test failed. Check your SetTestMode calls. \n\n\n";
                        return kFALSE;
                }
        }

Member Data Documentation

std::set<Int_t> FairEvtFilter::fAcceptedEventNumbers

protectedinherited

Definition at line 134 of file FairEvtFilter.h.

Referenced by PndEvtFilterOnInvMassCounts::EventMatches(), FairEvtFilterOnSingleParticleCounts::EventMatches(), FairEvtFilter::FairEvtFilter(), FairEvtFilter::ShowAcceptedEventNumbers(), and FairEvtFilter::TestPassed().

TDatabasePDG* FairEvtFilter::fdbPdg

protectedinherited

Definition at line 126 of file FairEvtFilter.h.

Referenced by PndEvtFilterOnInvMassCounts::EventMatches(), FairEvtFilter::FairEvtFilter(), FillList(), and FairEvtFilter::GetCharge().

Int_t FairEvtFilter::fEventNr

protectedinherited

Definition at line 136 of file FairEvtFilter.h.

std::set<Int_t> FairEvtFilter::fEvtNrsToAccept

protectedinherited

Definition at line 135 of file FairEvtFilter.h.

Referenced by FairEvtFilter::SetTestMode(), FairEvtFilter::ShowEvtNrsToAccept(), and FairEvtFilter::TestPassed().

TClonesArray* FairEvtFilter::fParticleList

protectedinherited

Definition at line 131 of file FairEvtFilter.h.

Referenced by PndEvtFilterOnInvMassCounts::EventMatches(), FairEvtFilterOnSingleParticleCounts::EventMatches(), FairEvtFilter::FairEvtFilter(), FillList(), FairEvtFilter::PrintAllTParticleInEvent(), and FairEvtFilter::SetListOfParticles().

Bool_t FairEvtFilter::fTestMode

protectedinherited

Definition at line 133 of file FairEvtFilter.h.

Referenced by FairEvtFilter::SetTestMode(), and FairEvtFilter::TestPassed().

Int_t FairEvtFilter::fVerbose

protectedinherited

Definition at line 132 of file FairEvtFilter.h.

Referenced by FairEvtFilterOnSingleParticleCounts::AcceptChargeCounter(), FairEvtFilterOnSingleParticleCounts::AcceptGeometry(), FairEvtFilterOnSingleParticleCounts::AcceptMomentum(), FairEvtFilterOnSingleParticleCounts::AcceptPdgCounter(), FairEvtFilterOnSingleParticleCounts::AndMinMaxCharge(), FairEvtFilterOnSingleParticleCounts::AndMinMaxGeom(), FairEvtFilterOnSingleParticleCounts::AndMinMaxMom(), FairEvtFilterOnSingleParticleCounts::AndMinMaxPdgCodes(), FairEvtFilterOnSingleParticleCounts::CountCharge(), FairEvtFilterOnSingleParticleCounts::CountPdg(), PndEvtFilterOnInvMassCounts::EventMatches(), FairEvtFilterOnSingleParticleCounts::EventMatches(), FairEvtFilter::GetCharge(), FairEvtFilterOnSingleParticleCounts::InitCounters(), and FairEvtFilter::SetVerbose().

const Double_t FairEvtFilter::kNoChargeSpecified = -999.9

staticprotectedinherited

Definition at line 129 of file FairEvtFilter.h.

Referenced by FillList(), FairEvtFilter::GetCharge(), and PndEvtFilterOnInvMassCounts::SetPdgCodesToCombine().

---

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

• PndEvtFilter.h
• PndEvtFilter.cxx