CVNImageUtils.h

Go to the documentation of this file.

#ifndef LCVN_IMAGE_UTILS_H
#define LCVN_IMAGE_UTILS_H

#include <vector>

#include "larrecodnn/CVN/func/PixelMap.h"

namespace lcvn {

  typedef std::vector<std::vector<unsigned char>> ViewVector;
  typedef std::vector<ViewVector> ImageVector;
  typedef std::vector<std::vector<float>> ViewVectorF;
  typedef std::vector<ViewVectorF> ImageVectorF;

  class CVNImageUtils {
  public:
    CVNImageUtils();
    CVNImageUtils(unsigned int nWires, unsigned int nTDCs, unsigned int nViews);

    void DisableRegionSelection();
    void EnableRegionSelection();

    unsigned char ConvertChargeToChar(float charge);

    void SetImageSize(unsigned int nWires, unsigned int nTDCs, unsigned int nViews);

    void SetViewReversal(bool reverseX, bool reverseY, bool reverseZ);
    void SetViewReversal(std::vector<bool> reverseViews);

    void SetLogScale(bool setLog);

    void SetPixelMapSize(unsigned int nWires, unsigned int nTDCs);

    void ConvertPixelMapToPixelArray(const PixelMap& pm, std::vector<unsigned char>& pix);

    void ConvertChargeVectorsToPixelArray(std::vector<float>& v0pe,
                                          std::vector<float>& v1pe,
                                          std::vector<float>& v2pe,
                                          std::vector<unsigned char>& pix);

    void ConvertPixelMapToImageVector(const PixelMap& pm, ImageVector& imageVec);

    void ConvertPixelMapToImageVectorF(const PixelMap& pm, ImageVectorF& imageVec);

    void ConvertChargeVectorsToImageVector(std::vector<float>& v0pe,
                                           std::vector<float>& v1pe,
                                           std::vector<float>& v2pe,
                                           ImageVector& imageVec);

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

    void ConvertPixelArrayToImageVectorF(const std::vector<unsigned char>& pixelArray,
                                         ImageVectorF& imageVec);

  private:
    void ConvertChargeVectorsToViewVectors(std::vector<float>& v0pe,
                                           std::vector<float>& v1pe,
                                           std::vector<float>& v2pe,
                                           ViewVector& view0,
                                           ViewVector& view1,
                                           ViewVector& view2);

    ImageVector BuildImageVector(ViewVector& v0, ViewVector& v1, ViewVector& v2);
    ImageVectorF BuildImageVectorF(ViewVectorF& v0, ViewVectorF& v1, ViewVectorF& v2);

    void GetMinMaxWires(std::vector<float>& wireCharges,
                        unsigned int& minWire,
                        unsigned int& maxWire);

    void GetMinMaxTDCs(std::vector<float>& tdcCharges, unsigned int& minTDC, unsigned int& maxTDC);

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

    ViewVectorF ConvertViewVecToViewVecF(ViewVector& view);

    ImageVectorF ConvertImageVecToImageVecF(ImageVector& image);

    unsigned int fNViews;

    unsigned int fNWires;

    unsigned int fNTDCs;

    unsigned int fPixelMapWires;
    unsigned int fPixelMapTDCs;

    std::vector<bool> fViewReverse;

    bool fDisableRegionSelection;

    bool fUseLogScale;
  };

}

#endif // CVN_IMAGE_UTILS_H