From f97c16f984db3cb901fafc7ca8bdc592b4e6d4ad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=81=95=E5=B1=B1?= Date: Tue, 11 Feb 2025 18:07:03 +0800 Subject: [PATCH 1/3] add python ASTPrint example --- example/python/ASTPrint.gdl | 90 +++++++++++++++++++++++++++++++++++++ 1 file changed, 90 insertions(+) create mode 100644 example/python/ASTPrint.gdl diff --git a/example/python/ASTPrint.gdl b/example/python/ASTPrint.gdl new file mode 100644 index 00000000..7dff8d2b --- /dev/null +++ b/example/python/ASTPrint.gdl @@ -0,0 +1,90 @@ +// script +use coref::python::* + +fn default_db() -> PythonDB { + return PythonDB::load("coref_python_src.db") +} + +// 递归查找以a为起点的语法树边，b 和 c 组成一条有向边，b 为父节点，c 为子节点，index 为这条边在树中距离 root 的层级 +fn searchEdge(a: CombineElement, b: CombineElement, c: CombineElement, index: int) -> bool { + if (a = b.getAnAncestorForIndex(index) || (b = a && index = 0)) { + if (b = c.getParent()) { + return true + } + } +} + +// 获取位置信息，如果该节点在 ast parser 中不存在位置信息，则会递归展示该节点的父节点的位置信息， +// 例如 Comprehension，Arguments，Withitem, DocstringComment 等类型 +fn getLoc(p: CombineElement, line: int, col: int) -> bool { + return line = p.getLocation().getStartLineNumber() && + col = p.getLocation().getStartColumnNumber() +} + +// 输出AST语法树的有向边，以及点的代码片段 +// 第一列是层数，从0开始 +// 第二列是当前边的父节点 +// 第三列是父节点的节点类型 +// 第四列是父节点的代码片段 +// 第五列是父节点起始行号 +// 第六列是父节点起始列号 +// 第七列是当前边的子节点 +// 第八列是子节点的节点类型 +// 第九列是子节点的代码片段 +// 第十列是子节点起始行号 +// 第十一列是子节点起始列号 +@output +fn out(filePath: string, + depth: int, + parent: CombineElement, + parentKind: string, + parentContent: string, + parentLine: int, + parentColumn: int, + child: CombineElement, + childKind: string, + childContent: string, + childLine: int, + childColumn: int) -> bool { + for (a in File(default_db())) { + if (filePath = a.getRelativePath() && + searchEdge(CombineElement(default_db()).find(a), parent, child, depth) && + childContent = child.print() && // 输出子节点的内容 + shortPrint(parent, parentContent) && // 优化输出父节点内容 + parentKind = parent.getType() && + childKind = child.getType() && + getLoc(parent, parentLine, parentColumn) && + getLoc(child, childLine, childColumn)) { + return true + } + } +} + +// 找到长度在 5 行以上的Expression, 可以调整 +fn isLongExpression(s: CombineElement) -> bool { + return Expression(default_db()).find(s).getSize().getNumberOfTotalLines() > 4 +} + +// 优化输出父节点 +fn shortPrint(p: CombineElement, n: string) -> bool { + if (isStatement(p)) { + return n = p.getType() + } + if (!isStatement(p)) { + if (isLongExpression(p)) { + return n = p.getType() + } + if (!isLongExpression(p)) { + return n = p.print() + } + } +} + +// 找到属于Statement的节点 +fn isStatement(s: CombineElement) -> bool { + for (b in Statement(default_db())) { + if (b.key_eq(s)) { + return true + } + } +} From 27e6f751e6c75ca0255d0efcca6e63ae3f09b110 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=81=95=E5=B1=B1?= Date: Wed, 26 Mar 2025 11:49:06 +0800 Subject: [PATCH 2/3] enable stats in souffle profiling log --- godel-script/godel-frontend/src/engine.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/godel-script/godel-frontend/src/engine.cpp b/godel-script/godel-frontend/src/engine.cpp index 84fc25c4..461fde86 100644 --- a/godel-script/godel-frontend/src/engine.cpp +++ b/godel-script/godel-frontend/src/engine.cpp @@ -397,8 +397,12 @@ void engine::run_souffle(const std::string& souffle_content, if (config.count(option::cli_enable_souffle_profiling)) { argv.push_back("--profile=souffle.prof.log"); argv.push_back("--profile-frequency"); + argv.push_back("--index-stats"); } + // enable souffle auto schedule, for experimental use only + // argv.push_back("--auto-schedule=souffle.prof.log"); + // null terminator argv.push_back(nullptr); From dbae57a84b577da87b50b777342eb056feb50866 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=81=95=E5=B1=B1?= Date: Thu, 17 Apr 2025 12:04:35 +0800 Subject: [PATCH 3/3] fix logic issue in godelscript documents --- doc/4_godelscript_language.en.md | 12 ++++++++++-- doc/4_godelscript_language.md | 12 ++++++++++-- 2 files changed, 20 insertions(+), 4 deletions(-) diff --git a/doc/4_godelscript_language.en.md b/doc/4_godelscript_language.en.md index 02b90462..7f6425d9 100644 --- a/doc/4_godelscript_language.en.md +++ b/doc/4_godelscript_language.en.md @@ -1172,10 +1172,18 @@ fn default_java_db() -> JavaDB { return JavaDB::load("coref_java_src.db") } +fn usedMethod(m: Method) -> bool { + for(c in CallableBinding(default_java_db())) { + if (c.getCallee().key_eq(m)) { + return true + } + } +} + // find unused methods fn unused_method(unused: string) -> bool { - for(c in Callable(default_java_db()), method in Callable(default_java_db()), caller in method.getCaller()) { - if (c != caller && unused = method.getSignature()) { + for(m in Method(default_java_db())) { + if (!usedMethod(m) && unused = m.getSignature()) { return true } } diff --git a/doc/4_godelscript_language.md b/doc/4_godelscript_language.md index 9e24a3c4..8d73cad0 100644 --- a/doc/4_godelscript_language.md +++ b/doc/4_godelscript_language.md @@ -1172,10 +1172,18 @@ fn default_java_db() -> JavaDB { return JavaDB::load("coref_java_src.db") } +fn usedMethod(m: Method) -> bool { + for(c in CallableBinding(default_java_db())) { + if (c.getCallee().key_eq(m)) { + return true + } + } +} + // find unused methods fn unused_method(unused: string) -> bool { - for(c in Callable(default_java_db()), method in Callable(default_java_db()), caller in method.getCaller()) { - if (c != caller && unused = method.getSignature()) { + for(m in Method(default_java_db())) { + if (!usedMethod(m) && unused = m.getSignature()) { return true } }