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 "SkMallocPixelRef.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.fWidth < 0 || info.fHeight < 0) {
  22             return false;
  23         }
  24         if (info.fWidth > maxDimension || info.fHeight > maxDimension) {
  25             return false;
  26         }
  27         if ((unsigned)info.fColorType > (unsigned)kLastEnum_SkColorType) {
  28             return false;
  29         }
  30         if ((unsigned)info.fAlphaType > (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.fHeight * 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 private:
  73     SkImage_Raster() : INHERITED(0, 0) {}
  74
  75     SkBitmap    fBitmap;
  76
  77     typedef SkImage_Base INHERITED;
  78 };
  79
  80 ///////////////////////////////////////////////////////////////////////////////
  81
  82 SkImage* SkImage_Raster::NewEmpty() {
  83     // Returns lazily created singleton
  84     static SkImage* gEmpty;
  85     if (NULL == gEmpty) {
  86         gEmpty = SkNEW(SkImage_Raster);
  87     }
  88     gEmpty->ref();
  89     return gEmpty;
  90 }
  91
  92 static void release_data(void* addr, void* context) {
  93     SkData* data = static_cast<SkData*>(context);
  94     data->unref();
  95 }
  96
  97 SkImage_Raster::SkImage_Raster(const Info& info, SkData* data, size_t rowBytes)
  98     : INHERITED(info.fWidth, info.fHeight)
  99 {
 100     data->ref();
 101     void* addr = const_cast<void*>(data->data());
 102     SkColorTable* ctable = NULL;
 103
 104     fBitmap.installPixels(info, addr, rowBytes, ctable, release_data, data);
 105     fBitmap.setImmutable();
 106     fBitmap.lockPixels();
 107 }
 108
 109 SkImage_Raster::SkImage_Raster(const Info& info, SkPixelRef* pr, size_t rowBytes)
 110     : INHERITED(info.fWidth, info.fHeight)
 111 {
 112     fBitmap.setInfo(info, rowBytes);
 113     fBitmap.setPixelRef(pr);
 114     fBitmap.lockPixels();
 115 }
 116
 117 SkImage_Raster::~SkImage_Raster() {}
 118
 119 SkShader* SkImage_Raster::onNewShader(SkShader::TileMode tileX, SkShader::TileMode tileY,
 120                                       const SkMatrix* localMatrix) const {
 121     return SkShader::CreateBitmapShader(fBitmap, tileX, tileY, localMatrix);
 122 }
 123
 124 void SkImage_Raster::onDraw(SkCanvas* canvas, SkScalar x, SkScalar y, const SkPaint* paint) const {
 125     canvas->drawBitmap(fBitmap, x, y, paint);
 126 }
 127
 128 void SkImage_Raster::onDrawRectToRect(SkCanvas* canvas, const SkRect* src, const SkRect& dst,
 129                                       const SkPaint* paint) const {
 130     canvas->drawBitmapRectToRect(fBitmap, src, dst, paint);
 131 }
 132
 133 bool SkImage_Raster::onReadPixels(SkBitmap* dst, const SkIRect& subset) const {
 134     if (dst->pixelRef()) {
 135         return this->INHERITED::onReadPixels(dst, subset);
 136     } else {
 137         SkBitmap src;
 138         if (!fBitmap.extractSubset(&src, subset)) {
 139             return false;
 140         }
 141         return src.copyTo(dst, src.colorType());
 142     }
 143 }
 144
 145 const void* SkImage_Raster::onPeekPixels(SkImageInfo* infoPtr, size_t* rowBytesPtr) const {
 146     const SkImageInfo info = fBitmap.info();
 147     if ((kUnknown_SkColorType == info.colorType()) || !fBitmap.getPixels()) {
 148         return NULL;
 149     }
 150     *infoPtr = info;
 151     *rowBytesPtr = fBitmap.rowBytes();
 152     return fBitmap.getPixels();
 153 }
 154
 155 bool SkImage_Raster::getROPixels(SkBitmap* dst) const {
 156     *dst = fBitmap;
 157     return true;
 158 }
 159
 160 ///////////////////////////////////////////////////////////////////////////////
 161
 162 SkImage* SkImage::NewRasterCopy(const SkImageInfo& info, const void* pixels, size_t rowBytes) {
 163     if (!SkImage_Raster::ValidArgs(info, rowBytes)) {
 164         return NULL;
 165     }
 166     if (0 == info.fWidth && 0 == info.fHeight) {
 167         return SkImage_Raster::NewEmpty();
 168     }
 169     // check this after empty-check
 170     if (NULL == pixels) {
 171         return NULL;
 172     }
 173
 174     // Here we actually make a copy of the caller's pixel data
 175     SkAutoDataUnref data(SkData::NewWithCopy(pixels, info.fHeight * rowBytes));
 176     return SkNEW_ARGS(SkImage_Raster, (info, data, rowBytes));
 177 }
 178
 179
 180 SkImage* SkImage::NewRasterData(const SkImageInfo& info, SkData* data, size_t rowBytes) {
 181     if (!SkImage_Raster::ValidArgs(info, rowBytes)) {
 182         return NULL;
 183     }
 184     if (0 == info.fWidth && 0 == info.fHeight) {
 185         return SkImage_Raster::NewEmpty();
 186     }
 187     // check this after empty-check
 188     if (NULL == data) {
 189         return NULL;
 190     }
 191
 192     // did they give us enough data?
 193     size_t size = info.fHeight * rowBytes;
 194     if (data->size() < size) {
 195         return NULL;
 196     }
 197
 198     return SkNEW_ARGS(SkImage_Raster, (info, data, rowBytes));
 199 }
 200
 201 SkImage* SkNewImageFromPixelRef(const SkImageInfo& info, SkPixelRef* pr,
 202                                 size_t rowBytes) {
 203     return SkNEW_ARGS(SkImage_Raster, (info, pr, rowBytes));
 204 }
 205
 206 SkPixelRef* SkBitmapImageGetPixelRef(SkImage* image) {
 207     return ((SkImage_Raster*)image)->getPixelRef();
 208 }