PndEvtFilterOnInvMassCounts.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----             PndEvtFilterOnInvMassCounts header file           -----
// -------------------------------------------------------------------------


#ifndef PndEvtFilterOnInvMassCounts_H_
#define PndEvtFilterOnInvMassCounts_H_

#include "PndEvtFilter.h"
#include "TMath.h"
#include <map>
#include <vector>
#include <iostream>
#include "RhoCandidate.h"
#include "RhoCandList.h"
#include "RhoMassParticleSelector.h"
#include "TLorentzVector.h"
#include "RhoMassParticleSelector.h"


class PndEvtFilterOnInvMassCounts : public PndEvtFilter
{

public:

        //Default constructor
        PndEvtFilterOnInvMassCounts();

        //Constructor with name and title
        PndEvtFilterOnInvMassCounts(const char* name, const char* title="PndEvtFilterOnInvMassCounts");

        // Destructor
        virtual ~PndEvtFilterOnInvMassCounts();

        // User interfaces
        // returns kTRUE if setting was added, otherwise returns kFALSE
        Bool_t SetPdgCodesToCombine( Int_t pdgCode1, Int_t pdgCode2, Int_t pdgCode3=kInvalidPdgCode, Int_t pdgCode4=kInvalidPdgCode, Int_t pdgCode5=kInvalidPdgCode );

        Bool_t SetMinMaxInvMass( Double_t min, Double_t max );
        Bool_t SetRhoMassParticleSelector( const char* name, Double_t cv, Double_t w, const char* type ); // same arguments as constructor of RhoMassParticleSelector

        Bool_t SetMinMaxCounts( Int_t min, Int_t max );
        Bool_t SetMinCounts( Int_t min ){ return SetMinMaxCounts(min, 9999); };
        Bool_t SetMaxCounts( Int_t max ){ return SetMinMaxCounts(0, max); };




        // checks if the particles in the event (fParticleList) suit the filter settings
        // kTRUE if event matches, kFALSE otherwise
        Bool_t EventMatches(Int_t evtNr);


        // returns kTRUE if any filter is turned on
        Bool_t FilterActive(){ return ( fInvMassRangeSet && fPgdCodesSet && fCountRangeSet ); };


private:
        // minimum and maximum acceptable multiplicities for the particle combinations
        std::pair<Int_t,Int_t> fCountsMinMax;
        // inv. mass selector which implements the minimum and maximum acceptable invariant mass range for the particle combinations
        RhoMassParticleSelector *fInvMassSel;
        // vector containing the pdg codes which should be combined along with its corresponding charges
        std::vector< std::pair< Int_t, Double_t > > fPdgCodesCharges;



        Bool_t fInvMassRangeSet; // is kTRUE if invariant mass is set
        Bool_t fPgdCodesSet; // is kTRUE if pdg codes are set
        Bool_t fCountRangeSet; // is kTRUE if min and max counts are set


        // constant holding an integer number which is not used as a pdg code
        // this serves as a place holder when constructing a vector from integers given by the user
        static const Int_t kInvalidPdgCode = 0;

        ClassDef(PndEvtFilterOnInvMassCounts,1);

};


#endif /* PndEvtFilterOnInvMassCounts_H_ */