WaveformRoiFinder_module.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// Class:       WaveformRoiFinder
// Plugin Type: producer (art v3_05_00)
// File:        WaveformRoiFinder_module.cc
//
// Authors:     Mike Wang (mwang@fnal.gov)
//              Lorenzo Uboldi (uboldi.lorenzo@gmail.com)
//              Tingjun Yang (tjyang@fnal.gov)
//              Wanwei Wu (wwu@fnal.gov)
//
// Generated at Fri Apr 10 23:30:12 2020 by Tingjun Yang using cetskelgen
// from cetlib version v3_10_00.
// Based on the Analyzer module written by Mike Wang.
////////////////////////////////////////////////////////////////////////

#include "art/Framework/Core/EDProducer.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 "art/Utilities/make_tool.h"
#include "canvas/Utilities/InputTag.h"
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

#include "larcore/Geometry/Geometry.h"
#include "lardataobj/RawData/RawDigit.h"
#include "lardataobj/RawData/raw.h"
#include "lardataobj/RecoBase/Wire.h"
#include "larrecodnn/ImagePatternAlgs/ToolInterfaces/IWaveformRecog.h"

#include <memory>

namespace nnet {
  class WaveformRoiFinder;
}

class nnet::WaveformRoiFinder : public art::EDProducer {
public:
  explicit WaveformRoiFinder(fhicl::ParameterSet const& p);
  // The compiler-generated destructor is fine for non-base
  // classes without bare pointers or other resource use.

  // Plugins should not be copied or assigned.
  WaveformRoiFinder(WaveformRoiFinder const&) = delete;
  WaveformRoiFinder(WaveformRoiFinder&&) = delete;
  WaveformRoiFinder& operator=(WaveformRoiFinder const&) = delete;
  WaveformRoiFinder& operator=(WaveformRoiFinder&&) = delete;

  // Required functions.
  void produce(art::Event& e) override;

private:
  art::InputTag fRawProducerLabel;
  art::InputTag fWireProducerLabel;

  std::vector<std::unique_ptr<wavrec_tool::IWaveformRecog>> fWaveformRecogToolVec;

  int fNPlanes;
  unsigned int fWaveformSize; // Full waveform size
};

nnet::WaveformRoiFinder::WaveformRoiFinder(fhicl::ParameterSet const& p)
  : EDProducer{p}
  , fRawProducerLabel(p.get<art::InputTag>("RawProducerLabel", ""))
  , fWireProducerLabel(p.get<art::InputTag>("WireProducerLabel", ""))
{
  // use either raw waveform or recob waveform
  if (fRawProducerLabel.empty() && fWireProducerLabel.empty()) {
    throw cet::exception("WaveformRoiFinder")
      << "Both RawProducerLabel and WireProducerLabel are empty";
  }

  if ((!fRawProducerLabel.empty()) && (!fWireProducerLabel.empty())) {
    throw cet::exception("WaveformRoiFinder")
      << "Only one of RawProducerLabel and WireProducerLabel should be set";
  }

  auto const* geo = lar::providerFrom<geo::Geometry>();
  fNPlanes = geo->Nplanes();

  // Signal/Noise waveform recognition tool
  fWaveformRecogToolVec.reserve(fNPlanes);
  auto const tool_psets = p.get<std::vector<fhicl::ParameterSet>>("WaveformRecogs");
  fWaveformSize = tool_psets[0].get<unsigned int>("WaveformSize");
  for (auto const& pset : tool_psets) {
    fWaveformRecogToolVec.push_back(art::make_tool<wavrec_tool::IWaveformRecog>(pset));
  }

  produces<std::vector<recob::Wire>>();
}

void
nnet::WaveformRoiFinder::produce(art::Event& e)
{
  art::Handle<std::vector<raw::RawDigit>> rawListHandle;
  std::vector<art::Ptr<raw::RawDigit>> rawlist;
  if (e.getByLabel(fRawProducerLabel, rawListHandle)) art::fill_ptr_vector(rawlist, rawListHandle);

  art::Handle<std::vector<recob::Wire>> wireListHandle;
  std::vector<art::Ptr<recob::Wire>> wirelist;
  if (e.getByLabel(fWireProducerLabel, wireListHandle))
    art::fill_ptr_vector(wirelist, wireListHandle);

  std::unique_ptr<std::vector<recob::Wire>> outwires(new std::vector<recob::Wire>);

  auto const* geo = lar::providerFrom<geo::Geometry>();

  //##############################
  //### Looping over the wires ###
  //##############################
  for (unsigned int ich = 0; ich < (rawlist.empty() ? wirelist.size() : rawlist.size()); ++ich) {

    std::vector<float> inputsignal(fWaveformSize);

    int view = -1;

    if (!wirelist.empty()) {
      const auto& wire = wirelist[ich];
      const auto& signal = wire->Signal();

      view = wire->View();

      for (size_t itck = 0; itck < inputsignal.size(); ++itck) {
        inputsignal[itck] = signal[itck];
      }
    }
    else if (!rawlist.empty()) {
      const auto& digitVec = rawlist[ich];

      view = geo->View(rawlist[ich]->Channel());

      std::vector<short> rawadc(fWaveformSize);
      raw::Uncompress(digitVec->ADCs(), rawadc, digitVec->GetPedestal(), digitVec->Compression());
      for (size_t itck = 0; itck < rawadc.size(); ++itck) {
        inputsignal[itck] = rawadc[itck] - digitVec->GetPedestal();
      }
    }

    // ... use waveform recognition CNN to perform inference on each window
    std::vector<bool> inroi(fWaveformSize, false);
    inroi = fWaveformRecogToolVec[view]->findROI(inputsignal);

    std::vector<float> sigs;
    int lastsignaltick = -1;
    int roistart = -1;

    recob::Wire::RegionsOfInterest_t rois(fWaveformSize);

    for (size_t i = 0; i < fWaveformSize; ++i) {
      if (inroi[i]) {
        if (sigs.empty()) {
          sigs.push_back(inputsignal[i]);
          lastsignaltick = i;
          roistart = i;
        }
        else {
          if (int(i) != lastsignaltick + 1) {
            rois.add_range(roistart, std::move(sigs));
            sigs.clear();
            sigs.push_back(inputsignal[i]);
            lastsignaltick = i;
            roistart = i;
          }
          else {
            sigs.push_back(inputsignal[i]);
            lastsignaltick = i;
          }
        }
      }
    }
    if (!sigs.empty()) { rois.add_range(roistart, std::move(sigs)); }
    if (!wirelist.empty()) {
      outwires->emplace_back(recob::Wire(rois, wirelist[ich]->Channel(), wirelist[ich]->View()));
    }
    else if (!rawlist.empty()) {
      outwires->emplace_back(
        recob::Wire(rois, rawlist[ich]->Channel(), geo->View(rawlist[ich]->Channel())));
    }
  }

  e.put(std::move(outwires));
}

DEFINE_ART_MODULE(nnet::WaveformRoiFinder)