TotemRPClusterProducer.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/CTPPSDigi/interface/TotemRPDigi.h"
 
#include "RecoPPS/Local/interface/TotemRPClusterProducerAlgorithm.h"
 
//----------------------------------------------------------------------------------------------------
 
class TotemRPClusterProducer : public edm::stream::EDProducer<> {
public:
  explicit TotemRPClusterProducer(const edm::ParameterSet& conf);
 
  ~TotemRPClusterProducer() override {}
 
  void produce(edm::Event& e, const edm::EventSetup& c) override;
  static void fillDescriptions(edm::ConfigurationDescriptions&);
 
private:
  edm::ParameterSet conf_;
  int verbosity_;
  edm::InputTag digiInputTag_;
  edm::EDGetTokenT<edm::DetSetVector<TotemRPDigi>> digiInputTagToken_;
 
  TotemRPClusterProducerAlgorithm algorithm_;
 
  void run(const edm::DetSetVector<TotemRPDigi>& input, edm::DetSetVector<TotemRPCluster>& output);
};
 
//----------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------
 
using namespace std;
using namespace edm;
 
//----------------------------------------------------------------------------------------------------
 
TotemRPClusterProducer::TotemRPClusterProducer(edm::ParameterSet const& conf) : conf_(conf), algorithm_(conf) {
  verbosity_ = conf.getParameter<int>("verbosity");
 
  digiInputTag_ = conf.getParameter<edm::InputTag>("tagDigi");
  digiInputTagToken_ = consumes<edm::DetSetVector<TotemRPDigi>>(digiInputTag_);
 
  produces<edm::DetSetVector<TotemRPCluster>>();
}
 
//----------------------------------------------------------------------------------------------------
 
void TotemRPClusterProducer::produce(edm::Event& e, const edm::EventSetup& es) {
  // get input
  edm::Handle<edm::DetSetVector<TotemRPDigi>> input;
  e.getByToken(digiInputTagToken_, input);
 
  // prepare output
  DetSetVector<TotemRPCluster> output;
 
  // run clusterisation
  if (!input->empty())
    run(*input, output);
 
  // save output to event
  e.put(make_unique<DetSetVector<TotemRPCluster>>(output));
}
 
//----------------------------------------------------------------------------------------------------
 
void TotemRPClusterProducer::run(const edm::DetSetVector<TotemRPDigi>& input,
                                 edm::DetSetVector<TotemRPCluster>& output) {
  for (const auto& ds_digi : input) {
    edm::DetSet<TotemRPCluster>& ds_cluster = output.find_or_insert(ds_digi.id);
 
    algorithm_.buildClusters(ds_digi.id, ds_digi.data, ds_cluster.data);
  }
}
 
//----------------------------------------------------------------------------------------------------
 
void TotemRPClusterProducer::fillDescriptions(edm::ConfigurationDescriptions& descr) {
  edm::ParameterSetDescription desc;
 
  desc.add<edm::InputTag>("tagDigi", edm::InputTag("totemRPRawToDigi", "TrackingStrip"))
      ->setComment("input digis collection to retrieve");
  desc.add<int>("verbosity", 0);
 
  descr.add("totemRPClusterProducer", desc);
}
 
//----------------------------------------------------------------------------------------------------
 
DEFINE_FWK_MODULE(TotemRPClusterProducer);