cnn::RegCNNImageUtils Class Reference

Class containing some utility functions for all things RegCNN. More...

#include <RegCNNImageUtils.h>

Public Member Functions
	RegCNNImageUtils ()
	RegCNNImageUtils (unsigned int nWires, unsigned int nTDCs, unsigned int nViews)
	~RegCNNImageUtils ()
void	SetViewReversal (bool reverseX, bool reverseY, bool reverseZ)
	Function to set any views that need reversing. More...
void	SetViewReversal (std::vector< bool > reverseViews)
void	SetPixelMapSize (unsigned int nWires, unsigned int nTDCs)
	Set the input pixel map size. More...
float	ConvertToScaledCharge (float charge)
void	ConvertPixelMapToImageVectorF (const RegPixelMap &pm, ImageVectorF &imageVec)
	Convert a pixel map into an image vector (float version) More...
void	ConvertChargeVectorsToImageVectorF (std::vector< float > &v0pe, std::vector< float > &v1pe, std::vector< float > &v2pe, ImageVectorF &imageVec)
	Float version of conversion for convenience of TF interface. More...

Private Member Functions
void	ConvertChargeVectorsToViewVectors (std::vector< float > &v0pe, std::vector< float > &v1pe, std::vector< float > &v2pe, ViewVectorF &view0, ViewVectorF &view1, ViewVectorF &view2)
	Base function for conversion of the Pixel Map to our required output format. More...
ImageVectorF	BuildImageVectorF (ViewVectorF v0, ViewVectorF v1, ViewVectorF v2)
	Make the image vector from the view vectors. More...
void	ReverseView (std::vector< float > &peVec)
	Funtion to actually reverse the view. More...

Private Attributes
unsigned int	fNViews
	Number of views of each event. More...
unsigned int	fPixelMapWires
	Input pixel map sizes. More...
unsigned int	fPixelMapTDCs
std::vector< bool >	fViewReverse
	Vector of bools to decide if any views need to be reversed. More...

Detailed Description

Class containing some utility functions for all things RegCNN.

Definition at line 22 of file RegCNNImageUtils.h.

Constructor & Destructor Documentation

cnn::RegCNNImageUtils::RegCNNImageUtils

(

)

Definition at line 6 of file RegCNNImageUtils.cxx.

                                     {
  // Set a default image size
  SetPixelMapSize(2880,500);
  fViewReverse = {false,false,false};
  fNViews = 3;
}

cnn::RegCNNImageUtils::RegCNNImageUtils	(	unsigned int	nWires,
		unsigned int	nTDCs,
		unsigned int	nViews
	)

Definition at line 13 of file RegCNNImageUtils.cxx.

                                                                                                 {
  SetPixelMapSize(2880,500);
}

cnn::RegCNNImageUtils::~RegCNNImageUtils

(

)

Definition at line 17 of file RegCNNImageUtils.cxx.

                                      {

}

Member Function Documentation

cnn::ImageVectorF cnn::RegCNNImageUtils::BuildImageVectorF ( cnn::ViewVectorF  v0, cnn::ViewVectorF  v1, cnn::ViewVectorF  v2  )

private

Make the image vector from the view vectors.

Definition at line 139 of file RegCNNImageUtils.cxx.

                                                                                                             {

  // Tensorflow wants things in the arrangement <wires, TDCs, views>
  cnn::ImageVectorF image;
  for(unsigned int w = 0; w < v0.size(); ++w){
    std::vector<std::vector<float> > wireVec;
    for(unsigned int t = 0; t < v0[0].size(); ++t){
      std::vector<float> timeVec;
      timeVec.push_back(v0[w][t]);
      timeVec.push_back(v1[w][t]);
      timeVec.push_back(v2[w][t]);
      wireVec.push_back(timeVec);
    } // Loop over tdcs
    image.push_back(wireVec);
  } // Loop over wires
  
  

  return image;
}

void cnn::RegCNNImageUtils::ConvertChargeVectorsToImageVectorF	(	std::vector< float > &	v0pe,
		std::vector< float > &	v1pe,
		std::vector< float > &	v2pe,
		cnn::ImageVectorF &	imageVec
	)

Float version of conversion for convenience of TF interface.

Definition at line 57 of file RegCNNImageUtils.cxx.

                                                                                                             {

  cnn::ViewVectorF view0;
  cnn::ViewVectorF view1;
  cnn::ViewVectorF view2;

  ConvertChargeVectorsToViewVectors(v0pe, v1pe, v2pe, view0, view1, view2);

  cnn::ImageVectorF newImage = BuildImageVectorF(view0,view1,view2);

  imageVec = newImage;
}

void cnn::RegCNNImageUtils::ConvertChargeVectorsToViewVectors ( std::vector< float > &  v0pe, std::vector< float > &  v1pe, std::vector< float > &  v2pe, cnn::ViewVectorF &  view0, cnn::ViewVectorF &  view1, cnn::ViewVectorF &  view2  )

private

Base function for conversion of the Pixel Map to our required output format.

