plotagon-director 0.6.0__py3-none-any.whl

plotagon_director/library/builder.py

@@ -0,0 +1,153 @@
+"""Construit library/data/library.json depuis le manifeste officiel local.
+
+Usage:
+    python -m plotagon_director.library.builder [chemin/vers/builtincontentmanifest.json]
+
+Par defaut, cherche le manifeste dans une installation Plotagon Studio situee
+dans le dossier parent du depot (ex. Desktop/plotagon/Plotagon Studio_Data).
+
+En plus du manifeste, extrait les positions d'acteurs de chaque scene depuis
+les bundles Content/scenes/*.pcf : chaque .pcf embarque un JSON listant les
+slots ("locationID", "displayName", "stance"...) — confirme par un export
+officiel a 2 personnages (positions "infodesk2"/"middle1" sur stores.tech).
+"""
+
+import json
+import re
+import sys
+from pathlib import Path
+
+from . import DATA_PATH
+
+DEFAULT_MANIFEST = (
+    Path(__file__).resolve().parents[3]
+    / "Plotagon Studio_Data" / "StreamingAssets" / "JSON"
+    / "ContentManifest" / "builtincontentmanifest.json"
+)
+
+EXPRESSION_PREFIX = "animation.dialogue."
+
+_LOCATION_PATTERN = re.compile(
+    rb'"locationID":\s*"([^"]+)",\s*"name":\s*"([^"]*)",\s*"displayName":\s*"([^"]*)"'
+)
+
+_STRING_PATTERN = re.compile(rb"[ -~]{3,}")
+_ID_PATTERN = re.compile(r"[a-z][a-z0-9_]{2,}")
+
+
+def extract_actions(animation_db_path, expressions):
+    """Extrait les actions (interactions) de AnimationMetaDatabase.animationdata.
+
+    La base (bundle UnityFS, lu via UnityPy) contient des enregistrements
+    [id, Nom, phrase verbale, reaction, stance] pour les animations de
+    dialogue ET les interactions. Les interactions = ids absents de la liste
+    des expressions du manifeste. Confirme par l'export officiel
+    'forcechoke' / "force chokes". Les ids idle* (internes) sont exclus.
+
+    Retourne {id: {"text": Nom, "phrase": phrase}} ou {} si UnityPy absent.
+    """
+    try:
+        import UnityPy
+    except ImportError:
+        print("UnityPy absent: actions non extraites (pip install UnityPy)")
+        return {}
+
+    env = UnityPy.load(str(animation_db_path))
+    raw = next(
+        o for o in env.objects if o.type.name == "MonoBehaviour"
+    ).get_raw_data()
+    strings = [m.group(0).decode() for m in _STRING_PATTERN.finditer(raw)]
+
+    records = {}
+    for i, s in enumerate(strings[:-2]):
+        if not _ID_PATTERN.fullmatch(s):
+            continue
+        display = strings[i + 1]
+        if display.lower().replace(" ", "").replace("_", "") != s.replace("_", ""):
+            continue
+        records.setdefault(s, {"text": display, "phrase": strings[i + 2]})
+
+    return {
+        k: v for k, v in sorted(records.items())
+        if k not in expressions and not k.startswith("idle")
+    }
+
+
+def extract_scene_positions(scenes_dir):
+    """Scanne Content/scenes/*.pcf et retourne {scene_id: [{id, text}, ...]}."""
+    positions = {}
+    for pcf in sorted(Path(scenes_dir).glob("*.pcf")):
+        # nom de fichier: <scene.id>.<hash>.pcf
+        scene_id = pcf.name.rsplit(".", 2)[0]
+        data = pcf.read_bytes()
+        slots = []
+        seen = set()
+        for m in _LOCATION_PATTERN.finditer(data):
+            loc_id = m.group(1).decode("utf-8", "replace")
+            if loc_id in seen or loc_id == "None":
+                continue
+            seen.add(loc_id)
+            slots.append({
+                "id": loc_id,
+                "text": m.group(3).decode("utf-8", "replace"),
+            })
+        if slots:
+            positions[scene_id] = slots
+    return positions
+
+
+def build_library(manifest_path):
+    with open(manifest_path, encoding="utf-8") as f:
+        manifest = json.load(f)
+
+    library = {
+        "source": str(manifest_path),
+        "scenes": sorted(manifest.get("scenes", [])),
+        "characters": sorted(
+            manifest.get("characters", []) + manifest.get("extracharacters", [])
+        ),
+        "expressions": sorted(
+            a[len(EXPRESSION_PREFIX):]
+            for a in manifest.get("dialogue", [])
+            if a.startswith(EXPRESSION_PREFIX)
+        ),
+        "voices": sorted(manifest.get("voices", [])),
+        "music": sorted(manifest.get("music", [])),
+        "sounds": sorted(manifest.get("sounds", [])),
+    }
+
+    scenes_dir = Path(manifest_path).parents[2] / "Content" / "scenes"
+    if scenes_dir.is_dir():
+        library["scene_positions"] = extract_scene_positions(scenes_dir)
+
+    animation_db = (
+        Path(manifest_path).parents[2] / "Animation" / "Standalone"
+        / "AnimationMetaDatabase.animationdata"
+    )
+    if animation_db.exists():
+        library["actions"] = extract_actions(animation_db, set(library["expressions"]))
+    return library
+
+
+def main():
+    manifest_path = Path(sys.argv[1]) if len(sys.argv) > 1 else DEFAULT_MANIFEST
+    if not manifest_path.exists():
+        sys.exit("Manifeste introuvable: %s" % manifest_path)
+
+    library = build_library(manifest_path)
+    DATA_PATH.parent.mkdir(parents=True, exist_ok=True)
+    with open(DATA_PATH, "w", encoding="utf-8") as f:
+        json.dump(library, f, ensure_ascii=False, indent=2)
+
+    for key in ("scenes", "characters", "expressions", "voices", "music", "sounds"):
+        print("%-12s %d" % (key, len(library[key])))
+    positions = library.get("scene_positions", {})
+    print("%-12s %d scenes (%d slots)" % (
+        "positions", len(positions), sum(len(v) for v in positions.values())
+    ))
+    print("%-12s %d" % ("actions", len(library.get("actions", {}))))
+    print("OK ->", DATA_PATH)
+
+
+if __name__ == "__main__":
+    main()