LSMASHVideoSource::LSMASHVideoSource
(
const char *source,
uint32_t track_number,
int threads,
int seek_mode,
uint32_t forward_seek_threshold,
int direct_rendering,
int stacked_format,
enum AVPixelFormat pixel_format,
IScriptEnvironment *env
)
{
memset( &vi, 0, sizeof(VideoInfo) );
memset( &vdh, 0, sizeof(libavsmash_video_decode_handler_t) );
memset( &voh, 0, sizeof(libavsmash_video_output_handler_t) );
format_ctx = NULL;
vdh.seek_mode = seek_mode;
vdh.forward_seek_threshold = forward_seek_threshold;
as_video_output_handler_t *as_vohp = (as_video_output_handler_t *)lw_malloc_zero( sizeof(as_video_output_handler_t) );
if( !as_vohp )
env->ThrowError( "LSMASHVideoSource: failed to allocate the AviSynth video output handler." );
as_vohp->vi = &vi;
as_vohp->env = env;
voh.private_handler = as_vohp;
voh.free_private_handler = as_free_video_output_handler;
get_video_track( source, track_number, threads, env );
lsmash_discard_boxes( vdh.root );
prepare_video_decoding( direct_rendering, stacked_format, pixel_format, env );
}
LSMASHAudioSource::LSMASHAudioSource
(
const char *source,
uint32_t track_number,
bool skip_priming,
uint64_t channel_layout,
int sample_rate,
IScriptEnvironment *env
)
{
memset( &vi, 0, sizeof(VideoInfo) );
memset( &adh, 0, sizeof(libavsmash_audio_decode_handler_t) );
memset( &aoh, 0, sizeof(libavsmash_audio_output_handler_t) );
format_ctx = NULL;
get_audio_track( source, track_number, skip_priming, env );
lsmash_discard_boxes( adh.root );
prepare_audio_decoding( channel_layout, sample_rate, env );
}
void VS_CC vs_libavsmashsource_create( const VSMap *in, VSMap *out, void *user_data, VSCore *core, const VSAPI *vsapi )
{
/* Get file name. */
const char *file_name = vsapi->propGetData( in, "source", 0, NULL );
if( !file_name )
{
vsapi->setError( out, "lsmas: failed to get source file name." );
return;
}
/* Allocate the handler of this plugin. */
lsmas_handler_t *hp = alloc_handler();
if( !hp )
{
vsapi->setError( out, "lsmas: failed to allocate the handler." );
return;
}
libavsmash_video_decode_handler_t *vdhp = hp->vdhp;
libavsmash_video_output_handler_t *vohp = hp->vohp;
vs_video_output_handler_t *vs_vohp = vs_allocate_video_output_handler( vohp );
if( !vs_vohp )
{
free_handler( &hp );
vsapi->setError( out, "lsmas: failed to allocate the VapourSynth video output handler." );
return;
}
vohp->private_handler = vs_vohp;
vohp->free_private_handler = lw_free;
/* Set up VapourSynth error handler. */
vs_basic_handler_t vsbh = { 0 };
vsbh.out = out;
vsbh.frame_ctx = NULL;
vsbh.vsapi = vsapi;
/* Set up log handler. */
lw_log_handler_t lh = { 0 };
lh.level = LW_LOG_FATAL;
lh.priv = &vsbh;
lh.show_log = set_error;
/* Open source file. */
uint32_t number_of_tracks = open_file( hp, file_name, &lh );
if( number_of_tracks == 0 )
{
vs_filter_free( hp, core, vsapi );
return;
}
/* Get options. */
int64_t track_number;
int64_t threads;
int64_t seek_mode;
int64_t seek_threshold;
int64_t variable_info;
int64_t direct_rendering;
int64_t fps_num;
int64_t fps_den;
const char *format;
const char *preferred_decoder_names;
set_option_int64 ( &track_number, 0, "track", in, vsapi );
set_option_int64 ( &threads, 0, "threads", in, vsapi );
set_option_int64 ( &seek_mode, 0, "seek_mode", in, vsapi );
set_option_int64 ( &seek_threshold, 10, "seek_threshold", in, vsapi );
set_option_int64 ( &variable_info, 0, "variable", in, vsapi );
set_option_int64 ( &direct_rendering, 0, "dr", in, vsapi );
set_option_int64 ( &fps_num, 0, "fpsnum", in, vsapi );
set_option_int64 ( &fps_den, 1, "fpsden", in, vsapi );
set_option_string( &format, NULL, "format", in, vsapi );
set_option_string( &preferred_decoder_names, NULL, "decoder", in, vsapi );
set_preferred_decoder_names_on_buf( hp->preferred_decoder_names_buf, preferred_decoder_names );
libavsmash_video_set_seek_mode ( vdhp, CLIP_VALUE( seek_mode, 0, 2 ) );
libavsmash_video_set_forward_seek_threshold ( vdhp, CLIP_VALUE( seek_threshold, 1, 999 ) );
libavsmash_video_set_preferred_decoder_names( vdhp, tokenize_preferred_decoder_names( hp->preferred_decoder_names_buf ) );
vohp->vfr2cfr = (fps_num > 0 && fps_den > 0);
vohp->cfr_num = (uint32_t)fps_num;
vohp->cfr_den = (uint32_t)fps_den;
vs_vohp->variable_info = CLIP_VALUE( variable_info, 0, 1 );
vs_vohp->direct_rendering = CLIP_VALUE( direct_rendering, 0, 1 ) && !format;
vs_vohp->vs_output_pixel_format = vs_vohp->variable_info ? pfNone : get_vs_output_pixel_format( format );
if( track_number && track_number > number_of_tracks )
{
vs_filter_free( hp, core, vsapi );
set_error_on_init( out, vsapi, "lsmas: the number of tracks equals %"PRIu32".", number_of_tracks );
return;
}
libavsmash_video_set_log_handler( vdhp, &lh );
/* Get video track. */
if( libavsmash_video_get_track( vdhp, track_number ) < 0 )
{
vs_filter_free( hp, core, vsapi );
return;
}
/* Set up decoders for this track. */
threads = threads >= 0 ? threads : 0;
if( prepare_video_decoding( hp, threads, out, core, vsapi ) < 0 )
{
vs_filter_free( hp, core, vsapi );
return;
}
lsmash_discard_boxes( libavsmash_video_get_root( vdhp ) );
vsapi->createFilter( in, out, "LibavSMASHSource", vs_filter_init, vs_filter_get_frame, vs_filter_free, fmUnordered, nfMakeLinear, hp, core );
return;
}
