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 | |
| 12 | SkDiscardablePixelRef::SkDiscardablePixelRef(SkImageGenerator* generator, |
| 13 | const SkImageInfo& info, |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 14 | size_t size, |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 15 | size_t rowBytes, |
| 16 | SkDiscardableMemory::Factory* fact) |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 17 | : fGenerator(generator) |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 18 | , fDMFactory(fact) |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 19 | , fInfo(info) |
| 20 | , fSize(size) |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 21 | , fRowBytes(rowBytes) |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 22 | , fDiscardableMemory(NULL) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 23 | SkASSERT(fGenerator != NULL); |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 24 | SkASSERT(fSize > 0); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 25 | SkASSERT(fRowBytes > 0); |
commit-bot@chromium.org | dd59799 | 2013-12-02 22:32:54 +0000 | [diff] [blame] | 26 | // The SkImageGenerator contract requires fGenerator to always |
| 27 | // decode the same image on each call to getPixels(). |
| 28 | this->setImmutable(); |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 29 | SkSafeRef(fDMFactory); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 30 | } |
| 31 | |
| 32 | SkDiscardablePixelRef::~SkDiscardablePixelRef() { |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 33 | SkDELETE(fDiscardableMemory); |
| 34 | SkSafeUnref(fDMFactory); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 35 | SkDELETE(fGenerator); |
| 36 | } |
| 37 | |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 38 | void* SkDiscardablePixelRef::onLockPixels(SkColorTable**) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 39 | if (fDiscardableMemory != NULL) { |
| 40 | if (fDiscardableMemory->lock()) { |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 41 | return fDiscardableMemory->data(); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 42 | } |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 43 | SkDELETE(fDiscardableMemory); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 44 | fDiscardableMemory = NULL; |
| 45 | } |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 46 | if (fDMFactory != NULL) { |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 47 | fDiscardableMemory = fDMFactory->create(fSize); |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 48 | } else { |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 49 | fDiscardableMemory = SkDiscardableMemory::Create(fSize); |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 50 | } |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 51 | if (NULL == fDiscardableMemory) { |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 52 | return NULL; // Memory allocation failed. |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 53 | } |
| 54 | void* pixels = fDiscardableMemory->data(); |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 55 | if (!fGenerator->getPixels(fInfo, pixels, fRowBytes)) { |
| 56 | return NULL; // TODO(halcanary) Find out correct thing to do. |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 57 | } |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 58 | return pixels; |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 59 | } |
| 60 | void SkDiscardablePixelRef::onUnlockPixels() { |
reed@google.com | 772d852 | 2013-12-12 16:27:12 +0000 | [diff] [blame^] | 61 | fDiscardableMemory->unlock(); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 62 | } |
| 63 | |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 64 | bool SkInstallDiscardablePixelRef(SkImageGenerator* generator, |
| 65 | SkBitmap* dst, |
| 66 | SkDiscardableMemory::Factory* factory) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 67 | SkImageInfo info; |
| 68 | SkASSERT(generator != NULL); |
| 69 | if ((NULL == generator) |
| 70 | || (!generator->getInfo(&info)) |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 71 | || (!dst->setConfig(info, 0))) { |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 72 | SkDELETE(generator); |
| 73 | return false; |
| 74 | } |
halcanary@google.com | edd370f | 2013-12-10 21:11:12 +0000 | [diff] [blame] | 75 | SkASSERT(dst->config() != SkBitmap::kNo_Config); |
| 76 | if (dst->empty()) { // Use a normal pixelref. |
| 77 | SkDELETE(generator); // Do not need this anymore. |
| 78 | return dst->allocPixels(NULL, NULL); |
| 79 | } |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 80 | SkAutoTUnref<SkDiscardablePixelRef> ref(SkNEW_ARGS(SkDiscardablePixelRef, |
| 81 | (generator, info, |
reed@google.com | 398337b | 2013-12-11 21:22:39 +0000 | [diff] [blame] | 82 | dst->getSize(), |
halcanary@google.com | 2c7c7ee | 2013-12-05 18:31:42 +0000 | [diff] [blame] | 83 | dst->rowBytes(), |
| 84 | factory))); |
halcanary@google.com | ad04eb4 | 2013-11-21 15:32:08 +0000 | [diff] [blame] | 85 | dst->setPixelRef(ref); |
| 86 | return true; |
| 87 | } |