agx-openplx 0.15.10__cp312-cp312-manylinux_2_39_x86_64.whl

openplx/__init__.py

@@ -0,0 +1,55 @@
+# pylint: disable=invalid-name
+"""
+__init__ module for OpenPLX package
+"""
+import os
+import sys
+import traceback
+# Use agx_env if available
+import importlib.util
+if importlib.util.find_spec("agx_env"):
+    import agx_env # pylint: disable=unused-import
+__AGXVERSION__ = "2.39.1.0"
+__version__ = "0.15.10"
+
+try:
+    import agx
+    if agx.__version__ != __AGXVERSION__:
+        print(f"This version of agx-openplx is compiled for AGX {__AGXVERSION__} and may crash with your {agx.__version__} version, "+
+              "update agx-openplx or AGX to make sure the versions are suited for eachother")
+except Exception as e: # pylint: disable=broad-exception-caught
+    traceback.print_exc()
+    print("Failed finding AGX Dynamics, have you run setup_env?")
+    sys.exit(255)
+
+if "DEBUG_AGXOPENPLX" in os.environ:
+    print("#### Using Debug build ####")
+    try:
+        # pylint: disable=relative-beyond-top-level, wildcard-import, unused-import, no-name-in-module
+        from .debug.api import *
+        from .debug import Core
+        from .debug import Math
+        from .debug import Physics
+        from .debug import Simulation
+        # pylint: enable=relative-beyond-top-level, wildcard-import, unused-import, no-name-in-module
+    except Exception as e: # pylint: disable=broad-exception-caught
+        traceback.print_exc()
+        print("Failed finding OpenPLX modules or libraries, did you set PYTHONPATH correctly? "+
+              "Should point to where OpenPLX directory with binaries are located.")
+        print("Also, make sure you are using the same Python version the libraries were built for.")
+        sys.exit(255)
+else:
+    try:
+        # pylint: disable=relative-beyond-top-level, wildcard-import, unused-import, no-name-in-module
+        from .api import *
+        from . import Core
+        from . import Math
+        from . import Physics
+        from . import Simulation
+        # pylint: enable=relative-beyond-top-level, wildcard-import, unused-import, no-name-in-module
+    except Exception as e: # pylint: disable=broad-exception-caught
+        traceback.print_exc()
+        print("Failed finding OpenPLX modules or libraries, did you set PYTHONPATH correctly? "+
+              "Should point to where OpenPLX directory with binaries are located.")
+        print("Also, make sure you are using the same Python version the libraries were built for.")
+        sys.exit(255)

openplx/agxtoopenplx.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+"""
+Command line utility that helps converting AGX files to OpenPLX files.
+"""
+import os
+import sys
+from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter, SUPPRESS
+import agx
+import agxIO
+import agxSDK
+from openplx import AgxToOpenPlxMapper, __version__, set_log_level
+from openplx.versionaction import VersionAction
+
+init = agx.AutoInit()
+
+
+def parse_args():
+    parser = ArgumentParser(description="Convert .agx and .aagx files to .openplx", formatter_class=ArgumentDefaultsHelpFormatter)
+    parser.add_argument("agxfile", help="the .agx or .aagx file to load")
+    parser.add_argument("--root-system-name", metavar="<root system name>", help="Override the name of the root system model")
+    parser.add_argument("--export-folder", metavar="<export folder>", help="Where to write exported trimesh:es", default="./")
+    parser.add_argument("--loglevel", choices=["trace", "debug", "info", "warn", "error", "critical", "off"], help="Set log level", default="info")
+    parser.add_argument("--version", help="Show version", action=VersionAction, nargs=0, default=SUPPRESS)
+    parser.add_argument("--realprecision", help="Set precision of generated real values", type=int, default=6)
+    return parser.parse_known_args()
+
+
+def run():
+    args, _ = parse_args()
+    set_log_level(args.loglevel)
+    simulation = agxSDK.Simulation()
+    assembly = agxSDK.AssemblyRef(agxSDK.Assembly())
+    if args.root_system_name is not None:
+        assembly.setName(args.root_system_name)
+    agxIO.readFile(args.agxfile, simulation, assembly.get())
+
+    # Export trimeshes relative to the current directory
+    export_folder = args.export_folder if os.path.isabs(args.export_folder) else os.path.join(os.getcwd(), args.export_folder)
+    if not os.path.exists(export_folder):
+        os.makedirs(export_folder)
+    mapper = AgxToOpenPlxMapper(export_folder, True, args.realprecision)
+    if ".agx" in args.agxfile:
+        output_file = os.path.basename(args.agxfile.replace(".agx", ".openplx"))
+    elif ".aagx" in args.agxfile:
+        output_file = os.path.basename(args.agxfile.replace(".aagx", ".openplx"))
+    else:
+        sys.exit("Invalid file format. Only .agx and .aagx files are supported.")
+
+
+    print("writing to", output_file)
+    with open(output_file, "w", encoding="utf8") as file:
+        file.writelines(mapper.assemblyToOpenPlx(assembly))
+
+if __name__ == '__main__':
+    run()

openplx/anytoopenplx.py

@@ -0,0 +1,44 @@
+#!/usr/bin/env python3
+"""
+Command line utility that helps converting supported formats to OpenPLX files.
+"""
+from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter, SUPPRESS
+from pathlib import Path
+import agx
+from openplx import register_plugins, parseWithPlugin, __version__, set_log_level
+from openplx.Core import OpenPlxContext, getRegisteredPlugins, StringVector
+from openplx.versionaction import VersionAction
+
+
+def parse_args():
+    parser = ArgumentParser(description="Convert any (supported) file to .openplx", formatter_class=ArgumentDefaultsHelpFormatter)
+    parser.add_argument("inputfile", metavar="<non-openplx-file>", help="the non-openplx file to convert")
+    # Next to will be added in the future, at which time agxtoopenplx will be obsolete
+    # parser.add_argument("--root-system-name", metavar="<root system name>", help="Override the name of the root system model")
+    # parser.add_argument("--export-folder", metavar="<export folder>", help="Where to write exported trimesh:es", default="./")
+    parser.add_argument("--loglevel", choices=["trace", "debug", "info", "warn", "error", "critical", "off"], help="Set log level", default="info")
+    parser.add_argument("--version", help="Show version", action=VersionAction, nargs=0, default=SUPPRESS)
+    parser.add_argument("--realprecision", help="Set precision of generated real values", type=int, default=6)
+    return parser.parse_known_args()
+
+def run():
+    init = agx.AutoInit()  # pylint: disable=W0612 # Unused variable 'init'
+    args, _ = parse_args()
+    set_log_level(args.loglevel)
+    context = OpenPlxContext(StringVector())
+    register_plugins(context, None)
+    plugins = getRegisteredPlugins(context)
+
+    extension_plugin_map = {ext: plugin for plugin in plugins for ext in plugin.extensions()}
+    extension = Path(args.inputfile).suffix
+    if extension not in extension_plugin_map:
+        print(f"No matching plugin found for {args.inputfile}, does not end in one of {list(extension_plugin_map.keys())}")
+        return
+    matching_plugin = extension_plugin_map[extension]
+    matching_plugin.setPrecision(args.realprecision)
+    openplx_string = parseWithPlugin(matching_plugin, args.inputfile)
+
+    print(openplx_string)
+
+if __name__ == '__main__':
+    run()