/*
 * pixmap.cc: 2D image class
 *
 * This file is part of Pyrit Ray Tracer.
 *
 * Copyright 2008  Radek Brich
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#include <stdio.h>

#include "config.h"
#include "pixmap.h"

#ifdef HAVE_PNG
#	include <png.h>
#   include <zlib.h>
#endif

unsigned char *Pixmap::getCharData() const
{
	unsigned char *cdata = new unsigned char[w*h*3];
	Float *fd = fdata;

#if 1 //def NO_SIMD
	for (unsigned char *cd = cdata; cd != cdata + w*h*3; cd++, fd++)
		if (*fd > 1.0)
			*cd = 255;
		else
			*cd = (unsigned char)(*fd * 255.0);
#else
	__m128 cmax = _mm_set_ps1(255.0);
	__m64 m;
	if (!refdata)
		for (unsigned char *cd = cdata; cd < cdata + w*h*3; cd += 4, fd += 4)
		{
			m = _mm_cvtps_pi16(_mm_mul_ps(cmax,
				_mm_min_ps(mOne, _mm_load_ps(fd))));
			for (int i = 0; i < 4; i++)
				cd[i] = ((unsigned char *)&m)[i<<1];
		}
	else // must use _mm_loadu_ instead of _mm_load_ as we do not know the alignment
		for (unsigned char *cd = cdata; cd < cdata + w*h*3; cd += 4, fd += 4)
		{
			m = _mm_cvtps_pi16(_mm_mul_ps(cmax,
				_mm_min_ps(mOne, _mm_loadu_ps(fd))));
			for (int i = 0; i < 4; i++)
				cd[i] = ((unsigned char *)&m)[i<<1];
		}
#endif

	return cdata;
}

int Pixmap::writePNG(const char *fname) const
{
#ifndef HAVE_PNG
	dbgmsg(0, "W %s not written: PNG support is disabled\n", fname);
	return -3;
#else
	int y;
	FILE *f;
	png_structp png;   /* PNG data */
	png_infop pnginfo; /* PNG info */
	unsigned char *cdata, *d;

	if ((f = fopen(fname, "wb")) == NULL)
		return -1;

	png = png_create_write_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
	if (!png) {
		fclose(f);
		return -2;
	}

	pnginfo = png_create_info_struct(png);
	if (!pnginfo) {
		fclose(f);
		png_destroy_write_struct(&png, 0);
		return -2;
	}

	if (setjmp(png_jmpbuf(png))) {
		fclose(f);
		png_destroy_info_struct(png, &pnginfo);
		png_destroy_write_struct(&png, &pnginfo);
		return -2;
	}

	/* predat knihovne PNG ukazatel na soubor */
	png_init_io(png, f);

	/* parametry PNG */
	png_set_compression_level(png, Z_BEST_COMPRESSION);
	png_set_IHDR(png, pnginfo, w, h, 8, PNG_COLOR_TYPE_RGB,
		PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
		PNG_FILTER_TYPE_DEFAULT);
	png_set_sRGB(png, pnginfo, PNG_sRGB_INTENT_PERCEPTUAL);
	png_set_sRGB_gAMA_and_cHRM(png, pnginfo, PNG_INFO_sRGB);

	/* zapsat hlavicku */
	png_write_info(png, pnginfo);

	/* zapsat data */
	d = cdata = getCharData();
	for (y = 0; y < h; y++, d += w * 3)
		png_write_row(png, (png_byte *) d);

	/* ukoncit soubor a uvolnit pomocne struktury */
	png_write_end(png, pnginfo);
	png_destroy_info_struct(png, &pnginfo);
	png_destroy_write_struct(&png, 0);

	fclose(f);

	delete[] cdata;

	return 0;
#endif
}