PhysicsTools/UtilAlgos/plugins/NTuplingDevice.cc - master - githubmirror/cmssw

PhysicsTools/UtilAlgos/plugins/NTuplingDevice.cc

84 строки2 KB

Christopher Jones

Switched to thread-friendly module types

29 сен 2022, 18:39

29 сен 2022, 18:39ca0d3dc

// -*- C++ -*-
//
// Package:    NTuplingDevice
// Class:      NTuplingDevice
//
/**\class NTuplingDevice NTuplingDevice.cc Workspace/NTuplingDevice/src/NTuplingDevice.cc

 Description: <one line class summary>

 Implementation:
     <Notes on implementation>
*/
//
// Original Author:  Jean-Roch Vlimant
//         Created:  Sun May 11 21:12:46 CEST 2008
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "PhysicsTools/UtilAlgos/interface/NTupler.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

//
// class decleration
//

class NTuplingDevice : public edm::one::EDProducer<edm::one::SharedResources> {
public:
  explicit NTuplingDevice(const edm::ParameterSet&);
  ~NTuplingDevice() override;

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

  // ----------member data ---------------------------
  std::unique_ptr<NTupler> ntupler_;
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
NTuplingDevice::NTuplingDevice(const edm::ParameterSet& iConfig) {
  usesResource(TFileService::kSharedResource);
  //this Ntupler can work with the InputTagDistributor, but should not be configured as such.
  edm::ParameterSet ntPset = iConfig.getParameter<edm::ParameterSet>("Ntupler");
  std::string ntuplerName = ntPset.getParameter<std::string>("ComponentName");
  ntupler_ = NTuplerFactory::get()->create(ntuplerName, ntPset);

  //register the leaves from the ntupler
  ntupler_->registerleaves(producesCollector());

  //put a dummy product if the ntupler does not output on edm
  produces<double>("dummy");
}

NTuplingDevice::~NTuplingDevice() {}

// ------------ method called to produce the data  ------------
void NTuplingDevice::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  ntupler_->fill(iEvent);
  iEvent.put(std::make_unique<double>(0.), "dummy");
}

//define this as a plug-in
DEFINE_FWK_MODULE(NTuplingDevice);