OLD | NEW |
| (Empty) |
1 #include "SkImageDecoder.h" | |
2 #include "SkColor.h" | |
3 #include "SkColorPriv.h" | |
4 #include "SkDither.h" | |
5 #include "SkMath.h" | |
6 #include "SkStream.h" | |
7 #include "SkTemplates.h" | |
8 #include "SkUtils.h" | |
9 | |
10 extern void ValidateHeap(); | |
11 | |
12 class SkPVJPEGImageDecoder : public SkImageDecoder { | |
13 protected: | |
14 virtual bool onDecode(SkStream* stream, SkBitmap* bm, | |
15 SkBitmap::Config pref, Mode); | |
16 | |
17 private: | |
18 enum { | |
19 STORAGE_SIZE = 8 * 1024 | |
20 }; | |
21 char fStorage[STORAGE_SIZE]; | |
22 }; | |
23 | |
24 SkImageDecoder* SkImageDecoder_PVJPEG_Factory(SkStream* stream) | |
25 { | |
26 return SkNEW(SkPVJPEGImageDecoder); | |
27 } | |
28 | |
29 #include "pvjpgdecoderinterface.h" | |
30 #include "pvjpgdecoder_factory.h" | |
31 | |
32 class AutoPVDelete { | |
33 public: | |
34 AutoPVDelete(PVJpgDecoderInterface* codec) : fCodec(codec) {} | |
35 ~AutoPVDelete() { | |
36 fCodec->Reset(); | |
37 PVJpgDecoderFactory::DeletePVJpgDecoder(fCodec); | |
38 } | |
39 private: | |
40 PVJpgDecoderInterface* fCodec; | |
41 }; | |
42 | |
43 class MyObserver : public MPVJpegDecObserver { | |
44 public: | |
45 MyObserver() : fCount(0) {} | |
46 ~MyObserver() { | |
47 if (fCount != 0) { | |
48 SkDebugf("--- pvjpeg left %d allocations\n", fCount); | |
49 } | |
50 } | |
51 | |
52 virtual void allocateBuffer(uint8* &buffer, int32 buffersize) { | |
53 ++fCount; | |
54 // we double the allocation to work around bug when height is odd | |
55 buffer = (uint8*)sk_malloc_throw(buffersize << 1); | |
56 SkDebugf("--- pvjpeg alloc [%d] %d addr=%p\n", fCount, buffersize, buff
er); | |
57 } | |
58 | |
59 virtual void deallocateBuffer(uint8 *buffer) { | |
60 SkDebugf("--- pvjpeg free [%d] addr=%p\n", fCount, buffer); | |
61 --fCount; | |
62 sk_free(buffer); | |
63 } | |
64 | |
65 private: | |
66 int fCount; | |
67 }; | |
68 | |
69 static void check_status(TPvJpgDecStatus status) { | |
70 if (TPVJPGDEC_SUCCESS != status) { | |
71 SkDEBUGF(("--- pvjpeg status %d\n", status)); | |
72 } | |
73 } | |
74 | |
75 static bool getFrame(PVJpgDecoderInterface* codec, SkBitmap* bitmap, | |
76 SkBitmap::Config prefConfig, SkImageDecoder::Mode mode) { | |
77 TPvJpgDecInfo info; | |
78 TPvJpgDecStatus status = codec->GetInfo(&info); | |
79 if (status != TPVJPGDEC_SUCCESS) | |
80 return false; | |
81 | |
82 int width = info.iWidth[0]; | |
83 int height = info.iHeight[0]; | |
84 | |
85 bitmap->setConfig(SkBitmap::kRGB_565_Config, width, height); | |
86 bitmap->setIsOpaque(true); | |
87 | |
88 if (SkImageDecoder::kDecodeBounds_Mode == mode) { | |
89 return true; | |
90 } | |
91 | |
92 SkASSERT(info.iNumComponent == 3); | |
93 | |
94 TPvJpgDecOutputFmt format; | |
95 format.iColorFormat = TPV_COLORFMT_RGB16; | |
96 format.iCropped.topLeftX = 0; | |
97 format.iCropped.topLeftY = 0; | |
98 format.iCropped.bottomRightX = width - 1; | |
99 format.iCropped.bottomRightY = height - 1; | |
100 format.iOutputPitch = bitmap->rowBytes() >> 1; | |
101 status = codec->SetOutput(&format); | |
102 if (status != TPVJPGDEC_SUCCESS) { | |
103 SkDebugf("--- PV SetOutput failed %d\n", status); | |
104 return false; | |
105 } | |
106 | |
107 TPvJpgDecFrame frame; | |
108 uint8* ptrs[3]; | |
109 int32 widths[3], heights[3]; | |
110 bzero(ptrs, sizeof(ptrs)); | |
111 frame.ptr = ptrs; | |
112 frame.iWidth = widths; | |
113 frame.iHeight = heights; | |
114 | |
115 status = codec->GetFrame(&frame); | |
116 if (status != TPVJPGDEC_SUCCESS) { | |
117 SkDebugf("--- PV GetFrame failed %d\n", status); | |
118 return false; | |
119 } | |
120 | |
121 bitmap->allocPixels(); | |
122 memcpy(bitmap->getPixels(), ptrs[0], bitmap->getSize()); | |
123 return true; | |
124 } | |
125 | |
126 class OsclCleanupper { | |
127 public: | |
128 OsclCleanupper() { | |
129 OsclBase::Init(); | |
130 OsclErrorTrap::Init(); | |
131 OsclMem::Init(); | |
132 } | |
133 ~OsclCleanupper() { | |
134 OsclMem::Cleanup(); | |
135 OsclErrorTrap::Cleanup(); | |
136 OsclBase::Cleanup(); | |
137 } | |
138 }; | |
139 | |
140 bool SkPVJPEGImageDecoder::onDecode(SkStream* stream, SkBitmap* decodedBitmap, | |
141 SkBitmap::Config prefConfig, Mode mode) | |
142 { | |
143 // do I need this guy? | |
144 OsclCleanupper oc; | |
145 | |
146 PVJpgDecoderInterface* codec = PVJpgDecoderFactory::CreatePVJpgDecoder(); | |
147 TPvJpgDecStatus status = codec->Init(); | |
148 check_status(status); | |
149 | |
150 MyObserver observer; // must create before autopvdelete | |
151 AutoPVDelete ad(codec); | |
152 | |
153 status = codec->SetObserver(&observer); | |
154 check_status(status); | |
155 | |
156 char* storage = fStorage; | |
157 int32 bytesInStorage = 0; | |
158 for (;;) | |
159 { | |
160 int32 bytesRead = stream->read(storage + bytesInStorage, | |
161 STORAGE_SIZE - bytesInStorage); | |
162 if (bytesRead <= 0) { | |
163 SkDEBUGF(("SkPVJPEGImageDecoder: stream read returned %d\n", bytesRe
ad)); | |
164 return false; | |
165 } | |
166 | |
167 // update bytesInStorage to account for the read() | |
168 bytesInStorage += bytesRead; | |
169 SkASSERT(bytesInStorage <= STORAGE_SIZE); | |
170 | |
171 // now call Decode to eat some of the bytes | |
172 int32 consumed = bytesInStorage; | |
173 status = codec->Decode((uint8*)storage, &consumed); | |
174 | |
175 SkASSERT(bytesInStorage >= consumed); | |
176 bytesInStorage -= consumed; | |
177 // now bytesInStorage is the remaining unread bytes | |
178 if (bytesInStorage > 0) { // slide the leftovers to the beginning | |
179 SkASSERT(storage == fStorage); | |
180 SkASSERT(consumed >= 0 && bytesInStorage >= 0); | |
181 SkASSERT((size_t)(consumed + bytesInStorage) <= sizeof(fStorage)); | |
182 SkASSERT(sizeof(fStorage) == STORAGE_SIZE); | |
183 // SkDebugf("-- memmov srcOffset=%d, numBytes=%d\n", consumed, bytesI
nStorage); | |
184 memmove(storage, storage + consumed, bytesInStorage); | |
185 } | |
186 | |
187 switch (status) { | |
188 case TPVJPGDEC_SUCCESS: | |
189 SkDEBUGF(("SkPVJPEGImageDecoder::Decode returned success?\n");) | |
190 return false; | |
191 case TPVJPGDEC_FRAME_READY: | |
192 case TPVJPGDEC_DONE: | |
193 return getFrame(codec, decodedBitmap, prefConfig, mode); | |
194 case TPVJPGDEC_FAIL: | |
195 case TPVJPGDEC_INVALID_MEMORY: | |
196 case TPVJPGDEC_INVALID_PARAMS: | |
197 case TPVJPGDEC_NO_IMAGE_DATA: | |
198 SkDEBUGF(("SkPVJPEGImageDecoder: failed to decode err=%d\n", status)
;) | |
199 return false; | |
200 case TPVJPGDEC_WAITING_FOR_INPUT: | |
201 break; // loop around and eat more from the stream | |
202 } | |
203 } | |
204 return false; | |
205 } | |
206 | |
OLD | NEW |