include/ElectronBuilder.h

#ifndef ELECTRONBUILDER_H_
#define ELECTRONBUILDER_H_

#include <vector>

#include <TTreeReaderArray.h>

#include <CollectionBuilder.h>
#include <Dataset.h>
#include <PhysicsObjects.h>
#include <Options.h>


class ElectronBuilder : public CollectionBuilder<Electron> {
 public:
  ElectronBuilder(Dataset &dataset, Options const &);

  std::vector<Electron> const &Get() const override;

  std::vector<Electron> const &GetLoose() const;

  std::vector<Electron> const &GetTight() const;

 private:
  void Build() const override;

  double minPtLoose_;

  double minPtTight_;

  double maxRelIsoLoose_;

  double maxRelIsoTight_;

  mutable std::vector<Electron> looseElectrons_;

  mutable std::vector<Electron> tightElectrons_;

  mutable TTreeReaderArray<float> srcPt_, srcEta_, srcPhi_, srcMass_, srcDeltaEtaSc_;
  mutable TTreeReaderArray<float> srcIsolation_;
  mutable TTreeReaderArray<int> srcCharge_;
  mutable TTreeReaderArray<bool> srcId_; // For MVA id
  mutable TTreeReaderArray<float> srcECorr_;
};


inline std::vector<Electron> const &ElectronBuilder::Get() const {
  return GetTight();
}

#endif  // ELECTRONBUILDER_H_

---