halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2013 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 "SkDiscardablePixelRef.h" |
| 9 | #include "SkDiscardableMemory.h" |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 10 | #include "SkImageGenerator.h" |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 11 | |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 12 | SkDiscardablePixelRef::SkDiscardablePixelRef(const SkImageInfo& info, |
| 13 | SkImageGenerator* generator, |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 14 | size_t rowBytes, |
| 15 | SkDiscardableMemory::Factory* fact) |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 16 | : INHERITED(info) |
| 17 | , fGenerator(generator) |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 18 | , fDMFactory(fact) |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 19 | , fRowBytes(rowBytes) |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 20 | , fDiscardableMemory(NULL) |
| 21 | { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 22 | SkASSERT(fGenerator != NULL); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 23 | SkASSERT(fRowBytes > 0); |
commit-bot@chromium.org | dd59799 | 2013-12-02 22:32:54 +0000 | [diff] [blame] | 24 | // The SkImageGenerator contract requires fGenerator to always |
| 25 | // decode the same image on each call to getPixels(). |
| 26 | this->setImmutable(); |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 27 | SkSafeRef(fDMFactory); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 28 | } |
| 29 | |
| 30 | SkDiscardablePixelRef::~SkDiscardablePixelRef() { |
reed@google.com | 1d0654f | 2013-12-12 22:37:32 +0000 | [diff] [blame] | 31 | if (this->isLocked()) { |
| 32 | fDiscardableMemory->unlock(); |
| 33 | } |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 34 | SkDELETE(fDiscardableMemory); |
| 35 | SkSafeUnref(fDMFactory); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 36 | SkDELETE(fGenerator); |
| 37 | } |
| 38 | |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 39 | bool SkDiscardablePixelRef::onNewLockPixels(LockRec* rec) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 40 | if (fDiscardableMemory != NULL) { |
| 41 | if (fDiscardableMemory->lock()) { |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 42 | rec->fPixels = fDiscardableMemory->data(); |
| 43 | rec->fColorTable = NULL; |
| 44 | rec->fRowBytes = fRowBytes; |
| 45 | return true; |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 46 | } |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 47 | SkDELETE(fDiscardableMemory); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 48 | fDiscardableMemory = NULL; |
| 49 | } |
skia.committer@gmail.com | 98272d9 | 2014-01-04 07:01:40 +0000 | [diff] [blame] | 50 | |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 51 | const size_t size = this->info().getSafeSize(fRowBytes); |
| 52 | |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 53 | if (fDMFactory != NULL) { |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 54 | fDiscardableMemory = fDMFactory->create(size); |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 55 | } else { |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 56 | fDiscardableMemory = SkDiscardableMemory::Create(size); |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 57 | } |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 58 | if (NULL == fDiscardableMemory) { |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 59 | return false; // Memory allocation failed. |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 60 | } |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 61 | |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 62 | void* pixels = fDiscardableMemory->data(); |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 63 | if (!fGenerator->getPixels(this->info(), pixels, fRowBytes)) { |
reed@google.com | c83a91f | 2013-12-13 13:41:14 +0000 | [diff] [blame] | 64 | fDiscardableMemory->unlock(); |
| 65 | SkDELETE(fDiscardableMemory); |
| 66 | fDiscardableMemory = NULL; |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 67 | return false; |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 68 | } |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 69 | |
| 70 | rec->fPixels = pixels; |
| 71 | rec->fColorTable = NULL; |
| 72 | rec->fRowBytes = fRowBytes; |
| 73 | return true; |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 74 | } |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame^] | 75 | |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 76 | void SkDiscardablePixelRef::onUnlockPixels() { |
reed@google.com | c83a91f | 2013-12-13 13:41:14 +0000 | [diff] [blame] | 77 | fDiscardableMemory->unlock(); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 78 | } |
| 79 | |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 80 | bool SkInstallDiscardablePixelRef(SkImageGenerator* generator, |
| 81 | SkBitmap* dst, |
| 82 | SkDiscardableMemory::Factory* factory) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 83 | SkImageInfo info; |
halcanary@google.com | 3d50ea1 | 2014-01-02 13:15:13 +0000 | [diff] [blame] | 84 | SkAutoTDelete<SkImageGenerator> autoGenerator(generator); |
| 85 | if ((NULL == autoGenerator.get()) |
| 86 | || (!autoGenerator->getInfo(&info)) |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 87 | || (!dst->setConfig(info, 0))) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 88 | return false; |
| 89 | } |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 90 | SkASSERT(dst->config() != SkBitmap::kNo_Config); |
halcanary@google.com | 3d50ea1 | 2014-01-02 13:15:13 +0000 | [diff] [blame] | 91 | if (dst->empty()) { // Use a normal pixelref. |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 92 | return dst->allocPixels(NULL, NULL); |
| 93 | } |
halcanary@google.com | 3d50ea1 | 2014-01-02 13:15:13 +0000 | [diff] [blame] | 94 | SkAutoTUnref<SkDiscardablePixelRef> ref( |
| 95 | SkNEW_ARGS(SkDiscardablePixelRef, |
| 96 | (info, autoGenerator.detach(), dst->rowBytes(), factory))); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 97 | dst->setPixelRef(ref); |
| 98 | return true; |
| 99 | } |