X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=main.cpp;h=3b2bcad7961dc53c5bd13ed8a8703602c2b5bfd3;hb=a88f299483ffe5068cd2828513078b9103325da8;hp=84837395879a43f10f4a897f772846d3e757966a;hpb=e61807327b9a1f98f39dd5e1496254905f78e581;p=movit

diff --git a/main.cpp b/main.cpp
index 8483739..3b2bcad 100644
--- a/main.cpp
+++ b/main.cpp
@@ -22,6 +22,7 @@
 #include <GL/glext.h>
 
 #include "effect.h"
+#include "effect_chain.h"
 #include "util.h"
 #include "widgets.h"
 #include "texture_enum.h"
@@ -37,165 +38,6 @@ float lift_r = 0.0f, lift_g = 0.0f, lift_b = 0.0f;
 float gamma_r = 1.0f, gamma_g = 1.0f, gamma_b = 1.0f;
 float gain_r = 1.0f, gain_g = 1.0f, gain_b = 1.0f;
 
-enum PixelFormat { FORMAT_RGB, FORMAT_RGBA };
-
-enum ColorSpace {
-	COLORSPACE_sRGB = 0,
-	COLORSPACE_REC_709 = 0,  // Same as sRGB.
-	COLORSPACE_REC_601_525 = 1,
-	COLORSPACE_REC_601_625 = 2,
-};
-
-enum GammaCurve {
-	GAMMA_LINEAR = 0,
-	GAMMA_sRGB = 1,
-	GAMMA_REC_601 = 2,
-	GAMMA_REC_709 = 2,  // Same as Rec. 601.
-};
-
-struct ImageFormat {
-	PixelFormat pixel_format;
-	ColorSpace color_space;
-	GammaCurve gamma_curve;
-};
-
-enum EffectId {
-	// Mostly for internal use.
-	GAMMA_CONVERSION = 0,
-	RGB_PRIMARIES_CONVERSION,
-
-	// Color.
-	LIFT_GAMMA_GAIN,
-};
-
-// Can alias on a float[3].
-struct RGBTriplet {
-	RGBTriplet(float r, float g, float b)
-		: r(r), g(g), b(b) {}
-
-	float r, g, b;
-};
-
-class GammaExpansionEffect : public Effect {
-public:
-	GammaExpansionEffect()
-		: source_curve(GAMMA_LINEAR)
-	{
-		register_int("source_curve", (int *)&source_curve);
-	}
-
-private:
-	GammaCurve source_curve;
-};
-
-class ColorSpaceConversionEffect : public Effect {
-public:
-	ColorSpaceConversionEffect()
-		: source_space(COLORSPACE_sRGB),
-		  destination_space(COLORSPACE_sRGB)
-	{
-		register_int("source_space", (int *)&source_space);
-		register_int("destination_space", (int *)&destination_space);
-	}
-
-private:
-	ColorSpace source_space, destination_space;
-};
-
-class ColorSpaceConversionEffect;
-
-class LiftGammaGainEffect : public Effect {
-public:
-	LiftGammaGainEffect()
-		: lift(0.0f, 0.0f, 0.0f),
-		  gamma(1.0f, 1.0f, 1.0f),
-		  gain(1.0f, 1.0f, 1.0f),
-		  saturation(1.0f)
-	{
-		register_vec3("lift", (float *)&lift);
-		register_vec3("gamma", (float *)&gamma);
-		register_vec3("gain", (float *)&gain);
-		register_float("saturation", &saturation);
-	}
-
-private:
-	RGBTriplet lift, gamma, gain;
-	float saturation;
-};
-
-class EffectChain {
-public:
-	EffectChain(unsigned width, unsigned height);
-	void add_input(const ImageFormat &format);
-
-	// The pointer is owned by EffectChain.
-	Effect *add_effect(EffectId effect);
-
-	void add_output(const ImageFormat &format);
-
-	void render(unsigned char *src, unsigned char *dst);
-
-private:
-	unsigned width, height;
-	ImageFormat input_format, output_format;
-	std::vector<Effect *> effects;
-
-	ColorSpace current_color_space;
-	GammaCurve current_gamma_curve;	
-};
-
-EffectChain::EffectChain(unsigned width, unsigned height)
-	: width(width), height(height) {}
-
-void EffectChain::add_input(const ImageFormat &format)
-{
-	input_format = format;
-	current_color_space = format.color_space;
-	current_gamma_curve = format.gamma_curve;
-}
-
-void EffectChain::add_output(const ImageFormat &format)
-{
-	output_format = format;
-}
-	
-Effect *instantiate_effect(EffectId effect)
-{
-	switch (effect) {
-	case GAMMA_CONVERSION:
-		return new GammaExpansionEffect();
-	case RGB_PRIMARIES_CONVERSION:
-		return new GammaExpansionEffect();
-	case LIFT_GAMMA_GAIN:
-		return new LiftGammaGainEffect();
-	}
-	assert(false);
-}
-
-Effect *EffectChain::add_effect(EffectId effect_id)
-{
-	Effect *effect = instantiate_effect(effect_id);
-
-	if (effect->needs_linear_light() && current_gamma_curve != GAMMA_LINEAR) {
-		GammaExpansionEffect *gamma_conversion = new GammaExpansionEffect();
-		gamma_conversion->set_int("source_curve", current_gamma_curve);
-		effects.push_back(gamma_conversion);
-		current_gamma_curve = GAMMA_LINEAR;
-	}
-
-	if (effect->needs_srgb_primaries() && current_color_space != COLORSPACE_sRGB) {
-		assert(current_gamma_curve == GAMMA_LINEAR);
-		ColorSpaceConversionEffect *colorspace_conversion = new ColorSpaceConversionEffect();
-		colorspace_conversion->set_int("source_space", current_color_space);
-		colorspace_conversion->set_int("destination_space", COLORSPACE_sRGB);
-		effects.push_back(colorspace_conversion);
-		current_color_space = COLORSPACE_sRGB;
-	}
-
-	effects.push_back(effect);
-	return effect;
-}
-
 GLhandleARB read_shader(const char* filename, GLenum type)
 {
 	static char buf[131072];