From 16806499fc13ec0a7664186a42a48d9154b2a8a3 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Thu, 19 Feb 2004 20:29:26 +0000 Subject: [PATCH] w32threads by (Gildas Bazin ) Originally committed as revision 2803 to svn://svn.ffmpeg.org/ffmpeg/trunk --- configure | 4 +- ffmpeg.c | 10 +-- libavcodec/Makefile | 4 ++ libavcodec/pthread.c | 14 ++--- libavcodec/w32thread.c | 134 +++++++++++++++++++++++++++++++++++++++++ 5 files changed, 153 insertions(+), 13 deletions(-) create mode 100644 libavcodec/w32thread.c diff --git a/configure b/configure index 191ce3c217..eb41868f51 100755 --- a/configure +++ b/configure @@ -707,7 +707,7 @@ if (zlibVersion() != ZLIB_VERSION) return 0; } EOF -$cc -o $TMPE $TMPC -lz 2> /dev/null || zlib="no" +$cc $CFLAGS -o $TMPE $TMPC -lz 2> /dev/null || zlib="no" # $TMPE 2> /dev/null > /dev/null || zlib="no" # XXX: more tests needed - runtime test fi @@ -1165,6 +1165,8 @@ fi if test "$mingw32" = "yes" ; then echo "#define CONFIG_WIN32 1" >> $TMPH echo "CONFIG_WIN32=yes" >> config.mak + echo "HAVE_W32THREADS=yes" >> config.mak + echo "#define HAVE_W32THREADS 1" >> $TMPH echo "#ifndef __MINGW32__" >> $TMPH echo "#define __MINGW32__ 1" >> $TMPH echo "#endif" >> $TMPH diff --git a/ffmpeg.c b/ffmpeg.c index a8c3a93f6f..dec1ca6a5e 100644 --- a/ffmpeg.c +++ b/ffmpeg.c @@ -2036,7 +2036,7 @@ static void opt_sc_threshold(const char *arg) static void opt_thread_count(const char *arg) { thread_count= atoi(arg); -#ifndef HAVE_PTHREADS +#if !defined(HAVE_PTHREADS) && !defined(HAVE_W32THREADS) fprintf(stderr, "Warning: not compiled with thread support, using thread emulation\n"); #endif } @@ -2399,9 +2399,9 @@ static void opt_output_file(const char *filename) exit(1); } avcodec_get_context_defaults(&st->codec); -#ifdef HAVE_PTHREADS +#if defined(HAVE_PTHREADS) || defined(HAVE_W32THREADS) if(thread_count>1) - avcodec_pthread_init(&st->codec, thread_count); + avcodec_thread_init(&st->codec, thread_count); #endif video_enc = &st->codec; @@ -2590,9 +2590,9 @@ static void opt_output_file(const char *filename) exit(1); } avcodec_get_context_defaults(&st->codec); -#ifdef HAVE_PTHREADS +#if defined(HAVE_PTHREADS) || defined(HAVE_W32THREADS) if(thread_count>1) - avcodec_pthread_init(&st->codec, thread_count); + avcodec_thread_init(&st->codec, thread_count); #endif audio_enc = &st->codec; diff --git a/libavcodec/Makefile b/libavcodec/Makefile index f90976c0a4..a464610d6d 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -38,6 +38,10 @@ ifeq ($(HAVE_PTHREADS),yes) OBJS+= pthread.o endif +ifeq ($(HAVE_W32THREADS),yes) +OBJS+= w32thread.o +endif + ifeq ($(AMR_WB),yes) OBJS+= amr.o amrwb_float/dec_acelp.o amrwb_float/dec_dtx.o amrwb_float/dec_gain.o \ amrwb_float/dec_if.o amrwb_float/dec_lpc.o amrwb_float/dec_main.o \ diff --git a/libavcodec/pthread.c b/libavcodec/pthread.c index 6f98dc41f4..770f9a54dd 100644 --- a/libavcodec/pthread.c +++ b/libavcodec/pthread.c @@ -54,10 +54,10 @@ static void * thread_func(void *v){ } /** - * free what has been allocated by avcodec_pthread_init(). - * must be called after decoding has finished, especially dont call while avcodec_pthread_execute() is running + * free what has been allocated by avcodec_thread_init(). + * must be called after decoding has finished, especially dont call while avcodec_thread_execute() is running */ -void avcodec_pthread_free(AVCodecContext *s){ +void avcodec_thread_free(AVCodecContext *s){ ThreadContext *c= s->thread_opaque; int i; @@ -77,7 +77,7 @@ void avcodec_pthread_free(AVCodecContext *s){ av_freep(&s->thread_opaque); } -int avcodec_pthread_execute(AVCodecContext *s, int (*func)(AVCodecContext *c2, void *arg2),void **arg, int *ret, int count){ +int avcodec_thread_execute(AVCodecContext *s, int (*func)(AVCodecContext *c2, void *arg2),void **arg, int *ret, int count){ ThreadContext *c= s->thread_opaque; int i, val; @@ -107,7 +107,7 @@ int avcodec_pthread_execute(AVCodecContext *s, int (*func)(AVCodecContext *c2, v return 0; } -int avcodec_pthread_init(AVCodecContext *s, int thread_count){ +int avcodec_thread_init(AVCodecContext *s, int thread_count){ int i; ThreadContext *c; @@ -130,10 +130,10 @@ int avcodec_pthread_init(AVCodecContext *s, int thread_count){ } //printf("init done\n"); fflush(stdout); - s->execute= avcodec_pthread_execute; + s->execute= avcodec_thread_execute; return 0; fail: - avcodec_pthread_free(s); + avcodec_thread_free(s); return -1; } diff --git a/libavcodec/w32thread.c b/libavcodec/w32thread.c new file mode 100644 index 0000000000..186e302dc8 --- /dev/null +++ b/libavcodec/w32thread.c @@ -0,0 +1,134 @@ +/* + * Copyright (c) 2004 Michael Niedermayer + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + * + */ +//#define DEBUG + +#include "avcodec.h" +#include "common.h" + +#define WIN32_LEAN_AND_MEAN +#include +#include + +typedef struct ThreadContext{ + AVCodecContext *avctx; + HANDLE thread; + HANDLE work_sem; + HANDLE done_sem; + int (*func)(AVCodecContext *c, void *arg); + void *arg; + int ret; +}ThreadContext; + + +static unsigned __stdcall thread_func(void *v){ + ThreadContext *c= v; + + for(;;){ +//printf("thread_func %X enter wait\n", (int)v); fflush(stdout); + WaitForSingleObject(c->work_sem, INFINITE); +//printf("thread_func %X after wait (func=%X)\n", (int)v, (int)c->func); fflush(stdout); + if(c->func) + c->ret= c->func(c->avctx, c->arg); + else + return 0; +//printf("thread_func %X signal complete\n", (int)v); fflush(stdout); + ReleaseSemaphore(c->done_sem, 1, 0); + } + + return 0; +} + +/** + * free what has been allocated by avcodec_thread_init(). + * must be called after decoding has finished, especially dont call while avcodec_thread_execute() is running + */ +void avcodec_thread_free(AVCodecContext *s){ + ThreadContext *c= s->thread_opaque; + int i; + + for(i=0; ithread_count; i++){ + + c[i].func= NULL; + ReleaseSemaphore(c[i].work_sem, 1, 0); + WaitForSingleObject(c[i].thread, INFINITE); + if(c[i].work_sem) CloseHandle(c[i].work_sem); + if(c[i].done_sem) CloseHandle(c[i].done_sem); + } + + av_freep(&s->thread_opaque); +} + +int avcodec_thread_execute(AVCodecContext *s, int (*func)(AVCodecContext *c2, void *arg2),void **arg, int *ret, int count){ + ThreadContext *c= s->thread_opaque; + int i; + + assert(s == c->avctx); + assert(count <= s->thread_count); + + /* note, we can be certain that this is not called with the same AVCodecContext by different threads at the same time */ + + for(i=0; ithread_count= thread_count; + + assert(!s->thread_opaque); + c= av_mallocz(sizeof(ThreadContext)*thread_count); + s->thread_opaque= c; + + for(i=0; ithread_count, NULL))) + goto fail; + if(!(c[i].done_sem = CreateSemaphore(NULL, 0, s->thread_count, NULL))) + goto fail; + +//printf("create thread %d\n", i); fflush(stdout); + c[i].thread = (HANDLE)_beginthreadex(NULL, 0, thread_func, &c[i], 0, &threadid ); + if( !c[i].thread ) goto fail; + } +//printf("init done\n"); fflush(stdout); + + s->execute= avcodec_thread_execute; + + return 0; +fail: + avcodec_thread_free(s); + return -1; +}