#include "hb.h"
-#include "ffmpeg/avcodec.h"
-#include "ffmpeg/swscale.h"
+#include "libavcodec/avcodec.h"
+#include "libavformat/avformat.h"
+#include "libswscale/swscale.h"
struct hb_handle_s
{
h->pause_lock = hb_lock_init();
/* libavcodec */
- avcodec_init();
- avcodec_register_all();
- av_register_codec_parser( &mpegaudio_parser);
+ av_register_all();
/* Start library thread */
hb_log( "hb_init: starting libhb thread" );
hb_register( &hb_deca52 );
hb_register( &hb_decdca );
hb_register( &hb_decavcodec );
+ hb_register( &hb_decavcodecv );
+ hb_register( &hb_decavcodecvi );
+ hb_register( &hb_decavcodecai );
hb_register( &hb_declpcm );
hb_register( &hb_encfaac );
hb_register( &hb_enclame );
* @param color_equal Sensitivity for detecting similar colors
* @param color_diff Sensitivity for detecting different colors
* @param threshold Sensitivity for flagging planes as combed
+ * @param prog_equal Sensitivity for detecting similar colors on progressive frames
+ * @param prog_diff Sensitivity for detecting different colors on progressive frames
+ * @param prog_threshold Sensitivity for flagging progressive frames as combed
*/
-int hb_detect_comb( hb_buffer_t * buf, int width, int height, int color_equal, int color_diff, int threshold )
+int hb_detect_comb( hb_buffer_t * buf, int width, int height, int color_equal, int color_diff, int threshold, int prog_equal, int prog_diff, int prog_threshold )
{
- int j, k, n, off, block, cc_1, cc_2, cc[3], flag[3];
+ int j, k, n, off, cc_1, cc_2, cc[3], flag[3] ;
uint16_t s1, s2, s3, s4;
cc_1 = 0; cc_2 = 0;
int offset = 0;
+
+ if ( buf->flags & 16 )
+ {
+ /* Frame is progressive, be more discerning. */
+ color_diff = prog_diff;
+ color_equal = prog_equal;
+ threshold = prog_threshold;
+ }
+ /* One pas for Y, one pass for Cb, one pass for Cr */
for( k = 0; k < 3; k++ )
{
- /* One pas for Y, one pass for Cb, one pass for Cr */
-
if( k == 1 )
{
/* Y has already been checked, now offset by Y's dimensions
offset = width * height;
width >>= 1;
height >>= 1;
- threshold >>= 1;
- color_equal >>= 1;
- color_diff >>= 1;
}
else if ( k == 2 )
{
offset *= 5/4;
}
- /* Look at one horizontal line at a time */
- block = width;
-
- for( j = 0; j < block; ++j )
+ for( j = 0; j < width; ++j )
{
off = 0;
{
/* Look at groups of 4 sequential horizontal lines */
s1 = ( ( buf->data + offset )[ off + j ] & 0xff );
- s2 = ( ( buf->data + offset )[ off + j + block ] & 0xff );
- s3 = ( ( buf->data + offset )[ off + j + 2 * block ] & 0xff );
- s4 = ( ( buf->data + offset )[ off + j + 3 * block ] & 0xff );
+ s2 = ( ( buf->data + offset )[ off + j + width ] & 0xff );
+ s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
+ s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
/* Note if the 1st and 2nd lines are more different in
color than the 1st and 3rd lines are similar in color.*/
++cc_2;
/* Now move down 2 horizontal lines before starting over.*/
- off += 2 * block;
+ off += 2 * width;
}
}
// compare results
- /* The final metric seems to be doing some kind of bits per pixel style calculation
- to decide whether or not enough lines showed alternating colors for the frame size. */
+ /* The final cc score for a plane is the percentage of combed pixels it contains.
+ Because sensitivity goes down to hundreths of a percent, multiply by 1000
+ so it will be easy to compare against the threhold value which is an integer. */
cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
-
- /* If the plane's cc score meets the threshold, flag it as combed. */
- flag[k] = 0;
- if ( cc[k] > threshold )
- {
- flag[k] = 1;
- }
}
-#if 0
-/* Debugging info */
-// if(flag)
- hb_log("flags: %i/%i/%i | cc0: %i | cc1: %i | cc2: %i", flag[0], flag[1], flag[2], cc[0], cc[1], cc[2]);
-#endif
- /* When more than one plane shows combing, tell the caller. */
- if (flag[0] || flag[1] || flag[2] )
+ /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
+ int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
+
+ /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
+ if( average_cc > threshold )
{
+#if 0
+ hb_log("Average %i combed (Threshold %i) %i/%i/%i | PTS: %lld (%fs) %s", average_cc, threshold, cc[0], cc[1], cc[2], buf->start, (float)buf->start / 90000, (buf->flags & 16) ? "Film" : "Video" );
+#endif
return 1;
}
+#if 0
+ hb_log("SKIPPED Average %i combed (Threshold %i) %i/%i/%i | PTS: %lld (%fs) %s", average_cc, threshold, cc[0], cc[1], cc[2], buf->start, (float)buf->start / 90000, (buf->flags & 16) ? "Film" : "Video" );
+#endif
+
+ /* Reaching this point means no combing detected. */
return 0;
-}
+}
/**
* Calculates job width and height for anamorphic content,
int pixel_aspect_width = job->pixel_aspect_width;
int pixel_aspect_height = job->pixel_aspect_height;
-
+
+ /* If a source was really 704*480 and hard matted with cropping
+ to 720*480, replace the PAR values with the ITU broadcast ones. */
if (cropped_width <= 706)
{
/* Handle ITU PARs */
pixel_aspect_width = 40;
pixel_aspect_height = 33;
}
- else
+ else if (aspect == 12)
{
/* It's 4:3 */
pixel_aspect_width = 10;
pixel_aspect_width = 16;
pixel_aspect_height = 11;
}
- else
+ else if (aspect == 12)
{
/* It's 4:3 */
pixel_aspect_width = 12;
OSDN Git Service
