TotemRPRecHitProducer.cc

Go to the documentation of this file.

/****************************************************************************
*
* This is a part of TOTEM offline software.
* Authors:
*   Hubert Niewiadomski
*   Jan Kašpar (jan.kaspar@gmail.com)
*
****************************************************************************/
 
#include "FWCore/Framework/interface/MakerMacros.h"
 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/Common/interface/DetSet.h"
#include "DataFormats/CTPPSReco/interface/TotemRPCluster.h"
#include "DataFormats/CTPPSReco/interface/TotemRPRecHit.h"
 
#include "RecoPPS/Local/interface/TotemRPRecHitProducerAlgorithm.h"
 
//----------------------------------------------------------------------------------------------------
 
class TotemRPRecHitProducer : public edm::stream::EDProducer<> {
public:
  explicit TotemRPRecHitProducer(const edm::ParameterSet& conf);
 
  ~TotemRPRecHitProducer() override {}
 
  void produce(edm::Event& e, const edm::EventSetup& c) override;
  static void fillDescriptions(edm::ConfigurationDescriptions&);
 
private:
  const edm::ParameterSet conf_;
  int verbosity_;
 
  TotemRPRecHitProducerAlgorithm algorithm_;
 
  edm::InputTag tagCluster_;
  edm::EDGetTokenT<edm::DetSetVector<TotemRPCluster>> tokenCluster_;
};
 
//----------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------
 
using namespace std;
using namespace edm;
 
//----------------------------------------------------------------------------------------------------
 
TotemRPRecHitProducer::TotemRPRecHitProducer(const edm::ParameterSet& conf) : conf_(conf), algorithm_(conf) {
  verbosity_ = conf.getParameter<int>("verbosity");
 
  tagCluster_ = conf.getParameter<edm::InputTag>("tagCluster");
  tokenCluster_ = consumes<edm::DetSetVector<TotemRPCluster>>(tagCluster_);
 
  produces<edm::DetSetVector<TotemRPRecHit>>();
}
 
//----------------------------------------------------------------------------------------------------
 
void TotemRPRecHitProducer::produce(edm::Event& e, const edm::EventSetup& es) {
  // get input
  edm::Handle<edm::DetSetVector<TotemRPCluster>> input;
  e.getByToken(tokenCluster_, input);
 
  // prepare output
  DetSetVector<TotemRPRecHit> output;
 
  // build reco hits
  for (auto& ids : *input) {
    DetSet<TotemRPRecHit>& ods = output.find_or_insert(ids.detId());
    algorithm_.buildRecoHits(ids, ods);
  }
 
  // save output
  e.put(make_unique<DetSetVector<TotemRPRecHit>>(output));
}
 
//----------------------------------------------------------------------------------------------------
 
void TotemRPRecHitProducer::fillDescriptions(edm::ConfigurationDescriptions& descr) {
  edm::ParameterSetDescription desc;
 
  desc.add<edm::InputTag>("tagCluster", edm::InputTag("totemRPClusterProducer"))
      ->setComment("input clusters collection to retrieve");
  desc.add<int>("verbosity", 0);
 
  descr.add("totemRPRecHitProducer", desc);
}
 
//----------------------------------------------------------------------------------------------------
 
DEFINE_FWK_MODULE(TotemRPRecHitProducer);