mirror of
https://pdfium.googlesource.com/pdfium
synced 2026-08-03 13:06:13 +08:00
9108588e63
Bug: pdfium:2127 Change-Id: I6958fd117a550a8c46a43265ee46370bd1615f38 Reviewed-on: https://pdfium-review.googlesource.com/c/pdfium/+/116851 Reviewed-by: Tom Sepez <tsepez@chromium.org> Reviewed-by: Thomas Sepez <tsepez@google.com> Commit-Queue: Lei Zhang <thestig@chromium.org>
55 lines
1.8 KiB
C++
55 lines
1.8 KiB
C++
// Copyright 2018 The PDFium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "testing/utils/bitmap_saver.h"
|
|
|
|
#include <fstream>
|
|
#include <vector>
|
|
|
|
#include "core/fxcrt/check.h"
|
|
#include "core/fxcrt/fx_safe_types.h"
|
|
#include "testing/image_diff/image_diff_png.h"
|
|
|
|
// static
|
|
void BitmapSaver::WriteBitmapToPng(FPDF_BITMAP bitmap,
|
|
const std::string& filename) {
|
|
const int stride = FPDFBitmap_GetStride(bitmap);
|
|
const int width = FPDFBitmap_GetWidth(bitmap);
|
|
const int height = FPDFBitmap_GetHeight(bitmap);
|
|
CHECK(stride >= 0);
|
|
CHECK(width >= 0);
|
|
CHECK(height >= 0);
|
|
FX_SAFE_FILESIZE size = stride;
|
|
size *= height;
|
|
auto input = pdfium::make_span(
|
|
static_cast<const uint8_t*>(FPDFBitmap_GetBuffer(bitmap)),
|
|
pdfium::ValueOrDieForType<size_t>(size));
|
|
|
|
std::vector<uint8_t> png;
|
|
int format = FPDFBitmap_GetFormat(bitmap);
|
|
if (format == FPDFBitmap_Gray) {
|
|
png = image_diff_png::EncodeGrayPNG(input, width, height, stride);
|
|
} else if (format == FPDFBitmap_BGR) {
|
|
png = image_diff_png::EncodeBGRPNG(input, width, height, stride);
|
|
} else {
|
|
png = image_diff_png::EncodeBGRAPNG(input, width, height, stride,
|
|
/*discard_transparency=*/false);
|
|
}
|
|
|
|
DCHECK(!png.empty());
|
|
DCHECK(filename.size() < 256u);
|
|
|
|
std::ofstream png_file;
|
|
png_file.open(filename, std::ios_base::out | std::ios_base::binary);
|
|
png_file.write(reinterpret_cast<char*>(&png.front()), png.size());
|
|
DCHECK(png_file.good());
|
|
png_file.close();
|
|
}
|
|
|
|
// static
|
|
void BitmapSaver::WriteBitmapToPng(CFX_DIBitmap* bitmap,
|
|
const std::string& filename) {
|
|
WriteBitmapToPng(reinterpret_cast<FPDF_BITMAP>(bitmap), filename);
|
|
}
|