skia2/tools/lua/ngrams.lua
Ravi Mistry 4c0ffe7820 [Tree Status] Update all references in Skia repo to tree-status
* Mainly updates to documentation.
* Fix some comments in PRESUBMIT.py.
* Delete tools/lua/trigger_ct_lua. It has not worked in many years.

Bug: skia:9962
Change-Id: If6f58f173f2c8bd3fc9bdfc4db440f42489fee08
Docs-Preview: https://skia.org/?cl=274597
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/274597
Commit-Queue: Ravi Mistry <rmistry@google.com>
Reviewed-by: Eric Boren <borenet@google.com>
2020-03-02 18:43:49 +00:00

66 lines
2.0 KiB
Lua

-- Generate n-grams of Skia API calls from SKPs.
-- To test this locally, run:
-- $ GYP_DEFINES="skia_shared_lib=1" make lua_pictures
-- $ out/Debug/lua_pictures -q -r $SKP_DIR -l tools/lua/ngrams.lua > /tmp/lua-output
-- $ lua tools/lua/ngrams_aggregate.lua
-- To run on Cluster Telemetry, copy and paste the contents of this file into
-- the box at https://ct.skia.org/lua_script/,
-- and paste the contents of ngrams_aggregate.lua into the "aggregator script"
-- box on the same page.
-- Change n as desired.
-- CHANGEME
local n = 3
-- CHANGEME
-- This algorithm uses a list-of-lists for each SKP. For API call, append a
-- list containing just the verb to the master list. Then, backtrack over the
-- last (n-1) sublists in the master list and append the verb to those
-- sublists. At the end of execution, the master list contains a sublist for
-- every verb in the SKP file. Each sublist has length n, with the exception of
-- the last n-1 sublists, which are discarded in the summarize() function,
-- which generates counts for each n-gram.
local ngrams = {}
local currentFile = ""
function sk_scrape_startcanvas(c, fileName)
currentFile = fileName
ngrams[currentFile] = {}
end
function sk_scrape_endcanvas(c, fileName)
end
function sk_scrape_accumulate(t)
table.insert(ngrams[currentFile], {t.verb})
for i = 1, n-1 do
local idx = #ngrams[currentFile] - i
if idx > 0 then
table.insert(ngrams[currentFile][idx], t.verb)
end
end
end
function sk_scrape_summarize()
-- Count the n-grams.
local counts = {}
for file, ngramsInFile in pairs(ngrams) do
for i = 1, #ngramsInFile - (n-1) do
local ngram = table.concat(ngramsInFile[i], " ")
if counts[ngram] == nil then
counts[ngram] = 1
else
counts[ngram] = counts[ngram] + 1
end
end
end
-- Write out code for aggregating.
for ngram, count in pairs(counts) do
io.write("if counts['", ngram, "'] == nil then counts['", ngram, "'] = ", count, " else counts['", ngram, "'] = counts['", ngram, "'] + ", count, " end\n")
end
end