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 (void) override
std::vector< std::string > const &	outputLabels (void) 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 (void) const
std::vector< float >	patchData1D (void) 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 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 (void) const
size_t	getAdcArea (void) const
float	poolMax (int wire, int drift, size_t r=0) const
	Pool max value in a patch around the wire/drift pixel. More...
float	poolSum (int wire, int drift, size_t r=0) const
	Pool sum of pixels in a patch around the wire/drift pixel. More...
unsigned int	Cryo (void) const
unsigned int	TPC (void) const
unsigned int	Plane (void) const
unsigned int	NWires (void) const
unsigned int	NScaledDrifts (void) const
unsigned int	NCachedDrifts (void) const
unsigned int	DriftWindow (void) const
float	ZeroLevel (void) const
	Level of zero ADC after scaling. More...
double	LifetimeCorrection (double tick) const

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

Protected Member Functions
void	downscaleMax (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
void	downscaleMaxMean (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
void	downscaleMean (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
void	downscale (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
size_t	getDriftIndex (float drift) const
bool	setWireData (std::vector< float > const &adc, size_t wireIdx)
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 void	resizeView (size_t wires, size_t drifts)

Protected Attributes
unsigned int	fCryo
unsigned int	fTPC
unsigned int	fPlane
unsigned int	fNWires
unsigned int	fNDrifts
unsigned int	fNScaledDrifts
unsigned int	fNCachedDrifts
std::vector< raw::ChannelID_t >	fWireChannels
std::vector< std::vector< float > >	fWireDriftData
std::vector< float >	fLifetimeCorrFactors
EDownscaleMode	fDownscaleMode
size_t	fDriftWindow
bool	fDownscaleFullView
float	fDriftWindowInv
calo::CalorimetryAlg	fCalorimetryAlg
geo::GeometryCore const *	fGeometry
detinfo::DetectorProperties const *	fDetProp

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)
void	deleteNNet (void)

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 89 of file PointIdAlg.h.

Member Enumeration Documentation

enum img::DataProviderAlg::EDownscaleMode

inherited

Enumerator
kMax
kMaxMean
kMean

Definition at line 49 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 154 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 ( void  )

override

Definition at line 185 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 156 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 178 of file PointIdAlg.h.

References bufferPatch().

Referenced by bufferPatch().

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

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

inlineinherited

Definition at line 157 of file DataProviderAlg.h.

References img::DataProviderAlg::fCryo.

{ return fCryo; }

void nnet::PointIdAlg::deleteNNet ( void  )

inlineprivate

Definition at line 181 of file PointIdAlg.h.

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

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

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

inlineprotectedinherited

Definition at line 189 of file DataProviderAlg.h.

References img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), 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: downscaleMean(dst, adc, tick0); break;
                case img::DataProviderAlg::kMaxMean: downscaleMaxMean(dst, adc, tick0); break;
                case img::DataProviderAlg::kMax: downscaleMax(dst, adc, tick0); break;
                default:throw cet::exception("img::DataProviderAlg") << "Downscale mode not supported." << std::endl; break;
            }
        }

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

protectedinherited

Definition at line 182 of file DataProviderAlg.cxx.

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

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

{
        size_t kStop = 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] * fLifetimeCorrFactors[k0 + tick0];
                for (size_t k = k0 + 1; k < k1; ++k)
                {
                        float ak = adc[k] * fLifetimeCorrFactors[k + tick0];
                        if (ak > max_adc) max_adc = ak;
                }

                dst[i] = max_adc;
        }
        scaleAdcSamples(dst);
}

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

protectedinherited

Definition at line 202 of file DataProviderAlg.cxx.

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

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

{
        size_t kStop = 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] * fLifetimeCorrFactors[k0 + tick0];
                for (size_t k = k0 + 1; k < k1; ++k)
                {
                        float ak = adc[k] * 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] * fLifetimeCorrFactors[max_idx - 1 + tick0]; n++; }
                if (max_idx + 1 < adc.size()) { max_adc += adc[max_idx + 1] * fLifetimeCorrFactors[max_idx + 1 + tick0]; n++; }

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

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

protectedinherited

Definition at line 227 of file DataProviderAlg.cxx.

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

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

