libmpdemux/demux_avs.c

   1 /*
   2  * Demuxer for avisynth
   3  * Copyright (c) 2005 Gianluigi Tiesi <sherpya@netfarm.it>
   4  *
   5  * This file is part of MPlayer.
   6  *
   7  * MPlayer is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * MPlayer is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public License
  18  * along with MPlayer; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20  */
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25 #include <limits.h>
  26
  27 #include "config.h"
  28 #include "mp_msg.h"
  29
  30 #include "stream/stream.h"
  31 #include "demuxer.h"
  32 #include "stheader.h"
  33 #include "libvo/fastmemcpy.h"
  34
  35 #include "loader/wine/winbase.h"
  36 #include "loader/wine/windef.h"
  37
  38 #ifdef WIN32_LOADER
  39 #include "loader/ldt_keeper.h"
  40 #endif
  41
  42 #include "demux_avs.h"
  43
  44 #define MAX_AVS_SIZE    16 * 1024 /* 16k should be enough */
  45
  46 typedef WINAPI AVS_ScriptEnvironment* (*imp_avs_create_script_environment)(int version);
  47 typedef WINAPI AVS_Value (*imp_avs_invoke)(AVS_ScriptEnvironment *, const char * name, AVS_Value args, const char** arg_names);
  48 typedef WINAPI const AVS_VideoInfo *(*imp_avs_get_video_info)(AVS_Clip *);
  49 typedef WINAPI AVS_Clip* (*imp_avs_take_clip)(AVS_Value, AVS_ScriptEnvironment *);
  50 typedef WINAPI void (*imp_avs_release_clip)(AVS_Clip *);
  51 typedef WINAPI AVS_VideoFrame* (*imp_avs_get_frame)(AVS_Clip *, int n);
  52 typedef WINAPI void (*imp_avs_release_video_frame)(AVS_VideoFrame *);
  53 typedef WINAPI int (*imp_avs_get_audio)(AVS_Clip *, void * buf, uint64_t start, uint64_t count);
  54
  55 #define Q(string) # string
  56 #define IMPORT_FUNC(x) \
  57     AVS->x = ( imp_##x ) GetProcAddress(AVS->dll, Q(x)); \
  58     if (!AVS->x) { mp_msg(MSGT_DEMUX,MSGL_V,"AVS: failed to load "Q(x)"()\n"); return 0; }
  59
  60 typedef struct tagAVS
  61 {
  62     AVS_ScriptEnvironment *avs_env;
  63     AVS_Value handler;
  64     AVS_Clip *clip;
  65     const AVS_VideoInfo *video_info;
  66 #ifdef WIN32_LOADER
  67     ldt_fs_t* ldt_fs;
  68 #endif
  69     HMODULE dll;
  70     int frameno;
  71     uint64_t sampleno;
  72     int init;
  73
  74     imp_avs_create_script_environment avs_create_script_environment;
  75     imp_avs_invoke avs_invoke;
  76     imp_avs_get_video_info avs_get_video_info;
  77     imp_avs_take_clip avs_take_clip;
  78     imp_avs_release_clip avs_release_clip;
  79     imp_avs_get_frame avs_get_frame;
  80     imp_avs_release_video_frame avs_release_video_frame;
  81     imp_avs_get_audio avs_get_audio;
  82 } AVS_T;
  83
  84 static AVS_T *initAVS(const char *filename)
  85 {
  86     AVS_T *AVS = malloc (sizeof(AVS_T));
  87     AVS_Value arg0 = avs_new_value_string(filename);
  88     AVS_Value args = avs_new_value_array(&arg0, 1);
  89
  90     memset(AVS, 0, sizeof(AVS_T));
  91
  92 #ifdef WIN32_LOADER
  93     AVS->ldt_fs = Setup_LDT_Keeper();
  94 #endif
  95
  96     AVS->dll = LoadLibraryA("avisynth.dll");
  97     if(!AVS->dll)
  98     {
  99         mp_msg(MSGT_DEMUX ,MSGL_V, "AVS: failed to load avisynth.dll\n");
 100         goto avs_err;
 101     }
 102
 103     /* Dynamic import of needed stuff from avisynth.dll */
 104     IMPORT_FUNC(avs_create_script_environment);
 105     IMPORT_FUNC(avs_invoke);
 106     IMPORT_FUNC(avs_get_video_info);
 107     IMPORT_FUNC(avs_take_clip);
 108     IMPORT_FUNC(avs_release_clip);
 109     IMPORT_FUNC(avs_get_frame);
 110     IMPORT_FUNC(avs_release_video_frame);
 111     IMPORT_FUNC(avs_get_audio);
 112
 113     AVS->avs_env = AVS->avs_create_script_environment(AVISYNTH_INTERFACE_VERSION);
 114     if (!AVS->avs_env)
 115     {
 116         mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_create_script_environment failed\n");
 117         goto avs_err;
 118     }
 119
 120
 121     AVS->handler = AVS->avs_invoke(AVS->avs_env, "Import", args, 0);
 122
 123     if (avs_is_error(AVS->handler))
 124     {
 125         mp_msg(MSGT_DEMUX, MSGL_V, "AVS: Avisynth error: %s\n", avs_as_string(AVS->handler));
 126         goto avs_err;
 127     }
 128
 129     if (!avs_is_clip(AVS->handler))
 130     {
 131         mp_msg(MSGT_DEMUX, MSGL_V, "AVS: Avisynth doesn't return a clip\n");
 132         goto avs_err;
 133     }
 134
 135     return AVS;
 136
 137 avs_err:
 138     if (AVS->dll) FreeLibrary(AVS->dll);
 139 #ifdef WIN32_LOADER
 140     Restore_LDT_Keeper(AVS->ldt_fs);
 141 #endif
 142     free(AVS);
 143     return NULL;
 144 }
 145
 146 /* Implement RGB MODES ?? */
 147 #if 0
 148 static __inline int get_mmioFOURCC(const AVS_VideoInfo *v)
 149 {
 150     if (avs_is_rgb(v)) return mmioFOURCC(8, 'R', 'G', 'B');
 151     if (avs_is_rgb24(v)) return mmioFOURCC(24, 'R', 'G', 'B');
 152     if (avs_is_rgb32(v)) return mmioFOURCC(32, 'R', 'G', 'B');
 153     if (avs_is_yv12(v)) return mmioFOURCC('Y', 'V', '1', '2');
 154     if (avs_is_yuy(v)) return mmioFOURCC('Y', 'U', 'Y', ' ');
 155     if (avs_is_yuy2(v)) return mmioFOURCC('Y', 'U', 'Y', '2');
 156     return 0;
 157 }
 158 #endif
 159
 160 static int demux_avs_fill_buffer(demuxer_t *demuxer, demux_stream_t *ds)
 161 {
 162     AVS_VideoFrame *curr_frame;
 163     demux_packet_t *dp = NULL;
 164     AVS_T *AVS = demuxer->priv;
 165
 166     if (ds == demuxer->video)
 167     {
 168         sh_video_t *sh_video = demuxer->video->sh;
 169         char *dst;
 170         int w, h;
 171         if (AVS->video_info->num_frames <= AVS->frameno) return 0; // EOF
 172
 173         curr_frame = AVS->avs_get_frame(AVS->clip, AVS->frameno);
 174         if (!curr_frame)
 175         {
 176             mp_msg(MSGT_DEMUX, MSGL_V, "AVS: error getting frame -- EOF??\n");
 177             return 0;
 178         }
 179         w = curr_frame->row_size;
 180         h = curr_frame->height;
 181
 182         dp = new_demux_packet(w * h + 2 * (w / 2) * (h / 2));
 183
 184         dp->pts=AVS->frameno / sh_video->fps;
 185
 186         dst = dp->buffer;
 187         memcpy_pic(dst, curr_frame->vfb->data + curr_frame->offset,
 188                    w, h, w, curr_frame->pitch);
 189         dst += w * h;
 190         w /= 2; h /= 2;
 191         memcpy_pic(dst, curr_frame->vfb->data + curr_frame->offsetV,
 192                    w, h, w, curr_frame->pitchUV);
 193         dst += w * h;
 194         memcpy_pic(dst, curr_frame->vfb->data + curr_frame->offsetU,
 195                    w, h, w, curr_frame->pitchUV);
 196         ds_add_packet(demuxer->video, dp);
 197
 198         AVS->frameno++;
 199         AVS->avs_release_video_frame(curr_frame);
 200     }
 201
 202     /* Audio */
 203     if (ds == demuxer->audio)
 204     {
 205         sh_audio_t *sh_audio = ds->sh;
 206         int samples = sh_audio->samplerate;
 207         uint64_t l;
 208         samples = FFMIN(samples, AVS->video_info->num_audio_samples - AVS->sampleno);
 209         if (!samples) return 0;
 210         l = samples * sh_audio->channels * sh_audio->samplesize;
 211         if (l > INT_MAX) {
 212             mp_msg(MSGT_DEMUX, MSGL_FATAL, "AVS: audio packet too big\n");
 213             return 0;
 214         }
 215         dp = new_demux_packet(l);
 216         dp->pts = AVS->sampleno / sh_audio->samplerate;
 217
 218         if (AVS->avs_get_audio(AVS->clip, dp->buffer, AVS->sampleno, samples))
 219         {
 220             mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_get_audio() failed\n");
 221             return 0;
 222         }
 223         ds_add_packet(demuxer->audio, dp);
 224
 225         AVS->sampleno += samples;
 226     }
 227
 228     return 1;
 229 }
 230
 231 static demuxer_t* demux_open_avs(demuxer_t* demuxer)
 232 {
 233     int found = 0;
 234     AVS_T *AVS = demuxer->priv;
 235     int audio_samplesize = 0;
 236     AVS->frameno = 0;
 237     AVS->sampleno = 0;
 238
 239     mp_msg(MSGT_DEMUX, MSGL_V, "AVS: demux_open_avs()\n");
 240     demuxer->seekable = 1;
 241
 242     AVS->clip = AVS->avs_take_clip(AVS->handler, AVS->avs_env);
 243     if(!AVS->clip)
 244     {
 245         mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_take_clip() failed\n");
 246         return NULL;
 247     }
 248
 249     AVS->video_info = AVS->avs_get_video_info(AVS->clip);
 250     if (!AVS->video_info)
 251     {
 252         mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_get_video_info() call failed\n");
 253         return NULL;
 254     }
 255
 256     if (!avs_is_yv12(AVS->video_info))
 257     {
 258         AVS->handler = AVS->avs_invoke(AVS->avs_env, "ConvertToYV12", avs_new_value_array(&AVS->handler, 1), 0);
 259         if (avs_is_error(AVS->handler))
 260         {
 261             mp_msg(MSGT_DEMUX, MSGL_V, "AVS: Cannot convert input video to YV12: %s\n", avs_as_string(AVS->handler));
 262             return NULL;
 263         }
 264
 265         AVS->clip = AVS->avs_take_clip(AVS->handler, AVS->avs_env);
 266
 267         if(!AVS->clip)
 268         {
 269             mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_take_clip() failed\n");
 270             return NULL;
 271         }
 272
 273         AVS->video_info = AVS->avs_get_video_info(AVS->clip);
 274         if (!AVS->video_info)
 275         {
 276             mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_get_video_info() call failed\n");
 277             return NULL;
 278         }
 279     }
 280
 281     // TODO check field-based ??
 282
 283     /* Video */
 284     if (avs_has_video(AVS->video_info))
 285     {
 286         sh_video_t *sh_video = new_sh_video(demuxer, 0);
 287         found = 1;
 288
 289         if (demuxer->video->id == -1) demuxer->video->id = 0;
 290         if (demuxer->video->id == 0)
 291         demuxer->video->sh = sh_video;
 292         sh_video->ds = demuxer->video;
 293
 294         sh_video->disp_w = AVS->video_info->width;
 295         sh_video->disp_h = AVS->video_info->height;
 296
 297         //sh_video->format = get_mmioFOURCC(AVS->video_info);
 298         sh_video->format = mmioFOURCC('Y', 'V', '1', '2');
 299         sh_video->fps = (double) AVS->video_info->fps_numerator / (double) AVS->video_info->fps_denominator;
 300         sh_video->frametime = 1.0 / sh_video->fps;
 301
 302         sh_video->bih = malloc(sizeof(*sh_video->bih) + (256 * 4));
 303         sh_video->bih->biCompression = sh_video->format;
 304         sh_video->bih->biBitCount = avs_bits_per_pixel(AVS->video_info);
 305         //sh_video->bih->biPlanes = 2;
 306
 307         sh_video->bih->biWidth = AVS->video_info->width;
 308         sh_video->bih->biHeight = AVS->video_info->height;
 309         sh_video->num_frames = 0;
 310         sh_video->num_frames_decoded = 0;
 311     }
 312
 313     /* Audio */
 314     if (avs_has_audio(AVS->video_info))
 315       switch (AVS->video_info->sample_type) {
 316         case AVS_SAMPLE_INT8:  audio_samplesize = 1; break;
 317         case AVS_SAMPLE_INT16: audio_samplesize = 2; break;
 318         case AVS_SAMPLE_INT24: audio_samplesize = 3; break;
 319         case AVS_SAMPLE_INT32:
 320         case AVS_SAMPLE_FLOAT: audio_samplesize = 4; break;
 321         default:
 322           mp_msg(MSGT_DEMUX, MSGL_ERR, "AVS: unknown audio type, disabling\n");
 323       }
 324     if (audio_samplesize)
 325     {
 326         sh_audio_t *sh_audio = new_sh_audio(demuxer, 0);
 327         found = 1;
 328         mp_msg(MSGT_DEMUX, MSGL_V, "AVS: Clip has audio -> Channels = %d - Freq = %d\n", AVS->video_info->nchannels, AVS->video_info->audio_samples_per_second);
 329
 330         if (demuxer->audio->id == -1) demuxer->audio->id = 0;
 331         if (demuxer->audio->id == 0)
 332         demuxer->audio->sh = sh_audio;
 333         sh_audio->ds = demuxer->audio;
 334
 335         sh_audio->wf = malloc(sizeof(*sh_audio->wf));
 336         sh_audio->wf->wFormatTag = sh_audio->format =
 337             (AVS->video_info->sample_type == AVS_SAMPLE_FLOAT) ? 0x3 : 0x1;
 338         sh_audio->wf->nChannels = sh_audio->channels = AVS->video_info->nchannels;
 339         sh_audio->wf->nSamplesPerSec = sh_audio->samplerate = AVS->video_info->audio_samples_per_second;
 340         sh_audio->samplesize = audio_samplesize;
 341         sh_audio->wf->nAvgBytesPerSec = sh_audio->channels * sh_audio->samplesize * sh_audio->samplerate;
 342         sh_audio->wf->nBlockAlign = sh_audio->channels * sh_audio->samplesize;
 343         sh_audio->wf->wBitsPerSample = sh_audio->samplesize * 8;
 344         sh_audio->wf->cbSize = 0;
 345         sh_audio->i_bps = sh_audio->wf->nAvgBytesPerSec;
 346     }
 347
 348     AVS->init = 1;
 349     if (found)
 350         return demuxer;
 351     else
 352         return NULL;
 353 }
 354
 355 static int demux_avs_control(demuxer_t *demuxer, int cmd, void *arg)
 356 {
 357     sh_video_t *sh_video=demuxer->video->sh;
 358     sh_audio_t *sh_audio=demuxer->audio->sh;
 359     AVS_T *AVS = demuxer->priv;
 360
 361     switch(cmd)
 362     {
 363         case DEMUXER_CTRL_GET_TIME_LENGTH:
 364         {
 365             double res = sh_video ? (double)AVS->video_info->num_frames / sh_video->fps : 0;
 366             if (sh_audio)
 367               res = FFMAX(res, (double)AVS->video_info->num_audio_samples / sh_audio->samplerate);
 368             *((double *)arg) = res;
 369             return DEMUXER_CTRL_OK;
 370         }
 371         case DEMUXER_CTRL_GET_PERCENT_POS:
 372         {
 373             if (sh_video)
 374             *((int *)arg) = AVS->frameno * 100 / AVS->video_info->num_frames;
 375             else
 376               *((int *)arg) = AVS->sampleno * 100 / AVS->video_info->num_audio_samples;
 377             return DEMUXER_CTRL_OK;
 378         }
 379     default:
 380         return DEMUXER_CTRL_NOTIMPL;
 381     }
 382 }
 383
 384 static void demux_close_avs(demuxer_t* demuxer)
 385 {
 386     AVS_T *AVS = demuxer->priv;
 387
 388     if (AVS)
 389     {
 390         if (AVS->dll)
 391         {
 392             if (AVS->clip)
 393                 AVS->avs_release_clip(AVS->clip);
 394             mp_msg(MSGT_DEMUX, MSGL_V, "AVS: Unloading avisynth.dll\n");
 395             FreeLibrary(AVS->dll);
 396         }
 397 #ifdef WIN32_LOADER
 398         Restore_LDT_Keeper(AVS->ldt_fs);
 399 #endif
 400         free(AVS);
 401     }
 402 }
 403
 404 static void demux_seek_avs(demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags)
 405 {
 406     sh_video_t *sh_video=demuxer->video->sh;
 407     sh_audio_t *sh_audio=demuxer->audio->sh;
 408     AVS_T *AVS = demuxer->priv;
 409     double video_pos = sh_video ?
 410                        (double)AVS->frameno / sh_video->fps :
 411                        (double)AVS->sampleno / sh_audio->samplerate;
 412     double duration = sh_video ?
 413                       (double)AVS->video_info->num_frames / sh_video->fps :
 414                       (double)AVS->video_info->num_audio_samples / sh_audio->samplerate;
 415
 416     //mp_msg(MSGT_DEMUX, MSGL_V, "AVS: seek rel_seek_secs = %f - flags = %x\n", rel_seek_secs, flags);
 417
 418     if (flags&SEEK_ABSOLUTE) video_pos=0;
 419     if (flags&SEEK_FACTOR) rel_seek_secs *= duration;
 420
 421     video_pos += rel_seek_secs;
 422     if (video_pos < 0) video_pos = 0;
 423
 424     if (sh_video) {
 425       AVS->frameno = FFMIN(video_pos * sh_video->fps,
 426                            AVS->video_info->num_frames);
 427       sh_video->num_frames_decoded = AVS->frameno;
 428       sh_video->num_frames = AVS->frameno;
 429     }
 430     video_pos += audio_delay;
 431     if (video_pos < 0) video_pos = 0;
 432     if (sh_audio)
 433       AVS->sampleno = FFMIN(video_pos * sh_audio->samplerate,
 434                             AVS->video_info->num_audio_samples);
 435 }
 436
 437 static int avs_check_file(demuxer_t *demuxer)
 438 {
 439     mp_msg(MSGT_DEMUX, MSGL_V, "AVS: avs_check_file - attempting to open file %s\n", demuxer->filename);
 440
 441     if (!demuxer->filename) return 0;
 442
 443     /* Avoid crazy memory eating when passing an mpg stream */
 444     if (demuxer->movi_end > MAX_AVS_SIZE)
 445     {
 446         mp_msg(MSGT_DEMUX,MSGL_V, "AVS: File is too big, aborting...\n");
 447         return 0;
 448     }
 449
 450     demuxer->priv = initAVS(demuxer->filename);
 451
 452     if (demuxer->priv)
 453     {
 454         mp_msg(MSGT_DEMUX,MSGL_V, "AVS: Init Ok\n");
 455         return DEMUXER_TYPE_AVS;
 456     }
 457     mp_msg(MSGT_DEMUX,MSGL_V, "AVS: Init failed\n");
 458     return 0;
 459 }
 460
 461
 462 const demuxer_desc_t demuxer_desc_avs = {
 463   "Avisynth demuxer",
 464   "avs",
 465   "AVS",
 466   "Gianluigi Tiesi",
 467   "Requires binary dll",
 468   DEMUXER_TYPE_AVS,
 469   0, // unsafe autodetect
 470   avs_check_file,
 471   demux_avs_fill_buffer,
 472   demux_open_avs,
 473   demux_close_avs,
 474   demux_seek_avs,
 475   demux_avs_control
 476 };