Skip to content

Commit

Permalink
Cleanup: formatting using autopep8
Browse files Browse the repository at this point in the history
  • Loading branch information
Germano Cavalcante committed Apr 19, 2022
1 parent 84367c7 commit 16c87dc
Show file tree
Hide file tree
Showing 9 changed files with 352 additions and 264 deletions.
21 changes: 15 additions & 6 deletions mesh_snap_utilities_line/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"blender": (3, 2, 0),
"location": "View3D > TOOLS > Line Tool",
"description": "Extends Blender Snap controls",
"doc_url" : "{BLENDER_MANUAL_URL}/addons/mesh/snap_utilities_line.html",
"doc_url": "{BLENDER_MANUAL_URL}/addons/mesh/snap_utilities_line.html",
"category": "Mesh",
}

Expand All @@ -28,15 +28,19 @@
from . import op_line
from . import keys

# autopep8: off
import bpy
from bpy.utils.toolsystem import ToolDef
# autopep8: on

if not __package__:
__package__ = "mesh_snap_utilities_line"


@ToolDef.from_fn
def tool_line():
import os

def draw_settings(context, layout, tool):
addon_prefs = context.preferences.addons[__package__].preferences

Expand All @@ -60,7 +64,7 @@ def draw_settings(context, layout, tool):
),
icon=os.path.join(icons_dir, "ops.mesh.snap_utilities_line"),
widget="MESH_GGT_snap_point",
#operator="mesh.snap_utilities_line",
# operator="mesh.snap_utilities_line",
keymap=keys.km_tool_snap_utilities_line,
draw_settings=draw_settings,
)
Expand Down Expand Up @@ -91,7 +95,7 @@ def register_snap_tools():
def unregister_snap_tools():
tools = get_tool_list('VIEW_3D', 'EDIT_MESH')

index = tools.index(tool_line) - 1 #None
index = tools.index(tool_line) - 1 # None
tools.pop(index)
tools.remove(tool_line)

Expand All @@ -109,11 +113,15 @@ def register_keymaps():

# TODO: find the user defined tool_mouse.
from bl_keymap_utils.io import keyconfig_init_from_data
keyconfig_init_from_data(kc_defaultconf, keys.generate_empty_snap_utilities_tools_keymaps())
keyconfig_init_from_data(kc_addonconf, keys.generate_snap_utilities_keymaps())
keyconfig_init_from_data(
kc_defaultconf, keys.generate_empty_snap_utilities_tools_keymaps())
keyconfig_init_from_data(
kc_addonconf, keys.generate_snap_utilities_keymaps())

#snap_modalkeymap = kc_addonconf.keymaps.find(keys.km_snap_utilities_modal_keymap)
#snap_modalkeymap.assign("MESH_OT_snap_utilities_line")
# snap_modalkeymap.assign("MESH_OT_snap_utilities_line")


def unregister_keymaps():
keyconfigs = bpy.context.window_manager.keyconfigs
defaultmap = getattr(keyconfigs.get("Blender"), "keymaps", None)
Expand Down Expand Up @@ -152,6 +160,7 @@ def unregister_keymaps():
widgets.SnapPointWidgetGroup,
)


def register():
for cls in classes:
bpy.utils.register_class(cls)
Expand Down
122 changes: 73 additions & 49 deletions mesh_snap_utilities_line/common_classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,14 @@
from mathutils import (
Vector,
Matrix,
)
)
from mathutils.geometry import intersect_point_line
from .drawing_utilities import SnapDrawn
from .common_utilities import (
convert_distance,
get_units_info,
location_3d_to_region_2d,
)
)


class SnapNavigation():
Expand All @@ -26,7 +26,6 @@ class SnapNavigation():
'_ndof_orbit_zoom',
'_ndof_pan')


@staticmethod
def debug_key(key):
for member in dir(key):
Expand All @@ -53,34 +52,44 @@ def __init__(self, context, use_ndof):

for key in context.window_manager.keyconfigs.user.keymaps['3D View'].keymap_items:
if key.idname == 'view3d.rotate':
self._rotate.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type, key.value))
self._rotate.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), key.type, key.value))
elif key.idname == 'view3d.move':
self._move.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type, key.value))
self._move.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), key.type, key.value))
elif key.idname == 'view3d.zoom':
if key.type == 'WHEELINMOUSE':
self._zoom.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), 'WHEELUPMOUSE', key.value, key.properties.delta))
self._zoom.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), 'WHEELUPMOUSE', key.value, key.properties.delta))
elif key.type == 'WHEELOUTMOUSE':
self._zoom.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), 'WHEELDOWNMOUSE', key.value, key.properties.delta))
self._zoom.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), 'WHEELDOWNMOUSE', key.value, key.properties.delta))
else:
self._zoom.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type, key.value, key.properties.delta))
self._zoom.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), key.type, key.value, key.properties.delta))

