c++ - 直接显示 ISampleGrabber : samples are upside-down and color channels reverse

Question

我必须使用 MS DirectShow 从相机捕捉视频帧(我只想要原始像素数据)。
我能够构建图形/过滤器网络(捕获设备过滤器和 ISampleGrabber)并实现回调(ISampleGrabberCB)。我收到大小合适的 sample 。

但是，它们总是上下颠倒(垂直翻转，即不旋转)并且颜色 channel 是 BGR 顺序(不是 RGB)。

我尝试将 BITMAPINFOHEADER 中的 biHeight 字段设置为正值和负值，但没有任何效果。根据 MSDN 文档，ISampleGrapper::SetMediaType() 无论如何都会忽略视频数据的格式 block 。

这是我看到的(用不同的相机记录的，不是 DS)，以及 DirectShow ISampleGrabber 给我的:“RGB”实际上分别是红色、绿色和蓝色:

我正在使用的代码示例，略有简化:

// Setting the media type...
AM_MEDIA_TYPE*   media_type   = 0 ;
this->ds.device_streamconfig->GetFormat(&media_type); // The IAMStreamConfig of the capture device
// Find the BMI header in the media type struct
BITMAPINFOHEADER* bmi_header;
if (media_type->formattype != FORMAT_VideoInfo) {
    bmi_header = &((VIDEOINFOHEADER*)media_type->pbFormat)->bmiHeader;
} else if (media_type->formattype != FORMAT_VideoInfo2) {
    bmi_header = &((VIDEOINFOHEADER2*)media_type->pbFormat)->bmiHeader;
} else {
    return false;
}
// Apply changes
media_type->subtype  = MEDIASUBTYPE_RGB24;
bmi_header->biWidth  = width;
bmi_header->biHeight = height;
// Set format to video device
this->ds.device_streamconfig->SetFormat(media_type);
// Set format for sample grabber
// bmi_header->biHeight = -(height); // tried this for either and both interfaces, no effect
this->ds.sample_grabber->SetMediaType(media_type);

// Connect filter pins
IPin* out_pin= getFilterPin(this->ds.device_filter, OUT,  0); // IBaseFilter interface for the capture device
IPin* in_pin = getFilterPin(this->ds.sample_grabber_filter,  IN,  0); // IBaseFilter interface for the sample grabber filter
out_pin->Connect(in_pin, media_type);

// Start capturing by callback
this->ds.sample_grabber->SetBufferSamples(false);
this->ds.sample_grabber->SetOneShot(false);
this->ds.sample_grabber->SetCallback(this, 1);
// start recording
this->ds.media_control->Run(); // IMediaControl interface

我正在检查每个函数的返回类型，没有发现任何错误。

感谢任何提示或想法。

我已经尝试过的事情:

1. 将捕获设备过滤器或样本采集器的 biHeight 字段设置为负值，或两者均设置为负值 - 没有任何效果。

2. 使用 IGraphBuilder 连接引脚 - 同样的问题。

3. 在更改媒体类型之前连接引脚 - 同样的问题。

4. 通过再次查询来检查过滤器是否实际应用了媒体类型 - 但它显然已应用或至少已存储。

5. 将图像解释为总字节反转(最后一个字节在前，第一个字节在后)- 然后它将水平翻转。

6. 检查是否是摄像机的问题 - 当我使用 VLC(DirectShow 捕获)对其进行测试时，它看起来很正常。

Answer 1

我注意到当使用 I420 色彩空间时，旋转消失了。此外，大多数当前的编解码器 (VP8) 都使用 I420 颜色空间作为原始 I/O 格式。

我在I420颜色空间写了一个简单的镜像框函数。

void Camera::OutputCallback(unsigned char* data, int len, uint32_t timestamp, void *instance_)
{
    Camera *instance = reinterpret_cast<Camera*>(instance_);

    Transport::RTPPacket packet;

    packet.rtpHeader.ts = timestamp;

    packet.payload = data;
    packet.payloadSize = len;

    if (instance->mirror)
    {
        Video::ResolutionValues rv = Video::GetValues(instance->resolution);
        int k = 0;

        // Chroma values
        for (int i = 0; i != rv.height; ++i)
        {
            for (int j = rv.width; j != 0; --j)
            {
                int l = ((rv.width * i) + j);
                instance->buffer[k++] = data[l];
            }
        }

        // U values
        for (int i = 0; i != rv.height/2; ++i)
        {
            for (int j = (rv.width/2); j != 0; --j)
            {
                int l = (((rv.width / 2) * i) + j) + rv.height*rv.width;
                instance->buffer[k++] = data[l];
            }
        }

        // V values
        for (int i = 0; i != rv.height / 2; ++i)
        {
            for (int j = (rv.width / 2); j != 0; --j)
            {
                int l = (((rv.width / 2) * i) + j) + rv.height*rv.width + (rv.width/2)*(rv.height/2);
                if (l == len)
                {
                    instance->buffer[k++] = 0;
                }
                else
                {
                    instance->buffer[k++] = data[l];
                }
            }
        }

        packet.payload = instance->buffer;
    }

    instance->receiver->Send(packet);
}