Friction Data Export

exporter/SynthesisFusionAddin/src/Parser/ExporterOptions.py

@@ -92,6 +92,9 @@ class ExporterOptions:
     # Always stored in kg regardless of 'preferredUnits'
     robotWeight: float = field(default=0.0)
 
+    frictionOverride: bool = field(default=False)
+    frictionOverrideCoeff: float | None = field(default=None)
+
     compressOutput: bool = field(default=True)
     exportAsPart: bool = field(default=False)
 

exporter/SynthesisFusionAddin/src/Parser/SynthesisParser/Parser.py

@@ -41,7 +41,11 @@ def export(self) -> bool:
 
             # Physical Props here when ready
 
-            # ts = time()
+            if self.exporterOptions.frictionOverride:
+                assembly_out.data.parts.parts_definitions.value.friction_override = (
+                    self.exporterOptions.frictionOverrideCoeff
+                )
+            #
 
             progressDialog = app.userInterface.createProgressDialog()
             progressDialog.cancelButtonText = "Cancel"

exporter/SynthesisFusionAddin/src/UI/ConfigCommand.py

@@ -4,7 +4,8 @@
 
 import logging
 import os
-import platform
+
+# import platform
 import traceback
 from enum import Enum
 
@@ -621,11 +622,13 @@ def notify(self, args):
             """
             Creates the advanced tab, which is the parent container for internal command inputs
             """
-            advancedSettings = INPUTS_ROOT.addTabCommandInput("advanced_settings", "Advanced")
+            advancedSettings: adsk.core.TabCommandInput = INPUTS_ROOT.addTabCommandInput(
+                "advanced_settings", "Advanced"
+            )
             advancedSettings.tooltip = (
                 "Additional Advanced Settings to change how your model will be translated into Unity."
             )
-            a_input = advancedSettings.children
+            a_input: adsk.core.CommandInputs = advancedSettings.children
 
             # ~~~~~~~~~~~~~~~~ EXPORTER SETTINGS ~~~~~~~~~~~~~~~~
             """
@@ -660,55 +663,38 @@ def notify(self, args):
             """
             Physics settings group command
             """
-            physicsSettings = a_input.addGroupCommandInput("physics_settings", "Physics Settings")
+            physicsSettings: adsk.core.GroupCommandInput = a_input.addGroupCommandInput(
+                "physics_settings", "Physics Settings"
+            )
 
-            physicsSettings.isExpanded = False
+            physicsSettings.isExpanded = True
             physicsSettings.isEnabled = True
-            physicsSettings.tooltip = "tooltip"  # TODO: update tooltip
-            physics_settings = physicsSettings.children
-
-            # AARD-1687
-            # Should also be commented out / removed?
-            # This would cause problems elsewhere but I can't tell i f
-            # this is even being used.
-            frictionOverrideTable = self.createTableInput(
-                "friction_override_table",
-                "",
-                physics_settings,
-                2,
-                "1:2",
-                1,
-                columnSpacing=25,
-            )
-            frictionOverrideTable.tablePresentationStyle = 2
-            # frictionOverrideTable.isFullWidth = True
+            physicsSettings.tooltip = "Settings relating to the custom physics of the robot, like the wheel friction"
+            physics_settings: adsk.core.CommandInputs = physicsSettings.children
 
-            frictionOverride = self.createBooleanInput(
+            frictionOverrideInput = self.createBooleanInput(
                 "friction_override",
-                "",
+                "Friction Override",
                 physics_settings,
-                checked=False,
+                checked=exporterOptions.frictionOverride,  # object is missing attribute
                 tooltip="Manually override the default friction values on the bodies in the assembly.",
                 enabled=True,
                 isCheckBox=False,
             )
-            frictionOverride.resourceFolder = IconPaths.stringIcons["friction_override-enabled"]
-            frictionOverride.isFullWidth = True
+            frictionOverrideInput.resourceFolder = IconPaths.stringIcons["friction_override-enabled"]
+            frictionOverrideInput.isFullWidth = True
 
             valueList = [1]
             for i in range(20):
                 valueList.append(i / 20)
 
-            frictionCoeff = physics_settings.addFloatSliderListCommandInput(
-                "friction_coeff_override", "Friction Coefficient", "", valueList
+            frictionCoeffSlider: adsk.core.FloatSliderCommandInput = physics_settings.addFloatSliderListCommandInput(
+                "friction_override_coeff", "Friction Coefficient", "", valueList
             )
-            frictionCoeff.isVisible = False
-            frictionCoeff.valueOne = 0.5
-            frictionCoeff.tooltip = "Friction coefficient of field element."
-            frictionCoeff.tooltipDescription = "<i>Friction coefficients range from 0 (ice) to 1 (rubber).</i>"
-
-            frictionOverrideTable.addCommandInput(frictionOverride, 0, 0)
-            frictionOverrideTable.addCommandInput(frictionCoeff, 0, 1)
+            frictionCoeffSlider.isVisible = True
+            frictionCoeffSlider.valueOne = 0.5
+            frictionCoeffSlider.tooltip = "Friction coefficient of field element."
+            frictionCoeffSlider.tooltipDescription = "<i>Friction coefficients range from 0 (ice) to 1 (rubber).</i>"
 
             # ~~~~~~~~~~~~~~~~ JOINT SETTINGS ~~~~~~~~~~~~~~~~
             """
