blob: 21ffd9943c9740531845aaa3a4cb569d15e85363 [file] [log] [blame]
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "printing/image.h"
#include "base/file_util.h"
#include "base/md5.h"
#include "base/string_number_conversions.h"
#include "gfx/codec/png_codec.h"
#include "gfx/rect.h"
#include "skia/ext/platform_device.h"
#if defined(OS_WIN)
#include "gfx/gdi_util.h" // EMF support
#elif defined(OS_MACOSX)
#include <ApplicationServices/ApplicationServices.h>
#include "base/scoped_cftyperef.h"
#endif
namespace {
// A simple class which temporarily overrides system settings.
// The bitmap image rendered via the PlayEnhMetaFile() function depends on
// some system settings.
// As a workaround for such dependency, this class saves the system settings
// and changes them. This class also restore the saved settings in its
// destructor.
class DisableFontSmoothing {
public:
explicit DisableFontSmoothing(bool disable) : enable_again_(false) {
if (disable) {
#if defined(OS_WIN)
BOOL enabled;
if (SystemParametersInfo(SPI_GETFONTSMOOTHING, 0, &enabled, 0) &&
enabled) {
if (SystemParametersInfo(SPI_SETFONTSMOOTHING, FALSE, NULL, 0))
enable_again_ = true;
}
#endif
}
}
~DisableFontSmoothing() {
if (enable_again_) {
#if defined(OS_WIN)
BOOL result = SystemParametersInfo(SPI_SETFONTSMOOTHING, TRUE, NULL, 0);
DCHECK(result);
#endif
}
}
private:
bool enable_again_;
DISALLOW_COPY_AND_ASSIGN(DisableFontSmoothing);
};
} // namespace
namespace printing {
Image::Image(const FilePath& path)
: row_length_(0),
ignore_alpha_(true) {
std::string data;
file_util::ReadFileToString(path, &data);
bool success = false;
if (path.MatchesExtension(FILE_PATH_LITERAL(".png"))) {
success = LoadPng(data);
} else if (path.MatchesExtension(FILE_PATH_LITERAL(".emf"))) {
success = LoadMetafile(data);
} else {
DCHECK(false);
}
if (!success) {
size_.SetSize(0, 0);
row_length_ = 0;
data_.clear();
}
}
Image::Image(const NativeMetafile& metafile)
: row_length_(0),
ignore_alpha_(true) {
LoadMetafile(metafile);
}
Image::Image(const Image& image)
: size_(image.size_),
row_length_(image.row_length_),
data_(image.data_),
ignore_alpha_(image.ignore_alpha_) {
}
std::string Image::checksum() const {
MD5Digest digest;
MD5Sum(&data_[0], data_.size(), &digest);
return base::HexEncode(&digest, sizeof(digest));
}
bool Image::SaveToPng(const FilePath& filepath) const {
DCHECK(!data_.empty());
std::vector<unsigned char> compressed;
bool success = gfx::PNGCodec::Encode(&*data_.begin(),
gfx::PNGCodec::FORMAT_BGRA,
size_.width(),
size_.height(),
row_length_,
true,
&compressed);
DCHECK(success && compressed.size());
if (success) {
int write_bytes = file_util::WriteFile(
filepath,
reinterpret_cast<char*>(&*compressed.begin()), compressed.size());
success = (write_bytes == static_cast<int>(compressed.size()));
DCHECK(success);
}
return success;
}
double Image::PercentageDifferent(const Image& rhs) const {
if (size_.width() == 0 || size_.height() == 0 ||
rhs.size_.width() == 0 || rhs.size_.height() == 0)
return 100.;
int width = std::min(size_.width(), rhs.size_.width());
int height = std::min(size_.height(), rhs.size_.height());
// Compute pixels different in the overlap
int pixels_different = 0;
for (int y = 0; y < height; ++y) {
for (int x = 0; x < width; ++x) {
uint32 lhs_pixel = pixel_at(x, y);
uint32 rhs_pixel = rhs.pixel_at(x, y);
if (lhs_pixel != rhs_pixel)
++pixels_different;
}
// Look for extra right lhs pixels. They should be white.
for (int x = width; x < size_.width(); ++x) {
uint32 lhs_pixel = pixel_at(x, y);
if (lhs_pixel != Color(SK_ColorWHITE))
++pixels_different;
}
// Look for extra right rhs pixels. They should be white.
for (int x = width; x < rhs.size_.width(); ++x) {
uint32 rhs_pixel = rhs.pixel_at(x, y);
if (rhs_pixel != Color(SK_ColorWHITE))
++pixels_different;
}
}
// Look for extra bottom lhs pixels. They should be white.
for (int y = height; y < size_.height(); ++y) {
for (int x = 0; x < size_.width(); ++x) {
uint32 lhs_pixel = pixel_at(x, y);
if (lhs_pixel != Color(SK_ColorWHITE))
++pixels_different;
}
}
// Look for extra bottom rhs pixels. They should be white.
for (int y = height; y < rhs.size_.height(); ++y) {
for (int x = 0; x < rhs.size_.width(); ++x) {
uint32 rhs_pixel = rhs.pixel_at(x, y);
if (rhs_pixel != Color(SK_ColorWHITE))
++pixels_different;
}
}
// Like the WebKit ImageDiff tool, we define percentage different in terms
// of the size of the 'actual' bitmap.
double total_pixels = static_cast<double>(size_.width()) *
static_cast<double>(height);
return static_cast<double>(pixels_different) / total_pixels * 100.;
}
bool Image::LoadPng(const std::string& compressed) {
int w;
int h;
bool success = gfx::PNGCodec::Decode(
reinterpret_cast<const unsigned char*>(compressed.c_str()),
compressed.size(), gfx::PNGCodec::FORMAT_BGRA, &data_, &w, &h);
size_.SetSize(w, h);
row_length_ = size_.width() * sizeof(uint32);
return success;
}
bool Image::LoadMetafile(const std::string& data) {
DCHECK(!data.empty());
#if defined(OS_WIN) || defined(OS_MACOSX)
NativeMetafile metafile;
metafile.CreateFromData(data.data(), data.size());
return LoadMetafile(metafile);
#else
NOTIMPLEMENTED();
return false;
#endif
}
bool Image::LoadMetafile(const NativeMetafile& metafile) {
#if defined(OS_WIN)
gfx::Rect rect(metafile.GetBounds());
DisableFontSmoothing disable_in_this_scope(true);
// Create a temporary HDC and bitmap to retrieve the rendered data.
HDC hdc = CreateCompatibleDC(NULL);
BITMAPV4HEADER hdr;
DCHECK_EQ(rect.x(), 0);
DCHECK_EQ(rect.y(), 0);
DCHECK_GE(rect.width(), 0); // Metafile could be empty.
DCHECK_GE(rect.height(), 0);
if (rect.width() > 0 && rect.height() > 0) {
size_ = rect.size();
gfx::CreateBitmapV4Header(rect.width(), rect.height(), &hdr);
void* bits;
HBITMAP bitmap = CreateDIBSection(hdc,
reinterpret_cast<BITMAPINFO*>(&hdr), 0,
&bits, NULL, 0);
DCHECK(bitmap);
DCHECK(SelectObject(hdc, bitmap));
skia::PlatformDevice::InitializeDC(hdc);
bool success = metafile.Playback(hdc, NULL);
row_length_ = size_.width() * sizeof(uint32);
size_t bytes = row_length_ * size_.height();
DCHECK(bytes);
data_.resize(bytes);
memcpy(&*data_.begin(), bits, bytes);
DeleteDC(hdc);
DeleteObject(bitmap);
return success;
}
#elif defined(OS_MACOSX)
// The printing system uses single-page metafiles (page indexes are 1-based).
const unsigned int page_number = 1;
gfx::Rect rect(metafile.GetPageBounds(page_number));
if (rect.width() > 0 && rect.height() > 0) {
size_ = rect.size();
row_length_ = size_.width() * sizeof(uint32);
size_t bytes = row_length_ * size_.height();
DCHECK(bytes);
data_.resize(bytes);
scoped_cftyperef<CGColorSpaceRef> color_space(
CGColorSpaceCreateWithName(kCGColorSpaceGenericRGB));
scoped_cftyperef<CGContextRef> bitmap_context(
CGBitmapContextCreate(&*data_.begin(), size_.width(), size_.height(),
8, row_length_, color_space,
kCGImageAlphaPremultipliedLast));
DCHECK(bitmap_context.get());
metafile.RenderPage(page_number, bitmap_context,
CGRectMake(0, 0, size_.width(), size_.height()),
true, false, false, false);
}
#else
NOTIMPLEMENTED();
#endif
return false;
}
} // namespace printing