从函数 return CComPtr 可以吗？

Question

我正在编写一个创建 IMFSample 并向其添加缓冲区的函数。我只想 return 新的 CComPtr 到 IMFSample 而不是将指针传递给参数中的指针。可以这样做吗？当函数退出时，对 IMFSample 的引用计数是否正确？我已经研究过，但未能找到有关此用法的任何示例或讨论。这是该函数的一个简化示例：

CComPtr<IMFSample> getSample() {
    HRESULT hr = S_OK;
    CComPtr<IMFSample> pSample = NULL;

    hr = MFCreateSample(&pSample);
    if (hr != S_OK) {
        // pSample should be released at exception.
        throw "MFCreateSample failed";
    }

    // ...Create buffer, Add buffer, etc...

    return pSample;
}

我在这里添加了一个更长的版本来帮助展示它如何通过使用异常来简化我的错误处理。我的程序的其余部分是使用异常构建的，所以我试图隔离 "COM" 部分，使它们尽可能地与其余部分更好地配合。

void decodeLoop() {
    CComPtr<IMFSample> pSample;

    while (noError) {
        try {
            pSample = getNALSample(decoderPkt, MAX_SIZE);
            processSample(pSample);
            pSample = NULL;
            ...
        }
        catch (MyException& e) {
            ...
        }
    }
}

CComPtr<IMFSample> getNALSample(DecoderPkt* decoderPkt, DWORD maxBuffSize) {
    HRESULT hr = S_OK;
    CComPtr<IMFSample> pSample = nullptr;
    CComPtr<IMFMediaBuffer> pMediaBuffer = nullptr;
    byte *pBuffer = nullptr;

    // MFCreate Sample
    hr = MFCreateSample(&pSample);
    if (hr != S_OK) 
        throw MyException(hr, "getNALSample::MFCreateSample failed");

    // Get buffer
    hr = MFCreateMemoryBuffer(maxBuffSize, &pMediaBuffer);
    if (hr != S_OK) 
        throw MyException(hr, "getNALSample::MFCreateMemoryBuffer failed");

    // Set up buffer pointer
    hr = pMediaBuffer->Lock(&pBuffer, nullptr, nullptr);
    if (hr != S_OK) 
        throw MyException(hr, "getNALSample::pMediaBuffer->Lock failed");

    // Build NAL from decoder que
    hr = pMediaBuffer->SetCurrentLength(buildNAL(pBuffer, decoderPkt));
    if (hr != S_OK) 
        throw MyException(hr, "getNALSample::pMediaBuffer->SetCurrentLength failed");

    hr = pMediaBuffer->Unlock();
    if (hr != S_OK) 
        throw MyException(hr, "getNALSample::pMediaBuffer->Unlock failed");

    // Add buffer to sample
    hr = pSample->AddBuffer(pMediaBuffer);
    if (hr != S_OK) 
        throw MyException(hr, "getNALSample::pSample->AddBuffer failed");

    return pSample;
}

谢谢。

Answer 1

它工作正常。

您看不到这种用法的原因是，它的设计方式看起来确实是一种 "private" 方法。

此外，它会抛出异常而不是返回 HRESULT。就个人而言，我总是更喜欢返回一个 HRESULT 而不是抛出（即使你随身携带了一个 ComPtr），因为它更容易处理。

这是一种（我承认更复杂）的方法，看起来更 "COM"（如果你愿意的话，可以更 "Microsoft"）：

HRESULT GetSample(REFIID riid, void**ppv) // you can add other parameters (in front, and leave riid and ppv last)
{
  HRESULT hr = S_OK;
  CComPtr<IMFSample> pSample;

  hr = MFCreateSample(&pSample);
  if (hr != S_OK) return hr; // or if (FAILED(hr) ? depends if you handle S_FALSE an other non error cases

  // ...Create buffer, Add buffer, etc...

  return pSample->QueryInterface(riid, ppv);
}

从函数 return CComPtr 可以吗？

Is it ok to return a CComPtr from a function?

com

atl

ms-media-foundation