kicad/3d-viewer/3d_rendering/image.h

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2015-2016 Mario Luzeiro <mrluzeiro@ua.pt>
 * Copyright (C) 2015-2020 KiCad Developers, see AUTHORS.txt for contributors.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, you may find one here:
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * or you may search the http://www.gnu.org website for the version 2 license,
 * or you may write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

 /**
 * @file image.h
 * @brief one 8bit-channel image definition.
 */

#ifndef IMAGE_H
#define IMAGE_H

#include <wx/string.h>

/// Image operation type
enum class IMAGE_OP
{
    RAW,
    ADD,
    SUB,
    DIF,
    MUL,
    AND,
    OR,
    XOR,
    BLEND50,
    MIN,
    MAX
};


/// Image wrap type enumeration
enum class IMAGE_WRAP
{
    ZERO,  ///< Coords that wraps are not evaluated
    CLAMP, ///< Coords are clamped to image size
    WRAP   ///< Coords are wrapped around
};


/// Filter type enumeration
enum class IMAGE_FILTER
{
    HIPASS,
    GAUSSIAN_BLUR,
    GAUSSIAN_BLUR2,
    INVERT_BLUR,
    CARTOON,
    EMBOSS,
    SHARPEN,
    MELT,
    SOBEL_GX,
    SOBEL_GY,
    BLUR_3X3,
};

/// 5x5 Filter struct parameters
typedef struct
{
    signed char    kernel[5][5];
    unsigned int   div;
    unsigned char  offset;
} S_FILTER;


/**
 * Manage an 8-bit channel image.
 */
class IMAGE
{
public:
    /**
     * Construct a IMAGE based on image size.
     *
     * @param aXsize x size
     * @param aYsize y size
     */
    IMAGE( unsigned int aXsize, unsigned int aYsize );

    /**
     * Construct a IMAGE based from an existing image.
     *
     * It will make a copy the \a aSrcImage.
     *
     * @param aSrcImage
     */
    IMAGE( const IMAGE& aSrcImage );

    ~IMAGE();

    /**
     * Set a value in a pixel position, position is clamped in accordance with the
     * current clamp settings.
     *
     * @param aX x position
     * @param aY y position
     * @param aValue value to set the pixel
     */
    void Setpixel( int aX, int aY, unsigned char aValue );

    /**
     * Get the pixel value from pixel position, position is clamped in accord with the
     * current clamp settings.
     *
     * @param aX x position
     * @param aY y position
     * @return unsigned char - pixel value
     */
    unsigned char Getpixel( int aX, int aY ) const;

    /**
     * Draw a horizontal line.
     *
     * @param aXStart x start position
     * @param aXEnd x end position
     * @param aY y position
     * @param aValue  value to add
     */
    void Hline( int aXStart, int aXEnd, int aY, unsigned char aValue );

    void CircleFilled( int aCx, int aCy, int aRadius, unsigned char aValue );

    /**
     * Perform a copy operation based on \a aOperation type.
     *
     * The available image operations.
     *  - IMAGE_OP::RAW        this <- aImgA
     *  - IMAGE_OP::ADD        this <- CLAMP(aImgA + aImgB)
     *  - IMAGE_OP::SUB        this <- CLAMP(aImgA - aImgB)
     *  - IMAGE_OP::DIF        this <- abs(aImgA - aImgB)
     *  - IMAGE_OP::MUL        this <- aImgA * aImgB
     *  - IMAGE_OP::AND        this <- aImgA & aImgB
     *  - IMAGE_OP::OR         this <- aImgA | aImgB
     *  - IMAGE_OP::XOR        this <- aImgA ^ aImgB
     *  - IMAGE_OP::BLEND50    this <- (aImgA + aImgB) / 2
     *  - IMAGE_OP::MIN        this <- (aImgA < aImgB) ? aImgA : aImgB
     *  - IMAGE_OP::MAX        this <- (aImgA > aImgB) ? aImgA : aImgB
     *
     * @param aImgA an image input.
     * @param aImgB an image input.
     * @param aOperation operation to perform
     */
    void CopyFull( const IMAGE* aImgA, const IMAGE* aImgB, IMAGE_OP aOperation );

    /**
     * Invert the values of this image <- (255 - this)
     */
    void Invert();

    /**
     * Apply a filter to the input image and store it in the image class.
     *
     * @param aInImg input image
     * @param aFilterType filter type to apply
     */
    void EfxFilter( IMAGE* aInImg, IMAGE_FILTER aFilterType );

    /**
     * Save image buffer to a PNG file into the working folder.
     *
     * Each RGB channel will have the 8bit-channel from the image.
     *
     * @param aFileName file name (without extension)
     */
    void SaveAsPNG( const wxString& aFileName ) const;

    /**
     * Set the current channel from a float normalized (0.0 - 1.0) buffer.
     *
     * this <- CLAMP(NormalizedFloat * 255)
     *
     * @param aNormalizedFloatArray a float array with the same size of the image
     */
    void SetPixelsFromNormalizedFloat( const float* aNormalizedFloatArray );

    /**
     * Get the image buffer pointer.
     *
     * @return unsigned char* the pointer of the buffer 8bit channel.
     */
    unsigned char* GetBuffer() const;

    unsigned int GetWidth() const { return m_width; }
    unsigned int GetHeight() const { return m_height; }

private:
    /**
     * Calculate the coordinates points in accord with the current clamping settings.
     *
     * @param aXo X coordinate to be converted (output).
     * @param aXo Y coordinate to be converted (output).
     * @return bool - true if the coordinates are inside the image, false otherwise.
     */
    bool wrapCoords( int* aXo, int* aYo ) const;

    void plot8CircleLines( int aCx, int aCy, int aX, int aY, unsigned char aValue );

    unsigned char*  m_pixels;           ///< buffer to store the image 8bit-channel
    unsigned int    m_width;            ///< width of the image
    unsigned int    m_height;           ///< height of the image
    unsigned int    m_wxh;              ///< width * height precalc value
    IMAGE_WRAP      m_wraping;          ///< current wrapping type
};

#endif   // IMAGE_H