Definition at line 71 of file RegCNNImageUtils.cxx.

                                                                                                           {

  // Reverse requested views
  if(fViewReverse[0]) ReverseView(v0pe);
  if(fViewReverse[1]) ReverseView(v1pe);
  if(fViewReverse[2]) ReverseView(v2pe);

  // Write the values to the three vectors
  for (unsigned int view = 0; view < fNViews; ++view){
    cnn::ViewVectorF viewChargeVec;
    //for (unsigned int wire = 0; wire < fPixelMapWires; ++wire){
    for (unsigned int wire = 0; wire < fPixelMapWires+1; ++wire){
      std::vector<float> wireTDCVec;
      //for (unsigned int time = 0; time < fPixelMapTDCs; ++time){
      for (unsigned int time = 0; time < fPixelMapTDCs+1; ++time) {

        // Get the index for the pixel map
        unsigned int element = time + fPixelMapTDCs * wire;
            // FIXME: needs to add additional dimensions because of TF model
            if (time < fPixelMapTDCs && wire < fPixelMapWires){
                    // Scale Charges
                    float val = 0;
                    if(view == 0){ val = ConvertToScaledCharge(v0pe[element]); }
                    if(view == 1){ val = ConvertToScaledCharge(v1pe[element]); }
                    if(view == 2){ val = ConvertToScaledCharge(v2pe[element]); }
                    wireTDCVec.push_back(val);
            } else {
            wireTDCVec.push_back(0.);
        }
      }
      viewChargeVec.push_back(wireTDCVec);
    }
    if(view == 0) view0 = viewChargeVec;
    if(view == 1) view1 = viewChargeVec;
    if(view == 2) view2 = viewChargeVec;
  }

  return;

}

void cnn::RegCNNImageUtils::ConvertPixelMapToImageVectorF	(	const RegPixelMap &	pm,
		cnn::ImageVectorF &	imageVec
	)

Convert a pixel map into an image vector (float version)

Definition at line 45 of file RegCNNImageUtils.cxx.

                                                                                                          {

  SetPixelMapSize(pm.fNWire,pm.fNTdc);

  // Strip out the charge vectors and use these
  std::vector<float> v0pe = pm.fPEX;
  std::vector<float> v1pe = pm.fPEY;
  std::vector<float> v2pe = pm.fPEZ;

  ConvertChargeVectorsToImageVectorF(v0pe, v1pe, v2pe, imageVec);
}

float cnn::RegCNNImageUtils::ConvertToScaledCharge

(

float

charge

)

Definition at line 41 of file RegCNNImageUtils.cxx.

                                                            {
  return charge/500.;
}

void cnn::RegCNNImageUtils::ReverseView ( std::vector< float > &  peVec )

private

Funtion to actually reverse the view.

Definition at line 114 of file RegCNNImageUtils.cxx.

                                                            {

  std::vector<float> vecCopy(peVec.size(),0.); 

  for (unsigned int w = 0; w < fPixelMapWires; ++w)
  {
    // Get our new plane number
    unsigned int newPlane = fPixelMapWires - w - 1;

    for (unsigned int t = 0; t < fPixelMapTDCs; ++t)
    {
      float val = peVec[t + fPixelMapTDCs * w];
      vecCopy[t + fPixelMapTDCs * newPlane] = val;
    }
  }

  // Copy the values back into the original vector
  for(unsigned int e = 0; e < peVec.size(); ++e){
    float val = vecCopy[e];
    peVec[e] = val;
  }

}

void cnn::RegCNNImageUtils::SetPixelMapSize	(	unsigned int	nWires,
		unsigned int	nTDCs
	)

Set the input pixel map size.

Definition at line 35 of file RegCNNImageUtils.cxx.

                                                                                {
  fPixelMapWires = nWires;
  fPixelMapTDCs = nTDCs;
}

void cnn::RegCNNImageUtils::SetViewReversal	(	bool	reverseX,
		bool	reverseY,
		bool	reverseZ
	)

Function to set any views that need reversing.

Definition at line 21 of file RegCNNImageUtils.cxx.

                                                                                    {
  fViewReverse = {reverseX,reverseY,reverseZ};
}

void cnn::RegCNNImageUtils::SetViewReversal

(

std::vector< bool >

reverseViews

)

Definition at line 25 of file RegCNNImageUtils.cxx.

                                                                     {
  if(reverseViews.size() != 3){
    std::cout << "Expected three views for view reversals... using defaults." << std::endl;
  }
  else{
    SetViewReversal(reverseViews[0],reverseViews[1],reverseViews[2]);
  }
  return;
}

Member Data Documentation

unsigned int cnn::RegCNNImageUtils::fNViews

private

Number of views of each event.

Definition at line 63 of file RegCNNImageUtils.h.

unsigned int cnn::RegCNNImageUtils::fPixelMapTDCs

private

Definition at line 67 of file RegCNNImageUtils.h.

unsigned int cnn::RegCNNImageUtils::fPixelMapWires

private

Input pixel map sizes.

Definition at line 66 of file RegCNNImageUtils.h.

std::vector<bool> cnn::RegCNNImageUtils::fViewReverse

private

Vector of bools to decide if any views need to be reversed.

Definition at line 70 of file RegCNNImageUtils.h.

---

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

• dunereco/dunereco/RegCNN/func/RegCNNImageUtils.h
• dunereco/dunereco/RegCNN/func/RegCNNImageUtils.cxx