png_toolkit.cpp

#define STB_IMAGE_IMPLEMENTATION
#define STB_IMAGE_WRITE_IMPLEMENTATION
#include <array>
#include "stb_image_write.h"
#include "png_toolkit.h"

png_toolkit::png_toolkit()
{
}

png_toolkit::~png_toolkit()
{
    stbi_image_free(imgData.pixels);
}
bool png_toolkit::load( const std::string &pictureName )
{
    imgData.pixels = stbi_load(pictureName.c_str(), &imgData.w, &imgData.h, &imgData.compPerPixel, 0);
    return imgData.pixels != nullptr;
}

bool png_toolkit::save( const std::string &pictureName )
{
    return stbi_write_png(pictureName.c_str(),
                   imgData.w, imgData.h,
                   imgData.compPerPixel,
                          imgData.pixels, 0) != 0;
}

image_data png_toolkit::getPixelData( void ) const
{
    return imgData;
}

void png_toolkit::paint()
{
	auto a = imgData.compPerPixel;
	int i, j;
	if ((imgData.compPerPixel == 3) || (imgData.compPerPixel == 4))
	for (i = (imgData.h) / 2; i < imgData.h; i++)
		for (j = 0; j < a * imgData.w; j += a)
		{
			imgData.pixels[i * imgData.w * a + j] = 255;
			imgData.pixels[1 + i * imgData.w * a + j] = 0;
			imgData.pixels[2 + i * imgData.w * a + j] = 0;
		}
}