nnet::EventImageData Class Reference

#include <EventImageData.h>

Public Member Functions
	EventImageData (size_t w, size_t d, bool saveDep)
void	addTpc (const TrainingDataAlg &dataAlg, size_t gw, bool flipw, size_t gd, bool flipd)
bool	findCrop (size_t max_area_cut, unsigned int &w0, unsigned int &w1, unsigned int &d0, unsigned int &d1) const
const std::vector< std::vector< float > > &	adcData (void) const
const std::vector< float > &	wireAdc (size_t widx) const
const std::vector< std::vector< float > > &	depData (void) const
const std::vector< float > &	wireDep (size_t widx) const
const std::vector< std::vector< int > > &	pdgData (void) const
const std::vector< int > &	wirePdg (size_t widx) const
void	setProjXY (const TrainingDataAlg &dataAlg, float x, float y, size_t gw, bool flipw, size_t gd, bool flipd)
float	getProjX (void) const
float	getProjY (void) const
int	getVtxX (void) const
int	getVtxY (void) const

Private Attributes
std::vector< std::vector< float > >	fAdc
std::vector< std::vector< float > >	fDeposit
std::vector< std::vector< int > >	fPdg
int	fVtxX
int	fVtxY
float	fProjX
float	fProjY
bool	fSaveDep

Detailed Description

Definition at line 28 of file EventImageData.h.

Constructor & Destructor Documentation

nnet::EventImageData::EventImageData	(	size_t	w,
		size_t	d,
		bool	saveDep
	)

Definition at line 51 of file EventImageData.cxx.

                                                                   :
    fVtxX(-9999), fVtxY(-9999),
    fProjX(-9999), fProjY(-9999),
    fSaveDep(saveDep)
{
    fAdc.resize(w, std::vector<float>(d, 0));
    if (saveDep) { fDeposit.resize(w, std::vector<float>(d, 0)); }
    fPdg.resize(w, std::vector<int>(d, 0));
}

Member Function Documentation

const std::vector< std::vector<float> >& nnet::EventImageData::adcData ( void  ) const

inline

Definition at line 36 of file EventImageData.h.

{ return fAdc; }

void nnet::EventImageData::addTpc	(	const TrainingDataAlg &	dataAlg,
		size_t	gw,
		bool	flipw,
		size_t	gd,
		bool	flipd
	)

Definition at line 71 of file EventImageData.cxx.

{
  float zero = dataAlg.ZeroLevel();
  for (size_t w = 0; w < dataAlg.NWires(); ++w)
  {
      size_t drift_size = dataAlg.NScaledDrifts();
      std::vector<float> & dstAdc = fAdc[gw + w];
      std::vector<float> & dstDep = fDeposit[gw + w];
      std::vector<int> & dstPdg = fPdg[gw + w];
      const float* srcAdc = 0;
      const float* srcDep = 0;
      const int* srcPdg = 0;
      if (flipw)
      {
          srcAdc = dataAlg.wireData(dataAlg.NWires() - w - 1).data();
          srcDep = dataAlg.wireEdep(dataAlg.NWires() - w - 1).data();
          srcPdg = dataAlg.wirePdg(dataAlg.NWires() - w - 1).data();
      }
      else
      {
          srcAdc = dataAlg.wireData(w).data();
          srcDep = dataAlg.wireEdep(w).data();
          srcPdg = dataAlg.wirePdg(w).data();
      }
      if (flipd)
      {
          for (size_t d = 0; d < drift_size; ++d) { dstAdc[gd + d] += srcAdc[drift_size - d - 1] - zero; }
          if (fSaveDep) { for (size_t d = 0; d < drift_size; ++d) { dstDep[gd + d] += srcDep[drift_size - d - 1]; } }
          for (size_t d = 0; d < drift_size; ++d)
          {
              int code = srcPdg[drift_size - d - 1];
              int best_pdg = code & nnet::TrainingDataAlg::kPdgMask;
              int vtx_flags = (dstPdg[gd + d] | code) & nnet::TrainingDataAlg::kVtxMask;
              dstPdg[gd + d] = vtx_flags | best_pdg; // now just overwrite pdg and keep all vtx flags
              
              if (code & nnet::TrainingDataAlg::kNuPri) { fVtxX = gw + w; fVtxY = gd + d; } // dstAdc[gd + d] = 127; }
          }
      }
      else
      {
          for (size_t d = 0; d < drift_size; ++d) { dstAdc[gd + d] += srcAdc[d] - zero; }
          if (fSaveDep) { for (size_t d = 0; d < drift_size; ++d) { dstDep[gd + d] += srcDep[d]; } }
          for (size_t d = 0; d < drift_size; ++d)
          {
              int code = srcPdg[d];
              int best_pdg = code & nnet::TrainingDataAlg::kPdgMask;
              int vtx_flags = (dstPdg[gd + d] | code) & nnet::TrainingDataAlg::kVtxMask;
              dstPdg[gd + d] = vtx_flags | best_pdg; // now just overwrite pdg and keep all vtx flags

              if (code & nnet::TrainingDataAlg::kNuPri) { fVtxX = gw + w; fVtxY = gd + d; } // dstAdc[gd + d] = 127; }
          }
      }
  }
}

