djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 1 | |
| 2 | /* |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 3 | * Copyright 2012 Google Inc. |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 4 | * |
| 5 | * Use of this source code is governed by a BSD-style license that can be |
| 6 | * found in the LICENSE file. |
| 7 | */ |
| 8 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 9 | #include "SkWriteBuffer.h" |
junov@chromium.org | ce65f38 | 2012-10-17 19:36:09 +0000 | [diff] [blame] | 10 | #include "SkBitmap.h" |
djsollen | 0b17d6c | 2014-11-13 12:52:35 -0800 | [diff] [blame] | 11 | #include "SkBitmapHeap.h" |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 12 | #include "SkData.h" |
reed@google.com | 672588b | 2014-01-08 15:42:01 +0000 | [diff] [blame] | 13 | #include "SkPixelRef.h" |
mike@reedtribe.org | 227b516 | 2012-08-12 19:25:08 +0000 | [diff] [blame] | 14 | #include "SkPtrRecorder.h" |
scroggo@google.com | 5a7c6be | 2012-10-04 21:46:08 +0000 | [diff] [blame] | 15 | #include "SkStream.h" |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 16 | #include "SkTypeface.h" |
| 17 | |
commit-bot@chromium.org | a2bd2d1 | 2014-01-30 22:16:32 +0000 | [diff] [blame] | 18 | SkWriteBuffer::SkWriteBuffer(uint32_t flags) |
| 19 | : fFlags(flags) |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 20 | , fFactorySet(nullptr) |
| 21 | , fNamedFactorySet(nullptr) |
| 22 | , fBitmapHeap(nullptr) |
| 23 | , fTFSet(nullptr) { |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 24 | } |
| 25 | |
commit-bot@chromium.org | a2bd2d1 | 2014-01-30 22:16:32 +0000 | [diff] [blame] | 26 | SkWriteBuffer::SkWriteBuffer(void* storage, size_t storageSize, uint32_t flags) |
| 27 | : fFlags(flags) |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 28 | , fFactorySet(nullptr) |
| 29 | , fNamedFactorySet(nullptr) |
commit-bot@chromium.org | 1938242 | 2014-01-14 20:51:26 +0000 | [diff] [blame] | 30 | , fWriter(storage, storageSize) |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 31 | , fBitmapHeap(nullptr) |
| 32 | , fTFSet(nullptr) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 33 | } |
| 34 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 35 | SkWriteBuffer::~SkWriteBuffer() { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 36 | SkSafeUnref(fFactorySet); |
| 37 | SkSafeUnref(fNamedFactorySet); |
djsollen@google.com | 21830d9 | 2012-08-07 19:49:41 +0000 | [diff] [blame] | 38 | SkSafeUnref(fBitmapHeap); |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 39 | SkSafeUnref(fTFSet); |
| 40 | } |
| 41 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 42 | void SkWriteBuffer::writeByteArray(const void* data, size_t size) { |
reed@google.com | 6fcd28b | 2014-02-04 16:03:51 +0000 | [diff] [blame] | 43 | fWriter.write32(SkToU32(size)); |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 44 | fWriter.writePad(data, size); |
| 45 | } |
| 46 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 47 | void SkWriteBuffer::writeBool(bool value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 48 | fWriter.writeBool(value); |
| 49 | } |
| 50 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 51 | void SkWriteBuffer::writeFixed(SkFixed value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 52 | fWriter.write32(value); |
| 53 | } |
| 54 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 55 | void SkWriteBuffer::writeScalar(SkScalar value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 56 | fWriter.writeScalar(value); |
| 57 | } |
| 58 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 59 | void SkWriteBuffer::writeScalarArray(const SkScalar* value, uint32_t count) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 60 | fWriter.write32(count); |
| 61 | fWriter.write(value, count * sizeof(SkScalar)); |
| 62 | } |
| 63 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 64 | void SkWriteBuffer::writeInt(int32_t value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 65 | fWriter.write32(value); |
| 66 | } |
| 67 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 68 | void SkWriteBuffer::writeIntArray(const int32_t* value, uint32_t count) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 69 | fWriter.write32(count); |
| 70 | fWriter.write(value, count * sizeof(int32_t)); |
| 71 | } |
| 72 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 73 | void SkWriteBuffer::writeUInt(uint32_t value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 74 | fWriter.write32(value); |
| 75 | } |
| 76 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 77 | void SkWriteBuffer::write32(int32_t value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 78 | fWriter.write32(value); |
| 79 | } |
| 80 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 81 | void SkWriteBuffer::writeString(const char* value) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 82 | fWriter.writeString(value); |
| 83 | } |
| 84 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 85 | void SkWriteBuffer::writeEncodedString(const void* value, size_t byteLength, |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 86 | SkPaint::TextEncoding encoding) { |
| 87 | fWriter.writeInt(encoding); |
reed@google.com | 6fcd28b | 2014-02-04 16:03:51 +0000 | [diff] [blame] | 88 | fWriter.writeInt(SkToU32(byteLength)); |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 89 | fWriter.write(value, byteLength); |
| 90 | } |
| 91 | |
| 92 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 93 | void SkWriteBuffer::writeColor(const SkColor& color) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 94 | fWriter.write32(color); |
| 95 | } |
| 96 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 97 | void SkWriteBuffer::writeColorArray(const SkColor* color, uint32_t count) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 98 | fWriter.write32(count); |
| 99 | fWriter.write(color, count * sizeof(SkColor)); |
| 100 | } |
| 101 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 102 | void SkWriteBuffer::writePoint(const SkPoint& point) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 103 | fWriter.writeScalar(point.fX); |
| 104 | fWriter.writeScalar(point.fY); |
| 105 | } |
| 106 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 107 | void SkWriteBuffer::writePointArray(const SkPoint* point, uint32_t count) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 108 | fWriter.write32(count); |
| 109 | fWriter.write(point, count * sizeof(SkPoint)); |
| 110 | } |
| 111 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 112 | void SkWriteBuffer::writeMatrix(const SkMatrix& matrix) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 113 | fWriter.writeMatrix(matrix); |
| 114 | } |
| 115 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 116 | void SkWriteBuffer::writeIRect(const SkIRect& rect) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 117 | fWriter.write(&rect, sizeof(SkIRect)); |
| 118 | } |
| 119 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 120 | void SkWriteBuffer::writeRect(const SkRect& rect) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 121 | fWriter.writeRect(rect); |
| 122 | } |
| 123 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 124 | void SkWriteBuffer::writeRegion(const SkRegion& region) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 125 | fWriter.writeRegion(region); |
| 126 | } |
| 127 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 128 | void SkWriteBuffer::writePath(const SkPath& path) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 129 | fWriter.writePath(path); |
| 130 | } |
| 131 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 132 | size_t SkWriteBuffer::writeStream(SkStream* stream, size_t length) { |
reed@google.com | 6fcd28b | 2014-02-04 16:03:51 +0000 | [diff] [blame] | 133 | fWriter.write32(SkToU32(length)); |
djsollen@google.com | e782f5e | 2013-01-23 13:40:40 +0000 | [diff] [blame] | 134 | size_t bytesWritten = fWriter.readFromStream(stream, length); |
| 135 | if (bytesWritten < length) { |
| 136 | fWriter.reservePad(length - bytesWritten); |
| 137 | } |
| 138 | return bytesWritten; |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 139 | } |
| 140 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 141 | bool SkWriteBuffer::writeToStream(SkWStream* stream) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 142 | return fWriter.writeToStream(stream); |
| 143 | } |
| 144 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 145 | static void write_encoded_bitmap(SkWriteBuffer* buffer, SkData* data, |
reed@google.com | 672588b | 2014-01-08 15:42:01 +0000 | [diff] [blame] | 146 | const SkIPoint& origin) { |
| 147 | buffer->writeUInt(SkToU32(data->size())); |
| 148 | buffer->getWriter32()->writePad(data->data(), data->size()); |
| 149 | buffer->write32(origin.fX); |
| 150 | buffer->write32(origin.fY); |
| 151 | } |
scroggo@google.com | 1b1bcc3 | 2013-05-21 20:31:23 +0000 | [diff] [blame] | 152 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 153 | void SkWriteBuffer::writeBitmap(const SkBitmap& bitmap) { |
scroggo@google.com | 74b7ffd | 2013-04-30 02:32:41 +0000 | [diff] [blame] | 154 | // Record the width and height. This way if readBitmap fails a dummy bitmap can be drawn at the |
| 155 | // right size. |
| 156 | this->writeInt(bitmap.width()); |
| 157 | this->writeInt(bitmap.height()); |
| 158 | |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 159 | // Record information about the bitmap in one of three ways, in order of priority: |
| 160 | // 1. If there is an SkBitmapHeap, store it in the heap. The client can avoid serializing the |
scroggo@google.com | 74b7ffd | 2013-04-30 02:32:41 +0000 | [diff] [blame] | 161 | // bitmap entirely or serialize it later as desired. A boolean value of true will be written |
| 162 | // to the stream to signify that a heap was used. |
scroggo@google.com | 1b1bcc3 | 2013-05-21 20:31:23 +0000 | [diff] [blame] | 163 | // 2. If there is a function for encoding bitmaps, use it to write an encoded version of the |
| 164 | // bitmap. After writing a boolean value of false, signifying that a heap was not used, write |
| 165 | // the size of the encoded data. A non-zero size signifies that encoded data was written. |
scroggo@google.com | 74b7ffd | 2013-04-30 02:32:41 +0000 | [diff] [blame] | 166 | // 3. Call SkBitmap::flatten. After writing a boolean value of false, signifying that a heap was |
| 167 | // not used, write a zero to signify that the data was not encoded. |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 168 | bool useBitmapHeap = fBitmapHeap != nullptr; |
scroggo@google.com | 74b7ffd | 2013-04-30 02:32:41 +0000 | [diff] [blame] | 169 | // Write a bool: true if the SkBitmapHeap is to be used, in which case the reader must use an |
| 170 | // SkBitmapHeapReader to read the SkBitmap. False if the bitmap was serialized another way. |
| 171 | this->writeBool(useBitmapHeap); |
| 172 | if (useBitmapHeap) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 173 | SkASSERT(nullptr == fPixelSerializer); |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 174 | int32_t slot = fBitmapHeap->insert(bitmap); |
| 175 | fWriter.write32(slot); |
| 176 | // crbug.com/155875 |
| 177 | // The generation ID is not required information. We write it to prevent collisions |
| 178 | // in SkFlatDictionary. It is possible to get a collision when a previously |
| 179 | // unflattened (i.e. stale) instance of a similar flattenable is in the dictionary |
| 180 | // and the instance currently being written is re-using the same slot from the |
| 181 | // bitmap heap. |
| 182 | fWriter.write32(bitmap.getGenerationID()); |
| 183 | return; |
| 184 | } |
reed@google.com | 672588b | 2014-01-08 15:42:01 +0000 | [diff] [blame] | 185 | |
scroggo | 895c43b | 2014-12-11 10:53:58 -0800 | [diff] [blame] | 186 | SkPixelRef* pixelRef = bitmap.pixelRef(); |
| 187 | if (pixelRef) { |
| 188 | // see if the pixelref already has an encoded version |
| 189 | SkAutoDataUnref existingData(pixelRef->refEncodedData()); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 190 | if (existingData.get() != nullptr) { |
scroggo | 895c43b | 2014-12-11 10:53:58 -0800 | [diff] [blame] | 191 | // Assumes that if the client did not set a serializer, they are |
| 192 | // happy to get the encoded data. |
| 193 | if (!fPixelSerializer || fPixelSerializer->useEncodedData(existingData->data(), |
| 194 | existingData->size())) { |
| 195 | write_encoded_bitmap(this, existingData, bitmap.pixelRefOrigin()); |
| 196 | return; |
| 197 | } |
scroggo@google.com | 5a7c6be | 2012-10-04 21:46:08 +0000 | [diff] [blame] | 198 | } |
reed@google.com | 672588b | 2014-01-08 15:42:01 +0000 | [diff] [blame] | 199 | |
scroggo | 895c43b | 2014-12-11 10:53:58 -0800 | [diff] [blame] | 200 | // see if the caller wants to manually encode |
reed | 92fc2ae | 2015-05-22 08:06:21 -0700 | [diff] [blame] | 201 | SkAutoPixmapUnlock result; |
| 202 | if (fPixelSerializer && bitmap.requestLock(&result)) { |
| 203 | const SkPixmap& pmap = result.pixmap(); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 204 | SkASSERT(nullptr == fBitmapHeap); |
reed | 92fc2ae | 2015-05-22 08:06:21 -0700 | [diff] [blame] | 205 | SkAutoDataUnref data(fPixelSerializer->encodePixels(pmap.info(), |
| 206 | pmap.addr(), |
| 207 | pmap.rowBytes())); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 208 | if (data.get() != nullptr) { |
scroggo | 895c43b | 2014-12-11 10:53:58 -0800 | [diff] [blame] | 209 | // if we have to "encode" the bitmap, then we assume there is no |
| 210 | // offset to share, since we are effectively creating a new pixelref |
| 211 | write_encoded_bitmap(this, data, SkIPoint::Make(0, 0)); |
| 212 | return; |
| 213 | } |
reed@google.com | 672588b | 2014-01-08 15:42:01 +0000 | [diff] [blame] | 214 | } |
| 215 | } |
| 216 | |
commit-bot@chromium.org | 968edca | 2014-05-23 13:21:55 +0000 | [diff] [blame] | 217 | this->writeUInt(0); // signal raw pixels |
| 218 | SkBitmap::WriteRawPixels(this, bitmap); |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 219 | } |
| 220 | |
reed | 871872f | 2015-06-22 12:48:26 -0700 | [diff] [blame] | 221 | static bool try_write_encoded(SkWriteBuffer* buffer, SkData* encoded) { |
| 222 | SkPixelSerializer* ps = buffer->getPixelSerializer(); |
| 223 | // Assumes that if the client did not set a serializer, they are |
| 224 | // happy to get the encoded data. |
| 225 | if (!ps || ps->useEncodedData(encoded->data(), encoded->size())) { |
| 226 | write_encoded_bitmap(buffer, encoded, SkIPoint::Make(0, 0)); |
| 227 | return true; |
| 228 | } |
| 229 | return false; |
| 230 | } |
| 231 | |
| 232 | void SkWriteBuffer::writeImage(const SkImage* image) { |
| 233 | this->writeInt(image->width()); |
| 234 | this->writeInt(image->height()); |
| 235 | |
| 236 | SkAutoTUnref<SkData> encoded(image->refEncoded()); |
| 237 | if (encoded && try_write_encoded(this, encoded)) { |
| 238 | return; |
| 239 | } |
| 240 | |
| 241 | encoded.reset(image->encode(SkImageEncoder::kPNG_Type, 100)); |
| 242 | if (encoded && try_write_encoded(this, encoded)) { |
| 243 | return; |
| 244 | } |
| 245 | |
| 246 | this->writeUInt(0); // signal no pixels (in place of the size of the encoded data) |
| 247 | } |
| 248 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 249 | void SkWriteBuffer::writeTypeface(SkTypeface* obj) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 250 | if (nullptr == obj || nullptr == fTFSet) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 251 | fWriter.write32(0); |
| 252 | } else { |
| 253 | fWriter.write32(fTFSet->add(obj)); |
| 254 | } |
| 255 | } |
| 256 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 257 | SkFactorySet* SkWriteBuffer::setFactoryRecorder(SkFactorySet* rec) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 258 | SkRefCnt_SafeAssign(fFactorySet, rec); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 259 | if (fNamedFactorySet != nullptr) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 260 | fNamedFactorySet->unref(); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 261 | fNamedFactorySet = nullptr; |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 262 | } |
| 263 | return rec; |
| 264 | } |
| 265 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 266 | SkNamedFactorySet* SkWriteBuffer::setNamedFactoryRecorder(SkNamedFactorySet* rec) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 267 | SkRefCnt_SafeAssign(fNamedFactorySet, rec); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 268 | if (fFactorySet != nullptr) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 269 | fFactorySet->unref(); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 270 | fFactorySet = nullptr; |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 271 | } |
| 272 | return rec; |
| 273 | } |
| 274 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 275 | SkRefCntSet* SkWriteBuffer::setTypefaceRecorder(SkRefCntSet* rec) { |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 276 | SkRefCnt_SafeAssign(fTFSet, rec); |
| 277 | return rec; |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 278 | } |
| 279 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 280 | void SkWriteBuffer::setBitmapHeap(SkBitmapHeap* bitmapHeap) { |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 281 | SkRefCnt_SafeAssign(fBitmapHeap, bitmapHeap); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 282 | if (bitmapHeap != nullptr) { |
| 283 | SkASSERT(nullptr == fPixelSerializer); |
| 284 | fPixelSerializer.reset(nullptr); |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 285 | } |
| 286 | } |
| 287 | |
scroggo | 895c43b | 2014-12-11 10:53:58 -0800 | [diff] [blame] | 288 | void SkWriteBuffer::setPixelSerializer(SkPixelSerializer* serializer) { |
| 289 | fPixelSerializer.reset(serializer); |
| 290 | if (serializer) { |
| 291 | serializer->ref(); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 292 | SkASSERT(nullptr == fBitmapHeap); |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 293 | SkSafeUnref(fBitmapHeap); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 294 | fBitmapHeap = nullptr; |
scroggo@google.com | 7c9d539 | 2012-12-10 15:40:55 +0000 | [diff] [blame] | 295 | } |
| 296 | } |
| 297 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 298 | void SkWriteBuffer::writeFlattenable(const SkFlattenable* flattenable) { |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 299 | /* |
| 300 | * If we have a factoryset, then the first 32bits tell us... |
| 301 | * 0: failure to write the flattenable |
scroggo@google.com | 0c3e5fe | 2012-08-01 19:34:20 +0000 | [diff] [blame] | 302 | * >0: (1-based) index into the SkFactorySet or SkNamedFactorySet |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 303 | * If we don't have a factoryset, then the first "ptr" is either the |
| 304 | * factory, or null for failure. |
| 305 | * |
| 306 | * The distinction is important, since 0-index is 32bits (always), but a |
| 307 | * 0-functionptr might be 32 or 64 bits. |
| 308 | */ |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 309 | if (nullptr == flattenable) { |
commit-bot@chromium.org | c0b7e10 | 2013-10-23 17:06:21 +0000 | [diff] [blame] | 310 | if (this->isValidating()) { |
| 311 | this->writeString(""); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 312 | } else if (fFactorySet != nullptr || fNamedFactorySet != nullptr) { |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 313 | this->write32(0); |
| 314 | } else { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 315 | this->writeFunctionPtr(nullptr); |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 316 | } |
| 317 | return; |
| 318 | } |
| 319 | |
commit-bot@chromium.org | 210a6aa | 2014-03-07 14:52:09 +0000 | [diff] [blame] | 320 | SkFlattenable::Factory factory = flattenable->getFactory(); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame^] | 321 | SkASSERT(factory != nullptr); |
commit-bot@chromium.org | 210a6aa | 2014-03-07 14:52:09 +0000 | [diff] [blame] | 322 | |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 323 | /* |
| 324 | * We can write 1 of 3 versions of the flattenable: |
| 325 | * 1. function-ptr : this is the fastest for the reader, but assumes that |
| 326 | * the writer and reader are in the same process. |
| 327 | * 2. index into fFactorySet : This is assumes the writer will later |
| 328 | * resolve the function-ptrs into strings for its reader. SkPicture |
| 329 | * does exactly this, by writing a table of names (matching the indices) |
| 330 | * up front in its serialized form. |
scroggo@google.com | 0c3e5fe | 2012-08-01 19:34:20 +0000 | [diff] [blame] | 331 | * 3. index into fNamedFactorySet. fNamedFactorySet will also store the |
| 332 | * name. SkGPipe uses this technique so it can write the name to its |
| 333 | * stream before writing the flattenable. |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 334 | */ |
commit-bot@chromium.org | c0b7e10 | 2013-10-23 17:06:21 +0000 | [diff] [blame] | 335 | if (this->isValidating()) { |
| 336 | this->writeString(flattenable->getTypeName()); |
| 337 | } else if (fFactorySet) { |
scroggo@google.com | 0c3e5fe | 2012-08-01 19:34:20 +0000 | [diff] [blame] | 338 | this->write32(fFactorySet->add(factory)); |
| 339 | } else if (fNamedFactorySet) { |
| 340 | int32_t index = fNamedFactorySet->find(factory); |
| 341 | this->write32(index); |
| 342 | if (0 == index) { |
| 343 | return; |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 344 | } |
| 345 | } else { |
| 346 | this->writeFunctionPtr((void*)factory); |
| 347 | } |
| 348 | |
scroggo@google.com | d8a6cc8 | 2012-09-12 18:53:49 +0000 | [diff] [blame] | 349 | // make room for the size of the flattened object |
djsollen@google.com | c73dd5c | 2012-08-07 15:54:32 +0000 | [diff] [blame] | 350 | (void)fWriter.reserve(sizeof(uint32_t)); |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 351 | // record the current size, so we can subtract after the object writes. |
reed@google.com | 6fcd28b | 2014-02-04 16:03:51 +0000 | [diff] [blame] | 352 | size_t offset = fWriter.bytesWritten(); |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 353 | // now flatten the object |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 354 | flattenable->flatten(*this); |
reed@google.com | 6fcd28b | 2014-02-04 16:03:51 +0000 | [diff] [blame] | 355 | size_t objSize = fWriter.bytesWritten() - offset; |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 356 | // record the obj's size |
commit-bot@chromium.org | 536ac5e | 2014-02-11 22:38:51 +0000 | [diff] [blame] | 357 | fWriter.overwriteTAt(offset - sizeof(uint32_t), SkToU32(objSize)); |
djsollen@google.com | 2b2ede3 | 2012-04-12 13:24:04 +0000 | [diff] [blame] | 358 | } |