blob: 22dd5679e3ec8d8f2486d108081e013c30a58dd8 [file] [log] [blame]
mtklein748ca3b2015-01-15 10:56:12 -08001#include "DMSrcSink.h"
2#include "SamplePipeControllers.h"
3#include "SkCommonFlags.h"
scroggof24f2242015-03-03 08:59:20 -08004#include "SkCodec.h"
mtklein748ca3b2015-01-15 10:56:12 -08005#include "SkDocument.h"
joshualitt5f5a8d72015-02-25 14:09:45 -08006#include "SkError.h"
mtklein748ca3b2015-01-15 10:56:12 -08007#include "SkMultiPictureDraw.h"
mtkleinad66f9b2015-02-13 15:11:10 -08008#include "SkNullCanvas.h"
mtklein748ca3b2015-01-15 10:56:12 -08009#include "SkOSFile.h"
10#include "SkPictureRecorder.h"
11#include "SkRandom.h"
fmalita2aafe6f2015-02-06 12:51:10 -080012#include "SkSVGCanvas.h"
scroggoa1193e42015-01-21 12:09:53 -080013#include "SkStream.h"
fmalita2aafe6f2015-02-06 12:51:10 -080014#include "SkXMLWriter.h"
mtklein748ca3b2015-01-15 10:56:12 -080015
scroggof24f2242015-03-03 08:59:20 -080016DEFINE_bool(codec, false, "Use SkCodec instead of SkImageDecoder");
17
mtklein748ca3b2015-01-15 10:56:12 -080018namespace DM {
19
mtklein748ca3b2015-01-15 10:56:12 -080020GMSrc::GMSrc(skiagm::GMRegistry::Factory factory) : fFactory(factory) {}
21
22Error GMSrc::draw(SkCanvas* canvas) const {
23 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
24 canvas->concat(gm->getInitialTransform());
25 gm->draw(canvas);
26 return "";
27}
28
29SkISize GMSrc::size() const {
30 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
31 return gm->getISize();
32}
33
34Name GMSrc::name() const {
35 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
36 return gm->getName();
37}
38
39/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
40
mtkleinedc93bc2015-01-30 13:22:23 -080041ImageSrc::ImageSrc(Path path, int divisor) : fPath(path), fDivisor(divisor) {}
mtklein748ca3b2015-01-15 10:56:12 -080042
43Error ImageSrc::draw(SkCanvas* canvas) const {
mtklein75d98fd2015-01-18 07:05:01 -080044 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
mtklein748ca3b2015-01-15 10:56:12 -080045 if (!encoded) {
46 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
47 }
mtkleinedc93bc2015-01-30 13:22:23 -080048 const SkColorType dstColorType = canvas->imageInfo().colorType();
49 if (fDivisor == 0) {
mtklein748ca3b2015-01-15 10:56:12 -080050 // Decode the full image.
51 SkBitmap bitmap;
scroggof24f2242015-03-03 08:59:20 -080052 if (FLAGS_codec) {
53 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
54 if (!codec) {
55 return SkStringPrintf("Couldn't decode %s.", fPath.c_str());
56 }
57 SkImageInfo info;
58 if (!codec->getInfo(&info)) {
59 return SkStringPrintf("Couldn't getInfo %s.", fPath.c_str());
60 }
61 info = info.makeColorType(dstColorType);
62 if (info.alphaType() == kUnpremul_SkAlphaType) {
63 // FIXME: Currently we cannot draw unpremultiplied sources.
64 info = info.makeAlphaType(kPremul_SkAlphaType);
65 }
66 if (!bitmap.tryAllocPixels(info)) {
67 return SkStringPrintf("Image(%s) is too large (%d x %d)\n", fPath.c_str(),
68 info.width(), info.height());
69 }
70 SkAutoLockPixels alp(bitmap);
71 const SkImageGenerator::Result result = codec->getPixels(info, bitmap.getPixels(),
72 bitmap.rowBytes());
scroggo56e25dd2015-03-05 11:46:40 -080073 switch (result) {
74 case SkImageGenerator::kSuccess:
75 // We consider incomplete to be valid, since we should still decode what is
76 // available.
77 case SkImageGenerator::kIncompleteInput:
78 break;
79 case SkImageGenerator::kInvalidConversion:
80 return Error::Nonfatal("Incompatible colortype conversion");
81 default:
82 // Everything else is considered a failure.
83 return SkStringPrintf("Couldn't getPixels %s.", fPath.c_str());
scroggof24f2242015-03-03 08:59:20 -080084 }
85 } else {
86 if (!SkImageDecoder::DecodeMemory(encoded->data(), encoded->size(), &bitmap,
87 dstColorType, SkImageDecoder::kDecodePixels_Mode)) {
88 return SkStringPrintf("Couldn't decode %s.", fPath.c_str());
89 }
scroggo56e25dd2015-03-05 11:46:40 -080090 if (kRGB_565_SkColorType == dstColorType && !bitmap.isOpaque()) {
91 // Do not draw a bitmap with alpha to a destination without alpha.
92 return Error::Nonfatal("Uninteresting to decode image with alpha into 565.");
93 }
mtklein748ca3b2015-01-15 10:56:12 -080094 }
mtklein75d98fd2015-01-18 07:05:01 -080095 encoded.reset((SkData*)NULL); // Might as well drop this when we're done with it.
mtklein748ca3b2015-01-15 10:56:12 -080096 canvas->drawBitmap(bitmap, 0,0);
97 return "";
98 }
mtkleinedc93bc2015-01-30 13:22:23 -080099 // Decode subsets. This is a little involved.
scroggoa1193e42015-01-21 12:09:53 -0800100 SkAutoTDelete<SkMemoryStream> stream(new SkMemoryStream(encoded));
101 SkAutoTDelete<SkImageDecoder> decoder(SkImageDecoder::Factory(stream.get()));
mtklein748ca3b2015-01-15 10:56:12 -0800102 if (!decoder) {
103 return SkStringPrintf("Can't find a good decoder for %s.", fPath.c_str());
104 }
scroggoa1193e42015-01-21 12:09:53 -0800105 stream->rewind();
mtklein748ca3b2015-01-15 10:56:12 -0800106 int w,h;
scroggoa1193e42015-01-21 12:09:53 -0800107 if (!decoder->buildTileIndex(stream.detach(), &w, &h) || w*h == 1) {
mtklein4089ef72015-03-05 08:40:28 -0800108 return Error::Nonfatal("Subset decoding not supported.");
mtklein748ca3b2015-01-15 10:56:12 -0800109 }
mtkleinedc93bc2015-01-30 13:22:23 -0800110
111 // Divide the image into subsets that cover the entire image.
112 if (fDivisor > w || fDivisor > h) {
113 return SkStringPrintf("divisor %d is too big for %s with dimensions (%d x %d)",
114 fDivisor, fPath.c_str(), w, h);
115 }
116 const int subsetWidth = w / fDivisor,
117 subsetHeight = h / fDivisor;
118 for (int y = 0; y < h; y += subsetHeight) {
119 for (int x = 0; x < w; x += subsetWidth) {
120 SkBitmap subset;
121 SkIRect rect = SkIRect::MakeXYWH(x, y, subsetWidth, subsetHeight);
122 if (!decoder->decodeSubset(&subset, rect, dstColorType)) {
123 return SkStringPrintf("Could not decode subset (%d, %d, %d, %d).",
124 x, y, x+subsetWidth, y+subsetHeight);
125 }
scroggo56e25dd2015-03-05 11:46:40 -0800126 if (kRGB_565_SkColorType == dstColorType && !subset.isOpaque()) {
127 // Do not draw a bitmap with alpha to a destination without alpha.
128 // This is not an error, but there is nothing interesting to show.
129
130 // This should only happen on the first iteration through the loop.
131 SkASSERT(0 == x && 0 == y);
132
133 return Error::Nonfatal("Uninteresting to decode image with alpha into 565.");
134 }
mtkleinedc93bc2015-01-30 13:22:23 -0800135 canvas->drawBitmap(subset, SkIntToScalar(x), SkIntToScalar(y));
mtklein748ca3b2015-01-15 10:56:12 -0800136 }
mtklein748ca3b2015-01-15 10:56:12 -0800137 }
138 return "";
139}
140
141SkISize ImageSrc::size() const {
mtklein75d98fd2015-01-18 07:05:01 -0800142 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
msarett36758742015-03-16 08:27:53 -0700143 if (FLAGS_codec) {
144 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
145 if (!codec) {
146 return SkISize::Make(0,0);
147 }
148 SkImageInfo info;
149 if (!codec->getInfo(&info)) {
150 return SkISize::Make(0,0);
151 }
152 return info.dimensions();
153 } else {
154 SkBitmap bitmap;
155 if (!encoded || !SkImageDecoder::DecodeMemory(encoded->data(),
156 encoded->size(),
157 &bitmap,
158 kUnknown_SkColorType,
159 SkImageDecoder::kDecodeBounds_Mode)) {
160 return SkISize::Make(0,0);
161 }
162 return bitmap.dimensions();
mtklein748ca3b2015-01-15 10:56:12 -0800163 }
mtklein748ca3b2015-01-15 10:56:12 -0800164}
165
mtklein9264a952015-01-20 10:11:53 -0800166Name ImageSrc::name() const {
mtkleinedc93bc2015-01-30 13:22:23 -0800167 return SkOSPath::Basename(fPath.c_str());
mtklein9264a952015-01-20 10:11:53 -0800168}
mtklein748ca3b2015-01-15 10:56:12 -0800169
170/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
171
mtkleinf4ba3212015-01-28 15:32:24 -0800172static const SkRect kSKPViewport = {0,0, 1000,1000};
173
mtklein8d17a132015-01-30 11:42:31 -0800174SKPSrc::SKPSrc(Path path) : fPath(path) {}
mtklein748ca3b2015-01-15 10:56:12 -0800175
176Error SKPSrc::draw(SkCanvas* canvas) const {
scroggoa1193e42015-01-21 12:09:53 -0800177 SkAutoTDelete<SkStream> stream(SkStream::NewFromFile(fPath.c_str()));
mtklein75d98fd2015-01-18 07:05:01 -0800178 if (!stream) {
mtklein748ca3b2015-01-15 10:56:12 -0800179 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
180 }
halcanary44906c62015-02-23 10:43:48 -0800181 SkAutoTUnref<SkPicture> pic(SkPicture::CreateFromStream(stream));
mtklein75d98fd2015-01-18 07:05:01 -0800182 if (!pic) {
183 return SkStringPrintf("Couldn't decode %s as a picture.", fPath.c_str());
184 }
185 stream.reset((SkStream*)NULL); // Might as well drop this when we're done with it.
mtkleinf4ba3212015-01-28 15:32:24 -0800186 canvas->clipRect(kSKPViewport);
mtklein748ca3b2015-01-15 10:56:12 -0800187 canvas->drawPicture(pic);
188 return "";
189}
190
191SkISize SKPSrc::size() const {
mtkleinf4ba3212015-01-28 15:32:24 -0800192 // This may be unnecessarily large.
193 return kSKPViewport.roundOut().size();
mtklein748ca3b2015-01-15 10:56:12 -0800194}
195
196Name SKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
197
198/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
199
mtkleinad66f9b2015-02-13 15:11:10 -0800200Error NullSink::draw(const Src& src, SkBitmap*, SkWStream*, SkString*) const {
201 SkAutoTDelete<SkCanvas> canvas(SkCreateNullCanvas());
202 return src.draw(canvas);
203}
204
205/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
206
mtkleinb9eb4ac2015-02-02 18:26:03 -0800207DEFINE_bool(gpuStats, false, "Append GPU stats to the log for each GPU task?");
208
mtklein82d28432015-01-15 12:46:02 -0800209GPUSink::GPUSink(GrContextFactory::GLContextType ct,
210 GrGLStandard api,
211 int samples,
212 bool dfText,
213 bool threaded)
mtklein748ca3b2015-01-15 10:56:12 -0800214 : fContextType(ct)
215 , fGpuAPI(api)
216 , fSampleCount(samples)
mtklein82d28432015-01-15 12:46:02 -0800217 , fUseDFText(dfText)
218 , fThreaded(threaded) {}
mtklein748ca3b2015-01-15 10:56:12 -0800219
220int GPUSink::enclave() const {
mtklein55e88b22015-01-21 15:50:13 -0800221 return fThreaded ? kAnyThread_Enclave : kGPU_Enclave;
mtklein748ca3b2015-01-15 10:56:12 -0800222}
223
joshualitt5f5a8d72015-02-25 14:09:45 -0800224void PreAbandonGpuContextErrorHandler(SkError, void*) {}
225
mtkleinb9eb4ac2015-02-02 18:26:03 -0800226Error GPUSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log) const {
mtklein55e88b22015-01-21 15:50:13 -0800227 GrContextFactory factory;
mtkleinf4ba3212015-01-28 15:32:24 -0800228 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -0800229 const SkImageInfo info =
230 SkImageInfo::Make(size.width(), size.height(), kN32_SkColorType, kPremul_SkAlphaType);
231 SkAutoTUnref<SkSurface> surface(
mtklein55e88b22015-01-21 15:50:13 -0800232 NewGpuSurface(&factory, fContextType, fGpuAPI, info, fSampleCount, fUseDFText));
mtklein748ca3b2015-01-15 10:56:12 -0800233 if (!surface) {
234 return "Could not create a surface.";
235 }
joshualitt5f5a8d72015-02-25 14:09:45 -0800236 if (FLAGS_preAbandonGpuContext) {
237 SkSetErrorCallback(&PreAbandonGpuContextErrorHandler, NULL);
238 factory.abandonContexts();
239 }
mtklein748ca3b2015-01-15 10:56:12 -0800240 SkCanvas* canvas = surface->getCanvas();
241 Error err = src.draw(canvas);
242 if (!err.isEmpty()) {
243 return err;
244 }
245 canvas->flush();
mtkleinb9eb4ac2015-02-02 18:26:03 -0800246 if (FLAGS_gpuStats) {
247 canvas->getGrContext()->dumpCacheStats(log);
248 canvas->getGrContext()->dumpGpuStats(log);
249 }
mtklein748ca3b2015-01-15 10:56:12 -0800250 dst->allocPixels(info);
joshualitt5f5a8d72015-02-25 14:09:45 -0800251 canvas->readPixels(dst, 0, 0);
mtklein55e88b22015-01-21 15:50:13 -0800252 if (FLAGS_abandonGpuContext) {
253 factory.abandonContexts();
254 }
mtklein748ca3b2015-01-15 10:56:12 -0800255 return "";
256}
257
258/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
259
halcanary47ef4d52015-03-03 09:13:09 -0800260static Error draw_skdocument(const Src& src, SkDocument* doc, SkWStream* dst) {
261 // Print the given DM:Src to a document, breaking on 8.5x11 pages.
262 SkASSERT(doc);
halcanaryfd4a9932015-01-28 11:45:58 -0800263 int width = src.size().width(),
264 height = src.size().height();
265
266 const int kLetterWidth = 612, // 8.5 * 72
267 kLetterHeight = 792; // 11 * 72
268 const SkRect letter = SkRect::MakeWH(SkIntToScalar(kLetterWidth),
269 SkIntToScalar(kLetterHeight));
270
271 int xPages = ((width - 1) / kLetterWidth) + 1;
272 int yPages = ((height - 1) / kLetterHeight) + 1;
273
274 for (int y = 0; y < yPages; ++y) {
275 for (int x = 0; x < xPages; ++x) {
276 int w = SkTMin(kLetterWidth, width - (x * kLetterWidth));
277 int h = SkTMin(kLetterHeight, height - (y * kLetterHeight));
278 SkCanvas* canvas =
279 doc->beginPage(SkIntToScalar(w), SkIntToScalar(h));
280 canvas->clipRect(letter);
281 canvas->translate(-letter.width() * x, -letter.height() * y);
282 Error err = src.draw(canvas);
283 if (!err.isEmpty()) {
284 return err;
285 }
286 doc->endPage();
287 }
mtklein748ca3b2015-01-15 10:56:12 -0800288 }
mtklein748ca3b2015-01-15 10:56:12 -0800289 doc->close();
halcanaryfd4a9932015-01-28 11:45:58 -0800290 dst->flush();
mtklein748ca3b2015-01-15 10:56:12 -0800291 return "";
292}
293
halcanary47ef4d52015-03-03 09:13:09 -0800294PDFSink::PDFSink() {}
295
296Error PDFSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
297 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(dst));
298 if (!doc) {
299 return "SkDocument::CreatePDF() returned NULL";
300 }
301 return draw_skdocument(src, doc.get(), dst);
302}
303
304/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
305
306XPSSink::XPSSink() {}
307
308Error XPSSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
309 SkAutoTUnref<SkDocument> doc(SkDocument::CreateXPS(dst));
310 if (!doc) {
311 return "SkDocument::CreateXPS() returned NULL";
312 }
313 return draw_skdocument(src, doc.get(), dst);
314}
mtklein748ca3b2015-01-15 10:56:12 -0800315/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
316
mtklein9c3f17d2015-01-28 11:35:18 -0800317SKPSink::SKPSink() {}
318
mtkleinb9eb4ac2015-02-02 18:26:03 -0800319Error SKPSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
mtklein9c3f17d2015-01-28 11:35:18 -0800320 SkSize size;
321 size = src.size();
322 SkPictureRecorder recorder;
323 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
324 if (!err.isEmpty()) {
325 return err;
326 }
327 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
328 pic->serialize(dst);
329 return "";
330}
331
332/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
333
mtklein8a4527e2015-01-31 20:00:58 -0800334SVGSink::SVGSink() {}
335
mtkleinb9eb4ac2015-02-02 18:26:03 -0800336Error SVGSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
fmalita2aafe6f2015-02-06 12:51:10 -0800337 SkAutoTDelete<SkXMLWriter> xmlWriter(SkNEW_ARGS(SkXMLStreamWriter, (dst)));
338 SkAutoTUnref<SkCanvas> canvas(SkSVGCanvas::Create(
339 SkRect::MakeWH(SkIntToScalar(src.size().width()), SkIntToScalar(src.size().height())),
340 xmlWriter));
341 return src.draw(canvas);
mtklein8a4527e2015-01-31 20:00:58 -0800342}
343
344/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
345
mtklein748ca3b2015-01-15 10:56:12 -0800346RasterSink::RasterSink(SkColorType colorType) : fColorType(colorType) {}
347
mtkleinb9eb4ac2015-02-02 18:26:03 -0800348Error RasterSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
mtkleinf4ba3212015-01-28 15:32:24 -0800349 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -0800350 // If there's an appropriate alpha type for this color type, use it, otherwise use premul.
351 SkAlphaType alphaType = kPremul_SkAlphaType;
352 (void)SkColorTypeValidateAlphaType(fColorType, alphaType, &alphaType);
353
354 dst->allocPixels(SkImageInfo::Make(size.width(), size.height(), fColorType, alphaType));
355 dst->eraseColor(SK_ColorTRANSPARENT);
356 SkCanvas canvas(*dst);
357 return src.draw(&canvas);
358}
359
360/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
361
mtkleind603b222015-02-17 11:13:33 -0800362static SkISize auto_compute_translate(SkMatrix* matrix, int srcW, int srcH) {
363 SkRect bounds = SkRect::MakeIWH(srcW, srcH);
364 matrix->mapRect(&bounds);
365 matrix->postTranslate(-bounds.x(), -bounds.y());
366 return SkISize::Make(SkScalarRoundToInt(bounds.width()), SkScalarRoundToInt(bounds.height()));
367}
368
mtklein748ca3b2015-01-15 10:56:12 -0800369ViaMatrix::ViaMatrix(SkMatrix matrix, Sink* sink) : fMatrix(matrix), fSink(sink) {}
370
mtkleinb9eb4ac2015-02-02 18:26:03 -0800371Error ViaMatrix::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -0800372 // We turn our arguments into a Src, then draw that Src into our Sink to fill bitmap or stream.
373 struct ProxySrc : public Src {
mtkleind603b222015-02-17 11:13:33 -0800374 const Src& fSrc;
375 SkMatrix fMatrix;
376 SkISize fSize;
377
378 ProxySrc(const Src& src, SkMatrix matrix) : fSrc(src), fMatrix(matrix) {
379 fSize = auto_compute_translate(&fMatrix, src.size().width(), src.size().height());
380 }
mtklein748ca3b2015-01-15 10:56:12 -0800381
382 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
383 canvas->concat(fMatrix);
384 return fSrc.draw(canvas);
385 }
mtkleind603b222015-02-17 11:13:33 -0800386 SkISize size() const SK_OVERRIDE { return fSize; }
mtklein748ca3b2015-01-15 10:56:12 -0800387 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
388 } proxy(src, fMatrix);
mtkleinb9eb4ac2015-02-02 18:26:03 -0800389 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800390}
391
mtkleind603b222015-02-17 11:13:33 -0800392// Undoes any flip or 90 degree rotate without changing the scale of the bitmap.
393// This should be pixel-preserving.
394ViaUpright::ViaUpright(SkMatrix matrix, Sink* sink) : fMatrix(matrix), fSink(sink) {}
395
396Error ViaUpright::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
397 Error err = fSink->draw(src, bitmap, stream, log);
398 if (!err.isEmpty()) {
399 return err;
400 }
401
402 SkMatrix inverse;
403 if (!fMatrix.rectStaysRect() || !fMatrix.invert(&inverse)) {
404 return "Cannot upright --matrix.";
405 }
406 SkMatrix upright = SkMatrix::I();
407 upright.setScaleX(SkScalarSignAsScalar(inverse.getScaleX()));
408 upright.setScaleY(SkScalarSignAsScalar(inverse.getScaleY()));
409 upright.setSkewX(SkScalarSignAsScalar(inverse.getSkewX()));
410 upright.setSkewY(SkScalarSignAsScalar(inverse.getSkewY()));
411
412 SkBitmap uprighted;
413 SkISize size = auto_compute_translate(&upright, bitmap->width(), bitmap->height());
414 uprighted.allocPixels(bitmap->info().makeWH(size.width(), size.height()));
415
416 SkCanvas canvas(uprighted);
417 canvas.concat(upright);
418 SkPaint paint;
419 paint.setXfermodeMode(SkXfermode::kSrc_Mode);
420 canvas.drawBitmap(*bitmap, 0, 0, &paint);
421
422 *bitmap = uprighted;
423 bitmap->lockPixels();
424 return "";
425}
426
mtklein748ca3b2015-01-15 10:56:12 -0800427/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
428
mtklein7edca212015-01-21 13:18:51 -0800429ViaPipe::ViaPipe(Sink* sink) : fSink(sink) {}
mtklein748ca3b2015-01-15 10:56:12 -0800430
mtkleinb9eb4ac2015-02-02 18:26:03 -0800431Error ViaPipe::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein7edca212015-01-21 13:18:51 -0800432 // We turn ourselves into another Src that draws our argument into bitmap/stream via pipe.
mtklein748ca3b2015-01-15 10:56:12 -0800433 struct ProxySrc : public Src {
434 const Src& fSrc;
mtklein7edca212015-01-21 13:18:51 -0800435 ProxySrc(const Src& src) : fSrc(src) {}
mtklein748ca3b2015-01-15 10:56:12 -0800436
437 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
438 SkISize size = this->size();
mtklein748ca3b2015-01-15 10:56:12 -0800439 PipeController controller(canvas, &SkImageDecoder::DecodeMemory);
440 SkGPipeWriter pipe;
mtklein7edca212015-01-21 13:18:51 -0800441 const uint32_t kFlags = 0; // We mirror SkDeferredCanvas, which doesn't use any flags.
442 return fSrc.draw(pipe.startRecording(&controller, kFlags, size.width(), size.height()));
mtklein748ca3b2015-01-15 10:56:12 -0800443 }
444 SkISize size() const SK_OVERRIDE { return fSrc.size(); }
445 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
mtklein7edca212015-01-21 13:18:51 -0800446 } proxy(src);
mtkleinb9eb4ac2015-02-02 18:26:03 -0800447 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800448}
449
450/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
451
452ViaSerialization::ViaSerialization(Sink* sink) : fSink(sink) {}
453
mtkleinb9eb4ac2015-02-02 18:26:03 -0800454Error ViaSerialization::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log)
455 const {
mtklein748ca3b2015-01-15 10:56:12 -0800456 // Record our Src into a picture.
457 SkSize size;
458 size = src.size();
459 SkPictureRecorder recorder;
460 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
461 if (!err.isEmpty()) {
462 return err;
463 }
464 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
465
466 // Serialize it and then deserialize it.
467 SkDynamicMemoryWStream wStream;
468 pic->serialize(&wStream);
scroggoa1193e42015-01-21 12:09:53 -0800469 SkAutoTDelete<SkStream> rStream(wStream.detachAsStream());
halcanary44906c62015-02-23 10:43:48 -0800470 SkAutoTUnref<SkPicture> deserialized(SkPicture::CreateFromStream(rStream));
mtklein748ca3b2015-01-15 10:56:12 -0800471
472 // Turn that deserialized picture into a Src, draw it into our Sink to fill bitmap or stream.
473 struct ProxySrc : public Src {
474 const SkPicture* fPic;
475 const SkISize fSize;
476 ProxySrc(const SkPicture* pic, SkISize size) : fPic(pic), fSize(size) {}
477
478 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
479 canvas->drawPicture(fPic);
480 return "";
481 }
482 SkISize size() const SK_OVERRIDE { return fSize; }
483 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
484 } proxy(deserialized, src.size());
mtkleinb9eb4ac2015-02-02 18:26:03 -0800485 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800486}
487
488/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
489
490ViaTiles::ViaTiles(int w, int h, SkBBHFactory* factory, Sink* sink)
491 : fW(w)
492 , fH(h)
493 , fFactory(factory)
494 , fSink(sink) {}
495
mtkleinb9eb4ac2015-02-02 18:26:03 -0800496Error ViaTiles::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -0800497 // Record our Src into a picture.
498 SkSize size;
499 size = src.size();
500 SkPictureRecorder recorder;
501 Error err = src.draw(recorder.beginRecording(size.width(), size.height(), fFactory.get()));
502 if (!err.isEmpty()) {
503 return err;
504 }
505 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
506
507 // Turn that picture into a Src that draws into our Sink via tiles + MPD.
508 struct ProxySrc : public Src {
509 const int fW, fH;
510 const SkPicture* fPic;
511 const SkISize fSize;
512 ProxySrc(int w, int h, const SkPicture* pic, SkISize size)
513 : fW(w), fH(h), fPic(pic), fSize(size) {}
514
515 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
516 const int xTiles = (fSize.width() + fW - 1) / fW,
517 yTiles = (fSize.height() + fH - 1) / fH;
518 SkMultiPictureDraw mpd(xTiles*yTiles);
519 SkTDArray<SkSurface*> surfaces;
520 surfaces.setReserve(xTiles*yTiles);
521
522 SkImageInfo info = canvas->imageInfo().makeWH(fW, fH);
523 for (int j = 0; j < yTiles; j++) {
524 for (int i = 0; i < xTiles; i++) {
525 // This lets our ultimate Sink determine the best kind of surface.
526 // E.g., if it's a GpuSink, the surfaces and images are textures.
527 SkSurface* s = canvas->newSurface(info);
528 if (!s) {
529 s = SkSurface::NewRaster(info); // Some canvases can't create surfaces.
530 }
531 surfaces.push(s);
532 SkCanvas* c = s->getCanvas();
533 c->translate(SkIntToScalar(-i * fW),
534 SkIntToScalar(-j * fH)); // Line up the canvas with this tile.
535 mpd.add(c, fPic);
536 }
537 }
538 mpd.draw();
539 for (int j = 0; j < yTiles; j++) {
540 for (int i = 0; i < xTiles; i++) {
541 SkAutoTUnref<SkImage> image(surfaces[i+xTiles*j]->newImageSnapshot());
542 canvas->drawImage(image, SkIntToScalar(i*fW), SkIntToScalar(j*fH));
543 }
544 }
545 surfaces.unrefAll();
546 return "";
547 }
548 SkISize size() const SK_OVERRIDE { return fSize; }
549 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
550 } proxy(fW, fH, pic, src.size());
mtkleinb9eb4ac2015-02-02 18:26:03 -0800551 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800552}
553
554} // namespace DM