/* -LICENSE-START-
** Copyright (c) 2023 Blackmagic Design
**
** Permission is hereby granted, free of charge, to any person or organization
** obtaining a copy of the software and accompanying documentation (the
** "Software") to use, reproduce, display, distribute, sub-license, execute,
** and transmit the Software, and to prepare derivative works of the Software,
** and to permit third-parties to whom the Software is furnished to do so, in
** accordance with:
**
** (1) if the Software is obtained from Blackmagic Design, the End User License
** Agreement for the Software Development Kit (“EULA”) available at
** https://www.blackmagicdesign.com/EULA/DeckLinkSDK; or
**
** (2) if the Software is obtained from any third party, such licensing terms
** as notified by that third party,
**
** and all subject to the following:
**
** (3) the copyright notices in the Software and this entire statement,
** including the above license grant, this restriction and the following
** disclaimer, must be included in all copies of the Software, in whole or in
** part, and all derivative works of the Software, unless such copies or
** derivative works are solely in the form of machine-executable object code
** generated by a source language processor.
**
** (4) THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
** OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
** FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
** SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
** FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
** ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
** DEALINGS IN THE SOFTWARE.
**
** A copy of the Software is available free of charge at
** https://www.blackmagicdesign.com/desktopvideo_sdk under the EULA.
**
** -LICENSE-END-
*/

#pragma once

#include <atomic>
#include <mutex>

#include "atlbase.h"
#include "mfidl.h"
#include "DeckLinkAPI_h.h"

static const GUID IID_DeckLinkMediaBuffer { 0x3ec04fc8, 0x30ca, 0x4aca, { 0x96, 0x7e, 0x50, 0x2, 0x60, 0x72, 0xf5, 0x54 } };

class DeckLinkMediaBuffer : public IDeckLinkVideoBuffer
{
public:
	explicit DeckLinkMediaBuffer(unsigned int bufferSize);
	virtual ~DeckLinkMediaBuffer() = default;

	// IUnknown interface
	virtual HRESULT			__stdcall QueryInterface(REFIID riid, void** ppvObject) override;
	virtual ULONG			__stdcall AddRef(void) override;
	virtual ULONG			__stdcall Release(void) override;

	// IDeckLinkVideoBuffer interface
	virtual HRESULT			__stdcall GetBytes(void** buffer) override;
	virtual HRESULT			__stdcall StartAccess(BMDBufferAccessFlags flags) override;
	virtual HRESULT			__stdcall EndAccess(BMDBufferAccessFlags flags) override;

	// Other public methods
	CComPtr<IMFMediaBuffer>	getBuffer(void);

private:
	std::atomic<ULONG>			m_refCount;
	std::mutex					m_mutex;
	CComPtr<IMFMediaBuffer>		m_mediaBuffer;
	BYTE*						m_accessedBuffer;
	UINT						m_accessedCount;
};
