cppgolf 0.1.0__tar.gz

cppgolf-0.1.0/PKG-INFO

@@ -0,0 +1,97 @@
+Metadata-Version: 2.4
+Name: cppgolf
+Version: 0.1.0
+Summary: C++ multi-file merge & code golf / minifier tool
+License: MIT
+Project-URL: Homepage, https://github.com/yourname/cppgolf
+Project-URL: Issues, https://github.com/yourname/cppgolf/issues
+Keywords: cpp,c++,golf,minify,code-golf,competitive-programming
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Code Generators
+Classifier: Topic :: Text Processing :: Filters
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: tree-sitter>=0.25
+Requires-Dist: tree-sitter-cpp>=0.23
+Provides-Extra: dev
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+
+# CPPGolf
+
+C++ 多文件合并 + 代码高尔夫（压缩）工具，专为竞技编程场景设计。
+
+## 安装
+
+```bash
+pip install cppgolf
+```
+
+## CLI 用法
+
+```bash
+cppgolf solution.cpp                       # 输出到 stdout
+cppgolf solution.cpp -o golf.cpp           # 输出到文件
+cppgolf solution.cpp -I include/ -o out.cpp
+cppgolf solution.cpp --rename              # 符号压缩
+```
+
+### 选项
+
+| 选项 | 说明 |
+|------|------|
+| `-o FILE` | 输出文件（默认 stdout） |
+| `-I DIR` | 追加 include 搜索目录（可多次） |
+| `--no-merge` | 跳过 `#include "..."` 内联 |
+| `--no-strip-comments` | 保留注释 |
+| `--no-compress-ws` | 保留空白格式 |
+| `--no-std-ns` | 不添加 `using namespace std` |
+| `--no-typedefs` | 不添加 `ll`/`ld` 等类型宏 |
+| `--keep-main-return` | 保留 `return 0;` |
+| `--keep-endl` | 保留 `endl` |
+| `--keep-inline` | 保留 `inline` 关键字 |
+| `--aggressive` | 去除单语句 if/for/while 花括号 |
+| `--shortcuts` | 高频 cout/cin → `#define` 缩写 |
+| `--rename` | 变量/成员名压缩为 a/b/aa/… |
+| `--stats` | 显示压缩率统计 |
+
+## Python API
+
+```python
+from cppgolf import process
+from pathlib import Path
+
+result = process(
+    Path("solution.cpp"),
+    include_dirs=[],          # 额外的 #include 搜索目录
+    rename_symbols=True,
+)
+print(result)
+```
+
+也可单独使用各 pass：
+
+```python
+from cppgolf import strip_comments, compress_whitespace, golf_rename_symbols
+
+code = open("a.cpp").read()
+code = strip_comments(code)
+code = golf_rename_symbols(code)
+code = compress_whitespace(code)
+```
+
+## 功能说明
+
+- **合并**：递归内联 `#include "..."` 本地头文件，去除 include guard / `#pragma once`，系统头去重
+- **去注释**：状态机感知字符串，支持 `//`、`/* */`、原始字符串 `R"(...)"` 
+- **语义压缩**：`std::` 消除、`long long→ll` 宏、`endl→"\n"`、去 `return 0;`、去 `inline`
+- **空白压缩**：token 级最小化，代码压为单行，预处理行保留换行
+- **符号重命名**：tree-sitter AST 驱动，仅重命名用户自定义变量/参数/成员名，不碰函数名/类型名/命名空间

cppgolf-0.1.0/README.md

