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_ */