| Index: tools/lua/ngrams.lua
|
| diff --git a/tools/lua/ngrams.lua b/tools/lua/ngrams.lua
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..c94ffb3b38fd5da51e57d2986557c1adaf4275b9
|
| --- /dev/null
|
| +++ b/tools/lua/ngrams.lua
|
| @@ -0,0 +1,62 @@
|
| +-- 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://skia-tree-status.appspot.com/skia-telemetry/lua_script,
|
| +-- and paste the contents of ngrams_aggregate.lua into the "aggregator script"
|
| +-- box on the same page.
|
| +
|
| +-- Change n as desired.
|
| +local n = 3
|
| +
|
| +-- 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
|
|
|