Valuemap.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.basecode
{
    [Serializable]
    public class Valuemap
    {
        int m_nChannels;
        int m_nHeight;
        int m_nWidth;
        int m_nChannelOffset;
        double[] m_rgData;
 
        public Valuemap(int nChannels, int nHeight, int nWidth, double[] rgData = null)
        {
            int nSize = nChannels * nHeight * nWidth;
            m_rgData = (rgData != null) ? rgData : new double[nSize];
            m_nChannels = nChannels;
            m_nHeight = nHeight;
            m_nWidth = nWidth;
            m_nChannelOffset = nHeight * nWidth;
        }
 
        public Valuemap(Valuemap data)
        {
            int nChannels = data.Channels;
            int nHeight = data.Height;
            int nWidth = data.Width;
 
            int nSize = nChannels * nHeight * nWidth;
            m_rgData = new double[nSize];
            m_nChannels = nChannels;
            m_nHeight = nHeight;
            m_nWidth = nWidth;
            m_nChannelOffset = nHeight * nWidth;
 
            Array.Copy(data.Values, m_rgData, nSize);
        }
 
        public void SetPixel(int nX, int nY, double clr)
        {
            int nIdx = nY * m_nWidth + nX;
            m_rgData[nIdx] = clr;
 
            if (m_nChannels == 3)
            {
                m_rgData[nIdx + m_nChannelOffset] = clr;
                m_rgData[nIdx + m_nChannelOffset * 2] = clr;
            }
        }
 
        public double GetPixel(int nX, int nY)
        {
            int nIdx = nY * m_nWidth + nX;
            return m_rgData[nIdx];
        }
 
        public void SetPixel(int nChannel, int nX, int nY, double clr)
        {
            int nIdx = (nChannel * m_nChannelOffset) + nY * m_nWidth + nX;
            m_rgData[nIdx] = clr;
        }
 
        public double GetPixel(int nChannel, int nX, int nY)
        {
            int nIdx = (nChannel * m_nChannelOffset) + nY * m_nWidth + nX;
            return m_rgData[nIdx];
        }
 
        public void SetAll(double dfVal)
        {
            for (int i = 0; i < m_rgData.Length; i++)
            {
                m_rgData[i] = dfVal;
            }
        }
 
        public void SetAll(int nChannel, double dfVal)
        {
            int nIdxStart = m_nChannelOffset * nChannel;
            int nCount = m_nChannelOffset;
 
            for (int i = nIdxStart; i < nIdxStart + nCount; i++)
            {
                m_rgData[i] = dfVal;
            }
        }
 
        public void Clear()
        {
            Array.Clear(m_rgData, 0, m_rgData.Length);
        }
 
        public double[] Values
        {
            get { return m_rgData; }
        }
 
        public int Channels
        {
            get { return m_nChannels; }
        }
 
        public int Height
        {
            get { return m_nHeight; }
        }
 
        public int Width
        {
            get { return m_nWidth; }
        }
    }
}