/* * Copyright 2013 Google Inc. * * Use of this source code is governed by a BSD-style license that can be * found in the LICENSE file. */ #include "include/core/SkData.h" #include "include/core/SkGraphics.h" #include "include/core/SkPicture.h" #include "include/core/SkStream.h" #include "include/utils/SkLua.h" #include "include/utils/SkLuaCanvas.h" #include "src/core/SkOSFile.h" #include "src/utils/SkOSPath.h" #include "tools/flags/CommandLineFlags.h" #include <stdlib.h> extern "C" { #include "lua.h" #include "lualib.h" #include "lauxlib.h" } static const char gStartCanvasFunc[] = "sk_scrape_startcanvas"; static const char gEndCanvasFunc[] = "sk_scrape_endcanvas"; static const char gAccumulateFunc[] = "sk_scrape_accumulate"; static const char gSummarizeFunc[] = "sk_scrape_summarize"; // Example usage for the modulo flag: // for i in {0..5}; do lua_pictures --skpPath SKP_PATH -l YOUR_SCRIPT --modulo $i 6 &; done static DEFINE_string(modulo, "", "[--modulo <remainder> <divisor>]: only run tests for which " "testIndex %% divisor == remainder."); static DEFINE_string2(skpPath, r, "", "Read this .skp file or .skp files from this dir"); static DEFINE_string2(luaFile, l, "", "File containing lua script to run"); static DEFINE_string2(headCode, s, "", "Optional lua code to call at beginning"); static DEFINE_string2(tailFunc, s, "", "Optional lua function to call at end"); static DEFINE_bool2(quiet, q, false, "Silence all non-error related output"); static sk_sp<SkPicture> load_picture(const char path[]) { std::unique_ptr<SkStream> stream = SkStream::MakeFromFile(path); if (stream) { return SkPicture::MakeFromStream(stream.get()); } return nullptr; } static void call_canvas(lua_State* L, SkLuaCanvas* canvas, const char pictureFile[], const char funcName[]) { lua_getglobal(L, funcName); if (!lua_isfunction(L, -1)) { int t = lua_type(L, -1); SkDebugf("--- expected %s function %d, ignoring.\n", funcName, t); lua_settop(L, -2); } else { canvas->pushThis(); lua_pushstring(L, pictureFile); if (lua_pcall(L, 2, 0, 0) != LUA_OK) { SkDebugf("lua err: %s\n", lua_tostring(L, -1)); } } } int main(int argc, char** argv) { CommandLineFlags::SetUsage("apply lua script to .skp files."); CommandLineFlags::Parse(argc, argv); if (FLAGS_skpPath.isEmpty()) { SkDebugf(".skp files or directories are required.\n"); exit(-1); } if (FLAGS_luaFile.isEmpty()) { SkDebugf("missing luaFile(s)\n"); exit(-1); } const char* summary = gSummarizeFunc; if (!FLAGS_tailFunc.isEmpty()) { summary = FLAGS_tailFunc[0]; } SkAutoGraphics ag; SkLua L(summary); for (int i = 0; i < FLAGS_luaFile.count(); ++i) { sk_sp<SkData> data(SkData::MakeFromFileName(FLAGS_luaFile[i])); if (!data) { data = SkData::MakeEmpty(); } if (!FLAGS_quiet) { SkDebugf("loading %s...\n", FLAGS_luaFile[i]); } if (!L.runCode(data->data(), data->size())) { SkDebugf("failed to load luaFile %s\n", FLAGS_luaFile[i]); exit(-1); } } if (!FLAGS_headCode.isEmpty()) { L.runCode(FLAGS_headCode[0]); } int moduloRemainder = -1; int moduloDivisor = -1; SkString moduloStr; if (FLAGS_modulo.count() == 2) { moduloRemainder = atoi(FLAGS_modulo[0]); moduloDivisor = atoi(FLAGS_modulo[1]); if (moduloRemainder < 0 || moduloDivisor <= 0 || moduloRemainder >= moduloDivisor) { SkDebugf("invalid modulo values.\n"); return -1; } } for (int i = 0; i < FLAGS_skpPath.count(); i ++) { SkTArray<SkString> paths; if (sk_isdir(FLAGS_skpPath[i])) { // Add all .skp in this directory. const SkString directory(FLAGS_skpPath[i]); SkString filename; SkOSFile::Iter iter(FLAGS_skpPath[i], "skp"); while(iter.next(&filename)) { paths.push_back() = SkOSPath::Join(directory.c_str(), filename.c_str()); } } else { // Add this as an .skp itself. paths.push_back() = FLAGS_skpPath[i]; } for (int i = 0; i < paths.count(); i++) { if (moduloRemainder >= 0) { if ((i % moduloDivisor) != moduloRemainder) { continue; } moduloStr.printf("[%d.%d] ", i, moduloDivisor); } const char* path = paths[i].c_str(); if (!FLAGS_quiet) { SkDebugf("scraping %s %s\n", path, moduloStr.c_str()); } auto pic(load_picture(path)); if (pic.get()) { std::unique_ptr<SkLuaCanvas> canvas( new SkLuaCanvas(SkScalarCeilToInt(pic->cullRect().width()), SkScalarCeilToInt(pic->cullRect().height()), L.get(), gAccumulateFunc)); call_canvas(L.get(), canvas.get(), path, gStartCanvasFunc); canvas->drawPicture(pic); call_canvas(L.get(), canvas.get(), path, gEndCanvasFunc); } else { SkDebugf("failed to load\n"); } } } return 0; }