elif self.use_ndof:
if key.idname == 'view3d.ndof_all':
self._ndof_all.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type))
self._ndof_all.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), key.type))
elif key.idname == 'view3d.ndof_orbit':
self._ndof_orbit.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type))
self._ndof_orbit.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), key.type))
elif key.idname == 'view3d.ndof_orbit_zoom':
self._ndof_orbit_zoom.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type))
self._ndof_orbit_zoom.add(
(self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type))
elif key.idname == 'view3d.ndof_pan':
self._ndof_pan.add((self.convert_to_flag(key.shift, key.ctrl, key.alt), key.type))

self._ndof_pan.add((self.convert_to_flag(
key.shift, key.ctrl, key.alt), key.type))

def run(self, context, event, snap_location):
evkey = (self.convert_to_flag(event.shift, event.ctrl, event.alt), event.type, event.value)
evkey = (self.convert_to_flag(event.shift, event.ctrl,
event.alt), event.type, event.value)

if evkey in self._rotate:
if snap_location:
bpy.ops.view3d.rotate_custom_pivot('INVOKE_DEFAULT', pivot=snap_location)
bpy.ops.view3d.rotate_custom_pivot(
'INVOKE_DEFAULT', pivot=snap_location)
else:
bpy.ops.view3d.rotate('INVOKE_DEFAULT', use_cursor_init=True)
return True
Expand All @@ -93,7 +102,8 @@ def run(self, context, event, snap_location):
if evkey == key[0:3]:
if key[3]:
if snap_location:
bpy.ops.view3d.zoom_custom_target('INVOKE_DEFAULT', delta=key[3], target=snap_location)
bpy.ops.view3d.zoom_custom_target(
'INVOKE_DEFAULT', delta=key[3], target=snap_location)
else:
bpy.ops.view3d.zoom('INVOKE_DEFAULT', delta=key[3])
else:
Expand Down Expand Up @@ -132,11 +142,11 @@ class CharMap:
"c", "m", "d", "k", "h", "a",
" ", "/", "*", "'", "\""
# "="
}
}
type = {
'BACK_SPACE', 'DEL',
'LEFT_ARROW', 'RIGHT_ARROW'
}
}

def __init__(self, context):
scale = context.scene.unit_settings.scale_length
Expand All @@ -155,28 +165,34 @@ def modal_(self, context, event):
pos = self.line_pos
if ascii == ",":
ascii = "."
self.length_entered = self.length_entered[:pos] + ascii + self.length_entered[pos:]
self.length_entered = self.length_entered[:pos] + \
ascii + self.length_entered[pos:]
self.line_pos += 1

if self.length_entered:
pos = self.line_pos
if type == 'BACK_SPACE':
self.length_entered = self.length_entered[:pos - 1] + self.length_entered[pos:]
self.length_entered = self.length_entered[:pos -
1] + self.length_entered[pos:]
self.line_pos -= 1

elif type == 'DEL':
self.length_entered = self.length_entered[:pos] + self.length_entered[pos + 1:]
self.length_entered = self.length_entered[:pos] + \
self.length_entered[pos + 1:]

elif type == 'LEFT_ARROW':
self.line_pos = (pos - 1) % (len(self.length_entered) + 1)
self.line_pos = (
pos - 1) % (len(self.length_entered) + 1)

elif type == 'RIGHT_ARROW':
self.line_pos = (pos + 1) % (len(self.length_entered) + 1)
self.line_pos = (
pos + 1) % (len(self.length_entered) + 1)

try:
self.length_entered_value = bpy.utils.units.to_value(self.unit_system, 'LENGTH', self.length_entered)
self.length_entered_value = bpy.utils.units.to_value(
self.unit_system, 'LENGTH', self.length_entered)
except: # ValueError:
self.length_entered_value = 0.0 #invalid
self.length_entered_value = 0.0 # invalid
#self.report({'INFO'}, "Operation not supported yet")
else:
self.length_entered_value = 0.0
Expand Down Expand Up @@ -236,7 +252,7 @@ def _constrain_set(self, mcursor):
vec = self.orientation[self.orientation_id][1]
type = 'Y'

else: # dot_z > dot_y and dot_z > dot_x:
else: # dot_z > dot_y and dot_z > dot_x:
vec = self.orientation[self.orientation_id][2]
type = 'Z'

Expand Down Expand Up @@ -280,7 +296,7 @@ def modal(self, event, shift_callback):
return True

def toggle(self):
self.rotMat = None # update
self.rotMat = None # update
if self.preferences.auto_constrain:
self.orientation_id = (self.orientation_id + 1) % 2
self.preferences.auto_constrain = self.orientation_id != 0
Expand All @@ -292,14 +308,18 @@ def update(self, region, rv3d, mcursor, center):
self.rotMat = rv3d.view_matrix.copy()

self.center = center.copy()
self.center_2d = location_3d_to_region_2d(region, rv3d, self.center)
self.center_2d = location_3d_to_region_2d(
region, rv3d, self.center)

