src/pixmap.cc
branchpyrit
changeset 88 f7edb3b90816
child 89 fcf1487b398b
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/pixmap.cc	Mon Apr 28 11:44:11 2008 +0200
@@ -0,0 +1,113 @@
+/*
+ * 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>
+
+#ifdef HAVE_PNG
+#	include <png.h>
+#endif
+
+#include "pixmap.h"
+
+unsigned char *Pixmap::getCharData() const
+{
+	unsigned char *cdata = new unsigned char[w*h*3];
+	Float *fd = fdata;
+	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);
+	}
+	return cdata;
+}
+
+int Pixmap::writePNG(const char *fname) const
+{
+#ifndef HAVE_PNG
+	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
+}