code.vuplus.com Git - vuplus_xbmc/blob - lib/DllAvFormat.h

   1 #pragma once
   2 /*
   3  *      Copyright (C) 2005-2013 Team XBMC
   4  *      http://www.xbmc.org
   5  *
   6  *  This Program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 2, or (at your option)
   9  *  any later version.
  10  *
  11  *  This Program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with XBMC; see the file COPYING.  If not, write to
  18  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  19  *  http://www.gnu.org/copyleft/gpl.html
  20  *
  21  */
  22
  23 #if (defined HAVE_CONFIG_H) && (!defined WIN32)
  24   #include "config.h"
  25 #endif
  26 #include "DynamicDll.h"
  27 #include "DllAvCodec.h"
  28
  29 extern "C" {
  30 #ifndef HAVE_MMX
  31 #define HAVE_MMX
  32 #endif
  33 #ifndef __STDC_CONSTANT_MACROS
  34 #define __STDC_CONSTANT_MACROS
  35 #endif
  36 #ifndef __GNUC__
  37 #pragma warning(disable:4244)
  38 #endif
  39 #if (defined USE_EXTERNAL_FFMPEG)
  40   #if (defined HAVE_LIBAVFORMAT_AVFORMAT_H)
  41     #include <libavformat/avformat.h>
  42   #else
  43     #include <ffmpeg/avformat.h>
  44   #endif
  45   /* xbmc_read_frame_flush() is defined for us in lib/xbmc-dll-symbols/DllAvFormat.c */
  46   void xbmc_read_frame_flush(AVFormatContext *s);
  47 #else
  48   #include "libavformat/avformat.h"
  49   #if defined(TARGET_DARWIN)
  50     void ff_read_frame_flush(AVFormatContext *s);    // internal replacement
  51     #define xbmc_read_frame_flush ff_read_frame_flush
  52   #endif
  53 #endif
  54 }
  55
  56 /* Flag introduced without a version bump */
  57 #ifndef AVSEEK_FORCE
  58 #define AVSEEK_FORCE 0x20000
  59 #endif
  60
  61 typedef int64_t offset_t;
  62
  63 class DllAvFormatInterface
  64 {
  65 public:
  66   virtual ~DllAvFormatInterface() {}
  67   virtual void av_register_all_dont_call(void)=0;
  68   virtual void avformat_network_init_dont_call(void)=0;
  69   virtual void avformat_network_deinit_dont_call(void)=0;
  70   virtual AVInputFormat *av_find_input_format(const char *short_name)=0;
  71   virtual void avformat_close_input(AVFormatContext **s)=0;
  72   virtual int av_read_frame(AVFormatContext *s, AVPacket *pkt)=0;
  73   virtual void av_read_frame_flush(AVFormatContext *s)=0;
  74   virtual int av_read_play(AVFormatContext *s)=0;
  75   virtual int av_read_pause(AVFormatContext *s)=0;
  76   virtual int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)=0;
  77 #if (!defined USE_EXTERNAL_FFMPEG) && (!defined TARGET_DARWIN)
  78   virtual int avformat_find_stream_info_dont_call(AVFormatContext *ic, AVDictionary **options)=0;
  79 #endif
  80   virtual int avformat_open_input(AVFormatContext **ps, const char *filename, AVInputFormat *fmt, AVDictionary **options)=0;
  81   virtual AVIOContext *avio_alloc_context(unsigned char *buffer, int buffer_size, int write_flag, void *opaque,
  82                             int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
  83                             int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
  84                             offset_t (*seek)(void *opaque, offset_t offset, int whence))=0;
  85   virtual AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened)=0;
  86   virtual AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max)=0;
  87   virtual int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt, const char *filename, void *logctx, unsigned int offset, unsigned int max_probe_size)=0;
  88   virtual void av_dump_format(AVFormatContext *ic, int index, const char *url, int is_output)=0;
  89   virtual int avio_open(AVIOContext **s, const char *filename, int flags)=0;
  90   virtual int avio_close(AVIOContext *s)=0;
  91   virtual int avio_open_dyn_buf(AVIOContext **s)=0;
  92   virtual int avio_close_dyn_buf(AVIOContext *s, uint8_t **pbuffer)=0;
  93   virtual offset_t avio_seek(AVIOContext *s, offset_t offset, int whence)=0;
  94   virtual int avio_read(AVIOContext *s, unsigned char *buf, int size)=0;
  95   virtual void avio_w8(AVIOContext *s, int b)=0;
  96   virtual void avio_write(AVIOContext *s, const unsigned char *buf, int size)=0;
  97   virtual void avio_wb24(AVIOContext *s, unsigned int val)=0;
  98   virtual void avio_wb32(AVIOContext *s, unsigned int val)=0;
  99   virtual void avio_wb16(AVIOContext *s, unsigned int val)=0;
 100   virtual AVFormatContext *avformat_alloc_context(void)=0;
 101   virtual AVStream *avformat_new_stream(AVFormatContext *s, AVCodec *c)=0;
 102   virtual AVOutputFormat *av_guess_format(const char *short_name, const char *filename, const char *mime_type)=0;
 103   virtual int avformat_write_header (AVFormatContext *s, AVDictionary **options)=0;
 104   virtual int av_write_trailer(AVFormatContext *s)=0;
 105   virtual int av_write_frame  (AVFormatContext *s, AVPacket *pkt)=0;
 106 };
 107
 108 #if (defined USE_EXTERNAL_FFMPEG) || (defined TARGET_DARWIN)
 109
 110 // Use direct mapping
 111 class DllAvFormat : public DllDynamic, DllAvFormatInterface
 112 {
 113 public:
 114   virtual ~DllAvFormat() {}
 115   virtual void av_register_all()
 116   {
 117     CSingleLock lock(DllAvCodec::m_critSection);
 118     return ::av_register_all();
 119   }
 120   virtual void av_register_all_dont_call() { *(volatile int* )0x0 = 0; }
 121   virtual void avformat_network_init_dont_call() { *(volatile int* )0x0 = 0; }
 122   virtual void avformat_network_deinit_dont_call() { *(volatile int* )0x0 = 0; }
 123   virtual AVInputFormat *av_find_input_format(const char *short_name) { return ::av_find_input_format(short_name); }
 124   virtual void avformat_close_input(AVFormatContext **s) { ::avformat_close_input(s); }
 125   virtual int av_read_frame(AVFormatContext *s, AVPacket *pkt) { return ::av_read_frame(s, pkt); }
 126   virtual void av_read_frame_flush(AVFormatContext *s) { ::xbmc_read_frame_flush(s); }
 127   virtual int av_read_play(AVFormatContext *s) { return ::av_read_play(s); }
 128   virtual int av_read_pause(AVFormatContext *s) { return ::av_read_pause(s); }
 129   virtual int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags) { return ::av_seek_frame(s, stream_index, timestamp, flags); }
 130   virtual int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
 131   {
 132     CSingleLock lock(DllAvCodec::m_critSection);
 133     return ::avformat_find_stream_info(ic, options);
 134   }
 135   virtual int avformat_open_input(AVFormatContext **ps, const char *filename, AVInputFormat *fmt, AVDictionary **options)
 136   { return ::avformat_open_input(ps, filename, fmt, options); }
 137   virtual AVIOContext *avio_alloc_context(unsigned char *buffer, int buffer_size, int write_flag, void *opaque,
 138                             int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
 139                             int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
 140                             offset_t (*seek)(void *opaque, offset_t offset, int whence)) { return ::avio_alloc_context(buffer, buffer_size, write_flag, opaque, read_packet, write_packet, seek); }
 141   virtual AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened) {return ::av_probe_input_format(pd, is_opened); }
 142   virtual AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max) {*score_max = 100; return ::av_probe_input_format(pd, is_opened); } // Use av_probe_input_format, this is not exported by ffmpeg's headers
 143   virtual int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt, const char *filename, void *logctx, unsigned int offset, unsigned int max_probe_size) { return ::av_probe_input_buffer(pb, fmt, filename, logctx, offset, max_probe_size); }
 144   virtual void av_dump_format(AVFormatContext *ic, int index, const char *url, int is_output) { ::av_dump_format(ic, index, url, is_output); }
 145   virtual int avio_open(AVIOContext **s, const char *filename, int flags) { return ::avio_open(s, filename, flags); }
 146   virtual int avio_close(AVIOContext *s) { return ::avio_close(s); }
 147   virtual int avio_open_dyn_buf(AVIOContext **s) { return ::avio_open_dyn_buf(s); }
 148   virtual int avio_close_dyn_buf(AVIOContext *s, uint8_t **pbuffer) { return ::avio_close_dyn_buf(s, pbuffer); }
 149   virtual offset_t avio_seek(AVIOContext *s, offset_t offset, int whence) { return ::avio_seek(s, offset, whence); }
 150   virtual int avio_read(AVIOContext *s, unsigned char *buf, int size) { return ::avio_read(s, buf, size); }
 151   virtual void avio_w8(AVIOContext *s, int b) { ::avio_w8(s, b); }
 152   virtual void avio_write(AVIOContext *s, const unsigned char *buf, int size) { ::avio_write(s, buf, size); }
 153   virtual void avio_wb24(AVIOContext *s, unsigned int val) { ::avio_wb24(s, val); }
 154   virtual void avio_wb32(AVIOContext *s, unsigned int val) { ::avio_wb32(s, val); }
 155   virtual void avio_wb16(AVIOContext *s, unsigned int val) { ::avio_wb16(s, val); }
 156   virtual AVFormatContext *avformat_alloc_context() { return ::avformat_alloc_context(); }
 157   virtual AVStream *avformat_new_stream(AVFormatContext *s, AVCodec *c) { return ::avformat_new_stream(s, c); }
 158   virtual AVOutputFormat *av_guess_format(const char *short_name, const char *filename, const char *mime_type) { return ::av_guess_format(short_name, filename, mime_type); }
 159   virtual int avformat_write_header (AVFormatContext *s, AVDictionary **options) { return ::avformat_write_header (s, options); }
 160   virtual int av_write_trailer(AVFormatContext *s) { return ::av_write_trailer(s); }
 161   virtual int av_write_frame  (AVFormatContext *s, AVPacket *pkt) { return ::av_write_frame(s, pkt); }
 162
 163   // DLL faking.
 164   virtual bool ResolveExports() { return true; }
 165   virtual bool Load() {
 166 #if !defined(TARGET_DARWIN)
 167     CLog::Log(LOGDEBUG, "DllAvFormat: Using libavformat system library");
 168 #endif
 169     CSingleLock lock(DllAvCodec::m_critSection);
 170     if (++m_avformat_refcnt == 1)
 171       ::avformat_network_init();
 172     return true;
 173   }
 174   virtual void Unload() {
 175     CSingleLock lock(DllAvCodec::m_critSection);
 176     if (--m_avformat_refcnt == 0)
 177       ::avformat_network_deinit();
 178   }
 179
 180 protected:
 181   static int m_avformat_refcnt;
 182 };
 183
 184 #else
 185
 186 class DllAvFormat : public DllDynamic, DllAvFormatInterface
 187 {
 188   DECLARE_DLL_WRAPPER(DllAvFormat, DLL_PATH_LIBAVFORMAT)
 189
 190   LOAD_SYMBOLS()
 191
 192   DEFINE_METHOD0(void, av_register_all_dont_call)
 193   DEFINE_METHOD0(void, avformat_network_init_dont_call)
 194   DEFINE_METHOD0(void, avformat_network_deinit_dont_call)
 195   DEFINE_METHOD1(AVInputFormat*, av_find_input_format, (const char *p1))
 196   DEFINE_METHOD1(void, avformat_close_input, (AVFormatContext **p1))
 197   DEFINE_METHOD1(int, av_read_play, (AVFormatContext *p1))
 198   DEFINE_METHOD1(int, av_read_pause, (AVFormatContext *p1))
 199   DEFINE_METHOD1(void, av_read_frame_flush, (AVFormatContext *p1))
 200   DEFINE_FUNC_ALIGNED2(int, __cdecl, av_read_frame, AVFormatContext *, AVPacket *)
 201   DEFINE_FUNC_ALIGNED4(int, __cdecl, av_seek_frame, AVFormatContext*, int, int64_t, int)
 202   DEFINE_FUNC_ALIGNED2(int, __cdecl, avformat_find_stream_info_dont_call, AVFormatContext*, AVDictionary **)
 203   DEFINE_FUNC_ALIGNED4(int, __cdecl, avformat_open_input, AVFormatContext **, const char *, AVInputFormat *, AVDictionary **)
 204   DEFINE_FUNC_ALIGNED2(AVInputFormat*, __cdecl, av_probe_input_format, AVProbeData*, int)
 205   DEFINE_FUNC_ALIGNED3(AVInputFormat*, __cdecl, av_probe_input_format2, AVProbeData*, int, int*)
 206   DEFINE_FUNC_ALIGNED6(int, __cdecl, av_probe_input_buffer, AVIOContext *, AVInputFormat **, const char *, void *, unsigned int, unsigned int)
 207   DEFINE_FUNC_ALIGNED3(int, __cdecl, avio_read, AVIOContext*, unsigned char *, int)
 208   DEFINE_FUNC_ALIGNED2(void, __cdecl, avio_w8, AVIOContext*, int)
 209   DEFINE_FUNC_ALIGNED3(void, __cdecl, avio_write, AVIOContext*, const unsigned char *, int)
 210   DEFINE_FUNC_ALIGNED2(void, __cdecl, avio_wb24, AVIOContext*, unsigned int)
 211   DEFINE_FUNC_ALIGNED2(void, __cdecl, avio_wb32, AVIOContext*, unsigned int)
 212   DEFINE_FUNC_ALIGNED2(void, __cdecl, avio_wb16, AVIOContext*, unsigned int)
 213   DEFINE_METHOD7(AVIOContext *, avio_alloc_context, (unsigned char *p1, int p2, int p3, void *p4,
 214                   int (*p5)(void *opaque, uint8_t *buf, int buf_size),
 215                   int (*p6)(void *opaque, uint8_t *buf, int buf_size),
 216                   offset_t (*p7)(void *opaque, offset_t offset, int whence)))
 217   DEFINE_METHOD4(void, av_dump_format, (AVFormatContext *p1, int p2, const char *p3, int p4))
 218   DEFINE_METHOD3(int, avio_open, (AVIOContext **p1, const char *p2, int p3))
 219   DEFINE_METHOD1(int, avio_close, (AVIOContext *p1))
 220   DEFINE_METHOD1(int, avio_open_dyn_buf, (AVIOContext **p1))
 221   DEFINE_METHOD2(int, avio_close_dyn_buf, (AVIOContext *p1, uint8_t **p2))
 222   DEFINE_METHOD3(offset_t, avio_seek, (AVIOContext *p1, offset_t p2, int p3))
 223   DEFINE_METHOD0(AVFormatContext *, avformat_alloc_context)
 224   DEFINE_METHOD2(AVStream *, avformat_new_stream, (AVFormatContext *p1, AVCodec *p2))
 225   DEFINE_METHOD3(AVOutputFormat *, av_guess_format, (const char *p1, const char *p2, const char *p3))
 226   DEFINE_METHOD2(int, avformat_write_header , (AVFormatContext *p1, AVDictionary **p2))
 227   DEFINE_METHOD1(int, av_write_trailer, (AVFormatContext *p1))
 228   DEFINE_METHOD2(int, av_write_frame  , (AVFormatContext *p1, AVPacket *p2))
 229   BEGIN_METHOD_RESOLVE()
 230     RESOLVE_METHOD_RENAME(av_register_all, av_register_all_dont_call)
 231     RESOLVE_METHOD_RENAME(avformat_network_init,   avformat_network_init_dont_call)
 232     RESOLVE_METHOD_RENAME(avformat_network_deinit, avformat_network_deinit_dont_call)
 233     RESOLVE_METHOD(av_find_input_format)
 234     RESOLVE_METHOD(avformat_close_input)
 235     RESOLVE_METHOD(av_read_frame)
 236     RESOLVE_METHOD(av_read_play)
 237     RESOLVE_METHOD(av_read_pause)
 238     RESOLVE_METHOD(av_read_frame_flush)
 239     RESOLVE_METHOD(av_seek_frame)
 240     RESOLVE_METHOD_RENAME(avformat_find_stream_info, avformat_find_stream_info_dont_call)
 241     RESOLVE_METHOD(avformat_open_input)
 242     RESOLVE_METHOD(avio_alloc_context)
 243     RESOLVE_METHOD(av_probe_input_format)
 244     RESOLVE_METHOD(av_probe_input_format2)
 245     RESOLVE_METHOD(av_probe_input_buffer)
 246     RESOLVE_METHOD(av_dump_format)
 247     RESOLVE_METHOD(avio_open)
 248     RESOLVE_METHOD(avio_close)
 249     RESOLVE_METHOD(avio_open_dyn_buf)
 250     RESOLVE_METHOD(avio_close_dyn_buf)
 251     RESOLVE_METHOD(avio_seek)
 252     RESOLVE_METHOD(avio_read)
 253     RESOLVE_METHOD(avio_w8)
 254     RESOLVE_METHOD(avio_write)
 255     RESOLVE_METHOD(avio_wb24)
 256     RESOLVE_METHOD(avio_wb32)
 257     RESOLVE_METHOD(avio_wb16)
 258     RESOLVE_METHOD(avformat_alloc_context)
 259     RESOLVE_METHOD(avformat_new_stream)
 260     RESOLVE_METHOD(av_guess_format)
 261     RESOLVE_METHOD(avformat_write_header)
 262     RESOLVE_METHOD(av_write_trailer)
 263     RESOLVE_METHOD(av_write_frame)
 264   END_METHOD_RESOLVE()
 265
 266   /* dependencies of libavformat */
 267   DllAvCodec m_dllAvCodec;
 268   // DllAvUtil loaded implicitely by m_dllAvCodec
 269
 270 public:
 271   void av_register_all()
 272   {
 273     CSingleLock lock(DllAvCodec::m_critSection);
 274     av_register_all_dont_call();
 275   }
 276   int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
 277   {
 278     CSingleLock lock(DllAvCodec::m_critSection);
 279     return avformat_find_stream_info_dont_call(ic, options);
 280   }
 281
 282   virtual bool Load()
 283   {
 284     if (!m_dllAvCodec.Load())
 285       return false;
 286     bool loaded = DllDynamic::Load();
 287
 288     CSingleLock lock(DllAvCodec::m_critSection);
 289     if (++m_avformat_refcnt == 1 && loaded)
 290       avformat_network_init_dont_call();
 291     return loaded;
 292   }
 293
 294   virtual void Unload()
 295   {
 296     CSingleLock lock(DllAvCodec::m_critSection);
 297     if (--m_avformat_refcnt == 0 && DllDynamic::IsLoaded())
 298       avformat_network_deinit_dont_call();
 299
 300     DllDynamic::Unload();
 301   }
 302
 303 protected:
 304   static int m_avformat_refcnt;
 305 };
 306
 307 #endif