Be more defensive about width/height/pitch given to FlatInput and YCbCrInput.

diff --git a/flat_input.h b/flat_input.h
index b337788f4edaa328c1c5f1a00f2b6b4787ff6245..bcaca50012facb2dcfb3d77879b11d99cac9a456 100644 (file)
--- a/flat_input.h
+++ b/flat_input.h
@@ -109,17 +109,20 @@ public:
        // you will need to re-set it after this call.
        void set_width(unsigned width)
        {
+               assert(width != 0);
                this->pitch = this->width = width;
                invalidate_pixel_data();
        }
 
        void set_height(unsigned height)
        {
+               assert(height != 0);
                this->height = height;
                invalidate_pixel_data();
        }
 
        void set_pitch(unsigned pitch) {
+               assert(pitch != 0);
                this->pitch = pitch;
                invalidate_pixel_data();
        }

diff --git a/ycbcr_input.h b/ycbcr_input.h
index 3aed51b3b19a90d9e0b0e03e6cfdc16fcfab726f..d0e71e277e27a235e759027e3a39b06e4990354c 100644 (file)
--- a/ycbcr_input.h
+++ b/ycbcr_input.h
@@ -83,6 +83,7 @@ public:
        // 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);
@@ -94,6 +95,7 @@ public:
 
        void set_height(unsigned height)
        {
+               assert(height != 0);
                this->height = height;
 
                assert(height % ycbcr_format.chroma_subsampling_y == 0);
@@ -105,6 +107,7 @@ public:
 
        void set_pitch(unsigned channel, unsigned pitch)
        {
+               assert(pitch != 0);
                assert(channel >= 0 && channel < num_channels);
                this->pitch[channel] = pitch;
                invalidate_pixel_data();