printing/image.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "printing/image.h"
   6
   7 #include <algorithm>
   8
   9 #include "base/files/file_util.h"
  10 #include "base/md5.h"
  11 #include "base/numerics/safe_conversions.h"
  12 #include "base/strings/string_number_conversions.h"
  13 #include "printing/metafile.h"
  14 #include "printing/pdf_metafile_skia.h"
  15 #include "third_party/skia/include/core/SkColor.h"
  16 #include "ui/gfx/codec/png_codec.h"
  17
  18 namespace printing {
  19
  20 Image::Image(const base::FilePath& path)
  21     : row_length_(0),
  22       ignore_alpha_(true) {
  23   std::string data;
  24   base::ReadFileToString(path, &data);
  25   bool success = false;
  26   if (path.MatchesExtension(FILE_PATH_LITERAL(".png"))) {
  27     success = LoadPng(data);
  28   } else if (path.MatchesExtension(FILE_PATH_LITERAL(".emf"))) {
  29     success = LoadMetafile(data);
  30   } else {
  31     DCHECK(false);
  32   }
  33   if (!success) {
  34     size_.SetSize(0, 0);
  35     row_length_ = 0;
  36     data_.clear();
  37   }
  38 }
  39
  40 Image::Image(const Metafile& metafile)
  41     : row_length_(0),
  42       ignore_alpha_(true) {
  43   LoadMetafile(metafile);
  44 }
  45
  46 Image::Image(const Image& image)
  47     : size_(image.size_),
  48       row_length_(image.row_length_),
  49       data_(image.data_),
  50       ignore_alpha_(image.ignore_alpha_) {
  51 }
  52
  53 Image::~Image() {}
  54
  55 std::string Image::checksum() const {
  56   base::MD5Digest digest;
  57   base::MD5Sum(&data_[0], data_.size(), &digest);
  58   return base::HexEncode(&digest, sizeof(digest));
  59 }
  60
  61 bool Image::SaveToPng(const base::FilePath& filepath) const {
  62   DCHECK(!data_.empty());
  63   std::vector<unsigned char> compressed;
  64   bool success = gfx::PNGCodec::Encode(&*data_.begin(),
  65                                        gfx::PNGCodec::FORMAT_BGRA,
  66                                        size_,
  67                                        row_length_,
  68                                        true,
  69                                        std::vector<gfx::PNGCodec::Comment>(),
  70                                        &compressed);
  71   DCHECK(success && compressed.size());
  72   if (success) {
  73     int write_bytes = base::WriteFile(
  74         filepath,
  75         reinterpret_cast<char*>(&*compressed.begin()),
  76         base::checked_cast<int>(compressed.size()));
  77     success = (write_bytes == static_cast<int>(compressed.size()));
  78     DCHECK(success);
  79   }
  80   return success;
  81 }
  82
  83 double Image::PercentageDifferent(const Image& rhs) const {
  84   if (size_.width() == 0 || size_.height() == 0 ||
  85     rhs.size_.width() == 0 || rhs.size_.height() == 0)
  86     return 100.;
  87
  88   int width = std::min(size_.width(), rhs.size_.width());
  89   int height = std::min(size_.height(), rhs.size_.height());
  90   // Compute pixels different in the overlap
  91   int pixels_different = 0;
  92   for (int y = 0; y < height; ++y) {
  93     for (int x = 0; x < width; ++x) {
  94       uint32 lhs_pixel = pixel_at(x, y);
  95       uint32 rhs_pixel = rhs.pixel_at(x, y);
  96       if (lhs_pixel != rhs_pixel)
  97         ++pixels_different;
  98     }
  99
 100     // Look for extra right lhs pixels. They should be white.
 101     for (int x = width; x < size_.width(); ++x) {
 102       uint32 lhs_pixel = pixel_at(x, y);
 103       if (lhs_pixel != Color(SK_ColorWHITE))
 104         ++pixels_different;
 105     }
 106
 107     // Look for extra right rhs pixels. They should be white.
 108     for (int x = width; x < rhs.size_.width(); ++x) {
 109       uint32 rhs_pixel = rhs.pixel_at(x, y);
 110       if (rhs_pixel != Color(SK_ColorWHITE))
 111         ++pixels_different;
 112     }
 113   }
 114
 115   // Look for extra bottom lhs pixels. They should be white.
 116   for (int y = height; y < size_.height(); ++y) {
 117     for (int x = 0; x < size_.width(); ++x) {
 118       uint32 lhs_pixel = pixel_at(x, y);
 119       if (lhs_pixel != Color(SK_ColorWHITE))
 120         ++pixels_different;
 121     }
 122   }
 123
 124   // Look for extra bottom rhs pixels. They should be white.
 125   for (int y = height; y < rhs.size_.height(); ++y) {
 126     for (int x = 0; x < rhs.size_.width(); ++x) {
 127       uint32 rhs_pixel = rhs.pixel_at(x, y);
 128       if (rhs_pixel != Color(SK_ColorWHITE))
 129         ++pixels_different;
 130     }
 131   }
 132
 133   // Like the WebKit ImageDiff tool, we define percentage different in terms
 134   // of the size of the 'actual' bitmap.
 135   double total_pixels = static_cast<double>(size_.width()) *
 136       static_cast<double>(height);
 137   return static_cast<double>(pixels_different) / total_pixels * 100.;
 138 }
 139
 140 bool Image::LoadPng(const std::string& compressed) {
 141   int w;
 142   int h;
 143   bool success = gfx::PNGCodec::Decode(
 144       reinterpret_cast<const unsigned char*>(compressed.c_str()),
 145       compressed.size(), gfx::PNGCodec::FORMAT_BGRA, &data_, &w, &h);
 146   size_.SetSize(w, h);
 147   row_length_ = size_.width() * sizeof(uint32);
 148   return success;
 149 }
 150
 151 bool Image::LoadMetafile(const std::string& data) {
 152   DCHECK(!data.empty());
 153   PdfMetafileSkia metafile;
 154   if (!metafile.InitFromData(data.data(),
 155                              base::checked_cast<uint32>(data.size())))
 156     return false;
 157   return LoadMetafile(metafile);
 158 }
 159
 160 }  // namespace printing