object_3d.py

class PrimitiveObject3D:
    """
    Abstract class that represents all 3d objects. These can be simple objects like Cubes and Spheres, or can be more complex objects that use a sketch and a length to extrude.
    """

    def __init__(self, sketch_profile='sketch.profiles[-1]', extrude_distance=None):
        self.object_type = 'Primitive Object3D'
        self.object_properties = {}
        self.sketch_profiles = [] # TODO use this to tie 3D objects to their 2D base
        self.alternative_properties = {} # key is the alternative property, value is a tuple of (object_property, adjustment_function) where object_property is the corresponding object property and adjustment_function is a function that accepts alternative_property and returns the correct value for the object_property

    def get_needed_properties(self):
        needed_properties = [elem for elem in self.object_properties if self.object_properties[elem] is None]
        return needed_properties # list of strings
    
    def is_complete(self):
        return len(self.get_needed_properties()) == 0
    
    def __str__(self) -> str:
        return_string = self.object_type[0].upper() + self.object_type[1 : ].lower() + ' || ' + ', '.join([elem[0] + ': ' + str(elem[1]) for elem in self.object_properties.items()])
        return return_string

    def to_string_fusion(self):
        """
        Returns a string that can be used to create the object using the Fusion API. Implemented in each subclass.
        """
        return_str = "" # TODO change this to return the sketch profile to be used
        return return_str
    
    def set_prop(self, property_name, value):
        if property_name in self.object_properties:
            self.object_properties[property_name] = value
            return True
        if property_name in self.alternative_properties:
            object_property, adjustment_function = self.alternative_properties[property_name]
            self.object_properties[object_property] = adjustment_function(value)
            return True
        raise Exception(f"{property_name} is not a property of {self.object_type}") # this method was given an invalid property name, probably by the TextParser class
    
    def set_next_prop(self, value):
        raise Exception('Invoked deprecated method')
        for key in self.properties:
            if self.properties[key] == None:
                self.set_prop(key, value)
                return None
        raise Exception(f'Attempted to add more properties than {self.object_type} has.')
    
    def get_prop(self, property_name):
        if property_name not in self.object_properties:
            raise Exception(f'{property_name} is not a property of {self.object_2d_name}')
        return self.object_properties[property_name]
    
    def get_prop_list(self):
        return set(self.object_properties.keys()).union(self.alternative_properties.keys())
    











"""
ExtrudedObject3D (below) is not currently functional. It is left in the codebase so it doesn't break imports.
"""

class ExtrudedObject3D:
    """
    Abstract class that represents all 3d objects. These can be simple objects like Cubes and Spheres, or can be more complex objects that use a sketch and a length to extrude.
    """

    def __init__(self, sketch_profile='sketch.profiles[-1]', extrude_distance=None):
        self.object_type = 'Extruded Sketch'
        self.properties = {'sketch_profile' : sketch_profile, 'extrude_distance' : extrude_distance}

    def get_needed_properties(self):
        needed_properties = [elem for elem in self.properties if self.properties[elem] is None]
        return needed_properties # list of strings
    
    def to_string_display(self):
        """
        Returns a string that can be used to display the object in a human-readable format.
        """
        return_string = self.object_type + ', '.join([elem for elem in self.properties.items()])
        return return_string

    def to_string_fusion(self):
        """
        Returns a string that can be used to create the object using the Fusion API. Implemented in each subclass.
        """
        return_str = f"        extrudeDistance = adsk.core.ValueInput.createByReal({self.properties['extrude_distance']})\n        extrude = rootComp.features.extrudeFeatures.addSimple({self.sketch_profile}, extrudeDistance, adsk.fusion.FeatureOperations.NewBodyFeatureOperation)"
        return return_str
    
    def set_prop(self, property_name, value):
        self.properties[property_name] = value