@@ -0,0 +1,70 @@
+# CPPGolf
+
+C++ 多文件合并 + 代码高尔夫（压缩）工具，专为竞技编程场景设计。
+
+## 安装
+
+```bash
+pip install cppgolf
+```
+
+## CLI 用法
+
+```bash
+cppgolf solution.cpp                       # 输出到 stdout
+cppgolf solution.cpp -o golf.cpp           # 输出到文件
+cppgolf solution.cpp -I include/ -o out.cpp
+cppgolf solution.cpp --rename              # 符号压缩
+```
+
+### 选项
+
+| 选项 | 说明 |
+|------|------|
+| `-o FILE` | 输出文件（默认 stdout） |
+| `-I DIR` | 追加 include 搜索目录（可多次） |
+| `--no-merge` | 跳过 `#include "..."` 内联 |
+| `--no-strip-comments` | 保留注释 |
+| `--no-compress-ws` | 保留空白格式 |
+| `--no-std-ns` | 不添加 `using namespace std` |
+| `--no-typedefs` | 不添加 `ll`/`ld` 等类型宏 |
+| `--keep-main-return` | 保留 `return 0;` |
+| `--keep-endl` | 保留 `endl` |
+| `--keep-inline` | 保留 `inline` 关键字 |
+| `--aggressive` | 去除单语句 if/for/while 花括号 |
+| `--shortcuts` | 高频 cout/cin → `#define` 缩写 |
+| `--rename` | 变量/成员名压缩为 a/b/aa/… |
+| `--stats` | 显示压缩率统计 |
+
+## Python API
+
+```python
+from cppgolf import process
+from pathlib import Path
+
+result = process(
+    Path("solution.cpp"),
+    include_dirs=[],          # 额外的 #include 搜索目录
+    rename_symbols=True,
+)
+print(result)
+```
+
+也可单独使用各 pass：
+
+```python
+from cppgolf import strip_comments, compress_whitespace, golf_rename_symbols
+
+code = open("a.cpp").read()
+code = strip_comments(code)
+code = golf_rename_symbols(code)
+code = compress_whitespace(code)
+```
+
+## 功能说明
+
+- **合并**：递归内联 `#include "..."` 本地头文件，去除 include guard / `#pragma once`，系统头去重
+- **去注释**：状态机感知字符串，支持 `//`、`/* */`、原始字符串 `R"(...)"` 
+- **语义压缩**：`std::` 消除、`long long→ll` 宏、`endl→"\n"`、去 `return 0;`、去 `inline`
+- **空白压缩**：token 级最小化，代码压为单行，预处理行保留换行
+- **符号重命名**：tree-sitter AST 驱动，仅重命名用户自定义变量/参数/成员名，不碰函数名/类型名/命名空间

cppgolf-0.1.0/cppgolf/__init__.py

@@ -0,0 +1,46 @@
+"""
+cppgolf — C++ multi-file merge & code golf tool
+
+公开 API：
+    process(input_file, include_dirs, **options) -> str
+    golf_rename_symbols(code) -> str
+    strip_comments(code) -> str
+    merge_files(filepath, include_dirs, visited, sys_includes) -> str
+    compress_whitespace(code) -> str
+    golf_std_namespace / golf_typedefs / golf_endl_to_newline /
+    golf_remove_main_return / golf_remove_inline /
+    golf_braces_single_stmt / golf_define_shortcuts
+"""
+
+from .strip_comments import strip_comments
+from .merge import merge_files, strip_include_guard
+from .whitespace import compress_whitespace
+from .transforms import (
+    golf_std_namespace,
+    golf_typedefs,
+    golf_remove_main_return,
+    golf_endl_to_newline,
+    golf_remove_inline,
+    golf_braces_single_stmt,
+    golf_define_shortcuts,
+)
+from .golf_rename import golf_rename_symbols
+from .__main__ import process
+
+__all__ = [
+    "process",
+    "strip_comments",
+    "merge_files",
+    "strip_include_guard",
+    "compress_whitespace",
+    "golf_std_namespace",
+    "golf_typedefs",
+    "golf_remove_main_return",
+    "golf_endl_to_newline",
+    "golf_remove_inline",
+    "golf_braces_single_stmt",
+    "golf_define_shortcuts",
+    "golf_rename_symbols",
+]
+
+__version__ = "0.1.0"

cppgolf-0.1.0/cppgolf/__main__.py

