dlls/winegstreamer/gst_private.h

   1 /*
   2  * GStreamer splitter + decoder, adapted from parser.c
   3  *
   4  * Copyright 2010 Maarten Lankhorst for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library 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 GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #ifndef __GST_PRIVATE_INCLUDED__
  22 #define __GST_PRIVATE_INCLUDED__
  23
  24 #include <assert.h>
  25 #include <limits.h>
  26 #include <stdarg.h>
  27 #include <stdbool.h>
  28 #include <stdint.h>
  29 #include <stdio.h>
  30
  31 #define COBJMACROS
  32 #define NONAMELESSSTRUCT
  33 #define NONAMELESSUNION
  34 #include "dshow.h"
  35 #include "mfidl.h"
  36 #include "wmsdk.h"
  37 #include "wine/debug.h"
  38 #include "wine/strmbase.h"
  39
  40 #include "unixlib.h"
  41
  42 bool array_reserve(void **elements, size_t *capacity, size_t count, size_t size);
  43
  44 static inline const char *debugstr_time(REFERENCE_TIME time)
  45 {
  46     ULONGLONG abstime = time >= 0 ? time : -time;
  47     unsigned int i = 0, j = 0;
  48     char buffer[23], rev[23];
  49
  50     while (abstime || i <= 8)
  51     {
  52         buffer[i++] = '0' + (abstime % 10);
  53         abstime /= 10;
  54         if (i == 7) buffer[i++] = '.';
  55     }
  56     if (time < 0) buffer[i++] = '-';
  57
  58     while (i--) rev[j++] = buffer[i];
  59     while (rev[j-1] == '0' && rev[j-2] != '.') --j;
  60     rev[j] = 0;
  61
  62     return wine_dbg_sprintf("%s", rev);
  63 }
  64
  65 #define MEDIATIME_FROM_BYTES(x) ((LONGLONG)(x) * 10000000)
  66
  67 struct wg_sample_queue;
  68
  69 HRESULT wg_sample_queue_create(struct wg_sample_queue **out);
  70 void wg_sample_queue_destroy(struct wg_sample_queue *queue);
  71 void wg_sample_queue_flush(struct wg_sample_queue *queue, bool all);
  72
  73 struct wg_parser *wg_parser_create(enum wg_parser_type type, bool unlimited_buffering);
  74 void wg_parser_destroy(struct wg_parser *parser);
  75
  76 HRESULT wg_parser_connect(struct wg_parser *parser, uint64_t file_size);
  77 void wg_parser_disconnect(struct wg_parser *parser);
  78
  79 bool wg_parser_get_next_read_offset(struct wg_parser *parser, uint64_t *offset, uint32_t *size);
  80 void wg_parser_push_data(struct wg_parser *parser, const void *data, uint32_t size);
  81
  82 uint32_t wg_parser_get_stream_count(struct wg_parser *parser);
  83 struct wg_parser_stream *wg_parser_get_stream(struct wg_parser *parser, uint32_t index);
  84
  85 void wg_parser_stream_get_preferred_format(struct wg_parser_stream *stream, struct wg_format *format);
  86 void wg_parser_stream_enable(struct wg_parser_stream *stream, const struct wg_format *format);
  87 void wg_parser_stream_disable(struct wg_parser_stream *stream);
  88
  89 bool wg_parser_stream_get_buffer(struct wg_parser_stream *stream, struct wg_parser_buffer *buffer);
  90 bool wg_parser_stream_copy_buffer(struct wg_parser_stream *stream,
  91         void *data, uint32_t offset, uint32_t size);
  92 void wg_parser_stream_release_buffer(struct wg_parser_stream *stream);
  93 void wg_parser_stream_notify_qos(struct wg_parser_stream *stream,
  94         bool underflow, double proportion, int64_t diff, uint64_t timestamp);
  95
  96 /* Returns the duration in 100-nanosecond units. */
  97 uint64_t wg_parser_stream_get_duration(struct wg_parser_stream *stream);
  98 /* start_pos and stop_pos are in 100-nanosecond units. */
  99 void wg_parser_stream_seek(struct wg_parser_stream *stream, double rate,
 100         uint64_t start_pos, uint64_t stop_pos, DWORD start_flags, DWORD stop_flags);
 101
 102 struct wg_transform *wg_transform_create(const struct wg_format *input_format,
 103         const struct wg_format *output_format);
 104 void wg_transform_destroy(struct wg_transform *transform);
 105 bool wg_transform_set_output_format(struct wg_transform *transform, struct wg_format *format);
 106
 107 unsigned int wg_format_get_max_size(const struct wg_format *format);
 108
 109 HRESULT avi_splitter_create(IUnknown *outer, IUnknown **out);
 110 HRESULT decodebin_parser_create(IUnknown *outer, IUnknown **out);
 111 HRESULT mpeg_audio_codec_create(IUnknown *outer, IUnknown **out);
 112 HRESULT mpeg_splitter_create(IUnknown *outer, IUnknown **out);
 113 HRESULT wave_parser_create(IUnknown *outer, IUnknown **out);
 114 HRESULT wma_decoder_create(IUnknown *outer, IUnknown **out);
 115 HRESULT resampler_create(IUnknown *outer, IUnknown **out);
 116 HRESULT color_convert_create(IUnknown *outer, IUnknown **out);
 117
 118 bool amt_from_wg_format(AM_MEDIA_TYPE *mt, const struct wg_format *format, bool wm);
 119 bool amt_to_wg_format(const AM_MEDIA_TYPE *mt, struct wg_format *format);
 120
 121 BOOL init_gstreamer(void);
 122
 123 extern HRESULT mfplat_get_class_object(REFCLSID rclsid, REFIID riid, void **obj);
 124 extern HRESULT mfplat_DllRegisterServer(void);
 125
 126 IMFMediaType *mf_media_type_from_wg_format(const struct wg_format *format);
 127 void mf_media_type_to_wg_format(IMFMediaType *type, struct wg_format *format);
 128
 129 HRESULT wg_sample_create_mf(IMFSample *sample, struct wg_sample **out);
 130 HRESULT wg_sample_create_quartz(IMediaSample *sample, struct wg_sample **out);
 131 void wg_sample_release(struct wg_sample *wg_sample);
 132
 133 HRESULT wg_transform_push_mf(struct wg_transform *transform, IMFSample *sample,
 134         struct wg_sample_queue *queue);
 135 HRESULT wg_transform_push_quartz(struct wg_transform *transform, struct wg_sample *sample,
 136         struct wg_sample_queue *queue);
 137 HRESULT wg_transform_read_mf(struct wg_transform *transform, struct wg_sample *sample,
 138         struct wg_format *format, DWORD *flags);
 139 HRESULT wg_transform_read_quartz(struct wg_transform *transform, struct wg_sample *sample);
 140
 141 HRESULT winegstreamer_stream_handler_create(REFIID riid, void **obj);
 142
 143 HRESULT h264_decoder_create(REFIID riid, void **ret);
 144 HRESULT video_processor_create(REFIID riid, void **ret);
 145
 146 struct wm_stream
 147 {
 148     struct wm_reader *reader;
 149     struct wg_parser_stream *wg_stream;
 150     struct wg_format format;
 151     WMT_STREAM_SELECTION selection;
 152     WORD index;
 153     bool eos;
 154     bool allocate_output;
 155     bool allocate_stream;
 156     /* Note that we only pretend to read compressed samples, and instead output
 157      * uncompressed samples regardless of whether we are configured to read
 158      * compressed samples. Rather, the behaviour of the reader objects differs
 159      * in nontrivial ways depending on this field. */
 160     bool read_compressed;
 161 };
 162
 163 struct wm_reader
 164 {
 165     IWMHeaderInfo3 IWMHeaderInfo3_iface;
 166     IWMLanguageList IWMLanguageList_iface;
 167     IWMPacketSize2 IWMPacketSize2_iface;
 168     IWMProfile3 IWMProfile3_iface;
 169     IWMReaderPlaylistBurn IWMReaderPlaylistBurn_iface;
 170     IWMReaderTimecode IWMReaderTimecode_iface;
 171     LONG refcount;
 172     CRITICAL_SECTION cs;
 173
 174     QWORD start_time;
 175
 176     IStream *source_stream;
 177     HANDLE file;
 178     HANDLE read_thread;
 179     bool read_thread_shutdown;
 180     struct wg_parser *wg_parser;
 181
 182     struct wm_stream *streams;
 183     WORD stream_count;
 184
 185     const struct wm_reader_ops *ops;
 186 };
 187
 188 struct wm_reader_ops
 189 {
 190     void *(*query_interface)(struct wm_reader *reader, REFIID iid);
 191     void (*destroy)(struct wm_reader *reader);
 192 };
 193
 194 void wm_reader_cleanup(struct wm_reader *reader);
 195 HRESULT wm_reader_close(struct wm_reader *reader);
 196 HRESULT wm_reader_get_max_stream_size(struct wm_reader *reader, WORD stream_number, DWORD *size);
 197 HRESULT wm_reader_get_output_format(struct wm_reader *reader, DWORD output,
 198         DWORD index, IWMOutputMediaProps **props);
 199 HRESULT wm_reader_get_output_format_count(struct wm_reader *reader, DWORD output, DWORD *count);
 200 HRESULT wm_reader_get_output_props(struct wm_reader *reader, DWORD output,
 201         IWMOutputMediaProps **props);
 202 struct wm_stream *wm_reader_get_stream_by_stream_number(struct wm_reader *reader,
 203         WORD stream_number);
 204 HRESULT wm_reader_get_stream_sample(struct wm_reader *reader, IWMReaderCallbackAdvanced *callback_advanced, WORD stream_number,
 205         INSSBuffer **ret_sample, QWORD *pts, QWORD *duration, DWORD *flags, WORD *ret_stream_number);
 206 HRESULT wm_reader_get_stream_selection(struct wm_reader *reader,
 207         WORD stream_number, WMT_STREAM_SELECTION *selection);
 208 void wm_reader_init(struct wm_reader *reader, const struct wm_reader_ops *ops);
 209 HRESULT wm_reader_open_file(struct wm_reader *reader, const WCHAR *filename);
 210 HRESULT wm_reader_open_stream(struct wm_reader *reader, IStream *stream);
 211 void wm_reader_seek(struct wm_reader *reader, QWORD start, LONGLONG duration);
 212 HRESULT wm_reader_set_allocate_for_output(struct wm_reader *reader, DWORD output, BOOL allocate);
 213 HRESULT wm_reader_set_allocate_for_stream(struct wm_reader *reader, WORD stream_number, BOOL allocate);
 214 HRESULT wm_reader_set_output_props(struct wm_reader *reader, DWORD output,
 215         IWMOutputMediaProps *props);
 216 HRESULT wm_reader_set_read_compressed(struct wm_reader *reader,
 217         WORD stream_number, BOOL compressed);
 218 HRESULT wm_reader_set_streams_selected(struct wm_reader *reader, WORD count,
 219         const WORD *stream_numbers, const WMT_STREAM_SELECTION *selections);
 220
 221 #endif /* __GST_PRIVATE_INCLUDED__ */