dlls/dmusic/buffer.c

   1 /*
   2  * IDirectMusicBuffer Implementation
   3  *
   4  * Copyright (C) 2003-2004 Rok Mandeljc
   5  * Copyright (C) 2012 Christian Costa
   6  *
   7  * This program 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  * This program 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
  18  * License along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "dmusic_private.h"
  23 #include "dmobject.h"
  24 #include "initguid.h"
  25 #include "dmksctrl.h"
  26
  27 WINE_DEFAULT_DEBUG_CHANNEL(dmusic);
  28
  29 static inline IDirectMusicBufferImpl *impl_from_IDirectMusicBuffer(IDirectMusicBuffer *iface)
  30 {
  31     return CONTAINING_RECORD(iface, IDirectMusicBufferImpl, IDirectMusicBuffer_iface);
  32 }
  33
  34 /* IDirectMusicBufferImpl IUnknown part: */
  35 static HRESULT WINAPI IDirectMusicBufferImpl_QueryInterface(LPDIRECTMUSICBUFFER iface, REFIID riid, LPVOID *ret_iface)
  36 {
  37     TRACE("(%p)->(%s, %p)\n", iface, debugstr_dmguid(riid), ret_iface);
  38
  39     if (IsEqualIID(riid, &IID_IUnknown) ||
  40         IsEqualIID(riid, &IID_IDirectMusicBuffer))
  41     {
  42         IDirectMusicBuffer_AddRef(iface);
  43         *ret_iface = iface;
  44         return S_OK;
  45     }
  46
  47     *ret_iface = NULL;
  48
  49     WARN("(%p)->(%s, %p): not found\n", iface, debugstr_dmguid(riid), ret_iface);
  50
  51     return E_NOINTERFACE;
  52 }
  53
  54 static ULONG WINAPI IDirectMusicBufferImpl_AddRef(LPDIRECTMUSICBUFFER iface)
  55 {
  56     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
  57     ULONG ref = InterlockedIncrement(&This->ref);
  58
  59     TRACE("(%p)->(): new ref = %u\n", iface, ref);
  60
  61     return ref;
  62 }
  63
  64 static ULONG WINAPI IDirectMusicBufferImpl_Release(LPDIRECTMUSICBUFFER iface)
  65 {
  66     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
  67     ULONG ref = InterlockedDecrement(&This->ref);
  68
  69     TRACE("(%p)->(): new ref = %u\n", iface, ref);
  70
  71     if (!ref) {
  72         HeapFree(GetProcessHeap(), 0, This->data);
  73         HeapFree(GetProcessHeap(), 0, This);
  74         DMUSIC_UnlockModule();
  75     }
  76
  77     return ref;
  78 }
  79
  80 /* IDirectMusicBufferImpl IDirectMusicBuffer part: */
  81 static HRESULT WINAPI IDirectMusicBufferImpl_Flush(LPDIRECTMUSICBUFFER iface)
  82 {
  83     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
  84
  85     TRACE("(%p)->()\n", iface);
  86
  87     This->write_pos = 0;
  88
  89     return S_OK;
  90 }
  91
  92 static HRESULT WINAPI IDirectMusicBufferImpl_TotalTime(LPDIRECTMUSICBUFFER iface, LPREFERENCE_TIME prtTime)
  93 {
  94     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
  95
  96     FIXME("(%p, %p): stub\n", This, prtTime);
  97
  98     return S_OK;
  99 }
 100
 101 static HRESULT WINAPI IDirectMusicBufferImpl_PackStructured(LPDIRECTMUSICBUFFER iface, REFERENCE_TIME ref_time, DWORD channel_group, DWORD channel_message)
 102 {
 103     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 104     DWORD new_write_pos = This->write_pos + DMUS_EVENT_SIZE(sizeof(channel_message));
 105     DMUS_EVENTHEADER *header;
 106
 107     TRACE("(%p)->(0x%s, %u, 0x%x)\n", iface, wine_dbgstr_longlong(ref_time), channel_group, channel_message);
 108
 109     if (new_write_pos > This->size)
 110         return DMUS_E_BUFFER_FULL;
 111
 112     /* Channel_message 0xZZYYXX (3 bytes) is a midi message where XX = status byte, YY = byte 1 and ZZ = byte 2 */
 113
 114     if (!(channel_message & 0x80))
 115     {
 116         /* Status byte MSB is always set */
 117         return DMUS_E_INVALID_EVENT;
 118     }
 119
 120     if (!This->write_pos)
 121         This->start_time = ref_time;
 122
 123     header = (DMUS_EVENTHEADER*)&This->data[This->write_pos];
 124     header->cbEvent = 3; /* Midi message takes 4 bytes space but only 3 are relevant */
 125     header->dwChannelGroup = channel_group;
 126     header->rtDelta = ref_time - This->start_time;
 127     header->dwFlags = DMUS_EVENT_STRUCTURED;
 128
 129     *(DWORD*)&header[1] = channel_message;
 130     This->write_pos = new_write_pos;
 131
 132     return S_OK;
 133 }
 134
 135 static HRESULT WINAPI IDirectMusicBufferImpl_PackUnstructured(IDirectMusicBuffer *iface,
 136         REFERENCE_TIME ref_time, DWORD channel_group, DWORD len, BYTE *data)
 137 {
 138     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 139     DWORD new_write_pos = This->write_pos + DMUS_EVENT_SIZE(len);
 140     DMUS_EVENTHEADER *header;
 141
 142     TRACE("(%p, 0x%s, %d, %d, %p)\n", This, wine_dbgstr_longlong(ref_time), channel_group, len, data);
 143
 144     if (new_write_pos > This->size)
 145         return DMUS_E_BUFFER_FULL;
 146
 147     if (!This->write_pos)
 148         This->start_time = ref_time;
 149
 150     header = (DMUS_EVENTHEADER*)&This->data[This->write_pos];
 151     header->cbEvent = len;
 152     header->dwChannelGroup = channel_group;
 153     header->rtDelta = ref_time - This->start_time;
 154     header->dwFlags = 0;
 155
 156     memcpy(&header[1], data, len);
 157     This->write_pos = new_write_pos;
 158
 159     return S_OK;
 160 }
 161
 162 static HRESULT WINAPI IDirectMusicBufferImpl_ResetReadPtr(LPDIRECTMUSICBUFFER iface)
 163 {
 164     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 165
 166     FIXME("(%p): stub\n", This);
 167
 168     return S_OK;
 169 }
 170
 171 static HRESULT WINAPI IDirectMusicBufferImpl_GetNextEvent(LPDIRECTMUSICBUFFER iface, LPREFERENCE_TIME prt, LPDWORD pdwChannelGroup, LPDWORD pdwLength, LPBYTE* ppData)
 172 {
 173     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 174
 175     FIXME("(%p, %p, %p, %p, %p): stub\n", This, prt, pdwChannelGroup, pdwLength, ppData);
 176
 177     return S_OK;
 178 }
 179
 180 static HRESULT WINAPI IDirectMusicBufferImpl_GetRawBufferPtr(LPDIRECTMUSICBUFFER iface, LPBYTE* data)
 181 {
 182     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 183
 184     TRACE("(%p)->(%p)\n", iface, data);
 185
 186     if (!data)
 187         return E_POINTER;
 188
 189     *data = This->data;
 190
 191     return S_OK;
 192 }
 193
 194 static HRESULT WINAPI IDirectMusicBufferImpl_GetStartTime(LPDIRECTMUSICBUFFER iface, LPREFERENCE_TIME ref_time)
 195 {
 196     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 197
 198     TRACE("(%p)->(%p)\n", iface, ref_time);
 199
 200     if (!ref_time)
 201         return E_POINTER;
 202     if (!This->write_pos)
 203         return DMUS_E_BUFFER_EMPTY;
 204
 205     *ref_time = This->start_time;
 206
 207     return S_OK;
 208 }
 209
 210 static HRESULT WINAPI IDirectMusicBufferImpl_GetUsedBytes(LPDIRECTMUSICBUFFER iface, LPDWORD used_bytes)
 211 {
 212     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 213
 214     TRACE("(%p)->(%p)\n", iface, used_bytes);
 215
 216     if (!used_bytes)
 217         return E_POINTER;
 218
 219     *used_bytes = This->write_pos;
 220
 221     return S_OK;
 222 }
 223
 224 static HRESULT WINAPI IDirectMusicBufferImpl_GetMaxBytes(LPDIRECTMUSICBUFFER iface, LPDWORD max_bytes)
 225 {
 226     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 227
 228     TRACE("(%p)->(%p)\n", iface, max_bytes);
 229
 230     if (!max_bytes)
 231         return E_POINTER;
 232
 233     *max_bytes = This->size;
 234
 235     return S_OK;
 236 }
 237
 238 static HRESULT WINAPI IDirectMusicBufferImpl_GetBufferFormat(LPDIRECTMUSICBUFFER iface, LPGUID format)
 239 {
 240     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 241
 242     TRACE("(%p)->(%p)\n", iface, format);
 243
 244     if (!format)
 245         return E_POINTER;
 246
 247     *format = This->format;
 248     return S_OK;
 249 }
 250
 251 static HRESULT WINAPI IDirectMusicBufferImpl_SetStartTime(LPDIRECTMUSICBUFFER iface, REFERENCE_TIME ref_time)
 252 {
 253     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 254
 255     TRACE("(%p)->(0x%s)\n", This, wine_dbgstr_longlong(ref_time));
 256
 257     This->start_time = ref_time;
 258
 259     return S_OK;
 260 }
 261
 262 static HRESULT WINAPI IDirectMusicBufferImpl_SetUsedBytes(LPDIRECTMUSICBUFFER iface, DWORD used_bytes)
 263 {
 264     IDirectMusicBufferImpl *This = impl_from_IDirectMusicBuffer(iface);
 265
 266     TRACE("(%p)->(%u)\n", iface, used_bytes);
 267
 268     if (used_bytes > This->size)
 269         return DMUS_E_BUFFER_FULL;
 270
 271     This->write_pos = used_bytes;
 272
 273     return S_OK;
 274 }
 275
 276 static const IDirectMusicBufferVtbl DirectMusicBuffer_Vtbl = {
 277         IDirectMusicBufferImpl_QueryInterface,
 278         IDirectMusicBufferImpl_AddRef,
 279         IDirectMusicBufferImpl_Release,
 280         IDirectMusicBufferImpl_Flush,
 281         IDirectMusicBufferImpl_TotalTime,
 282         IDirectMusicBufferImpl_PackStructured,
 283         IDirectMusicBufferImpl_PackUnstructured,
 284         IDirectMusicBufferImpl_ResetReadPtr,
 285         IDirectMusicBufferImpl_GetNextEvent,
 286         IDirectMusicBufferImpl_GetRawBufferPtr,
 287         IDirectMusicBufferImpl_GetStartTime,
 288         IDirectMusicBufferImpl_GetUsedBytes,
 289         IDirectMusicBufferImpl_GetMaxBytes,
 290         IDirectMusicBufferImpl_GetBufferFormat,
 291         IDirectMusicBufferImpl_SetStartTime,
 292         IDirectMusicBufferImpl_SetUsedBytes
 293 };
 294
 295 HRESULT DMUSIC_CreateDirectMusicBufferImpl(LPDMUS_BUFFERDESC desc, LPVOID* ret_iface)
 296 {
 297     IDirectMusicBufferImpl* dmbuffer;
 298
 299     TRACE("(%p, %p)\n", desc, ret_iface);
 300
 301     *ret_iface = NULL;
 302
 303     dmbuffer = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectMusicBufferImpl));
 304     if (!dmbuffer)
 305         return E_OUTOFMEMORY;
 306
 307     dmbuffer->IDirectMusicBuffer_iface.lpVtbl = &DirectMusicBuffer_Vtbl;
 308     dmbuffer->ref = 1;
 309
 310     if (IsEqualGUID(&desc->guidBufferFormat, &GUID_NULL))
 311         dmbuffer->format = KSDATAFORMAT_SUBTYPE_MIDI;
 312     else
 313         dmbuffer->format = desc->guidBufferFormat;
 314     dmbuffer->size = (desc->cbBuffer + 3) & ~3; /* Buffer size must be multiple of 4 bytes */
 315
 316     dmbuffer->data = HeapAlloc(GetProcessHeap(), 0, dmbuffer->size);
 317     if (!dmbuffer->data) {
 318         HeapFree(GetProcessHeap(), 0, dmbuffer);
 319         return E_OUTOFMEMORY;
 320     }
 321
 322     DMUSIC_LockModule();
 323     *ret_iface = &dmbuffer->IDirectMusicBuffer_iface;
 324
 325     return S_OK;
 326 }