# HG changeset patch
# User John Tsiombikas <nuclear@member.fsf.org>
# Date 1350824207 -10800
# Node ID 5981917093ff685dfdc12f5d42d497047f1858a7
# Parent  67d33003862935f0201bd07087059b9cd4e6a751
color grading palette output done, all is left is the input

diff -r 67d330038629 -r 5981917093ff prototype/src/colgrade.cc
--- a/prototype/src/colgrade.cc	Sun Oct 21 02:11:23 2012 +0300
+++ b/prototype/src/colgrade.cc	Sun Oct 21 15:56:47 2012 +0300
@@ -1,10 +1,13 @@
 #include <stdio.h>
+#include <string.h>
 #include "opengl.h"
 #include "colgrade.h"
+#include "imago2.h"
 
 GradePalette::GradePalette()
 {
 	tex = 0;
+	palette = 0;
 }
 
 GradePalette::~GradePalette()
@@ -18,18 +21,14 @@
 
 	destroy();
 
-	glGenTextures(1, &tex);
-	glBindTexture(GL_TEXTURE_3D, tex);
-	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
-	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
-	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_S, clamp);
-	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_T, clamp);
-	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_R, clamp);
-	glTexImage3D(GL_TEXTURE_3D, 0, GL_RGB, sz, sz, sz, 0, GL_RGB, GL_UNSIGNED_BYTE, 0);
+	this->size = sz;
 
-	unsigned char *scanline = new unsigned char[sz * 3];
+	int nbytes = sz * sz * sz * 3;
+	printf("allocating %d bytes for a color grading palette\n", nbytes);
+	palette = new unsigned char[nbytes];
 
-	size = sz;
+	unsigned char *scanline = palette;
+
 	for(int i=0; i<sz; i++) {	// for each slice...
 		int b = 255 * i / (sz - 1);
 		for(int j=0; j<sz; j++) {	// for each scanline...
@@ -42,12 +41,19 @@
 				scanline[k * 3 + 1] = g;
 				scanline[k * 3 + 2] = b;
 			}
-			glTexSubImage3D(GL_TEXTURE_3D, 0, 0, j, i, sz, 1, 1, GL_RGB,
-					GL_UNSIGNED_BYTE, scanline);
+			scanline += sz * 3;
 		}
 	}
 
-	delete [] scanline;
+	glGenTextures(1, &tex);
+	glBindTexture(GL_TEXTURE_3D, tex);
+	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
+	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
+	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_S, clamp);
+	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_T, clamp);
+	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_R, clamp);
+	glTexImage3D(GL_TEXTURE_3D, 0, GL_RGB, sz, sz, sz, 0, GL_RGB, GL_UNSIGNED_BYTE, palette);
+
 	return true;
 }
 
@@ -55,12 +61,58 @@
 {
 	if(tex) {
 		glDeleteTextures(1, &tex);
+		tex = 0;
 	}
+	delete [] palette;
+	palette = 0;
 }
 
 bool GradePalette::save_shot(const char *fname) const
 {
-	return false;	// TODO
+	if(!tex) {
+		return false;
+	}
+
+	int slice_bytes = size * size * 3;
+	int nslices = size;
+
+	int vp[4];
+	glGetIntegerv(GL_VIEWPORT, vp);
+
+	int xsz = vp[2];
+	int ysz = vp[3];
+
+	// allocate more scanlines to cram the palette slices in there
+	unsigned char *img = new unsigned char[xsz * (ysz + nslices) * 3];
+
+	glReadPixels(0, 0, vp[2], vp[3], GL_RGB, GL_UNSIGNED_BYTE, img);
+
+	// invert the image
+	unsigned char *top = img;
+	unsigned char *bot = img + xsz * (ysz - 1) * 3;
+
+	while(top < bot) {
+		for(int j=0; j<xsz * 3; j++) {
+			unsigned char tmp = top[j];
+			top[j] = bot[j];
+			bot[j] = tmp;
+		}
+		top += xsz * 3;
+		bot -= xsz * 3;
+	}
+
+	unsigned char *pal = img + xsz * ysz * 3;
+	memset(pal, 0, nslices * xsz * 3);
+
+	for(int i=0; i<nslices; i++) {
+		memcpy(pal, palette + i * slice_bytes, slice_bytes);
+		pal += xsz * 3;
+	}
+
+	int res = img_save_pixels(fname, img, xsz, ysz + nslices, IMG_FMT_RGB24);
+	delete [] img;
+
+	return res == 0;
 }
 
 bool GradePalette::load_shot(const char *fname)
diff -r 67d330038629 -r 5981917093ff prototype/src/colgrade.h
--- a/prototype/src/colgrade.h	Sun Oct 21 02:11:23 2012 +0300
+++ b/prototype/src/colgrade.h	Sun Oct 21 15:56:47 2012 +0300
@@ -3,6 +3,7 @@
 
 class GradePalette {
 private:
+	unsigned char *palette;
 	unsigned int tex;
 	int size;
 
diff -r 67d330038629 -r 5981917093ff prototype/src/main.cc
--- a/prototype/src/main.cc	Sun Oct 21 02:11:23 2012 +0300
+++ b/prototype/src/main.cc	Sun Oct 21 15:56:47 2012 +0300
@@ -43,6 +43,7 @@
 static float stereo_eye_sep = stereo_focus_dist / 30.0;
 
 static bool show_con;
+static bool save_grade_shot;
 
 static AudioSource *move_sound;
 static OggVorbisStream *music;
@@ -245,6 +246,14 @@
 	if(show_con) {
 		draw_cmdcon();
 	}
+
+	if(save_grade_shot) {
+		GradePalette *pal = rend->get_grade_palette();
+		if(!pal->save_shot("gradeshot.ppm")) {
+			fprintf(stderr, "failed to save color grading shot\n");
+		}
+		save_grade_shot = false;
+	}
 }
 
 void view_matrix(int eye)
@@ -396,6 +405,10 @@
 		}
 		break;
 
+	case 'g':
+		save_grade_shot = true;
+		break;
+
 	default:
 		break;
 	}
diff -r 67d330038629 -r 5981917093ff prototype/src/renderer.cc
--- a/prototype/src/renderer.cc	Sun Oct 21 02:11:23 2012 +0300
+++ b/prototype/src/renderer.cc	Sun Oct 21 15:56:47 2012 +0300
@@ -199,6 +199,16 @@
 	return true;
 }
 
+GradePalette *Renderer::get_grade_palette()
+{
+	return &gradepal;
+}
+
+const GradePalette *Renderer::get_grade_palette() const
+{
+	return &gradepal;
+}
+
 
 // ---- fallback forward renderer ----
 FwdRenderer::FwdRenderer()
diff -r 67d330038629 -r 5981917093ff prototype/src/renderer.h
--- a/prototype/src/renderer.h	Sun Oct 21 02:11:23 2012 +0300
+++ b/prototype/src/renderer.h	Sun Oct 21 15:56:47 2012 +0300
@@ -32,6 +32,9 @@
 	virtual void render_pre(const Level *level) const;
 	virtual void render(const Level *level) const = 0;
 	virtual void render_post(const Level *level) const;
+
+	virtual GradePalette *get_grade_palette();
+	virtual const GradePalette *get_grade_palette() const;
 };