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 "RecoCTPPS/TotemRPLocal/interface/TotemRPClusterProducerAlgorithm.h"
 
//----------------------------------------------------------------------------------------------------

class TotemRPClusterProducer : public edm::stream::EDProducer<>
{
  public:
  
    explicit TotemRPClusterProducer(const edm::ParameterSet& conf);
  
    virtual ~TotemRPClusterProducer() {}
  
    virtual 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->size())
    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);