@@ -0,0 +1,148 @@
+"""cppgolf.__main__ — CLI 入口，支持 python -m cppgolf 和 cppgolf 命令"""
+import sys
+import argparse
+from pathlib import Path
+
+from .strip_comments import strip_comments
+from .merge import merge_files
+from .whitespace import compress_whitespace
+from .transforms import (
+    golf_std_namespace, golf_typedefs, golf_remove_main_return,
+    golf_endl_to_newline, golf_remove_inline,
+    golf_braces_single_stmt, golf_define_shortcuts,
+)
+from .golf_rename import golf_rename_symbols
+
+
+def process(
+    input_file: Path,
+    include_dirs: list,
+    *,
+    no_merge: bool = False,
+    no_strip_comments: bool = False,
+    no_compress_ws: bool = False,
+    no_std_ns: bool = False,
+    no_typedefs: bool = False,
+    keep_main_return: bool = False,
+    keep_endl: bool = False,
+    keep_inline: bool = False,
+    aggressive: bool = False,
+    define_shortcuts: bool = False,
+    rename_symbols: bool = False,
+) -> str:
+    sys_includes: list = []
+    visited: set = set()
+
+    if not no_merge:
+        merged = merge_files(input_file, list(include_dirs), visited, sys_includes)
+        code = ''.join(sys_includes) + merged
+    else:
+        code = input_file.read_text(encoding='utf-8-sig', errors='replace')
+
+    if not no_strip_comments:
+        code = strip_comments(code)
+
+    if not keep_endl:
+        code = golf_endl_to_newline(code)
+    if not no_std_ns:
+        code = golf_std_namespace(code)
+    if not no_typedefs:
+        code = golf_typedefs(code)
+    if not keep_main_return:
+        code = golf_remove_main_return(code)
+    if not keep_inline:
+        code = golf_remove_inline(code)
+    if aggressive:
+        code = golf_braces_single_stmt(code)
+    if define_shortcuts:
+        code = golf_define_shortcuts(code)
+    if rename_symbols:
+        code = golf_rename_symbols(code)
+
+    if not no_compress_ws:
+        code = compress_whitespace(code)
+
+    return code.strip() + '\n'
+
+
+def build_parser() -> argparse.ArgumentParser:
+    p = argparse.ArgumentParser(
+        prog='cppgolf',
+        description='C++ 多文件合并 + 代码高尔夫工具',
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""示例：
+  cppgolf solution.cpp
+  cppgolf solution.cpp -o golf.cpp
+  cppgolf solution.cpp -I include/ --rename --stats
+""",
+    )
+    p.add_argument('input', type=Path, help='入口 C++ 文件')
+    p.add_argument('-o', '--output', type=Path, default=None, help='输出文件（默认 stdout）')
+    p.add_argument('-I', '--include', dest='include_dirs', action='append',
+                   type=Path, default=[], metavar='DIR', help='追加 include 目录（可多次）')
+
+    g = p.add_argument_group('功能开关（默认全部开启）')
+    g.add_argument('--no-merge',          action='store_true', help='跳过多文件合并')
+    g.add_argument('--no-strip-comments', action='store_true', help='保留注释')
+    g.add_argument('--no-compress-ws',    action='store_true', help='保留空白格式')
+    g.add_argument('--no-std-ns',         action='store_true', help='不添加 using namespace std')
+    g.add_argument('--no-typedefs',       action='store_true', help='不添加 ll/ld 等类型宏')
+    g.add_argument('--keep-main-return',  action='store_true', help='保留 main 末尾 return 0')
+    g.add_argument('--keep-endl',         action='store_true', help='保留 endl')
+    g.add_argument('--keep-inline',       action='store_true', help='保留 inline 关键字')
+
+    g2 = p.add_argument_group('激进优化（有风险，默认关闭）')
+    g2.add_argument('--aggressive', action='store_true',
+                    help='单语句 if/for/while 去花括号')
+    g2.add_argument('--shortcuts', dest='define_shortcuts', action='store_true',
+                    help='高频 cout/cin 用 #define 缩写')
+    g2.add_argument('--rename', dest='rename_symbols', action='store_true',
+                    help='将用户变量/成员名压缩为短名（需要 tree-sitter-cpp）')
+
+    p.add_argument('--stats', action='store_true', help='显示压缩率统计')
+    return p
+
+
+def main():
+    parser = build_parser()
+    args = parser.parse_args()
+
+    if not args.input.exists():
+        print(f'错误：文件不存在 —— {args.input}', file=sys.stderr)
+        sys.exit(1)
+
+    original_size = args.input.stat().st_size
+    result = process(
+        args.input, args.include_dirs,
+        no_merge=args.no_merge,
+        no_strip_comments=args.no_strip_comments,
+        no_compress_ws=args.no_compress_ws,
+        no_std_ns=args.no_std_ns,
+        no_typedefs=args.no_typedefs,
+        keep_main_return=args.keep_main_return,
+        keep_endl=args.keep_endl,
+        keep_inline=args.keep_inline,
+        aggressive=args.aggressive,
+        define_shortcuts=args.define_shortcuts,
+        rename_symbols=args.rename_symbols,
+    )
+
+    def print_stats(final_size: int):
+        ratio = (1 - final_size / original_size) * 100 if original_size else 0
+        print(f'[统计] 原始：{original_size} B  →  高尔夫后：{final_size} B  （压缩 {ratio:.1f}%）',
+              file=sys.stderr)
+
+    if args.output:
+        args.output.write_text(result, encoding='utf-8')
+        if args.stats:
+            print_stats(args.output.stat().st_size)
+        else:
+            print(f'已写入：{args.output}', file=sys.stderr)
+    else:
+        if args.stats:
+            print_stats(len(result.encode('utf-8')))
+        sys.stdout.write(result)
+
+
+if __name__ == '__main__':
+    main()