X-Git-Url: http://research.m1stereo.tv/gitweb?a=blobdiff_plain;f=src%2Fmodules%2Fcore%2Ffilter_resize.c;h=bd8841b07cc1c5aa610fe16a17178ad009b485de;hb=b6232c93db568b3becc00019ca18e31bd92e54b5;hp=dab7fdfc04d0141f91dbe017ecf2f60c304a820b;hpb=bbc63e064f2874ea16a4c8cf0e6331fb73f3ffb5;p=melted diff --git a/src/modules/core/filter_resize.c b/src/modules/core/filter_resize.c index dab7fdf..bd8841b 100644 --- a/src/modules/core/filter_resize.c +++ b/src/modules/core/filter_resize.c @@ -31,18 +31,123 @@ static int filter_get_image( mlt_frame this, uint8_t **image, mlt_image_format *format, int *width, int *height, int writable ) { + int error = 0; + + // Get the properties from the frame + mlt_properties properties = MLT_FRAME_PROPERTIES( this ); + + // Pop the top of stack now + mlt_filter filter = mlt_frame_pop_service( this ); + + // Retrieve the aspect ratio + double aspect_ratio = mlt_deque_pop_back_double( MLT_FRAME_IMAGE_STACK( this ) ); + + // Assign requested width/height from our subordinate int owidth = *width; int oheight = *height; - mlt_frame_get_image( this, image, format, &owidth, &oheight, 0 ); - if ( *width == 0 ) - *width = 720; - if ( *height == 0 ) - *height = 576; - if ( !strcmp( mlt_properties_get( mlt_frame_properties( this ), "resize.scale" ), "affine" ) ) - *image = mlt_frame_rescale_yuv422( this, *width, *height ); - else - *image = mlt_frame_resize_yuv422( this, *width, *height ); - return 0; + + // Check for the special case - no aspect ratio means no problem :-) + if ( aspect_ratio == 0.0 ) + aspect_ratio = mlt_properties_get_double( properties, "consumer_aspect_ratio" ); + + // Reset the aspect ratio + mlt_properties_set_double( properties, "aspect_ratio", aspect_ratio ); + + // Hmmm... + char *rescale = mlt_properties_get( properties, "rescale.interp" ); + if ( rescale != NULL && !strcmp( rescale, "none" ) ) + return mlt_frame_get_image( this, image, format, width, height, writable ); + + if ( mlt_properties_get_int( properties, "distort" ) == 0 ) + { + // Normalise the input and out display aspect + int normalised_width = mlt_properties_get_int( properties, "normalised_width" ); + int normalised_height = mlt_properties_get_int( properties, "normalised_height" ); + int real_width = mlt_properties_get_int( properties, "real_width" ); + int real_height = mlt_properties_get_int( properties, "real_height" ); + if ( real_width == 0 ) + real_width = mlt_properties_get_int( properties, "width" ); + if ( real_height == 0 ) + real_height = mlt_properties_get_int( properties, "height" ); + double input_ar = aspect_ratio * real_width / real_height; + double output_ar = mlt_properties_get_double( properties, "consumer_aspect_ratio" ) * owidth / oheight; + + //fprintf( stderr, "normalised %dx%d output %dx%d %f %f\n", normalised_width, normalised_height, owidth, oheight, ( float )output_ar, ( float )mlt_properties_get_double( properties, "consumer_aspect_ratio" ) * owidth / oheight ); + + // Optimised for the input_ar > output_ar case (e.g. widescreen on standard) + int scaled_width = input_ar / output_ar * normalised_width + 0.5; + int scaled_height = normalised_height; + + // Now ensure that our images fit in the output frame + if ( scaled_width > normalised_width ) + { + scaled_width = normalised_width; + scaled_height = output_ar / input_ar * normalised_height + 0.5; + } + + // Now calculate the actual image size that we want + owidth = scaled_width * owidth / normalised_width; + oheight = scaled_height * oheight / normalised_height; + + // Tell frame we have conformed the aspect to the consumer + mlt_frame_set_aspect_ratio( this, mlt_properties_get_double( properties, "consumer_aspect_ratio" ) ); + } + + mlt_properties_set_int( properties, "distort", 0 ); + + // Now pass on the calculations down the line + mlt_properties_set_int( properties, "resize_width", *width ); + mlt_properties_set_int( properties, "resize_height", *height ); + + // Now get the image + error = mlt_frame_get_image( this, image, format, &owidth, &oheight, writable ); + + // We only know how to process yuv422 at the moment + if ( error == 0 && *format == mlt_image_yuv422 && *image != NULL ) + { + // Get the requested scale operation + char *op = mlt_properties_get( MLT_FILTER_PROPERTIES( filter ), "scale" ); + + // Correct field order if needed + if ( mlt_properties_get_int( properties, "top_field_first" ) == 1 ) + { + // Get the input image, width and height + int size; + uint8_t *image = mlt_properties_get_data( properties, "image", &size ); + + // Keep the original image around to be destroyed on frame close + mlt_properties_rename( properties, "image", "original_image" ); + + // Duplicate the last line in the field to avoid artifact + memcpy( image + oheight * owidth * 2, image + oheight * owidth * 2 - owidth * 4, owidth * 2 ); + + // Offset the image pointer by one line + image += owidth * 2; + size -= owidth * 2; + + // Set the new image pointer with no destructor + mlt_properties_set_data( properties, "image", image, size, NULL, NULL ); + + // Set the normalised field order + mlt_properties_set_int( properties, "top_field_first", 0 ); + } + + if ( !strcmp( op, "affine" ) ) + { + *image = mlt_frame_rescale_yuv422( this, *width, *height ); + } + else if ( strcmp( op, "none" ) != 0 ) + { + *image = mlt_frame_resize_yuv422( this, *width, *height ); + } + else + { + *width = owidth; + *height = oheight; + } + } + + return error; } /** Filter processing. @@ -50,22 +155,28 @@ static int filter_get_image( mlt_frame this, uint8_t **image, mlt_image_format * static mlt_frame filter_process( mlt_filter this, mlt_frame frame ) { + // Store the aspect ratio reported by the source + mlt_deque_push_back_double( MLT_FRAME_IMAGE_STACK( frame ), mlt_frame_get_aspect_ratio( frame ) ); + + // Push this on to the service stack + mlt_frame_push_service( frame, this ); + + // Push the get_image method on to the stack mlt_frame_push_get_image( frame, filter_get_image ); - mlt_properties_set( mlt_frame_properties( frame ), "resize.scale", mlt_properties_get( mlt_filter_properties( this ), "scale" ) ); + return frame; } /** Constructor for the filter. */ -mlt_filter filter_resize_init( void *arg ) +mlt_filter filter_resize_init( char *arg ) { mlt_filter this = calloc( sizeof( struct mlt_filter_s ), 1 ); if ( mlt_filter_init( this, this ) == 0 ) { this->process = filter_process; - mlt_properties_set( mlt_filter_properties( this ), "scale", "off" ); + mlt_properties_set( MLT_FILTER_PROPERTIES( this ), "scale", arg == NULL ? "off" : arg ); } return this; } -