michael@0: /*
michael@0:  * Copyright 2012 Google Inc.
michael@0:  *
michael@0:  * Use of this source code is governed by a BSD-style license that can be
michael@0:  * found in the LICENSE file.
michael@0:  */
michael@0: 
michael@0: #include "SkImage_Base.h"
michael@0: #include "SkImagePriv.h"
michael@0: #include "SkBitmap.h"
michael@0: #include "SkCanvas.h"
michael@0: #include "SkData.h"
michael@0: #include "SkMallocPixelRef.h"
michael@0: 
michael@0: class SkImage_Raster : public SkImage_Base {
michael@0: public:
michael@0:     static bool ValidArgs(const Info& info, size_t rowBytes) {
michael@0:         const int maxDimension = SK_MaxS32 >> 2;
michael@0:         const size_t kMaxPixelByteSize = SK_MaxS32;
michael@0: 
michael@0:         if (info.fWidth < 0 || info.fHeight < 0) {
michael@0:             return false;
michael@0:         }
michael@0:         if (info.fWidth > maxDimension || info.fHeight > maxDimension) {
michael@0:             return false;
michael@0:         }
michael@0:         if ((unsigned)info.fColorType > (unsigned)kLastEnum_SkColorType) {
michael@0:             return false;
michael@0:         }
michael@0:         if ((unsigned)info.fAlphaType > (unsigned)kLastEnum_SkAlphaType) {
michael@0:             return false;
michael@0:         }
michael@0: 
michael@0:         if (SkImageInfoToBitmapConfig(info) == SkBitmap::kNo_Config) {
michael@0:             return false;
michael@0:         }
michael@0: 
michael@0:         // TODO: check colorspace
michael@0: 
michael@0:         if (rowBytes < SkImageMinRowBytes(info)) {
michael@0:             return false;
michael@0:         }
michael@0: 
michael@0:         int64_t size = (int64_t)info.fHeight * rowBytes;
michael@0:         if (size > (int64_t)kMaxPixelByteSize) {
michael@0:             return false;
michael@0:         }
michael@0:         return true;
michael@0:     }
michael@0: 
michael@0:     static SkImage* NewEmpty();
michael@0: 
michael@0:     SkImage_Raster(const SkImageInfo&, SkData*, size_t rb);
michael@0:     virtual ~SkImage_Raster();
michael@0: 
michael@0:     virtual void onDraw(SkCanvas*, SkScalar, SkScalar, const SkPaint*) SK_OVERRIDE;
michael@0:     virtual void onDrawRectToRect(SkCanvas*, const SkRect*, const SkRect&, const SkPaint*) SK_OVERRIDE;
michael@0:     virtual bool onReadPixels(SkBitmap*, const SkIRect&) const SK_OVERRIDE;
michael@0:     virtual const void* onPeekPixels(SkImageInfo*, size_t* /*rowBytes*/) const SK_OVERRIDE;
michael@0:     virtual bool getROPixels(SkBitmap*) const SK_OVERRIDE;
michael@0: 
michael@0:     // exposed for SkSurface_Raster via SkNewImageFromPixelRef
michael@0:     SkImage_Raster(const SkImageInfo&, SkPixelRef*, size_t rowBytes);
michael@0: 
michael@0:     SkPixelRef* getPixelRef() const { return fBitmap.pixelRef(); }
michael@0: 
michael@0: private:
michael@0:     SkImage_Raster() : INHERITED(0, 0) {}
michael@0: 
michael@0:     SkBitmap    fBitmap;
michael@0: 
michael@0:     typedef SkImage_Base INHERITED;
michael@0: };
michael@0: 
michael@0: ///////////////////////////////////////////////////////////////////////////////
michael@0: 
michael@0: SkImage* SkImage_Raster::NewEmpty() {
michael@0:     // Returns lazily created singleton
michael@0:     static SkImage* gEmpty;
michael@0:     if (NULL == gEmpty) {
michael@0:         gEmpty = SkNEW(SkImage_Raster);
michael@0:     }
michael@0:     gEmpty->ref();
michael@0:     return gEmpty;
michael@0: }
michael@0: 
michael@0: static void release_data(void* addr, void* context) {
michael@0:     SkData* data = static_cast<SkData*>(context);
michael@0:     data->unref();
michael@0: }
michael@0: 
michael@0: SkImage_Raster::SkImage_Raster(const Info& info, SkData* data, size_t rowBytes)
michael@0:     : INHERITED(info.fWidth, info.fHeight)
michael@0: {
michael@0:     data->ref();
michael@0:     void* addr = const_cast<void*>(data->data());
michael@0: 
michael@0:     fBitmap.installPixels(info, addr, rowBytes, release_data, data);
michael@0:     fBitmap.setImmutable();
michael@0:     fBitmap.lockPixels();
michael@0: }
michael@0: 
michael@0: SkImage_Raster::SkImage_Raster(const Info& info, SkPixelRef* pr, size_t rowBytes)
michael@0:     : INHERITED(info.fWidth, info.fHeight)
michael@0: {
michael@0:     fBitmap.setConfig(info, rowBytes);
michael@0:     fBitmap.setPixelRef(pr);
michael@0:     fBitmap.lockPixels();
michael@0: }
michael@0: 
michael@0: SkImage_Raster::~SkImage_Raster() {}
michael@0: 
michael@0: void SkImage_Raster::onDraw(SkCanvas* canvas, SkScalar x, SkScalar y, const SkPaint* paint) {
michael@0:     canvas->drawBitmap(fBitmap, x, y, paint);
michael@0: }
michael@0: 
michael@0: void SkImage_Raster::onDrawRectToRect(SkCanvas* canvas, const SkRect* src,
michael@0:                                       const SkRect& dst, const SkPaint* paint) {
michael@0:     canvas->drawBitmapRectToRect(fBitmap, src, dst, paint);
michael@0: }
michael@0: 
michael@0: bool SkImage_Raster::onReadPixels(SkBitmap* dst, const SkIRect& subset) const {
michael@0:     if (dst->pixelRef()) {
michael@0:         return this->INHERITED::onReadPixels(dst, subset);
michael@0:     } else {
michael@0:         SkBitmap src;
michael@0:         if (!fBitmap.extractSubset(&src, subset)) {
michael@0:             return false;
michael@0:         }
michael@0:         return src.copyTo(dst, src.colorType());
michael@0:     }
michael@0: }
michael@0: 
michael@0: const void* SkImage_Raster::onPeekPixels(SkImageInfo* infoPtr,
michael@0:                                          size_t* rowBytesPtr) const {
michael@0:     SkImageInfo info;
michael@0:     if (!fBitmap.asImageInfo(&info) || !fBitmap.getPixels()) {
michael@0:         return NULL;
michael@0:     }
michael@0:     *infoPtr = info;
michael@0:     *rowBytesPtr = fBitmap.rowBytes();
michael@0:     return fBitmap.getPixels();
michael@0: }
michael@0: 
michael@0: bool SkImage_Raster::getROPixels(SkBitmap* dst) const {
michael@0:     *dst = fBitmap;
michael@0:     return true;
michael@0: }
michael@0: 
michael@0: ///////////////////////////////////////////////////////////////////////////////
michael@0: 
michael@0: SkImage* SkImage::NewRasterCopy(const SkImageInfo& info, const void* pixels, size_t rowBytes) {
michael@0:     if (!SkImage_Raster::ValidArgs(info, rowBytes)) {
michael@0:         return NULL;
michael@0:     }
michael@0:     if (0 == info.fWidth && 0 == info.fHeight) {
michael@0:         return SkImage_Raster::NewEmpty();
michael@0:     }
michael@0:     // check this after empty-check
michael@0:     if (NULL == pixels) {
michael@0:         return NULL;
michael@0:     }
michael@0: 
michael@0:     // Here we actually make a copy of the caller's pixel data
michael@0:     SkAutoDataUnref data(SkData::NewWithCopy(pixels, info.fHeight * rowBytes));
michael@0:     return SkNEW_ARGS(SkImage_Raster, (info, data, rowBytes));
michael@0: }
michael@0: 
michael@0: 
michael@0: SkImage* SkImage::NewRasterData(const SkImageInfo& info, SkData* data, size_t rowBytes) {
michael@0:     if (!SkImage_Raster::ValidArgs(info, rowBytes)) {
michael@0:         return NULL;
michael@0:     }
michael@0:     if (0 == info.fWidth && 0 == info.fHeight) {
michael@0:         return SkImage_Raster::NewEmpty();
michael@0:     }
michael@0:     // check this after empty-check
michael@0:     if (NULL == data) {
michael@0:         return NULL;
michael@0:     }
michael@0: 
michael@0:     // did they give us enough data?
michael@0:     size_t size = info.fHeight * rowBytes;
michael@0:     if (data->size() < size) {
michael@0:         return NULL;
michael@0:     }
michael@0: 
michael@0:     return SkNEW_ARGS(SkImage_Raster, (info, data, rowBytes));
michael@0: }
michael@0: 
michael@0: SkImage* SkNewImageFromPixelRef(const SkImageInfo& info, SkPixelRef* pr,
michael@0:                                 size_t rowBytes) {
michael@0:     return SkNEW_ARGS(SkImage_Raster, (info, pr, rowBytes));
michael@0: }
michael@0: 
michael@0: SkPixelRef* SkBitmapImageGetPixelRef(SkImage* image) {
michael@0:     return ((SkImage_Raster*)image)->getPixelRef();
michael@0: }