HighEnergySliceRemoval_module.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// Class:       HighEnergySliceRemoval
// Module Type: filter
// File:        HighEnergySliceRemoval_module.cc
//
// Generated at Tue May 17 14:20:30 2016 by Martin Frank using artmod
// from cetpkgsupport v1_08_07.
////////////////////////////////////////////////////////////////////////

#include <art/Framework/Core/EDFilter.h>
#include <art/Framework/Core/ModuleMacros.h>
#include <art/Framework/Principal/Event.h>
#include <art/Framework/Principal/Handle.h>
#include <art/Framework/Principal/Run.h>
#include <art/Framework/Principal/SubRun.h>
// #include <canvas/Utilities/InputTag.h>
#include <art/Utilities/InputTag.h>
#include <fhiclcpp/ParameterSet.h>
#include <messagefacility/MessageLogger/MessageLogger.h>

#include "RecoBase/Cluster.h"

#include <memory>

namespace mono {
  class HighEnergySliceRemoval;
}

class mono::HighEnergySliceRemoval : public art::EDFilter {
public:
  explicit HighEnergySliceRemoval(fhicl::ParameterSet const & p);

  HighEnergySliceRemoval(HighEnergySliceRemoval const &) = delete;
  HighEnergySliceRemoval(HighEnergySliceRemoval &&) = delete;
  HighEnergySliceRemoval & operator = (HighEnergySliceRemoval const &) = delete;
  HighEnergySliceRemoval & operator = (HighEnergySliceRemoval &&) = delete;

  bool filter(art::Event & e) override;

private:
  bool is_high_energy_slice(rb::Cluster const& slice) const;

  double high_energy_adc_;
  std::string slice_label_;
};



mono::HighEnergySliceRemoval::HighEnergySliceRemoval
(fhicl::ParameterSet const & p) :
  high_energy_adc_(p.get<double>("high_energy_adc")),
  slice_label_    (p.get<std::string>("slice_label"))
{
  produces<std::vector<rb::Cluster> >();

  std::cout << "mono::HighEnergySliceRemoval: high_energy_adc = "
            << high_energy_adc_
            << std::endl;
}



bool mono::HighEnergySliceRemoval::filter(art::Event & e)
{
  art::Handle<std::vector<rb::Cluster> > in_slices;
  e.getByLabel(slice_label_, in_slices);

  std::unique_ptr<std::vector<rb::Cluster> > out_slices
    (new std::vector<rb::Cluster>(*in_slices));

  // we are using rb::Cluster's noise flag to identify high energy slices
  // that we consider noise for now
  for (auto & slice : *out_slices)
    if (is_high_energy_slice(slice))
      slice.SetNoise(true);

  e.put(std::move(out_slices));
  return true;
}



bool mono::HighEnergySliceRemoval::is_high_energy_slice
(rb::Cluster const& slice) const
{
  if (slice.IsNoise())
    return false;

  if (slice.TotalADC() < high_energy_adc_)
    return false;

  return true;
}



DEFINE_ART_MODULE(mono::HighEnergySliceRemoval)