D3D11如何正确从交换链/渲染目标复制像素并保存文件

核心问题 #

D3D11的GPU纹理行步长（Row Pitch）不是简单的宽度×像素字节数——硬件会按固定规则（通常是256字节）对齐内存。非16:9这类标准分辨率时，步长和实际行像素字节数不一致，直接按分辨率宽度复制内存就会导致像素错位，这就是你遇到的异常根源。

正确实现方案 #

1. 创建CPU可访问的中转纹理 #

不能直接读取交换链的后台缓冲区（它通常是GPU独占、不可读的），必须先创建一个staging纹理作为中间载体：

ID3D11Texture2D* CreateReadbackTexture(ID3D11Device* device, UINT width, UINT height, DXGI_FORMAT format) {
    D3D11_TEXTURE2D_DESC desc{};
    desc.Width = width;
    desc.Height = height;
    desc.MipLevels = 1;
    desc.ArraySize = 1;
    desc.Format = format;
    desc.SampleDesc.Count = 1;
    desc.SampleDesc.Quality = 0;
    desc.Usage = D3D11_USAGE_STAGING; // 标记为CPU可访问的 staging 纹理
    desc.BindFlags = 0;
    desc.CPUAccessFlags = D3D11_CPU_ACCESS_READ;
    desc.MiscFlags = 0;

    ID3D11Texture2D* readbackTex = nullptr;
    device->CreateTexture2D(&desc, nullptr, &readbackTex);
    return readbackTex;
}

2. 复制后台缓冲区到中转纹理 #

用CopyResource完成GPU内部的内容复制：

void CopyBackBufferToReadback(ID3D11DeviceContext* context, ID3D11Texture2D* backBuffer, ID3D11Texture2D* readbackTex) {
    context->CopyResource(readbackTex, backBuffer);
}

3. 映射纹理并处理步长复制 #

映射staging纹理后，必须用实际返回的RowPitch作为源数据的行间隔，逐行复制像素到目标缓冲区：

bool ExtractPixelData(ID3D11DeviceContext* context, ID3D11Texture2D* readbackTex, UINT width, UINT height, std::vector<unsigned char>& outPixels) {
    D3D11_MAPPED_SUBRESOURCE mapped{};
    HRESULT hr = context->Map(readbackTex, 0, D3D11_MAP_READ, 0, &mapped);
    if (FAILED(hr)) return false;

    // 假设像素格式为RGBA8（4字节/像素），其他格式需调整字节数
    const UINT pixelByteSize = 4;
    const UINT targetRowBytes = width * pixelByteSize;
    outPixels.resize(height * targetRowBytes);

    unsigned char* srcPtr = static_cast<unsigned char*>(mapped.pData);
    unsigned char* dstPtr = outPixels.data();
    // 逐行复制，跳过步长中多余的对齐字节
    for (UINT y = 0; y < height; ++y) {
        memcpy(dstPtr, srcPtr, targetRowBytes);
        srcPtr += mapped.RowPitch;
        dstPtr += targetRowBytes;
    }

    context->Unmap(readbackTex, 0);
    return true;
}

4. 用stb_image_write保存图片 #

把处理后的像素数组传入stb，注意通道数匹配：

void SavePixelDataToPNG(const std::string& path, UINT width, UINT height, const std::vector<unsigned char>& pixels) {
    // 保存RGB格式（忽略Alpha通道）
    std::vector<unsigned char> rgbPixels(width * height * 3);
    for (UINT i = 0; i < width * height; ++i) {
        rgbPixels[i*3] = pixels[i*4];     // R
        rgbPixels[i*3+1] = pixels[i*4+1]; // G
        rgbPixels[i*3+2] = pixels[i*4+2]; // B
    }
    stbi_write_png(path.c_str(), width, height, 3, rgbPixels.data(), width * 3);

    // 如果要保存RGBA格式，直接用：
    // stbi_write_png(path.c_str(), width, height, 4, pixels.data(), width * 4);
}

完整调用流程 #

bool CaptureBackBuffer(ID3D11Device* device, ID3D11DeviceContext* context, IDXGISwapChain* swapChain, const std::string& savePath) {
    // 获取交换链后台缓冲区
    ID3D11Texture2D* backBuffer = nullptr;
    HRESULT hr = swapChain->GetBuffer(0, IID_PPV_ARGS(&backBuffer));
    if (FAILED(hr)) return false;

    // 获取后台缓冲区的分辨率和格式
    D3D11_TEXTURE2D_DESC backBufferDesc{};
    backBuffer->GetDesc(&backBufferDesc);
    UINT width = backBufferDesc.Width;
    UINT height = backBufferDesc.Height;
    DXGI_FORMAT format = backBufferDesc.Format;

    // 注意：如果后台缓冲区格式不是RGBA8，需创建对应格式的readback纹理，否则颜色会错乱
    // 比如常见的B8G8R8A8格式，提取像素时要调整通道顺序

    // 创建读取纹理
    ID3D11Texture2D* readbackTex = CreateReadbackTexture(device, width, height, format);
    if (!readbackTex) {
        backBuffer->Release();
        return false;
    }

    // 复制内容
    CopyBackBufferToReadback(context, backBuffer, readbackTex);

    // 提取像素数据
    std::vector<unsigned char> pixels;
    bool success = ExtractPixelData(context, readbackTex, width, height, pixels);

    // 保存图片
    if (success) {
        SavePixelDataToPNG(savePath, width, height, pixels);
    }

    // 释放资源
    readbackTex->Release();
    backBuffer->Release();
    return success;
}

关键注意点 #

• 格式匹配：如果后台缓冲区是DXGI_FORMAT_B8G8R8A8_UNORM这类格式，提取像素时要调换R和B通道的顺序，否则颜色会反转。
• 步长必须用RowPitch：绝对不能用width×像素字节数代替mapped.RowPitch，这是适配任意分辨率的核心。
• 资源释放：所有创建的D3D资源必须调用Release()，避免内存泄漏。

内容的提问来源于stack exchange，提问作者Zermil