def OriTransform(): if cad.GetNumSelected() == 0: wx.GetApp().PickObjects('Pick objects to transform') if cad.GetNumSelected() == 0: return config = HeeksConfig() selected_items = cad.GetSelectedObjects() cad.ClearSelection(False) o1 = PickOrigin('Pick From Origin') if o1 == None: wx.MessageBox("You didn't pick an origin\nTransform Cancelled!") return o2 = PickOrigin('Pick To Origin') if o2 == None: wx.MessageBox("You didn't pick an origin\nTransform Cancelled!") return cad.StartHistory() m1 = o1.GetOrigin() mat = m1.Inverse() m2 = o2.GetOrigin() mat.Multiply(m2) for object in selected_items: cad.TransformUndoably(object, mat) cad.EndHistory()
def Mirror(): if cad.GetNumSelected() == 0: wx.GetApp().PickObjects('Pick objects to mirror') if cad.GetNumSelected() == 0: return config = HeeksConfig() selected_items = cad.GetSelectedObjects() cad.ClearSelection(False) copy = config.ReadBool("MirrorCopy", False) axis = geom.Point3D(0, 0, 1) pos = geom.Point3D(0, 0, 0) axis.x = config.ReadFloat("MirrorAxisX", 1.0) axis.y = config.ReadFloat("MirrorAxisY", 0.0) axis.z = config.ReadFloat("MirrorAxisZ", 0.0) pos.x = config.ReadFloat("MirrorPosX", 0.0) pos.y = config.ReadFloat("MirrorPosY", 0.0) pos.z = config.ReadFloat("MirrorPosZ", 0.0) result, axis, pos, copy = InputMirrorPlane(axis, pos, copy, 'Mirror') if not result: return config.WriteBool("MirrorCopy", copy) config.WriteFloat("MirrorAxisX", axis.x) config.WriteFloat("MirrorAxisY", axis.y) config.WriteFloat("MirrorAxisZ", axis.z) config.WriteFloat("MirrorPosX", pos.x) config.WriteFloat("MirrorPosY", pos.y) config.WriteFloat("MirrorPosZ", pos.z) cad.StartHistory() mat = geom.Matrix() x, y = axis.ArbitraryAxes() axis_mat = geom.Matrix(pos, x, y) inv_mat = axis_mat.Inverse() mat.Multiply(inv_mat) # transform so axis goes to z axis mat.Scale3(1.0, 1.0, -1.0) # mirror in z axis mat.Multiply(axis_mat) # transform back to where it was for object in selected_items: if copy: if object.CanBeCopied(): object = object.MakeACopy() object.Transform(mat) cad.AddUndoably(object) else: cad.TransformUndoably(object, mat) cad.EndHistory()
def Scale(): centre_Pos = geom.Point3D(0, 0, 0) if cad.GetNumSelected() == 0: wx.GetApp().PickObjects('Pick objects to scale') if cad.GetNumSelected() == 0: return config = HeeksConfig() selected_items = cad.GetSelectedObjects() cad.ClearSelection(False) scale_factor = config.ReadFloat('ScaleFactor', 1.0) copy = config.ReadBool("ScaleCopy", False) pos = geom.Point3D(0, 0, 0) pos.x = config.ReadFloat("ScaleAboutPosX", 0.0) pos.y = config.ReadFloat("ScaleAboutPosY", 0.0) pos.z = config.ReadFloat("ScaleAboutPosZ", 0.0) result, pos, scale_factor, copy = InputScale(pos, scale_factor, copy, 'Scale') if not result: return config.WriteFloat("ScaleFactor", scale_factor) config.WriteBool("ScaleCopy", copy) config.WriteFloat("ScaleAboutPosX", pos.x) config.WriteFloat("ScaleAboutPosY", pos.y) config.WriteFloat("ScaleAboutPosZ", pos.z) cad.StartHistory() mat = geom.Matrix() mat.Translate(-pos) mat.Scale(scale_factor) mat.Translate(pos) for object in selected_items: if copy: if object.CanBeCopied(): object = object.MakeACopy() object.Transform(mat) cad.AddUndoably(object) else: cad.TransformUndoably(object, mat) cad.EndHistory()
def OnSweep(self, event): if cad.GetNumSelected() == 0: self.PickObjects('Pick Some Sketches, Faces or Circles') if not self.CheckForNumberOrMore( 1, [ cad.OBJECT_TYPE_SKETCH, step.GetFaceType(), cad.OBJECT_TYPE_CIRCLE ], 'Pick one or more sketches, faces or circles, to sweep\n( hold down Ctrl key to select more than one solid )', 'Sweep'): return sweep_objects = cad.GetSelectedObjects() cad.ClearSelection(True) self.PickObjects('Pick a Sketch to Sweep Along') if not self.CheckForNumberOrMore(1, [cad.OBJECT_TYPE_SKETCH], 'Pick one sketch to sweep along', 'Sweep'): return sweep_profile = cad.GetSelectedObjects()[0] cad.StartHistory() step.CreateSweep(sweep_objects, sweep_profile, cad.Color(128, 150, 128)) cad.EndHistory()
def OnRevolve(self, event): if cad.GetNumSelected() == 0: self.PickObjects('Pick Some Sketches, Faces or Circles') if not self.CheckForNumberOrMore( 1, [ cad.OBJECT_TYPE_SKETCH, step.GetFaceType(), cad.OBJECT_TYPE_CIRCLE ], 'Pick one or more sketches, faces or circles, to create an extruded body from\n( hold down Ctrl key to select more than one solid )', 'Revolve'): return config = HeeksConfig() angle = config.ReadFloat('RevolutionAngle', 360.0) extrude_makes_a_solid = config.ReadBool('ExtrudeToSolid') success, angle, extrude_makes_a_solid = self.InputRevolutionAngle( angle, extrude_makes_a_solid) if success: config.WriteFloat('RevolutionAngle', angle) config.WriteBool('ExtrudeToSolid', extrude_makes_a_solid) step.CreateExtrusion(angle, extrude_makes_a_solid, True, 0.0, cad.Color(128, 128, 150))
def OnExtrude(self, event): if cad.GetNumSelected() == 0: self.PickObjects('Pick Some Sketches, Faces or Circles') if not self.CheckForNumberOrMore( 1, [ cad.OBJECT_TYPE_SKETCH, step.GetFaceType(), cad.OBJECT_TYPE_CIRCLE ], 'Pick one or more sketches, faces or circles, to create an extruded body from\n( hold down Ctrl key to select more than one solid )', 'Extude'): return config = HeeksConfig() height = config.ReadFloat('ExtrusionHeight', 10.0) taper_angle = config.ReadFloat('ExtrusionTaperAngle', 0.0) extrude_makes_a_solid = config.ReadBool('ExtrudeToSolid') success, height, extrude_makes_a_solid, taper_angle = self.InputExtrusionHeight( height, extrude_makes_a_solid, taper_angle) if success: config.WriteFloat('ExtrusionHeight', height) config.WriteFloat('ExtrusionTaperAngle', taper_angle) config.WriteBool('ExtrudeToSolid', extrude_makes_a_solid) step.CreateExtrusion(height, extrude_makes_a_solid, False, taper_angle, cad.Color(128, 128, 128))
def OnKeyDown(self, e): k = e.GetKeyCode() if k == wx.WXK_ESCAPE and self.frame.IsFullScreen(): self.ShowFullScreen(False) return True else: key_code = Key.KeyCodeFromWx(e) if wx.GetApp().input_mode_object.OnKeyDown(key_code): return True if k == wx.WXK_DELETE: if cad.GetNumSelected() > 0: self.DeleteMarkedList() elif k == wx.WXK_RETURN: if self.inMainLoop: self.ExitMainLoop() self.Repaint() elif k == ord('Z'): if e.ControlDown(): if e.ShiftDown(): self.OnRedo(e) else: self.OnUndo(e) elif k == ord('X'): if e.ControlDown(): self.OnCut(e) elif k == ord('C'): if e.ControlDown(): self.OnCopy(e) elif k == ord('V'): if e.ControlDown(): self.OnPaste(e) else: return False return True
def PickSketch(self): cad.ClearSelection() wx.GetApp().PickObjects('Pick a sketch', cad.OBJECT_TYPE_SKETCH, True) self.cmbSketch.Recreate() self.Fit() id = 0 if cad.GetNumSelected() > 0: id = cad.GetSelectedObjects()[0].GetID() self.cmbSketch.SelectById(id)
def CheckForNumberOrMore(self, min_num, types, msg, caption): num_found = 0 for object in cad.GetSelectedObjects(): for type in types: if object.GetType() == type: num_found += 1 break if num_found < min_num: filter = cad.Filter() for t in types: filter.AddType(t) objects = self.PickObjects(msg, filter, False) if cad.GetNumSelected() < min_num: wx.MessageBox(msg) return False return True
def Rotate(copy=False): #rotation axis - Z axis by default axis_Dir = geom.Point3D(0, 0, 1) line_Pos = geom.Point3D(0, 0, 0) if cad.GetNumSelected() == 0: wx.GetApp().PickObjects('Pick objects to rotate') if cad.GetNumSelected() == 0: return config = HeeksConfig() ncopies = None if copy: ncopies = config.ReadInt('RotateNumCopies', 1) RemoveUncopyable() if cad.GetNumSelected() == 0: return selected_items = cad.GetSelectedObjects() cad.ClearSelection(False) angle = config.ReadFloat('RotateAngle', 90.0) # enter angle, plane and position axis = geom.Point3D(0, 0, 1) pos = geom.Point3D(0, 0, 0) axis.x = config.ReadFloat("RotateAxisX", 0.0) axis.y = config.ReadFloat("RotateAxisY", 0.0) axis.z = config.ReadFloat("RotateAxisZ", 1.0) pos.x = config.ReadFloat("RotatePosX", 0.0) pos.y = config.ReadFloat("RotatePosY", 0.0) pos.z = config.ReadFloat("RotatePosZ", 0.0) axial_shift = config.ReadFloat('RotateAxialShift', 0.0) result, angle, axis, pos, ncopies, axial_shift = InputAngleWithPlane( angle, axis, pos, ncopies, axial_shift, 'Rotate') if not result: return if copy: if ncopies < 1: return config.WriteInt("RotateNumCopies", ncopies) config.WriteFloat("RotateAngle", angle) config.WriteFloat("RotateAxialShift", axial_shift) config.WriteFloat("RotateAxisX", axis.x) config.WriteFloat("RotateAxisY", axis.y) config.WriteFloat("RotateAxisZ", axis.z) config.WriteFloat("RotatePosX", pos.x) config.WriteFloat("RotatePosY", pos.y) config.WriteFloat("RotatePosZ", pos.z) cad.StartHistory() if copy: for i in range(0, ncopies): mat = geom.Matrix() mat.Translate(-pos) mat.RotateAxis(angle * 0.0174532925199433 * (i + 1), axis) mat.Translate(pos) for object in selected_items: new_object = object.MakeACopy() object.GetOwner().Add(new_object) cad.TransformUndoably(new_object, mat) else: mat = geom.Matrix() mat.Translate(-pos) mat.RotateAxis(angle * 0.0174532925199433, axis) mat.Translate(pos) for object in selected_items: cad.TransformUndoably(object, mat) cad.EndHistory()
def Translate(copy=False): if cad.GetNumSelected() == 0: wx.GetApp().PickObjects('Pick objects to move') if cad.GetNumSelected() == 0: return config = HeeksConfig() ncopies = None if copy: ncopies = config.ReadInt('TranslateNumCopies', 1) RemoveUncopyable() if cad.GetNumSelected() == 0: return selected_items = cad.GetSelectedObjects() cad.ClearSelection(False) fromp = geom.Point3D() to = geom.Point3D() fromp.x = config.ReadFloat("TranslateFromX", 0.0) fromp.y = config.ReadFloat("TranslateFromY", 0.0) fromp.z = config.ReadFloat("TranslateFromZ", 0.0) to.x = config.ReadFloat("TranslateToX", 0.0) to.y = config.ReadFloat("TranslateToY", 0.0) to.z = config.ReadFloat("TranslateToZ", 0.0) result, fromp, to, ncopies = InputFromAndTo(fromp, to, ncopies, 'Move Translate') if not result: return if copy: if ncopies < 1: return config.WriteInt("TranslateNumCopies", ncopies) config.WriteFloat("TranslateFromX", fromp.x) config.WriteFloat("TranslateFromY", fromp.y) config.WriteFloat("TranslateFromZ", fromp.z) config.WriteFloat("TranslateToX", to.x) config.WriteFloat("TranslateToY", to.y) config.WriteFloat("TranslateToZ", to.z) cad.StartHistory() if copy: for i in range(0, ncopies): mat = geom.Matrix() mat.Translate((to - fromp) * (i + 1)) for object in selected_items: new_object = object.MakeACopy() object.GetOwner().Add(new_object) cad.TransformUndoably(new_object, mat) else: mat = geom.Matrix() mat.Translate(to - fromp) for object in selected_items: cad.TransformUndoably(object, mat) cad.EndHistory()
def OnUpdateCopy(self, e): e.Enable(cad.GetNumSelected() > 0)
def OnUpdateDelete(self, e): e.Enable(cad.GetNumSelected() > 0)