2 * Copyright 2018 Alistair Leslie-Hughes
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 #include "mfplat_private.h"
31 WINE_DEFAULT_DEBUG_CHANNEL(mfplat
);
33 #define ALIGN_SIZE(size, alignment) (((size) + (alignment)) & ~((alignment)))
35 typedef void (*p_copy_image_func
)(BYTE
*dest
, LONG dest_stride
, const BYTE
*src
, LONG src_stride
, DWORD width
, DWORD lines
);
39 IMFMediaBuffer IMFMediaBuffer_iface
;
40 IMF2DBuffer2 IMF2DBuffer2_iface
;
41 IMFDXGIBuffer IMFDXGIBuffer_iface
;
42 IMFGetService IMFGetService_iface
;
59 p_copy_image_func copy_image
;
63 IDirect3DSurface9
*surface
;
68 ID3D11Texture2D
*texture
;
69 unsigned int sub_resource_idx
;
70 ID3D11Texture2D
*rb_texture
;
71 D3D11_MAPPED_SUBRESOURCE map_desc
;
72 struct attributes attributes
;
78 static void copy_image(const struct buffer
*buffer
, BYTE
*dest
, LONG dest_stride
, const BYTE
*src
,
79 LONG src_stride
, DWORD width
, DWORD lines
)
81 MFCopyImage(dest
, dest_stride
, src
, src_stride
, width
, lines
);
83 if (buffer
->_2d
.copy_image
)
85 dest
+= dest_stride
* lines
;
86 src
+= src_stride
* lines
;
87 buffer
->_2d
.copy_image(dest
, dest_stride
, src
, src_stride
, width
, lines
);
91 static void copy_image_nv12(BYTE
*dest
, LONG dest_stride
, const BYTE
*src
, LONG src_stride
, DWORD width
, DWORD lines
)
93 MFCopyImage(dest
, dest_stride
, src
, src_stride
, width
, lines
/ 2);
96 static void copy_image_imc1(BYTE
*dest
, LONG dest_stride
, const BYTE
*src
, LONG src_stride
, DWORD width
, DWORD lines
)
98 MFCopyImage(dest
, dest_stride
, src
, src_stride
, width
/ 2, lines
);
101 static void copy_image_imc2(BYTE
*dest
, LONG dest_stride
, const BYTE
*src
, LONG src_stride
, DWORD width
, DWORD lines
)
103 MFCopyImage(dest
, dest_stride
, src
, src_stride
, width
/ 2, lines
/ 2);
104 MFCopyImage(dest
+ dest_stride
/ 2, dest_stride
, src
+ src_stride
/ 2, src_stride
, width
/ 2, lines
/ 2);
107 static inline struct buffer
*impl_from_IMFMediaBuffer(IMFMediaBuffer
*iface
)
109 return CONTAINING_RECORD(iface
, struct buffer
, IMFMediaBuffer_iface
);
112 static struct buffer
*impl_from_IMF2DBuffer2(IMF2DBuffer2
*iface
)
114 return CONTAINING_RECORD(iface
, struct buffer
, IMF2DBuffer2_iface
);
117 static struct buffer
*impl_from_IMFGetService(IMFGetService
*iface
)
119 return CONTAINING_RECORD(iface
, struct buffer
, IMFGetService_iface
);
122 static struct buffer
*impl_from_IMFDXGIBuffer(IMFDXGIBuffer
*iface
)
124 return CONTAINING_RECORD(iface
, struct buffer
, IMFDXGIBuffer_iface
);
127 static HRESULT WINAPI
memory_buffer_QueryInterface(IMFMediaBuffer
*iface
, REFIID riid
, void **out
)
129 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
131 TRACE("%p, %s, %p.\n", iface
, debugstr_guid(riid
), out
);
133 if (IsEqualIID(riid
, &IID_IMFMediaBuffer
) ||
134 IsEqualIID(riid
, &IID_IUnknown
))
136 *out
= &buffer
->IMFMediaBuffer_iface
;
137 IMFMediaBuffer_AddRef(iface
);
141 WARN("Unsupported %s.\n", debugstr_guid(riid
));
143 return E_NOINTERFACE
;
146 static ULONG WINAPI
memory_buffer_AddRef(IMFMediaBuffer
*iface
)
148 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
149 ULONG refcount
= InterlockedIncrement(&buffer
->refcount
);
151 TRACE("%p, refcount %lu.\n", buffer
, refcount
);
156 static ULONG WINAPI
memory_buffer_Release(IMFMediaBuffer
*iface
)
158 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
159 ULONG refcount
= InterlockedDecrement(&buffer
->refcount
);
161 TRACE("%p, refcount %lu.\n", iface
, refcount
);
165 if (buffer
->d3d9_surface
.surface
)
166 IDirect3DSurface9_Release(buffer
->d3d9_surface
.surface
);
167 if (buffer
->dxgi_surface
.texture
)
169 ID3D11Texture2D_Release(buffer
->dxgi_surface
.texture
);
170 if (buffer
->dxgi_surface
.rb_texture
)
171 ID3D11Texture2D_Release(buffer
->dxgi_surface
.rb_texture
);
172 clear_attributes_object(&buffer
->dxgi_surface
.attributes
);
174 DeleteCriticalSection(&buffer
->cs
);
175 free(buffer
->_2d
.linear_buffer
);
176 _aligned_free(buffer
->data
);
183 static HRESULT WINAPI
memory_buffer_Lock(IMFMediaBuffer
*iface
, BYTE
**data
, DWORD
*max_length
, DWORD
*current_length
)
185 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
187 TRACE("%p, %p %p, %p.\n", iface
, data
, max_length
, current_length
);
192 *data
= buffer
->data
;
194 *max_length
= buffer
->max_length
;
196 *current_length
= buffer
->current_length
;
201 static HRESULT WINAPI
memory_buffer_Unlock(IMFMediaBuffer
*iface
)
203 TRACE("%p.\n", iface
);
208 static HRESULT WINAPI
memory_buffer_GetCurrentLength(IMFMediaBuffer
*iface
, DWORD
*current_length
)
210 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
212 TRACE("%p.\n", iface
);
217 *current_length
= buffer
->current_length
;
222 static HRESULT WINAPI
memory_buffer_SetCurrentLength(IMFMediaBuffer
*iface
, DWORD current_length
)
224 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
226 TRACE("%p, %lu.\n", iface
, current_length
);
228 if (current_length
> buffer
->max_length
)
231 buffer
->current_length
= current_length
;
236 static HRESULT WINAPI
memory_buffer_GetMaxLength(IMFMediaBuffer
*iface
, DWORD
*max_length
)
238 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
240 TRACE("%p, %p.\n", iface
, max_length
);
245 *max_length
= buffer
->max_length
;
250 static const IMFMediaBufferVtbl memory_1d_buffer_vtbl
=
252 memory_buffer_QueryInterface
,
253 memory_buffer_AddRef
,
254 memory_buffer_Release
,
256 memory_buffer_Unlock
,
257 memory_buffer_GetCurrentLength
,
258 memory_buffer_SetCurrentLength
,
259 memory_buffer_GetMaxLength
,
262 static HRESULT WINAPI
memory_1d_2d_buffer_QueryInterface(IMFMediaBuffer
*iface
, REFIID riid
, void **out
)
264 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
266 TRACE("%p, %s, %p.\n", iface
, debugstr_guid(riid
), out
);
268 if (IsEqualIID(riid
, &IID_IMFMediaBuffer
) ||
269 IsEqualIID(riid
, &IID_IUnknown
))
271 *out
= &buffer
->IMFMediaBuffer_iface
;
273 else if (IsEqualIID(riid
, &IID_IMF2DBuffer2
) ||
274 IsEqualIID(riid
, &IID_IMF2DBuffer
))
276 *out
= &buffer
->IMF2DBuffer2_iface
;
278 else if (IsEqualIID(riid
, &IID_IMFGetService
))
280 *out
= &buffer
->IMFGetService_iface
;
284 WARN("Unsupported interface %s.\n", debugstr_guid(riid
));
286 return E_NOINTERFACE
;
289 IUnknown_AddRef((IUnknown
*)*out
);
293 static HRESULT WINAPI
memory_1d_2d_buffer_Lock(IMFMediaBuffer
*iface
, BYTE
**data
, DWORD
*max_length
, DWORD
*current_length
)
295 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
298 TRACE("%p, %p, %p, %p.\n", iface
, data
, max_length
, current_length
);
303 /* Allocate linear buffer and return it as a copy of current content. Maximum and current length are
304 unrelated to 2D buffer maximum allocate length, or maintained current length. */
306 EnterCriticalSection(&buffer
->cs
);
308 if (!buffer
->_2d
.linear_buffer
&& buffer
->_2d
.locks
)
309 hr
= MF_E_INVALIDREQUEST
;
310 else if (!buffer
->_2d
.linear_buffer
)
312 if (!(buffer
->_2d
.linear_buffer
= malloc(buffer
->_2d
.plane_size
)))
316 copy_image(buffer
, buffer
->_2d
.linear_buffer
, buffer
->_2d
.width
, buffer
->data
, buffer
->_2d
.pitch
,
317 buffer
->_2d
.width
, buffer
->_2d
.height
);
323 *data
= buffer
->_2d
.linear_buffer
;
325 *max_length
= buffer
->_2d
.plane_size
;
327 *current_length
= buffer
->_2d
.plane_size
;
330 LeaveCriticalSection(&buffer
->cs
);
335 static HRESULT WINAPI
memory_1d_2d_buffer_Unlock(IMFMediaBuffer
*iface
)
337 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
339 TRACE("%p.\n", iface
);
341 EnterCriticalSection(&buffer
->cs
);
343 if (buffer
->_2d
.linear_buffer
&& !--buffer
->_2d
.locks
)
345 copy_image(buffer
, buffer
->data
, buffer
->_2d
.pitch
, buffer
->_2d
.linear_buffer
, buffer
->_2d
.width
,
346 buffer
->_2d
.width
, buffer
->_2d
.height
);
348 free(buffer
->_2d
.linear_buffer
);
349 buffer
->_2d
.linear_buffer
= NULL
;
352 LeaveCriticalSection(&buffer
->cs
);
357 static const IMFMediaBufferVtbl memory_1d_2d_buffer_vtbl
=
359 memory_1d_2d_buffer_QueryInterface
,
360 memory_buffer_AddRef
,
361 memory_buffer_Release
,
362 memory_1d_2d_buffer_Lock
,
363 memory_1d_2d_buffer_Unlock
,
364 memory_buffer_GetCurrentLength
,
365 memory_buffer_SetCurrentLength
,
366 memory_buffer_GetMaxLength
,
369 static HRESULT WINAPI
d3d9_surface_buffer_Lock(IMFMediaBuffer
*iface
, BYTE
**data
, DWORD
*max_length
, DWORD
*current_length
)
371 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
374 TRACE("%p, %p, %p, %p.\n", iface
, data
, max_length
, current_length
);
379 EnterCriticalSection(&buffer
->cs
);
381 if (!buffer
->_2d
.linear_buffer
&& buffer
->_2d
.locks
)
382 hr
= MF_E_INVALIDREQUEST
;
383 else if (!buffer
->_2d
.linear_buffer
)
387 if (!(buffer
->_2d
.linear_buffer
= malloc(buffer
->_2d
.plane_size
)))
392 hr
= IDirect3DSurface9_LockRect(buffer
->d3d9_surface
.surface
, &rect
, NULL
, 0);
395 copy_image(buffer
, buffer
->_2d
.linear_buffer
, buffer
->_2d
.width
, rect
.pBits
, rect
.Pitch
,
396 buffer
->_2d
.width
, buffer
->_2d
.height
);
397 IDirect3DSurface9_UnlockRect(buffer
->d3d9_surface
.surface
);
405 *data
= buffer
->_2d
.linear_buffer
;
407 *max_length
= buffer
->_2d
.plane_size
;
409 *current_length
= buffer
->_2d
.plane_size
;
412 LeaveCriticalSection(&buffer
->cs
);
417 static HRESULT WINAPI
d3d9_surface_buffer_Unlock(IMFMediaBuffer
*iface
)
419 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
422 TRACE("%p.\n", iface
);
424 EnterCriticalSection(&buffer
->cs
);
426 if (!buffer
->_2d
.linear_buffer
)
427 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
428 else if (!--buffer
->_2d
.locks
)
432 if (SUCCEEDED(hr
= IDirect3DSurface9_LockRect(buffer
->d3d9_surface
.surface
, &rect
, NULL
, 0)))
434 copy_image(buffer
, rect
.pBits
, rect
.Pitch
, buffer
->_2d
.linear_buffer
, buffer
->_2d
.width
,
435 buffer
->_2d
.width
, buffer
->_2d
.height
);
436 IDirect3DSurface9_UnlockRect(buffer
->d3d9_surface
.surface
);
439 free(buffer
->_2d
.linear_buffer
);
440 buffer
->_2d
.linear_buffer
= NULL
;
443 LeaveCriticalSection(&buffer
->cs
);
448 static HRESULT WINAPI
d3d9_surface_buffer_SetCurrentLength(IMFMediaBuffer
*iface
, DWORD current_length
)
450 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
452 TRACE("%p, %lu.\n", iface
, current_length
);
454 buffer
->current_length
= current_length
;
459 static const IMFMediaBufferVtbl d3d9_surface_1d_buffer_vtbl
=
461 memory_1d_2d_buffer_QueryInterface
,
462 memory_buffer_AddRef
,
463 memory_buffer_Release
,
464 d3d9_surface_buffer_Lock
,
465 d3d9_surface_buffer_Unlock
,
466 memory_buffer_GetCurrentLength
,
467 d3d9_surface_buffer_SetCurrentLength
,
468 memory_buffer_GetMaxLength
,
471 static HRESULT WINAPI
memory_2d_buffer_QueryInterface(IMF2DBuffer2
*iface
, REFIID riid
, void **obj
)
473 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
474 return IMFMediaBuffer_QueryInterface(&buffer
->IMFMediaBuffer_iface
, riid
, obj
);
477 static ULONG WINAPI
memory_2d_buffer_AddRef(IMF2DBuffer2
*iface
)
479 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
480 return IMFMediaBuffer_AddRef(&buffer
->IMFMediaBuffer_iface
);
483 static ULONG WINAPI
memory_2d_buffer_Release(IMF2DBuffer2
*iface
)
485 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
486 return IMFMediaBuffer_Release(&buffer
->IMFMediaBuffer_iface
);
489 static HRESULT
memory_2d_buffer_lock(struct buffer
*buffer
, BYTE
**scanline0
, LONG
*pitch
,
490 BYTE
**buffer_start
, DWORD
*buffer_length
)
494 if (buffer
->_2d
.linear_buffer
)
495 hr
= MF_E_UNEXPECTED
;
499 *scanline0
= buffer
->_2d
.scanline0
;
500 *pitch
= buffer
->_2d
.pitch
;
502 *buffer_start
= buffer
->data
;
504 *buffer_length
= buffer
->max_length
;
510 static HRESULT WINAPI
memory_2d_buffer_Lock2D(IMF2DBuffer2
*iface
, BYTE
**scanline0
, LONG
*pitch
)
512 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
515 TRACE("%p, %p, %p.\n", iface
, scanline0
, pitch
);
517 if (!scanline0
|| !pitch
)
520 EnterCriticalSection(&buffer
->cs
);
522 hr
= memory_2d_buffer_lock(buffer
, scanline0
, pitch
, NULL
, NULL
);
524 LeaveCriticalSection(&buffer
->cs
);
529 static HRESULT WINAPI
memory_2d_buffer_Unlock2D(IMF2DBuffer2
*iface
)
531 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
534 TRACE("%p.\n", iface
);
536 EnterCriticalSection(&buffer
->cs
);
538 if (!buffer
->_2d
.linear_buffer
)
540 if (buffer
->_2d
.locks
)
543 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
546 LeaveCriticalSection(&buffer
->cs
);
551 static HRESULT WINAPI
memory_2d_buffer_GetScanline0AndPitch(IMF2DBuffer2
*iface
, BYTE
**scanline0
, LONG
*pitch
)
553 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
556 TRACE("%p, %p, %p.\n", iface
, scanline0
, pitch
);
558 if (!scanline0
|| !pitch
)
561 EnterCriticalSection(&buffer
->cs
);
563 if (buffer
->_2d
.linear_buffer
|| !buffer
->_2d
.locks
)
564 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
567 *scanline0
= buffer
->_2d
.scanline0
;
568 *pitch
= buffer
->_2d
.pitch
;
571 LeaveCriticalSection(&buffer
->cs
);
576 static HRESULT WINAPI
memory_2d_buffer_IsContiguousFormat(IMF2DBuffer2
*iface
, BOOL
*is_contiguous
)
578 TRACE("%p, %p.\n", iface
, is_contiguous
);
583 *is_contiguous
= FALSE
;
588 static HRESULT WINAPI
memory_2d_buffer_GetContiguousLength(IMF2DBuffer2
*iface
, DWORD
*length
)
590 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
592 TRACE("%p, %p.\n", iface
, length
);
597 *length
= buffer
->_2d
.plane_size
;
602 static HRESULT WINAPI
memory_2d_buffer_ContiguousCopyTo(IMF2DBuffer2
*iface
, BYTE
*dest_buffer
, DWORD dest_length
)
604 FIXME("%p, %p, %lu.\n", iface
, dest_buffer
, dest_length
);
609 static HRESULT WINAPI
memory_2d_buffer_ContiguousCopyFrom(IMF2DBuffer2
*iface
, const BYTE
*src_buffer
, DWORD src_length
)
611 FIXME("%p, %p, %lu.\n", iface
, src_buffer
, src_length
);
616 static HRESULT WINAPI
memory_2d_buffer_Lock2DSize(IMF2DBuffer2
*iface
, MF2DBuffer_LockFlags flags
, BYTE
**scanline0
,
617 LONG
*pitch
, BYTE
**buffer_start
, DWORD
*buffer_length
)
619 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
622 TRACE("%p, %#x, %p, %p, %p, %p.\n", iface
, flags
, scanline0
, pitch
, buffer_start
, buffer_length
);
624 if (!scanline0
|| !pitch
|| !buffer_start
|| !buffer_length
)
627 EnterCriticalSection(&buffer
->cs
);
629 hr
= memory_2d_buffer_lock(buffer
, scanline0
, pitch
, buffer_start
, buffer_length
);
631 LeaveCriticalSection(&buffer
->cs
);
636 static HRESULT WINAPI
memory_2d_buffer_Copy2DTo(IMF2DBuffer2
*iface
, IMF2DBuffer2
*dest_buffer
)
638 FIXME("%p, %p.\n", iface
, dest_buffer
);
643 static const IMF2DBuffer2Vtbl memory_2d_buffer_vtbl
=
645 memory_2d_buffer_QueryInterface
,
646 memory_2d_buffer_AddRef
,
647 memory_2d_buffer_Release
,
648 memory_2d_buffer_Lock2D
,
649 memory_2d_buffer_Unlock2D
,
650 memory_2d_buffer_GetScanline0AndPitch
,
651 memory_2d_buffer_IsContiguousFormat
,
652 memory_2d_buffer_GetContiguousLength
,
653 memory_2d_buffer_ContiguousCopyTo
,
654 memory_2d_buffer_ContiguousCopyFrom
,
655 memory_2d_buffer_Lock2DSize
,
656 memory_2d_buffer_Copy2DTo
,
659 static HRESULT WINAPI
d3d9_surface_buffer_Lock2D(IMF2DBuffer2
*iface
, BYTE
**scanline0
, LONG
*pitch
)
661 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
664 TRACE("%p, %p, %p.\n", iface
, scanline0
, pitch
);
666 if (!scanline0
|| !pitch
)
669 EnterCriticalSection(&buffer
->cs
);
671 if (buffer
->_2d
.linear_buffer
)
672 hr
= MF_E_UNEXPECTED
;
673 else if (!buffer
->_2d
.locks
)
675 hr
= IDirect3DSurface9_LockRect(buffer
->d3d9_surface
.surface
, &buffer
->d3d9_surface
.rect
, NULL
, 0);
681 *scanline0
= buffer
->d3d9_surface
.rect
.pBits
;
682 *pitch
= buffer
->d3d9_surface
.rect
.Pitch
;
685 LeaveCriticalSection(&buffer
->cs
);
690 static HRESULT WINAPI
d3d9_surface_buffer_Unlock2D(IMF2DBuffer2
*iface
)
692 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
695 TRACE("%p.\n", iface
);
697 EnterCriticalSection(&buffer
->cs
);
699 if (buffer
->_2d
.locks
)
701 if (!--buffer
->_2d
.locks
)
703 IDirect3DSurface9_UnlockRect(buffer
->d3d9_surface
.surface
);
704 memset(&buffer
->d3d9_surface
.rect
, 0, sizeof(buffer
->d3d9_surface
.rect
));
708 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
710 LeaveCriticalSection(&buffer
->cs
);
715 static HRESULT WINAPI
d3d9_surface_buffer_GetScanline0AndPitch(IMF2DBuffer2
*iface
, BYTE
**scanline0
, LONG
*pitch
)
717 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
720 TRACE("%p, %p, %p.\n", iface
, scanline0
, pitch
);
722 if (!scanline0
|| !pitch
)
725 EnterCriticalSection(&buffer
->cs
);
727 if (!buffer
->_2d
.locks
)
731 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
735 *scanline0
= buffer
->d3d9_surface
.rect
.pBits
;
736 *pitch
= buffer
->d3d9_surface
.rect
.Pitch
;
739 LeaveCriticalSection(&buffer
->cs
);
744 static HRESULT WINAPI
d3d9_surface_buffer_Lock2DSize(IMF2DBuffer2
*iface
, MF2DBuffer_LockFlags flags
, BYTE
**scanline0
,
745 LONG
*pitch
, BYTE
**buffer_start
, DWORD
*buffer_length
)
747 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
750 TRACE("%p, %#x, %p, %p, %p, %p.\n", iface
, flags
, scanline0
, pitch
, buffer_start
, buffer_length
);
752 if (!scanline0
|| !pitch
|| !buffer_start
|| !buffer_length
)
755 EnterCriticalSection(&buffer
->cs
);
757 if (buffer
->_2d
.linear_buffer
)
758 hr
= MF_E_UNEXPECTED
;
759 else if (!buffer
->_2d
.locks
)
761 hr
= IDirect3DSurface9_LockRect(buffer
->d3d9_surface
.surface
, &buffer
->d3d9_surface
.rect
, NULL
, 0);
767 *scanline0
= buffer
->d3d9_surface
.rect
.pBits
;
768 *pitch
= buffer
->d3d9_surface
.rect
.Pitch
;
770 *buffer_start
= *scanline0
;
772 *buffer_length
= buffer
->d3d9_surface
.rect
.Pitch
* buffer
->_2d
.height
;
775 LeaveCriticalSection(&buffer
->cs
);
780 static const IMF2DBuffer2Vtbl d3d9_surface_buffer_vtbl
=
782 memory_2d_buffer_QueryInterface
,
783 memory_2d_buffer_AddRef
,
784 memory_2d_buffer_Release
,
785 d3d9_surface_buffer_Lock2D
,
786 d3d9_surface_buffer_Unlock2D
,
787 d3d9_surface_buffer_GetScanline0AndPitch
,
788 memory_2d_buffer_IsContiguousFormat
,
789 memory_2d_buffer_GetContiguousLength
,
790 memory_2d_buffer_ContiguousCopyTo
,
791 memory_2d_buffer_ContiguousCopyFrom
,
792 d3d9_surface_buffer_Lock2DSize
,
793 memory_2d_buffer_Copy2DTo
,
796 static HRESULT WINAPI
memory_2d_buffer_gs_QueryInterface(IMFGetService
*iface
, REFIID riid
, void **obj
)
798 struct buffer
*buffer
= impl_from_IMFGetService(iface
);
799 return IMFMediaBuffer_QueryInterface(&buffer
->IMFMediaBuffer_iface
, riid
, obj
);
802 static ULONG WINAPI
memory_2d_buffer_gs_AddRef(IMFGetService
*iface
)
804 struct buffer
*buffer
= impl_from_IMFGetService(iface
);
805 return IMFMediaBuffer_AddRef(&buffer
->IMFMediaBuffer_iface
);
808 static ULONG WINAPI
memory_2d_buffer_gs_Release(IMFGetService
*iface
)
810 struct buffer
*buffer
= impl_from_IMFGetService(iface
);
811 return IMFMediaBuffer_Release(&buffer
->IMFMediaBuffer_iface
);
814 static HRESULT WINAPI
memory_2d_buffer_gs_GetService(IMFGetService
*iface
, REFGUID service
, REFIID riid
, void **obj
)
816 TRACE("%p, %s, %s, %p.\n", iface
, debugstr_guid(service
), debugstr_guid(riid
), obj
);
821 static const IMFGetServiceVtbl memory_2d_buffer_gs_vtbl
=
823 memory_2d_buffer_gs_QueryInterface
,
824 memory_2d_buffer_gs_AddRef
,
825 memory_2d_buffer_gs_Release
,
826 memory_2d_buffer_gs_GetService
,
829 static HRESULT WINAPI
d3d9_surface_buffer_gs_GetService(IMFGetService
*iface
, REFGUID service
, REFIID riid
, void **obj
)
831 struct buffer
*buffer
= impl_from_IMFGetService(iface
);
833 TRACE("%p, %s, %s, %p.\n", iface
, debugstr_guid(service
), debugstr_guid(riid
), obj
);
835 if (IsEqualGUID(service
, &MR_BUFFER_SERVICE
))
837 return IDirect3DSurface9_QueryInterface(buffer
->d3d9_surface
.surface
, riid
, obj
);
843 static const IMFGetServiceVtbl d3d9_surface_buffer_gs_vtbl
=
845 memory_2d_buffer_gs_QueryInterface
,
846 memory_2d_buffer_gs_AddRef
,
847 memory_2d_buffer_gs_Release
,
848 d3d9_surface_buffer_gs_GetService
,
851 static HRESULT WINAPI
dxgi_1d_2d_buffer_QueryInterface(IMFMediaBuffer
*iface
, REFIID riid
, void **out
)
853 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
855 TRACE("%p, %s, %p.\n", iface
, debugstr_guid(riid
), out
);
857 if (IsEqualIID(riid
, &IID_IMFMediaBuffer
) ||
858 IsEqualIID(riid
, &IID_IUnknown
))
860 *out
= &buffer
->IMFMediaBuffer_iface
;
862 else if (IsEqualIID(riid
, &IID_IMF2DBuffer2
) ||
863 IsEqualIID(riid
, &IID_IMF2DBuffer
))
865 *out
= &buffer
->IMF2DBuffer2_iface
;
867 else if (IsEqualIID(riid
, &IID_IMFDXGIBuffer
))
869 *out
= &buffer
->IMFDXGIBuffer_iface
;
873 WARN("Unsupported interface %s.\n", debugstr_guid(riid
));
875 return E_NOINTERFACE
;
878 IUnknown_AddRef((IUnknown
*)*out
);
882 static HRESULT
dxgi_surface_buffer_create_readback_texture(struct buffer
*buffer
)
884 D3D11_TEXTURE2D_DESC texture_desc
;
885 ID3D11Device
*device
;
888 if (buffer
->dxgi_surface
.rb_texture
)
891 ID3D11Texture2D_GetDevice(buffer
->dxgi_surface
.texture
, &device
);
893 ID3D11Texture2D_GetDesc(buffer
->dxgi_surface
.texture
, &texture_desc
);
894 texture_desc
.Usage
= D3D11_USAGE_STAGING
;
895 texture_desc
.BindFlags
= 0;
896 texture_desc
.CPUAccessFlags
= D3D11_CPU_ACCESS_READ
| D3D11_CPU_ACCESS_WRITE
;
897 texture_desc
.MiscFlags
= 0;
898 texture_desc
.MipLevels
= 1;
899 if (FAILED(hr
= ID3D11Device_CreateTexture2D(device
, &texture_desc
, NULL
, &buffer
->dxgi_surface
.rb_texture
)))
900 WARN("Failed to create readback texture, hr %#lx.\n", hr
);
902 ID3D11Device_Release(device
);
907 static HRESULT
dxgi_surface_buffer_map(struct buffer
*buffer
)
909 ID3D11DeviceContext
*immediate_context
;
910 ID3D11Device
*device
;
913 if (FAILED(hr
= dxgi_surface_buffer_create_readback_texture(buffer
)))
916 ID3D11Texture2D_GetDevice(buffer
->dxgi_surface
.texture
, &device
);
917 ID3D11Device_GetImmediateContext(device
, &immediate_context
);
918 ID3D11DeviceContext_CopySubresourceRegion(immediate_context
, (ID3D11Resource
*)buffer
->dxgi_surface
.rb_texture
,
919 0, 0, 0, 0, (ID3D11Resource
*)buffer
->dxgi_surface
.texture
, buffer
->dxgi_surface
.sub_resource_idx
, NULL
);
921 memset(&buffer
->dxgi_surface
.map_desc
, 0, sizeof(buffer
->dxgi_surface
.map_desc
));
922 if (FAILED(hr
= ID3D11DeviceContext_Map(immediate_context
, (ID3D11Resource
*)buffer
->dxgi_surface
.rb_texture
,
923 0, D3D11_MAP_READ_WRITE
, 0, &buffer
->dxgi_surface
.map_desc
)))
925 WARN("Failed to map readback texture, hr %#lx.\n", hr
);
928 ID3D11DeviceContext_Release(immediate_context
);
929 ID3D11Device_Release(device
);
934 static void dxgi_surface_buffer_unmap(struct buffer
*buffer
)
936 ID3D11DeviceContext
*immediate_context
;
937 ID3D11Device
*device
;
939 ID3D11Texture2D_GetDevice(buffer
->dxgi_surface
.texture
, &device
);
940 ID3D11Device_GetImmediateContext(device
, &immediate_context
);
941 ID3D11DeviceContext_Unmap(immediate_context
, (ID3D11Resource
*)buffer
->dxgi_surface
.rb_texture
, 0);
942 memset(&buffer
->dxgi_surface
.map_desc
, 0, sizeof(buffer
->dxgi_surface
.map_desc
));
944 ID3D11DeviceContext_CopySubresourceRegion(immediate_context
, (ID3D11Resource
*)buffer
->dxgi_surface
.texture
,
945 buffer
->dxgi_surface
.sub_resource_idx
, 0, 0, 0, (ID3D11Resource
*)buffer
->dxgi_surface
.rb_texture
, 0, NULL
);
947 ID3D11DeviceContext_Release(immediate_context
);
948 ID3D11Device_Release(device
);
951 static HRESULT WINAPI
dxgi_surface_buffer_Lock(IMFMediaBuffer
*iface
, BYTE
**data
, DWORD
*max_length
,
952 DWORD
*current_length
)
954 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
957 TRACE("%p, %p, %p, %p.\n", iface
, data
, max_length
, current_length
);
962 EnterCriticalSection(&buffer
->cs
);
964 if (!buffer
->_2d
.linear_buffer
&& buffer
->_2d
.locks
)
965 hr
= MF_E_INVALIDREQUEST
;
966 else if (!buffer
->_2d
.linear_buffer
)
968 if (!(buffer
->_2d
.linear_buffer
= malloc(buffer
->_2d
.plane_size
)))
973 hr
= dxgi_surface_buffer_map(buffer
);
976 copy_image(buffer
, buffer
->_2d
.linear_buffer
, buffer
->_2d
.width
, buffer
->dxgi_surface
.map_desc
.pData
,
977 buffer
->dxgi_surface
.map_desc
.RowPitch
, buffer
->_2d
.width
, buffer
->_2d
.height
);
985 *data
= buffer
->_2d
.linear_buffer
;
987 *max_length
= buffer
->_2d
.plane_size
;
989 *current_length
= buffer
->_2d
.plane_size
;
992 LeaveCriticalSection(&buffer
->cs
);
997 static HRESULT WINAPI
dxgi_surface_buffer_Unlock(IMFMediaBuffer
*iface
)
999 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
1002 TRACE("%p.\n", iface
);
1004 EnterCriticalSection(&buffer
->cs
);
1006 if (!buffer
->_2d
.linear_buffer
)
1007 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
1008 else if (!--buffer
->_2d
.locks
)
1010 copy_image(buffer
, buffer
->dxgi_surface
.map_desc
.pData
, buffer
->dxgi_surface
.map_desc
.RowPitch
,
1011 buffer
->_2d
.linear_buffer
, buffer
->_2d
.width
, buffer
->_2d
.width
, buffer
->_2d
.height
);
1012 dxgi_surface_buffer_unmap(buffer
);
1014 free(buffer
->_2d
.linear_buffer
);
1015 buffer
->_2d
.linear_buffer
= NULL
;
1018 LeaveCriticalSection(&buffer
->cs
);
1023 static HRESULT WINAPI
dxgi_surface_buffer_SetCurrentLength(IMFMediaBuffer
*iface
, DWORD current_length
)
1025 struct buffer
*buffer
= impl_from_IMFMediaBuffer(iface
);
1027 TRACE("%p, %lu.\n", iface
, current_length
);
1029 buffer
->current_length
= current_length
;
1034 static HRESULT WINAPI
dxgi_surface_buffer_Lock2D(IMF2DBuffer2
*iface
, BYTE
**scanline0
, LONG
*pitch
)
1036 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
1039 TRACE("%p, %p, %p.\n", iface
, scanline0
, pitch
);
1041 if (!scanline0
|| !pitch
)
1044 EnterCriticalSection(&buffer
->cs
);
1046 if (buffer
->_2d
.linear_buffer
)
1047 hr
= MF_E_UNEXPECTED
;
1048 else if (!buffer
->_2d
.locks
++)
1049 hr
= dxgi_surface_buffer_map(buffer
);
1053 *scanline0
= buffer
->dxgi_surface
.map_desc
.pData
;
1054 *pitch
= buffer
->dxgi_surface
.map_desc
.RowPitch
;
1057 LeaveCriticalSection(&buffer
->cs
);
1062 static HRESULT WINAPI
dxgi_surface_buffer_Unlock2D(IMF2DBuffer2
*iface
)
1064 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
1067 TRACE("%p.\n", iface
);
1069 EnterCriticalSection(&buffer
->cs
);
1071 if (buffer
->_2d
.locks
)
1073 if (!--buffer
->_2d
.locks
)
1074 dxgi_surface_buffer_unmap(buffer
);
1077 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
1079 LeaveCriticalSection(&buffer
->cs
);
1084 static HRESULT WINAPI
dxgi_surface_buffer_GetScanline0AndPitch(IMF2DBuffer2
*iface
, BYTE
**scanline0
, LONG
*pitch
)
1086 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
1089 TRACE("%p, %p, %p.\n", iface
, scanline0
, pitch
);
1091 if (!scanline0
|| !pitch
)
1094 EnterCriticalSection(&buffer
->cs
);
1096 if (!buffer
->_2d
.locks
)
1100 hr
= HRESULT_FROM_WIN32(ERROR_WAS_UNLOCKED
);
1104 *scanline0
= buffer
->dxgi_surface
.map_desc
.pData
;
1105 *pitch
= buffer
->dxgi_surface
.map_desc
.RowPitch
;
1108 LeaveCriticalSection(&buffer
->cs
);
1113 static HRESULT WINAPI
dxgi_surface_buffer_Lock2DSize(IMF2DBuffer2
*iface
, MF2DBuffer_LockFlags flags
,
1114 BYTE
**scanline0
, LONG
*pitch
, BYTE
**buffer_start
, DWORD
*buffer_length
)
1116 struct buffer
*buffer
= impl_from_IMF2DBuffer2(iface
);
1119 TRACE("%p, %#x, %p, %p, %p, %p.\n", iface
, flags
, scanline0
, pitch
, buffer_start
, buffer_length
);
1121 if (!scanline0
|| !pitch
|| !buffer_start
|| !buffer_length
)
1124 EnterCriticalSection(&buffer
->cs
);
1126 if (buffer
->_2d
.linear_buffer
)
1127 hr
= MF_E_UNEXPECTED
;
1128 else if (!buffer
->_2d
.locks
++)
1129 hr
= dxgi_surface_buffer_map(buffer
);
1133 *scanline0
= buffer
->dxgi_surface
.map_desc
.pData
;
1134 *pitch
= buffer
->dxgi_surface
.map_desc
.RowPitch
;
1136 *buffer_start
= *scanline0
;
1138 *buffer_length
= buffer
->dxgi_surface
.map_desc
.RowPitch
* buffer
->_2d
.height
;
1141 LeaveCriticalSection(&buffer
->cs
);
1146 static HRESULT WINAPI
dxgi_buffer_QueryInterface(IMFDXGIBuffer
*iface
, REFIID riid
, void **obj
)
1148 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1149 return IMFMediaBuffer_QueryInterface(&buffer
->IMFMediaBuffer_iface
, riid
, obj
);
1152 static ULONG WINAPI
dxgi_buffer_AddRef(IMFDXGIBuffer
*iface
)
1154 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1155 return IMFMediaBuffer_AddRef(&buffer
->IMFMediaBuffer_iface
);
1158 static ULONG WINAPI
dxgi_buffer_Release(IMFDXGIBuffer
*iface
)
1160 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1161 return IMFMediaBuffer_Release(&buffer
->IMFMediaBuffer_iface
);
1164 static HRESULT WINAPI
dxgi_buffer_GetResource(IMFDXGIBuffer
*iface
, REFIID riid
, void **obj
)
1166 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1168 TRACE("%p, %s, %p.\n", iface
, debugstr_guid(riid
), obj
);
1170 return ID3D11Texture2D_QueryInterface(buffer
->dxgi_surface
.texture
, riid
, obj
);
1173 static HRESULT WINAPI
dxgi_buffer_GetSubresourceIndex(IMFDXGIBuffer
*iface
, UINT
*index
)
1175 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1177 TRACE("%p, %p.\n", iface
, index
);
1182 *index
= buffer
->dxgi_surface
.sub_resource_idx
;
1187 static HRESULT WINAPI
dxgi_buffer_GetUnknown(IMFDXGIBuffer
*iface
, REFIID guid
, REFIID riid
, void **object
)
1189 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1191 TRACE("%p, %s, %s, %p.\n", iface
, debugstr_guid(guid
), debugstr_guid(riid
), object
);
1193 if (attributes_GetUnknown(&buffer
->dxgi_surface
.attributes
, guid
, riid
, object
) == MF_E_ATTRIBUTENOTFOUND
)
1194 return MF_E_NOT_FOUND
;
1199 static HRESULT WINAPI
dxgi_buffer_SetUnknown(IMFDXGIBuffer
*iface
, REFIID guid
, IUnknown
*data
)
1201 struct buffer
*buffer
= impl_from_IMFDXGIBuffer(iface
);
1204 TRACE("%p, %s, %p.\n", iface
, debugstr_guid(guid
), data
);
1206 EnterCriticalSection(&buffer
->dxgi_surface
.attributes
.cs
);
1209 if (SUCCEEDED(attributes_GetItem(&buffer
->dxgi_surface
.attributes
, guid
, NULL
)))
1210 hr
= HRESULT_FROM_WIN32(ERROR_OBJECT_ALREADY_EXISTS
);
1212 hr
= attributes_SetUnknown(&buffer
->dxgi_surface
.attributes
, guid
, data
);
1216 attributes_DeleteItem(&buffer
->dxgi_surface
.attributes
, guid
);
1218 LeaveCriticalSection(&buffer
->dxgi_surface
.attributes
.cs
);
1223 static const IMFMediaBufferVtbl dxgi_surface_1d_buffer_vtbl
=
1225 dxgi_1d_2d_buffer_QueryInterface
,
1226 memory_buffer_AddRef
,
1227 memory_buffer_Release
,
1228 dxgi_surface_buffer_Lock
,
1229 dxgi_surface_buffer_Unlock
,
1230 memory_buffer_GetCurrentLength
,
1231 dxgi_surface_buffer_SetCurrentLength
,
1232 memory_buffer_GetMaxLength
,
1235 static const IMF2DBuffer2Vtbl dxgi_surface_buffer_vtbl
=
1237 memory_2d_buffer_QueryInterface
,
1238 memory_2d_buffer_AddRef
,
1239 memory_2d_buffer_Release
,
1240 dxgi_surface_buffer_Lock2D
,
1241 dxgi_surface_buffer_Unlock2D
,
1242 dxgi_surface_buffer_GetScanline0AndPitch
,
1243 memory_2d_buffer_IsContiguousFormat
,
1244 memory_2d_buffer_GetContiguousLength
,
1245 memory_2d_buffer_ContiguousCopyTo
,
1246 memory_2d_buffer_ContiguousCopyFrom
,
1247 dxgi_surface_buffer_Lock2DSize
,
1248 memory_2d_buffer_Copy2DTo
,
1251 static const IMFDXGIBufferVtbl dxgi_buffer_vtbl
=
1253 dxgi_buffer_QueryInterface
,
1255 dxgi_buffer_Release
,
1256 dxgi_buffer_GetResource
,
1257 dxgi_buffer_GetSubresourceIndex
,
1258 dxgi_buffer_GetUnknown
,
1259 dxgi_buffer_SetUnknown
,
1262 static HRESULT
memory_buffer_init(struct buffer
*buffer
, DWORD max_length
, DWORD alignment
,
1263 const IMFMediaBufferVtbl
*vtbl
)
1265 if (alignment
< MF_16_BYTE_ALIGNMENT
)
1266 alignment
= MF_16_BYTE_ALIGNMENT
;
1269 if (alignment
& (alignment
- 1))
1272 alignment
|= alignment
>> 1;
1273 alignment
|= alignment
>> 2;
1274 alignment
|= alignment
>> 4;
1275 alignment
|= alignment
>> 8;
1276 alignment
|= alignment
>> 16;
1280 if (!(buffer
->data
= _aligned_malloc(max_length
, alignment
)))
1281 return E_OUTOFMEMORY
;
1282 memset(buffer
->data
, 0, max_length
);
1284 buffer
->IMFMediaBuffer_iface
.lpVtbl
= vtbl
;
1285 buffer
->refcount
= 1;
1286 buffer
->max_length
= max_length
;
1287 buffer
->current_length
= 0;
1288 InitializeCriticalSection(&buffer
->cs
);
1293 static HRESULT
create_1d_buffer(DWORD max_length
, DWORD alignment
, IMFMediaBuffer
**buffer
)
1295 struct buffer
*object
;
1303 if (!(object
= calloc(1, sizeof(*object
))))
1304 return E_OUTOFMEMORY
;
1306 hr
= memory_buffer_init(object
, max_length
, alignment
, &memory_1d_buffer_vtbl
);
1313 *buffer
= &object
->IMFMediaBuffer_iface
;
1318 static p_copy_image_func
get_2d_buffer_copy_func(DWORD fourcc
)
1320 if (fourcc
== MAKEFOURCC('N','V','1','2'))
1321 return copy_image_nv12
;
1322 if (fourcc
== MAKEFOURCC('I','M','C','1') || fourcc
== MAKEFOURCC('I','M','C','3'))
1323 return copy_image_imc1
;
1324 if (fourcc
== MAKEFOURCC('I','M','C','2') || fourcc
== MAKEFOURCC('I','M','C','4'))
1325 return copy_image_imc2
;
1329 static HRESULT
create_2d_buffer(DWORD width
, DWORD height
, DWORD fourcc
, BOOL bottom_up
, IMFMediaBuffer
**buffer
)
1331 unsigned int stride
, max_length
;
1332 unsigned int row_alignment
;
1333 struct buffer
*object
;
1345 memcpy(&subtype
, &MFVideoFormat_Base
, sizeof(subtype
));
1346 subtype
.Data1
= fourcc
;
1348 if (!(stride
= mf_format_get_stride(&subtype
, width
, &is_yuv
)))
1349 return MF_E_INVALIDMEDIATYPE
;
1351 if (is_yuv
&& bottom_up
)
1352 return MF_E_INVALIDMEDIATYPE
;
1354 if (FAILED(hr
= MFGetPlaneSize(fourcc
, width
, height
, &plane_size
)))
1357 if (!(object
= calloc(1, sizeof(*object
))))
1358 return E_OUTOFMEMORY
;
1362 case MAKEFOURCC('I','M','C','1'):
1363 case MAKEFOURCC('I','M','C','2'):
1364 case MAKEFOURCC('I','M','C','3'):
1365 case MAKEFOURCC('I','M','C','4'):
1366 case MAKEFOURCC('Y','V','1','2'):
1367 row_alignment
= MF_128_BYTE_ALIGNMENT
;
1370 row_alignment
= MF_64_BYTE_ALIGNMENT
;
1373 pitch
= ALIGN_SIZE(stride
, row_alignment
);
1377 case MAKEFOURCC('I','M','C','1'):
1378 case MAKEFOURCC('I','M','C','3'):
1379 max_length
= pitch
* height
* 2;
1382 case MAKEFOURCC('N','V','1','2'):
1383 case MAKEFOURCC('Y','V','1','2'):
1384 case MAKEFOURCC('I','M','C','2'):
1385 case MAKEFOURCC('I','M','C','4'):
1386 max_length
= pitch
* height
* 3 / 2;
1389 max_length
= pitch
* height
;
1392 if (FAILED(hr
= memory_buffer_init(object
, max_length
, row_alignment
, &memory_1d_2d_buffer_vtbl
)))
1398 object
->IMF2DBuffer2_iface
.lpVtbl
= &memory_2d_buffer_vtbl
;
1399 object
->IMFGetService_iface
.lpVtbl
= &memory_2d_buffer_gs_vtbl
;
1400 object
->_2d
.plane_size
= plane_size
;
1401 object
->_2d
.width
= stride
;
1402 object
->_2d
.height
= height
;
1403 object
->_2d
.pitch
= bottom_up
? -pitch
: pitch
;
1404 object
->_2d
.scanline0
= bottom_up
? object
->data
+ pitch
* (object
->_2d
.height
- 1) : object
->data
;
1405 object
->_2d
.copy_image
= get_2d_buffer_copy_func(fourcc
);
1407 *buffer
= &object
->IMFMediaBuffer_iface
;
1412 static HRESULT
create_d3d9_surface_buffer(IUnknown
*surface
, BOOL bottom_up
, IMFMediaBuffer
**buffer
)
1414 struct buffer
*object
;
1415 D3DSURFACE_DESC desc
;
1416 unsigned int stride
;
1420 IDirect3DSurface9_GetDesc((IDirect3DSurface9
*)surface
, &desc
);
1421 TRACE("format %#x, %u x %u.\n", desc
.Format
, desc
.Width
, desc
.Height
);
1423 memcpy(&subtype
, &MFVideoFormat_Base
, sizeof(subtype
));
1424 subtype
.Data1
= desc
.Format
;
1426 if (!(stride
= mf_format_get_stride(&subtype
, desc
.Width
, &is_yuv
)))
1427 return MF_E_INVALIDMEDIATYPE
;
1429 if (!(object
= calloc(1, sizeof(*object
))))
1430 return E_OUTOFMEMORY
;
1432 object
->IMFMediaBuffer_iface
.lpVtbl
= &d3d9_surface_1d_buffer_vtbl
;
1433 object
->IMF2DBuffer2_iface
.lpVtbl
= &d3d9_surface_buffer_vtbl
;
1434 object
->IMFGetService_iface
.lpVtbl
= &d3d9_surface_buffer_gs_vtbl
;
1435 object
->refcount
= 1;
1436 InitializeCriticalSection(&object
->cs
);
1437 object
->d3d9_surface
.surface
= (IDirect3DSurface9
*)surface
;
1438 IUnknown_AddRef(surface
);
1440 MFGetPlaneSize(desc
.Format
, desc
.Width
, desc
.Height
, &object
->_2d
.plane_size
);
1441 object
->_2d
.width
= stride
;
1442 object
->_2d
.height
= desc
.Height
;
1443 object
->max_length
= object
->_2d
.plane_size
;
1444 object
->_2d
.copy_image
= get_2d_buffer_copy_func(desc
.Format
);
1446 *buffer
= &object
->IMFMediaBuffer_iface
;
1451 static HRESULT
create_dxgi_surface_buffer(IUnknown
*surface
, unsigned int sub_resource_idx
,
1452 BOOL bottom_up
, IMFMediaBuffer
**buffer
)
1454 struct buffer
*object
;
1455 D3D11_TEXTURE2D_DESC desc
;
1456 ID3D11Texture2D
*texture
;
1457 unsigned int stride
;
1463 if (FAILED(hr
= IUnknown_QueryInterface(surface
, &IID_ID3D11Texture2D
, (void **)&texture
)))
1465 WARN("Failed to get texture interface, hr %#lx.\n", hr
);
1469 ID3D11Texture2D_GetDesc(texture
, &desc
);
1470 TRACE("format %#x, %u x %u.\n", desc
.Format
, desc
.Width
, desc
.Height
);
1472 memcpy(&subtype
, &MFVideoFormat_Base
, sizeof(subtype
));
1473 subtype
.Data1
= format
= MFMapDXGIFormatToDX9Format(desc
.Format
);
1475 if (!(stride
= mf_format_get_stride(&subtype
, desc
.Width
, &is_yuv
)))
1477 ID3D11Texture2D_Release(texture
);
1478 return MF_E_INVALIDMEDIATYPE
;
1481 if (!(object
= calloc(1, sizeof(*object
))))
1483 ID3D11Texture2D_Release(texture
);
1484 return E_OUTOFMEMORY
;
1487 object
->IMFMediaBuffer_iface
.lpVtbl
= &dxgi_surface_1d_buffer_vtbl
;
1488 object
->IMF2DBuffer2_iface
.lpVtbl
= &dxgi_surface_buffer_vtbl
;
1489 object
->IMFDXGIBuffer_iface
.lpVtbl
= &dxgi_buffer_vtbl
;
1490 object
->refcount
= 1;
1491 InitializeCriticalSection(&object
->cs
);
1492 object
->dxgi_surface
.texture
= texture
;
1493 object
->dxgi_surface
.sub_resource_idx
= sub_resource_idx
;
1495 MFGetPlaneSize(format
, desc
.Width
, desc
.Height
, &object
->_2d
.plane_size
);
1496 object
->_2d
.width
= stride
;
1497 object
->_2d
.height
= desc
.Height
;
1498 object
->max_length
= object
->_2d
.plane_size
;
1499 object
->_2d
.copy_image
= get_2d_buffer_copy_func(format
);
1501 if (FAILED(hr
= init_attributes_object(&object
->dxgi_surface
.attributes
, 0)))
1503 IMFMediaBuffer_Release(&object
->IMFMediaBuffer_iface
);
1507 *buffer
= &object
->IMFMediaBuffer_iface
;
1512 /***********************************************************************
1513 * MFCreateMemoryBuffer (mfplat.@)
1515 HRESULT WINAPI
MFCreateMemoryBuffer(DWORD max_length
, IMFMediaBuffer
**buffer
)
1517 TRACE("%lu, %p.\n", max_length
, buffer
);
1519 return create_1d_buffer(max_length
, MF_1_BYTE_ALIGNMENT
, buffer
);
1522 /***********************************************************************
1523 * MFCreateAlignedMemoryBuffer (mfplat.@)
1525 HRESULT WINAPI
MFCreateAlignedMemoryBuffer(DWORD max_length
, DWORD alignment
, IMFMediaBuffer
**buffer
)
1527 TRACE("%lu, %lu, %p.\n", max_length
, alignment
, buffer
);
1529 return create_1d_buffer(max_length
, alignment
, buffer
);
1532 /***********************************************************************
1533 * MFCreate2DMediaBuffer (mfplat.@)
1535 HRESULT WINAPI
MFCreate2DMediaBuffer(DWORD width
, DWORD height
, DWORD fourcc
, BOOL bottom_up
, IMFMediaBuffer
**buffer
)
1537 TRACE("%lu, %lu, %s, %d, %p.\n", width
, height
, debugstr_fourcc(fourcc
), bottom_up
, buffer
);
1539 return create_2d_buffer(width
, height
, fourcc
, bottom_up
, buffer
);
1542 /***********************************************************************
1543 * MFCreateDXSurfaceBuffer (mfplat.@)
1545 HRESULT WINAPI
MFCreateDXSurfaceBuffer(REFIID riid
, IUnknown
*surface
, BOOL bottom_up
, IMFMediaBuffer
**buffer
)
1547 TRACE("%s, %p, %d, %p.\n", debugstr_guid(riid
), surface
, bottom_up
, buffer
);
1549 if (!IsEqualIID(riid
, &IID_IDirect3DSurface9
))
1550 return E_INVALIDARG
;
1552 return create_d3d9_surface_buffer(surface
, bottom_up
, buffer
);
1555 /***********************************************************************
1556 * MFCreateDXGISurfaceBuffer (mfplat.@)
1558 HRESULT WINAPI
MFCreateDXGISurfaceBuffer(REFIID riid
, IUnknown
*surface
, UINT subresource
, BOOL bottom_up
,
1559 IMFMediaBuffer
**buffer
)
1561 TRACE("%s, %p, %u, %d, %p.\n", debugstr_guid(riid
), surface
, subresource
, bottom_up
, buffer
);
1563 if (!IsEqualIID(riid
, &IID_ID3D11Texture2D
))
1564 return E_INVALIDARG
;
1566 return create_dxgi_surface_buffer(surface
, subresource
, bottom_up
, buffer
);
1569 static unsigned int buffer_get_aligned_length(unsigned int length
, unsigned int alignment
)
1571 length
= (length
+ alignment
) / alignment
;
1572 length
*= alignment
;
1577 HRESULT WINAPI
MFCreateMediaBufferFromMediaType(IMFMediaType
*media_type
, LONGLONG duration
, DWORD min_length
,
1578 DWORD alignment
, IMFMediaBuffer
**buffer
)
1580 UINT32 length
= 0, block_alignment
;
1581 LONGLONG avg_length
;
1585 TRACE("%p, %s, %lu, %lu, %p.\n", media_type
, debugstr_time(duration
), min_length
, alignment
, buffer
);
1588 return E_INVALIDARG
;
1590 if (FAILED(hr
= IMFMediaType_GetMajorType(media_type
, &major
)))
1593 if (IsEqualGUID(&major
, &MFMediaType_Audio
))
1595 block_alignment
= 0;
1596 if (FAILED(IMFMediaType_GetUINT32(media_type
, &MF_MT_AUDIO_BLOCK_ALIGNMENT
, &block_alignment
)))
1597 WARN("Block alignment was not specified.\n");
1599 if (block_alignment
)
1606 if (SUCCEEDED(IMFMediaType_GetUINT32(media_type
, &MF_MT_AUDIO_AVG_BYTES_PER_SECOND
, &length
)))
1609 avg_length
= length
* duration
/ (10 * 1000 * 1000);
1613 alignment
= max(16, alignment
);
1615 length
= buffer_get_aligned_length(avg_length
+ 1, alignment
);
1616 length
= buffer_get_aligned_length(length
, block_alignment
);
1621 length
= max(length
, min_length
);
1623 return create_1d_buffer(length
, MF_1_BYTE_ALIGNMENT
, buffer
);
1626 FIXME("Major type %s is not supported.\n", debugstr_guid(&major
));