jak2: add .gd files and level info for some ps3 test levels (#2776)

scripts/gen-dgo-txt.py

@@ -0,0 +1,107 @@
+import json
+import argparse
+import os, shutil, sys
+from glob import glob
+from pathlib import Path
+
+# generate dgo.txt and gd files out of ps3 test level dgo folders
+# usage: python gen-dgo-txt.py <path to dgo folder>
+
+dgo_json = {}
+
+def get_dgo_files(path):
+    return glob(path + "/**/*.go", recursive=True)
+
+def remove_extensions(file_list):
+    for file in file_list:
+        if (file.endswith("-bt.go")):
+            os.rename(file, file[:-3].replace("-bt", "-vis"))
+        elif file.endswith(".go"):
+            os.rename(file, file[:-3])
+
+def move_to_root_folder(root_path, cur_path):
+    for filename in os.listdir(cur_path):
+        if os.path.isfile(os.path.join(cur_path, filename)):
+            shutil.move(os.path.join(cur_path, filename), os.path.join(root_path, filename))
+        elif os.path.isdir(os.path.join(cur_path, filename)):
+            move_to_root_folder(root_path, os.path.join(cur_path, filename))
+        else:
+            sys.exit("Should never reach here.")
+    # remove empty folders
+    if cur_path != root_path:
+        os.rmdir(cur_path)
+
+def gen_json_for_dgo(folder_name, files):
+    dgo_name = os.path.basename(Path(folder_name)) + ".DGO"
+    dgo = {
+        "file_name": dgo_name,
+        "internal_name": dgo_name,
+        "objects": []
+    }
+
+    for file in files:
+        internal_name = ""
+        if "." in file:
+            internal_name = Path(file).name.split(".")[0]
+        else:
+            internal_name = file
+        unique_name = internal_name
+        if (internal_name.endswith("-ag")):
+            internal_name = internal_name.replace("-ag", "")
+        elif (internal_name.endswith("-bt")):
+            internal_name = internal_name.replace("-bt", "-vis")
+            unique_name = unique_name.replace("-bt", "-vis")
+        dgo["objects"].append({
+            "internal_name": internal_name,
+            "unique_name": unique_name
+        })
+    return dgo
+
+def gen_gd_file(dgo_name, file_list):
+    # ("SKATEPARK.DGO"
+    #  ("drill-turret-ext-ag.go"
+    #   "drill-turret-int-ag.go"
+    #   "jak-pole+0-ag.go"
+    #   "lazerfence-ag.go"
+    #   "lazerpit-ag.go"
+    #   "mech-ag.go"
+    #   "tpage-243.go"
+    #   "tpage-354.go"
+    #   "tpage-3279.go"
+    #   "skatepark-bt.go"
+    #  ))
+    file_names = []
+    for file in file_list:
+        file = file.replace("-bt", "-vis")
+        file_names.append(os.path.basename(Path(file)))
+    buffer = ""
+    buffer += "(" + "\"" + os.path.basename(Path(dgo_name)) + ".DGO" + "\"" + "\n  (\n"
+    for file in file_names:
+        buffer += "    \"" + file + "\"\n"
+    buffer += "  ))\n"
+    # print("dgo " + os.path.basename(Path(dgo_name)) + " gd:\n" + buffer + "\n")
+    with open(dgo_name + "/" + os.path.basename(Path(dgo_name)).lower() + ".gd", "a+") as f:
+        f.write(buffer)
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("dgo_folder", type=str)
+    args = parser.parse_args()
+
+    # generate gd files for all dgo folders in current directory
+    # folders = [name for name in os.listdir(os.getcwd()) if os.path.isdir(os.path.join(os.getcwd(), name))]
+    # for folder in folders:
+    #     gen_gd_file(folder, get_dgo_files(folder))
+
+    files = get_dgo_files(args.dgo_folder)
+    gen_gd_file(args.dgo_folder, files)
+    dgo_json = gen_json_for_dgo(args.dgo_folder, files)
+    # print(dgo_json)
+    with open (args.dgo_folder + "/" + os.path.basename(Path(args.dgo_folder)) + ".DGO.txt", "a+") as f:
+        json.dump(dgo_json, f, indent=4)
+    move_to_root_folder(args.dgo_folder, args.dgo_folder)
+    file_list_after = glob(args.dgo_folder + "/**/*.go", recursive=True)
+    remove_extensions(file_list_after)
+
+if __name__ == "__main__":
+    main()