import FreeCAD
import Part
import Import

doc = FreeCAD.newDocument()


def genPlacementMatrix():
    yaw = 5000
    pitch = 3000
    roll = 2200

    pos = FreeCAD.Vector(0, 0, 0)
    rot = FreeCAD.Rotation(yaw, pitch, roll)
    center = FreeCAD.Vector(0, 0, 0)
    newplace = FreeCAD.Placement(pos, rot, center)

    return newplace

# Generate a placement matrix
newplace = genPlacementMatrix()

stem_radius = 2
stem_length = 5
cyl = FreeCAD.ActiveDocument.addObject("Part::Feature", "Cylinder")
cyl_pos_vec = FreeCAD.Vector(0, 0, 0)
cyl_dir_vec = FreeCAD.Vector(0, 0, -1)
cyl_shape = Part.makeCylinder(
    stem_radius, stem_length, cyl_pos_vec, cyl_dir_vec)
# Set the placement to the cylinder
cyl_shape.Placement = newplace
cyl.Shape = cyl_shape

# Create a box, don't worry about placement here
box_length = 2
box_width = 2
box_height = 2
box = FreeCAD.ActiveDocument.addObject("Part::Feature", "Box")
box.Shape = Part.makeBox(box_length, box_width, box_height)

# Create a list of the cyl and box part features
shapeobjs = [cyl, box]
# Set the dump path
dump_path_1 = 'D:/GoogleDrive/Product/Connections/FreeCAD/LocalTests/dump1.stp'
# Do the dumping
Import.export(shapeobjs, dump_path_1)

#Test with the equivalent gui version of this command
import ImportGui
dump_path_2 = 'D:/GoogleDrive/Product/Connections/FreeCAD/LocalTests/dump2.stp'
ImportGui.export(shapeobjs, dump_path_2)