vec = self.center + self.orientation[self.orientation_id][0]
self.projected_vecs[0] = location_3d_to_region_2d(region, rv3d, vec) - self.center_2d
self.projected_vecs[0] = location_3d_to_region_2d(
region, rv3d, vec) - self.center_2d
vec = self.center + self.orientation[self.orientation_id][1]
self.projected_vecs[1] = location_3d_to_region_2d(region, rv3d, vec) - self.center_2d
self.projected_vecs[1] = location_3d_to_region_2d(
region, rv3d, vec) - self.center_2d
vec = self.center + self.orientation[self.orientation_id][2]
self.projected_vecs[2] = location_3d_to_region_2d(region, rv3d, vec) - self.center_2d
self.projected_vecs[2] = location_3d_to_region_2d(
region, rv3d, vec) - self.center_2d

self.projected_vecs[0].normalize()
self.projected_vecs[1].normalize()
Expand Down Expand Up @@ -331,12 +351,12 @@ class SnapUtilities:
"""

constrain_keys = {
'X': Vector((1,0,0)),
'Y': Vector((0,1,0)),
'Z': Vector((0,0,1)),
'X': Vector((1, 0, 0)),
'Y': Vector((0, 1, 0)),
'Z': Vector((0, 0, 1)),
'RIGHT_SHIFT': 'shift',
'LEFT_SHIFT': 'shift',
}
}

snapwidgets = []
constrain = None
Expand Down Expand Up @@ -364,7 +384,6 @@ def set_contrain(context, key):

SnapUtilities.constrain = key


def snap_context_update_and_return_moving_objects(self, context):
moving_objects = set()
moving_snp_objects = set()
Expand Down Expand Up @@ -405,7 +424,6 @@ def snap_context_update_and_return_moving_objects(self, context):
del children
return moving_objects, moving_snp_objects


def snap_context_update(self, context):
def visible_objects_and_duplis():
if self.preferences.outer_verts:
Expand All @@ -425,12 +443,12 @@ def visible_objects_and_duplis():
for obj, matrix in visible_objects_and_duplis():
self.sctx.add_obj(obj, matrix)


def snap_context_init(self, context, snap_edge_and_vert=True):
from .snap_context_l import global_snap_context_get

#Create Snap Context
self.sctx = global_snap_context_get(context.evaluated_depsgraph_get(), context.region, context.space_data)
# Create Snap Context
self.sctx = global_snap_context_get(
context.evaluated_depsgraph_get(), context.region, context.space_data)
ui_scale = context.preferences.system.ui_scale
self.sctx.set_pixel_dist(12 * ui_scale)

Expand All @@ -448,7 +466,7 @@ def snap_context_init(self, context, snap_edge_and_vert=True):
self.normal = widget.normal

else:
#init these variables to avoid errors
# init these variables to avoid errors
self.obj = context.active_object
self.bm = None
self.geom = None
Expand All @@ -458,7 +476,7 @@ def snap_context_init(self, context, snap_edge_and_vert=True):
preferences = context.preferences.addons[__package__].preferences
self.preferences = preferences

#Init DrawCache
# Init DrawCache
self.draw_cache = SnapDrawn(
preferences.out_color,
preferences.face_color,
Expand All @@ -467,26 +485,32 @@ def snap_context_init(self, context, snap_edge_and_vert=True):
preferences.center_color,
preferences.perpendicular_color,
preferences.constrain_shift_color,
tuple(context.preferences.themes[0].user_interface.axis_x) + (1.0,),
tuple(context.preferences.themes[0].user_interface.axis_y) + (1.0,),
tuple(context.preferences.themes[0].user_interface.axis_z) + (1.0,),
tuple(
context.preferences.themes[0].user_interface.axis_x) + (1.0,),
tuple(
context.preferences.themes[0].user_interface.axis_y) + (1.0,),
tuple(
context.preferences.themes[0].user_interface.axis_z) + (1.0,),
self.sctx.rv3d,
ui_scale)

self.snap_vert = self.snap_edge = snap_edge_and_vert

shading = context.space_data.shading
self.snap_face = not (snap_edge_and_vert and (shading.show_xray or shading.type == 'WIREFRAME'))
self.snap_face = not (snap_edge_and_vert and (
shading.show_xray or shading.type == 'WIREFRAME'))

self.sctx.set_snap_mode(self.snap_vert, self.snap_edge, self.snap_face)

#Configure the unit of measure
# Configure the unit of measure
unit_system = context.scene.unit_settings.system
scale = context.scene.unit_settings.scale_length
scale /= context.space_data.overlay.grid_scale
self.rd = bpy.utils.units.to_value(unit_system, 'LENGTH', str(1 / scale))
self.rd = bpy.utils.units.to_value(
unit_system, 'LENGTH', str(1 / scale))

self.incremental = bpy.utils.units.to_value(unit_system, 'LENGTH', str(self.preferences.incremental))
self.incremental = bpy.utils.units.to_value(
unit_system, 'LENGTH', str(self.preferences.incremental))

def snap_to_grid(self):
if self.type == 'OUT' and self.preferences.increments_grid:
Expand Down
Loading

0 comments on commit 16c87dc

Please sign in to comment.