Function writeScreenshot

layersvt/screenshot.cpp:1207–1266

Save an image to a PPM image file. Returns true if file is successfully written, false otherwise.

// Save an image to a PPM image file.
// Returns true if file is successfully written, false otherwise.
static bool writeScreenshot(ScreenshotQueueData &data) {
    PROFILE("screenshot.write");
    // Map the final image so that the CPU can read it.
    const VkImageSubresource sr = {VK_IMAGE_ASPECT_COLOR_BIT, 0, 0};
    VkSubresourceLayout srLayout;
    const char *pixels;
    if (data.image3 == VK_NULL_HANDLE) {
        data.pTableDevice->GetImageSubresourceLayout(data.device, data.image2, &sr, &srLayout);
        VkResult err = data.pTableDevice->MapMemory(data.device, data.mem2, 0, VK_WHOLE_SIZE, 0, (void **)&pixels);
        if (VK_SUCCESS != err) return false;
    } else {
        data.pTableDevice->GetImageSubresourceLayout(data.device, data.image3, &sr, &srLayout);
        VkResult err = data.pTableDevice->MapMemory(data.device, data.mem3, 0, VK_WHOLE_SIZE, 0, (void **)&pixels);
        if (VK_SUCCESS != err) return false;
    }

    pixels += srLayout.offset;

    string fileName;
    if (settings.targetFolder.empty()) {
        fileName = to_string(data.frameNumber);
    } else {
        fileName = settings.targetFolder;
        fileName += "/" + to_string(data.frameNumber);
    }

    bool writeResult;
    switch (settings.screenshotExtension) {
        case Settings::ScreenshotExtension::PPM:
            fileName += ".ppm";
            writeResult = writePPM(fileName.c_str(), pixels, data.dstWidth, data.dstHeight, data.dstNumChannels, srLayout.rowPitch);
            break;
        case Settings::ScreenshotExtension::PAM:
            fileName += ".pam";
            writeResult = writePAM(fileName.c_str(), pixels, data.dstWidth, data.dstHeight, data.dstNumChannels, srLayout.rowPitch);
            break;
    }

    if (data.image3 == VK_NULL_HANDLE) {
        data.pTableDevice->UnmapMemory(data.device, data.mem2);
    } else {
        data.pTableDevice->UnmapMemory(data.device, data.mem3);
    }

    if (!writeResult) {
#ifdef ANDROID
        __android_log_print(ANDROID_LOG_ERROR, "screenshot", "Failed to write image: %s", fileName.c_str());
#else
        fprintf(stderr, "screenshot: Failed to write image: %s\n", fileName.c_str());
#endif
        return false;
    }
#ifdef ANDROID
    __android_log_print(ANDROID_LOG_INFO, "screenshot", "Saved image: %s", fileName.c_str());
#else
    printf("screenshot: Saved image: %s \n", fileName.c_str());
    fflush(stdout);
#endif