EventShapeVarsProducer.cc

Go to the documentation of this file.

#include "PhysicsTools/CandAlgos/plugins/EventShapeVarsProducer.h"

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

#include "PhysicsTools/CandUtils/interface/Thrust.h"

EventShapeVarsProducer::EventShapeVarsProducer(const edm::ParameterSet& cfg)
{
  srcToken_ = consumes<edm::View<reco::Candidate> >(cfg.getParameter<edm::InputTag>("src"));
  r_ = cfg.exists("r") ? cfg.getParameter<double>("r") : 2.;

  produces<double>("thrust");
  //produces<double>("oblateness");
  produces<double>("isotropy");
  produces<double>("circularity");
  produces<double>("sphericity");
  produces<double>("aplanarity");
  produces<double>("C");
  produces<double>("D");

}

void put(edm::Event& evt, double value, const char* instanceName)
{
  evt.put(std::make_unique<double>(value), instanceName);
}

void EventShapeVarsProducer::produce(edm::Event& evt, const edm::EventSetup&)
{
  //std::cout << "<EventShapeVarsProducer::produce>:" << std::endl;

  edm::Handle<edm::View<reco::Candidate> > objects;
  evt.getByToken(srcToken_, objects);

  Thrust thrustAlgo(objects->begin(), objects->end());
  put(evt, thrustAlgo.thrust(), "thrust");
  //put(evt, thrustAlgo.oblateness(), "oblateness");

  EventShapeVariables eventShapeVarsAlgo(*objects);
  put(evt, eventShapeVarsAlgo.isotropy(), "isotropy");
  put(evt, eventShapeVarsAlgo.circularity(), "circularity");
  put(evt, eventShapeVarsAlgo.sphericity(r_), "sphericity");
  put(evt, eventShapeVarsAlgo.aplanarity(r_), "aplanarity");
  put(evt, eventShapeVarsAlgo.C(r_), "C");
  put(evt, eventShapeVarsAlgo.D(r_), "D");
}

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

DEFINE_FWK_MODULE(EventShapeVarsProducer);