test_ast.py

# test.py
import json
from pathlib import Path
from src.backend.app.services import ScoreService
from src.scorelang.ast_score.nodes import ScoreDocumentNode
from src.scorelang.core.pipeline_context import PipelineContext


def main():
    sample_score_text = (
"""
# 最凉州
@ 沙陀调
% 来源：三五要录
% 录入：冯氏羊肉馆
% 审校：冯氏羊肉馆
% 日期：2025.10.12
= 凉或作梁 拍子廿 可弹四反 合拍子八十
= 终帖加拍子 南宫横笛谱云打三度拍子
= 中曲 古乐 南宫谱同    龙吟抄云内宴？
= 音声 用之
## 第一段
@ 沙陀调
{二}
{也/pz}
{七}
{言（七言）/pz}
{一/y/b/pz}
{之}
{四/pz}
{之}
{合（八）/h}
{之/hh/pz}
{七（三七）}
{之/hh}
{四/hh/pz}
{言（七言）/b}
{言/h}
{言（七）/hh/pz}
{言(七言)}
{八/pz}
{五}
{八/pz}
{二}
{也/pz}
{言（七）/b}
{七/h}
{七（三）/hh/pz}
{七（三七）/y/pz}
{之}
{四/h}
{七（三）/hh/pz}
{二}
{也/h}
{七（三）/hh/pz}
{二/b}
{也/pz}
{也/y/r}


"""
    )
    sample_score_text_2 = (
"""
# 唐琵琶谱数字化测试范例
@ 沙陀调
% 来源：三五要录
% 录入：冯氏羊肉馆
% 审校：冯氏羊肉馆
% 日期：2025.10.12
= 用于测试字体
= 这一行用于测试文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行文本换行换行
## 主谱字展示
= 这里展示所有的主谱字
{一}
{二}
{三}
{四}
= 去掉只拍子也能正常排版，顺便展示中间插入注释的效果
{五}
{六/pz}
{七}
{八/pz}
{九}
{十/pz}
{匕}
{卜/pz}
{敷}
{乙/pz}
{言}
{合/pz}
{斗}
{乞/pz}
{之}
{也/pz}


## 小谱字展示
= 这里展示所有的附加谱字和其排版效果
{一/h}
{二/hh}
{三/pz}
{四/y/pz}
{五/b}
{六（五六）/h/py}
{七/hh/pz}
{八/py}
{九/pz/py}
{乙/ls}
{三/le}
{七/f/h}
{三/hh/pz}

## 第一段
@ 沙陀调
{二}
{也/pz}
{七}
{言（七言）/pz}
{一/y/b/pz}
{之}
{四/pz}
{之}
{合（八）/h}
{之/hh/pz}
{七（三七）}
{之/hh}
{四/hh/pz}
{言（七言）/b}
{言/h}
{言（七）/hh/pz}
{言(七言)}
{八/pz}
{五}
{八/pz}
{二}
{也/pz}
{言（七）/b}
{七/h}
{七（三）/hh/pz}
{七（三七）/y/pz}
{之}
{四/h}
{七（三）/hh/pz}
{二}
{也/h}
{七（三）/hh/pz}
{二/b}
{也/pz}
{也/y/r}

## 第二段
@ 盘涉调
{二}    
{也/pz}
{七}
{言（七言）/pz}
{一/b/y/pz}
{之}
{四/pz}
{之}
{合（八）/h}
{之/hh/pz}
{七（三七）}
{之/hh}
{四/hh/pz}
{言（七言）/b}
{言/h}
{言（七）/hh/pz}
{言（七言）}
{八/pz}
{言（七言）}
{五/hh}
{八/hh/pz}
{八/y/pz}
{言（七言）/b}
{之/hh}
{四/hh/pz}
{合（八）}
{一/hh}
{之/hh/pz}
{之/y/pz}
{八}
{合（八合）/pz}
{一/b}
{之/pz}
{之/y/pz}
{之}
{四/r}
## 第三段

"""
    )



    # 初始化解析器
    service = ScoreService()
    context: PipelineContext = PipelineContext()
    context.set_raw_text(sample_score_text)
    # 解析成 AST
    context = service.process_score(context,"pipa")
    score_document:ScoreDocumentNode = context.node
    # 3.输出到控制台
    print("=== AST ===")
    #print(score_document)

    # 4.保存为 JSON
    json_data = json.dumps(score_document.to_dict(), ensure_ascii=False, indent=2)

    with open("score_document.json", "w", encoding="utf-8") as f:
        f.write(json_data)

    print("=== 已保存 score_document.json ===")
    save_path = str(Path(__file__).parent)
    # 5.渲染为图片
    service.render_score(context,"pipa","image",save_path)
    service.convert_score(context,"pipa","midi",save_path)


if __name__ == "__main__":
    main()