blob: 06e1fe746e148b035eb5bbcaba59e39646d49823 [file] [log] [blame]
scroggo478652e2015-03-25 07:11:02 -07001/*
2 * Copyright 2015 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
mtklein748ca3b2015-01-15 10:56:12 -08008#include "DMSrcSink.h"
9#include "SamplePipeControllers.h"
10#include "SkCommonFlags.h"
scroggof24f2242015-03-03 08:59:20 -080011#include "SkCodec.h"
mtklein748ca3b2015-01-15 10:56:12 -080012#include "SkDocument.h"
joshualitt5f5a8d72015-02-25 14:09:45 -080013#include "SkError.h"
mtklein748ca3b2015-01-15 10:56:12 -080014#include "SkMultiPictureDraw.h"
mtkleinad66f9b2015-02-13 15:11:10 -080015#include "SkNullCanvas.h"
mtklein748ca3b2015-01-15 10:56:12 -080016#include "SkOSFile.h"
mtkleinffa901a2015-03-16 10:38:07 -070017#include "SkPictureData.h"
mtklein748ca3b2015-01-15 10:56:12 -080018#include "SkPictureRecorder.h"
19#include "SkRandom.h"
fmalita2aafe6f2015-02-06 12:51:10 -080020#include "SkSVGCanvas.h"
scroggoa1193e42015-01-21 12:09:53 -080021#include "SkStream.h"
fmalita2aafe6f2015-02-06 12:51:10 -080022#include "SkXMLWriter.h"
mtklein748ca3b2015-01-15 10:56:12 -080023
24namespace DM {
25
mtklein748ca3b2015-01-15 10:56:12 -080026GMSrc::GMSrc(skiagm::GMRegistry::Factory factory) : fFactory(factory) {}
27
28Error GMSrc::draw(SkCanvas* canvas) const {
29 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
30 canvas->concat(gm->getInitialTransform());
31 gm->draw(canvas);
32 return "";
33}
34
35SkISize GMSrc::size() const {
36 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
37 return gm->getISize();
38}
39
40Name GMSrc::name() const {
41 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
42 return gm->getName();
43}
44
45/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
46
scroggo9b77ddd2015-03-19 06:03:39 -070047CodecSrc::CodecSrc(Path path) : fPath(path) {}
mtklein748ca3b2015-01-15 10:56:12 -080048
scroggo9b77ddd2015-03-19 06:03:39 -070049Error CodecSrc::draw(SkCanvas* canvas) const {
50 SkImageInfo canvasInfo;
51 if (NULL == canvas->peekPixels(&canvasInfo, NULL)) {
52 // TODO: Once we implement GPU paths (e.g. JPEG YUV), we should use a deferred decode to
53 // let the GPU handle it.
54 return Error::Nonfatal("No need to test decoding to non-raster backend.");
55 }
56
mtklein75d98fd2015-01-18 07:05:01 -080057 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
mtklein748ca3b2015-01-15 10:56:12 -080058 if (!encoded) {
59 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
60 }
scroggo9b77ddd2015-03-19 06:03:39 -070061
62 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
63 if (!codec) {
64 return SkStringPrintf("Couldn't decode %s.", fPath.c_str());
65 }
66
reed3ef71e32015-03-19 08:31:14 -070067 SkImageInfo decodeInfo = codec->getInfo().makeColorType(canvasInfo.colorType());
scroggo9b77ddd2015-03-19 06:03:39 -070068 if (decodeInfo.alphaType() == kUnpremul_SkAlphaType) {
69 // FIXME: Currently we cannot draw unpremultiplied sources.
70 decodeInfo = decodeInfo.makeAlphaType(kPremul_SkAlphaType);
71 }
72
73 SkBitmap bitmap;
74 if (!bitmap.tryAllocPixels(decodeInfo)) {
75 return SkStringPrintf("Image(%s) is too large (%d x %d)\n", fPath.c_str(),
76 decodeInfo.width(), decodeInfo.height());
77 }
78
79 SkAutoLockPixels alp(bitmap);
80 switch (codec->getPixels(decodeInfo, bitmap.getPixels(), bitmap.rowBytes())) {
81 case SkImageGenerator::kSuccess:
82 // We consider incomplete to be valid, since we should still decode what is
83 // available.
84 case SkImageGenerator::kIncompleteInput:
85 canvas->drawBitmap(bitmap, 0, 0);
86 return "";
87 case SkImageGenerator::kInvalidConversion:
88 return Error::Nonfatal("Incompatible colortype conversion");
89 default:
90 // Everything else is considered a failure.
91 return SkStringPrintf("Couldn't getPixels %s.", fPath.c_str());
92 }
93}
94
95SkISize CodecSrc::size() const {
96 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
97 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
msarett9bde9182015-03-25 05:27:48 -070098 if (NULL != codec) {
99 return codec->getInfo().dimensions();
100 } else {
101 return SkISize::Make(0, 0);
102 }
scroggo9b77ddd2015-03-19 06:03:39 -0700103}
104
105Name CodecSrc::name() const {
106 return SkOSPath::Basename(fPath.c_str());
107}
108
109/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
110
111ImageSrc::ImageSrc(Path path, int divisor) : fPath(path), fDivisor(divisor) {}
112
113Error ImageSrc::draw(SkCanvas* canvas) const {
114 SkImageInfo canvasInfo;
115 if (NULL == canvas->peekPixels(&canvasInfo, NULL)) {
116 // TODO: Instead, use lazy decoding to allow the GPU to handle cases like YUV.
117 return Error::Nonfatal("No need to test decoding to non-raster backend.");
118 }
119
120 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
121 if (!encoded) {
122 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
123 }
124 const SkColorType dstColorType = canvasInfo.colorType();
mtkleinedc93bc2015-01-30 13:22:23 -0800125 if (fDivisor == 0) {
mtklein748ca3b2015-01-15 10:56:12 -0800126 // Decode the full image.
127 SkBitmap bitmap;
scroggo9b77ddd2015-03-19 06:03:39 -0700128 if (!SkImageDecoder::DecodeMemory(encoded->data(), encoded->size(), &bitmap,
129 dstColorType, SkImageDecoder::kDecodePixels_Mode)) {
130 return SkStringPrintf("Couldn't decode %s.", fPath.c_str());
131 }
132 if (kRGB_565_SkColorType == dstColorType && !bitmap.isOpaque()) {
133 // Do not draw a bitmap with alpha to a destination without alpha.
134 return Error::Nonfatal("Uninteresting to decode image with alpha into 565.");
mtklein748ca3b2015-01-15 10:56:12 -0800135 }
mtklein75d98fd2015-01-18 07:05:01 -0800136 encoded.reset((SkData*)NULL); // Might as well drop this when we're done with it.
mtklein748ca3b2015-01-15 10:56:12 -0800137 canvas->drawBitmap(bitmap, 0,0);
138 return "";
139 }
mtkleinedc93bc2015-01-30 13:22:23 -0800140 // Decode subsets. This is a little involved.
scroggoa1193e42015-01-21 12:09:53 -0800141 SkAutoTDelete<SkMemoryStream> stream(new SkMemoryStream(encoded));
142 SkAutoTDelete<SkImageDecoder> decoder(SkImageDecoder::Factory(stream.get()));
mtklein748ca3b2015-01-15 10:56:12 -0800143 if (!decoder) {
144 return SkStringPrintf("Can't find a good decoder for %s.", fPath.c_str());
145 }
scroggoa1193e42015-01-21 12:09:53 -0800146 stream->rewind();
mtklein748ca3b2015-01-15 10:56:12 -0800147 int w,h;
scroggoa1193e42015-01-21 12:09:53 -0800148 if (!decoder->buildTileIndex(stream.detach(), &w, &h) || w*h == 1) {
mtklein4089ef72015-03-05 08:40:28 -0800149 return Error::Nonfatal("Subset decoding not supported.");
mtklein748ca3b2015-01-15 10:56:12 -0800150 }
mtkleinedc93bc2015-01-30 13:22:23 -0800151
152 // Divide the image into subsets that cover the entire image.
153 if (fDivisor > w || fDivisor > h) {
154 return SkStringPrintf("divisor %d is too big for %s with dimensions (%d x %d)",
155 fDivisor, fPath.c_str(), w, h);
156 }
157 const int subsetWidth = w / fDivisor,
158 subsetHeight = h / fDivisor;
159 for (int y = 0; y < h; y += subsetHeight) {
160 for (int x = 0; x < w; x += subsetWidth) {
161 SkBitmap subset;
162 SkIRect rect = SkIRect::MakeXYWH(x, y, subsetWidth, subsetHeight);
163 if (!decoder->decodeSubset(&subset, rect, dstColorType)) {
164 return SkStringPrintf("Could not decode subset (%d, %d, %d, %d).",
165 x, y, x+subsetWidth, y+subsetHeight);
166 }
scroggo56e25dd2015-03-05 11:46:40 -0800167 if (kRGB_565_SkColorType == dstColorType && !subset.isOpaque()) {
168 // Do not draw a bitmap with alpha to a destination without alpha.
169 // This is not an error, but there is nothing interesting to show.
170
171 // This should only happen on the first iteration through the loop.
172 SkASSERT(0 == x && 0 == y);
173
174 return Error::Nonfatal("Uninteresting to decode image with alpha into 565.");
175 }
mtkleinedc93bc2015-01-30 13:22:23 -0800176 canvas->drawBitmap(subset, SkIntToScalar(x), SkIntToScalar(y));
mtklein748ca3b2015-01-15 10:56:12 -0800177 }
mtklein748ca3b2015-01-15 10:56:12 -0800178 }
179 return "";
180}
181
182SkISize ImageSrc::size() const {
mtklein75d98fd2015-01-18 07:05:01 -0800183 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
scroggo9b77ddd2015-03-19 06:03:39 -0700184 SkBitmap bitmap;
185 if (!encoded || !SkImageDecoder::DecodeMemory(encoded->data(),
186 encoded->size(),
187 &bitmap,
188 kUnknown_SkColorType,
189 SkImageDecoder::kDecodeBounds_Mode)) {
190 return SkISize::Make(0,0);
mtklein748ca3b2015-01-15 10:56:12 -0800191 }
scroggo9b77ddd2015-03-19 06:03:39 -0700192 return bitmap.dimensions();
mtklein748ca3b2015-01-15 10:56:12 -0800193}
194
mtklein9264a952015-01-20 10:11:53 -0800195Name ImageSrc::name() const {
mtkleinedc93bc2015-01-30 13:22:23 -0800196 return SkOSPath::Basename(fPath.c_str());
mtklein9264a952015-01-20 10:11:53 -0800197}
mtklein748ca3b2015-01-15 10:56:12 -0800198
199/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
200
mtkleinf4ba3212015-01-28 15:32:24 -0800201static const SkRect kSKPViewport = {0,0, 1000,1000};
202
mtklein8d17a132015-01-30 11:42:31 -0800203SKPSrc::SKPSrc(Path path) : fPath(path) {}
mtklein748ca3b2015-01-15 10:56:12 -0800204
205Error SKPSrc::draw(SkCanvas* canvas) const {
scroggoa1193e42015-01-21 12:09:53 -0800206 SkAutoTDelete<SkStream> stream(SkStream::NewFromFile(fPath.c_str()));
mtklein75d98fd2015-01-18 07:05:01 -0800207 if (!stream) {
mtklein748ca3b2015-01-15 10:56:12 -0800208 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
209 }
halcanary44906c62015-02-23 10:43:48 -0800210 SkAutoTUnref<SkPicture> pic(SkPicture::CreateFromStream(stream));
mtklein75d98fd2015-01-18 07:05:01 -0800211 if (!pic) {
212 return SkStringPrintf("Couldn't decode %s as a picture.", fPath.c_str());
213 }
214 stream.reset((SkStream*)NULL); // Might as well drop this when we're done with it.
mtkleinf4ba3212015-01-28 15:32:24 -0800215 canvas->clipRect(kSKPViewport);
mtklein748ca3b2015-01-15 10:56:12 -0800216 canvas->drawPicture(pic);
217 return "";
218}
219
220SkISize SKPSrc::size() const {
mtkleinffa901a2015-03-16 10:38:07 -0700221 SkAutoTDelete<SkStream> stream(SkStream::NewFromFile(fPath.c_str()));
222 if (!stream) {
223 return SkISize::Make(0,0);
224 }
225 SkPictInfo info;
226 if (!SkPicture::InternalOnly_StreamIsSKP(stream, &info)) {
227 return SkISize::Make(0,0);
228 }
229 SkRect viewport = kSKPViewport;
230 if (!viewport.intersect(info.fCullRect)) {
231 return SkISize::Make(0,0);
232 }
233 return viewport.roundOut().size();
mtklein748ca3b2015-01-15 10:56:12 -0800234}
235
236Name SKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
237
238/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
239
mtkleinad66f9b2015-02-13 15:11:10 -0800240Error NullSink::draw(const Src& src, SkBitmap*, SkWStream*, SkString*) const {
241 SkAutoTDelete<SkCanvas> canvas(SkCreateNullCanvas());
242 return src.draw(canvas);
243}
244
245/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
246
mtkleinb9eb4ac2015-02-02 18:26:03 -0800247DEFINE_bool(gpuStats, false, "Append GPU stats to the log for each GPU task?");
248
mtklein82d28432015-01-15 12:46:02 -0800249GPUSink::GPUSink(GrContextFactory::GLContextType ct,
250 GrGLStandard api,
251 int samples,
252 bool dfText,
253 bool threaded)
mtklein748ca3b2015-01-15 10:56:12 -0800254 : fContextType(ct)
255 , fGpuAPI(api)
256 , fSampleCount(samples)
mtklein82d28432015-01-15 12:46:02 -0800257 , fUseDFText(dfText)
258 , fThreaded(threaded) {}
mtklein748ca3b2015-01-15 10:56:12 -0800259
260int GPUSink::enclave() const {
mtklein55e88b22015-01-21 15:50:13 -0800261 return fThreaded ? kAnyThread_Enclave : kGPU_Enclave;
mtklein748ca3b2015-01-15 10:56:12 -0800262}
263
joshualitt5f5a8d72015-02-25 14:09:45 -0800264void PreAbandonGpuContextErrorHandler(SkError, void*) {}
265
mtkleinb9eb4ac2015-02-02 18:26:03 -0800266Error GPUSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log) const {
mtklein55e88b22015-01-21 15:50:13 -0800267 GrContextFactory factory;
mtkleinf4ba3212015-01-28 15:32:24 -0800268 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -0800269 const SkImageInfo info =
270 SkImageInfo::Make(size.width(), size.height(), kN32_SkColorType, kPremul_SkAlphaType);
271 SkAutoTUnref<SkSurface> surface(
mtklein55e88b22015-01-21 15:50:13 -0800272 NewGpuSurface(&factory, fContextType, fGpuAPI, info, fSampleCount, fUseDFText));
mtklein748ca3b2015-01-15 10:56:12 -0800273 if (!surface) {
274 return "Could not create a surface.";
275 }
joshualitt5f5a8d72015-02-25 14:09:45 -0800276 if (FLAGS_preAbandonGpuContext) {
277 SkSetErrorCallback(&PreAbandonGpuContextErrorHandler, NULL);
278 factory.abandonContexts();
279 }
mtklein748ca3b2015-01-15 10:56:12 -0800280 SkCanvas* canvas = surface->getCanvas();
281 Error err = src.draw(canvas);
282 if (!err.isEmpty()) {
283 return err;
284 }
285 canvas->flush();
mtkleinb9eb4ac2015-02-02 18:26:03 -0800286 if (FLAGS_gpuStats) {
287 canvas->getGrContext()->dumpCacheStats(log);
288 canvas->getGrContext()->dumpGpuStats(log);
289 }
mtklein748ca3b2015-01-15 10:56:12 -0800290 dst->allocPixels(info);
joshualitt5f5a8d72015-02-25 14:09:45 -0800291 canvas->readPixels(dst, 0, 0);
mtklein55e88b22015-01-21 15:50:13 -0800292 if (FLAGS_abandonGpuContext) {
293 factory.abandonContexts();
294 }
mtklein748ca3b2015-01-15 10:56:12 -0800295 return "";
296}
297
298/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
299
halcanary47ef4d52015-03-03 09:13:09 -0800300static Error draw_skdocument(const Src& src, SkDocument* doc, SkWStream* dst) {
301 // Print the given DM:Src to a document, breaking on 8.5x11 pages.
302 SkASSERT(doc);
halcanaryfd4a9932015-01-28 11:45:58 -0800303 int width = src.size().width(),
304 height = src.size().height();
305
306 const int kLetterWidth = 612, // 8.5 * 72
307 kLetterHeight = 792; // 11 * 72
308 const SkRect letter = SkRect::MakeWH(SkIntToScalar(kLetterWidth),
309 SkIntToScalar(kLetterHeight));
310
311 int xPages = ((width - 1) / kLetterWidth) + 1;
312 int yPages = ((height - 1) / kLetterHeight) + 1;
313
314 for (int y = 0; y < yPages; ++y) {
315 for (int x = 0; x < xPages; ++x) {
316 int w = SkTMin(kLetterWidth, width - (x * kLetterWidth));
317 int h = SkTMin(kLetterHeight, height - (y * kLetterHeight));
318 SkCanvas* canvas =
319 doc->beginPage(SkIntToScalar(w), SkIntToScalar(h));
320 canvas->clipRect(letter);
321 canvas->translate(-letter.width() * x, -letter.height() * y);
322 Error err = src.draw(canvas);
323 if (!err.isEmpty()) {
324 return err;
325 }
326 doc->endPage();
327 }
mtklein748ca3b2015-01-15 10:56:12 -0800328 }
mtklein748ca3b2015-01-15 10:56:12 -0800329 doc->close();
halcanaryfd4a9932015-01-28 11:45:58 -0800330 dst->flush();
mtklein748ca3b2015-01-15 10:56:12 -0800331 return "";
332}
333
halcanary47ef4d52015-03-03 09:13:09 -0800334PDFSink::PDFSink() {}
335
336Error PDFSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
337 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(dst));
338 if (!doc) {
339 return "SkDocument::CreatePDF() returned NULL";
340 }
341 return draw_skdocument(src, doc.get(), dst);
342}
343
344/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
345
346XPSSink::XPSSink() {}
347
348Error XPSSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
349 SkAutoTUnref<SkDocument> doc(SkDocument::CreateXPS(dst));
350 if (!doc) {
351 return "SkDocument::CreateXPS() returned NULL";
352 }
353 return draw_skdocument(src, doc.get(), dst);
354}
mtklein748ca3b2015-01-15 10:56:12 -0800355/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
356
mtklein9c3f17d2015-01-28 11:35:18 -0800357SKPSink::SKPSink() {}
358
mtkleinb9eb4ac2015-02-02 18:26:03 -0800359Error SKPSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
mtklein9c3f17d2015-01-28 11:35:18 -0800360 SkSize size;
361 size = src.size();
362 SkPictureRecorder recorder;
363 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
364 if (!err.isEmpty()) {
365 return err;
366 }
367 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
368 pic->serialize(dst);
369 return "";
370}
371
372/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
373
mtklein8a4527e2015-01-31 20:00:58 -0800374SVGSink::SVGSink() {}
375
mtkleinb9eb4ac2015-02-02 18:26:03 -0800376Error SVGSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
fmalita2aafe6f2015-02-06 12:51:10 -0800377 SkAutoTDelete<SkXMLWriter> xmlWriter(SkNEW_ARGS(SkXMLStreamWriter, (dst)));
378 SkAutoTUnref<SkCanvas> canvas(SkSVGCanvas::Create(
379 SkRect::MakeWH(SkIntToScalar(src.size().width()), SkIntToScalar(src.size().height())),
380 xmlWriter));
381 return src.draw(canvas);
mtklein8a4527e2015-01-31 20:00:58 -0800382}
383
384/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
385
mtklein748ca3b2015-01-15 10:56:12 -0800386RasterSink::RasterSink(SkColorType colorType) : fColorType(colorType) {}
387
mtkleinb9eb4ac2015-02-02 18:26:03 -0800388Error RasterSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
mtkleinf4ba3212015-01-28 15:32:24 -0800389 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -0800390 // If there's an appropriate alpha type for this color type, use it, otherwise use premul.
391 SkAlphaType alphaType = kPremul_SkAlphaType;
392 (void)SkColorTypeValidateAlphaType(fColorType, alphaType, &alphaType);
393
394 dst->allocPixels(SkImageInfo::Make(size.width(), size.height(), fColorType, alphaType));
395 dst->eraseColor(SK_ColorTRANSPARENT);
396 SkCanvas canvas(*dst);
397 return src.draw(&canvas);
398}
399
400/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
401
mtkleind603b222015-02-17 11:13:33 -0800402static SkISize auto_compute_translate(SkMatrix* matrix, int srcW, int srcH) {
403 SkRect bounds = SkRect::MakeIWH(srcW, srcH);
404 matrix->mapRect(&bounds);
405 matrix->postTranslate(-bounds.x(), -bounds.y());
406 return SkISize::Make(SkScalarRoundToInt(bounds.width()), SkScalarRoundToInt(bounds.height()));
407}
408
mtklein748ca3b2015-01-15 10:56:12 -0800409ViaMatrix::ViaMatrix(SkMatrix matrix, Sink* sink) : fMatrix(matrix), fSink(sink) {}
410
mtkleinb9eb4ac2015-02-02 18:26:03 -0800411Error ViaMatrix::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -0800412 // We turn our arguments into a Src, then draw that Src into our Sink to fill bitmap or stream.
413 struct ProxySrc : public Src {
mtkleind603b222015-02-17 11:13:33 -0800414 const Src& fSrc;
415 SkMatrix fMatrix;
416 SkISize fSize;
417
418 ProxySrc(const Src& src, SkMatrix matrix) : fSrc(src), fMatrix(matrix) {
419 fSize = auto_compute_translate(&fMatrix, src.size().width(), src.size().height());
420 }
mtklein748ca3b2015-01-15 10:56:12 -0800421
422 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
423 canvas->concat(fMatrix);
424 return fSrc.draw(canvas);
425 }
mtkleind603b222015-02-17 11:13:33 -0800426 SkISize size() const SK_OVERRIDE { return fSize; }
mtklein748ca3b2015-01-15 10:56:12 -0800427 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
428 } proxy(src, fMatrix);
mtkleinb9eb4ac2015-02-02 18:26:03 -0800429 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800430}
431
mtkleind603b222015-02-17 11:13:33 -0800432// Undoes any flip or 90 degree rotate without changing the scale of the bitmap.
433// This should be pixel-preserving.
434ViaUpright::ViaUpright(SkMatrix matrix, Sink* sink) : fMatrix(matrix), fSink(sink) {}
435
436Error ViaUpright::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
437 Error err = fSink->draw(src, bitmap, stream, log);
438 if (!err.isEmpty()) {
439 return err;
440 }
441
442 SkMatrix inverse;
443 if (!fMatrix.rectStaysRect() || !fMatrix.invert(&inverse)) {
444 return "Cannot upright --matrix.";
445 }
446 SkMatrix upright = SkMatrix::I();
447 upright.setScaleX(SkScalarSignAsScalar(inverse.getScaleX()));
448 upright.setScaleY(SkScalarSignAsScalar(inverse.getScaleY()));
449 upright.setSkewX(SkScalarSignAsScalar(inverse.getSkewX()));
450 upright.setSkewY(SkScalarSignAsScalar(inverse.getSkewY()));
451
452 SkBitmap uprighted;
453 SkISize size = auto_compute_translate(&upright, bitmap->width(), bitmap->height());
454 uprighted.allocPixels(bitmap->info().makeWH(size.width(), size.height()));
455
456 SkCanvas canvas(uprighted);
457 canvas.concat(upright);
458 SkPaint paint;
459 paint.setXfermodeMode(SkXfermode::kSrc_Mode);
460 canvas.drawBitmap(*bitmap, 0, 0, &paint);
461
462 *bitmap = uprighted;
463 bitmap->lockPixels();
464 return "";
465}
466
mtklein748ca3b2015-01-15 10:56:12 -0800467/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
468
mtklein7edca212015-01-21 13:18:51 -0800469ViaPipe::ViaPipe(Sink* sink) : fSink(sink) {}
mtklein748ca3b2015-01-15 10:56:12 -0800470
mtkleinb9eb4ac2015-02-02 18:26:03 -0800471Error ViaPipe::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein7edca212015-01-21 13:18:51 -0800472 // We turn ourselves into another Src that draws our argument into bitmap/stream via pipe.
mtklein748ca3b2015-01-15 10:56:12 -0800473 struct ProxySrc : public Src {
474 const Src& fSrc;
mtklein7edca212015-01-21 13:18:51 -0800475 ProxySrc(const Src& src) : fSrc(src) {}
mtklein748ca3b2015-01-15 10:56:12 -0800476
477 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
478 SkISize size = this->size();
mtklein748ca3b2015-01-15 10:56:12 -0800479 PipeController controller(canvas, &SkImageDecoder::DecodeMemory);
480 SkGPipeWriter pipe;
mtklein7edca212015-01-21 13:18:51 -0800481 const uint32_t kFlags = 0; // We mirror SkDeferredCanvas, which doesn't use any flags.
482 return fSrc.draw(pipe.startRecording(&controller, kFlags, size.width(), size.height()));
mtklein748ca3b2015-01-15 10:56:12 -0800483 }
484 SkISize size() const SK_OVERRIDE { return fSrc.size(); }
485 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
mtklein7edca212015-01-21 13:18:51 -0800486 } proxy(src);
mtkleinb9eb4ac2015-02-02 18:26:03 -0800487 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800488}
489
490/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
491
492ViaSerialization::ViaSerialization(Sink* sink) : fSink(sink) {}
493
mtkleinb9eb4ac2015-02-02 18:26:03 -0800494Error ViaSerialization::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log)
495 const {
mtklein748ca3b2015-01-15 10:56:12 -0800496 // Record our Src into a picture.
497 SkSize size;
498 size = src.size();
499 SkPictureRecorder recorder;
500 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
501 if (!err.isEmpty()) {
502 return err;
503 }
504 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
505
506 // Serialize it and then deserialize it.
507 SkDynamicMemoryWStream wStream;
508 pic->serialize(&wStream);
scroggoa1193e42015-01-21 12:09:53 -0800509 SkAutoTDelete<SkStream> rStream(wStream.detachAsStream());
halcanary44906c62015-02-23 10:43:48 -0800510 SkAutoTUnref<SkPicture> deserialized(SkPicture::CreateFromStream(rStream));
mtklein748ca3b2015-01-15 10:56:12 -0800511
512 // Turn that deserialized picture into a Src, draw it into our Sink to fill bitmap or stream.
513 struct ProxySrc : public Src {
514 const SkPicture* fPic;
515 const SkISize fSize;
516 ProxySrc(const SkPicture* pic, SkISize size) : fPic(pic), fSize(size) {}
517
518 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
519 canvas->drawPicture(fPic);
520 return "";
521 }
522 SkISize size() const SK_OVERRIDE { return fSize; }
523 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
524 } proxy(deserialized, src.size());
mtkleinb9eb4ac2015-02-02 18:26:03 -0800525 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800526}
527
528/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
529
530ViaTiles::ViaTiles(int w, int h, SkBBHFactory* factory, Sink* sink)
531 : fW(w)
532 , fH(h)
533 , fFactory(factory)
534 , fSink(sink) {}
535
mtkleinb9eb4ac2015-02-02 18:26:03 -0800536Error ViaTiles::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -0800537 // Record our Src into a picture.
538 SkSize size;
539 size = src.size();
540 SkPictureRecorder recorder;
541 Error err = src.draw(recorder.beginRecording(size.width(), size.height(), fFactory.get()));
542 if (!err.isEmpty()) {
543 return err;
544 }
545 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
546
547 // Turn that picture into a Src that draws into our Sink via tiles + MPD.
548 struct ProxySrc : public Src {
549 const int fW, fH;
550 const SkPicture* fPic;
551 const SkISize fSize;
552 ProxySrc(int w, int h, const SkPicture* pic, SkISize size)
553 : fW(w), fH(h), fPic(pic), fSize(size) {}
554
555 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
556 const int xTiles = (fSize.width() + fW - 1) / fW,
557 yTiles = (fSize.height() + fH - 1) / fH;
558 SkMultiPictureDraw mpd(xTiles*yTiles);
559 SkTDArray<SkSurface*> surfaces;
560 surfaces.setReserve(xTiles*yTiles);
561
562 SkImageInfo info = canvas->imageInfo().makeWH(fW, fH);
563 for (int j = 0; j < yTiles; j++) {
564 for (int i = 0; i < xTiles; i++) {
565 // This lets our ultimate Sink determine the best kind of surface.
566 // E.g., if it's a GpuSink, the surfaces and images are textures.
567 SkSurface* s = canvas->newSurface(info);
568 if (!s) {
569 s = SkSurface::NewRaster(info); // Some canvases can't create surfaces.
570 }
571 surfaces.push(s);
572 SkCanvas* c = s->getCanvas();
573 c->translate(SkIntToScalar(-i * fW),
574 SkIntToScalar(-j * fH)); // Line up the canvas with this tile.
575 mpd.add(c, fPic);
576 }
577 }
578 mpd.draw();
579 for (int j = 0; j < yTiles; j++) {
580 for (int i = 0; i < xTiles; i++) {
581 SkAutoTUnref<SkImage> image(surfaces[i+xTiles*j]->newImageSnapshot());
582 canvas->drawImage(image, SkIntToScalar(i*fW), SkIntToScalar(j*fH));
583 }
584 }
585 surfaces.unrefAll();
586 return "";
587 }
588 SkISize size() const SK_OVERRIDE { return fSize; }
589 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
590 } proxy(fW, fH, pic, src.size());
mtkleinb9eb4ac2015-02-02 18:26:03 -0800591 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800592}
593
594} // namespace DM