| |
| /* |
| * Copyright 2008 The Android Open Source Project |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| |
| #ifndef SkWriter32_DEFINED |
| #define SkWriter32_DEFINED |
| |
| #include "SkMatrix.h" |
| #include "SkPath.h" |
| #include "SkPoint.h" |
| #include "SkRRect.h" |
| #include "SkRect.h" |
| #include "SkRegion.h" |
| #include "SkScalar.h" |
| #include "SkStream.h" |
| #include "SkTDArray.h" |
| #include "SkTypes.h" |
| |
| class SkWriter32 : SkNoncopyable { |
| public: |
| /** |
| * The caller can specify an initial block of storage, which the caller manages. |
| * |
| * SkWriter32 will try to back reserve and write calls with this external storage until the |
| * first time an allocation doesn't fit. From then it will use dynamically allocated storage. |
| * This used to be optional behavior, but pipe now relies on it. |
| */ |
| SkWriter32(void* external = NULL, size_t externalBytes = 0) { |
| this->reset(external, externalBytes); |
| } |
| |
| // return the current offset (will always be a multiple of 4) |
| size_t bytesWritten() const { return fCount * 4; } |
| |
| SK_ATTR_DEPRECATED("use bytesWritten") |
| size_t size() const { return this->bytesWritten(); } |
| |
| void reset(void* external = NULL, size_t externalBytes = 0) { |
| SkASSERT(SkIsAlign4((uintptr_t)external)); |
| SkASSERT(SkIsAlign4(externalBytes)); |
| fExternal = (uint32_t*)external; |
| fExternalLimit = externalBytes/4; |
| fCount = 0; |
| fInternal.rewind(); |
| } |
| |
| // If all data written is contiguous, then this returns a pointer to it, otherwise NULL. |
| // This will work if we've only written to the externally supplied block of storage, or if we've |
| // only written to our internal dynamic storage, but will fail if we have written into both. |
| const uint32_t* contiguousArray() const { |
| if (this->externalCount() == 0) { |
| return fInternal.begin(); |
| } else if (fInternal.isEmpty()) { |
| return fExternal; |
| } |
| return NULL; |
| } |
| |
| // size MUST be multiple of 4 |
| uint32_t* reserve(size_t size) { |
| SkASSERT(SkAlign4(size) == size); |
| const int count = size/4; |
| |
| uint32_t* p; |
| // Once we start writing to fInternal, we never write to fExternal again. |
| // This simplifies tracking what data is where. |
| if (fInternal.isEmpty() && this->externalCount() + count <= fExternalLimit) { |
| p = fExternal + fCount; |
| } else { |
| p = fInternal.append(count); |
| } |
| |
| fCount += count; |
| return p; |
| } |
| |
| // return the address of the 4byte int at the specified offset (which must |
| // be a multiple of 4. This does not allocate any new space, so the returned |
| // address is only valid for 1 int. |
| uint32_t* peek32(size_t offset) { |
| SkASSERT(SkAlign4(offset) == offset); |
| const int count = offset/4; |
| SkASSERT(count < fCount); |
| |
| if (count < this->externalCount()) { |
| return fExternal + count; |
| } |
| return &fInternal[count - this->externalCount()]; |
| } |
| |
| bool writeBool(bool value) { |
| this->write32(value); |
| return value; |
| } |
| |
| void writeInt(int32_t value) { |
| this->write32(value); |
| } |
| |
| void write8(int32_t value) { |
| *(int32_t*)this->reserve(sizeof(value)) = value & 0xFF; |
| } |
| |
| void write16(int32_t value) { |
| *(int32_t*)this->reserve(sizeof(value)) = value & 0xFFFF; |
| } |
| |
| void write32(int32_t value) { |
| *(int32_t*)this->reserve(sizeof(value)) = value; |
| } |
| |
| void writePtr(void* value) { |
| *(void**)this->reserve(sizeof(value)) = value; |
| } |
| |
| void writeScalar(SkScalar value) { |
| *(SkScalar*)this->reserve(sizeof(value)) = value; |
| } |
| |
| void writePoint(const SkPoint& pt) { |
| *(SkPoint*)this->reserve(sizeof(pt)) = pt; |
| } |
| |
| void writeRect(const SkRect& rect) { |
| *(SkRect*)this->reserve(sizeof(rect)) = rect; |
| } |
| |
| void writeIRect(const SkIRect& rect) { |
| *(SkIRect*)this->reserve(sizeof(rect)) = rect; |
| } |
| |
| void writeRRect(const SkRRect& rrect) { |
| rrect.writeToMemory(this->reserve(SkRRect::kSizeInMemory)); |
| } |
| |
| void writePath(const SkPath& path) { |
| size_t size = path.writeToMemory(NULL); |
| SkASSERT(SkAlign4(size) == size); |
| path.writeToMemory(this->reserve(size)); |
| } |
| |
| void writeMatrix(const SkMatrix& matrix) { |
| size_t size = matrix.writeToMemory(NULL); |
| SkASSERT(SkAlign4(size) == size); |
| matrix.writeToMemory(this->reserve(size)); |
| } |
| |
| void writeRegion(const SkRegion& rgn) { |
| size_t size = rgn.writeToMemory(NULL); |
| SkASSERT(SkAlign4(size) == size); |
| rgn.writeToMemory(this->reserve(size)); |
| } |
| |
| // write count bytes (must be a multiple of 4) |
| void writeMul4(const void* values, size_t size) { |
| this->write(values, size); |
| } |
| |
| /** |
| * Write size bytes from values. size must be a multiple of 4, though |
| * values need not be 4-byte aligned. |
| */ |
| void write(const void* values, size_t size) { |
| SkASSERT(SkAlign4(size) == size); |
| // TODO: If we're going to spill from fExternal to fInternal, we might want to fill |
| // fExternal as much as possible before writing to fInternal. |
| memcpy(this->reserve(size), values, size); |
| } |
| |
| /** |
| * Reserve size bytes. Does not need to be 4 byte aligned. The remaining space (if any) will be |
| * filled in with zeroes. |
| */ |
| uint32_t* reservePad(size_t size) { |
| uint32_t* p = this->reserve(SkAlign4(size)); |
| uint8_t* tail = (uint8_t*)p + size; |
| switch (SkAlign4(size) - size) { |
| default: SkDEBUGFAIL("SkAlign4(x) - x should always be 0, 1, 2, or 3."); |
| case 3: *tail++ = 0x00; // fallthrough is intentional |
| case 2: *tail++ = 0x00; // fallthrough is intentional |
| case 1: *tail++ = 0x00; |
| case 0: ;/*nothing to do*/ |
| } |
| return p; |
| } |
| |
| /** |
| * Write size bytes from src, and pad to 4 byte alignment with zeroes. |
| */ |
| void writePad(const void* src, size_t size) { |
| memcpy(this->reservePad(size), src, size); |
| } |
| |
| /** |
| * Writes a string to the writer, which can be retrieved with |
| * SkReader32::readString(). |
| * The length can be specified, or if -1 is passed, it will be computed by |
| * calling strlen(). The length must be < max size_t. |
| * |
| * If you write NULL, it will be read as "". |
| */ |
| void writeString(const char* str, size_t len = (size_t)-1); |
| |
| /** |
| * Computes the size (aligned to multiple of 4) need to write the string |
| * in a call to writeString(). If the length is not specified, it will be |
| * computed by calling strlen(). |
| */ |
| static size_t WriteStringSize(const char* str, size_t len = (size_t)-1); |
| |
| /** |
| * Move the cursor back to offset bytes from the beginning. |
| * This has the same restrictions as peek32: offset must be <= size() and |
| * offset must be a multiple of 4. |
| */ |
| void rewindToOffset(size_t offset) { |
| SkASSERT(SkAlign4(offset) == offset); |
| const int count = offset/4; |
| if (count < this->externalCount()) { |
| fInternal.setCount(0); |
| } else { |
| fInternal.setCount(count - this->externalCount()); |
| } |
| fCount = count; |
| } |
| |
| // copy into a single buffer (allocated by caller). Must be at least size() |
| void flatten(void* dst) const { |
| const size_t externalBytes = this->externalCount()*4; |
| memcpy(dst, fExternal, externalBytes); |
| dst = (uint8_t*)dst + externalBytes; |
| memcpy(dst, fInternal.begin(), fInternal.bytes()); |
| } |
| |
| bool writeToStream(SkWStream* stream) const { |
| return stream->write(fExternal, this->externalCount()*4) |
| && stream->write(fInternal.begin(), fInternal.bytes()); |
| } |
| |
| // read from the stream, and write up to length bytes. Return the actual |
| // number of bytes written. |
| size_t readFromStream(SkStream* stream, size_t length) { |
| return stream->read(this->reservePad(length), length); |
| } |
| |
| private: |
| // Number of uint32_t written into fExternal. <= fExternalLimit. |
| int externalCount() const { return fCount - fInternal.count(); } |
| |
| int fCount; // Total number of uint32_t written. |
| int fExternalLimit; // Number of uint32_t we can write to fExternal. |
| uint32_t* fExternal; // Unmanaged memory block. |
| SkTDArray<uint32_t> fInternal; // Managed memory block. |
| }; |
| |
| /** |
| * Helper class to allocated SIZE bytes as part of the writer, and to provide |
| * that storage to the constructor as its initial storage buffer. |
| * |
| * This wrapper ensures proper alignment rules are met for the storage. |
| */ |
| template <size_t SIZE> class SkSWriter32 : public SkWriter32 { |
| public: |
| SkSWriter32() : SkWriter32(fData.fStorage, SIZE) {} |
| |
| private: |
| union { |
| void* fPtrAlignment; |
| double fDoubleAlignment; |
| char fStorage[SIZE]; |
| } fData; |
| }; |
| |
| #endif |