123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 |
- /*
- * 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;
- }
|