const std::vector< std::vector<float> >& nnet::EventImageData::depData ( void  ) const

inline

Definition at line 39 of file EventImageData.h.

{ return fDeposit; }

bool nnet::EventImageData::findCrop	(	size_t	max_area_cut,
		unsigned int &	w0,
		unsigned int &	w1,
		unsigned int &	d0,
		unsigned int &	d1
	)		const

Definition at line 126 of file EventImageData.cxx.

{
  size_t max_cut = max_area_cut / 4;
  float adcThr = 10;

  w0 = 0;
  size_t cut = 0;
  while (w0 < fAdc.size())
  {
      for (auto const d : fAdc[w0]) { if (d > adcThr) cut++; }
      if (cut < max_cut) w0++;
      else break;
  }
  w1 = fAdc.size() - 1;
  cut = 0;
  while (w1 > w0)
  {
      for (auto const d : fAdc[w1]) { if (d > adcThr) cut++; }
      if (cut < max_cut) w1--;
      else break;
  }
  w1++;

  d0 = 0;
  cut = 0;
  while (d0 < fAdc.front().size())
  {
      for (size_t i = w0; i < w1; ++i) { if (fAdc[i][d0] > adcThr) cut++; }
      if (cut < max_cut) d0++;
      else break;
  }
  d1 = fAdc.front().size() - 1;
  cut = 0;
  while (d1 > d0)
  {
      for (size_t i = w0; i < w1; ++i) { if (fAdc[i][d1] > adcThr) cut++; }
      if (cut < max_cut) d1--;
      else break;
  }
  d1++;

  unsigned int margin = 32;
  if ((w1 - w0 > 8) && (d1 - d0 > 8))
  {
      if (w0 < margin) w0 = 0;
      else w0 -= margin;

      if (w1 > fAdc.size() - margin) w1 = fAdc.size();
      else w1 += margin;
      
      if (d0 < margin) d0 = 0;
      else d0 -= margin;
      
      if (d1 > fAdc.front().size() - margin) d1 = fAdc.front().size();
      else d1 += margin;
      
      return true;
  }
  else return false;
}

float nnet::EventImageData::getProjX ( void  ) const

inline

Definition at line 46 of file EventImageData.h.

{ return fProjX; }

float nnet::EventImageData::getProjY ( void  ) const

inline

Definition at line 47 of file EventImageData.h.

{ return fProjY; }

int nnet::EventImageData::getVtxX ( void  ) const

inline

Definition at line 49 of file EventImageData.h.

{ return fVtxX; }

int nnet::EventImageData::getVtxY ( void  ) const

inline

Definition at line 50 of file EventImageData.h.

{ return fVtxY; }

const std::vector< std::vector<int> >& nnet::EventImageData::pdgData ( void  ) const

inline

Definition at line 42 of file EventImageData.h.

{ return fPdg; }

void nnet::EventImageData::setProjXY	(	const TrainingDataAlg &	dataAlg,
		float	x,
		float	y,
		size_t	gw,
		bool	flipw,
		size_t	gd,
		bool	flipd
	)

Definition at line 62 of file EventImageData.cxx.

{
  if (flipw) { fProjX = gw + dataAlg.NWires() - x - 1; }
  else { fProjX = gw + x; }

  if (flipd) { fProjY = gd + dataAlg.NScaledDrifts() - y/dataAlg.DriftWindow() - 1; }
  else { fProjY = gd + y/dataAlg.DriftWindow(); }
}

const std::vector<float>& nnet::EventImageData::wireAdc ( size_t  widx ) const

inline

Definition at line 37 of file EventImageData.h.

{ return fAdc[widx]; }

const std::vector<float>& nnet::EventImageData::wireDep ( size_t  widx ) const

inline

Definition at line 40 of file EventImageData.h.

{ return fDeposit[widx]; }

const std::vector<int>& nnet::EventImageData::wirePdg ( size_t  widx ) const

inline

Definition at line 43 of file EventImageData.h.

{ return fPdg[widx]; }

Member Data Documentation

std::vector< std::vector<float> > nnet::EventImageData::fAdc

private

Definition at line 53 of file EventImageData.h.

std::vector< std::vector<float> > nnet::EventImageData::fDeposit

private

Definition at line 53 of file EventImageData.h.

std::vector< std::vector<int> > nnet::EventImageData::fPdg

private

Definition at line 54 of file EventImageData.h.

float nnet::EventImageData::fProjX

private

Definition at line 56 of file EventImageData.h.

float nnet::EventImageData::fProjY

private

Definition at line 56 of file EventImageData.h.

bool nnet::EventImageData::fSaveDep

private

Definition at line 57 of file EventImageData.h.

int nnet::EventImageData::fVtxX

private

Definition at line 55 of file EventImageData.h.

int nnet::EventImageData::fVtxY

private

Definition at line 55 of file EventImageData.h.

---

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

• dunereco/dunereco/CVN/adcutils/EventImageData.h
• dunereco/dunereco/CVN/adcutils/EventImageData.cxx