CODEWITHSUNDEEP
c++windowswin-universal-appvideo-processingms-media-foundation
Sergey
Sergey

Reputation: 1590

How to duplicate IMFSample object

I need to duplicate IMFSample object so that I will have separate independent objects. Microsoft API does not provide easy way to duplicate IMFSample video frame.

Upvotes: 2

Views: 799

Answers (1)

Sergey
Sergey

Reputation: 1590

Here is a sample code:

#define MmfThrowIfError(hrVal) \
{ \
    HRESULT hrMmfTraceInternal = (hrVal); \
    if (FAILED(hrMmfTraceInternal)) \
    { \
        /* LOG_F(LS_ERROR)<<"MMF hr: 0x"<<std::hex<<hrMmfTraceInternal; */ \
        throw ref new ::Platform::Exception(hrMmfTraceInternal); \
    } \
}

MFComPtr<IMFMediaBuffer> DuplicateBuffer(const MFComPtr<IMFMediaBuffer>& srcBuf)
{
    byte* srcByteBuffer = nullptr;
    DWORD srcBuffMaxLen = 0;
    DWORD srcBuffCurrLen = 0;
    MmfThrowIfError(srcBuf->Lock(&srcByteBuffer, &srcBuffMaxLen, &srcBuffCurrLen));

    MFComPtr<IMFMediaBuffer> destBuf = nullptr;
    MmfThrowIfError(MFCreateMemoryBuffer(srcBuffCurrLen, &destBuf));

    byte* destByteBuffer = nullptr;
    MmfThrowIfError(destBuf->Lock(&destByteBuffer, nullptr, nullptr));
    memcpy(destByteBuffer, srcByteBuffer, srcBuffCurrLen);
    MmfThrowIfError(destBuf->Unlock());
    MmfThrowIfError(srcBuf->Unlock());

    MmfThrowIfError(destBuf->SetCurrentLength(srcBuffCurrLen));
    return destBuf;
}

SampleComPtr DuplicateSample(const SampleComPtr& sample)
{
    if (!sample)
        return nullptr;

    DWORD sampleFlags = 0;
    LONGLONG llVideoTimeStamp = 0;
    LONGLONG llSampleDuration = 0;
    MmfThrowIfError(sample->GetSampleFlags(&sampleFlags));
    MmfThrowIfError(sample->GetSampleTime(&llVideoTimeStamp));
    MmfThrowIfError(sample->GetSampleDuration(&llSampleDuration));

    SampleComPtr outSample;
    MFCreateSample(&outSample);
    MmfThrowIfError(outSample->SetSampleFlags(sampleFlags));
    MmfThrowIfError(outSample->SetSampleTime(llVideoTimeStamp));
    MmfThrowIfError(outSample->SetSampleDuration(llSampleDuration));

    DWORD bufferCount = 0;
    MmfThrowIfError(sample->GetBufferCount(&bufferCount));

    for (DWORD index = 0; index < bufferCount; ++index)
    {
        MFComPtr<IMFMediaBuffer> srcBuf = nullptr;
        MmfThrowIfError(sample->GetBufferByIndex(index, &srcBuf));

        MFComPtr<IMFMediaBuffer> reConstructedBuffer = DuplicateBuffer(srcBuf);
        srcBuf = nullptr;
        MmfThrowIfError(outSample->AddBuffer(reConstructedBuffer.Get()));
    }

    return outSample;
}

Upvotes: 2

Related Questions