Fix an issue where a (cached) shader program could be used from multiple

[movit] / ycbcr_input.h

diff --git a/ycbcr_input.h b/ycbcr_input.h
index 282b55aaac2d1fe19dd3ef7bae9649e04ab1a54a..d0e71e277e27a235e759027e3a39b06e4990354c 100644 (file)
--- a/ycbcr_input.h
+++ b/ycbcr_input.h
@@ -79,12 +79,50 @@ public:
 
        void invalidate_pixel_data();
 
-       void set_pitch(unsigned channel, unsigned pitch) {
+       // Note: Sets pitch to width, so even if your pitch is unchanged,
+       // you will need to re-set it after this call.
+       void set_width(unsigned width)
+       {
+               assert(width != 0);
+               this->width = width;
+
+               assert(width % ycbcr_format.chroma_subsampling_x == 0);
+               pitch[0] = widths[0] = width;
+               pitch[1] = widths[1] = width / ycbcr_format.chroma_subsampling_x;
+               pitch[2] = widths[2] = width / ycbcr_format.chroma_subsampling_x;
+               invalidate_pixel_data();
+       }
+
+       void set_height(unsigned height)
+       {
+               assert(height != 0);
+               this->height = height;
+
+               assert(height % ycbcr_format.chroma_subsampling_y == 0);
+               heights[0] = height;
+               heights[1] = height / ycbcr_format.chroma_subsampling_y;
+               heights[2] = height / ycbcr_format.chroma_subsampling_y;
+               invalidate_pixel_data();
+       }
+
+       void set_pitch(unsigned channel, unsigned pitch)
+       {
+               assert(pitch != 0);
                assert(channel >= 0 && channel < num_channels);
                this->pitch[channel] = pitch;
                invalidate_pixel_data();
        }
 
+       // Tells the input to use the specific OpenGL texture as pixel data for the given
+       // channel. The comments on FlatInput::set_texture_num() also apply here, except
+       // that this input generally does not use mipmaps.
+       void set_texture_num(unsigned channel, GLuint texture_num)
+       {
+               possibly_release_texture(channel);
+               this->texture_num[channel] = texture_num;
+               this->owns_texture[channel] = false;
+       }
+
        virtual void inform_added(EffectChain *chain)
        {
                resource_pool = chain->get_resource_pool();
@@ -93,6 +131,9 @@ public:
        bool set_int(const std::string& key, int value);
 
 private:
+       // Release the texture in the given channel if we have any, and it is owned by us.
+       void possibly_release_texture(unsigned channel);
+
        ImageFormat image_format;
        YCbCrFormat ycbcr_format;
        GLuint num_channels;
@@ -103,6 +144,7 @@ private:
        unsigned width, height, widths[3], heights[3];
        const unsigned char *pixel_data[3];
        unsigned pitch[3];
+       bool owns_texture[3];
        ResourcePool *resource_pool;
 };