keyar@chromium.org | b3fb7c1 | 2012-08-20 21:02:49 +0000 | [diff] [blame] | 1 | /* |
| 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.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 8 | #include "PictureRenderer.h" |
| 9 | #include "SamplePipeControllers.h" |
| 10 | #include "SkCanvas.h" |
| 11 | #include "SkDevice.h" |
keyar@chromium.org | 9299ede | 2012-08-21 19:05:08 +0000 | [diff] [blame^] | 12 | #include "SkImageEncoder.h" |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 13 | #include "SkGPipe.h" |
| 14 | #include "SkPicture.h" |
keyar@chromium.org | 9299ede | 2012-08-21 19:05:08 +0000 | [diff] [blame^] | 15 | #include "SkString.h" |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 16 | #include "SkTDArray.h" |
| 17 | #include "SkTypes.h" |
| 18 | #include "picture_utils.h" |
| 19 | |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 20 | #if SK_SUPPORT_GPU |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 21 | #include "SkGpuDevice.h" |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 22 | #endif |
| 23 | |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 24 | namespace sk_tools { |
| 25 | |
| 26 | enum { |
| 27 | kDefaultTileWidth = 256, |
| 28 | kDefaultTileHeight = 256 |
| 29 | }; |
| 30 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 31 | void PictureRenderer::init(SkPicture* pict) { |
keyar@chromium.org | 78a35c5 | 2012-08-20 15:03:44 +0000 | [diff] [blame] | 32 | SkASSERT(NULL == fPicture); |
| 33 | SkASSERT(NULL == fCanvas.get()); |
| 34 | if (fPicture != NULL || NULL != fCanvas.get()) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 35 | return; |
| 36 | } |
| 37 | |
| 38 | SkASSERT(pict != NULL); |
keyar@chromium.org | 78a35c5 | 2012-08-20 15:03:44 +0000 | [diff] [blame] | 39 | if (NULL == pict) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 40 | return; |
| 41 | } |
| 42 | |
| 43 | fPicture = pict; |
keyar@chromium.org | a474ce3 | 2012-08-20 15:03:57 +0000 | [diff] [blame] | 44 | fCanvas.reset(this->setupCanvas()); |
| 45 | } |
| 46 | |
| 47 | SkCanvas* PictureRenderer::setupCanvas() { |
| 48 | return this->setupCanvas(fPicture->width(), fPicture->height()); |
| 49 | } |
| 50 | |
| 51 | SkCanvas* PictureRenderer::setupCanvas(int width, int height) { |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 52 | switch(fDeviceType) { |
| 53 | case kBitmap_DeviceType: { |
| 54 | SkBitmap bitmap; |
keyar@chromium.org | a474ce3 | 2012-08-20 15:03:57 +0000 | [diff] [blame] | 55 | sk_tools::setup_bitmap(&bitmap, width, height); |
| 56 | return SkNEW_ARGS(SkCanvas, (bitmap)); |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 57 | break; |
| 58 | } |
| 59 | #if SK_SUPPORT_GPU |
| 60 | case kGPU_DeviceType: { |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 61 | SkAutoTUnref<SkGpuDevice> device(SkNEW_ARGS(SkGpuDevice, |
keyar@chromium.org | 0612564 | 2012-08-20 15:03:33 +0000 | [diff] [blame] | 62 | (fGrContext, SkBitmap::kARGB_8888_Config, |
keyar@chromium.org | a474ce3 | 2012-08-20 15:03:57 +0000 | [diff] [blame] | 63 | width, height))); |
| 64 | return SkNEW_ARGS(SkCanvas, (device.get())); |
keyar@chromium.org | 4ea96c5 | 2012-08-20 15:03:29 +0000 | [diff] [blame] | 65 | break; |
| 66 | } |
| 67 | #endif |
| 68 | default: |
| 69 | SkASSERT(0); |
| 70 | } |
keyar@chromium.org | a474ce3 | 2012-08-20 15:03:57 +0000 | [diff] [blame] | 71 | |
| 72 | return NULL; |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 73 | } |
| 74 | |
| 75 | void PictureRenderer::end() { |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 76 | this->resetState(); |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 77 | fPicture = NULL; |
| 78 | fCanvas.reset(NULL); |
| 79 | } |
| 80 | |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 81 | void PictureRenderer::resetState() { |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 82 | #if SK_SUPPORT_GPU |
| 83 | if (this->isUsingGpuDevice()) { |
| 84 | SkGLContext* glContext = fGrContextFactory.getGLContext( |
| 85 | GrContextFactory::kNative_GLContextType); |
| 86 | SK_GL(*glContext, Finish()); |
| 87 | } |
| 88 | #endif |
| 89 | } |
| 90 | |
| 91 | void PictureRenderer::finishDraw() { |
keyar@chromium.org | bffacc0 | 2012-08-20 15:04:07 +0000 | [diff] [blame] | 92 | SkASSERT(fCanvas.get() != NULL); |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 93 | if (NULL == fCanvas.get()) { |
keyar@chromium.org | bffacc0 | 2012-08-20 15:04:07 +0000 | [diff] [blame] | 94 | return; |
| 95 | } |
| 96 | |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 97 | fCanvas->flush(); |
| 98 | |
keyar@chromium.org | a40c20d | 2012-08-20 15:04:12 +0000 | [diff] [blame] | 99 | #if SK_SUPPORT_GPU |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 100 | if (this->isUsingGpuDevice()) { |
| 101 | SkGLContext* glContext = fGrContextFactory.getGLContext( |
| 102 | GrContextFactory::kNative_GLContextType); |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 103 | |
| 104 | SkASSERT(glContext != NULL); |
| 105 | if (NULL == glContext) { |
| 106 | return; |
| 107 | } |
| 108 | |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 109 | SK_GL(*glContext, Finish()); |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 110 | } |
keyar@chromium.org | a40c20d | 2012-08-20 15:04:12 +0000 | [diff] [blame] | 111 | #endif |
keyar@chromium.org | 77a5522 | 2012-08-20 15:03:47 +0000 | [diff] [blame] | 112 | } |
| 113 | |
keyar@chromium.org | 9299ede | 2012-08-21 19:05:08 +0000 | [diff] [blame^] | 114 | bool PictureRenderer::write(const SkString& path) const { |
| 115 | SkASSERT(fCanvas.get() != NULL); |
| 116 | SkASSERT(fPicture != NULL); |
| 117 | if (NULL == fCanvas.get() || NULL == fPicture) { |
| 118 | return false; |
| 119 | } |
| 120 | |
| 121 | SkBitmap bitmap; |
| 122 | sk_tools::setup_bitmap(&bitmap, fPicture->width(), fPicture->height()); |
| 123 | |
| 124 | fCanvas->readPixels(&bitmap, 0, 0); |
| 125 | sk_tools::force_all_opaque(bitmap); |
| 126 | |
| 127 | return SkImageEncoder::EncodeFile(path.c_str(), bitmap, SkImageEncoder::kPNG_Type, 100); |
| 128 | } |
| 129 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 130 | void PipePictureRenderer::render() { |
| 131 | SkASSERT(fCanvas.get() != NULL); |
| 132 | SkASSERT(fPicture != NULL); |
keyar@chromium.org | 78a35c5 | 2012-08-20 15:03:44 +0000 | [diff] [blame] | 133 | if (NULL == fCanvas.get() || NULL == fPicture) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 134 | return; |
| 135 | } |
| 136 | |
| 137 | PipeController pipeController(fCanvas.get()); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 138 | SkGPipeWriter writer; |
| 139 | SkCanvas* pipeCanvas = writer.startRecording(&pipeController); |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 140 | pipeCanvas->drawPicture(*fPicture); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 141 | writer.endRecording(); |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 142 | this->finishDraw(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 143 | } |
| 144 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 145 | void SimplePictureRenderer::render() { |
| 146 | SkASSERT(fCanvas.get() != NULL); |
| 147 | SkASSERT(fPicture != NULL); |
keyar@chromium.org | 78a35c5 | 2012-08-20 15:03:44 +0000 | [diff] [blame] | 148 | if (NULL == fCanvas.get() || NULL == fPicture) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 149 | return; |
| 150 | } |
| 151 | |
| 152 | fCanvas->drawPicture(*fPicture); |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 153 | this->finishDraw(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 154 | } |
| 155 | |
| 156 | TiledPictureRenderer::TiledPictureRenderer() |
| 157 | : fTileWidth(kDefaultTileWidth) |
| 158 | , fTileHeight(kDefaultTileHeight) {} |
| 159 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 160 | void TiledPictureRenderer::init(SkPicture* pict) { |
| 161 | SkASSERT(pict != NULL); |
keyar@chromium.org | 78a35c5 | 2012-08-20 15:03:44 +0000 | [diff] [blame] | 162 | SkASSERT(0 == fTiles.count()); |
| 163 | if (NULL == pict || fTiles.count() != 0) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 164 | return; |
| 165 | } |
| 166 | |
| 167 | this->INHERITED::init(pict); |
keyar@chromium.org | cc6e5ef | 2012-07-27 20:09:26 +0000 | [diff] [blame] | 168 | |
| 169 | if (fTileWidthPercentage > 0) { |
robertphillips@google.com | 5d8d186 | 2012-08-15 14:36:41 +0000 | [diff] [blame] | 170 | fTileWidth = sk_float_ceil2int(float(fTileWidthPercentage * fPicture->width() / 100)); |
keyar@chromium.org | cc6e5ef | 2012-07-27 20:09:26 +0000 | [diff] [blame] | 171 | } |
| 172 | if (fTileHeightPercentage > 0) { |
robertphillips@google.com | 5d8d186 | 2012-08-15 14:36:41 +0000 | [diff] [blame] | 173 | fTileHeight = sk_float_ceil2int(float(fTileHeightPercentage * fPicture->height() / 100)); |
keyar@chromium.org | cc6e5ef | 2012-07-27 20:09:26 +0000 | [diff] [blame] | 174 | } |
| 175 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 176 | this->setupTiles(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 177 | } |
| 178 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 179 | void TiledPictureRenderer::render() { |
| 180 | SkASSERT(fCanvas.get() != NULL); |
| 181 | SkASSERT(fPicture != NULL); |
keyar@chromium.org | 78a35c5 | 2012-08-20 15:03:44 +0000 | [diff] [blame] | 182 | if (NULL == fCanvas.get() || NULL == fPicture) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 183 | return; |
| 184 | } |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 185 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 186 | this->drawTiles(); |
| 187 | this->copyTilesToCanvas(); |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 188 | this->finishDraw(); |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 189 | } |
| 190 | |
| 191 | void TiledPictureRenderer::end() { |
| 192 | this->deleteTiles(); |
| 193 | this->INHERITED::end(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 194 | } |
| 195 | |
| 196 | TiledPictureRenderer::~TiledPictureRenderer() { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 197 | this->deleteTiles(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 198 | } |
| 199 | |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 200 | void TiledPictureRenderer::clipTile(SkCanvas* tile) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 201 | SkRect clip = SkRect::MakeWH(SkIntToScalar(fPicture->width()), |
| 202 | SkIntToScalar(fPicture->height())); |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 203 | tile->clipRect(clip); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 204 | } |
| 205 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 206 | void TiledPictureRenderer::addTile(int tile_x_start, int tile_y_start) { |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 207 | SkCanvas* tile = this->setupCanvas(fTileWidth, fTileHeight); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 208 | |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 209 | tile->translate(SkIntToScalar(-tile_x_start), SkIntToScalar(-tile_y_start)); |
| 210 | this->clipTile(tile); |
| 211 | |
| 212 | fTiles.push(tile); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 213 | } |
| 214 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 215 | void TiledPictureRenderer::setupTiles() { |
| 216 | for (int tile_y_start = 0; tile_y_start < fPicture->height(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 217 | tile_y_start += fTileHeight) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 218 | for (int tile_x_start = 0; tile_x_start < fPicture->width(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 219 | tile_x_start += fTileWidth) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 220 | this->addTile(tile_x_start, tile_y_start); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 221 | } |
| 222 | } |
| 223 | } |
| 224 | |
| 225 | void TiledPictureRenderer::deleteTiles() { |
| 226 | for (int i = 0; i < fTiles.count(); ++i) { |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 227 | SkDELETE(fTiles[i]); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 228 | } |
| 229 | |
| 230 | fTiles.reset(); |
| 231 | } |
| 232 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 233 | void TiledPictureRenderer::drawTiles() { |
keyar@chromium.org | 163b567 | 2012-08-01 17:53:29 +0000 | [diff] [blame] | 234 | for (int i = 0; i < fTiles.count(); ++i) { |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 235 | fTiles[i]->drawPicture(*(fPicture)); |
keyar@chromium.org | 163b567 | 2012-08-01 17:53:29 +0000 | [diff] [blame] | 236 | } |
| 237 | } |
| 238 | |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 239 | void TiledPictureRenderer::finishDraw() { |
keyar@chromium.org | 275be53 | 2012-08-20 15:04:00 +0000 | [diff] [blame] | 240 | for (int i = 0; i < fTiles.count(); ++i) { |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 241 | fTiles[i]->flush(); |
keyar@chromium.org | 275be53 | 2012-08-20 15:04:00 +0000 | [diff] [blame] | 242 | } |
| 243 | |
keyar@chromium.org | 28136b3 | 2012-08-20 15:04:15 +0000 | [diff] [blame] | 244 | #if SK_SUPPORT_GPU |
| 245 | if (this->isUsingGpuDevice()) { |
| 246 | SkGLContext* glContext = fGrContextFactory.getGLContext( |
| 247 | GrContextFactory::kNative_GLContextType); |
| 248 | |
| 249 | SkASSERT(glContext != NULL); |
| 250 | if (NULL == glContext) { |
| 251 | return; |
| 252 | } |
| 253 | |
| 254 | SK_GL(*glContext, Finish()); |
| 255 | } |
| 256 | #endif |
keyar@chromium.org | 275be53 | 2012-08-20 15:04:00 +0000 | [diff] [blame] | 257 | } |
| 258 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 259 | void TiledPictureRenderer::copyTilesToCanvas() { |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 260 | int tile_index = 0; |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 261 | for (int tile_y_start = 0; tile_y_start < fPicture->height(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 262 | tile_y_start += fTileHeight) { |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 263 | for (int tile_x_start = 0; tile_x_start < fPicture->width(); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 264 | tile_x_start += fTileWidth) { |
| 265 | SkASSERT(tile_index < fTiles.count()); |
keyar@chromium.org | da652c2 | 2012-08-20 22:04:07 +0000 | [diff] [blame] | 266 | SkBitmap source = fTiles[tile_index]->getDevice()->accessBitmap(false); |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 267 | fCanvas->drawBitmap(source, |
| 268 | SkIntToScalar(tile_x_start), |
| 269 | SkIntToScalar(tile_y_start)); |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 270 | ++tile_index; |
| 271 | } |
| 272 | } |
| 273 | } |
| 274 | |
| 275 | } |