summaryrefslogtreecommitdiffstats
path: root/libhb/hb.c
diff options
context:
space:
mode:
authorprigaux <[email protected]>2007-02-11 20:00:15 +0000
committerprigaux <[email protected]>2007-02-11 20:00:15 +0000
commita74fd5a1ec37e53fe7008f022639af6ede0e82c9 (patch)
tree0a6a32640a840b72653149aa11efa264674df976 /libhb/hb.c
parentf45dfeb2afb1b0bf46054959e8ea9494f3b7d14f (diff)
Merge the 0.8.0_mpeg4ip branch into the trunk
git-svn-id: svn://svn.handbrake.fr/HandBrake/trunk@285 b64f7644-9d1e-0410-96f1-a4d463321fa5
Diffstat (limited to 'libhb/hb.c')
-rw-r--r--libhb/hb.c722
1 files changed, 0 insertions, 722 deletions
diff --git a/libhb/hb.c b/libhb/hb.c
deleted file mode 100644
index 98e014f0d..000000000
--- a/libhb/hb.c
+++ /dev/null
@@ -1,722 +0,0 @@
-/* $Id: hb.c,v 1.43 2005/04/27 21:05:24 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_handle_s
-{
- /* The "Check for update" thread */
- int build;
- char version[16];
- hb_thread_t * update_thread;
-
- /* This thread's only purpose is to check other threads'
- states */
- volatile int die;
- hb_thread_t * main_thread;
- int pid;
-
- /* DVD/file scan thread */
- hb_list_t * list_title;
- hb_thread_t * scan_thread;
-
- /* The thread which processes the jobs. Others threads are launched
- from this one (see work.c) */
- hb_list_t * jobs;
- int job_count;
- volatile int work_die;
- int work_error;
- hb_thread_t * work_thread;
-
- int cpu_count;
-
- hb_lock_t * state_lock;
- hb_state_t state;
-
- int paused;
- hb_lock_t * pause_lock;
-};
-
-hb_work_object_t * hb_objects = NULL;
-
-static void thread_func( void * );
-
-/**
- * Registers work objects, by adding the work object to a liked list.
- * @param w Handle to hb_work_object_t to register.
- */
-void hb_register( hb_work_object_t * w )
-{
- w->next = hb_objects;
- hb_objects = w;
-}
-
-/**
- * libhb initialization routine.
- * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
- * @param update_check signals libhb to check for updated version from HandBrake website.
- * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
- */
-hb_handle_t * hb_init_real( int verbose, int update_check )
-{
- hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
- uint64_t date;
-
- /* See hb_log() in common.c */
- if( verbose > HB_DEBUG_NONE )
- {
- putenv( "HB_DEBUG=1" );
- av_log_set_level(AV_LOG_DEBUG);
- }
-
- /* Check for an update on the website if asked to */
- h->build = -1;
-
- if( update_check )
- {
- hb_log( "hb_init: checking for updates" );
- date = hb_get_date();
- h->update_thread = hb_update_init( &h->build, h->version );
-
- for( ;; )
- {
- if( hb_thread_has_exited( h->update_thread ) )
- {
- /* Immediate success or failure */
- hb_thread_close( &h->update_thread );
- break;
- }
- if( hb_get_date() > date + 1000 )
- {
- /* Still nothing after one second. Connection problem,
- let the thread die */
- hb_log( "hb_init: connection problem, not waiting for "
- "update_thread" );
- break;
- }
- hb_snooze( 500 );
- }
- }
-
- /* CPU count detection */
- hb_log( "hb_init: checking cpu count" );
- h->cpu_count = hb_get_cpu_count();
-
- h->list_title = hb_list_init();
- h->jobs = hb_list_init();
-
- h->state_lock = hb_lock_init();
- h->state.state = HB_STATE_IDLE;
-
- h->pause_lock = hb_lock_init();
-
- /* libavcodec */
- avcodec_init();
- register_avcodec( &mpeg4_encoder );
- register_avcodec( &mp2_decoder );
- register_avcodec( &ac3_encoder );
-
- /* Start library thread */
- hb_log( "hb_init: starting libhb thread" );
- h->die = 0;
- h->main_thread = hb_thread_init( "libhb", thread_func, h,
- HB_NORMAL_PRIORITY );
-
- return h;
-}
-
-/**
- * Returns current version of libhb.
- * @param h Handle to hb_handle_t.
- * @return character array of version number.
- */
-char * hb_get_version( hb_handle_t * h )
-{
- return HB_VERSION;
-}
-
-/**
- * Returns current build of libhb.
- * @param h Handle to hb_handle_t.
- * @return character array of build number.
- */
-int hb_get_build( hb_handle_t * h )
-{
- return HB_BUILD;
-}
-
-/**
- * Checks for needed update.
- * @param h Handle to hb_handle_t.
- * @param version Pointer to handle where version will be copied.
- * @return update indicator.
- */
-int hb_check_update( hb_handle_t * h, char ** version )
-{
- *version = ( h->build < 0 ) ? NULL : h->version;
- return h->build;
-}
-
-/**
- * Sets the cpu count to the desired value.
- * @param h Handle to hb_handle_t
- * @param cpu_count Number of CPUs to use.
- */
-void hb_set_cpu_count( hb_handle_t * h, int cpu_count )
-{
- cpu_count = MAX( 1, cpu_count );
- cpu_count = MIN( cpu_count, 8 );
- h->cpu_count = cpu_count;
-}
-
-/**
- * Initializes a scan of the by calling hb_scan_init
- * @param h Handle to hb_handle_t
- * @param path location of VIDEO_TS folder.
- * @param title_index Desired title to scan. 0 for all titles.
- */
-void hb_scan( hb_handle_t * h, const char * path, int title_index )
-{
- hb_title_t * title;
-
- /* Clean up from previous scan */
- while( ( title = hb_list_item( h->list_title, 0 ) ) )
- {
- hb_list_rem( h->list_title, title );
- hb_title_close( &title );
- }
-
- hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
- h->scan_thread = hb_scan_init( h, path, title_index, h->list_title );
-}
-
-/**
- * Returns the list of titles found.
- * @param h Handle to hb_handle_t
- * @return Handle to hb_list_t of the title list.
- */
-hb_list_t * hb_get_titles( hb_handle_t * h )
-{
- return h->list_title;
-}
-
-/**
- * Create preview image of desired title a index of picture.
- * @param h Handle to hb_handle_t.
- * @param title Handle to hb_title_t of desired title.
- * @param picture Index in title.
- * @param buffer Handle to buufer were inage will be drawn.
- */
-void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
- uint8_t * buffer )
-{
- hb_job_t * job = title->job;
- char filename[1024];
- FILE * file;
- uint8_t * buf1, * buf2, * buf3, * buf4, * pen;
- uint32_t * p32;
- AVPicture pic1, pic2, pic3, pic4;
- ImgReSampleContext * context;
- int i;
-
- buf1 = malloc( title->width * title->height * 3 / 2 );
- buf2 = malloc( title->width * title->height * 3 / 2 );
- buf3 = malloc( title->width * title->height * 3 / 2 );
- buf4 = malloc( title->width * title->height * 4 );
- avpicture_fill( &pic1, buf1, PIX_FMT_YUV420P,
- title->width, title->height );
- avpicture_fill( &pic2, buf2, PIX_FMT_YUV420P,
- title->width, title->height );
- avpicture_fill( &pic3, buf3, PIX_FMT_YUV420P,
- job->width, job->height );
- avpicture_fill( &pic4, buf4, PIX_FMT_RGBA32,
- job->width, job->height );
-
- memset( filename, 0, 1024 );
-
- hb_get_tempory_filename( h, filename, "%x%d",
- (int) title, picture );
-
- file = fopen( filename, "r" );
- if( !file )
- {
- hb_log( "hb_get_preview: fopen failed" );
- return;
- }
-
- fread( buf1, title->width * title->height * 3 / 2, 1, file );
- fclose( file );
-
- 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 )
- {
- avpicture_deinterlace( &pic2, &pic1, PIX_FMT_YUV420P,
- title->width, title->height );
- img_resample( context, &pic3, &pic2 );
- }
- else
- {
- img_resample( context, &pic3, &pic1 );
- }
- img_convert( &pic4, PIX_FMT_RGBA32, &pic3, PIX_FMT_YUV420P,
- job->width, job->height );
-
- /* Gray background */
- p32 = (uint32_t *) buffer;
- for( i = 0; i < ( title->width + 2 ) * ( title->height + 2 ); i++ )
- {
- p32[i] = 0xFF808080;
- }
-
- /* Draw the picture, centered, and draw the cropping zone */
- pen = buffer + ( title->height - job->height ) *
- ( title->width + 2 ) * 2 + ( title->width - job->width ) * 2;
- memset( pen, 0xFF, 4 * ( job->width + 2 ) );
- pen += 4 * ( title->width + 2 );
- for( i = 0; i < job->height; i++ )
- {
- uint8_t * nextLine;
- nextLine = pen + 4 * ( title->width + 2 );
- memset( pen, 0xFF, 4 );
- pen += 4;
- memcpy( pen, buf4 + 4 * job->width * i, 4 * job->width );
- pen += 4 * job->width;
- memset( pen, 0xFF, 4 );
- pen = nextLine;
- }
- memset( pen, 0xFF, 4 * ( job->width + 2 ) );
-
- free( buf1 );
- free( buf2 );
- free( buf3 );
- free( buf4 );
-}
-
-/**
- * Calculates job width, height, and cropping parameters.
- * @param job Handle to hb_job_t.
- * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
- * @param pixels Maximum desired pixel count.
- */
-void hb_set_size( hb_job_t * job, int aspect, int pixels )
-{
- hb_title_t * title = job->title;
-
- int croppedWidth = title->width - title->crop[2] - title->crop[3];
- int croppedHeight = title->height - title->crop[0] - title->crop[1];
- int croppedAspect = title->aspect * title->height * croppedWidth /
- croppedHeight / title->width;
- int addCrop;
- int i, w, h;
-
- if( aspect <= 0 )
- {
- /* Keep the best possible aspect ratio */
- aspect = croppedAspect;
- }
-
- /* Crop if necessary to obtain the desired ratio */
- memcpy( job->crop, title->crop, 4 * sizeof( int ) );
- if( aspect < croppedAspect )
- {
- /* Need to crop on the left and right */
- addCrop = croppedWidth - aspect * croppedHeight * title->width /
- title->aspect / title->height;
- if( addCrop & 3 )
- {
- addCrop = ( addCrop + 1 ) / 2;
- job->crop[2] += addCrop;
- job->crop[3] += addCrop;
- }
- else if( addCrop & 2 )
- {
- addCrop /= 2;
- job->crop[2] += addCrop - 1;
- job->crop[3] += addCrop + 1;
- }
- else
- {
- addCrop /= 2;
- job->crop[2] += addCrop;
- job->crop[3] += addCrop;
- }
- }
- else if( aspect > croppedAspect )
- {
- /* Need to crop on the top and bottom */
- addCrop = croppedHeight - croppedWidth * title->aspect *
- title->height / aspect / title->width;
- if( addCrop & 3 )
- {
- addCrop = ( addCrop + 1 ) / 2;
- job->crop[0] += addCrop;
- job->crop[1] += addCrop;
- }
- else if( addCrop & 2 )
- {
- addCrop /= 2;
- job->crop[0] += addCrop - 1;
- job->crop[1] += addCrop + 1;
- }
- else
- {
- addCrop /= 2;
- job->crop[0] += addCrop;
- job->crop[1] += addCrop;
- }
- }
-
- /* Compute a resolution from the number of pixels and aspect */
- for( i = 0;; i++ )
- {
- w = 16 * i;
- h = MULTIPLE_16( w * HB_ASPECT_BASE / aspect );
- if( w * h > pixels )
- {
- break;
- }
- }
- i--;
- job->width = 16 * i;
- job->height = MULTIPLE_16( 16 * i * HB_ASPECT_BASE / aspect );
-}
-
-/**
- * Returns the number of jobs in the queue.
- * @param h Handle to hb_handle_t.
- * @return Number of jobs.
- */
-int hb_count( hb_handle_t * h )
-{
- return hb_list_count( h->jobs );
-}
-
-/**
- * Returns handle to job at index i within the job list.
- * @param h Handle to hb_handle_t.
- * @param i Index of job.
- * @returns Handle to hb_job_t of desired job.
- */
-hb_job_t * hb_job( hb_handle_t * h, int i )
-{
- return hb_list_item( h->jobs, i );
-}
-
-/**
- * Adds a job to the job list.
- * @param h Handle to hb_handle_t.
- * @param job Handle to hb_job_t.
- */
-void hb_add( hb_handle_t * h, hb_job_t * job )
-{
- hb_job_t * job_copy;
- hb_title_t * title, * title_copy;
- hb_chapter_t * chapter, * chapter_copy;
- hb_audio_t * audio, * audio_copy;
- hb_subtitle_t * subtitle, * subtitle_copy;
- int i;
-
- /* Copy the title */
- title = job->title;
- title_copy = malloc( sizeof( hb_title_t ) );
- memcpy( title_copy, title, sizeof( hb_title_t ) );
-
- title_copy->list_chapter = hb_list_init();
- for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
- {
- chapter = hb_list_item( title->list_chapter, i );
- chapter_copy = malloc( sizeof( hb_chapter_t ) );
- memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
- hb_list_add( title_copy->list_chapter, chapter_copy );
- }
-
- /* Copy the audio track(s) we want */
- title_copy->list_audio = hb_list_init();
-
- /* Do nothing about audio during first pass */
- if( job->pass != 1 )
- {
- for( i = 0; i < 8; i++ )
- {
- if( job->audios[i] < 0 )
- {
- break;
- }
- if( ( audio = hb_list_item( title->list_audio, job->audios[i] ) ) )
- {
- audio_copy = malloc( sizeof( hb_audio_t ) );
- memcpy( audio_copy, audio, sizeof( hb_audio_t ) );
- hb_list_add( title_copy->list_audio, audio_copy );
- }
- }
- }
-
- /* Copy the subtitle we want (or not) */
- title_copy->list_subtitle = hb_list_init();
- if( ( subtitle = hb_list_item( title->list_subtitle, job->subtitle ) ) )
- {
- subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
- memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
- hb_list_add( title_copy->list_subtitle, subtitle_copy );
- }
-
- /* Copy the job */
- job_copy = calloc( sizeof( hb_job_t ), 1 );
- memcpy( job_copy, job, sizeof( hb_job_t ) );
- job_copy->title = title_copy;
- job_copy->file = strdup( job->file );
- job_copy->h = h;
- job_copy->pause = h->pause_lock;
-
- /* Add the job to the list */
- hb_list_add( h->jobs, job_copy );
-}
-
-/**
- * Removes a job from the job list.
- * @param h Handle to hb_handle_t.
- * @param job Handle to hb_job_t.
- */
-void hb_rem( hb_handle_t * h, hb_job_t * job )
-{
- hb_list_rem( h->jobs, job );
-
- /* XXX free everything XXX */
-}
-
-/**
- * Starts the conversion process.
- * Sets state to HB_STATE_WORKING.
- * calls hb_work_init, to launch work thread. Stores handle to work thread.
- * @param h Handle to hb_handle_t.
- */
-void hb_start( hb_handle_t * h )
-{
- /* XXX Hack */
- h->job_count = hb_list_count( h->jobs );
-
- hb_lock( h->state_lock );
- h->state.state = HB_STATE_WORKING;
-#define p h->state.param.working
- p.progress = 0.0;
- p.job_cur = 1;
- p.job_count = h->job_count;
- p.rate_cur = 0.0;
- p.rate_avg = 0.0;
- p.hours = -1;
- p.minutes = -1;
- p.seconds = -1;
-#undef p
- hb_unlock( h->state_lock );
-
- h->paused = 0;
-
- h->work_die = 0;
- h->work_thread = hb_work_init( h->jobs, h->cpu_count,
- &h->work_die, &h->work_error );
-}
-
-/**
- * Pauses the conversion process.
- * @param h Handle to hb_handle_t.
- */
-void hb_pause( hb_handle_t * h )
-{
- if( !h->paused )
- {
- hb_lock( h->pause_lock );
- h->paused = 1;
-
- hb_lock( h->state_lock );
- h->state.state = HB_STATE_PAUSED;
- hb_unlock( h->state_lock );
- }
-}
-
-/**
- * Resumes the conversion process.
- * @param h Handle to hb_handle_t.
- */
-void hb_resume( hb_handle_t * h )
-{
- if( h->paused )
- {
- hb_unlock( h->pause_lock );
- h->paused = 0;
- }
-}
-
-/**
- * Stops the conversion process.
- * @param h Handle to hb_handle_t.
- */
-void hb_stop( hb_handle_t * h )
-{
- h->work_die = 1;
-
- hb_resume( h );
-}
-
-/**
- * Returns the state of the conversion process.
- * @param h Handle to hb_handle_t.
- * @param s Handle to hb_state_t which to copy the state data.
- */
-void hb_get_state( hb_handle_t * h, hb_state_t * s )
-{
- hb_lock( h->state_lock );
-
- memcpy( s, &h->state, sizeof( hb_state_t ) );
- h->state.state = HB_STATE_IDLE;
-
- hb_unlock( h->state_lock );
-}
-
-/**
- * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init_real.
- * @param _h Pointer to handle to hb_handle_t.
- */
-void hb_close( hb_handle_t ** _h )
-{
- hb_handle_t * h = *_h;
- hb_title_t * title;
-
- h->die = 1;
- hb_thread_close( &h->main_thread );
-
- while( ( title = hb_list_item( h->list_title, 0 ) ) )
- {
- hb_list_rem( h->list_title, title );
- hb_title_close( &title );
- }
- hb_list_close( &h->list_title );
-
- hb_list_close( &h->jobs );
- hb_lock_close( &h->state_lock );
- hb_lock_close( &h->pause_lock );
- free( h );
- *_h = NULL;
-}
-
-/**
- * Monitors the state of the update, scan, and work threads.
- * Sets scan done state when scan thread exits.
- * Sets work done state when work thread exits.
- * @param _h Handle to hb_handle_t
- */
-static void thread_func( void * _h )
-{
- hb_handle_t * h = (hb_handle_t *) _h;
- char dirname[1024];
- DIR * dir;
- struct dirent * entry;
-
- h->pid = getpid();
-
- /* Create folder for temporary files */
- memset( dirname, 0, 1024 );
- hb_get_tempory_directory( h, dirname );
-
- hb_mkdir( dirname );
-
- while( !h->die )
- {
- /* In case the check_update thread hangs, it'll die sooner or
- later. Then, we join it here */
- if( h->update_thread &&
- hb_thread_has_exited( h->update_thread ) )
- {
- hb_thread_close( &h->update_thread );
- }
-
- /* Check if the scan thread is done */
- if( h->scan_thread &&
- hb_thread_has_exited( h->scan_thread ) )
- {
- hb_thread_close( &h->scan_thread );
-
- hb_log( "libhb: scan thread found %d valid title(s)",
- hb_list_count( h->list_title ) );
- hb_lock( h->state_lock );
- h->state.state = HB_STATE_SCANDONE;
- hb_unlock( h->state_lock );
- }
-
- /* Check if the work thread is done */
- if( h->work_thread &&
- hb_thread_has_exited( h->work_thread ) )
- {
- hb_thread_close( &h->work_thread );
-
- hb_log( "libhb: work result = %d",
- h->work_error );
- hb_lock( h->state_lock );
- h->state.state = HB_STATE_WORKDONE;
- h->state.param.workdone.error = h->work_error;
- hb_unlock( h->state_lock );
- }
-
- hb_snooze( 50 );
- }
-
- if( h->work_thread )
- {
- hb_stop( h );
- hb_thread_close( &h->work_thread );
- }
-
- /* Remove temp folder */
- dir = opendir( dirname );
- while( ( entry = readdir( dir ) ) )
- {
- char filename[1024];
- if( entry->d_name[0] == '.' )
- {
- continue;
- }
- memset( filename, 0, 1024 );
- snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
- unlink( filename );
- }
- closedir( dir );
- rmdir( dirname );
-}
-
-/**
- * Returns the PID.
- * @param h Handle to hb_handle_t
- */
-int hb_get_pid( hb_handle_t * h )
-{
- return h->pid;
-}
-
-/**
- * Sets the current state.
- * @param h Handle to hb_handle_t
- * @param s Handle to new hb_state_t
- */
-void hb_set_state( hb_handle_t * h, hb_state_t * s )
-{
- hb_lock( h->pause_lock );
- hb_lock( h->state_lock );
- memcpy( &h->state, s, sizeof( hb_state_t ) );
- if( h->state.state == HB_STATE_WORKING )
- {
- /* XXX Hack */
- h->state.param.working.job_cur =
- h->job_count - hb_list_count( h->jobs );
- h->state.param.working.job_count = h->job_count;
- }
- hb_unlock( h->state_lock );
- hb_unlock( h->pause_lock );
-}