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
"""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", "n"}) #new
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", "n"}) #new
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, v)")
label(text="D: Duplicate Geometry (d, i, v)")
label(text="E: Extrude Geometry (d, i, v)")
label(text="G: Grab (Move) (a, d, i, p, v)")
label(text="N: New Vertex (a, d, i, p, v)")
label(text="M: Maths Functions (a, d, p, o, x, y, z)")
label(text="P: Pivot Point (a, d, i, p, v)")
label(text="V: Extrude Vertice Only (a, d, i, p, v)")
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="n: Work in View Normal Axis")
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:
# View relative mode
if pg.plane == "XZ":
values = [0.0, values[0], 0.0]
elif pg.plane == "YZ":
values = [values[0], 0.0, 0.0]
elif pg.plane == "XY":
values = [0.0, 0.0, values[0]]
else:
if "-" in values[0]:
values = [0.0, 0.0, values[0][1:]]
else:
values = [0.0, 0.0, f"-{values[0]}"]
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 Vertex, 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
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", "n"}:
if pg.plane == "LO" and mode in {"d", "n"}:
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":
for v in [v for v in bm.verts if v.select]:
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", "n"}:
if mode in {"d", "n"}:
# 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 in {"d", "n"}:
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
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
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.
"""
if not obj.mode == "EDIT":
pg.error = PDT_ERR_EXTEDIT
context.window_manager.popup_menu(oops, title="Error", icon="ERROR")
return
# Delta/Relative Coordinates
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 in {"d", "n"}:
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")
try:
vector_delta = vector_build(context, pg, obj, operation, values, 3)
except: