当源过滤器停止发送时从相机过滤器获取连续帧?

时间:2013-07-28 10:22:48

标签: c++ visual-c++ filter transform directshow

enter image description here

大家好,

我有一个转换滤镜,它有两个输入,一个来自摄像头,另一个来自文件源。在我的变换滤镜中,我混合了两个来源的输入。

转换过滤器派生自CTransformFilter

class CWMTransformFilter : public CTransformFilter

和额外的引脚派生自:CTransformInputPin(内部派生自CBaseInputPin)

class CFileInputPin : public CTransformInputPin

在我的情况下,发生的事情是,如果文件来源很小(假设10秒),我也会从相机输入10秒钟,之后相机会停止向输入引脚发送帧。

所以我现在需要的是:
1。即使源过滤器停止发送,如何通知相机发送帧? 2.如何在停止播放源文件时重新启动源过滤器?(类似于循环播放文件)

更新

STDMETHODIMP CFileInputPin::EndOfStream()
{
    //return CTransformInputPin::EndOfStream();
    return S_OK;
}

STDMETHODIMP CFileInputPin::Receive(IMediaSample* pSample)
{
    HRESULT hr;
    BYTE* pBufferIn;
    long lBufferLength, lBufferSize;
    hr = CBaseInputPin::Receive(pSample);
    if (FAILED(hr))
    {
        printf("Error !!\n");
        return hr;
    }
    hr = pSample->GetPointer(&pBufferIn);
    DWORD stat = WaitForSingleObject(m_pFil->m_QSem,0L);
    BOOL bSem = FALSE;
    if( WAIT_OBJECT_0 == stat )
    {
        BYTE *pBuf = (BYTE *) malloc( Wsize*2 );
        memcpy(pBuf,pBufferIn,Wsize); //lBufferLength);
        sEncodedFrame CurFrame={pBuf,Wsize};
        m_pFil->m_Q.push(CurFrame); //push it onto the queue

        bSem = ReleaseSemaphore(m_pFil->m_QSem,1,NULL);
        if(!bSem)
        {
            printf("ReleaseSemaphore error: %d \n", GetLastError());
        }

        return S_OK;
    }
    else
    {
        printf("Cant Receive frame 0x%x \n",stat);
        return E_FAIL;
    }
    return S_OK;
}

HRESULT CWMTransformFilter::Transform(IMediaSample *pSource, IMediaSample *pDest)
{
    unsigned char r,g,b;
    unsigned char y,u,v;

    BYTE *pBufferIn, *pBufferOut, *pBuf;
    HRESULT hr = pSource->GetPointer(&pBufferIn);
    hr = pDest->GetPointer(&pBufferOut);
    if (FAILED(hr))
    {
        return hr;
    }
    long srclen = pSource->GetActualDataLength();
    long dstlen = pDest->GetActualDataLength();

    LONG pLastCnt;
    BOOL bSem = FALSE;
    //printf("Waiting to fill buffer %d\n",pSource);
    //return S_OK;
    //try
    //{
    while(1)
    {
        //if(1)
        DWORD ret = WaitForSingleObject(m_QSem,0L);
        if(ret != WAIT_OBJECT_0)
        {
            printf("Get error %d \n",GetLastError());
        }
        if( WAIT_OBJECT_0 == ret )
        {

            sEncodedFrame Frame;
            if( m_Q.empty() == false )
            {
                Frame = m_Q.front();
                m_Q.pop();
                pBuf = (BYTE*) malloc(dstlen*2);
                pLastCnt = Frame.iValidSize;
                printf("Copy onto queue \n");
                memcpy(pBuf,Frame.pFrame,pLastCnt); //Frame.iValidSize);
                free(Frame.pFrame);
                //delete &Frame;
                bSem = ReleaseSemaphore(m_QSem,1,NULL);
                if(!bSem)
                {
                    printf("ReleaseSemaphore error: %d \n", GetLastError());
                }
                hr = S_OK;
                break;
            }
            else
            {
                bSem = ReleaseSemaphore(m_QSem,1,NULL);
                if(!bSem)
                {
                    printf("ReleaseSemaphore error: %d \n", GetLastError());
                }
                //return S_OK;
                hr = E_FAIL;
            }
        }
        else
        {
            //return S_OK;
            hr = E_FAIL;
        }
    }

    for(i = 0; i < windowWidth*2*windowHeight ; i+=4)
    {
     y = pBufferIn[i];
     u = pBufferIn[i+1];
     v = pBufferIn[i+3];
     r = y + 1.4075 * (v - 128);
     g = y - 0.3455 * (u - 128) - (0.7169 * (v - 128));
     b = y + 1.7790 * (u - 128);

    if(((r > b) &&(g > b)) && (g <= 200) )
    {
        pBufferIn[i] = pBuf[i];
        pBufferIn[i+1] = pBuf[i+1];
        pBufferIn[i+2] = pBuf[i+2];
        pBufferIn[i+3] = pBuf[i+3];
    }
}

    // Process the data.
    memcpy(pBufferOut,pBufferIn,pSource->GetSize()); //after blend

    pDest->SetActualDataLength(pDest->GetSize());
    pDest->SetSyncPoint(TRUE);
    return S_OK;
}

CFileInputPin :: Receive是我从文件输入引脚接收样本的地方,
CFileInputPin :: EndOfStream(),通知样本已完成 CWMTransformFilter :: Transform(),是将样本分配给渲染器的分支。

谢谢,
希亚姆

1 个答案:

答案 0 :(得分:0)

这个问题有很多棘手的问题。

首先,第一个兴趣点是自定义双输入滤波器。无论输入过滤器做什么,这个转换过滤器都可以排除 - 它完全取决于它的实现是否允许两个输入支路流,或者它将阻止其中一个。常见的(典型)规则是,如果过滤器有2个以上的输入,则它是两个中的任何一个:

  1. 其中一个流是主流,其他输入如果有意义或被丢弃则要么进行处理
  2. 过滤器阻止输入输入分支以保持获取具有匹配时间戳的数据,然后在内部处理过程中合并流。
  3. 输入流通常是样本序列,然后是EOS通知。特别是,如果其中一个源未发送EOS,或者转换过滤器未正确处理,则可能会发生冻结。

    第二大问题是寻求。您通常不会寻求图表的一部分。但是,这正是你想要做的。您可以尝试独立地寻找源过滤器来定位自己的搜索接口,否则您可以实现一个缓冲区并保存它发送的所有内容,并在源发送EOS后无限循环。对此没有任何建议 - 您决定在您的方案中适合什么。

    另一种选择是拆分图形并将它们桥接,以便您可以常规方式搜索/重新启动源图形。