src/third_party/skia/src/image/SkImage_Raster.cpp

   1 /*
   2  * Copyright 2012 Google Inc.
   3  *
   4  * Use of this source code is governed by a BSD-style license that can be
   5  * found in the LICENSE file.
   6  */
   7
   8 #include "SkImage_Base.h"
   9 #include "SkImagePriv.h"
  10 #include "SkBitmap.h"
  11 #include "SkCanvas.h"
  12 #include "SkData.h"
  13 #include "SkDecodingImageGenerator.h"
  14
  15 class SkImage_Raster : public SkImage_Base {
  16 public:
  17     static bool ValidArgs(const Info& info, size_t rowBytes) {
  18         const int maxDimension = SK_MaxS32 >> 2;
  19         const size_t kMaxPixelByteSize = SK_MaxS32;
  20
  21         if (info.width() < 0 || info.height() < 0) {
  22             return false;
  23         }
  24         if (info.width() > maxDimension || info.height() > maxDimension) {
  25             return false;
  26         }
  27         if ((unsigned)info.colorType() > (unsigned)kLastEnum_SkColorType) {
  28             return false;
  29         }
  30         if ((unsigned)info.alphaType() > (unsigned)kLastEnum_SkAlphaType) {
  31             return false;
  32         }
  33
  34         if (kUnknown_SkColorType == info.colorType()) {
  35             return false;
  36         }
  37
  38         // TODO: check colorspace
  39
  40         if (rowBytes < SkImageMinRowBytes(info)) {
  41             return false;
  42         }
  43
  44         int64_t size = (int64_t)info.height() * rowBytes;
  45         if (size > (int64_t)kMaxPixelByteSize) {
  46             return false;
  47         }
  48         return true;
  49     }
  50
  51     static SkImage* NewEmpty();
  52
  53     SkImage_Raster(const SkImageInfo&, SkData*, size_t rb);
  54     virtual ~SkImage_Raster();
  55
  56     virtual void onDraw(SkCanvas*, SkScalar, SkScalar, const SkPaint*) const SK_OVERRIDE;
  57     virtual void onDrawRectToRect(SkCanvas*, const SkRect*, const SkRect&,
  58                                   const SkPaint*) const SK_OVERRIDE;
  59     virtual bool onReadPixels(SkBitmap*, const SkIRect&) const SK_OVERRIDE;
  60     virtual const void* onPeekPixels(SkImageInfo*, size_t* /*rowBytes*/) const SK_OVERRIDE;
  61     virtual bool getROPixels(SkBitmap*) const SK_OVERRIDE;
  62
  63     // exposed for SkSurface_Raster via SkNewImageFromPixelRef
  64     SkImage_Raster(const SkImageInfo&, SkPixelRef*, size_t rowBytes);
  65
  66     SkPixelRef* getPixelRef() const { return fBitmap.pixelRef(); }
  67
  68     virtual SkShader* onNewShader(SkShader::TileMode,
  69                                   SkShader::TileMode,
  70                                   const SkMatrix* localMatrix) const SK_OVERRIDE;
  71
  72     virtual bool isOpaque() const SK_OVERRIDE;
  73
  74     SkImage_Raster(const SkBitmap& bm)
  75         : INHERITED(bm.width(), bm.height())
  76         , fBitmap(bm) {}
  77
  78 private:
  79     SkImage_Raster() : INHERITED(0, 0) {}
  80
  81     SkBitmap    fBitmap;
  82
  83     typedef SkImage_Base INHERITED;
  84 };
  85
  86 ///////////////////////////////////////////////////////////////////////////////
  87
  88 SkImage* SkImage_Raster::NewEmpty() {
  89     // Returns lazily created singleton
  90     static SkImage* gEmpty;
  91     if (NULL == gEmpty) {
  92         gEmpty = SkNEW(SkImage_Raster);
  93     }
  94     gEmpty->ref();
  95     return gEmpty;
  96 }
  97
  98 static void release_data(void* addr, void* context) {
  99     SkData* data = static_cast<SkData*>(context);
 100     data->unref();
 101 }
 102
 103 SkImage_Raster::SkImage_Raster(const Info& info, SkData* data, size_t rowBytes)
 104     : INHERITED(info.width(), info.height())
 105 {
 106     data->ref();
 107     void* addr = const_cast<void*>(data->data());
 108     SkColorTable* ctable = NULL;
 109
 110     fBitmap.installPixels(info, addr, rowBytes, ctable, release_data, data);
 111     fBitmap.setImmutable();
 112     fBitmap.lockPixels();
 113 }
 114
 115 SkImage_Raster::SkImage_Raster(const Info& info, SkPixelRef* pr, size_t rowBytes)
 116     : INHERITED(info.width(), info.height())
 117 {
 118     fBitmap.setInfo(info, rowBytes);
 119     fBitmap.setPixelRef(pr);
 120     fBitmap.lockPixels();
 121 }
 122
 123 SkImage_Raster::~SkImage_Raster() {}
 124
 125 SkShader* SkImage_Raster::onNewShader(SkShader::TileMode tileX, SkShader::TileMode tileY,
 126                                       const SkMatrix* localMatrix) const {
 127     return SkShader::CreateBitmapShader(fBitmap, tileX, tileY, localMatrix);
 128 }
 129
 130 void SkImage_Raster::onDraw(SkCanvas* canvas, SkScalar x, SkScalar y, const SkPaint* paint) const {
 131     canvas->drawBitmap(fBitmap, x, y, paint);
 132 }
 133
 134 void SkImage_Raster::onDrawRectToRect(SkCanvas* canvas, const SkRect* src, const SkRect& dst,
 135                                       const SkPaint* paint) const {
 136     canvas->drawBitmapRectToRect(fBitmap, src, dst, paint);
 137 }
 138
 139 bool SkImage_Raster::onReadPixels(SkBitmap* dst, const SkIRect& subset) const {
 140     if (dst->pixelRef()) {
 141         return this->INHERITED::onReadPixels(dst, subset);
 142     } else {
 143         SkBitmap src;
 144         if (!fBitmap.extractSubset(&src, subset)) {
 145             return false;
 146         }
 147         return src.copyTo(dst, src.colorType());
 148     }
 149 }
 150
 151 const void* SkImage_Raster::onPeekPixels(SkImageInfo* infoPtr, size_t* rowBytesPtr) const {
 152     const SkImageInfo info = fBitmap.info();
 153     if ((kUnknown_SkColorType == info.colorType()) || !fBitmap.getPixels()) {
 154         return NULL;
 155     }
 156     *infoPtr = info;
 157     *rowBytesPtr = fBitmap.rowBytes();
 158     return fBitmap.getPixels();
 159 }
 160
 161 bool SkImage_Raster::getROPixels(SkBitmap* dst) const {
 162     *dst = fBitmap;
 163     return true;
 164 }
 165
 166 ///////////////////////////////////////////////////////////////////////////////
 167
 168 SkImage* SkImage::NewRasterCopy(const SkImageInfo& info, const void* pixels, size_t rowBytes) {
 169     if (!SkImage_Raster::ValidArgs(info, rowBytes)) {
 170         return NULL;
 171     }
 172     if (0 == info.width() && 0 == info.height()) {
 173         return SkImage_Raster::NewEmpty();
 174     }
 175     // check this after empty-check
 176     if (NULL == pixels) {
 177         return NULL;
 178     }
 179
 180     // Here we actually make a copy of the caller's pixel data
 181     SkAutoDataUnref data(SkData::NewWithCopy(pixels, info.height() * rowBytes));
 182     return SkNEW_ARGS(SkImage_Raster, (info, data, rowBytes));
 183 }
 184
 185
 186 SkImage* SkImage::NewRasterData(const SkImageInfo& info, SkData* data, size_t rowBytes) {
 187     if (!SkImage_Raster::ValidArgs(info, rowBytes)) {
 188         return NULL;
 189     }
 190     if (0 == info.width() && 0 == info.height()) {
 191         return SkImage_Raster::NewEmpty();
 192     }
 193     // check this after empty-check
 194     if (NULL == data) {
 195         return NULL;
 196     }
 197
 198     // did they give us enough data?
 199     size_t size = info.height() * rowBytes;
 200     if (data->size() < size) {
 201         return NULL;
 202     }
 203
 204     return SkNEW_ARGS(SkImage_Raster, (info, data, rowBytes));
 205 }
 206
 207 SkImage* SkImage::NewFromGenerator(SkImageGenerator* generator) {
 208     SkBitmap bitmap;
 209     if (!SkInstallDiscardablePixelRef(generator, &bitmap)) {
 210         return NULL;
 211     }
 212     return SkNEW_ARGS(SkImage_Raster, (bitmap));
 213 }
 214
 215 SkImage* SkNewImageFromPixelRef(const SkImageInfo& info, SkPixelRef* pr,
 216                                 size_t rowBytes) {
 217     return SkNEW_ARGS(SkImage_Raster, (info, pr, rowBytes));
 218 }
 219
 220 SkPixelRef* SkBitmapImageGetPixelRef(SkImage* image) {
 221     return ((SkImage_Raster*)image)->getPixelRef();
 222 }
 223
 224 bool SkImage_Raster::isOpaque() const {
 225     return fBitmap.isOpaque();
 226 }