{
        size_t kStop = 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)
                {
                        sum_adc += adc[k] * fLifetimeCorrFactors[k + tick0];
                }
        dst[i] = sum_adc * fDriftWindowInv;
        }
        scaleAdcSamples(dst);
}

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

inlineinherited

Definition at line 164 of file DataProviderAlg.h.

References img::DataProviderAlg::fDriftWindow.

{ return fDriftWindow; }

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

static

Definition at line 303 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 ( void  ) const

inlineinherited

Definition at line 149 of file DataProviderAlg.h.

References img::DataProviderAlg::fAdcAreaOverThr, img::DataProviderAlg::poolMax(), and img::DataProviderAlg::poolSum().

{ return fAdcAreaOverThr; }

double img::DataProviderAlg::getAdcSum ( void  ) const

inlineinherited

Definition at line 148 of file DataProviderAlg.h.

References img::DataProviderAlg::fAdcSumOverThr.

{ return fAdcSumOverThr; }

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

inlineprotectedinherited

Definition at line 200 of file DataProviderAlg.h.

References img::DataProviderAlg::fDriftWindowInv, img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::resizeView(), img::DataProviderAlg::setWireData(), and lar::dump::vector().

Referenced by img::DataProviderAlg::getPixelOrZero(), img::DataProviderAlg::poolMax(), and img::DataProviderAlg::poolSum().

    {
        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 114 of file DataProviderAlg.h.

References img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::patchFromDownsampledView(), and img::DataProviderAlg::patchFromOriginalView().

        {
                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;
                else
                {
                        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 136 of file DataProviderAlg.h.

References img::DataProviderAlg::fNCachedDrifts, img::DataProviderAlg::fWireDriftData, and img::DataProviderAlg::getDriftIndex().

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

        if ((widx < fWireDriftData.size()) &&
            (didx < fNCachedDrifts))
        {
            return fWireDriftData[widx][didx];
        }
        else { 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 285 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 327 of file PointIdAlg.cxx.

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

{
        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 < fNWires - marginW) &&
            (scaledDrift >= marginD) && (scaledDrift < 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 267 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 ( double  tick ) const

inlineinherited

Definition at line 169 of file DataProviderAlg.h.

References img::DataProviderAlg::fCalorimetryAlg, and calo::CalorimetryAlg::LifetimeCorrection().

{ return fCalorimetryAlg.LifetimeCorrection(tick); }

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

inlineinherited

Definition at line 163 of file DataProviderAlg.h.

References img::DataProviderAlg::fNCachedDrifts.

{ return fNCachedDrifts; }

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

inlineinherited

Definition at line 162 of file DataProviderAlg.h.

References img::DataProviderAlg::fNScaledDrifts.

{ return fNScaledDrifts; }

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

inlineinherited

Definition at line 161 of file DataProviderAlg.h.

References img::DataProviderAlg::fNWires.

{ return fNWires; }

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

inline

network output labels

Definition at line 123 of file PointIdAlg.h.

References lar::dump::vector().

{ return fNNetOutputs; }

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

inline

Definition at line 136 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 ( void  ) const

inline

Definition at line 135 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 400 of file DataProviderAlg.cxx.

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

Referenced by img::DataProviderAlg::getDriftIndex(), and img::DataProviderAlg::getPatch().

{
        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 = fWireDriftData.size();
        for (int w = w0, wpatch = 0; w < w1; ++w, ++wpatch)
        {
                auto & dst = patch[wpatch];
                if ((w >= 0) && (w < wsize))
                {
                        auto & src = 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 442 of file DataProviderAlg.cxx.

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

Referenced by img::DataProviderAlg::getDriftIndex(), and img::DataProviderAlg::getPatch().

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

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

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

        std::vector<float> tmp(dsize);
        int wsize = fWireDriftData.size();
        for (int w = w0, wpatch = 0; w < w1; ++w, ++wpatch)
        {
                if ((w >= 0) && (w < wsize))
                {
                        auto & src = 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);
                }

                downscale(patch[wpatch], tmp, d0);
        }

        return true;
}

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

inlineinherited

Definition at line 159 of file DataProviderAlg.h.

References img::DataProviderAlg::fPlane.

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 133 of file DataProviderAlg.cxx.

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

Referenced by img::DataProviderAlg::getAdcArea(), 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)fNCachedDrifts) { d1 = fNCachedDrifts - 1; }

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

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

    return max_adc;
}

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

inherited

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

Definition at line 159 of file DataProviderAlg.cxx.

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

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

{
    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)fNCachedDrifts) { d1 = fNCachedDrifts - 1; }

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

    float sum = 0;
    for (int w = w0; w <= w1; ++w)
    {
        auto const * col = fWireDriftData[w].data();
        for (int d = d0; d <= d1; ++d) { sum += col[d]; }
    }

    return sum;
}

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 198 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 222 of file PointIdAlg.cxx.

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

{
        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 245 of file PointIdAlg.cxx.

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

{
        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 ( void  )

private

Definition at line 191 of file PointIdAlg.cxx.

References fPatchSizeD, fPatchSizeW, and fWireDriftPatch.

Referenced by PointIdAlg().

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

void img::DataProviderAlg::resizeView ( size_t  wires, size_t  drifts  )

protectedvirtualinherited

Reimplemented in nnet::TrainingDataAlg.

Definition at line 103 of file DataProviderAlg.cxx.

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

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

{
    fNWires = wires; fNDrifts = drifts;
    fNScaledDrifts = drifts / fDriftWindow;

    if (fDownscaleFullView) { fNCachedDrifts = fNScaledDrifts; }
    else { fNCachedDrifts = fNDrifts; }

    fWireChannels.resize(wires);
    std::fill(fWireChannels.begin(), fWireChannels.end(), raw::InvalidChannelID);

    fWireDriftData.resize(wires);
    for (auto & w : fWireDriftData)
    {
        w.resize(fNCachedDrifts);
        std::fill(w.begin(), w.end(), fAdcZero);
    }

    fLifetimeCorrFactors.resize(fNDrifts);
    if (fCalibrateLifetime)
    {
        for (size_t t = 0; t < fNDrifts; ++t) { fLifetimeCorrFactors[t] = fCalorimetryAlg.LifetimeCorrection(t); }
    }
    else
    {
        for (size_t t = 0; t < fNDrifts; ++t) { fLifetimeCorrFactors[t] = 1.0; }
    }
}

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

protectedinherited

Definition at line 245 of file DataProviderAlg.cxx.

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

Referenced by img::DataProviderAlg::getDriftIndex(), and img::DataProviderAlg::setWireDriftData().

{
        if (wireIdx >= fWireDriftData.size()) return false;
        auto & wData = fWireDriftData[wireIdx];

    if (fDownscaleFullView)
    {
        if (!adc.empty()) { downscale(wData, adc, 0); }
        else { return false; }
    }
    else
    {
        if (adc.empty()) { return false; }
        else if (adc.size() <= wData.size()) { std::copy(adc.begin(), adc.end(), wData.begin()); }
        else { std::copy(adc.begin(), adc.begin()+wData.size(), wData.begin()); }
    }
    return true;
}

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

inherited

Definition at line 265 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::fCryo, img::DataProviderAlg::fDetProp, img::DataProviderAlg::fGeometry, img::DataProviderAlg::fPlane, img::DataProviderAlg::fTPC, img::DataProviderAlg::fWireChannels, detinfo::DetectorProperties::NumberTimeSamples(), geo::GeometryCore::Nwires(), img::DataProviderAlg::Plane(), img::DataProviderAlg::resizeView(), and img::DataProviderAlg::setWireData().

Referenced by img::DataProviderAlg::DataProviderAlg(), 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(plane, tpc, cryo);
        size_t ndrifts = fDetProp->NumberTimeSamples();

        resizeView(nwires, ndrifts);

    auto const & channelStatus = art::ServiceHandle< lariov::ChannelStatusService >()->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
                            }

                            if (!setWireData(adc, w_idx))
                            {
                                mf::LogWarning("DataProviderAlg") << "Wire data not set.";
                                continue; // also not critical, try to set other wires
                            }

                for (auto v : adc) { if (v >= fAdcSumThr) { fAdcSumOverThr += v; fAdcAreaOverThr++; } }

                            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 ( void  ) const

inlineinherited

Definition at line 158 of file DataProviderAlg.h.

References img::DataProviderAlg::fTPC.

{ return fTPC; }

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

inlineinherited

Definition at line 109 of file DataProviderAlg.h.

References img::DataProviderAlg::fWireDriftData.

{ return fWireDriftData[widx]; }

float img::DataProviderAlg::ZeroLevel ( void  ) const

inlineinherited

Level of zero ADC after scaling.

Definition at line 167 of file DataProviderAlg.h.

References img::DataProviderAlg::fAdcZero.

{ return fAdcZero; }

Member Data Documentation

calo::CalorimetryAlg img::DataProviderAlg::fCalorimetryAlg

protectedinherited

Definition at line 214 of file DataProviderAlg.h.

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

unsigned int img::DataProviderAlg::fCryo

protectedinherited

Definition at line 172 of file DataProviderAlg.h.

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

size_t nnet::PointIdAlg::fCurrentScaledDrift

mutableprivate

Definition at line 155 of file PointIdAlg.h.

Referenced by isCurrentPatch().

size_t nnet::PointIdAlg::fCurrentWireIdx

mutableprivate

Definition at line 155 of file PointIdAlg.h.

Referenced by isCurrentPatch().

detinfo::DetectorProperties const* img::DataProviderAlg::fDetProp

protectedinherited

Definition at line 218 of file DataProviderAlg.h.

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

bool img::DataProviderAlg::fDownscaleFullView

protectedinherited

Definition at line 183 of file DataProviderAlg.h.

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

EDownscaleMode img::DataProviderAlg::fDownscaleMode

protectedinherited

Definition at line 179 of file DataProviderAlg.h.

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

size_t img::DataProviderAlg::fDriftWindow

protectedinherited

Definition at line 182 of file DataProviderAlg.h.

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

float img::DataProviderAlg::fDriftWindowInv

protectedinherited

Definition at line 184 of file DataProviderAlg.h.

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

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

protectedinherited

Definition at line 217 of file DataProviderAlg.h.

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

std::vector<float> img::DataProviderAlg::fLifetimeCorrFactors

protectedinherited

Definition at line 177 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), and img::DataProviderAlg::resizeView().

unsigned int img::DataProviderAlg::fNCachedDrifts

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::getPixelOrZero(), img::DataProviderAlg::NCachedDrifts(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), nnet::TrainingDataAlg::resizeView(), and nnet::TrainingDataAlg::setWireEdepsAndLabels().

unsigned int img::DataProviderAlg::fNDrifts

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

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

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

private

Definition at line 150 of file PointIdAlg.h.

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

std::string nnet::PointIdAlg::fNNetModelFilePath

private

Definition at line 148 of file PointIdAlg.h.

Referenced by PointIdAlg().

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

private

Definition at line 149 of file PointIdAlg.h.

Referenced by PointIdAlg().

unsigned int img::DataProviderAlg::fNScaledDrifts

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

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

unsigned int img::DataProviderAlg::fNWires

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

Referenced by isInsideFiducialRegion(), img::DataProviderAlg::NWires(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), and nnet::TrainingDataAlg::setEventData().

size_t nnet::PointIdAlg::fPatchSizeD

private

Definition at line 153 of file PointIdAlg.h.

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

size_t nnet::PointIdAlg::fPatchSizeW

private

Definition at line 153 of file PointIdAlg.h.

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

unsigned int img::DataProviderAlg::fPlane

protectedinherited

Definition at line 172 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::Plane(), img::DataProviderAlg::scaleAdcSample(), img::DataProviderAlg::scaleAdcSamples(), and img::DataProviderAlg::setWireDriftData().

unsigned int img::DataProviderAlg::fTPC

protectedinherited

Definition at line 172 of file DataProviderAlg.h.

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

std::vector< raw::ChannelID_t > img::DataProviderAlg::fWireChannels

protectedinherited

Definition at line 175 of file DataProviderAlg.h.

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

std::vector< std::vector<float> > img::DataProviderAlg::fWireDriftData

protectedinherited

Definition at line 176 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::applyBlur(), img::DataProviderAlg::getPixelOrZero(), img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), img::DataProviderAlg::setWireData(), and img::DataProviderAlg::wireData().

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

mutableprivate

Definition at line 152 of file PointIdAlg.h.

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

---

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

• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.h
• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.cxx