X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=flat_input.cpp;h=5978ca3b51c4116441933537a6168707f210d44a;hp=0a8139224c2feb037fa177fa31f32dfcbcde627c;hb=bd98e07fa5ba3cb783e8e9d16ab4cc93a9fa3100;hpb=358df88f77e4f8ae2ead71a06dc36bf5402d9b5d

diff --git a/flat_input.cpp b/flat_input.cpp
index 0a81392..5978ca3 100644
--- a/flat_input.cpp
+++ b/flat_input.cpp
@@ -1,19 +1,22 @@
 #include <string.h>
 #include <assert.h>
-#include <GL/glew.h>
+#include <epoxy/gl.h>
 
 #include "effect_util.h"
 #include "flat_input.h"
 #include "resource_pool.h"
 #include "util.h"
 
+using namespace std;
+
+namespace movit {
+
 FlatInput::FlatInput(ImageFormat image_format, MovitPixelFormat pixel_format, GLenum type, unsigned width, unsigned height)
 	: image_format(image_format),
           pixel_format(pixel_format),
 	  type(type),
 	  pbo(0),
 	  texture_num(0),
-	  finalized(false),
 	  output_linear_gamma(false),
 	  needs_mipmaps(false),
 	  width(width),
@@ -21,7 +24,7 @@ FlatInput::FlatInput(ImageFormat image_format, MovitPixelFormat pixel_format, GL
 	  pitch(width),
 	  pixel_data(NULL)
 {
-	assert(type == GL_FLOAT || type == GL_UNSIGNED_BYTE);
+	assert(type == GL_FLOAT || type == GL_HALF_FLOAT || type == GL_UNSIGNED_SHORT || type == GL_UNSIGNED_BYTE);
 	register_int("output_linear_gamma", &output_linear_gamma);
 	register_int("needs_mipmaps", &needs_mipmaps);
 }
@@ -33,43 +36,70 @@ FlatInput::~FlatInput()
 	}
 }
 
-void FlatInput::finalize()
-{
-	// Translate the input format to OpenGL's enums.
-	if (type == GL_FLOAT) {
-		internal_format = GL_RGBA32F_ARB;
-	} else if (output_linear_gamma) {
-		assert(type == GL_UNSIGNED_BYTE);
-		internal_format = GL_SRGB8_ALPHA8;
-	} else {
-		assert(type == GL_UNSIGNED_BYTE);
-		internal_format = GL_RGBA8;
-	}
-	if (pixel_format == FORMAT_RGB) {
-		format = GL_RGB;
-	} else if (pixel_format == FORMAT_RGBA_PREMULTIPLIED_ALPHA ||
-	           pixel_format == FORMAT_RGBA_POSTMULTIPLIED_ALPHA) {
-		format = GL_RGBA;
-	} else if (pixel_format == FORMAT_BGR) {
-		format = GL_BGR;
-	} else if (pixel_format == FORMAT_BGRA_PREMULTIPLIED_ALPHA ||
-	           pixel_format == FORMAT_BGRA_POSTMULTIPLIED_ALPHA) {
-		format = GL_BGRA;
-	} else if (pixel_format == FORMAT_GRAYSCALE) {
-		format = GL_LUMINANCE;
-	} else {
-		assert(false);
-	}
-
-	finalized = true;
-}
-	
-void FlatInput::set_gl_state(GLuint glsl_program_num, const std::string& prefix, unsigned *sampler_num)
+void FlatInput::set_gl_state(GLuint glsl_program_num, const string& prefix, unsigned *sampler_num)
 {
 	glActiveTexture(GL_TEXTURE0 + *sampler_num);
 	check_error();
 
 	if (texture_num == 0) {
+		// Translate the input format to OpenGL's enums.
+		GLint internal_format;
+		GLenum format;
+		if (type == GL_FLOAT) {
+			if (pixel_format == FORMAT_R) {
+				internal_format = GL_R32F;
+			} else if (pixel_format == FORMAT_RG) {
+				internal_format = GL_RG32F;
+			} else {
+				internal_format = GL_RGBA32F;
+			}
+		} else if (type == GL_HALF_FLOAT) {
+			if (pixel_format == FORMAT_R) {
+				internal_format = GL_R16F;
+			} else if (pixel_format == FORMAT_RG) {
+				internal_format = GL_RG16F;
+			} else {
+				internal_format = GL_RGBA16F;
+			}
+		} else if (type == GL_UNSIGNED_SHORT) {
+			if (pixel_format == FORMAT_R) {
+				internal_format = GL_R16;
+			} else if (pixel_format == FORMAT_RG) {
+				internal_format = GL_RG16;
+			} else {
+				internal_format = GL_RGBA16;
+			}
+		} else if (output_linear_gamma) {
+			assert(type == GL_UNSIGNED_BYTE);
+			internal_format = GL_SRGB8_ALPHA8;
+		} else {
+			assert(type == GL_UNSIGNED_BYTE);
+			if (pixel_format == FORMAT_R) {
+				internal_format = GL_R8;
+			} else if (pixel_format == FORMAT_RG) {
+				internal_format = GL_RG8;
+			} else {
+				internal_format = GL_RGBA8;
+			}
+		}
+		if (pixel_format == FORMAT_RGB) {
+			format = GL_RGB;
+		} else if (pixel_format == FORMAT_RGBA_PREMULTIPLIED_ALPHA ||
+			   pixel_format == FORMAT_RGBA_POSTMULTIPLIED_ALPHA) {
+			format = GL_RGBA;
+		} else if (pixel_format == FORMAT_BGR) {
+			format = GL_BGR;
+		} else if (pixel_format == FORMAT_BGRA_PREMULTIPLIED_ALPHA ||
+			   pixel_format == FORMAT_BGRA_POSTMULTIPLIED_ALPHA) {
+			format = GL_BGRA;
+		} else if (pixel_format == FORMAT_GRAYSCALE) {
+			format = GL_LUMINANCE;
+		} else if (pixel_format == FORMAT_RG) {
+			format = GL_RG;
+		} else {
+			assert(false);
+		}
+
 		// (Re-)upload the texture.
 		texture_num = resource_pool->create_2d_texture(internal_format, width, height);
 		glBindTexture(GL_TEXTURE_2D, texture_num);
@@ -108,7 +138,7 @@ void FlatInput::set_gl_state(GLuint glsl_program_num, const std::string& prefix,
 	++*sampler_num;
 }
 
-std::string FlatInput::output_fragment_shader()
+string FlatInput::output_fragment_shader()
 {
 	return read_file("flat_input.frag");
 }
@@ -120,3 +150,5 @@ void FlatInput::invalidate_pixel_data()
 		texture_num = 0;
 	}
 }
+
+}  // namespace movit