render_pictures_main.cpp revision 795cd47342509da7924e703bc5c8503cbf00b10a 
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
8#include "SkBitmap.h"
9#include "SkCanvas.h"
10#include "SkColorPriv.h"
11#include "SkDevice.h"
12#include "SkImageEncoder.h"
13#include "SkOSFile.h"
14#include "SkPicture.h"
15#include "SkStream.h"
16#include "SkString.h"
17#include "SkTArray.h"
18#include "PictureRenderer.h"
19#include "picture_utils.h"
20
21static void usage(const char* argv0) {
22    SkDebugf("SkPicture rendering tool\n");
23    SkDebugf("\n"
24"Usage: \n"
25"     %s <input>... <outputDir> \n"
26"     [--mode pipe | simple | tile width[%] height[%]]"
27, argv0);
28    SkDebugf("\n\n");
29    SkDebugf(
30"     input:     A list of directories and files to use as input. Files are\n"
31"                expected to have the .skp extension.\n\n");
32    SkDebugf(
33"     outputDir: directory to write the rendered images.\n\n");
34    SkDebugf(
35"     --mode pipe | simple | tile width[%] height[%]: Run in the\n"
36"                corresponding mode. Default is simple.\n");
37    SkDebugf(
38"                     pipe, Render using a SkGPipe.\n");
39    SkDebugf(
40"                     simple, Render using the default rendering method.\n");
41    SkDebugf(
42"                     tile width[%] height[%], Do a simple render using tiles\n"
43"                                              with the given dimensions.\n");
44}
45
46static void make_output_filepath(SkString* path, const SkString& dir,
47                                 const SkString& name) {
48    sk_tools::make_filepath(path, dir, name);
49    path->remove(path->size() - 3, 3);
50    path->append("png");
51}
52
53/* since PNG insists on unpremultiplying our alpha, we take no precision chances
54    and force all pixels to be 100% opaque, otherwise on compare we may not get
55    a perfect match.
56 */
57static void force_all_opaque(const SkBitmap& bitmap) {
58    SkAutoLockPixels lock(bitmap);
59    for (int y = 0; y < bitmap.height(); y++) {
60        for (int x = 0; x < bitmap.width(); x++) {
61            *bitmap.getAddr32(x, y) |= (SK_A32_MASK << SK_A32_SHIFT);
62        }
63    }
64}
65
66static bool write_bitmap(const SkString& path, const SkBitmap& bitmap) {
67    SkBitmap copy;
68    bitmap.copyTo(&copy, SkBitmap::kARGB_8888_Config);
69    force_all_opaque(copy);
70    return SkImageEncoder::EncodeFile(path.c_str(), copy,
71                                      SkImageEncoder::kPNG_Type, 100);
72}
73
74static void write_output(const SkString& outputDir, const SkString& inputFilename,
75                         const SkBitmap& bitmap) {
76    SkString outputPath;
77    make_output_filepath(&outputPath, outputDir, inputFilename);
78    bool isWritten = write_bitmap(outputPath, bitmap);
79    if (!isWritten) {
80        SkDebugf("Could not write to file %s\n", outputPath.c_str());
81    }
82}
83
84static void render_picture(const SkString& inputPath, const SkString& outputDir,
85                           sk_tools::PictureRenderer& renderer) {
86    SkString inputFilename;
87    sk_tools::get_basename(&inputFilename, inputPath);
88
89    SkFILEStream inputStream;
90    inputStream.setPath(inputPath.c_str());
91    if (!inputStream.isValid()) {
92        SkDebugf("Could not open file %s\n", inputPath.c_str());
93        return;
94    }
95
96    SkPicture picture(&inputStream);
97    SkBitmap bitmap;
98    sk_tools::setup_bitmap(&bitmap, picture.width(), picture.height());
99    SkCanvas canvas(bitmap);
100
101    renderer.init(picture);
102    renderer.render(&picture, &canvas);
103    write_output(outputDir, inputFilename, bitmap);
104}
105
106static void process_input(const SkString& input, const SkString& outputDir,
107                          sk_tools::PictureRenderer& renderer) {
108    SkOSFile::Iter iter(input.c_str(), "skp");
109    SkString inputFilename;
110
111    if (iter.next(&inputFilename)) {
112        do {
113            SkString inputPath;
114            sk_tools::make_filepath(&inputPath, input, inputFilename);
115            render_picture(inputPath, outputDir, renderer);
116        } while(iter.next(&inputFilename));
117    } else {
118        SkString inputPath(input);
119        render_picture(inputPath, outputDir, renderer);
120    }
121}
122
123static void parse_commandline(int argc, char* const argv[], SkTArray<SkString>* inputs,
124                              sk_tools::PictureRenderer*& renderer){
125    const char* argv0 = argv[0];
126    char* const* stop = argv + argc;
127
128    for (++argv; argv < stop; ++argv) {
129        if (0 == strcmp(*argv, "--mode")) {
130            SkDELETE(renderer);
131
132            ++argv;
133            if (argv >= stop) {
134                SkDebugf("Missing mode for --mode\n");
135                usage(argv0);
136                exit(-1);
137            }
138
139            if (0 == strcmp(*argv, "pipe")) {
140                renderer = SkNEW(sk_tools::PipePictureRenderer);
141            } else if (0 == strcmp(*argv, "simple")) {
142                renderer = SkNEW(sk_tools::SimplePictureRenderer);
143            } else if (0 == strcmp(*argv, "tile")) {
144                sk_tools::TiledPictureRenderer* tileRenderer =
145                    SkNEW(sk_tools::TiledPictureRenderer);
146                ++argv;
147                if (argv >= stop) {
148                    SkDELETE(tileRenderer);
149                    SkDebugf("Missing width for --mode tile\n");
150                    usage(argv0);
151                    exit(-1);
152                }
153
154                if (sk_tools::is_percentage(*argv)) {
155                    tileRenderer->setTileWidthPercentage(atof(*argv));
156                    if (!(tileRenderer->getTileWidthPercentage() > 0)) {
157                        SkDELETE(tileRenderer);
158                        SkDebugf("--mode tile must be given a width percentage > 0\n");
159                        exit(-1);
160                    }
161                } else {
162                    tileRenderer->setTileWidth(atoi(*argv));
163                    if (!(tileRenderer->getTileWidth() > 0)) {
164                        SkDELETE(tileRenderer);
165                        SkDebugf("--mode tile must be given a width > 0\n");
166                        exit(-1);
167                    }
168                }
169
170                ++argv;
171                if (argv >= stop) {
172                    SkDELETE(tileRenderer);
173                    SkDebugf("Missing height for --mode tile\n");
174                    usage(argv0);
175                    exit(-1);
176                }
177
178                if (sk_tools::is_percentage(*argv)) {
179                    tileRenderer->setTileHeightPercentage(atof(*argv));
180                    if (!(tileRenderer->getTileHeightPercentage() > 0)) {
181                        SkDELETE(tileRenderer);
182                        SkDebugf(
183                            "--mode tile must be given a height percentage > 0\n");
184                        exit(-1);
185                    }
186                } else {
187                    tileRenderer->setTileHeight(atoi(*argv));
188                    if (!(tileRenderer->getTileHeight() > 0)) {
189                        SkDELETE(tileRenderer);
190                        SkDebugf("--mode tile must be given a height > 0\n");
191                        exit(-1);
192                    }
193                }
194
195                renderer = tileRenderer;
196            } else {
197                SkDebugf("%s is not a valid mode for --mode\n", *argv);
198                usage(argv0);
199                exit(-1);
200            }
201        } else if ((0 == strcmp(*argv, "-h")) || (0 == strcmp(*argv, "--help"))) {
202            SkDELETE(renderer);
203            usage(argv0);
204            exit(-1);
205        } else {
206            inputs->push_back(SkString(*argv));
207        }
208    }
209
210    if (inputs->count() < 2) {
211        SkDELETE(renderer);
212        usage(argv0);
213        exit(-1);
214    }
215
216    if (NULL == renderer) {
217        renderer = SkNEW(sk_tools::SimplePictureRenderer);
218    }
219}
220
221int main(int argc, char* const argv[]) {
222    SkTArray<SkString> inputs;
223    sk_tools::PictureRenderer* renderer = NULL;
224
225    parse_commandline(argc, argv, &inputs, renderer);
226    SkString outputDir = inputs[inputs.count() - 1];
227    SkASSERT(renderer);
228
229    for (int i = 0; i < inputs.count() - 1; i ++) {
230        process_input(inputs[i], outputDir, *renderer);
231    }
232
233    SkDELETE(renderer);
234}
235