/*
 * Copyright (c) 2020-2021 https://gitee.com/fsfzp888
 * All rights reserved
 */

#ifndef SAMPLE_GRABBER_H
#define SAMPLE_GRABBER_H

#include <dshow.h>
#include <rpc.h>
#include <rpcndr.h>

struct GUIDHolder
{
    static constexpr IID IID_ISampleGrabberCB          = {0x0579154A, 0x2B53, 0x4994, {0xB0, 0xD0, 0xE7, 0x73, 0x14, 0x8E, 0xFF, 0x85}};
    static constexpr IID IID_ISampleGrabber            = {0x6B652FFF, 0x11FE, 0x4fce, {0x92, 0xAD, 0x02, 0x66, 0xB5, 0xD7, 0xC7, 0x8F}};
    static constexpr CLSID CLSID_SampleGrabber         = {0xC1F400A0, 0x3F08, 0x11d3, {0x9F, 0x0B, 0x00, 0x60, 0x08, 0x03, 0x9E, 0x37}};
    static constexpr CLSID CLSID_NullRenderer          = {0xC1F400A4, 0x3F08, 0x11d3, {0x9F, 0x0B, 0x00, 0x60, 0x08, 0x03, 0x9E, 0x37}};
    static constexpr CLSID CLSID_VideoEffects1Category = {0xcc7bfb42, 0xf175, 0x11d1, {0xa3, 0x92, 0x0, 0xe0, 0x29, 0x1f, 0x39, 0x59}};
    static constexpr CLSID CLSID_VideoEffects2Category = {0xcc7bfb43, 0xf175, 0x11d1, {0xa3, 0x92, 0x0, 0xe0, 0x29, 0x1f, 0x39, 0x59}};
    static constexpr CLSID CLSID_AudioEffects1Category = {0xcc7bfb44, 0xf175, 0x11d1, {0xa3, 0x92, 0x0, 0xe0, 0x29, 0x1f, 0x39, 0x59}};
    static constexpr CLSID CLSID_AudioEffects2Category = {0xcc7bfb45, 0xf175, 0x11d1, {0xa3, 0x92, 0x0, 0xe0, 0x29, 0x1f, 0x39, 0x59}};
};

MIDL_INTERFACE("0579154A-2B53-4994-B0D0-E773148EFF85")
ISampleGrabberCB : public IUnknown
{
  public:
    virtual HRESULT STDMETHODCALLTYPE SampleCB(double SampleTime, IMediaSample *pSample) = 0;

    virtual HRESULT STDMETHODCALLTYPE BufferCB(double SampleTime, BYTE *pBuffer, long BufferLen) = 0;
};

MIDL_INTERFACE("6B652FFF-11FE-4fce-92AD-0266B5D7C78F")
ISampleGrabber : public IUnknown
{
  public:
    virtual HRESULT STDMETHODCALLTYPE SetOneShot(BOOL OneShot) = 0;

    virtual HRESULT STDMETHODCALLTYPE SetMediaType(const AM_MEDIA_TYPE *pType) = 0;

    virtual HRESULT STDMETHODCALLTYPE GetConnectedMediaType(AM_MEDIA_TYPE * pType) = 0;

    virtual HRESULT STDMETHODCALLTYPE SetBufferSamples(BOOL BufferThem) = 0;

    virtual HRESULT STDMETHODCALLTYPE GetCurrentBuffer(
        /* [out][in] */ long *pBufferSize,
        /* [out] */ long *pBuffer) = 0;

    virtual HRESULT STDMETHODCALLTYPE GetCurrentSample(
        /* [retval][out] */ IMediaSample * *ppSample) = 0;

    virtual HRESULT STDMETHODCALLTYPE SetCallback(ISampleGrabberCB * pCallback, long WhichMethodToCallback) = 0;
};

#endif