include/JetBuilder.h

#ifndef HZZ2L2NU_INCLUDE_JETBUILDER_H_
#define HZZ2L2NU_INCLUDE_JETBUILDER_H_

#include <initializer_list>
#include <optional>
#include <vector>

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

#include <CollectionBuilder.h>
#include <Dataset.h>
#include <GenJetBuilder.h>
#include <JetCorrector.h>
#include <Options.h>
#include <PhysicsObjects.h>
#include <PileUpIdFilter.h>


class JetBuilder : public CollectionBuilder<Jet> {
 public:
  JetBuilder(
      Dataset &dataset, Options const &options, TabulatedRngEngine &rngEngine,
      PileUpIdFilter const *pileUpIdFilter = nullptr);

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

  std::vector<Jet> const &GetRejected() const;

  void SetGenJetBuilder(GenJetBuilder const *genJetBuilder);

 private:
  void AddType1Correction(
      TLorentzVector const &rawP4, double area,
      double jecOrig, double jecNew, double jerFactor,
      double emFraction, double muonFraction) const;

  void Build() const override;

  GenJet const *FindGenMatch(TLorentzVector const &p4,
                             double ptResolution) const;

  double GetJerFactor(TLorentzVector const &corrP4, int rngChannel) const;

  void ProcessJets() const;

  void ProcessSoftJets() const;

  bool SetPileUpInfo(Jet &jet, int index) const;

  GenJetBuilder const *genJetBuilder_;

  PileUpIdFilter const *pileUpIdFilter_;

  double minPt_;

  double maxAbsEta_;

  double minPtType1Corr_;

  bool ptMissEeNoise_;

  bool ptMissJer_;

  bool ptMissPogJets_;

  double pileUpIdMinPt_, pileUpIdMaxPt_;

  mutable std::vector<Jet> jets_;

  mutable std::vector<Jet> rejectedJets_;

  bool isSim_;

  int jetIdBit_;

  JetCorrector jetCorrector_;

  mutable TTreeReaderArray<float> srcPt_, srcEta_, srcPhi_, srcMass_;
  mutable TTreeReaderArray<float> srcArea_, srcRawFactor_;
  mutable TTreeReaderArray<float> srcChEmEF_, srcNeEmEF_, srcMuonFraction_;
  mutable TTreeReaderArray<float> srcBTag_;
  mutable TTreeReaderArray<int> srcId_, srcPileUpId_;
  mutable TTreeReaderValue<float> puRho_;
  mutable std::optional<TTreeReaderArray<int>> srcHadronFlavour_,
      srcPartonFlavour_, srcGenJetIdx_;

  // Properties of soft jets, which are used in the type 1 correction of ptmiss
  mutable TTreeReaderArray<float> softRawPt_, softEta_, softPhi_, softArea_;
  mutable TTreeReaderArray<float> softMuonFraction_;
};

#endif  // HZZ2L2NU_INCLUDE_JETBUILDER_H_

---