wavrec_tool::IWaveformRecog Class Reference

#include <IWaveformRecog.h>

Inheritance diagram for wavrec_tool::IWaveformRecog:

Public Member Functions
virtual	~IWaveformRecog () noexcept=default
virtual std::vector< std::vector< float > >	predictWaveformType (const std::vector< std::vector< float >> &) const =0
std::vector< bool >	findROI (const std::vector< float > &adcin) const
std::vector< float >	predROI (const std::vector< float > &adcin) const

Protected Member Functions
std::string	findFile (const char *fileName) const
void	setupWaveRecRoiParams (const fhicl::ParameterSet &pset)

Private Member Functions
std::vector< std::vector< float > >	scanWaveform (const std::vector< float > &adcin) const

Private Attributes
std::vector< float >	scalevec
std::vector< float >	meanvec
float	fCnnMean
float	fCnnScale
float	fCnnPredCut
unsigned int	fWaveformSize
unsigned int	fWindowSize
unsigned int	fStrideLength
unsigned int	fNumStrides
unsigned int	fLastWindowSize

Detailed Description

Definition at line 9 of file IWaveformRecog.h.

Constructor & Destructor Documentation

virtual wavrec_tool::IWaveformRecog::~IWaveformRecog ( )

virtualdefaultnoexcept

Member Function Documentation

std::string wavrec_tool::IWaveformRecog::findFile ( const char *  fileName ) const

inlineprotected

Definition at line 71 of file IWaveformRecog.h.

    {
      std::string fname_out;
      cet::search_path sp("FW_SEARCH_PATH");
      if (!sp.find_file(fileName, fname_out)) {
        struct stat buffer;
        if (stat(fileName, &buffer) == 0) { fname_out = fileName; }
        else {
          throw art::Exception(art::errors::NotFound)
            << "Could not find the model file " << fileName;
        }
      }
      return fname_out;
    }

std::vector<bool> wavrec_tool::IWaveformRecog::findROI ( const std::vector< float > &  adcin ) const

inline

Definition at line 23 of file IWaveformRecog.h.

    {
      std::vector<bool> bvec(fWaveformSize, false);
      if (adcin.size() != fWaveformSize) { return bvec; }

      std::vector<std::vector<float>> predv = scanWaveform(adcin);

      // .. set to true all bins in the output vector that are in windows identified as signals
      int j1;
      for (unsigned int i = 0; i < fNumStrides; i++) {
        j1 = i * fStrideLength;
        if (predv[i][0] > fCnnPredCut) { std::fill_n(bvec.begin() + j1, fWindowSize, true); }
      }
      // .. last window is a special case
      if (predv[fNumStrides][0] > fCnnPredCut) {
        j1 = fNumStrides * fStrideLength;
        std::fill_n(bvec.begin() + j1, fLastWindowSize, true);
      }
      return bvec;
    }

virtual std::vector<std::vector<float> > wavrec_tool::IWaveformRecog::predictWaveformType ( const std::vector< std::vector< float >> &  ) const

pure virtual

Implemented in wavrec_tool::WaveformRecogTf.

std::vector<float> wavrec_tool::IWaveformRecog::predROI ( const std::vector< float > &  adcin ) const

inline

Definition at line 50 of file IWaveformRecog.h.

    {
      std::vector<float> fvec(fWaveformSize, 0.);
      if (adcin.size() != fWaveformSize) { return fvec; }

      std::vector<std::vector<float>> predv = scanWaveform(adcin);

      // .. set value in each bin of output vector to the prediction for the window it is in
      int j1;
      for (unsigned int i = 0; i < fNumStrides; i++) {
        j1 = i * fStrideLength;
        std::fill_n(fvec.begin() + j1, fWindowSize, predv[i][0]);
      }
      // .. last window is a special case
      j1 = fNumStrides * fStrideLength;
      std::fill_n(fvec.begin() + j1, fLastWindowSize, predv[fNumStrides][0]);
      return fvec;
    }

std::vector<std::vector<float> > wavrec_tool::IWaveformRecog::scanWaveform ( const std::vector< float > &  adcin ) const

inlineprivate

Definition at line 167 of file IWaveformRecog.h.

    {
      // .. rescale input waveform for CNN
      std::vector<float> adc(fWaveformSize);
      for (size_t itck = 0; itck < fWaveformSize; ++itck) {
        adc[itck] = (adcin[itck] - meanvec[itck]) / scalevec[itck];
      }

      // .. create a vector of windows
      std::vector<std::vector<float>> wwv(fNumStrides + 1, std::vector<float>(fWindowSize, 0.));

      // .. fill each window with adc values
      unsigned int j1, j2, k;
      for (unsigned int i = 0; i < fNumStrides; i++) {
        j1 = i * fStrideLength;
        j2 = j1 + fWindowSize;
        k = 0;
        for (unsigned int j = j1; j < j2; j++) {
          wwv[i][k] = adc[j];
          k++;
        }
      }
      // .. last window is a special case
      j1 = fNumStrides * fStrideLength;
      j2 = j1 + fLastWindowSize;
      k = 0;
      for (unsigned int j = j1; j < j2; j++) {
        wwv[fNumStrides][k] = adc[j];
        k++;
      }

      // ... use waveform recognition CNN to perform inference on each window
      return predictWaveformType(wwv);
    }