@@ -999,12 +985,6 @@ def notify(self, args):
                 self.log.error("Could not execute configuration due to failure")
                 return
 
-            export_as_part_boolean = (
-                eventArgs.command.commandInputs.itemById("advanced_settings")
-                .children.itemById("exporter_settings")
-                .children.itemById("export_as_part")
-            ).value
-
             processedFileName = gm.app.activeDocument.name.replace(" ", "_")
             dropdownExportMode = INPUTS_ROOT.itemById("mode")
             if dropdownExportMode.selectedItem.index == 0:
@@ -1167,13 +1147,36 @@ def notify(self, args):
             elif dropdownExportMode.selectedItem.index == 1:
                 _mode = ExportMode.FIELD
 
+            """
+            Advanced Settings
+            """
             global compress
             compress = (
                 eventArgs.command.commandInputs.itemById("advanced_settings")
                 .children.itemById("exporter_settings")
                 .children.itemById("compress")
             ).value
 
+            export_as_part_boolean = (
+                eventArgs.command.commandInputs.itemById("advanced_settings")
+                .children.itemById("exporter_settings")
+                .children.itemById("export_as_part")
+            ).value
+
+            frictionOverrideButton: adsk.core.BoolValueCommandInput = (
+                eventArgs.command.commandInputs.itemById("advanced_settings")
+                .children.itemById("physics_settings")
+                .children.itemById("friction_override")
+            )
+            frictionSlider: adsk.core.FloatSliderCommandInput = (
+                eventArgs.command.commandInputs.itemById("advanced_settings")
+                .children.itemById("physics_settings")
+                .children.itemById("friction_override_coeff")
+            )
+
+            frictionOverrideCoeff = frictionSlider.valueOne
+            frictionOverride = frictionSlider.isVisible
+
             exporterOptions = ExporterOptions(
                 savepath,
                 name,
@@ -1187,9 +1190,11 @@ def notify(self, args):
                 exportMode=_mode,
                 compressOutput=compress,
                 exportAsPart=export_as_part_boolean,
+                frictionOverride=frictionOverride,
+                frictionOverrideCoeff=frictionOverrideCoeff,
             )
 
-            Parser(exporterOptions).export()
+            _: bool = Parser(exporterOptions).export()
             exporterOptions.writeToDesign()
         except:
             if gm.ui:
@@ -1628,7 +1633,7 @@ def notify(self, args):
             inputs = cmdInput.commandInputs
             onSelect = gm.handlers[3]
 
-            frictionCoeff = INPUTS_ROOT.itemById("friction_coeff_override")
+            frictionCoeff = INPUTS_ROOT.itemById("friction_override_coeff")
 
             wheelSelect = inputs.itemById("wheel_select")
             jointSelect = inputs.itemById("joint_select")

exporter/SynthesisFusionAddin/src/configure.py

@@ -7,6 +7,7 @@
 
 from .strings import INTERNAL_ID
 from .Types.OString import OString
+from typing import Any
 
 try:
     config = ConfigParser()
@@ -49,7 +50,7 @@ def setAnalytics(enabled: bool):
     return True
 
 
-def get_value(key: str, section="main") -> any:
+def get_value(key: str, section="main") -> Any:
     """Gets a value from the config file
 
     Args:
@@ -62,7 +63,7 @@ def get_value(key: str, section="main") -> any:
     return config.get(section, key)
 
 
-def write_configuration(section: str, key: str, value: any):
+def write_configuration(section: str, key: str, value: Any):
     """Write a configuration flag to the ini file
 
     Args: