nnet::PointIdAlg Class Reference

#include "PointIdAlg.h"

Inheritance diagram for nnet::PointIdAlg:

Classes
struct	Config

Public Types
enum	EDownscaleMode { kMax = 1, kMaxMean = 2, kMean = 3 }

Public Member Functions
	PointIdAlg (const fhicl::ParameterSet &pset)
	PointIdAlg (const Config &config)
	~PointIdAlg () override
std::vector< std::string > const &	outputLabels () const
	network output labels More...
float	predictIdValue (unsigned int wire, float drift, size_t outIdx=0) const
	calculate single-value prediction (2-class probability) for [wire, drift] point More...
std::vector< float >	predictIdVector (unsigned int wire, float drift) const
	calculate multi-class probabilities for [wire, drift] point More...
std::vector< std::vector< float > >	predictIdVectors (std::vector< std::pair< unsigned int, float >> points) const
std::vector< std::vector< float > > const &	patchData2D () const
std::vector< float >	patchData1D () const
bool	isInsideFiducialRegion (unsigned int wire, float drift) const
bool	isCurrentPatch (unsigned int wire, float drift) const
bool	isSamePatch (unsigned int wire1, float drift1, unsigned int wire2, float drift2) const
	test if two wire/drift coordinates point to the same patch More...
bool	setWireDriftData (const detinfo::DetectorClocksData &clock_data, const detinfo::DetectorPropertiesData &det_prop, const std::vector< recob::Wire > &wires, unsigned int plane, unsigned int tpc, unsigned int cryo)
std::vector< float > const &	wireData (size_t widx) const
std::vector< std::vector< float > >	getPatch (size_t wire, float drift, size_t patchSizeW, size_t patchSizeD) const
float	getPixelOrZero (int wire, int drift) const
double	getAdcSum () const
size_t	getAdcArea () const
float	poolMax (int wire, int drift, size_t r=0) const
	Pool max value in a patch around the wire/drift pixel. More...
unsigned int	Cryo () const
	Pool sum of pixels in a patch around the wire/drift pixel. More...
unsigned int	TPC () const
unsigned int	Plane () const
unsigned int	NWires () const
unsigned int	NScaledDrifts () const
unsigned int	NCachedDrifts () const
unsigned int	DriftWindow () const
float	ZeroLevel () const
	Level of zero ADC after scaling. More...
double	LifetimeCorrection (detinfo::DetectorClocksData const &clock_data, detinfo::DetectorPropertiesData const &det_prop, double tick) const

Static Public Member Functions
static std::vector< float >	flattenData2D (std::vector< std::vector< float >> const &patch)

Protected Member Functions
std::vector< float >	downscaleMax (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const
std::vector< float >	downscaleMaxMean (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const
std::vector< float >	downscaleMean (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const
std::vector< float >	downscale (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const
size_t	getDriftIndex (float drift) const
std::optional< std::vector< float > >	setWireData (std::vector< float > const &adc, size_t wireIdx) const
bool	patchFromDownsampledView (size_t wire, float drift, size_t size_w, size_t size_d, std::vector< std::vector< float >> &patch) const
bool	patchFromOriginalView (size_t wire, float drift, size_t size_w, size_t size_d, std::vector< std::vector< float >> &patch) const
virtual DataProviderAlgView	resizeView (detinfo::DetectorClocksData const &clock_data, detinfo::DetectorPropertiesData const &det_prop, size_t wires, size_t drifts)

Protected Attributes
DataProviderAlgView	fAlgView
EDownscaleMode	fDownscaleMode
size_t	fDriftWindow
bool	fDownscaleFullView
float	fDriftWindowInv
calo::CalorimetryAlg	fCalorimetryAlg
geo::GeometryCore const *	fGeometry

Private Member Functions
bool	bufferPatch (size_t wire, float drift, std::vector< std::vector< float >> &patch) const
bool	bufferPatch (size_t wire, float drift) const
void	resizePatch ()
void	deleteNNet ()

Private Attributes
std::string	fNNetModelFilePath
std::vector< std::string >	fNNetOutputs
nnet::ModelInterface *	fNNet
std::vector< std::vector< float > >	fWireDriftPatch
size_t	fPatchSizeW
size_t	fPatchSizeD
size_t	fCurrentWireIdx
size_t	fCurrentScaledDrift

Detailed Description

Definition at line 98 of file PointIdAlg.h.

Member Enumeration Documentation

enum img::DataProviderAlg::EDownscaleMode

inherited

Enumerator
kMax
kMaxMean
kMean

Definition at line 64 of file DataProviderAlg.h.

{ kMax = 1, kMaxMean = 2, kMean = 3 };

Constructor & Destructor Documentation

nnet::PointIdAlg::PointIdAlg ( const fhicl::ParameterSet &  pset )

inline

Definition at line 114 of file PointIdAlg.h.

: PointIdAlg(fhicl::Table<Config>(pset, {})()) {}

nnet::PointIdAlg::PointIdAlg

(

const Config &

config

)

Definition at line 173 of file PointIdAlg.cxx.

References deleteNNet(), fNNet, fNNetModelFilePath, fNNetOutputs, nnet::PointIdAlg::Config::NNetModelFile, nnet::PointIdAlg::Config::NNetOutputs, and resizePatch().

  : img::DataProviderAlg(config)
  , fNNet(0)
  , fPatchSizeW(config.PatchSizeW())
  , fPatchSizeD(config.PatchSizeD())
  , fCurrentWireIdx(99999)
  , fCurrentScaledDrift(99999)
{
  fNNetModelFilePath = config.NNetModelFile();
  fNNetOutputs = config.NNetOutputs();

  deleteNNet();

  if ((fNNetModelFilePath.length() > 5) &&
      (fNNetModelFilePath.compare(fNNetModelFilePath.length() - 5, 5, ".nnet") == 0)) {
    fNNet = new nnet::KerasModelInterface(fNNetModelFilePath.c_str());
  }
  else if ((fNNetModelFilePath.length() > 3) &&
           (fNNetModelFilePath.compare(fNNetModelFilePath.length() - 3, 3, ".pb") == 0)) {
    fNNet = new nnet::TfModelInterface(fNNetModelFilePath.c_str());
  }
  else {
    mf::LogError("PointIdAlg") << "File name extension not supported.";
  }

  if (!fNNet) { throw cet::exception("nnet::PointIdAlg") << "Loading model from file failed."; }

  resizePatch();
}

nnet::PointIdAlg::~PointIdAlg ( )

override

Definition at line 204 of file PointIdAlg.cxx.

References deleteNNet().

{
  deleteNNet();
}

Member Function Documentation

bool nnet::PointIdAlg::bufferPatch ( size_t  wire, float  drift, std::vector< std::vector< float >> &  patch  ) const

inlineprivate

Definition at line 158 of file PointIdAlg.h.

Referenced by predictIdValue(), predictIdVector(), and predictIdVectors().

  {
    if (fDownscaleFullView) {
      size_t sd = (size_t)(drift / fDriftWindow);
      if ((fCurrentWireIdx == wire) && (fCurrentScaledDrift == sd))
        return true; // still within the current position

      fCurrentWireIdx = wire;
      fCurrentScaledDrift = sd;

      return patchFromDownsampledView(wire, drift, fPatchSizeW, fPatchSizeD, patch);
    }
    else {
      if ((fCurrentWireIdx == wire) && (fCurrentScaledDrift == drift))
        return true; // still within the current position

      fCurrentWireIdx = wire;
      fCurrentScaledDrift = drift;

      return patchFromOriginalView(wire, drift, fPatchSizeW, fPatchSizeD, patch);
    }
  }

bool nnet::PointIdAlg::bufferPatch ( size_t  wire, float  drift  ) const

inlineprivate

Definition at line 180 of file PointIdAlg.h.

  {
    return bufferPatch(wire, drift, fWireDriftPatch);
  }

unsigned int img::DataProviderAlg::Cryo ( ) const

inlineinherited

Pool sum of pixels in a patch around the wire/drift pixel.

Definition at line 161 of file DataProviderAlg.h.

Referenced by nnet::TrainingDataAlg::collectVtxFlags().

{ return fCryo; }

void nnet::PointIdAlg::deleteNNet ( )

inlineprivate

Definition at line 186 of file PointIdAlg.h.

Referenced by PointIdAlg(), and ~PointIdAlg().

  {
    if (fNNet) delete fNNet;
    fNNet = 0;
  }

std::vector<float> img::DataProviderAlg::downscale ( std::size_t  dst_size, std::vector< float > const &  adc, size_t  tick0  ) const

inlineprotectedinherited

Definition at line 198 of file DataProviderAlg.h.

References img::DataProviderAlg::kMax, img::DataProviderAlg::kMaxMean, and img::DataProviderAlg::kMean.

Referenced by img::DataProviderAlg::patchFromOriginalView(), and img::DataProviderAlg::setWireData().

  {
    switch (fDownscaleMode) {
    case img::DataProviderAlg::kMean: return downscaleMean(dst_size, adc, tick0);
    case img::DataProviderAlg::kMaxMean: return downscaleMaxMean(dst_size, adc, tick0);
    case img::DataProviderAlg::kMax: return downscaleMax(dst_size, adc, tick0);
    }
    throw cet::exception("img::DataProviderAlg") << "Downscale mode not supported." << std::endl;
  }

std::vector< float > img::DataProviderAlg::downscaleMax ( std::size_t  dst_size, std::vector< float > const &  adc, size_t  tick0  ) const

protectedinherited

Definition at line 196 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlgView::fLifetimeCorrFactors, and img::DataProviderAlg::scaleAdcSamples().

{
  size_t kStop = dst_size;
  std::vector<float> result(dst_size);
  if (adc.size() < kStop) { kStop = adc.size(); }
  for (size_t i = 0, k0 = 0; i < kStop; ++i, k0 += fDriftWindow) {
    size_t k1 = k0 + fDriftWindow;

    float max_adc = adc[k0] * fAlgView.fLifetimeCorrFactors[k0 + tick0];
    for (size_t k = k0 + 1; k < k1; ++k) {
      float ak = adc[k] * fAlgView.fLifetimeCorrFactors[k + tick0];
      if (ak > max_adc) max_adc = ak;
    }
    result[i] = max_adc;
  }
  scaleAdcSamples(result);
  return result;
}

std::vector< float > img::DataProviderAlg::downscaleMaxMean ( std::size_t  dst_size, std::vector< float > const &  adc, size_t  tick0  ) const

protectedinherited

Definition at line 217 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlgView::fLifetimeCorrFactors, n, and img::DataProviderAlg::scaleAdcSamples().

{
  size_t kStop = dst_size;
  std::vector<float> result(dst_size);
  if (adc.size() < kStop) { kStop = adc.size(); }
  for (size_t i = 0, k0 = 0; i < kStop; ++i, k0 += fDriftWindow) {
    size_t k1 = k0 + fDriftWindow;
    size_t max_idx = k0;
    float max_adc = adc[k0] * fAlgView.fLifetimeCorrFactors[k0 + tick0];
    for (size_t k = k0 + 1; k < k1; ++k) {
      float ak = adc[k] * fAlgView.fLifetimeCorrFactors[k + tick0];
      if (ak > max_adc) {
        max_adc = ak;
        max_idx = k;
      }
    }

    size_t n = 1;
    if (max_idx > 0) {
      max_adc += adc[max_idx - 1] * fAlgView.fLifetimeCorrFactors[max_idx - 1 + tick0];
      n++;
    }
    if (max_idx + 1 < adc.size()) {
      max_adc += adc[max_idx + 1] * fAlgView.fLifetimeCorrFactors[max_idx + 1 + tick0];
      n++;
    }

    result[i] = max_adc / n;
  }
  scaleAdcSamples(result);
  return result;
}

std::vector< float > img::DataProviderAlg::downscaleMean ( std::size_t  dst_size, std::vector< float > const &  adc, size_t  tick0  ) const

protectedinherited

Definition at line 252 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fDriftWindowInv, img::DataProviderAlgView::fLifetimeCorrFactors, and img::DataProviderAlg::scaleAdcSamples().

{
  size_t kStop = dst_size;
  std::vector<float> result(dst_size);
  if (adc.size() < kStop) { kStop = adc.size(); }
  for (size_t i = 0, k0 = 0; i < kStop; ++i, k0 += fDriftWindow) {
    size_t k1 = k0 + fDriftWindow;

    float sum_adc = 0;
    for (size_t k = k0; k < k1; ++k) {
      if (k + tick0 < fAlgView.fLifetimeCorrFactors.size())
        sum_adc += adc[k] * fAlgView.fLifetimeCorrFactors[k + tick0];
    }
    result[i] = sum_adc * fDriftWindowInv;
  }
  scaleAdcSamples(result);
  return result;
}

unsigned int img::DataProviderAlg::DriftWindow ( ) const

inlineinherited

Definition at line 168 of file DataProviderAlg.h.

{ return fDriftWindow; }

std::vector< float > nnet::PointIdAlg::flattenData2D ( std::vector< std::vector< float >> const &  patch )

static

Definition at line 309 of file PointIdAlg.cxx.

References d, and w.

{
  std::vector<float> flat;
  if (patch.empty() || patch.front().empty()) {
    mf::LogError("DataProviderAlg") << "Patch is empty.";
    return flat;
  }

  flat.resize(patch.size() * patch.front().size());

  for (size_t w = 0, i = 0; w < patch.size(); ++w) {
    auto const& wire = patch[w];
    for (size_t d = 0; d < wire.size(); ++d, ++i) {
      flat[i] = wire[d];
    }
  }

  return flat;
}

size_t img::DataProviderAlg::getAdcArea ( ) const

inlineinherited

Definition at line 153 of file DataProviderAlg.h.

References r.

{ return fAdcAreaOverThr; }

double img::DataProviderAlg::getAdcSum ( ) const

inlineinherited

Definition at line 152 of file DataProviderAlg.h.

{ return fAdcSumOverThr; }

size_t img::DataProviderAlg::getDriftIndex ( float  drift ) const

inlineprotectedinherited

Definition at line 210 of file DataProviderAlg.h.

References lar::dump::vector().

Referenced by img::DataProviderAlg::poolMax().

  {
    if (fDownscaleFullView)
      return (size_t)(drift * fDriftWindowInv);
    else
      return (size_t)drift;
  }

std::vector<std::vector<float> > img::DataProviderAlg::getPatch ( size_t  wire, float  drift, size_t  patchSizeW, size_t  patchSizeD  ) const

inlineinherited

Return patch of data centered on the wire and drift, witht the size in (downscaled) pixels givent with patchSizeW and patchSizeD. Pad with the zero-level calue if patch extends beyond the event projection.

Definition at line 122 of file DataProviderAlg.h.

  {
    bool ok = false;
    std::vector<std::vector<float>> patch;
    if (fDownscaleFullView) {
      ok = patchFromDownsampledView(wire, drift, patchSizeW, patchSizeD, patch);
    }
    else {
      ok = patchFromOriginalView(wire, drift, patchSizeW, patchSizeD, patch);
    }

    if (ok) return patch;
    throw cet::exception("img::DataProviderAlg") << "Patch filling failed." << std::endl;
  }

float img::DataProviderAlg::getPixelOrZero ( int  wire, int  drift  ) const

inlineinherited

Return value from the ADC buffer, or zero if coordinates are out of the view; will scale the drift according to the downscale settings.

Definition at line 142 of file DataProviderAlg.h.

  {
    size_t didx = getDriftIndex(drift), widx = (size_t)wire;

    if ((widx < fAlgView.fWireDriftData.size()) && (didx < fAlgView.fNCachedDrifts)) {
      return fAlgView.fWireDriftData[widx][didx];
    }
    return 0;
  }

bool nnet::PointIdAlg::isCurrentPatch	(	unsigned int	wire,
		float	drift
	)		const

test if wire/drift coordinates point to the current patch (so maybe the cnn output does not need to be recalculated)

Definition at line 293 of file PointIdAlg.cxx.

References fCurrentScaledDrift, fCurrentWireIdx, img::DataProviderAlg::fDownscaleFullView, and img::DataProviderAlg::fDriftWindow.

{
  if (fDownscaleFullView) {
    size_t sd = (size_t)(drift / fDriftWindow);
    if ((fCurrentWireIdx == wire) && (fCurrentScaledDrift == sd))
      return true; // still within the current position
  }
  else {
    if ((fCurrentWireIdx == wire) && (fCurrentScaledDrift == drift))
      return true; // still within the current position
  }

  return false; // not a current position
}

bool nnet::PointIdAlg::isInsideFiducialRegion	(	unsigned int	wire,
		float	drift
	)		const

Definition at line 330 of file PointIdAlg.cxx.

References img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlgView::fNScaledDrifts, img::DataProviderAlgView::fNWires, fPatchSizeD, and fPatchSizeW.

Referenced by nnet::EmTrackClusterId2out::produce(), nnet::EmTrackClusterId::produce(), and nnet::EmTrackMichelId::produce().

{
  size_t marginW = fPatchSizeW / 8; // fPatchSizeX/2 will make patch always completely filled
  size_t marginD = fPatchSizeD / 8;

  size_t scaledDrift = (size_t)(drift / fDriftWindow);
  if ((wire >= marginW) && (wire < fAlgView.fNWires - marginW) && (scaledDrift >= marginD) &&
      (scaledDrift < fAlgView.fNScaledDrifts - marginD))
    return true;
  else
    return false;
}

bool nnet::PointIdAlg::isSamePatch	(	unsigned int	wire1,
		float	drift1,
		unsigned int	wire2,
		float	drift2
	)		const

test if two wire/drift coordinates point to the same patch

Definition at line 276 of file PointIdAlg.cxx.

References img::DataProviderAlg::fDownscaleFullView, and img::DataProviderAlg::fDriftWindow.

{
  if (fDownscaleFullView) {
    size_t sd1 = (size_t)(drift1 / fDriftWindow);
    size_t sd2 = (size_t)(drift2 / fDriftWindow);
    if ((wire1 == wire2) && (sd1 == sd2)) return true; // the same position
  }
  else {
    if ((wire1 == wire2) && ((size_t)drift1 == (size_t)drift2)) return true; // the same position
  }

  return false; // not the same position
}

double img::DataProviderAlg::LifetimeCorrection ( detinfo::DetectorClocksData const &  clock_data, detinfo::DetectorPropertiesData const &  det_prop, double  tick  ) const

inlineinherited

Definition at line 173 of file DataProviderAlg.h.

  {
    return fCalorimetryAlg.LifetimeCorrection(clock_data, det_prop, tick);
  }

unsigned int img::DataProviderAlg::NCachedDrifts ( ) const

inlineinherited

Definition at line 167 of file DataProviderAlg.h.

{ return fAlgView.fNCachedDrifts; }

unsigned int img::DataProviderAlg::NScaledDrifts ( ) const

inlineinherited

Definition at line 166 of file DataProviderAlg.h.

Referenced by nnet::PointIdTrainingData::analyze().

{ return fAlgView.fNScaledDrifts; }

unsigned int img::DataProviderAlg::NWires ( ) const

inlineinherited

Definition at line 165 of file DataProviderAlg.h.

Referenced by nnet::PointIdTrainingData::analyze().

{ return fAlgView.fNWires; }

std::vector<std::string> const& nnet::PointIdAlg::outputLabels ( ) const

inline

network output labels

Definition at line 121 of file PointIdAlg.h.

References lar::dump::vector().

Referenced by nnet::EmTrackClusterId2out::produce(), nnet::EmTrackClusterId::produce(), and nnet::EmTrackMichelId::produce().

{ return fNNetOutputs; }

std::vector<float> nnet::PointIdAlg::patchData1D ( ) const

inline

Definition at line 135 of file PointIdAlg.h.

  {
    return flattenData2D(fWireDriftPatch);
  } // flat vector made of the patch data, wire after wire

std::vector<std::vector<float> > const& nnet::PointIdAlg::patchData2D ( ) const

inline

Definition at line 134 of file PointIdAlg.h.

{ return fWireDriftPatch; }

bool img::DataProviderAlg::patchFromDownsampledView ( size_t  wire, float  drift, size_t  size_w, size_t  size_d, std::vector< std::vector< float >> &  patch  ) const

protectedinherited

Definition at line 439 of file DataProviderAlg.cxx.

References d, img::DataProviderAlg::fAdcZero, img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDriftWindow, trkf::fill(), img::DataProviderAlgView::fWireDriftData, and w.

{
  int halfSizeW = size_w / 2;
  int halfSizeD = size_d / 2;

  int w0 = wire - halfSizeW;
  int w1 = wire + halfSizeW;

  size_t sd = (size_t)(drift / fDriftWindow);
  int d0 = sd - halfSizeD;
  int d1 = sd + halfSizeD;

  int wsize = fAlgView.fWireDriftData.size();
  for (int w = w0, wpatch = 0; w < w1; ++w, ++wpatch) {
    auto& dst = patch[wpatch];
    if ((w >= 0) && (w < wsize)) {
      auto& src = fAlgView.fWireDriftData[w];
      int dsize = src.size();
      for (int d = d0, dpatch = 0; d < d1; ++d, ++dpatch) {
        if ((d >= 0) && (d < dsize)) { dst[dpatch] = src[d]; }
        else {
          dst[dpatch] = fAdcZero;
        }
      }
    }
    else {
      std::fill(dst.begin(), dst.end(), fAdcZero);
    }
  }

  return true;
}

bool img::DataProviderAlg::patchFromOriginalView ( size_t  wire, float  drift, size_t  size_w, size_t  size_d, std::vector< std::vector< float >> &  patch  ) const

protectedinherited

Definition at line 476 of file DataProviderAlg.cxx.

References d, img::DataProviderAlg::downscale(), img::DataProviderAlg::fAdcZero, img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDriftWindow, trkf::fill(), img::DataProviderAlgView::fWireDriftData, util::size(), tmp, and w.

{
  int dsize = fDriftWindow * size_d;
  int halfSizeW = size_w / 2;
  int halfSizeD = dsize / 2;

  int w0 = wire - halfSizeW;
  int w1 = wire + halfSizeW;

  int d0 = int(drift) - halfSizeD;
  int d1 = int(drift) + halfSizeD;

  if (d0 < 0) d0 = 0;

  std::vector<float> tmp(dsize);
  int wsize = fAlgView.fWireDriftData.size();
  for (int w = w0, wpatch = 0; w < w1; ++w, ++wpatch) {
    if ((w >= 0) && (w < wsize)) {
      auto& src = fAlgView.fWireDriftData[w];
      int src_size = src.size();
      for (int d = d0, dpatch = 0; d < d1; ++d, ++dpatch) {
        if ((d >= 0) && (d < src_size)) { tmp[dpatch] = src[d]; }
        else {
          tmp[dpatch] = fAdcZero;
        }
      }
    }
    else {
      std::fill(tmp.begin(), tmp.end(), fAdcZero);
    }
    patch[wpatch] = downscale(patch[wpatch].size(), tmp, d0);
  }

  return true;
}

unsigned int img::DataProviderAlg::Plane ( ) const

inlineinherited

Definition at line 163 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::setWireDriftData(), pma::ProjectionMatchingAlg::validate_on_adc(), and pma::ProjectionMatchingAlg::validate_on_adc_test().

{ return fPlane; }

float img::DataProviderAlg::poolMax ( int  wire, int  drift, size_t  r = 0  ) const

inherited

Pool max value in a patch around the wire/drift pixel.

Definition at line 145 of file DataProviderAlg.cxx.

References col, d, img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlgView::fNCachedDrifts, img::DataProviderAlgView::fNWires, img::DataProviderAlgView::fWireDriftData, img::DataProviderAlg::getDriftIndex(), r, and w.

Referenced by pma::ProjectionMatchingAlg::validate_on_adc(), and pma::ProjectionMatchingAlg::validate_on_adc_test().

{
  size_t rw = r, rd = r;
  if (!fDownscaleFullView) { rd *= fDriftWindow; }

  size_t didx = getDriftIndex(drift);
  int d0 = didx - rd;
  if (d0 < 0) { d0 = 0; }
  int d1 = didx + rd;
  if (d1 >= (int)fAlgView.fNCachedDrifts) { d1 = fAlgView.fNCachedDrifts - 1; }

  int w0 = wire - rw;
  if (w0 < 0) { w0 = 0; }
  int w1 = wire + rw;
  if (w1 >= (int)fAlgView.fNWires) { w1 = fAlgView.fNWires - 1; }

  float adc, max_adc = 0;
  for (int w = w0; w <= w1; ++w) {
    auto const* col = fAlgView.fWireDriftData[w].data();
    for (int d = d0; d <= d1; ++d) {
      adc = col[d];
      if (adc > max_adc) { max_adc = adc; }
    }
  }

  return max_adc;
}

float nnet::PointIdAlg::predictIdValue	(	unsigned int	wire,
		float	drift,
		size_t	outIdx = 0
	)		const

calculate single-value prediction (2-class probability) for [wire, drift] point

Definition at line 218 of file PointIdAlg.cxx.

References bufferPatch(), fNNet, fWireDriftPatch, and nnet::ModelInterface::Run().

{
  float result = 0.;

  if (!bufferPatch(wire, drift)) {
    mf::LogError("PointIdAlg") << "Patch buffering failed.";
    return result;
  }

  if (fNNet) {
    auto out = fNNet->Run(fWireDriftPatch);
    if (!out.empty()) { result = out[outIdx]; }
    else {
      mf::LogError("PointIdAlg") << "Problem with applying model to input.";
    }
  }

  return result;
}

std::vector< float > nnet::PointIdAlg::predictIdVector	(	unsigned int	wire,
		float	drift
	)		const

calculate multi-class probabilities for [wire, drift] point

Definition at line 239 of file PointIdAlg.cxx.

References bufferPatch(), fNNet, fWireDriftPatch, and nnet::ModelInterface::Run().

Referenced by nnet::ParticleDecayId::DetectDecay().

{
  std::vector<float> result;

  if (!bufferPatch(wire, drift)) {
    mf::LogError("PointIdAlg") << "Patch buffering failed.";
    return result;
  }

  if (fNNet) {
    result = fNNet->Run(fWireDriftPatch);
    if (result.empty()) { mf::LogError("PointIdAlg") << "Problem with applying model to input."; }
  }

  return result;
}

std::vector< std::vector< float > > nnet::PointIdAlg::predictIdVectors

(

std::vector< std::pair< unsigned int, float >>

points

)

const

Definition at line 257 of file PointIdAlg.cxx.

References bufferPatch(), fNNet, fPatchSizeD, fPatchSizeW, and nnet::ModelInterface::Run().

Referenced by nnet::EmTrackClusterId2out::produce(), nnet::EmTrackClusterId::produce(), and nnet::EmTrackMichelId::produce().

{
  if (points.empty() || !fNNet) { return std::vector<std::vector<float>>(); }

  std::vector<std::vector<std::vector<float>>> inps(
    points.size(), std::vector<std::vector<float>>(fPatchSizeW, std::vector<float>(fPatchSizeD)));
  for (size_t i = 0; i < points.size(); ++i) {
    unsigned int wire = points[i].first;
    float drift = points[i].second;
    if (!bufferPatch(wire, drift, inps[i])) {
      throw cet::exception("PointIdAlg") << "Patch buffering failed" << std::endl;
    }
  }

  return fNNet->Run(inps);
}

void nnet::PointIdAlg::resizePatch ( )

private

Definition at line 210 of file PointIdAlg.cxx.

References fPatchSizeD, fPatchSizeW, fWireDriftPatch, and r.

Referenced by PointIdAlg().

{
  fWireDriftPatch.resize(fPatchSizeW);
  for (auto& r : fWireDriftPatch)
    r.resize(fPatchSizeD);
}

img::DataProviderAlgView img::DataProviderAlg::resizeView ( detinfo::DetectorClocksData const &  clock_data, detinfo::DetectorPropertiesData const &  det_prop, size_t  wires, size_t  drifts  )

protectedvirtualinherited

Reimplemented in nnet::TrainingDataAlg.

Definition at line 114 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fAdcZero, img::DataProviderAlg::fCalibrateLifetime, img::DataProviderAlg::fCalorimetryAlg, img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlgView::fLifetimeCorrFactors, img::DataProviderAlgView::fNCachedDrifts, img::DataProviderAlgView::fNDrifts, img::DataProviderAlgView::fNScaledDrifts, img::DataProviderAlgView::fNWires, img::DataProviderAlgView::fWireChannels, img::DataProviderAlgView::fWireDriftData, raw::InvalidChannelID, and calo::CalorimetryAlg::LifetimeCorrection().

Referenced by nnet::TrainingDataAlg::resizeView(), and img::DataProviderAlg::setWireDriftData().

{
  img::DataProviderAlgView result;
  result.fNWires = wires;
  result.fNDrifts = drifts;
  result.fNScaledDrifts = drifts / fDriftWindow;
  result.fNCachedDrifts = fDownscaleFullView ? result.fNScaledDrifts : drifts;

  result.fWireChannels.resize(wires, raw::InvalidChannelID);

  result.fWireDriftData.resize(wires, std::vector<float>(result.fNCachedDrifts, fAdcZero));

  result.fLifetimeCorrFactors.resize(drifts);
  if (fCalibrateLifetime) {
    for (size_t t = 0; t < drifts; ++t) {
      result.fLifetimeCorrFactors[t] = fCalorimetryAlg.LifetimeCorrection(clock_data, det_prop, t);
    }
  }
  else {
    for (size_t t = 0; t < drifts; ++t) {
      result.fLifetimeCorrFactors[t] = 1.0;
    }
  }
  return result;
}

std::optional< std::vector< float > > img::DataProviderAlg::setWireData ( std::vector< float > const &  adc, size_t  wireIdx  ) const

protectedinherited

Definition at line 273 of file DataProviderAlg.cxx.

References img::DataProviderAlg::downscale(), img::DataProviderAlg::fAlgView, img::DataProviderAlg::fDownscaleFullView, and img::DataProviderAlgView::fWireDriftData.

Referenced by img::DataProviderAlg::setWireDriftData().

{
  if (wireIdx >= fAlgView.fWireDriftData.size()) { return std::nullopt; }
  auto& wData = fAlgView.fWireDriftData[wireIdx];

  if (fDownscaleFullView) {
    if (!adc.empty()) { return downscale(wData.size(), adc, 0); }
    return std::nullopt;
  }
  if (adc.empty()) { return std::nullopt; }
  if (adc.size() <= wData.size()) { return adc; }
  return std::vector<float>(adc.begin(), adc.begin() + wData.size());
}

bool img::DataProviderAlg::setWireDriftData ( const detinfo::DetectorClocksData &  clock_data, const detinfo::DetectorPropertiesData &  det_prop, const std::vector< recob::Wire > &  wires, unsigned int  plane, unsigned int  tpc, unsigned int  cryo  )

inherited

Definition at line 289 of file DataProviderAlg.cxx.

References img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::applyBlur(), geo::GeometryCore::ChannelToWire(), img::DataProviderAlg::fAdcAreaOverThr, img::DataProviderAlg::fAdcSumOverThr, img::DataProviderAlg::fAdcSumThr, img::DataProviderAlg::fAlgView, img::DataProviderAlg::fCryo, img::DataProviderAlg::fGeometry, img::DataProviderAlg::fPlane, img::DataProviderAlg::fTPC, img::DataProviderAlgView::fWireChannels, img::DataProviderAlgView::fWireDriftData, detinfo::DetectorPropertiesData::NumberTimeSamples(), geo::GeometryCore::Nwires(), img::DataProviderAlg::Plane(), img::DataProviderAlg::resizeView(), and img::DataProviderAlg::setWireData().

Referenced by nnet::ParticleDecayId::DetectDecay(), nnet::EmTrackClusterId2out::produce(), nnet::EmTrackClusterId::produce(), nnet::EmTrackMichelId::produce(), nnet::TrainingDataAlg::setDataEventData(), and nnet::TrainingDataAlg::setEventData().

{
  mf::LogInfo("DataProviderAlg") << "Create image for cryo:" << cryo << " tpc:" << tpc
                                 << " plane:" << plane;

  fCryo = cryo;
  fTPC = tpc;
  fPlane = plane;

  fAdcSumOverThr = 0;
  fAdcAreaOverThr = 0;

  size_t nwires = fGeometry->Nwires({cryo, tpc, plane});
  size_t ndrifts = det_prop.NumberTimeSamples();

  fAlgView = resizeView(clock_data, det_prop, nwires, ndrifts);

  auto const& channelStatus =
    art::ServiceHandle<lariov::ChannelStatusService const>()->GetProvider();

  bool allWrong = true;
  for (auto const& wire : wires) {
    auto wireChannelNumber = wire.Channel();
    if (!channelStatus.IsGood(wireChannelNumber)) { continue; }

    size_t w_idx = 0;
    for (auto const& id : fGeometry->ChannelToWire(wireChannelNumber)) {
      if ((id.Plane == plane) && (id.TPC == tpc) && (id.Cryostat == cryo)) {
        w_idx = id.Wire;

        auto adc = wire.Signal();
        if (adc.size() < ndrifts) {
          mf::LogWarning("DataProviderAlg") << "Wire ADC vector size lower than NumberTimeSamples.";
          continue; // not critical, maybe other wires are OK, so continue
        }
        auto wire_data = setWireData(adc, w_idx);
        if (!wire_data) {
          mf::LogWarning("DataProviderAlg") << "Wire data not set.";
          continue; // also not critical, try to set other wires
        }
        fAlgView.fWireDriftData[w_idx] = *wire_data;
        for (auto v : adc) {
          if (v >= fAdcSumThr) {
            fAdcSumOverThr += v;
            fAdcAreaOverThr++;
          }
        }

        fAlgView.fWireChannels[w_idx] = wireChannelNumber;
        allWrong = false;
      }
    }
  }
  if (allWrong) {
    mf::LogError("DataProviderAlg")
      << "Wires data not set in the cryo:" << cryo << " tpc:" << tpc << " plane:" << plane;
    return false;
  }

  applyBlur();
  addWhiteNoise();
  addCoherentNoise();

  return true;
}

unsigned int img::DataProviderAlg::TPC ( ) const

inlineinherited

Definition at line 162 of file DataProviderAlg.h.

Referenced by nnet::TrainingDataAlg::collectVtxFlags().

{ return fTPC; }

std::vector<float> const& img::DataProviderAlg::wireData ( size_t  widx ) const

inlineinherited

Definition at line 117 of file DataProviderAlg.h.

Referenced by nnet::PointIdTrainingData::analyze().

{ return fAlgView.fWireDriftData[widx]; }

float img::DataProviderAlg::ZeroLevel ( ) const

inlineinherited

Level of zero ADC after scaling.

Definition at line 171 of file DataProviderAlg.h.

{ return fAdcZero; }

Member Data Documentation

DataProviderAlgView img::DataProviderAlg::fAlgView

protectedinherited

Definition at line 181 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::applyBlur(), img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), isInsideFiducialRegion(), img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::poolMax(), nnet::TrainingDataAlg::setDataEventData(), nnet::TrainingDataAlg::setEventData(), img::DataProviderAlg::setWireData(), img::DataProviderAlg::setWireDriftData(), and nnet::TrainingDataAlg::setWireEdepsAndLabels().

calo::CalorimetryAlg img::DataProviderAlg::fCalorimetryAlg

protectedinherited

Definition at line 238 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg(), and img::DataProviderAlg::resizeView().

size_t nnet::PointIdAlg::fCurrentScaledDrift

mutableprivate

Definition at line 157 of file PointIdAlg.h.

Referenced by isCurrentPatch().

size_t nnet::PointIdAlg::fCurrentWireIdx

mutableprivate

Definition at line 157 of file PointIdAlg.h.

Referenced by isCurrentPatch().

bool img::DataProviderAlg::fDownscaleFullView

protectedinherited

Definition at line 186 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::DataProviderAlg(), isCurrentPatch(), isSamePatch(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::resizeView(), img::DataProviderAlg::setWireData(), and nnet::TrainingDataAlg::setWireEdepsAndLabels().

EDownscaleMode img::DataProviderAlg::fDownscaleMode

protectedinherited

Definition at line 182 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg().

size_t img::DataProviderAlg::fDriftWindow

protectedinherited

Definition at line 185 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::DataProviderAlg(), img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), isCurrentPatch(), isInsideFiducialRegion(), isSamePatch(), img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::resizeView(), and nnet::TrainingDataAlg::setWireEdepsAndLabels().

float img::DataProviderAlg::fDriftWindowInv

protectedinherited

Definition at line 187 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg(), and img::DataProviderAlg::downscaleMean().

geo::GeometryCore const* img::DataProviderAlg::fGeometry

protectedinherited

Definition at line 241 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg(), nnet::TrainingDataAlg::getProjection(), and img::DataProviderAlg::setWireDriftData().

nnet::ModelInterface* nnet::PointIdAlg::fNNet

private

Definition at line 152 of file PointIdAlg.h.

Referenced by PointIdAlg(), predictIdValue(), predictIdVector(), and predictIdVectors().

std::string nnet::PointIdAlg::fNNetModelFilePath

private

Definition at line 150 of file PointIdAlg.h.

Referenced by PointIdAlg().

std::vector<std::string> nnet::PointIdAlg::fNNetOutputs

private

Definition at line 151 of file PointIdAlg.h.

Referenced by PointIdAlg().

size_t nnet::PointIdAlg::fPatchSizeD

private

Definition at line 155 of file PointIdAlg.h.

Referenced by isInsideFiducialRegion(), predictIdVectors(), and resizePatch().

size_t nnet::PointIdAlg::fPatchSizeW

private

Definition at line 155 of file PointIdAlg.h.

Referenced by isInsideFiducialRegion(), predictIdVectors(), and resizePatch().

std::vector<std::vector<float> > nnet::PointIdAlg::fWireDriftPatch

mutableprivate

Definition at line 154 of file PointIdAlg.h.

Referenced by predictIdValue(), predictIdVector(), and resizePatch().

---

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

• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.h
• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.cxx