diff options
Diffstat (limited to 'libhb/render.c')
| -rw-r--r-- | libhb/render.c | 167 |
1 files changed, 167 insertions, 0 deletions
diff --git a/libhb/render.c b/libhb/render.c new file mode 100644 index 000000000..914fa3995 --- /dev/null +++ b/libhb/render.c @@ -0,0 +1,167 @@ +/* $Id: render.c,v 1.17 2005/04/14 17:37:54 titer Exp $ + + This file is part of the HandBrake source code. + Homepage: <http://handbrake.m0k.org/>. + It may be used under the terms of the GNU General Public License. */ + +#include "hb.h" + +#include "ffmpeg/avcodec.h" + +struct hb_work_object_s +{ + HB_WORK_COMMON; + + hb_job_t * job; + + ImgReSampleContext * context; + AVPicture pic_raw; + AVPicture pic_deint; + AVPicture pic_render; + hb_buffer_t * buf_deint; +}; + +static void ApplySub( hb_job_t * job, hb_buffer_t * buf, + hb_buffer_t ** _sub ) +{ + hb_buffer_t * sub = *_sub; + hb_title_t * title = job->title; + int i, j, offset_top, offset_left; + uint8_t * lum, * alpha, * out; + + if( !sub ) + { + return; + } + + if( sub->width > title->width - job->crop[0] - job->crop[1] - 40 || + sub->height > title->height - job->crop[2] - job->crop[3] - 40 ) + { + /* The subtitle won't fit */ + hb_buffer_close( _sub ); + return; + } + + /* If necessary, move the subtitle so it is 20 pixels far from each + border of the cropped picture */ + offset_top = sub->y; + offset_top = MAX( offset_top, job->crop[0] + 20 ); + offset_top = MIN( offset_top, + title->height - job->crop[1] - 20 - sub->height ); + offset_left = sub->x; + offset_left = MAX( offset_left, job->crop[2] + 20 ); + offset_left = MIN( offset_left, + title->width - job->crop[3] - 20 - sub->width ); + + lum = sub->data; + alpha = lum + sub->width * sub->height; + out = buf->data + offset_top * title->width + offset_left; + + for( i = 0; i < sub->height; i++ ) + { + for( j = 0; j < sub->width; j++ ) + { + out[j] = ( (uint16_t) out[j] * ( 16 - (uint16_t) alpha[j] ) + + (uint16_t) lum[j] * (uint16_t) alpha[j] ) >> 4; + } + lum += sub->width; + alpha += sub->width; + out += title->width; + } + + hb_buffer_close( _sub ); +} + +static int Work( hb_work_object_t * w, hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out ) +{ + hb_job_t * job = w->job; + hb_title_t * title = job->title; + hb_buffer_t * in = *buf_in, * buf; + + avpicture_fill( &w->pic_raw, in->data, PIX_FMT_YUV420P, + title->width, title->height ); + + buf = hb_buffer_init( 3 * job->width * job->height / 2 ); + buf->start = in->start; + buf->stop = in->stop; + + if( job->deinterlace && w->context ) + { + avpicture_fill( &w->pic_render, buf->data, PIX_FMT_YUV420P, + job->width, job->height ); + avpicture_deinterlace( &w->pic_deint, &w->pic_raw, + PIX_FMT_YUV420P, title->width, + title->height ); + ApplySub( job, w->buf_deint, &in->sub ); + img_resample( w->context, &w->pic_render, &w->pic_deint ); + } + else if( job->deinterlace ) + { + avpicture_fill( &w->pic_deint, buf->data, PIX_FMT_YUV420P, + job->width, job->height ); + avpicture_deinterlace( &w->pic_deint, &w->pic_raw, + PIX_FMT_YUV420P, title->width, + title->height ); + ApplySub( job, buf, &in->sub ); + } + else if( w->context ) + { + ApplySub( job, in, &in->sub ); + avpicture_fill( &w->pic_render, buf->data, PIX_FMT_YUV420P, + job->width, job->height ); + img_resample( w->context, &w->pic_render, &w->pic_raw ); + } + else + { + hb_buffer_close( &buf ); + ApplySub( job, in, &in->sub ); + buf = in; + *buf_in = NULL; + } + + (*buf_out) = buf; + + return HB_WORK_OK; +} + +static void Close( hb_work_object_t ** _w ) +{ + hb_work_object_t * w = *_w; + free( w->name ); + free( w ); + *_w = NULL; +} + +hb_work_object_t * hb_work_render_init( hb_job_t * job ) +{ + hb_title_t * title; + + hb_work_object_t * w = calloc( sizeof( hb_work_object_t ), 1 ); + w->name = strdup( "Renderer" ); + w->work = Work; + w->close = Close; + + title = job->title; + + w->job = job; + + if( job->crop[0] || job->crop[1] || job->crop[2] || job->crop[3] || + job->width != title->width || job->height != title->height ) + { + w->context = img_resample_full_init( + job->width, job->height, title->width, title->height, + job->crop[0], job->crop[1], job->crop[2], job->crop[3], + 0, 0, 0, 0 ); + } + + if( job->deinterlace ) + { + /* Allocate a constant buffer used for deinterlacing */ + w->buf_deint = hb_buffer_init( 3 * title->width * + title->height / 2 ); + avpicture_fill( &w->pic_deint, w->buf_deint->data, + PIX_FMT_YUV420P, title->width, title->height ); + } + return w; +} |