void wavrec_tool::IWaveformRecog::setupWaveRecRoiParams ( const fhicl::ParameterSet &  pset )

inlineprotected

Definition at line 87 of file IWaveformRecog.h.

    {
      fCnnPredCut = pset.get<float>("CnnPredCut", 0.5);
      fWaveformSize = pset.get<unsigned int>("WaveformSize", 0); // 6000
      std::string fMeanFilename = pset.get<std::string>("MeanFilename", "");
      std::string fScaleFilename = pset.get<std::string>("ScaleFilename", "");

      // ... load the mean and scale (std) vectors
      if (!fMeanFilename.empty() && !fScaleFilename.empty()) {
        float val;
        std::ifstream meanfile(findFile(fMeanFilename.c_str()).c_str());
        if (meanfile.is_open()) {
          while (meanfile >> val)
            meanvec.push_back(val);
          meanfile.close();
          if (meanvec.size() != fWaveformSize) {
            throw cet::exception("WaveformRecogTf_tool")
              << "vector of mean values does not match waveform size, exiting" << std::endl;
          }
        }
        else {
          throw cet::exception("WaveformRecogTf_tool")
            << "failed opening StdScaler mean file, exiting" << std::endl;
        }
        std::ifstream scalefile(findFile(fScaleFilename.c_str()).c_str());
        if (scalefile.is_open()) {
          while (scalefile >> val)
            scalevec.push_back(val);
          scalefile.close();
          if (scalevec.size() != fWaveformSize) {
            throw cet::exception("WaveformRecogTf_tool")
              << "vector of scale values does not match waveform size, exiting" << std::endl;
          }
        }
        else {
          throw cet::exception("WaveformRecogTf_tool")
            << "failed opening StdScaler scale file, exiting" << std::endl;
        }
      }
      else {
        fCnnMean = pset.get<float>("CnnMean", 0.);
        fCnnScale = pset.get<float>("CnnScale", 1.);
        meanvec.resize(fWaveformSize);
        std::fill(meanvec.begin(), meanvec.end(), fCnnMean);
        scalevec.resize(fWaveformSize);
        std::fill(scalevec.begin(), scalevec.end(), fCnnScale);
      }

      fWindowSize = pset.get<unsigned int>("ScanWindowSize", 0); // 200
      fStrideLength = pset.get<unsigned int>("StrideLength", 0); // 150

      if (fWaveformSize > 0 && fWindowSize > 0) {
        float dmn =
          fWaveformSize - fWindowSize; // dist between trail edge of 1st win & last data point
        fNumStrides = std::ceil(dmn / float(fStrideLength)); // # strides to scan entire waveform
        unsigned int overshoot = fNumStrides * fStrideLength + fWindowSize - fWaveformSize;
        fLastWindowSize = fWindowSize - overshoot;
        unsigned int numwindows = fNumStrides + 1;
        std::cout << " !!!!! WaveformRoiFinder: WindowSize = " << fWindowSize
                  << ", StrideLength = " << fStrideLength
                  << ", dmn/StrideLength = " << dmn / fStrideLength << std::endl;
        std::cout << "   dmn = " << dmn << ", NumStrides = " << fNumStrides
                  << ", overshoot = " << overshoot << ", LastWindowSize = " << fLastWindowSize
                  << ", numwindows = " << numwindows << std::endl;
      }
    }

Member Data Documentation

float wavrec_tool::IWaveformRecog::fCnnMean

private

Definition at line 157 of file IWaveformRecog.h.

float wavrec_tool::IWaveformRecog::fCnnPredCut

private

Definition at line 159 of file IWaveformRecog.h.

float wavrec_tool::IWaveformRecog::fCnnScale

private

Definition at line 158 of file IWaveformRecog.h.

unsigned int wavrec_tool::IWaveformRecog::fLastWindowSize

private

Definition at line 164 of file IWaveformRecog.h.

unsigned int wavrec_tool::IWaveformRecog::fNumStrides

private

Definition at line 163 of file IWaveformRecog.h.

unsigned int wavrec_tool::IWaveformRecog::fStrideLength

private

Definition at line 162 of file IWaveformRecog.h.

unsigned int wavrec_tool::IWaveformRecog::fWaveformSize

private

Definition at line 160 of file IWaveformRecog.h.

unsigned int wavrec_tool::IWaveformRecog::fWindowSize

private

Definition at line 161 of file IWaveformRecog.h.

std::vector<float> wavrec_tool::IWaveformRecog::meanvec

private

Definition at line 156 of file IWaveformRecog.h.

std::vector<float> wavrec_tool::IWaveformRecog::scalevec

private

Definition at line 155 of file IWaveformRecog.h.

---

The documentation for this class was generated from the following file:

• larrecodnn/larrecodnn/ImagePatternAlgs/ToolInterfaces/IWaveformRecog.h