2009-11-27 07:37:18 +01:00
|
|
|
/*****************************************************************************
|
2010-09-17 13:03:27 +02:00
|
|
|
* avs.c: avisynth input
|
2009-11-27 07:37:18 +01:00
|
|
|
*****************************************************************************
|
2010-09-17 13:03:27 +02:00
|
|
|
* Copyright (C) 2009-2010 x264 project
|
2009-11-27 07:37:18 +01:00
|
|
|
*
|
|
|
|
* Authors: Steven Walters <kemuri9@gmail.com>
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program 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 General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA.
|
2010-09-17 13:03:27 +02:00
|
|
|
*
|
|
|
|
* This program is also available under a commercial proprietary license.
|
|
|
|
* For more information, contact us at licensing@x264.com.
|
2009-11-27 07:37:18 +01:00
|
|
|
*****************************************************************************/
|
|
|
|
|
2010-06-26 22:28:49 +02:00
|
|
|
#include "input.h"
|
2009-11-27 07:37:18 +01:00
|
|
|
#include <windows.h>
|
2010-06-26 22:28:49 +02:00
|
|
|
#define FAIL_IF_ERROR( cond, ... ) FAIL_IF_ERR( cond, "avs", __VA_ARGS__ )
|
2009-11-27 07:37:18 +01:00
|
|
|
|
2010-07-17 23:43:37 +02:00
|
|
|
#define AVSC_NO_DECLSPEC
|
2009-11-27 07:37:18 +01:00
|
|
|
#undef EXTERN_C
|
|
|
|
#include "extras/avisynth_c.h"
|
2010-07-17 23:43:37 +02:00
|
|
|
#define AVSC_DECLARE_FUNC(name) name##_func name
|
2009-11-27 07:37:18 +01:00
|
|
|
|
|
|
|
/* AVS uses a versioned interface to control backwards compatibility */
|
2010-07-17 23:43:37 +02:00
|
|
|
/* YV12 support is required, which was added in 2.5 */
|
|
|
|
#define AVS_INTERFACE_25 2
|
2009-11-27 07:37:18 +01:00
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
#if HAVE_SWSCALE
|
|
|
|
#include <libavutil/pixfmt.h>
|
|
|
|
#endif
|
|
|
|
|
2009-11-27 07:37:18 +01:00
|
|
|
/* maximum size of the sequence of filters to try on non script files */
|
|
|
|
#define AVS_MAX_SEQUENCE 5
|
|
|
|
|
2010-05-06 19:03:31 +02:00
|
|
|
#define LOAD_AVS_FUNC(name, continue_on_fail)\
|
2009-11-27 07:37:18 +01:00
|
|
|
{\
|
|
|
|
h->func.name = (void*)GetProcAddress( h->library, #name );\
|
|
|
|
if( !continue_on_fail && !h->func.name )\
|
|
|
|
goto fail;\
|
|
|
|
}
|
|
|
|
|
|
|
|
typedef struct
|
|
|
|
{
|
|
|
|
AVS_Clip *clip;
|
|
|
|
AVS_ScriptEnvironment *env;
|
|
|
|
HMODULE library;
|
2009-12-08 20:36:25 +01:00
|
|
|
int num_frames;
|
2009-11-27 07:37:18 +01:00
|
|
|
struct
|
|
|
|
{
|
2010-07-17 23:43:37 +02:00
|
|
|
AVSC_DECLARE_FUNC( avs_clip_get_error );
|
|
|
|
AVSC_DECLARE_FUNC( avs_create_script_environment );
|
|
|
|
AVSC_DECLARE_FUNC( avs_delete_script_environment );
|
|
|
|
AVSC_DECLARE_FUNC( avs_get_frame );
|
|
|
|
AVSC_DECLARE_FUNC( avs_get_video_info );
|
|
|
|
AVSC_DECLARE_FUNC( avs_function_exists );
|
|
|
|
AVSC_DECLARE_FUNC( avs_invoke );
|
|
|
|
AVSC_DECLARE_FUNC( avs_release_clip );
|
|
|
|
AVSC_DECLARE_FUNC( avs_release_value );
|
|
|
|
AVSC_DECLARE_FUNC( avs_release_video_frame );
|
|
|
|
AVSC_DECLARE_FUNC( avs_take_clip );
|
2009-11-27 07:37:18 +01:00
|
|
|
} func;
|
|
|
|
} avs_hnd_t;
|
|
|
|
|
|
|
|
/* load the library and functions we require from it */
|
2010-07-17 23:43:37 +02:00
|
|
|
static int x264_avs_load_library( avs_hnd_t *h )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
|
|
|
h->library = LoadLibrary( "avisynth" );
|
|
|
|
if( !h->library )
|
|
|
|
return -1;
|
|
|
|
LOAD_AVS_FUNC( avs_clip_get_error, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_create_script_environment, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_delete_script_environment, 1 );
|
|
|
|
LOAD_AVS_FUNC( avs_get_frame, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_get_video_info, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_function_exists, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_invoke, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_release_clip, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_release_value, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_release_video_frame, 0 );
|
|
|
|
LOAD_AVS_FUNC( avs_take_clip, 0 );
|
|
|
|
return 0;
|
|
|
|
fail:
|
|
|
|
FreeLibrary( h->library );
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* generate a filter sequence to try based on the filename extension */
|
|
|
|
static void avs_build_filter_sequence( char *filename_ext, const char *filter[AVS_MAX_SEQUENCE+1] )
|
|
|
|
{
|
2010-03-31 10:44:07 +02:00
|
|
|
int i = 0;
|
2009-11-27 07:37:18 +01:00
|
|
|
const char *all_purpose[] = { "FFmpegSource2", "DSS2", "DirectShowSource", 0 };
|
|
|
|
if( !strcasecmp( filename_ext, "avi" ) )
|
|
|
|
filter[i++] = "AVISource";
|
|
|
|
if( !strcasecmp( filename_ext, "d2v" ) )
|
|
|
|
filter[i++] = "MPEG2Source";
|
|
|
|
if( !strcasecmp( filename_ext, "dga" ) )
|
|
|
|
filter[i++] = "AVCSource";
|
2010-03-31 10:44:07 +02:00
|
|
|
for( int j = 0; all_purpose[j] && i < AVS_MAX_SEQUENCE; j++ )
|
2009-11-27 07:37:18 +01:00
|
|
|
filter[i++] = all_purpose[j];
|
|
|
|
}
|
|
|
|
|
2010-04-07 04:08:21 +02:00
|
|
|
static AVS_Value update_clip( avs_hnd_t *h, const AVS_VideoInfo **vi, AVS_Value res, AVS_Value release )
|
|
|
|
{
|
|
|
|
h->func.avs_release_clip( h->clip );
|
|
|
|
h->clip = h->func.avs_take_clip( res, h->env );
|
|
|
|
h->func.avs_release_value( release );
|
|
|
|
*vi = h->func.avs_get_video_info( h->clip );
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
LAVF/FFMS input support, native VFR timestamp handling
libx264 now takes three new API parameters.
b_vfr_input tells x264 whether or not the input is VFR, and is 1 by default.
i_timebase_num and i_timebase_den pass the timebase to x264.
x264_picture_t now returns the DTS of each frame: the calling app need not calculate it anymore.
Add libavformat and FFMS2 input support: requires libav* and ffms2 libraries respectively.
FFMS2 is _STRONGLY_ preferred over libavformat: we encourage all distributions to compile with FFMS2 support if at all possible.
FFMS2 can be found at http://code.google.com/p/ffmpegsource/.
--index, a new x264cli option, allows the user to store (or load) an FFMS2 index file for future use, to avoid re-indexing in the future.
Overhaul the muxers to pass through timestamps instead of assuming CFR.
Also overhaul muxers to correctly use b_annexb and b_repeat_headers to simplify the code.
Remove VFW input support, since it's now pretty much redundant with native AVS support and LAVF support.
Finally, overhaul a large part of the x264cli internals.
--force-cfr, a new x264cli option, allows the user to force the old method of timestamp handling. May be useful in case of a source with broken timestamps.
Avisynth, YUV, and Y4M input are all still CFR. LAVF or FFMS2 must be used for VFR support.
Do note that this patch does *not* add VFR ratecontrol yet.
Support for telecined input is also somewhat dubious at the moment.
Large parts of this patch by Mike Gurlitz <mike.gurlitz@gmail.com>, Steven Walters <kemuri9@gmail.com>, and Yusuke Nakamura <muken.the.vfrmaniac@gmail.com>.
2009-12-28 16:42:17 +01:00
|
|
|
static int open_file( char *psz_filename, hnd_t *p_handle, video_info_t *info, cli_input_opt_t *opt )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
|
|
|
FILE *fh = fopen( psz_filename, "r" );
|
|
|
|
if( !fh )
|
|
|
|
return -1;
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( !x264_is_regular_file( fh ), "AVS input is incompatible with non-regular file `%s'\n", psz_filename );
|
2009-11-27 07:37:18 +01:00
|
|
|
fclose( fh );
|
|
|
|
|
|
|
|
avs_hnd_t *h = malloc( sizeof(avs_hnd_t) );
|
|
|
|
if( !h )
|
|
|
|
return -1;
|
2010-07-17 23:43:37 +02:00
|
|
|
FAIL_IF_ERROR( x264_avs_load_library( h ), "failed to load avisynth\n" )
|
|
|
|
h->env = h->func.avs_create_script_environment( AVS_INTERFACE_25 );
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( !h->env, "failed to initiate avisynth\n" )
|
2009-11-27 07:37:18 +01:00
|
|
|
AVS_Value arg = avs_new_value_string( psz_filename );
|
|
|
|
AVS_Value res;
|
|
|
|
char *filename_ext = get_filename_extension( psz_filename );
|
|
|
|
|
|
|
|
if( !strcasecmp( filename_ext, "avs" ) )
|
|
|
|
{
|
|
|
|
res = h->func.avs_invoke( h->env, "Import", arg, NULL );
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( avs_is_error( res ), "%s\n", avs_as_string( res ) )
|
2009-12-11 04:48:51 +01:00
|
|
|
/* check if the user is using a multi-threaded script and apply distributor if necessary.
|
|
|
|
adapted from avisynth's vfw interface */
|
|
|
|
AVS_Value mt_test = h->func.avs_invoke( h->env, "GetMTMode", avs_new_value_bool( 0 ), NULL );
|
|
|
|
int mt_mode = avs_is_int( mt_test ) ? avs_as_int( mt_test ) : 0;
|
|
|
|
h->func.avs_release_value( mt_test );
|
|
|
|
if( mt_mode > 0 && mt_mode < 5 )
|
|
|
|
{
|
|
|
|
AVS_Value temp = h->func.avs_invoke( h->env, "Distributor", res, NULL );
|
|
|
|
h->func.avs_release_value( res );
|
|
|
|
res = temp;
|
|
|
|
}
|
2009-11-27 07:37:18 +01:00
|
|
|
}
|
|
|
|
else /* non script file */
|
|
|
|
{
|
|
|
|
/* cycle through known source filters to find one that works */
|
|
|
|
const char *filter[AVS_MAX_SEQUENCE+1] = { 0 };
|
|
|
|
avs_build_filter_sequence( filename_ext, filter );
|
|
|
|
int i;
|
|
|
|
for( i = 0; filter[i]; i++ )
|
|
|
|
{
|
2010-06-26 22:28:49 +02:00
|
|
|
x264_cli_log( "avs", X264_LOG_INFO, "trying %s... ", filter[i] );
|
2009-11-27 07:37:18 +01:00
|
|
|
if( !h->func.avs_function_exists( h->env, filter[i] ) )
|
|
|
|
{
|
2010-06-26 22:28:49 +02:00
|
|
|
x264_cli_printf( X264_LOG_INFO, "not found\n" );
|
2009-11-27 07:37:18 +01:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if( !strncasecmp( filter[i], "FFmpegSource", 12 ) )
|
|
|
|
{
|
2010-06-26 22:28:49 +02:00
|
|
|
x264_cli_printf( X264_LOG_INFO, "indexing... " );
|
2009-11-27 07:37:18 +01:00
|
|
|
fflush( stderr );
|
|
|
|
}
|
|
|
|
res = h->func.avs_invoke( h->env, filter[i], arg, NULL );
|
|
|
|
if( !avs_is_error( res ) )
|
|
|
|
{
|
2010-06-26 22:28:49 +02:00
|
|
|
x264_cli_printf( X264_LOG_INFO, "succeeded\n" );
|
2009-11-27 07:37:18 +01:00
|
|
|
break;
|
|
|
|
}
|
2010-06-26 22:28:49 +02:00
|
|
|
x264_cli_printf( X264_LOG_INFO, "failed\n" );
|
2009-11-27 07:37:18 +01:00
|
|
|
}
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( !filter[i], "unable to find source filter to open `%s'\n", psz_filename )
|
2009-11-27 07:37:18 +01:00
|
|
|
}
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( !avs_is_clip( res ), "`%s' didn't return a video clip\n", psz_filename )
|
2009-11-27 07:37:18 +01:00
|
|
|
h->clip = h->func.avs_take_clip( res, h->env );
|
|
|
|
const AVS_VideoInfo *vi = h->func.avs_get_video_info( h->clip );
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( !avs_has_video( vi ), "`%s' has no video data\n", psz_filename )
|
2010-04-07 04:08:21 +02:00
|
|
|
/* if the clip is made of fields instead of frames, call weave to make them frames */
|
|
|
|
if( avs_is_field_based( vi ) )
|
|
|
|
{
|
2010-06-26 22:28:49 +02:00
|
|
|
x264_cli_log( "avs", X264_LOG_WARNING, "detected fieldbased (separated) input, weaving to frames\n" );
|
2010-04-07 04:08:21 +02:00
|
|
|
AVS_Value tmp = h->func.avs_invoke( h->env, "Weave", res, NULL );
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( avs_is_error( tmp ), "couldn't weave fields into frames\n" )
|
2010-04-07 04:08:21 +02:00
|
|
|
res = update_clip( h, &vi, tmp, res );
|
2010-04-13 14:44:37 +02:00
|
|
|
info->interlaced = 1;
|
|
|
|
info->tff = avs_is_tff( vi );
|
2010-04-07 04:08:21 +02:00
|
|
|
}
|
2010-07-17 23:43:37 +02:00
|
|
|
#if !HAVE_SWSCALE
|
|
|
|
/* if swscale is not available, convert CSPs to yv12 */
|
|
|
|
if( !avs_is_yv12( vi ) )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
2010-07-17 23:43:37 +02:00
|
|
|
x264_cli_log( "avs", X264_LOG_WARNING, "converting input clip to YV12\n" );
|
2010-07-05 23:37:47 +02:00
|
|
|
FAIL_IF_ERROR( vi->width&1 || vi->height&1, "input clip width or height not divisible by 2 (%dx%d)\n", vi->width, vi->height )
|
2009-11-27 07:37:18 +01:00
|
|
|
const char *arg_name[2] = { NULL, "interlaced" };
|
LAVF/FFMS input support, native VFR timestamp handling
libx264 now takes three new API parameters.
b_vfr_input tells x264 whether or not the input is VFR, and is 1 by default.
i_timebase_num and i_timebase_den pass the timebase to x264.
x264_picture_t now returns the DTS of each frame: the calling app need not calculate it anymore.
Add libavformat and FFMS2 input support: requires libav* and ffms2 libraries respectively.
FFMS2 is _STRONGLY_ preferred over libavformat: we encourage all distributions to compile with FFMS2 support if at all possible.
FFMS2 can be found at http://code.google.com/p/ffmpegsource/.
--index, a new x264cli option, allows the user to store (or load) an FFMS2 index file for future use, to avoid re-indexing in the future.
Overhaul the muxers to pass through timestamps instead of assuming CFR.
Also overhaul muxers to correctly use b_annexb and b_repeat_headers to simplify the code.
Remove VFW input support, since it's now pretty much redundant with native AVS support and LAVF support.
Finally, overhaul a large part of the x264cli internals.
--force-cfr, a new x264cli option, allows the user to force the old method of timestamp handling. May be useful in case of a source with broken timestamps.
Avisynth, YUV, and Y4M input are all still CFR. LAVF or FFMS2 must be used for VFR support.
Do note that this patch does *not* add VFR ratecontrol yet.
Support for telecined input is also somewhat dubious at the moment.
Large parts of this patch by Mike Gurlitz <mike.gurlitz@gmail.com>, Steven Walters <kemuri9@gmail.com>, and Yusuke Nakamura <muken.the.vfrmaniac@gmail.com>.
2009-12-28 16:42:17 +01:00
|
|
|
AVS_Value arg_arr[2] = { res, avs_new_value_bool( info->interlaced ) };
|
2009-11-27 07:37:18 +01:00
|
|
|
AVS_Value res2 = h->func.avs_invoke( h->env, "ConvertToYV12", avs_new_value_array( arg_arr, 2 ), arg_name );
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( avs_is_error( res2 ), "couldn't convert input clip to YV12\n" )
|
2010-04-07 04:08:21 +02:00
|
|
|
res = update_clip( h, &vi, res2, res );
|
2009-11-27 07:37:18 +01:00
|
|
|
}
|
2010-07-17 23:43:37 +02:00
|
|
|
#endif
|
2009-11-27 07:37:18 +01:00
|
|
|
h->func.avs_release_value( res );
|
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
info->width = vi->width;
|
|
|
|
info->height = vi->height;
|
LAVF/FFMS input support, native VFR timestamp handling
libx264 now takes three new API parameters.
b_vfr_input tells x264 whether or not the input is VFR, and is 1 by default.
i_timebase_num and i_timebase_den pass the timebase to x264.
x264_picture_t now returns the DTS of each frame: the calling app need not calculate it anymore.
Add libavformat and FFMS2 input support: requires libav* and ffms2 libraries respectively.
FFMS2 is _STRONGLY_ preferred over libavformat: we encourage all distributions to compile with FFMS2 support if at all possible.
FFMS2 can be found at http://code.google.com/p/ffmpegsource/.
--index, a new x264cli option, allows the user to store (or load) an FFMS2 index file for future use, to avoid re-indexing in the future.
Overhaul the muxers to pass through timestamps instead of assuming CFR.
Also overhaul muxers to correctly use b_annexb and b_repeat_headers to simplify the code.
Remove VFW input support, since it's now pretty much redundant with native AVS support and LAVF support.
Finally, overhaul a large part of the x264cli internals.
--force-cfr, a new x264cli option, allows the user to force the old method of timestamp handling. May be useful in case of a source with broken timestamps.
Avisynth, YUV, and Y4M input are all still CFR. LAVF or FFMS2 must be used for VFR support.
Do note that this patch does *not* add VFR ratecontrol yet.
Support for telecined input is also somewhat dubious at the moment.
Large parts of this patch by Mike Gurlitz <mike.gurlitz@gmail.com>, Steven Walters <kemuri9@gmail.com>, and Yusuke Nakamura <muken.the.vfrmaniac@gmail.com>.
2009-12-28 16:42:17 +01:00
|
|
|
info->fps_num = vi->fps_numerator;
|
|
|
|
info->fps_den = vi->fps_denominator;
|
2010-07-05 23:37:47 +02:00
|
|
|
h->num_frames = info->num_frames = vi->num_frames;
|
|
|
|
info->thread_safe = 1;
|
|
|
|
#if HAVE_SWSCALE
|
|
|
|
if( avs_is_rgb32( vi ) )
|
|
|
|
info->csp = X264_CSP_BGRA | X264_CSP_VFLIP;
|
|
|
|
else if( avs_is_rgb24( vi ) )
|
|
|
|
info->csp = X264_CSP_BGR | X264_CSP_VFLIP;
|
|
|
|
else if( avs_is_yuy2( vi ) )
|
|
|
|
info->csp = PIX_FMT_YUYV422 | X264_CSP_OTHER;
|
2010-07-17 23:43:37 +02:00
|
|
|
else if( avs_is_yv24( vi ) )
|
|
|
|
info->csp = X264_CSP_I444;
|
|
|
|
else if( avs_is_yv16( vi ) )
|
|
|
|
info->csp = X264_CSP_I422;
|
|
|
|
else if( avs_is_yv12( vi ) )
|
|
|
|
info->csp = X264_CSP_I420;
|
|
|
|
else if( avs_is_yv411( vi ) )
|
|
|
|
info->csp = PIX_FMT_YUV411P | X264_CSP_OTHER;
|
|
|
|
else if( avs_is_y8( vi ) )
|
|
|
|
info->csp = PIX_FMT_GRAY8 | X264_CSP_OTHER;
|
|
|
|
else
|
|
|
|
info->csp = X264_CSP_NONE;
|
2010-07-05 23:37:47 +02:00
|
|
|
#else
|
|
|
|
info->csp = X264_CSP_I420;
|
|
|
|
#endif
|
LAVF/FFMS input support, native VFR timestamp handling
libx264 now takes three new API parameters.
b_vfr_input tells x264 whether or not the input is VFR, and is 1 by default.
i_timebase_num and i_timebase_den pass the timebase to x264.
x264_picture_t now returns the DTS of each frame: the calling app need not calculate it anymore.
Add libavformat and FFMS2 input support: requires libav* and ffms2 libraries respectively.
FFMS2 is _STRONGLY_ preferred over libavformat: we encourage all distributions to compile with FFMS2 support if at all possible.
FFMS2 can be found at http://code.google.com/p/ffmpegsource/.
--index, a new x264cli option, allows the user to store (or load) an FFMS2 index file for future use, to avoid re-indexing in the future.
Overhaul the muxers to pass through timestamps instead of assuming CFR.
Also overhaul muxers to correctly use b_annexb and b_repeat_headers to simplify the code.
Remove VFW input support, since it's now pretty much redundant with native AVS support and LAVF support.
Finally, overhaul a large part of the x264cli internals.
--force-cfr, a new x264cli option, allows the user to force the old method of timestamp handling. May be useful in case of a source with broken timestamps.
Avisynth, YUV, and Y4M input are all still CFR. LAVF or FFMS2 must be used for VFR support.
Do note that this patch does *not* add VFR ratecontrol yet.
Support for telecined input is also somewhat dubious at the moment.
Large parts of this patch by Mike Gurlitz <mike.gurlitz@gmail.com>, Steven Walters <kemuri9@gmail.com>, and Yusuke Nakamura <muken.the.vfrmaniac@gmail.com>.
2009-12-28 16:42:17 +01:00
|
|
|
info->vfr = 0;
|
2009-11-27 07:37:18 +01:00
|
|
|
|
|
|
|
*p_handle = h;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
static int picture_alloc( cli_pic_t *pic, int csp, int width, int height )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
2010-07-05 23:37:47 +02:00
|
|
|
if( x264_cli_pic_alloc( pic, X264_CSP_NONE, width, height ) )
|
|
|
|
return -1;
|
|
|
|
pic->img.csp = csp;
|
|
|
|
const x264_cli_csp_t *cli_csp = x264_cli_get_csp( csp );
|
2010-07-17 23:43:37 +02:00
|
|
|
if( cli_csp )
|
|
|
|
pic->img.planes = cli_csp->planes;
|
|
|
|
#if HAVE_SWSCALE
|
|
|
|
else if( csp == (PIX_FMT_YUV411P | X264_CSP_OTHER) )
|
|
|
|
pic->img.planes = 3;
|
|
|
|
else
|
|
|
|
pic->img.planes = 1; //y8 and yuy2 are one plane
|
|
|
|
#endif
|
2009-11-27 07:37:18 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
static int read_frame( cli_pic_t *pic, hnd_t handle, int i_frame )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
2010-07-05 23:37:47 +02:00
|
|
|
static const int plane[3] = { AVS_PLANAR_Y, AVS_PLANAR_U, AVS_PLANAR_V };
|
2009-11-27 07:37:18 +01:00
|
|
|
avs_hnd_t *h = handle;
|
2009-12-08 20:36:25 +01:00
|
|
|
if( i_frame >= h->num_frames )
|
|
|
|
return -1;
|
2010-07-05 23:37:47 +02:00
|
|
|
AVS_VideoFrame *frm = pic->opaque = h->func.avs_get_frame( h->clip, i_frame );
|
2009-11-27 07:37:18 +01:00
|
|
|
const char *err = h->func.avs_clip_get_error( h->clip );
|
2010-06-26 22:28:49 +02:00
|
|
|
FAIL_IF_ERROR( err, "%s occurred while reading frame %d\n", err, i_frame )
|
2010-07-05 23:37:47 +02:00
|
|
|
for( int i = 0; i < pic->img.planes; i++ )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
|
|
|
/* explicitly cast away the const attribute to avoid a warning */
|
2010-07-05 23:37:47 +02:00
|
|
|
pic->img.plane[i] = (uint8_t*)avs_get_read_ptr_p( frm, plane[i] );
|
|
|
|
pic->img.stride[i] = avs_get_pitch_p( frm, plane[i] );
|
2009-11-27 07:37:18 +01:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
static int release_frame( cli_pic_t *pic, hnd_t handle )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
|
|
|
avs_hnd_t *h = handle;
|
|
|
|
h->func.avs_release_video_frame( pic->opaque );
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
static void picture_clean( cli_pic_t *pic )
|
2009-11-27 07:37:18 +01:00
|
|
|
{
|
2010-07-05 23:37:47 +02:00
|
|
|
memset( pic, 0, sizeof(cli_pic_t) );
|
2009-11-27 07:37:18 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
static int close_file( hnd_t handle )
|
|
|
|
{
|
|
|
|
avs_hnd_t *h = handle;
|
|
|
|
h->func.avs_release_clip( h->clip );
|
|
|
|
if( h->func.avs_delete_script_environment )
|
|
|
|
h->func.avs_delete_script_environment( h->env );
|
|
|
|
FreeLibrary( h->library );
|
|
|
|
free( h );
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-07-05 23:37:47 +02:00
|
|
|
const cli_input_t avs_input = { open_file, picture_alloc, read_frame, release_frame, picture_clean, close_file };
|