render_pictures can now do tiling based rendering.

Review URL: https://codereview.appspot.com/6401057

git-svn-id: http://skia.googlecode.com/svn/trunk@4707 2bbb7eff-a529-9590-31e7-b0007b416f81

1 files changed, 97 insertions, 2 deletions

diff --git a/tools/render_pictures_main.cpp b/tools/render_pictures_main.cpp
index 5356cb631a..5572b286d3 100644
--- a/tools/render_pictures_main.cpp
+++ b/tools/render_pictures_main.cpp
@@ -10,6 +10,7 @@
 #include "SamplePipeControllers.h"
 #include "SkCanvas.h"
 #include "SkColorPriv.h"
+#include "SkDevice.h"
 #include "SkGPipe.h"
 #include "SkImageEncoder.h"
 #include "SkOSFile.h"
@@ -17,23 +18,36 @@
 #include "SkStream.h"
 #include "SkString.h"
 #include "SkTArray.h"
+#include "SkTDArray.h"
 #include "picture_utils.h"
 
-typedef void (*RenderFunc) (SkPicture*, SkBitmap*);
+enum {
+    kDefaultTileWidth = 256,
+    kDefaultTileHeight = 256
+};
 
 static void usage(const char* argv0) {
     SkDebugf("SkPicture rendering tool\n");
     SkDebugf("\n"
 "Usage: \n"
-"     %s <input>... <outputDir> \n\n"
+"     %s <input>... <outputDir> \n"
+"     [--pipe | --tile]"
 , argv0);
+    SkDebugf("\n\n");
     SkDebugf(
 "     input:     A list of directories and files to use as input.\n"
 "                    Files are expected to have the .skp extension.\n");
     SkDebugf(
 "     outputDir: directory to write the rendered images.\n");
+    SkDebugf(
+"     --pipe : Render using a SkGPipe\n");
+    SkDebugf(
+"     --tile : Render using tiles.\n");
 }
 
+static void simple_render(SkPicture* pict, SkBitmap* bitmap);
+typedef void (*RenderFunc) (SkPicture*, SkBitmap*);
+
 static void make_output_filepath(SkString* path, const SkString& dir,
                                  const SkString& name) {
     sk_tools::make_filepath(path, dir, name);
@@ -94,6 +108,76 @@ static void pipe_render(SkPicture* picture, SkBitmap* bitmap) {
     pipe_run(picture, &canvas);
 }
 
+struct TileInfo {
+    SkBitmap* fBitmap;
+    SkCanvas* fCanvas;
+};
+
+// Clips the tile to an area that is completely in what the SkPicture says is the
+// drawn-to area. This is mostly important for tiles on the right and bottom edges
+// as they may go over this area and the picture may have some commands that
+// draw outside of this area and so should not actually be written.
+static void clip_tile(SkPicture* picture, const TileInfo& tile) {
+    SkRect clip = SkRect::MakeWH(SkIntToScalar(picture->width()),
+                                 SkIntToScalar(picture->height()));
+    tile.fCanvas->clipRect(clip);
+}
+
+static void add_tile(SkPicture* picture, SkTDArray<TileInfo>* tiles,
+                     int tile_x_start, int tile_y_start) {
+    TileInfo* tile = tiles->push();
+
+    tile->fBitmap = new SkBitmap();
+    sk_tools::setup_bitmap(tile->fBitmap, kDefaultTileWidth, kDefaultTileHeight);
+
+    tile->fCanvas = new SkCanvas(*(tile->fBitmap));
+    tile->fCanvas->translate(SkIntToScalar(-tile_x_start), SkIntToScalar(-tile_y_start));
+    clip_tile(picture, *tile);
+}
+
+static void setup_tiles(SkPicture* picture, SkTDArray<TileInfo>* tiles) {
+    for (int tile_y_start = 0; tile_y_start < picture->height();
+         tile_y_start += kDefaultTileWidth) {
+        for (int tile_x_start = 0; tile_x_start < picture->width();
+             tile_x_start += kDefaultTileHeight) {
+            add_tile(picture, tiles, tile_x_start, tile_y_start);
+        }
+    }
+}
+
+static void copy_tiles_to_bitmap(const SkTDArray<TileInfo>& tiles, SkBitmap* bitmap) {
+    SkCanvas destination(*bitmap);
+
+    int tile_index = 0;
+    for (int tile_y_start = 0; tile_y_start < bitmap->height();
+         tile_y_start += kDefaultTileWidth) {
+        for (int tile_x_start = 0; tile_x_start < bitmap->width();
+             tile_x_start += kDefaultTileHeight) {
+            SkBitmap source = tiles[tile_index].fCanvas->getDevice()->accessBitmap(false);
+            destination.drawBitmap(source, tile_x_start, tile_y_start);
+            ++tile_index;
+        }
+    }
+}
+
+static void tile_render(SkPicture* picture, SkBitmap* bitmap) {
+    sk_tools::setup_bitmap(bitmap, picture->width(), picture->height());
+
+    SkTDArray<TileInfo> tiles;
+    setup_tiles(picture, &tiles);
+
+    for (int i = 0; i < tiles.count(); ++i) {
+        tiles[i].fCanvas->drawPicture(*picture);
+    }
+
+    copy_tiles_to_bitmap(tiles, bitmap);
+
+    for (int i = 0; i < tiles.count(); ++i) {
+        delete tiles[i].fCanvas;
+        delete tiles[i].fBitmap;
+    }
+}
+
 static void render_picture(const SkString& inputPath, const SkString& outputDir,
                            RenderFunc renderFunc) {
     SkString inputFilename;
@@ -137,15 +221,26 @@ static void parse_commandline(int argc, char* const argv[], SkTArray<SkString>*
     for (++argv; argv < stop; ++argv) {
         if (0 == strcmp(*argv, "--pipe")) {
             *renderFunc = pipe_render;
+        } else if (0 == strcmp(*argv, "--tile")) {
+            *renderFunc = tile_render;
+        } else if ((0 == strcmp(*argv, "-h")) || (0 == strcmp(*argv, "--help"))) {
+            usage(argv0);
+            exit(-1);
         } else {
             inputs->push_back(SkString(*argv));
         }
     }
+
+    if (inputs->count() < 2) {
+        usage(argv0);
+        exit(-1);
+    }
 }
 
 int main(int argc, char* const argv[]) {
     SkTArray<SkString> inputs;
     RenderFunc renderFunc = simple_render;
+
     parse_commandline(argc, argv, &inputs, &renderFunc);
     SkString outputDir = inputs[inputs.count() - 1];