import NemAll_Python_Geometry as geo
import NemAll_Python_BaseElements as base
import NemAll_Python_BasisElements as basis
import NemAll_Python_Utility as util
import NemAll_Python_IFW_ElementAdapter as elem_adpt
import NemAll_Python_Usr_Parts as usr_parts

# Define the parameters for the column
def define_parameters():
    param_list = [
        usr_parts.CreateUserVariable('Height', 'Column Height', 3000.0, min_value=1000.0, max_value=10000.0),
        usr_parts.CreateUserVariable('Width', 'Column Width', 300.0, min_value=100.0, max_value=2000.0),
        usr_parts.CreateUserVariable('Depth', 'Column Depth', 300.0, min_value=100.0, max_value=2000.0)
    ]
    return param_list

# Create the column geometry
def create_column_geometry(params):
    height = params['Height']
    width = params['Width']
    depth = params['Depth']
    
    # Define the points of the column
    points = [
        geo.Point3D(0, 0, 0),
        geo.Point3D(width, 0, 0),
        geo.Point3D(width, depth, 0),
        geo.Point3D(0, depth, 0)
    ]
    
    # Create the base polygon
    base_polygon = geo.Polygon3D(points)
    
    # Create the column as a prism (extrude the base polygon)
    column = basis.BasisElement.CreatePrism(base_polygon, height)
    
    return column

# Main function to create the PythonPart
def main():
    # Define the parameters
    param_list = define_parameters()
    
    # Create the geometry
    column = create_column_geometry(param_list)
    
    # Insert the column into the Allplan model
    util.AllplanUtility.CreateElement(column)

# Run the main function to execute the script
if __name__ == '__main__':
    main()