reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2013 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 | |
reed@google.com | 3597b73 | 2013-05-22 20:12:50 +0000 | [diff] [blame] | 8 | #include "SkLua.h" |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 9 | #include "SkLuaCanvas.h" |
| 10 | #include "SkPicture.h" |
| 11 | #include "SkCommandLineFlags.h" |
| 12 | #include "SkGraphics.h" |
| 13 | #include "SkStream.h" |
| 14 | #include "SkData.h" |
| 15 | #include "picture_utils.h" |
| 16 | #include "SkOSFile.h" |
| 17 | #include "SkImageDecoder.h" |
| 18 | |
bungeman | 60e0fee | 2015-08-26 05:15:46 -0700 | [diff] [blame] | 19 | #include <stdlib.h> |
| 20 | |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 21 | extern "C" { |
| 22 | #include "lua.h" |
| 23 | #include "lualib.h" |
| 24 | #include "lauxlib.h" |
| 25 | } |
| 26 | |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 27 | static const char gStartCanvasFunc[] = "sk_scrape_startcanvas"; |
| 28 | static const char gEndCanvasFunc[] = "sk_scrape_endcanvas"; |
| 29 | static const char gAccumulateFunc[] = "sk_scrape_accumulate"; |
| 30 | static const char gSummarizeFunc[] = "sk_scrape_summarize"; |
| 31 | |
commit-bot@chromium.org | aaeb87d | 2013-06-21 21:45:20 +0000 | [diff] [blame] | 32 | // Example usage for the modulo flag: |
| 33 | // for i in {0..5}; do lua_pictures --skpPath SKP_PATH -l YOUR_SCRIPT --modulo $i 6 &; done |
| 34 | DEFINE_string(modulo, "", "[--modulo <remainder> <divisor>]: only run tests for which " |
| 35 | "testIndex %% divisor == remainder."); |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 36 | DEFINE_string2(skpPath, r, "", "Read this .skp file or .skp files from this dir"); |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 37 | DEFINE_string2(luaFile, l, "", "File containing lua script to run"); |
reed@google.com | e2aad27 | 2013-05-31 19:46:02 +0000 | [diff] [blame] | 38 | DEFINE_string2(headCode, s, "", "Optional lua code to call at beginning"); |
| 39 | DEFINE_string2(tailFunc, s, "", "Optional lua function to call at end"); |
commit-bot@chromium.org | bdc772e | 2014-03-19 19:23:17 +0000 | [diff] [blame] | 40 | DEFINE_bool2(quiet, q, false, "Silence all non-error related output"); |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 41 | |
| 42 | static SkPicture* load_picture(const char path[]) { |
scroggo | a1193e4 | 2015-01-21 12:09:53 -0800 | [diff] [blame] | 43 | SkAutoTDelete<SkStream> stream(SkStream::NewFromFile(path)); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 44 | SkPicture* pic = nullptr; |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 45 | if (stream.get()) { |
msarett | 8715d47 | 2016-02-17 10:02:29 -0800 | [diff] [blame^] | 46 | pic = SkPicture::CreateFromStream(stream.get()); |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 47 | } |
| 48 | return pic; |
| 49 | } |
| 50 | |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 51 | static void call_canvas(lua_State* L, SkLuaCanvas* canvas, |
| 52 | const char pictureFile[], const char funcName[]) { |
| 53 | lua_getglobal(L, funcName); |
mike@reedtribe.org | f02fe3d | 2013-05-21 12:20:39 +0000 | [diff] [blame] | 54 | if (!lua_isfunction(L, -1)) { |
| 55 | int t = lua_type(L, -1); |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 56 | SkDebugf("--- expected %s function %d, ignoring.\n", funcName, t); |
mike@reedtribe.org | f02fe3d | 2013-05-21 12:20:39 +0000 | [diff] [blame] | 57 | lua_settop(L, -2); |
| 58 | } else { |
| 59 | canvas->pushThis(); |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 60 | lua_pushstring(L, pictureFile); |
| 61 | if (lua_pcall(L, 2, 0, 0) != LUA_OK) { |
mike@reedtribe.org | f02fe3d | 2013-05-21 12:20:39 +0000 | [diff] [blame] | 62 | SkDebugf("lua err: %s\n", lua_tostring(L, -1)); |
| 63 | } |
| 64 | } |
| 65 | } |
| 66 | |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 67 | int tool_main(int argc, char** argv); |
| 68 | int tool_main(int argc, char** argv) { |
| 69 | SkCommandLineFlags::SetUsage("apply lua script to .skp files."); |
| 70 | SkCommandLineFlags::Parse(argc, argv); |
| 71 | |
| 72 | if (FLAGS_skpPath.isEmpty()) { |
| 73 | SkDebugf(".skp files or directories are required.\n"); |
| 74 | exit(-1); |
| 75 | } |
| 76 | if (FLAGS_luaFile.isEmpty()) { |
mike@reedtribe.org | 0e59b79 | 2013-05-21 03:24:37 +0000 | [diff] [blame] | 77 | SkDebugf("missing luaFile(s)\n"); |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 78 | exit(-1); |
| 79 | } |
| 80 | |
reed@google.com | e2aad27 | 2013-05-31 19:46:02 +0000 | [diff] [blame] | 81 | const char* summary = gSummarizeFunc; |
| 82 | if (!FLAGS_tailFunc.isEmpty()) { |
| 83 | summary = FLAGS_tailFunc[0]; |
| 84 | } |
| 85 | |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 86 | SkAutoGraphics ag; |
reed@google.com | e2aad27 | 2013-05-31 19:46:02 +0000 | [diff] [blame] | 87 | SkLua L(summary); |
mike@reedtribe.org | 0e59b79 | 2013-05-21 03:24:37 +0000 | [diff] [blame] | 88 | |
| 89 | for (int i = 0; i < FLAGS_luaFile.count(); ++i) { |
reed | 9594da1 | 2014-09-12 12:12:27 -0700 | [diff] [blame] | 90 | SkAutoDataUnref data(SkData::NewFromFileName(FLAGS_luaFile[i])); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 91 | if (nullptr == data.get()) { |
reed | 9594da1 | 2014-09-12 12:12:27 -0700 | [diff] [blame] | 92 | data.reset(SkData::NewEmpty()); |
| 93 | } |
commit-bot@chromium.org | bdc772e | 2014-03-19 19:23:17 +0000 | [diff] [blame] | 94 | if (!FLAGS_quiet) { |
| 95 | SkDebugf("loading %s...\n", FLAGS_luaFile[i]); |
| 96 | } |
reed@google.com | 3597b73 | 2013-05-22 20:12:50 +0000 | [diff] [blame] | 97 | if (!L.runCode(data->data(), data->size())) { |
mike@reedtribe.org | 0e59b79 | 2013-05-21 03:24:37 +0000 | [diff] [blame] | 98 | SkDebugf("failed to load luaFile %s\n", FLAGS_luaFile[i]); |
| 99 | exit(-1); |
| 100 | } |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 101 | } |
| 102 | |
reed@google.com | e2aad27 | 2013-05-31 19:46:02 +0000 | [diff] [blame] | 103 | if (!FLAGS_headCode.isEmpty()) { |
| 104 | L.runCode(FLAGS_headCode[0]); |
| 105 | } |
skia.committer@gmail.com | 26da7f0 | 2013-06-01 07:01:39 +0000 | [diff] [blame] | 106 | |
commit-bot@chromium.org | aaeb87d | 2013-06-21 21:45:20 +0000 | [diff] [blame] | 107 | int moduloRemainder = -1; |
| 108 | int moduloDivisor = -1; |
| 109 | SkString moduloStr; |
| 110 | |
| 111 | if (FLAGS_modulo.count() == 2) { |
| 112 | moduloRemainder = atoi(FLAGS_modulo[0]); |
| 113 | moduloDivisor = atoi(FLAGS_modulo[1]); |
| 114 | if (moduloRemainder < 0 || moduloDivisor <= 0 || moduloRemainder >= moduloDivisor) { |
| 115 | SkDebugf("invalid modulo values.\n"); |
| 116 | return -1; |
| 117 | } |
| 118 | } |
| 119 | |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 120 | for (int i = 0; i < FLAGS_skpPath.count(); i ++) { |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 121 | SkTArray<SkString> paths; |
| 122 | if (sk_isdir(FLAGS_skpPath[i])) { |
| 123 | // Add all .skp in this directory. |
| 124 | const SkString directory(FLAGS_skpPath[i]); |
| 125 | SkString filename; |
| 126 | SkOSFile::Iter iter(FLAGS_skpPath[i], "skp"); |
| 127 | while(iter.next(&filename)) { |
tfarina | a8e2e15 | 2014-07-28 19:26:58 -0700 | [diff] [blame] | 128 | paths.push_back() = SkOSPath::Join(directory.c_str(), filename.c_str()); |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 129 | } |
| 130 | } else { |
| 131 | // Add this as an .skp itself. |
| 132 | paths.push_back() = FLAGS_skpPath[i]; |
| 133 | } |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 134 | |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 135 | for (int i = 0; i < paths.count(); i++) { |
commit-bot@chromium.org | aaeb87d | 2013-06-21 21:45:20 +0000 | [diff] [blame] | 136 | if (moduloRemainder >= 0) { |
| 137 | if ((i % moduloDivisor) != moduloRemainder) { |
| 138 | continue; |
| 139 | } |
| 140 | moduloStr.printf("[%d.%d] ", i, moduloDivisor); |
| 141 | } |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 142 | const char* path = paths[i].c_str(); |
commit-bot@chromium.org | bdc772e | 2014-03-19 19:23:17 +0000 | [diff] [blame] | 143 | if (!FLAGS_quiet) { |
| 144 | SkDebugf("scraping %s %s\n", path, moduloStr.c_str()); |
| 145 | } |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 146 | |
mike@reedtribe.org | cef454e | 2013-05-21 01:49:06 +0000 | [diff] [blame] | 147 | SkAutoTUnref<SkPicture> pic(load_picture(path)); |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 148 | if (pic.get()) { |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 149 | SkAutoTUnref<SkLuaCanvas> canvas( |
robertphillips | a8d7f0b | 2014-08-29 08:03:56 -0700 | [diff] [blame] | 150 | new SkLuaCanvas(SkScalarCeilToInt(pic->cullRect().width()), |
| 151 | SkScalarCeilToInt(pic->cullRect().height()), |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 152 | L.get(), gAccumulateFunc)); |
mike@reedtribe.org | f02fe3d | 2013-05-21 12:20:39 +0000 | [diff] [blame] | 153 | |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 154 | call_canvas(L.get(), canvas.get(), path, gStartCanvasFunc); |
robertphillips | 9b14f26 | 2014-06-04 05:40:44 -0700 | [diff] [blame] | 155 | canvas->drawPicture(pic); |
commit-bot@chromium.org | 3da6c56 | 2013-06-18 18:35:58 +0000 | [diff] [blame] | 156 | call_canvas(L.get(), canvas.get(), path, gEndCanvasFunc); |
reed@google.com | 2d51677 | 2013-05-21 16:05:53 +0000 | [diff] [blame] | 157 | |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 158 | } else { |
mike@reedtribe.org | cef454e | 2013-05-21 01:49:06 +0000 | [diff] [blame] | 159 | SkDebugf("failed to load\n"); |
reed@google.com | dff7e11 | 2013-05-15 19:34:20 +0000 | [diff] [blame] | 160 | } |
| 161 | } |
| 162 | } |
| 163 | return 0; |
| 164 | } |
| 165 | |
| 166 | #if !defined SK_BUILD_FOR_IOS |
| 167 | int main(int argc, char * const argv[]) { |
| 168 | return tool_main(argc, (char**) argv); |
| 169 | } |
| 170 | #endif |