ShallowTree.h

Go to the documentation of this file.

#ifndef Shallow_Tree_h
#define Shallow_Tree_h

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include <string>
#include <vector>
#include <TTree.h>

class ShallowTree : public edm::EDAnalyzer {
private:
  void beginJob() override;
  void analyze(const edm::Event&, const edm::EventSetup&) override;
  void endJob() override {}

  template <class T>
  void eat(edm::BranchDescription const* desc) {
    consumes<T>(edm::InputTag(desc->moduleLabel(), desc->productInstanceName()));
  }

  class BranchConnector {
  public:
    virtual ~BranchConnector(){};
    virtual void connect(const edm::Event&) = 0;
  };

  template <class T>
  class TypedBranchConnector : public BranchConnector {
  private:
    std::string ml;   //module label
    std::string pin;  //product instance name
    T object_;
    T* object_ptr_;

  public:
    TypedBranchConnector(edm::BranchDescription const*, std::string, TTree*);
    void connect(const edm::Event&) override;
  };

  edm::Service<TFileService> fs_;
  TTree* tree_;
  std::vector<BranchConnector*> connectors_;

public:
  explicit ShallowTree(const edm::ParameterSet& iConfig);  // : pset(iConfig) {}

  enum LEAFTYPE {
    BOOL = 1,
    BOOL_V,
    SHORT,
    SHORT_V,
    U_SHORT,
    U_SHORT_V,
    INT,
    INT_V,
    U_INT,
    U_INT_V,
    FLOAT,
    FLOAT_V,
    DOUBLE,
    DOUBLE_V,
    LONG,
    LONG_V,
    U_LONG,
    U_LONG_V,
    CHAR,
    CHAR_V,
    U_CHAR,
    U_CHAR_V
  };
};

#endif