Newer
Older
# ***** BEGIN GPL LICENSE BLOCK *****
#
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ***** END GPL LICENCE BLOCK *****
#
# -----------------------------------------------------------------------
# Author: Alan Odom (Clockmender), Rune Morling (ermo) Copyright (c) 2019
# -----------------------------------------------------------------------
#
import bpy
import bmesh
from .pdt_functions import (
debug,
from .pdt_command_functions import (
vector_build,
join_two_vertices,
set_angle_distance_two,
set_angle_distance_three,
origin_to_cursor,
taper,
placement_normal,
)
from .pdt_msg_strings import (
PDT_ERR_ADDVEDIT,
PDT_ERR_BADFLETTER,
PDT_ERR_CHARS_NUM,
PDT_ERR_DUPEDIT,
PDT_ERR_EXTEDIT,
PDT_ERR_FACE_SEL,
PDT_ERR_FILEDIT,
PDT_ERR_NON_VALID,
PDT_ERR_NO_SEL_GEOM,
PDT_ERR_SEL_1_EDGE,
PDT_ERR_SEL_1_EDGEM,
PDT_ERR_SPLITEDIT,
PDT_ERR_BADMATHS,
PDT_OBJ_MODE_ERROR,
PDT_ERR_SEL_4_VERTS,
PDT_ERR_INT_LINES,
PDT_LAB_PLANE,
from .pdt_bix import add_line_to_bisection
from .pdt_etof import extend_vertex
from .pdt_xall import intersect_all
from . import pdt_exception
PDT_SelectionError = pdt_exception.SelectionError
PDT_InvalidVector = pdt_exception.InvalidVector
PDT_CommandFailure = pdt_exception.CommandFailure
PDT_ObjectModeError = pdt_exception.ObjectModeError
PDT_MathsError = pdt_exception.MathsError
PDT_IntersectionError = pdt_exception.IntersectionError
PDT_NoObjectError = pdt_exception.NoObjectError
PDT_FeatureError = pdt_exception.FeatureError
class PDT_OT_CommandReRun(Operator):
"""Repeat Current Displayed Command."""
bl_idname = "pdt.command_rerun"
bl_label = "Re-run Current Command"
bl_options = {"REGISTER", "UNDO"}
def execute(self, context):
"""Repeat Current Command Line Input.
Args:
context: Blender bpy.context instance.
Returns:
Nothing.
"""
command_run(self, context)
return {"FINISHED"}
def command_run(self, context):
"""Run Command String as input into Command Line.
Note:
Uses pg.command, pg.error & many other 'pg.' variables to set PDT menu items,
or alter functions
Command Format; Operation(single letter) Mode(single letter) Values(up to 3 values
separated by commas)
Example; CD0.4,0.6,1.1 - Moves Cursor Delta XYZ = 0.4,0.6,1.1 from Current Position/Active
Vertex/Object Origin
Example; SP35 - Splits active Edge at 35% of separation between edge's vertices
Valid First Letters (as 'operation' - pg.command[0])
C = Cursor, G = Grab(move), N = New Vertex, V = Extrude Vertices Only,
E = Extrude geometry, P = Move Pivot Point, D = Duplicate geometry, S = Split Edges
Capitals and lower case letters are both allowed
Valid Second Letters (as 'mode' - pg.command[1])
A = Absolute XYZ, D = Delta XYZ, I = Distance at Angle, P = Percent
X = X Delta, Y = Y, Delta Z, = Z Delta, O = Output (Maths Operation only)
V = Vertex Bevel, E = Edge Bevel, I = Intersect then Bevel
Capitals and lower case letters are both allowed
Valid Values (pdt_command[2:])
Only Integers and Floats, missing values are set to 0, appropriate length checks are
performed as Values is split by commas.
Example; CA,,3 - Cursor to Absolute, is re-interpreted as CA0,0,3
Exception for Maths Operation, Values section is evaluated as Maths expression
Example; madegrees(atan(3/4)) - sets PDT Angle to smallest angle of 3,4,5 Triangle;
(36.8699 degrees)
Args:
context: Blender bpy.context instance.
Returns:
Nothing.
"""
scene = context.scene
pg = scene.pdt_pg
# Check Object Type & Mode First
obj = context.view_layer.objects.active
if obj is not None and command[0].upper() not in {"M", "?", "HELP"}:
if obj.mode not in {"OBJECT", "EDIT"} or obj.type not in {"MESH", "EMPTY"}:
pg.error = PDT_OBJ_MODE_ERROR
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_ObjectModeError
if command == "?" or command.lower() == "help":
# fmt: off
context.window_manager.popup_menu(pdt_help, title="PDT Command Line Help", icon="INFO")
# fmt: on
return
if command.upper() == "J2V":
if command.upper() == "AD2":
set_angle_distance_two(context)
if command.upper() == "AD3":
set_angle_distance_three(context)
if command.upper() == "OTC":
if command.upper() == "TAP":
if command.upper() == "BIS":
add_line_to_bisection(context)
if command.upper() == "ETF":
if command.upper() == "INTALL":
if command.upper()[1:] == "NML":
placement_normal(context, command.upper()[0])
if command.upper()[1:] == "CEN":
placement_arc_centre(context, command.upper()[0])
if command.upper()[1:] == "INT":
placement_intersect(context, command.upper()[0])
pg.error = PDT_ERR_CHARS_NUM
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
# Check First Letter
operation = command[0].upper()
if operation not in {"C", "D", "E", "F", "G", "N", "M", "P", "V", "S"}:
pg.error = PDT_ERR_BADFLETTER
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
(operation == "F" and mode not in {"v", "e", "i"})
or (operation in {"D", "E"} and mode not in {"d", "i"})
or (operation == "M" and mode not in {"a", "d", "i", "p", "o", "x", "y", "z"})
or (operation not in {"D", "E", "F", "M"} and mode not in {"a", "d", "i", "p"})
):
pg.error = f"'{mode}' {PDT_ERR_NON_VALID} '{operation}'"
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
# --------------
# Maths Operation
if operation == "M":
try:
command_maths(context, mode, pg, command[2:], mode)
# -----------------------------------------------------
# Not a Maths Operation, so let's parse the command line
try:
pg, values, obj, obj_loc, bm, verts = command_parse(context)
except PDT_SelectionError:
return
# ---------------------
# Cursor or Pivot Point
if operation in {"C", "P"}:
try:
move_cursor_pivot(context, pg, operation, mode, obj, verts, values)
except PDT_CommandFailure:
return
# ------------------------
# Move Vertices or Objects
try:
move_entities(context, pg, operation, mode, obj, bm, verts, values)
except PDT_CommandFailure:
return
# --------------
# Add New Vertex
try:
add_new_vertex(context, pg, operation, mode, obj, bm, verts, values)
except PDT_CommandFailure:
return
# -----------
# Split Edges
try:
split_edges(context, pg, operation, mode, obj, obj_loc, bm, values)
except PDT_CommandFailure:
return
# ----------------
# Extrude Vertices
try:
extrude_vertices(context, pg, operation, mode, obj, obj_loc, bm, verts, values)
except PDT_CommandFailure:
return
# ----------------
# Extrude Geometry
try:
extrude_geometry(context, pg, operation, mode, obj, bm, values)
except PDT_CommandFailure:
return
# ------------------
# Duplicate Geometry
try:
duplicate_geometry(context, pg, operation, mode, obj, bm, values)
except PDT_CommandFailure:
return
# ---------------
# Fillet Geometry
try:
fillet_geometry(context, pg, mode, obj, bm, verts, values)
except PDT_CommandFailure:
return
def pdt_help(self, context):
"""Display PDT Command Line help in a pop-up.
Args:
context: Blender bpy.context instance
Returns:
Nothing.
"""
label = self.layout.label
label(text="Primary Letters (Available Secondary Letters):")
label(text="")
label(text="C: Cursor (a, d, i, p)")
label(text="D: Duplicate Geometry (d, i)")
label(text="E: Extrude Geometry (d, i)")
label(text="G: Grab (Move) (a, d, i, p)")
label(text="N: New Vertex (a, d, i, p)")
label(text="M: Maths Functions (a, d, p, o, x, y, z)")
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
label(text="P: Pivot Point (a, d, i, p)")
label(text="V: Extrude Vertice Only (a, d, i, p)")
label(text="S: Split Edges (a, d, i, p)")
label(text="?: Quick Help")
label(text="")
label(text="Secondary Letters:")
label(text="")
label(text="- General Options:")
label(text="a: Absolute (Global) Coordinates e.g. 1,3,2")
label(text="d: Delta (Relative) Coordinates, e.g. 0.5,0,1.2")
label(text="i: Directional (Polar) Coordinates e.g. 2.6,45")
label(text="p: Percent e.g. 67.5")
label(text="- Fillet Options:")
label(text="v: Fillet Vertices")
label(text="e: Fillet Edges")
label(text="i: Fillet & Intersect 2 Disconnected Edges")
label(text="- Math Options:")
label(text="x, y, z: Send result to X, Y and Z input fields in PDT Design")
label(text="d, a, p: Send result to Distance, Angle or Percent input field in PDT Design")
label(text="o: Send Maths Calculation to Output")
label(text="")
label(text="Note that commands are case-insensitive: ED = Ed = eD = ed")
label(text="")
label(text="Examples:")
label(text="")
label(text="ed0.5,,0.6")
label(text="'- Extrude Geometry Delta 0.5 in X, 0 in Y, 0.6 in Z")
label(text="")
label(text="fe0.1,4,0.5")
label(text="'- Fillet Edges")
label(text="'- Radius: 0.1 (float) -- the radius (or offset) of the bevel/fillet")
label(text="'- Segments: 4 (int) -- choosing an even amount of segments gives better geometry")
label(text="'- Profile: 0.5 (float[0.0;1.0]) -- 0.5 (default) yields a circular, convex shape")
label(text="")
label(text="More Information at:")
label(text="https://github.com/Clockmender/Precision-Drawing-Tools/wiki")
def command_maths(context, mode, pg, expression, output_target):
"""Evaluates Maths Input.
Args:
context: Blender bpy.context instance.
mode: The Operation Mode, e.g. a for Absolute
pg: PDT Parameters Group - our variables
expression: The Maths component of the command input e.g. sqrt(56)
output_target: The output variable box on the UI
Returns:
Nothing.
"""
namespace = {}
namespace.update(vars(math))
try:
maths_result = eval(expression, namespace, namespace)
except:
pg.error = PDT_ERR_BADMATHS
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_MathsError
decimal_places = context.preferences.addons[__package__].preferences.pdt_input_round
pg.cartesian_coords.x = round(maths_result, decimal_places)
pg.cartesian_coords.y = round(maths_result, decimal_places)
pg.cartesian_coords.z = round(maths_result, decimal_places)
pg.distance = round(maths_result, decimal_places)
pg.angle = round(maths_result, decimal_places)
pg.percent = round(maths_result, decimal_places)
pg.maths_output = round(maths_result, decimal_places)
"""Parse Command Input.
Args:
context: Blender bpy.context instance.
Returns:
pg: PDT Parameters Group - our variables
values_out: The Output Values as a list of numbers
obj: The Active Object
obj_loc: The object's location in 3D space
bm: The object's Bmesh
verts: The object's selected vertices, or selected history vertices.
scene = context.scene
pg = scene.pdt_pg
command = pg.command.strip()
operation = command[0].upper()
mode = command[1].lower()
good = True
except ValueError:
decimal_places = context.preferences.addons[__package__].preferences.pdt_input_round
values_out = [str(round(float(v), decimal_places)) for v in values]
obj_loc = Vector((0,0,0))
verts = []
if mode_sel == 'REL' and operation not in {"C", "P"}:
pg.select = 'SEL'
mode_sel = 'SEL'
if mode == "a" and operation not in {"C", "P"}:
# Place new Vetex, or Extrude Vertices by Absolute Coords.
if mode_sel == 'REL':
pg.select = 'SEL'
if obj is not None:
if obj.mode == "EDIT":
bm = bmesh.from_edit_mesh(obj.data)
obj_loc = obj.matrix_world.decompose()[0]
verts = []
else:
pg.error = PDT_OBJ_MODE_ERROR
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_ObjectModeError
else:
pg.error = PDT_ERR_NO_ACT_OBJ
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_NoObjectError
if mode_sel == 'SEL' and mode not in {"a"}:
# All other options except Cursor or Pivot by Absolute
# These options require no object, etc.
bm, good = obj_check(obj, scene, operation)
if good and obj.mode == 'EDIT':
obj_loc = obj.matrix_world.decompose()[0]
if len(bm.select_history) == 0 or operation == "G":
verts = [v for v in bm.verts if v.select]
if len(verts) == 0:
pg.error = PDT_ERR_NO_SEL_GEOM
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_SelectionError
else:
verts = bm.select_history
debug(f"obj: {obj}, bm: {bm}, obj_loc: {obj_loc}")
def move_cursor_pivot(context, pg, operation, mode, obj, verts, values):
"""Moves Cursor & Pivot Point.
Args:
context: Blender bpy.context instance.
pg: PDT Parameters Group - our variables
operation: The Operation e.g. Create New Vertex
mode: The Operation Mode, e.g. a for Absolute
obj: The Active Object
verts: The object's selected vertices, or selected history vertices
values: The parameters passed e.g. 1,4,3 for Cartesian Coordinates
Returns:
Nothing.
"""
# Absolute/Global Coordinates, or Delta/Relative Coordinates
if mode in {"a", "d"}:
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
# Direction/Polar Coordinates
elif mode == "i":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
# Percent Options
else:
# Must be Percent
try:
vector_delta = vector_build(context, pg, obj, operation, values, 1)
except:
raise PDT_InvalidVector
scene = context.scene
mode_sel = pg.select
obj_loc = Vector((0,0,0))
if obj is not None:
obj_loc = obj.matrix_world.decompose()[0]
if mode == "a":
if operation == "C":
scene.cursor.location = vector_delta
elif operation == "P":
pg.pivot_loc = vector_delta
elif mode in {"d", "i"}:
if pg.plane == "LO" and mode == "d":
vector_delta = view_coords(vector_delta.x, vector_delta.y, vector_delta.z)
elif pg.plane == "LO" and mode == "i":
vector_delta = view_dir(pg.distance, pg.angle)
if mode_sel == "REL":
if operation == "C":
scene.cursor.location = scene.cursor.location + vector_delta
else:
pg.pivot_loc = pg.pivot_loc + vector_delta
elif mode_sel == "SEL":
if operation == "C":
scene.cursor.location = verts[-1].co + obj_loc + vector_delta
else:
pg.pivot_loc = verts[-1].co + obj_loc + vector_delta
if obj.mode == "OBJECT":
if operation == "C":
scene.cursor.location = obj_loc + vector_delta
else:
pg.pivot_loc = obj_loc + vector_delta
else:
# Must be Percent
if obj.mode == "EDIT":
if operation == "C":
scene.cursor.location = obj_loc + vector_delta
pg.pivot_loc = obj_loc + vector_delta
if obj.mode == "OBJECT":
if operation == "C":
scene.cursor.location = vector_delta
else:
pg.pivot_loc = vector_delta
def move_entities(context, pg, operation, mode, obj, bm, verts, values):
"""Moves Entities.
Args:
context: Blender bpy.context instance.
pg: PDT Parameters Group - our variables
operation: The Operation e.g. Create New Vertex
mode: The Operation Mode, e.g. a for Absolute
obj: The Active Object
verts: The object's selected vertices, or selected history vertices
values: The parameters passed e.g. 1,4,3 for Cartesian Coordinates
Returns:
Nothing.
"""
# Absolute/Global Coordinates
if mode == "a":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
if obj.mode == "EDIT":
bmesh.ops.remove_doubles(
bm, verts=[v for v in bm.verts if v.select], dist=0.0001
)
if obj.mode == "OBJECT":
for ob in context.view_layer.objects.selected:
ob.location = vector_delta
elif mode in {"d", "i"}:
if mode == "d":
# Delta/Relative Coordinates
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
else:
# Direction/Polar Coordinates
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
if pg.plane == "LO" and mode == "d":
vector_delta = view_coords(vector_delta.x, vector_delta.y, vector_delta.z)
elif pg.plane == "LO" and mode == "i":
vector_delta = view_dir(pg.distance, pg.angle)
if obj.mode == "EDIT":
bmesh.ops.translate(
bm, verts=[v for v in bm.verts if v.select], vec=vector_delta
)
if obj.mode == "OBJECT":
for ob in context.view_layer.objects.selected:
ob.location = obj_loc + vector_delta
# Percent Options Only Other Choice
else:
try:
vector_delta = vector_build(context, pg, obj, operation, values, 1)
except:
raise PDT_InvalidVector
if obj.mode == 'EDIT':
verts[-1].co = vector_delta
if obj.mode == "OBJECT":
obj.location = vector_delta
if obj.mode == 'EDIT':
bmesh.update_edit_mesh(obj.data)
bm.select_history.clear()
def add_new_vertex(context, pg, operation, mode, obj, bm, verts, values):
"""Add New Vertex.
Args:
context: Blender bpy.context instance.
pg, operation, mode, obj, bm, verts, values
Returns:
Nothing.
"""
obj_loc = obj.matrix_world.decompose()[0]
if not obj.mode == "EDIT":
pg.error = PDT_ERR_ADDVEDIT
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_SelectionError
if mode not in {"a"}:
if not isinstance(verts[0], bmesh.types.BMVert):
pg.error = PDT_ERR_VERT_MODE
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
raise PDT_FeatureError
# Absolute/Global Coordinates
if mode == "a":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
new_vertex = bm.verts.new(vector_delta - obj_loc)
# Delta/Relative Coordinates
elif mode == "d":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
if pg.plane == "LO":
vector_delta = view_coords(vector_delta.x, vector_delta.y, vector_delta.z)
new_vertex = bm.verts.new(verts[-1].co + vector_delta)
# Direction/Polar Coordinates
elif mode == "i":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
if pg.plane == "LO":
vector_delta = view_dir(pg.distance, pg.angle)
new_vertex = bm.verts.new(verts[-1].co + vector_delta)
# Percent Options Only Other Choice
else:
try:
vector_delta = vector_build(context, pg, obj, operation, values, 1)
except:
raise PDT_InvalidVector
new_vertex = bm.verts.new(vector_delta)
for v in [v for v in bm.verts if v.select]:
v.select_set(False)
new_vertex.select_set(True)
bmesh.update_edit_mesh(obj.data)
bm.select_history.clear()
def split_edges(context, pg, operation, mode, obj, obj_loc, bm, values):
"""Split Edges.
Args:
context: Blender bpy.context instance.
pg: PDT Parameters Group - our variables
operation: The Operation e.g. Create New Vertex
mode: The Operation Mode, e.g. a for Absolute
obj: The Active Object
obj_loc: The object's location in 3D space
bm: The object's Bmesh
verts: The object's selected vertices, or selected history vertices
values: The parameters passed e.g. 1,4,3 for Cartesian Coordinates
Returns:
Nothing.
"""
if not obj.mode == "EDIT":
pg.error = PDT_ERR_SPLITEDIT
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
# Absolute/Global Coordinates
if mode == "a":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
edges = [e for e in bm.edges if e.select]
if len(edges) != 1:
pg.error = f"{PDT_ERR_SEL_1_EDGE} {len(edges)})"
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
geom = bmesh.ops.subdivide_edges(bm, edges=edges, cuts=1)
new_verts = [v for v in geom["geom_split"] if isinstance(v, bmesh.types.BMVert)]
new_vertex = new_verts[0]
new_vertex.co = vector_delta - obj_loc
# Delta/Relative Coordinates
elif mode == "d":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
edges = [e for e in bm.edges if e.select]
faces = [f for f in bm.faces if f.select]
if len(faces) != 0:
pg.error = PDT_ERR_FACE_SEL
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
if len(edges) < 1:
pg.error = f"{PDT_ERR_SEL_1_EDGEM} {len(edges)})"
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
geom = bmesh.ops.subdivide_edges(bm, edges=edges, cuts=1)
new_verts = [v for v in geom["geom_split"] if isinstance(v, bmesh.types.BMVert)]
bmesh.ops.translate(bm, verts=new_verts, vec=vector_delta)
# Directional/Polar Coordinates
elif mode == "i":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
edges = [e for e in bm.edges if e.select]
faces = [f for f in bm.faces if f.select]
if len(faces) != 0:
pg.error = PDT_ERR_FACE_SEL
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
if len(edges) < 1:
pg.error = f"{PDT_ERR_SEL_1_EDGEM} {len(edges)})"
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
geom = bmesh.ops.subdivide_edges(bm, edges=edges, cuts=1)
new_verts = [v for v in geom["geom_split"] if isinstance(v, bmesh.types.BMVert)]
bmesh.ops.translate(bm, verts=new_verts, vec=vector_delta)
# Percent Options
elif mode == "p":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 1)
except:
raise PDT_InvalidVector
edges = [e for e in bm.edges if e.select]
faces = [f for f in bm.faces if f.select]
if len(faces) != 0:
pg.error = PDT_ERR_FACE_SEL
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
if len(edges) != 1:
pg.error = f"{PDT_ERR_SEL_1_EDGEM} {len(edges)})"
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
geom = bmesh.ops.subdivide_edges(bm, edges=edges, cuts=1)
new_verts = [v for v in geom["geom_split"] if isinstance(v, bmesh.types.BMVert)]
new_vertex = new_verts[0]
new_vertex.co = vector_delta
for v in [v for v in bm.verts if v.select]:
v.select_set(False)
for v in new_verts:
v.select_set(False)
bmesh.update_edit_mesh(obj.data)
bm.select_history.clear()
def extrude_vertices(context, pg, operation, mode, obj, obj_loc, bm, verts, values):
"""Extrude Vertices.
Args:
context: Blender bpy.context instance.
pg: PDT Parameters Group - our variables
operation: The Operation e.g. Create New Vertex
mode: The Operation Mode, e.g. a for Absolute
obj: The Active Object
obj_loc: The object's location in 3D space
bm: The object's Bmesh
verts: The object's selected vertices, or selected history vertices
values: The parameters passed e.g. 1,4,3 for Cartesian Coordinates
Returns:
Nothing.
"""
if not obj.mode == "EDIT":
pg.error = PDT_ERR_EXTEDIT
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
# Absolute/Global Coordinates
if mode == "a":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
new_vertex = bm.verts.new(vector_delta - obj_loc)
verts = [v for v in bm.verts if v.select].copy()
for v in verts:
bm.edges.new([v, new_vertex])
v.select_set(False)
new_vertex.select_set(True)
bmesh.ops.remove_doubles(
bm, verts=[v for v in bm.verts if v.select], dist=0.0001
# Delta/Relative Coordinates
elif mode == "d":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
if pg.plane == "LO":
vector_delta = view_coords(vector_delta.x, vector_delta.y, vector_delta.z)
for v in verts:
new_vertex = bm.verts.new(v.co)
new_vertex.co = new_vertex.co + vector_delta
bm.edges.new([v, new_vertex])
v.select_set(False)
new_vertex.select_set(True)
# Direction/Polar Coordinates
elif mode == "i":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
if pg.plane == "LO":
vector_delta = view_dir(pg.distance, pg.angle)
for v in verts:
new_vertex = bm.verts.new(v.co)
new_vertex.co = new_vertex.co + vector_delta
bm.edges.new([v, new_vertex])
v.select_set(False)
new_vertex.select_set(True)
# Percent Options
elif mode == "p":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 1)
except:
raise PDT_InvalidVector
verts = [v for v in bm.verts if v.select].copy()
new_vertex = bm.verts.new(vector_delta)
for v in [v for v in bm.verts if v.select]:
bm.edges.new([v, new_vertex])
v.select_set(False)
else:
bm.edges.new([verts[-1], new_vertex])
new_vertex.select_set(True)
bmesh.update_edit_mesh(obj.data)
def extrude_geometry(context, pg, operation, mode, obj, bm, values):
"""Extrude Geometry.
Args:
context: Blender bpy.context instance.
pg: PDT Parameters Group - our variables
operation: The Operation e.g. Create New Vertex
mode: The Operation Mode, e.g. a for Absolute
obj: The Active Object
bm: The object's Bmesh
values: The parameters passed e.g. 1,4,3 for Cartesian Coordinates
Returns:
Nothing.
"""
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
if not obj.mode == "EDIT":
pg.error = PDT_ERR_EXTEDIT
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
# Delta/Relative Coordinates
if mode == "d":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
# Direction/Polar Coordinates
elif mode == "i":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
ret = bmesh.ops.extrude_face_region(
bm,
geom=(
[f for f in bm.faces if f.select]
+ [e for e in bm.edges if e.select]
+ [v for v in bm.verts if v.select]
),
use_select_history=True,
)
geom_extr = ret["geom"]
verts_extr = [v for v in geom_extr if isinstance(v, bmesh.types.BMVert)]
edges_extr = [e for e in geom_extr if isinstance(e, bmesh.types.BMEdge)]
faces_extr = [f for f in geom_extr if isinstance(f, bmesh.types.BMFace)]
del ret
if pg.plane == "LO" and mode == "d":
vector_delta = view_coords(vector_delta.x, vector_delta.y, vector_delta.z)
elif pg.plane == "LO" and mode == "i":
vector_delta = view_dir(pg.distance, pg.angle)
bmesh.ops.translate(bm, verts=verts_extr, vec=vector_delta)
update_sel(bm, verts_extr, edges_extr, faces_extr)
bmesh.update_edit_mesh(obj.data)
bm.select_history.clear()
def duplicate_geometry(context, pg, operation, mode, obj, bm, values):
"""Duplicate Geometry.
Args:
context: Blender bpy.context instance.
pg: PDT Parameters Group - our variables
operation: The Operation e.g. Create New Vertex
mode: The Operation Mode, e.g. a for Absolute
obj: The Active Object
bm: The object's Bmesh
values: The parameters passed e.g. 1,4,3 for Cartesian Coordinates
Returns:
Nothing.
"""
if not obj.mode == "EDIT":
pg.error = PDT_ERR_DUPEDIT
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
# Delta/Relative Coordinates
if mode == "d":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except:
raise PDT_InvalidVector
# Direction/Polar Coordinates
elif mode == "i":
try:
vector_delta = vector_build(context, pg, obj, operation, values, 2)
except:
raise PDT_InvalidVector
ret = bmesh.ops.duplicate(
bm,
geom=(
[f for f in bm.faces if f.select]
+ [e for e in bm.edges if e.select]
+ [v for v in bm.verts if v.select]
),