blob: 7bbdff73bcb765a31831702db7aa2a9162c8e409 [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"
mtkleinffa901a2015-03-16 10:38:07 -070010#include "SkPictureData.h"
mtklein748ca3b2015-01-15 10:56:12 -080011#include "SkPictureRecorder.h"
12#include "SkRandom.h"
fmalita2aafe6f2015-02-06 12:51:10 -080013#include "SkSVGCanvas.h"
scroggoa1193e42015-01-21 12:09:53 -080014#include "SkStream.h"
fmalita2aafe6f2015-02-06 12:51:10 -080015#include "SkXMLWriter.h"
mtklein748ca3b2015-01-15 10:56:12 -080016
17namespace DM {
18
mtklein748ca3b2015-01-15 10:56:12 -080019GMSrc::GMSrc(skiagm::GMRegistry::Factory factory) : fFactory(factory) {}
20
21Error GMSrc::draw(SkCanvas* canvas) const {
22 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
23 canvas->concat(gm->getInitialTransform());
24 gm->draw(canvas);
25 return "";
26}
27
28SkISize GMSrc::size() const {
29 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
30 return gm->getISize();
31}
32
33Name GMSrc::name() const {
34 SkAutoTDelete<skiagm::GM> gm(fFactory(NULL));
35 return gm->getName();
36}
37
38/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
39
scroggo9b77ddd2015-03-19 06:03:39 -070040CodecSrc::CodecSrc(Path path) : fPath(path) {}
mtklein748ca3b2015-01-15 10:56:12 -080041
scroggo9b77ddd2015-03-19 06:03:39 -070042Error CodecSrc::draw(SkCanvas* canvas) const {
43 SkImageInfo canvasInfo;
44 if (NULL == canvas->peekPixels(&canvasInfo, NULL)) {
45 // TODO: Once we implement GPU paths (e.g. JPEG YUV), we should use a deferred decode to
46 // let the GPU handle it.
47 return Error::Nonfatal("No need to test decoding to non-raster backend.");
48 }
49
mtklein75d98fd2015-01-18 07:05:01 -080050 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
mtklein748ca3b2015-01-15 10:56:12 -080051 if (!encoded) {
52 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
53 }
scroggo9b77ddd2015-03-19 06:03:39 -070054
55 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
56 if (!codec) {
57 return SkStringPrintf("Couldn't decode %s.", fPath.c_str());
58 }
59
60 SkImageInfo decodeInfo;
61 if (!codec->getInfo(&decodeInfo)) {
62 return SkStringPrintf("Couldn't getInfo %s.", fPath.c_str());
63 }
64
65 decodeInfo = decodeInfo.makeColorType(canvasInfo.colorType());
66 if (decodeInfo.alphaType() == kUnpremul_SkAlphaType) {
67 // FIXME: Currently we cannot draw unpremultiplied sources.
68 decodeInfo = decodeInfo.makeAlphaType(kPremul_SkAlphaType);
69 }
70
71 SkBitmap bitmap;
72 if (!bitmap.tryAllocPixels(decodeInfo)) {
73 return SkStringPrintf("Image(%s) is too large (%d x %d)\n", fPath.c_str(),
74 decodeInfo.width(), decodeInfo.height());
75 }
76
77 SkAutoLockPixels alp(bitmap);
78 switch (codec->getPixels(decodeInfo, bitmap.getPixels(), bitmap.rowBytes())) {
79 case SkImageGenerator::kSuccess:
80 // We consider incomplete to be valid, since we should still decode what is
81 // available.
82 case SkImageGenerator::kIncompleteInput:
83 canvas->drawBitmap(bitmap, 0, 0);
84 return "";
85 case SkImageGenerator::kInvalidConversion:
86 return Error::Nonfatal("Incompatible colortype conversion");
87 default:
88 // Everything else is considered a failure.
89 return SkStringPrintf("Couldn't getPixels %s.", fPath.c_str());
90 }
91}
92
93SkISize CodecSrc::size() const {
94 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
95 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
96 SkImageInfo info;
97 if (codec && codec->getInfo(&info)) {
98 return info.dimensions();
99 }
100 return SkISize::Make(0,0);
101}
102
103Name CodecSrc::name() const {
104 return SkOSPath::Basename(fPath.c_str());
105}
106
107/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
108
109ImageSrc::ImageSrc(Path path, int divisor) : fPath(path), fDivisor(divisor) {}
110
111Error ImageSrc::draw(SkCanvas* canvas) const {
112 SkImageInfo canvasInfo;
113 if (NULL == canvas->peekPixels(&canvasInfo, NULL)) {
114 // TODO: Instead, use lazy decoding to allow the GPU to handle cases like YUV.
115 return Error::Nonfatal("No need to test decoding to non-raster backend.");
116 }
117
118 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
119 if (!encoded) {
120 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
121 }
122 const SkColorType dstColorType = canvasInfo.colorType();
mtkleinedc93bc2015-01-30 13:22:23 -0800123 if (fDivisor == 0) {
mtklein748ca3b2015-01-15 10:56:12 -0800124 // Decode the full image.
125 SkBitmap bitmap;
scroggo9b77ddd2015-03-19 06:03:39 -0700126 if (!SkImageDecoder::DecodeMemory(encoded->data(), encoded->size(), &bitmap,
127 dstColorType, SkImageDecoder::kDecodePixels_Mode)) {
128 return SkStringPrintf("Couldn't decode %s.", fPath.c_str());
129 }
130 if (kRGB_565_SkColorType == dstColorType && !bitmap.isOpaque()) {
131 // Do not draw a bitmap with alpha to a destination without alpha.
132 return Error::Nonfatal("Uninteresting to decode image with alpha into 565.");
mtklein748ca3b2015-01-15 10:56:12 -0800133 }
mtklein75d98fd2015-01-18 07:05:01 -0800134 encoded.reset((SkData*)NULL); // Might as well drop this when we're done with it.
mtklein748ca3b2015-01-15 10:56:12 -0800135 canvas->drawBitmap(bitmap, 0,0);
136 return "";
137 }
mtkleinedc93bc2015-01-30 13:22:23 -0800138 // Decode subsets. This is a little involved.
scroggoa1193e42015-01-21 12:09:53 -0800139 SkAutoTDelete<SkMemoryStream> stream(new SkMemoryStream(encoded));
140 SkAutoTDelete<SkImageDecoder> decoder(SkImageDecoder::Factory(stream.get()));
mtklein748ca3b2015-01-15 10:56:12 -0800141 if (!decoder) {
142 return SkStringPrintf("Can't find a good decoder for %s.", fPath.c_str());
143 }
scroggoa1193e42015-01-21 12:09:53 -0800144 stream->rewind();
mtklein748ca3b2015-01-15 10:56:12 -0800145 int w,h;
scroggoa1193e42015-01-21 12:09:53 -0800146 if (!decoder->buildTileIndex(stream.detach(), &w, &h) || w*h == 1) {
mtklein4089ef72015-03-05 08:40:28 -0800147 return Error::Nonfatal("Subset decoding not supported.");
mtklein748ca3b2015-01-15 10:56:12 -0800148 }
mtkleinedc93bc2015-01-30 13:22:23 -0800149
150 // Divide the image into subsets that cover the entire image.
151 if (fDivisor > w || fDivisor > h) {
152 return SkStringPrintf("divisor %d is too big for %s with dimensions (%d x %d)",
153 fDivisor, fPath.c_str(), w, h);
154 }
155 const int subsetWidth = w / fDivisor,
156 subsetHeight = h / fDivisor;
157 for (int y = 0; y < h; y += subsetHeight) {
158 for (int x = 0; x < w; x += subsetWidth) {
159 SkBitmap subset;
160 SkIRect rect = SkIRect::MakeXYWH(x, y, subsetWidth, subsetHeight);
161 if (!decoder->decodeSubset(&subset, rect, dstColorType)) {
162 return SkStringPrintf("Could not decode subset (%d, %d, %d, %d).",
163 x, y, x+subsetWidth, y+subsetHeight);
164 }
scroggo56e25dd2015-03-05 11:46:40 -0800165 if (kRGB_565_SkColorType == dstColorType && !subset.isOpaque()) {
166 // Do not draw a bitmap with alpha to a destination without alpha.
167 // This is not an error, but there is nothing interesting to show.
168
169 // This should only happen on the first iteration through the loop.
170 SkASSERT(0 == x && 0 == y);
171
172 return Error::Nonfatal("Uninteresting to decode image with alpha into 565.");
173 }
mtkleinedc93bc2015-01-30 13:22:23 -0800174 canvas->drawBitmap(subset, SkIntToScalar(x), SkIntToScalar(y));
mtklein748ca3b2015-01-15 10:56:12 -0800175 }
mtklein748ca3b2015-01-15 10:56:12 -0800176 }
177 return "";
178}
179
180SkISize ImageSrc::size() const {
mtklein75d98fd2015-01-18 07:05:01 -0800181 SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
scroggo9b77ddd2015-03-19 06:03:39 -0700182 SkBitmap bitmap;
183 if (!encoded || !SkImageDecoder::DecodeMemory(encoded->data(),
184 encoded->size(),
185 &bitmap,
186 kUnknown_SkColorType,
187 SkImageDecoder::kDecodeBounds_Mode)) {
188 return SkISize::Make(0,0);
mtklein748ca3b2015-01-15 10:56:12 -0800189 }
scroggo9b77ddd2015-03-19 06:03:39 -0700190 return bitmap.dimensions();
mtklein748ca3b2015-01-15 10:56:12 -0800191}
192
mtklein9264a952015-01-20 10:11:53 -0800193Name ImageSrc::name() const {
mtkleinedc93bc2015-01-30 13:22:23 -0800194 return SkOSPath::Basename(fPath.c_str());
mtklein9264a952015-01-20 10:11:53 -0800195}
mtklein748ca3b2015-01-15 10:56:12 -0800196
197/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
198
mtkleinf4ba3212015-01-28 15:32:24 -0800199static const SkRect kSKPViewport = {0,0, 1000,1000};
200
mtklein8d17a132015-01-30 11:42:31 -0800201SKPSrc::SKPSrc(Path path) : fPath(path) {}
mtklein748ca3b2015-01-15 10:56:12 -0800202
203Error SKPSrc::draw(SkCanvas* canvas) const {
scroggoa1193e42015-01-21 12:09:53 -0800204 SkAutoTDelete<SkStream> stream(SkStream::NewFromFile(fPath.c_str()));
mtklein75d98fd2015-01-18 07:05:01 -0800205 if (!stream) {
mtklein748ca3b2015-01-15 10:56:12 -0800206 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
207 }
halcanary44906c62015-02-23 10:43:48 -0800208 SkAutoTUnref<SkPicture> pic(SkPicture::CreateFromStream(stream));
mtklein75d98fd2015-01-18 07:05:01 -0800209 if (!pic) {
210 return SkStringPrintf("Couldn't decode %s as a picture.", fPath.c_str());
211 }
212 stream.reset((SkStream*)NULL); // Might as well drop this when we're done with it.
mtkleinf4ba3212015-01-28 15:32:24 -0800213 canvas->clipRect(kSKPViewport);
mtklein748ca3b2015-01-15 10:56:12 -0800214 canvas->drawPicture(pic);
215 return "";
216}
217
218SkISize SKPSrc::size() const {
mtkleinffa901a2015-03-16 10:38:07 -0700219 SkAutoTDelete<SkStream> stream(SkStream::NewFromFile(fPath.c_str()));
220 if (!stream) {
221 return SkISize::Make(0,0);
222 }
223 SkPictInfo info;
224 if (!SkPicture::InternalOnly_StreamIsSKP(stream, &info)) {
225 return SkISize::Make(0,0);
226 }
227 SkRect viewport = kSKPViewport;
228 if (!viewport.intersect(info.fCullRect)) {
229 return SkISize::Make(0,0);
230 }
231 return viewport.roundOut().size();
mtklein748ca3b2015-01-15 10:56:12 -0800232}
233
234Name SKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
235
236/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
237
mtkleinad66f9b2015-02-13 15:11:10 -0800238Error NullSink::draw(const Src& src, SkBitmap*, SkWStream*, SkString*) const {
239 SkAutoTDelete<SkCanvas> canvas(SkCreateNullCanvas());
240 return src.draw(canvas);
241}
242
243/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
244
mtkleinb9eb4ac2015-02-02 18:26:03 -0800245DEFINE_bool(gpuStats, false, "Append GPU stats to the log for each GPU task?");
246
mtklein82d28432015-01-15 12:46:02 -0800247GPUSink::GPUSink(GrContextFactory::GLContextType ct,
248 GrGLStandard api,
249 int samples,
250 bool dfText,
251 bool threaded)
mtklein748ca3b2015-01-15 10:56:12 -0800252 : fContextType(ct)
253 , fGpuAPI(api)
254 , fSampleCount(samples)
mtklein82d28432015-01-15 12:46:02 -0800255 , fUseDFText(dfText)
256 , fThreaded(threaded) {}
mtklein748ca3b2015-01-15 10:56:12 -0800257
258int GPUSink::enclave() const {
mtklein55e88b22015-01-21 15:50:13 -0800259 return fThreaded ? kAnyThread_Enclave : kGPU_Enclave;
mtklein748ca3b2015-01-15 10:56:12 -0800260}
261
joshualitt5f5a8d72015-02-25 14:09:45 -0800262void PreAbandonGpuContextErrorHandler(SkError, void*) {}
263
mtkleinb9eb4ac2015-02-02 18:26:03 -0800264Error GPUSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log) const {
mtklein55e88b22015-01-21 15:50:13 -0800265 GrContextFactory factory;
mtkleinf4ba3212015-01-28 15:32:24 -0800266 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -0800267 const SkImageInfo info =
268 SkImageInfo::Make(size.width(), size.height(), kN32_SkColorType, kPremul_SkAlphaType);
269 SkAutoTUnref<SkSurface> surface(
mtklein55e88b22015-01-21 15:50:13 -0800270 NewGpuSurface(&factory, fContextType, fGpuAPI, info, fSampleCount, fUseDFText));
mtklein748ca3b2015-01-15 10:56:12 -0800271 if (!surface) {
272 return "Could not create a surface.";
273 }
joshualitt5f5a8d72015-02-25 14:09:45 -0800274 if (FLAGS_preAbandonGpuContext) {
275 SkSetErrorCallback(&PreAbandonGpuContextErrorHandler, NULL);
276 factory.abandonContexts();
277 }
mtklein748ca3b2015-01-15 10:56:12 -0800278 SkCanvas* canvas = surface->getCanvas();
279 Error err = src.draw(canvas);
280 if (!err.isEmpty()) {
281 return err;
282 }
283 canvas->flush();
mtkleinb9eb4ac2015-02-02 18:26:03 -0800284 if (FLAGS_gpuStats) {
285 canvas->getGrContext()->dumpCacheStats(log);
286 canvas->getGrContext()->dumpGpuStats(log);
287 }
mtklein748ca3b2015-01-15 10:56:12 -0800288 dst->allocPixels(info);
joshualitt5f5a8d72015-02-25 14:09:45 -0800289 canvas->readPixels(dst, 0, 0);
mtklein55e88b22015-01-21 15:50:13 -0800290 if (FLAGS_abandonGpuContext) {
291 factory.abandonContexts();
292 }
mtklein748ca3b2015-01-15 10:56:12 -0800293 return "";
294}
295
296/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
297
halcanary47ef4d52015-03-03 09:13:09 -0800298static Error draw_skdocument(const Src& src, SkDocument* doc, SkWStream* dst) {
299 // Print the given DM:Src to a document, breaking on 8.5x11 pages.
300 SkASSERT(doc);
halcanaryfd4a9932015-01-28 11:45:58 -0800301 int width = src.size().width(),
302 height = src.size().height();
303
304 const int kLetterWidth = 612, // 8.5 * 72
305 kLetterHeight = 792; // 11 * 72
306 const SkRect letter = SkRect::MakeWH(SkIntToScalar(kLetterWidth),
307 SkIntToScalar(kLetterHeight));
308
309 int xPages = ((width - 1) / kLetterWidth) + 1;
310 int yPages = ((height - 1) / kLetterHeight) + 1;
311
312 for (int y = 0; y < yPages; ++y) {
313 for (int x = 0; x < xPages; ++x) {
314 int w = SkTMin(kLetterWidth, width - (x * kLetterWidth));
315 int h = SkTMin(kLetterHeight, height - (y * kLetterHeight));
316 SkCanvas* canvas =
317 doc->beginPage(SkIntToScalar(w), SkIntToScalar(h));
318 canvas->clipRect(letter);
319 canvas->translate(-letter.width() * x, -letter.height() * y);
320 Error err = src.draw(canvas);
321 if (!err.isEmpty()) {
322 return err;
323 }
324 doc->endPage();
325 }
mtklein748ca3b2015-01-15 10:56:12 -0800326 }
mtklein748ca3b2015-01-15 10:56:12 -0800327 doc->close();
halcanaryfd4a9932015-01-28 11:45:58 -0800328 dst->flush();
mtklein748ca3b2015-01-15 10:56:12 -0800329 return "";
330}
331
halcanary47ef4d52015-03-03 09:13:09 -0800332PDFSink::PDFSink() {}
333
334Error PDFSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
335 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(dst));
336 if (!doc) {
337 return "SkDocument::CreatePDF() returned NULL";
338 }
339 return draw_skdocument(src, doc.get(), dst);
340}
341
342/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
343
344XPSSink::XPSSink() {}
345
346Error XPSSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
347 SkAutoTUnref<SkDocument> doc(SkDocument::CreateXPS(dst));
348 if (!doc) {
349 return "SkDocument::CreateXPS() returned NULL";
350 }
351 return draw_skdocument(src, doc.get(), dst);
352}
mtklein748ca3b2015-01-15 10:56:12 -0800353/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
354
mtklein9c3f17d2015-01-28 11:35:18 -0800355SKPSink::SKPSink() {}
356
mtkleinb9eb4ac2015-02-02 18:26:03 -0800357Error SKPSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
mtklein9c3f17d2015-01-28 11:35:18 -0800358 SkSize size;
359 size = src.size();
360 SkPictureRecorder recorder;
361 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
362 if (!err.isEmpty()) {
363 return err;
364 }
365 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
366 pic->serialize(dst);
367 return "";
368}
369
370/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
371
mtklein8a4527e2015-01-31 20:00:58 -0800372SVGSink::SVGSink() {}
373
mtkleinb9eb4ac2015-02-02 18:26:03 -0800374Error SVGSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
fmalita2aafe6f2015-02-06 12:51:10 -0800375 SkAutoTDelete<SkXMLWriter> xmlWriter(SkNEW_ARGS(SkXMLStreamWriter, (dst)));
376 SkAutoTUnref<SkCanvas> canvas(SkSVGCanvas::Create(
377 SkRect::MakeWH(SkIntToScalar(src.size().width()), SkIntToScalar(src.size().height())),
378 xmlWriter));
379 return src.draw(canvas);
mtklein8a4527e2015-01-31 20:00:58 -0800380}
381
382/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
383
mtklein748ca3b2015-01-15 10:56:12 -0800384RasterSink::RasterSink(SkColorType colorType) : fColorType(colorType) {}
385
mtkleinb9eb4ac2015-02-02 18:26:03 -0800386Error RasterSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
mtkleinf4ba3212015-01-28 15:32:24 -0800387 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -0800388 // If there's an appropriate alpha type for this color type, use it, otherwise use premul.
389 SkAlphaType alphaType = kPremul_SkAlphaType;
390 (void)SkColorTypeValidateAlphaType(fColorType, alphaType, &alphaType);
391
392 dst->allocPixels(SkImageInfo::Make(size.width(), size.height(), fColorType, alphaType));
393 dst->eraseColor(SK_ColorTRANSPARENT);
394 SkCanvas canvas(*dst);
395 return src.draw(&canvas);
396}
397
398/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
399
mtkleind603b222015-02-17 11:13:33 -0800400static SkISize auto_compute_translate(SkMatrix* matrix, int srcW, int srcH) {
401 SkRect bounds = SkRect::MakeIWH(srcW, srcH);
402 matrix->mapRect(&bounds);
403 matrix->postTranslate(-bounds.x(), -bounds.y());
404 return SkISize::Make(SkScalarRoundToInt(bounds.width()), SkScalarRoundToInt(bounds.height()));
405}
406
mtklein748ca3b2015-01-15 10:56:12 -0800407ViaMatrix::ViaMatrix(SkMatrix matrix, Sink* sink) : fMatrix(matrix), fSink(sink) {}
408
mtkleinb9eb4ac2015-02-02 18:26:03 -0800409Error ViaMatrix::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -0800410 // We turn our arguments into a Src, then draw that Src into our Sink to fill bitmap or stream.
411 struct ProxySrc : public Src {
mtkleind603b222015-02-17 11:13:33 -0800412 const Src& fSrc;
413 SkMatrix fMatrix;
414 SkISize fSize;
415
416 ProxySrc(const Src& src, SkMatrix matrix) : fSrc(src), fMatrix(matrix) {
417 fSize = auto_compute_translate(&fMatrix, src.size().width(), src.size().height());
418 }
mtklein748ca3b2015-01-15 10:56:12 -0800419
420 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
421 canvas->concat(fMatrix);
422 return fSrc.draw(canvas);
423 }
mtkleind603b222015-02-17 11:13:33 -0800424 SkISize size() const SK_OVERRIDE { return fSize; }
mtklein748ca3b2015-01-15 10:56:12 -0800425 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
426 } proxy(src, fMatrix);
mtkleinb9eb4ac2015-02-02 18:26:03 -0800427 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800428}
429
mtkleind603b222015-02-17 11:13:33 -0800430// Undoes any flip or 90 degree rotate without changing the scale of the bitmap.
431// This should be pixel-preserving.
432ViaUpright::ViaUpright(SkMatrix matrix, Sink* sink) : fMatrix(matrix), fSink(sink) {}
433
434Error ViaUpright::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
435 Error err = fSink->draw(src, bitmap, stream, log);
436 if (!err.isEmpty()) {
437 return err;
438 }
439
440 SkMatrix inverse;
441 if (!fMatrix.rectStaysRect() || !fMatrix.invert(&inverse)) {
442 return "Cannot upright --matrix.";
443 }
444 SkMatrix upright = SkMatrix::I();
445 upright.setScaleX(SkScalarSignAsScalar(inverse.getScaleX()));
446 upright.setScaleY(SkScalarSignAsScalar(inverse.getScaleY()));
447 upright.setSkewX(SkScalarSignAsScalar(inverse.getSkewX()));
448 upright.setSkewY(SkScalarSignAsScalar(inverse.getSkewY()));
449
450 SkBitmap uprighted;
451 SkISize size = auto_compute_translate(&upright, bitmap->width(), bitmap->height());
452 uprighted.allocPixels(bitmap->info().makeWH(size.width(), size.height()));
453
454 SkCanvas canvas(uprighted);
455 canvas.concat(upright);
456 SkPaint paint;
457 paint.setXfermodeMode(SkXfermode::kSrc_Mode);
458 canvas.drawBitmap(*bitmap, 0, 0, &paint);
459
460 *bitmap = uprighted;
461 bitmap->lockPixels();
462 return "";
463}
464
mtklein748ca3b2015-01-15 10:56:12 -0800465/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
466
mtklein7edca212015-01-21 13:18:51 -0800467ViaPipe::ViaPipe(Sink* sink) : fSink(sink) {}
mtklein748ca3b2015-01-15 10:56:12 -0800468
mtkleinb9eb4ac2015-02-02 18:26:03 -0800469Error ViaPipe::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein7edca212015-01-21 13:18:51 -0800470 // We turn ourselves into another Src that draws our argument into bitmap/stream via pipe.
mtklein748ca3b2015-01-15 10:56:12 -0800471 struct ProxySrc : public Src {
472 const Src& fSrc;
mtklein7edca212015-01-21 13:18:51 -0800473 ProxySrc(const Src& src) : fSrc(src) {}
mtklein748ca3b2015-01-15 10:56:12 -0800474
475 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
476 SkISize size = this->size();
mtklein748ca3b2015-01-15 10:56:12 -0800477 PipeController controller(canvas, &SkImageDecoder::DecodeMemory);
478 SkGPipeWriter pipe;
mtklein7edca212015-01-21 13:18:51 -0800479 const uint32_t kFlags = 0; // We mirror SkDeferredCanvas, which doesn't use any flags.
480 return fSrc.draw(pipe.startRecording(&controller, kFlags, size.width(), size.height()));
mtklein748ca3b2015-01-15 10:56:12 -0800481 }
482 SkISize size() const SK_OVERRIDE { return fSrc.size(); }
483 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
mtklein7edca212015-01-21 13:18:51 -0800484 } proxy(src);
mtkleinb9eb4ac2015-02-02 18:26:03 -0800485 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800486}
487
488/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
489
490ViaSerialization::ViaSerialization(Sink* sink) : fSink(sink) {}
491
mtkleinb9eb4ac2015-02-02 18:26:03 -0800492Error ViaSerialization::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log)
493 const {
mtklein748ca3b2015-01-15 10:56:12 -0800494 // Record our Src into a picture.
495 SkSize size;
496 size = src.size();
497 SkPictureRecorder recorder;
498 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
499 if (!err.isEmpty()) {
500 return err;
501 }
502 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
503
504 // Serialize it and then deserialize it.
505 SkDynamicMemoryWStream wStream;
506 pic->serialize(&wStream);
scroggoa1193e42015-01-21 12:09:53 -0800507 SkAutoTDelete<SkStream> rStream(wStream.detachAsStream());
halcanary44906c62015-02-23 10:43:48 -0800508 SkAutoTUnref<SkPicture> deserialized(SkPicture::CreateFromStream(rStream));
mtklein748ca3b2015-01-15 10:56:12 -0800509
510 // Turn that deserialized picture into a Src, draw it into our Sink to fill bitmap or stream.
511 struct ProxySrc : public Src {
512 const SkPicture* fPic;
513 const SkISize fSize;
514 ProxySrc(const SkPicture* pic, SkISize size) : fPic(pic), fSize(size) {}
515
516 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
517 canvas->drawPicture(fPic);
518 return "";
519 }
520 SkISize size() const SK_OVERRIDE { return fSize; }
521 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
522 } proxy(deserialized, src.size());
mtkleinb9eb4ac2015-02-02 18:26:03 -0800523 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800524}
525
526/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
527
528ViaTiles::ViaTiles(int w, int h, SkBBHFactory* factory, Sink* sink)
529 : fW(w)
530 , fH(h)
531 , fFactory(factory)
532 , fSink(sink) {}
533
mtkleinb9eb4ac2015-02-02 18:26:03 -0800534Error ViaTiles::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -0800535 // Record our Src into a picture.
536 SkSize size;
537 size = src.size();
538 SkPictureRecorder recorder;
539 Error err = src.draw(recorder.beginRecording(size.width(), size.height(), fFactory.get()));
540 if (!err.isEmpty()) {
541 return err;
542 }
543 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
544
545 // Turn that picture into a Src that draws into our Sink via tiles + MPD.
546 struct ProxySrc : public Src {
547 const int fW, fH;
548 const SkPicture* fPic;
549 const SkISize fSize;
550 ProxySrc(int w, int h, const SkPicture* pic, SkISize size)
551 : fW(w), fH(h), fPic(pic), fSize(size) {}
552
553 Error draw(SkCanvas* canvas) const SK_OVERRIDE {
554 const int xTiles = (fSize.width() + fW - 1) / fW,
555 yTiles = (fSize.height() + fH - 1) / fH;
556 SkMultiPictureDraw mpd(xTiles*yTiles);
557 SkTDArray<SkSurface*> surfaces;
558 surfaces.setReserve(xTiles*yTiles);
559
560 SkImageInfo info = canvas->imageInfo().makeWH(fW, fH);
561 for (int j = 0; j < yTiles; j++) {
562 for (int i = 0; i < xTiles; i++) {
563 // This lets our ultimate Sink determine the best kind of surface.
564 // E.g., if it's a GpuSink, the surfaces and images are textures.
565 SkSurface* s = canvas->newSurface(info);
566 if (!s) {
567 s = SkSurface::NewRaster(info); // Some canvases can't create surfaces.
568 }
569 surfaces.push(s);
570 SkCanvas* c = s->getCanvas();
571 c->translate(SkIntToScalar(-i * fW),
572 SkIntToScalar(-j * fH)); // Line up the canvas with this tile.
573 mpd.add(c, fPic);
574 }
575 }
576 mpd.draw();
577 for (int j = 0; j < yTiles; j++) {
578 for (int i = 0; i < xTiles; i++) {
579 SkAutoTUnref<SkImage> image(surfaces[i+xTiles*j]->newImageSnapshot());
580 canvas->drawImage(image, SkIntToScalar(i*fW), SkIntToScalar(j*fH));
581 }
582 }
583 surfaces.unrefAll();
584 return "";
585 }
586 SkISize size() const SK_OVERRIDE { return fSize; }
587 Name name() const SK_OVERRIDE { sk_throw(); return ""; } // No one should be calling this.
588 } proxy(fW, fH, pic, src.size());
mtkleinb9eb4ac2015-02-02 18:26:03 -0800589 return fSink->draw(proxy, bitmap, stream, log);
mtklein748ca3b2015-01-15 10:56:12 -0800590}
591
592} // namespace DM