nuclear@72: #include <stdio.h>
nuclear@74: #include <string.h>
nuclear@72: #include "opengl.h"
nuclear@72: #include "colgrade.h"
nuclear@74: #include "imago2.h"
nuclear@72: 
nuclear@72: GradePalette::GradePalette()
nuclear@72: {
nuclear@72: 	tex = 0;
nuclear@74: 	palette = 0;
nuclear@72: }
nuclear@72: 
nuclear@72: GradePalette::~GradePalette()
nuclear@72: {
nuclear@72: 	destroy();
nuclear@72: }
nuclear@72: 
nuclear@72: bool GradePalette::create(int sz)
nuclear@72: {
nuclear@72: 	unsigned int clamp = GLEW_ARB_texture_border_clamp ? GL_CLAMP_TO_EDGE : GL_CLAMP;
nuclear@72: 
nuclear@72: 	destroy();
nuclear@72: 
nuclear@74: 	this->size = sz;
nuclear@72: 
nuclear@74: 	int nbytes = sz * sz * sz * 3;
nuclear@74: 	printf("allocating %d bytes for a color grading palette\n", nbytes);
nuclear@74: 	palette = new unsigned char[nbytes];
nuclear@72: 
nuclear@74: 	unsigned char *scanline = palette;
nuclear@74: 
nuclear@72: 	for(int i=0; i<sz; i++) {	// for each slice...
nuclear@72: 		int b = 255 * i / (sz - 1);
nuclear@72: 		for(int j=0; j<sz; j++) {	// for each scanline...
nuclear@72: 			int g = 255 * j / (sz - 1);
nuclear@72: 
nuclear@72: 			for(int k=0; k<sz; k++) {
nuclear@72: 				int r = 255 * k / (sz - 1);
nuclear@72: 
nuclear@72: 				scanline[k * 3] = r;
nuclear@72: 				scanline[k * 3 + 1] = g;
nuclear@72: 				scanline[k * 3 + 2] = b;
nuclear@72: 			}
nuclear@74: 			scanline += sz * 3;
nuclear@72: 		}
nuclear@72: 	}
nuclear@72: 
nuclear@74: 	glGenTextures(1, &tex);
nuclear@74: 	glBindTexture(GL_TEXTURE_3D, tex);
nuclear@74: 	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
nuclear@74: 	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
nuclear@74: 	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_S, clamp);
nuclear@74: 	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_T, clamp);
nuclear@74: 	glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_R, clamp);
nuclear@74: 	glTexImage3D(GL_TEXTURE_3D, 0, GL_RGB, sz, sz, sz, 0, GL_RGB, GL_UNSIGNED_BYTE, palette);
nuclear@74: 
nuclear@72: 	return true;
nuclear@72: }
nuclear@72: 
nuclear@72: void GradePalette::destroy()
nuclear@72: {
nuclear@72: 	if(tex) {
nuclear@72: 		glDeleteTextures(1, &tex);
nuclear@74: 		tex = 0;
nuclear@72: 	}
nuclear@74: 	delete [] palette;
nuclear@74: 	palette = 0;
nuclear@72: }
nuclear@72: 
nuclear@72: bool GradePalette::save_shot(const char *fname) const
nuclear@72: {
nuclear@74: 	if(!tex) {
nuclear@74: 		return false;
nuclear@74: 	}
nuclear@74: 
nuclear@74: 	int slice_bytes = size * size * 3;
nuclear@74: 	int nslices = size;
nuclear@74: 
nuclear@74: 	int vp[4];
nuclear@74: 	glGetIntegerv(GL_VIEWPORT, vp);
nuclear@74: 
nuclear@74: 	int xsz = vp[2];
nuclear@74: 	int ysz = vp[3];
nuclear@74: 
nuclear@74: 	// allocate more scanlines to cram the palette slices in there
nuclear@74: 	unsigned char *img = new unsigned char[xsz * (ysz + nslices) * 3];
nuclear@74: 
nuclear@74: 	glReadPixels(0, 0, vp[2], vp[3], GL_RGB, GL_UNSIGNED_BYTE, img);
nuclear@74: 
nuclear@74: 	// invert the image
nuclear@74: 	unsigned char *top = img;
nuclear@74: 	unsigned char *bot = img + xsz * (ysz - 1) * 3;
nuclear@74: 
nuclear@74: 	while(top < bot) {
nuclear@74: 		for(int j=0; j<xsz * 3; j++) {
nuclear@74: 			unsigned char tmp = top[j];
nuclear@74: 			top[j] = bot[j];
nuclear@74: 			bot[j] = tmp;
nuclear@74: 		}
nuclear@74: 		top += xsz * 3;
nuclear@74: 		bot -= xsz * 3;
nuclear@74: 	}
nuclear@74: 
nuclear@74: 	unsigned char *pal = img + xsz * ysz * 3;
nuclear@74: 	memset(pal, 0, nslices * xsz * 3);
nuclear@74: 
nuclear@74: 	for(int i=0; i<nslices; i++) {
nuclear@74: 		memcpy(pal, palette + i * slice_bytes, slice_bytes);
nuclear@74: 		pal += xsz * 3;
nuclear@74: 	}
nuclear@74: 
nuclear@74: 	int res = img_save_pixels(fname, img, xsz, ysz + nslices, IMG_FMT_RGB24);
nuclear@74: 	delete [] img;
nuclear@74: 
nuclear@74: 	return res == 0;
nuclear@72: }
nuclear@72: 
nuclear@72: bool GradePalette::load_shot(const char *fname)
nuclear@72: {
nuclear@72: 	return false;	// TODO
nuclear@72: }
nuclear@72: 
nuclear@72: unsigned int GradePalette::get_texture() const
nuclear@72: {
nuclear@72: 	return tex;
nuclear@72: }