blob: ab8727470c67ab805260fc974a0b0287d6ddf3e4 [file] [log] [blame]
keyar@chromium.orgb3fb7c12012-08-20 21:02:49 +00001/*
2 * Copyright 2012 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
keyar@chromium.org451bb9f2012-07-26 17:27:57 +00008#include "PictureRenderer.h"
scroggo@google.com58b4ead2012-08-31 16:15:22 +00009#include "picture_utils.h"
keyar@chromium.org451bb9f2012-07-26 17:27:57 +000010#include "SamplePipeControllers.h"
11#include "SkCanvas.h"
12#include "SkDevice.h"
13#include "SkGPipe.h"
scroggo@google.com58b4ead2012-08-31 16:15:22 +000014#if SK_SUPPORT_GPU
15#include "SkGpuDevice.h"
16#endif
17#include "SkGraphics.h"
18#include "SkImageEncoder.h"
caryclark@google.coma3622372012-11-06 21:26:13 +000019#include "SkMaskFilter.h"
keyar@chromium.orgea826952012-08-23 15:24:13 +000020#include "SkMatrix.h"
keyar@chromium.org451bb9f2012-07-26 17:27:57 +000021#include "SkPicture.h"
junov@chromium.org9313ca42012-11-02 18:11:49 +000022#include "SkRTree.h"
keyar@chromium.orgea826952012-08-23 15:24:13 +000023#include "SkScalar.h"
scroggo@google.coma9e3a362012-11-07 17:52:48 +000024#include "SkStream.h"
keyar@chromium.org9299ede2012-08-21 19:05:08 +000025#include "SkString.h"
scroggo@google.com58b4ead2012-08-31 16:15:22 +000026#include "SkTemplates.h"
junov@chromium.org3cb834b2012-12-13 16:39:53 +000027#include "SkTileGridPicture.h"
keyar@chromium.org451bb9f2012-07-26 17:27:57 +000028#include "SkTDArray.h"
scroggo@google.com58b4ead2012-08-31 16:15:22 +000029#include "SkThreadUtils.h"
keyar@chromium.org451bb9f2012-07-26 17:27:57 +000030#include "SkTypes.h"
reed@google.comfe7b1ed2012-11-29 21:00:39 +000031#include "SkData.h"
32#include "SkPictureUtils.h"
keyar@chromium.org4ea96c52012-08-20 15:03:29 +000033
keyar@chromium.org451bb9f2012-07-26 17:27:57 +000034namespace sk_tools {
35
36enum {
37 kDefaultTileWidth = 256,
38 kDefaultTileHeight = 256
39};
40
keyar@chromium.org9d696c02012-08-07 17:11:33 +000041void PictureRenderer::init(SkPicture* pict) {
keyar@chromium.org78a35c52012-08-20 15:03:44 +000042 SkASSERT(NULL == fPicture);
43 SkASSERT(NULL == fCanvas.get());
44 if (fPicture != NULL || NULL != fCanvas.get()) {
keyar@chromium.org9d696c02012-08-07 17:11:33 +000045 return;
46 }
47
48 SkASSERT(pict != NULL);
keyar@chromium.org78a35c52012-08-20 15:03:44 +000049 if (NULL == pict) {
keyar@chromium.org9d696c02012-08-07 17:11:33 +000050 return;
51 }
52
53 fPicture = pict;
junov@chromium.org9313ca42012-11-02 18:11:49 +000054 fPicture->ref();
keyar@chromium.orga474ce32012-08-20 15:03:57 +000055 fCanvas.reset(this->setupCanvas());
56}
57
caryclark@google.coma3622372012-11-06 21:26:13 +000058class FlagsDrawFilter : public SkDrawFilter {
59public:
60 FlagsDrawFilter(PictureRenderer::DrawFilterFlags* flags) :
61 fFlags(flags) {}
62
reed@google.com971aca72012-11-26 20:26:54 +000063 virtual bool filter(SkPaint* paint, Type t) {
caryclark@google.coma3622372012-11-06 21:26:13 +000064 paint->setFlags(paint->getFlags() & ~fFlags[t] & SkPaint::kAllFlags);
reed@google.com457d8a72012-12-18 18:20:44 +000065 if (PictureRenderer::kBlur_DrawFilterFlag & fFlags[t]) {
caryclark@google.coma3622372012-11-06 21:26:13 +000066 SkMaskFilter* maskFilter = paint->getMaskFilter();
67 SkMaskFilter::BlurInfo blurInfo;
68 if (maskFilter && maskFilter->asABlur(&blurInfo)) {
reed@google.com457d8a72012-12-18 18:20:44 +000069 paint->setMaskFilter(NULL);
caryclark@google.coma3622372012-11-06 21:26:13 +000070 }
71 }
72 if (PictureRenderer::kHinting_DrawFilterFlag & fFlags[t]) {
73 paint->setHinting(SkPaint::kNo_Hinting);
74 } else if (PictureRenderer::kSlightHinting_DrawFilterFlag & fFlags[t]) {
75 paint->setHinting(SkPaint::kSlight_Hinting);
76 }
reed@google.com971aca72012-11-26 20:26:54 +000077 return true;
caryclark@google.coma3622372012-11-06 21:26:13 +000078 }
79
80private:
81 PictureRenderer::DrawFilterFlags* fFlags;
82};
83
scroggo@google.com82ec0b02012-12-17 19:25:54 +000084static void setUpFilter(SkCanvas* canvas, PictureRenderer::DrawFilterFlags* drawFilters) {
caryclark@google.coma3622372012-11-06 21:26:13 +000085 if (drawFilters && !canvas->getDrawFilter()) {
86 canvas->setDrawFilter(SkNEW_ARGS(FlagsDrawFilter, (drawFilters)))->unref();
caryclark@google.come3e940c2012-11-07 16:42:17 +000087 if (drawFilters[0] & PictureRenderer::kAAClip_DrawFilterFlag) {
88 canvas->setAllowSoftClip(false);
89 }
caryclark@google.coma3622372012-11-06 21:26:13 +000090 }
caryclark@google.coma3622372012-11-06 21:26:13 +000091}
92
keyar@chromium.orga474ce32012-08-20 15:03:57 +000093SkCanvas* PictureRenderer::setupCanvas() {
scroggo@google.comc0d5e542012-12-13 21:40:48 +000094 const int width = this->getViewWidth();
95 const int height = this->getViewHeight();
96 return this->setupCanvas(width, height);
keyar@chromium.orga474ce32012-08-20 15:03:57 +000097}
98
99SkCanvas* PictureRenderer::setupCanvas(int width, int height) {
caryclark@google.coma3622372012-11-06 21:26:13 +0000100 SkCanvas* canvas;
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000101 switch(fDeviceType) {
102 case kBitmap_DeviceType: {
103 SkBitmap bitmap;
keyar@chromium.orga474ce32012-08-20 15:03:57 +0000104 sk_tools::setup_bitmap(&bitmap, width, height);
caryclark@google.coma3622372012-11-06 21:26:13 +0000105 canvas = SkNEW_ARGS(SkCanvas, (bitmap));
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000106 }
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000107 break;
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000108#if SK_SUPPORT_GPU
109 case kGPU_DeviceType: {
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000110 SkAutoTUnref<SkGpuDevice> device(SkNEW_ARGS(SkGpuDevice,
keyar@chromium.org06125642012-08-20 15:03:33 +0000111 (fGrContext, SkBitmap::kARGB_8888_Config,
keyar@chromium.orga474ce32012-08-20 15:03:57 +0000112 width, height)));
caryclark@google.coma3622372012-11-06 21:26:13 +0000113 canvas = SkNEW_ARGS(SkCanvas, (device.get()));
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000114 }
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000115 break;
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000116#endif
117 default:
118 SkASSERT(0);
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000119 return NULL;
keyar@chromium.org4ea96c52012-08-20 15:03:29 +0000120 }
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000121 setUpFilter(canvas, fDrawFilters);
122 this->scaleToScaleFactor(canvas);
123 return canvas;
124}
keyar@chromium.orga474ce32012-08-20 15:03:57 +0000125
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000126void PictureRenderer::scaleToScaleFactor(SkCanvas* canvas) {
127 SkASSERT(canvas != NULL);
128 if (fScaleFactor != SK_Scalar1) {
129 canvas->scale(fScaleFactor, fScaleFactor);
130 }
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000131}
132
133void PictureRenderer::end() {
keyar@chromium.org77a55222012-08-20 15:03:47 +0000134 this->resetState();
junov@chromium.org9313ca42012-11-02 18:11:49 +0000135 SkSafeUnref(fPicture);
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000136 fPicture = NULL;
137 fCanvas.reset(NULL);
138}
139
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000140int PictureRenderer::getViewWidth() {
141 SkASSERT(fPicture != NULL);
142 int width = fPicture->width();
143 if (fViewport.width() > 0) {
144 width = SkMin32(width, fViewport.width());
145 }
146 return width;
147}
148
149int PictureRenderer::getViewHeight() {
150 SkASSERT(fPicture != NULL);
151 int height = fPicture->height();
152 if (fViewport.height() > 0) {
153 height = SkMin32(height, fViewport.height());
154 }
155 return height;
156}
157
junov@chromium.org9313ca42012-11-02 18:11:49 +0000158/** Converts fPicture to a picture that uses a BBoxHierarchy.
159 * PictureRenderer subclasses that are used to test picture playback
160 * should call this method during init.
161 */
162void PictureRenderer::buildBBoxHierarchy() {
163 SkASSERT(NULL != fPicture);
164 if (kNone_BBoxHierarchyType != fBBoxHierarchyType && NULL != fPicture) {
165 SkPicture* newPicture = this->createPicture();
166 SkCanvas* recorder = newPicture->beginRecording(fPicture->width(), fPicture->height(),
167 this->recordFlags());
168 fPicture->draw(recorder);
169 newPicture->endRecording();
170 fPicture->unref();
171 fPicture = newPicture;
172 }
173}
174
keyar@chromium.org77a55222012-08-20 15:03:47 +0000175void PictureRenderer::resetState() {
keyar@chromium.org28136b32012-08-20 15:04:15 +0000176#if SK_SUPPORT_GPU
177 if (this->isUsingGpuDevice()) {
178 SkGLContext* glContext = fGrContextFactory.getGLContext(
179 GrContextFactory::kNative_GLContextType);
keyar@chromium.org28136b32012-08-20 15:04:15 +0000180
181 SkASSERT(glContext != NULL);
182 if (NULL == glContext) {
183 return;
184 }
185
scroggo@google.com9a412522012-09-07 15:21:18 +0000186 fGrContext->flush();
keyar@chromium.org77a55222012-08-20 15:03:47 +0000187 SK_GL(*glContext, Finish());
keyar@chromium.org77a55222012-08-20 15:03:47 +0000188 }
keyar@chromium.orga40c20d2012-08-20 15:04:12 +0000189#endif
keyar@chromium.org77a55222012-08-20 15:03:47 +0000190}
191
junov@chromium.org9313ca42012-11-02 18:11:49 +0000192uint32_t PictureRenderer::recordFlags() {
193 return kNone_BBoxHierarchyType == fBBoxHierarchyType ? 0 :
194 SkPicture::kOptimizeForClippedPlayback_RecordingFlag;
195}
196
scroggo@google.comb6e806b2012-10-03 17:32:33 +0000197/**
198 * Write the canvas to the specified path.
199 * @param canvas Must be non-null. Canvas to be written to a file.
200 * @param path Path for the file to be written. Should have no extension; write() will append
201 * an appropriate one. Passed in by value so it can be modified.
202 * @return bool True if the Canvas is written to a file.
203 */
204static bool write(SkCanvas* canvas, SkString path) {
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000205 SkASSERT(canvas != NULL);
scroggo@google.comb6e806b2012-10-03 17:32:33 +0000206 if (NULL == canvas) {
keyar@chromium.org9299ede2012-08-21 19:05:08 +0000207 return false;
208 }
209
210 SkBitmap bitmap;
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000211 SkISize size = canvas->getDeviceSize();
212 sk_tools::setup_bitmap(&bitmap, size.width(), size.height());
keyar@chromium.org9299ede2012-08-21 19:05:08 +0000213
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000214 canvas->readPixels(&bitmap, 0, 0);
keyar@chromium.org9299ede2012-08-21 19:05:08 +0000215 sk_tools::force_all_opaque(bitmap);
216
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000217 // Since path is passed in by value, it is okay to modify it.
218 path.append(".png");
keyar@chromium.org9299ede2012-08-21 19:05:08 +0000219 return SkImageEncoder::EncodeFile(path.c_str(), bitmap, SkImageEncoder::kPNG_Type, 100);
220}
221
scroggo@google.comb6e806b2012-10-03 17:32:33 +0000222/**
223 * If path is non NULL, append number to it, and call write(SkCanvas*, SkString) to write the
224 * provided canvas to a file. Returns true if path is NULL or if write() succeeds.
225 */
226static bool writeAppendNumber(SkCanvas* canvas, const SkString* path, int number) {
227 if (NULL == path) {
228 return true;
229 }
230 SkString pathWithNumber(*path);
231 pathWithNumber.appendf("%i", number);
232 return write(canvas, pathWithNumber);
233}
234
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000235///////////////////////////////////////////////////////////////////////////////////////////////
236
djsollen@google.comfd9720c2012-11-06 16:54:40 +0000237SkCanvas* RecordPictureRenderer::setupCanvas(int width, int height) {
238 // defer the canvas setup until the render step
239 return NULL;
240}
241
scroggo@google.coma9e3a362012-11-07 17:52:48 +0000242static bool PNGEncodeBitmapToStream(SkWStream* wStream, const SkBitmap& bm) {
243 return SkImageEncoder::EncodeStream(wStream, bm, SkImageEncoder::kPNG_Type, 100);
244}
245
246bool RecordPictureRenderer::render(const SkString* path) {
junov@chromium.org9313ca42012-11-02 18:11:49 +0000247 SkAutoTUnref<SkPicture> replayer(this->createPicture());
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000248 SkCanvas* recorder = replayer->beginRecording(this->getViewWidth(), this->getViewHeight(),
junov@chromium.org9313ca42012-11-02 18:11:49 +0000249 this->recordFlags());
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000250 this->scaleToScaleFactor(recorder);
scroggo@google.com9a412522012-09-07 15:21:18 +0000251 fPicture->draw(recorder);
junov@chromium.org9313ca42012-11-02 18:11:49 +0000252 replayer->endRecording();
scroggo@google.coma9e3a362012-11-07 17:52:48 +0000253 if (path != NULL) {
254 // Record the new picture as a new SKP with PNG encoded bitmaps.
255 SkString skpPath(*path);
256 // ".skp" was removed from 'path' before being passed in here.
257 skpPath.append(".skp");
258 SkFILEWStream stream(skpPath.c_str());
259 replayer->serialize(&stream, &PNGEncodeBitmapToStream);
260 return true;
261 }
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000262 return false;
scroggo@google.com9a412522012-09-07 15:21:18 +0000263}
264
scroggo@google.com0a049b82012-11-02 22:01:26 +0000265SkString RecordPictureRenderer::getConfigNameInternal() {
266 return SkString("record");
267}
268
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000269///////////////////////////////////////////////////////////////////////////////////////////////
270
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000271bool PipePictureRenderer::render(const SkString* path) {
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000272 SkASSERT(fCanvas.get() != NULL);
273 SkASSERT(fPicture != NULL);
keyar@chromium.org78a35c52012-08-20 15:03:44 +0000274 if (NULL == fCanvas.get() || NULL == fPicture) {
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000275 return false;
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000276 }
277
278 PipeController pipeController(fCanvas.get());
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000279 SkGPipeWriter writer;
280 SkCanvas* pipeCanvas = writer.startRecording(&pipeController);
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000281 pipeCanvas->drawPicture(*fPicture);
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000282 writer.endRecording();
scroggo@google.com9a412522012-09-07 15:21:18 +0000283 fCanvas->flush();
borenet@google.com070d3542012-10-26 13:26:55 +0000284 if (NULL != path) {
285 return write(fCanvas, *path);
286 }
287 return true;
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000288}
289
scroggo@google.com0a049b82012-11-02 22:01:26 +0000290SkString PipePictureRenderer::getConfigNameInternal() {
291 return SkString("pipe");
292}
293
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000294///////////////////////////////////////////////////////////////////////////////////////////////
295
junov@chromium.org9313ca42012-11-02 18:11:49 +0000296void SimplePictureRenderer::init(SkPicture* picture) {
297 INHERITED::init(picture);
298 this->buildBBoxHierarchy();
299}
300
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000301bool SimplePictureRenderer::render(const SkString* path) {
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000302 SkASSERT(fCanvas.get() != NULL);
303 SkASSERT(fPicture != NULL);
keyar@chromium.org78a35c52012-08-20 15:03:44 +0000304 if (NULL == fCanvas.get() || NULL == fPicture) {
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000305 return false;
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000306 }
307
308 fCanvas->drawPicture(*fPicture);
scroggo@google.com9a412522012-09-07 15:21:18 +0000309 fCanvas->flush();
borenet@google.com070d3542012-10-26 13:26:55 +0000310 if (NULL != path) {
311 return write(fCanvas, *path);
312 }
313 return true;
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000314}
315
scroggo@google.com0a049b82012-11-02 22:01:26 +0000316SkString SimplePictureRenderer::getConfigNameInternal() {
317 return SkString("simple");
318}
319
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000320///////////////////////////////////////////////////////////////////////////////////////////////
321
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000322TiledPictureRenderer::TiledPictureRenderer()
scroggo@google.coma62da2f2012-11-02 21:28:12 +0000323 : fTileWidth(kDefaultTileWidth)
rileya@google.comb947b912012-08-29 17:35:07 +0000324 , fTileHeight(kDefaultTileHeight)
rileya@google.coma04dc022012-09-10 19:01:38 +0000325 , fTileWidthPercentage(0.0)
rileya@google.comb947b912012-08-29 17:35:07 +0000326 , fTileHeightPercentage(0.0)
scroggo@google.comcbcef702012-12-13 22:09:28 +0000327 , fTileMinPowerOf2Width(0)
328 , fCurrentTileOffset(-1)
329 , fTilesX(0)
330 , fTilesY(0) { }
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000331
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000332void TiledPictureRenderer::init(SkPicture* pict) {
333 SkASSERT(pict != NULL);
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000334 SkASSERT(0 == fTileRects.count());
335 if (NULL == pict || fTileRects.count() != 0) {
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000336 return;
337 }
338
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000339 // Do not call INHERITED::init(), which would create a (potentially large) canvas which is not
340 // used by bench_pictures.
341 fPicture = pict;
junov@chromium.org9313ca42012-11-02 18:11:49 +0000342 fPicture->ref();
scroggo@google.coma62da2f2012-11-02 21:28:12 +0000343 this->buildBBoxHierarchy();
keyar@chromium.orgcc6e5ef2012-07-27 20:09:26 +0000344
345 if (fTileWidthPercentage > 0) {
robertphillips@google.com5d8d1862012-08-15 14:36:41 +0000346 fTileWidth = sk_float_ceil2int(float(fTileWidthPercentage * fPicture->width() / 100));
keyar@chromium.orgcc6e5ef2012-07-27 20:09:26 +0000347 }
348 if (fTileHeightPercentage > 0) {
robertphillips@google.com5d8d1862012-08-15 14:36:41 +0000349 fTileHeight = sk_float_ceil2int(float(fTileHeightPercentage * fPicture->height() / 100));
keyar@chromium.orgcc6e5ef2012-07-27 20:09:26 +0000350 }
351
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000352 if (fTileMinPowerOf2Width > 0) {
353 this->setupPowerOf2Tiles();
354 } else {
355 this->setupTiles();
356 }
scroggo@google.comcbcef702012-12-13 22:09:28 +0000357 fCanvas.reset(this->setupCanvas(fTileWidth, fTileHeight));
358 // Initialize to -1 so that the first call to nextTile will set this up to draw tile 0 on the
359 // first call to drawCurrentTile.
360 fCurrentTileOffset = -1;
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000361}
362
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000363void TiledPictureRenderer::end() {
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000364 fTileRects.reset();
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000365 this->INHERITED::end();
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000366}
367
keyar@chromium.org9d696c02012-08-07 17:11:33 +0000368void TiledPictureRenderer::setupTiles() {
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000369 // Only use enough tiles to cover the viewport
370 const int width = this->getViewWidth();
371 const int height = this->getViewHeight();
372
scroggo@google.comcbcef702012-12-13 22:09:28 +0000373 fTilesX = fTilesY = 0;
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000374 for (int tile_y_start = 0; tile_y_start < height; tile_y_start += fTileHeight) {
scroggo@google.comcbcef702012-12-13 22:09:28 +0000375 fTilesY++;
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000376 for (int tile_x_start = 0; tile_x_start < width; tile_x_start += fTileWidth) {
scroggo@google.comcbcef702012-12-13 22:09:28 +0000377 if (0 == tile_y_start) {
378 // Only count tiles in the X direction on the first pass.
379 fTilesX++;
380 }
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000381 *fTileRects.append() = SkRect::MakeXYWH(SkIntToScalar(tile_x_start),
382 SkIntToScalar(tile_y_start),
383 SkIntToScalar(fTileWidth),
384 SkIntToScalar(fTileHeight));
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000385 }
386 }
387}
388
scroggo@google.comcbcef702012-12-13 22:09:28 +0000389bool TiledPictureRenderer::tileDimensions(int &x, int &y) {
390 if (fTileRects.count() == 0 || NULL == fPicture) {
391 return false;
392 }
393 x = fTilesX;
394 y = fTilesY;
395 return true;
396}
397
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000398// The goal of the powers of two tiles is to minimize the amount of wasted tile
399// space in the width-wise direction and then minimize the number of tiles. The
400// constraints are that every tile must have a pixel width that is a power of
401// two and also be of some minimal width (that is also a power of two).
402//
scroggo@google.com58b4ead2012-08-31 16:15:22 +0000403// This is solved by first taking our picture size and rounding it up to the
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000404// multiple of the minimal width. The binary representation of this rounded
405// value gives us the tiles we need: a bit of value one means we need a tile of
406// that size.
407void TiledPictureRenderer::setupPowerOf2Tiles() {
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000408 // Only use enough tiles to cover the viewport
409 const int width = this->getViewWidth();
410 const int height = this->getViewHeight();
411
412 int rounded_value = width;
413 if (width % fTileMinPowerOf2Width != 0) {
414 rounded_value = width - (width % fTileMinPowerOf2Width) + fTileMinPowerOf2Width;
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000415 }
416
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000417 int num_bits = SkScalarCeilToInt(SkScalarLog2(SkIntToScalar(width)));
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000418 int largest_possible_tile_size = 1 << num_bits;
419
scroggo@google.comcbcef702012-12-13 22:09:28 +0000420 fTilesX = fTilesY = 0;
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000421 // The tile height is constant for a particular picture.
scroggo@google.comc0d5e542012-12-13 21:40:48 +0000422 for (int tile_y_start = 0; tile_y_start < height; tile_y_start += fTileHeight) {
scroggo@google.comcbcef702012-12-13 22:09:28 +0000423 fTilesY++;
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000424 int tile_x_start = 0;
425 int current_width = largest_possible_tile_size;
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000426 // Set fTileWidth to be the width of the widest tile, so that each canvas is large enough
427 // to draw each tile.
428 fTileWidth = current_width;
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000429
430 while (current_width >= fTileMinPowerOf2Width) {
431 // It is very important this is a bitwise AND.
432 if (current_width & rounded_value) {
scroggo@google.comcbcef702012-12-13 22:09:28 +0000433 if (0 == tile_y_start) {
434 // Only count tiles in the X direction on the first pass.
435 fTilesX++;
436 }
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000437 *fTileRects.append() = SkRect::MakeXYWH(SkIntToScalar(tile_x_start),
438 SkIntToScalar(tile_y_start),
439 SkIntToScalar(current_width),
440 SkIntToScalar(fTileHeight));
keyar@chromium.orgf4959ab2012-08-23 20:53:25 +0000441 tile_x_start += current_width;
442 }
443
444 current_width >>= 1;
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000445 }
446 }
447}
448
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000449/**
450 * Draw the specified playback to the canvas translated to rectangle provided, so that this mini
451 * canvas represents the rectangle's portion of the overall picture.
452 * Saves and restores so that the initial clip and matrix return to their state before this function
453 * is called.
454 */
455template<class T>
456static void DrawTileToCanvas(SkCanvas* canvas, const SkRect& tileRect, T* playback) {
457 int saveCount = canvas->save();
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000458 // Translate so that we draw the correct portion of the picture.
459 // Perform a postTranslate so that the scaleFactor does not interfere with the positioning.
460 SkMatrix mat(canvas->getTotalMatrix());
461 mat.postTranslate(-tileRect.fLeft, -tileRect.fTop);
462 canvas->setMatrix(mat);
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000463 playback->draw(canvas);
464 canvas->restoreToCount(saveCount);
465 canvas->flush();
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000466}
467
scroggo@google.com58b4ead2012-08-31 16:15:22 +0000468///////////////////////////////////////////////////////////////////////////////////////////////
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000469
scroggo@google.comcbcef702012-12-13 22:09:28 +0000470bool TiledPictureRenderer::nextTile(int &i, int &j) {
471 if (++fCurrentTileOffset < fTileRects.count()) {
472 i = fCurrentTileOffset % fTilesX;
473 j = fCurrentTileOffset / fTilesX;
474 return true;
475 }
476 return false;
477}
478
479void TiledPictureRenderer::drawCurrentTile() {
480 SkASSERT(fCurrentTileOffset >= 0 && fCurrentTileOffset < fTileRects.count());
481 DrawTileToCanvas(fCanvas, fTileRects[fCurrentTileOffset], fPicture);
482}
483
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000484bool TiledPictureRenderer::render(const SkString* path) {
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000485 SkASSERT(fPicture != NULL);
486 if (NULL == fPicture) {
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000487 return false;
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000488 }
489
scroggo@google.coma62da2f2012-11-02 21:28:12 +0000490 bool success = true;
491 for (int i = 0; i < fTileRects.count(); ++i) {
scroggo@google.comcbcef702012-12-13 22:09:28 +0000492 DrawTileToCanvas(fCanvas, fTileRects[i], fPicture);
scroggo@google.coma62da2f2012-11-02 21:28:12 +0000493 if (NULL != path) {
scroggo@google.comcbcef702012-12-13 22:09:28 +0000494 success &= writeAppendNumber(fCanvas, path, i);
scroggo@google.com58b4ead2012-08-31 16:15:22 +0000495 }
keyar@chromium.org163b5672012-08-01 17:53:29 +0000496 }
scroggo@google.coma62da2f2012-11-02 21:28:12 +0000497 return success;
keyar@chromium.org163b5672012-08-01 17:53:29 +0000498}
499
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000500SkCanvas* TiledPictureRenderer::setupCanvas(int width, int height) {
501 SkCanvas* canvas = this->INHERITED::setupCanvas(width, height);
502 SkASSERT(fPicture != NULL);
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000503 // Clip the tile to an area that is completely inside both the SkPicture and the viewport. This
504 // is mostly important for tiles on the right and bottom edges as they may go over this area and
505 // the picture may have some commands that draw outside of this area and so should not actually
506 // be written.
507 // Uses a clipRegion so that it will be unaffected by the scale factor, which may have been set
508 // by INHERITED::setupCanvas.
509 SkRegion clipRegion;
510 clipRegion.setRect(0, 0, this->getViewWidth(), this->getViewHeight());
511 canvas->clipRegion(clipRegion);
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000512 return canvas;
513}
scroggo@google.com0a049b82012-11-02 22:01:26 +0000514
515SkString TiledPictureRenderer::getConfigNameInternal() {
516 SkString name;
517 if (fTileMinPowerOf2Width > 0) {
518 name.append("pow2tile_");
519 name.appendf("%i", fTileMinPowerOf2Width);
520 } else {
521 name.append("tile_");
522 if (fTileWidthPercentage > 0) {
523 name.appendf("%.f%%", fTileWidthPercentage);
524 } else {
525 name.appendf("%i", fTileWidth);
526 }
527 }
528 name.append("x");
529 if (fTileHeightPercentage > 0) {
530 name.appendf("%.f%%", fTileHeightPercentage);
531 } else {
532 name.appendf("%i", fTileHeight);
533 }
534 return name;
535}
536
scroggo@google.coma62da2f2012-11-02 21:28:12 +0000537///////////////////////////////////////////////////////////////////////////////////////////////
538
539// Holds all of the information needed to draw a set of tiles.
540class CloneData : public SkRunnable {
541
542public:
543 CloneData(SkPicture* clone, SkCanvas* canvas, SkTDArray<SkRect>& rects, int start, int end,
544 SkRunnable* done)
545 : fClone(clone)
546 , fCanvas(canvas)
547 , fPath(NULL)
548 , fRects(rects)
549 , fStart(start)
550 , fEnd(end)
551 , fSuccess(NULL)
552 , fDone(done) {
553 SkASSERT(fDone != NULL);
554 }
555
556 virtual void run() SK_OVERRIDE {
557 SkGraphics::SetTLSFontCacheLimit(1024 * 1024);
558 for (int i = fStart; i < fEnd; i++) {
559 DrawTileToCanvas(fCanvas, fRects[i], fClone);
560 if (fPath != NULL && !writeAppendNumber(fCanvas, fPath, i)
561 && fSuccess != NULL) {
562 *fSuccess = false;
563 // If one tile fails to write to a file, do not continue drawing the rest.
564 break;
565 }
566 }
567 fDone->run();
568 }
569
570 void setPathAndSuccess(const SkString* path, bool* success) {
571 fPath = path;
572 fSuccess = success;
573 }
574
575private:
576 // All pointers unowned.
577 SkPicture* fClone; // Picture to draw from. Each CloneData has a unique one which
578 // is threadsafe.
579 SkCanvas* fCanvas; // Canvas to draw to. Reused for each tile.
580 const SkString* fPath; // If non-null, path to write the result to as a PNG.
581 SkTDArray<SkRect>& fRects; // All tiles of the picture.
582 const int fStart; // Range of tiles drawn by this thread.
583 const int fEnd;
584 bool* fSuccess; // Only meaningful if path is non-null. Shared by all threads,
585 // and only set to false upon failure to write to a PNG.
586 SkRunnable* fDone;
587};
588
589MultiCorePictureRenderer::MultiCorePictureRenderer(int threadCount)
590: fNumThreads(threadCount)
591, fThreadPool(threadCount)
592, fCountdown(threadCount) {
593 // Only need to create fNumThreads - 1 clones, since one thread will use the base
594 // picture.
595 fPictureClones = SkNEW_ARRAY(SkPicture, fNumThreads - 1);
596 fCloneData = SkNEW_ARRAY(CloneData*, fNumThreads);
597}
598
599void MultiCorePictureRenderer::init(SkPicture *pict) {
600 // Set fPicture and the tiles.
601 this->INHERITED::init(pict);
602 for (int i = 0; i < fNumThreads; ++i) {
603 *fCanvasPool.append() = this->setupCanvas(this->getTileWidth(), this->getTileHeight());
604 }
605 // Only need to create fNumThreads - 1 clones, since one thread will use the base picture.
606 fPicture->clone(fPictureClones, fNumThreads - 1);
607 // Populate each thread with the appropriate data.
608 // Group the tiles into nearly equal size chunks, rounding up so we're sure to cover them all.
609 const int chunkSize = (fTileRects.count() + fNumThreads - 1) / fNumThreads;
610
611 for (int i = 0; i < fNumThreads; i++) {
612 SkPicture* pic;
613 if (i == fNumThreads-1) {
614 // The last set will use the original SkPicture.
615 pic = fPicture;
616 } else {
617 pic = &fPictureClones[i];
618 }
619 const int start = i * chunkSize;
620 const int end = SkMin32(start + chunkSize, fTileRects.count());
621 fCloneData[i] = SkNEW_ARGS(CloneData,
622 (pic, fCanvasPool[i], fTileRects, start, end, &fCountdown));
623 }
624}
625
626bool MultiCorePictureRenderer::render(const SkString *path) {
627 bool success = true;
628 if (path != NULL) {
629 for (int i = 0; i < fNumThreads-1; i++) {
630 fCloneData[i]->setPathAndSuccess(path, &success);
631 }
632 }
633
634 fCountdown.reset(fNumThreads);
635 for (int i = 0; i < fNumThreads; i++) {
636 fThreadPool.add(fCloneData[i]);
637 }
638 fCountdown.wait();
639
640 return success;
641}
642
643void MultiCorePictureRenderer::end() {
644 for (int i = 0; i < fNumThreads - 1; i++) {
645 SkDELETE(fCloneData[i]);
646 fCloneData[i] = NULL;
647 }
648
649 fCanvasPool.unrefAll();
650
651 this->INHERITED::end();
652}
653
654MultiCorePictureRenderer::~MultiCorePictureRenderer() {
655 // Each individual CloneData was deleted in end.
656 SkDELETE_ARRAY(fCloneData);
657 SkDELETE_ARRAY(fPictureClones);
658}
scroggo@google.combcdf2ec2012-09-20 14:42:33 +0000659
scroggo@google.com0a049b82012-11-02 22:01:26 +0000660SkString MultiCorePictureRenderer::getConfigNameInternal() {
661 SkString name = this->INHERITED::getConfigNameInternal();
662 name.appendf("_multi_%i_threads", fNumThreads);
663 return name;
664}
665
scroggo@google.comacfb30e2012-09-18 14:32:35 +0000666///////////////////////////////////////////////////////////////////////////////////////////////
scroggo@google.com9a412522012-09-07 15:21:18 +0000667
668void PlaybackCreationRenderer::setup() {
junov@chromium.org9313ca42012-11-02 18:11:49 +0000669 fReplayer.reset(this->createPicture());
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000670 SkCanvas* recorder = fReplayer->beginRecording(this->getViewWidth(), this->getViewHeight(),
junov@chromium.org9313ca42012-11-02 18:11:49 +0000671 this->recordFlags());
scroggo@google.com82ec0b02012-12-17 19:25:54 +0000672 this->scaleToScaleFactor(recorder);
scroggo@google.com9a412522012-09-07 15:21:18 +0000673 fPicture->draw(recorder);
674}
675
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000676bool PlaybackCreationRenderer::render(const SkString*) {
junov@chromium.org9313ca42012-11-02 18:11:49 +0000677 fReplayer->endRecording();
scroggo@google.com81f9d2e2012-09-20 14:54:21 +0000678 // Since this class does not actually render, return false.
679 return false;
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000680}
681
scroggo@google.com0a049b82012-11-02 22:01:26 +0000682SkString PlaybackCreationRenderer::getConfigNameInternal() {
683 return SkString("playback_creation");
684}
685
junov@chromium.org9313ca42012-11-02 18:11:49 +0000686///////////////////////////////////////////////////////////////////////////////////////////////
687// SkPicture variants for each BBoxHierarchy type
688
689class RTreePicture : public SkPicture {
690public:
691 virtual SkBBoxHierarchy* createBBoxHierarchy() const SK_OVERRIDE{
692 static const int kRTreeMinChildren = 6;
693 static const int kRTreeMaxChildren = 11;
694 SkScalar aspectRatio = SkScalarDiv(SkIntToScalar(fWidth),
695 SkIntToScalar(fHeight));
696 return SkRTree::Create(kRTreeMinChildren, kRTreeMaxChildren,
697 aspectRatio);
698 }
699};
700
701SkPicture* PictureRenderer::createPicture() {
702 switch (fBBoxHierarchyType) {
703 case kNone_BBoxHierarchyType:
704 return SkNEW(SkPicture);
705 case kRTree_BBoxHierarchyType:
706 return SkNEW(RTreePicture);
junov@chromium.org7b537062012-11-06 18:58:43 +0000707 case kTileGrid_BBoxHierarchyType:
junov@chromium.org3cb834b2012-12-13 16:39:53 +0000708 return SkNEW_ARGS(SkTileGridPicture, (fGridWidth, fGridHeight, fPicture->width(),
709 fPicture->height()));
junov@chromium.org9313ca42012-11-02 18:11:49 +0000710 }
711 SkASSERT(0); // invalid bbhType
712 return NULL;
keyar@chromium.org451bb9f2012-07-26 17:27:57 +0000713}
junov@chromium.org9313ca42012-11-02 18:11:49 +0000714
reed@google.comfe7b1ed2012-11-29 21:00:39 +0000715///////////////////////////////////////////////////////////////////////////////
716
717class GatherRenderer : public PictureRenderer {
718public:
719 virtual bool render(const SkString* path) SK_OVERRIDE {
720 SkRect bounds = SkRect::MakeWH(SkIntToScalar(fPicture->width()),
721 SkIntToScalar(fPicture->height()));
722 SkData* data = SkPictureUtils::GatherPixelRefs(fPicture, bounds);
723 SkSafeUnref(data);
skia.committer@gmail.comc7b4be72012-12-11 02:01:20 +0000724
reed@google.comfe7b1ed2012-11-29 21:00:39 +0000725 return NULL == path; // we don't have anything to write
726 }
skia.committer@gmail.comc7b4be72012-12-11 02:01:20 +0000727
reed@google.comfe7b1ed2012-11-29 21:00:39 +0000728private:
729 virtual SkString getConfigNameInternal() SK_OVERRIDE {
730 return SkString("gather_pixelrefs");
731 }
732};
733
734PictureRenderer* CreateGatherPixelRefsRenderer() {
735 return SkNEW(GatherRenderer);
736}
skia.committer@gmail.comc3d7d902012-11-30 02:01:24 +0000737
reed@google.com5a34fd32012-12-10 16:05:09 +0000738///////////////////////////////////////////////////////////////////////////////
739
740class PictureCloneRenderer : public PictureRenderer {
741public:
742 virtual bool render(const SkString* path) SK_OVERRIDE {
743 for (int i = 0; i < 100; ++i) {
744 SkPicture* clone = fPicture->clone();
745 SkSafeUnref(clone);
746 }
skia.committer@gmail.comc7b4be72012-12-11 02:01:20 +0000747
reed@google.com5a34fd32012-12-10 16:05:09 +0000748 return NULL == path; // we don't have anything to write
749 }
skia.committer@gmail.comc7b4be72012-12-11 02:01:20 +0000750
reed@google.com5a34fd32012-12-10 16:05:09 +0000751private:
752 virtual SkString getConfigNameInternal() SK_OVERRIDE {
753 return SkString("picture_clone");
754 }
755};
756
757PictureRenderer* CreatePictureCloneRenderer() {
758 return SkNEW(PictureCloneRenderer);
759}
760
junov@chromium.org9313ca42012-11-02 18:11:49 +0000761} // namespace sk_tools