MuBaseFlatTableProducer.h

Go to the documentation of this file.

#ifndef Mu_MuBaseFlatTableProducer_h
#define Mu_MuBaseFlatTableProducer_h

#include "DPGAnalysis/MuonTools/interface/MuNtupleUtils.h"

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "DataFormats/NanoAOD/interface/FlatTable.h"

#include <string>

class MuBaseFlatTableProducer : public edm::stream::EDProducer<> {
public:
  explicit MuBaseFlatTableProducer(const edm::ParameterSet &);

  void beginRun(const edm::Run &run, const edm::EventSetup &config) final;

  void produce(edm::Event &, const edm::EventSetup &) final;

  void endRun(const edm::Run &, const edm::EventSetup &) final {}

protected:
  std::string m_name;

  virtual void getFromES(const edm::Run &run, const edm::EventSetup &environment) {}

  virtual void getFromES(const edm::EventSetup &environment) {}

  virtual void fillTable(edm::Event &ev) = 0;

  static constexpr int DEFAULT_INT_VAL{-999};

  static constexpr int8_t DEFAULT_INT8_VAL{-99};

  static constexpr int DEFAULT_INT_VAL_POS{-1};

  static constexpr double DEFAULT_DOUBLE_VAL{-999.0};

  static constexpr double DEFAULT_DOUBLE_VAL_POS{-1.0};

  template <typename T>
  void addColumn(std::unique_ptr<nanoaod::FlatTable> &table,
                 const std::string name,
                 const std::vector<T> &vec,
                 const std::string descr) {
    table->template addColumn<T, std::vector<T>>(name, vec, descr);
  }
};

#endif