include/MeKinFilter.h

#ifndef MEKINFILTER_H_
#define MEKINFILTER_H_

#include <memory>

#include <TTreeReaderValue.h>
#include <TTreeReaderArray.h>

#include <Dataset.h>


class MeKinFilter {
 public:
  MeKinFilter(Dataset &dataset);

  bool operator()() const;

 private:
  double ComputeHt() const;

  bool enabled_;

  double minValue_, maxValue_;

  mutable std::unique_ptr<TTreeReaderValue<UInt_t>> srcNumPart_;
  mutable std::unique_ptr<TTreeReaderArray<Int_t>> srcPdgId_;
  mutable std::unique_ptr<TTreeReaderArray<Int_t>> srcMotherIndex_;
  mutable std::unique_ptr<TTreeReaderArray<Float_t>> srcPartPt_;
};

#endif  // MEKINFILTER_H_

---