OLD | NEW |
1 | 1 |
2 /* | 2 /* |
3 * Copyright 2010 Google Inc. | 3 * Copyright 2010 Google Inc. |
4 * | 4 * |
5 * Use of this source code is governed by a BSD-style license that can be | 5 * Use of this source code is governed by a BSD-style license that can be |
6 * found in the LICENSE file. | 6 * found in the LICENSE file. |
7 */ | 7 */ |
8 | 8 |
9 | 9 |
10 #include "SkData.h" | 10 #include "SkData.h" |
(...skipping 12 matching lines...) Expand all Loading... |
23 this->setData(stream); | 23 this->setData(stream); |
24 } | 24 } |
25 | 25 |
26 SkPDFStream::SkPDFStream(SkData* data) : fState(kUnused_State) { | 26 SkPDFStream::SkPDFStream(SkData* data) : fState(kUnused_State) { |
27 this->setData(data); | 27 this->setData(data); |
28 } | 28 } |
29 | 29 |
30 SkPDFStream::SkPDFStream(const SkPDFStream& pdfStream) | 30 SkPDFStream::SkPDFStream(const SkPDFStream& pdfStream) |
31 : SkPDFDict(), | 31 : SkPDFDict(), |
32 fState(kUnused_State) { | 32 fState(kUnused_State) { |
33 this->setData(pdfStream.fData.get()); | 33 this->setData(pdfStream.fDataStream.get()); |
34 bool removeLength = true; | 34 bool removeLength = true; |
35 // Don't uncompress an already compressed stream, but we could. | 35 // Don't uncompress an already compressed stream, but we could. |
36 if (pdfStream.fState == kCompressed_State) { | 36 if (pdfStream.fState == kCompressed_State) { |
37 fState = kCompressed_State; | 37 fState = kCompressed_State; |
38 removeLength = false; | 38 removeLength = false; |
39 } | 39 } |
40 this->mergeFrom(pdfStream); | 40 this->mergeFrom(pdfStream); |
41 if (removeLength) { | 41 if (removeLength) { |
42 this->remove("Length"); | 42 this->remove("Length"); |
43 } | 43 } |
44 } | 44 } |
45 | 45 |
46 SkPDFStream::~SkPDFStream() {} | 46 SkPDFStream::~SkPDFStream() {} |
47 | 47 |
48 void SkPDFStream::emitObject(SkWStream* stream, SkPDFCatalog* catalog, | 48 void SkPDFStream::emitObject(SkWStream* stream, SkPDFCatalog* catalog, |
49 bool indirect) { | 49 bool indirect) { |
50 if (indirect) { | 50 if (indirect) { |
51 return emitIndirectObject(stream, catalog); | 51 return emitIndirectObject(stream, catalog); |
52 } | 52 } |
53 SkAutoMutexAcquire lock(fMutex); // multiple threads could be calling emit | 53 SkAutoMutexAcquire lock(fMutex); // multiple threads could be calling emit |
54 if (!this->populate(catalog)) { | 54 if (!this->populate(catalog)) { |
55 return fSubstitute->emitObject(stream, catalog, indirect); | 55 return fSubstitute->emitObject(stream, catalog, indirect); |
56 } | 56 } |
57 | 57 |
58 this->INHERITED::emitObject(stream, catalog, false); | 58 this->INHERITED::emitObject(stream, catalog, false); |
59 stream->writeText(" stream\n"); | 59 stream->writeText(" stream\n"); |
60 if (fData.get()) { | 60 stream->writeStream(fDataStream.get(), fDataStream->getLength()); |
61 stream->write(fData->data(), fData->size()); | 61 SkAssertResult(fDataStream->rewind()); |
62 } | |
63 stream->writeText("\nendstream"); | 62 stream->writeText("\nendstream"); |
64 } | 63 } |
65 | 64 |
66 size_t SkPDFStream::getOutputSize(SkPDFCatalog* catalog, bool indirect) { | 65 size_t SkPDFStream::getOutputSize(SkPDFCatalog* catalog, bool indirect) { |
67 if (indirect) { | 66 if (indirect) { |
68 return getIndirectOutputSize(catalog); | 67 return getIndirectOutputSize(catalog); |
69 } | 68 } |
70 SkAutoMutexAcquire lock(fMutex); // multiple threads could be calling emit | 69 SkAutoMutexAcquire lock(fMutex); // multiple threads could be calling emit |
71 if (!this->populate(catalog)) { | 70 if (!this->populate(catalog)) { |
72 return fSubstitute->getOutputSize(catalog, indirect); | 71 return fSubstitute->getOutputSize(catalog, indirect); |
73 } | 72 } |
74 | 73 |
75 return this->INHERITED::getOutputSize(catalog, false) + | 74 return this->INHERITED::getOutputSize(catalog, false) + |
76 strlen(" stream\n\nendstream") + this->dataSize(); | 75 strlen(" stream\n\nendstream") + this->dataSize(); |
77 } | 76 } |
78 | 77 |
79 SkPDFStream::SkPDFStream() : fState(kUnused_State) {} | 78 SkPDFStream::SkPDFStream() : fState(kUnused_State) {} |
80 | 79 |
81 void SkPDFStream::setData(SkData* data) { | 80 void SkPDFStream::setData(SkData* data) { |
82 fData.reset(SkSafeRef(data)); | 81 fMemoryStream.setData(data); |
| 82 if (&fMemoryStream != fDataStream.get()) { |
| 83 fDataStream.reset(SkRef(&fMemoryStream)); |
| 84 } |
83 } | 85 } |
84 | 86 |
85 void SkPDFStream::setData(SkStream* stream) { | 87 void SkPDFStream::setData(SkStream* stream) { |
86 // Code assumes that the stream starts at the beginning and is rewindable. | 88 // Code assumes that the stream starts at the beginning and is rewindable. |
| 89 if (&fMemoryStream == fDataStream.get()) { |
| 90 SkASSERT(&fMemoryStream != stream); |
| 91 fMemoryStream.setData(NULL); |
| 92 } |
| 93 SkASSERT(0 == fMemoryStream.getLength()); |
87 if (stream) { | 94 if (stream) { |
88 SkASSERT(stream->getPosition() == 0); | 95 // SkStreamRewindableFromSkStream will try stream->duplicate(). |
89 fData.reset(SkCopyStreamToData(stream)); | 96 fDataStream.reset(SkStreamRewindableFromSkStream(stream)); |
90 SkAssertResult(stream->rewind()); | 97 SkASSERT(fDataStream.get()); |
91 } else { | 98 } else { |
92 fData.reset(NULL); | 99 fDataStream.reset(SkRef(&fMemoryStream)); |
93 } | 100 } |
94 } | 101 } |
95 | 102 |
96 size_t SkPDFStream::dataSize() const { | 103 size_t SkPDFStream::dataSize() const { |
97 return fData.get() ? fData->size() : 0; | 104 SkASSERT(fDataStream->hasLength()); |
| 105 return fDataStream->getLength(); |
98 } | 106 } |
99 | 107 |
100 bool SkPDFStream::populate(SkPDFCatalog* catalog) { | 108 bool SkPDFStream::populate(SkPDFCatalog* catalog) { |
101 if (fState == kUnused_State) { | 109 if (fState == kUnused_State) { |
102 if (!skip_compression(catalog) && SkFlate::HaveFlate()) { | 110 if (!skip_compression(catalog) && SkFlate::HaveFlate()) { |
103 SkDynamicMemoryWStream compressedData; | 111 SkDynamicMemoryWStream compressedData; |
104 | 112 |
105 SkAssertResult(SkFlate::Deflate(fData.get(), &compressedData)); | 113 SkAssertResult( |
| 114 SkFlate::Deflate(fDataStream.get(), &compressedData)); |
| 115 SkAssertResult(fDataStream->rewind()); |
106 if (compressedData.getOffset() < this->dataSize()) { | 116 if (compressedData.getOffset() < this->dataSize()) { |
107 fData.reset(compressedData.copyToData()); | 117 this->setData(compressedData.detachAsStream()); |
108 insertName("Filter", "FlateDecode"); | 118 insertName("Filter", "FlateDecode"); |
109 } | 119 } |
110 fState = kCompressed_State; | 120 fState = kCompressed_State; |
111 } else { | 121 } else { |
112 fState = kNoCompression_State; | 122 fState = kNoCompression_State; |
113 } | 123 } |
114 insertInt("Length", this->dataSize()); | 124 insertInt("Length", this->dataSize()); |
115 } else if (fState == kNoCompression_State && !skip_compression(catalog) && | 125 } else if (fState == kNoCompression_State && !skip_compression(catalog) && |
116 SkFlate::HaveFlate()) { | 126 SkFlate::HaveFlate()) { |
117 if (!fSubstitute.get()) { | 127 if (!fSubstitute.get()) { |
118 fSubstitute.reset(new SkPDFStream(*this)); | 128 fSubstitute.reset(new SkPDFStream(*this)); |
119 catalog->setSubstitute(this, fSubstitute.get()); | 129 catalog->setSubstitute(this, fSubstitute.get()); |
120 } | 130 } |
121 return false; | 131 return false; |
122 } | 132 } |
123 return true; | 133 return true; |
124 } | 134 } |
OLD | NEW |