yyc12345/BallanceBlenderHelper
1
0
Fork 0
Code Issues Pull Requests Releases Activity

[fix] fix fatal error when use flatten uv for sink floor top.

Browse Source 
- default behavior of flatten uv only suit for normal floor
- add extra property called scale correction to allow use specific custom uv scale.
This commit is contained in:
yyc12345 2023-01-28 20:44:39 +08:00
parent c9e51c9b6a
commit a300ddbb49
1 changed files with 28 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
ballance_blender_plugin/MODS_flatten_uv.py
View File

@ -1,5 +1,6 @@
import bpy,mathutils import bpy,mathutils
import bmesh import bmesh
import math
from . import UTILS_functions from . import UTILS_functions
class BALLANCE_OT_flatten_uv(bpy.types.Operator): class BALLANCE_OT_flatten_uv(bpy.types.Operator):
@ -8,6 +9,19 @@ class BALLANCE_OT_flatten_uv(bpy.types.Operator):
bl_label = "Flatten UV" bl_label = "Flatten UV"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
normal_scale_correction = 5.0
sink_scale_correction = 5.0 * (math.sqrt(2.5 ** 2 + 0.7 ** 2) / 2.5)
scale_correction: bpy.props.EnumProperty(
name="Scale Correction",
description="Choose your UV scale.",
items=(
("NORMAL", "Normal Floor", "Normal floor scale, 5.0"),
("SINK", "Sink Floor", "Sink floor scale, around 5.19")
),
default='NORMAL',
)
reference_edge : bpy.props.IntProperty( reference_edge : bpy.props.IntProperty(
name="Reference edge", name="Reference edge",
description="The references edge of UV. It will be placed in V axis.", description="The references edge of UV. It will be placed in V axis.",
@ -28,23 +42,26 @@ class BALLANCE_OT_flatten_uv(bpy.types.Operator):
return False return False
return True return True
""" def get_scale_correction(self):
def invoke(self, context, event): if self.scale_correction == 'NORMAL':
wm = context.window_manager return BALLANCE_OT_flatten_uv.normal_scale_correction
return wm.invoke_props_dialog(self) elif self.scale_correction == 'SINK':
""" return BALLANCE_OT_flatten_uv.sink_scale_correction
else:
raise Exception("Unknow scale correction.")
def execute(self, context): def execute(self, context):
no_processed_count = _real_flatten_uv(bpy.context.active_object.data, self.reference_edge) no_processed_count = _real_flatten_uv(bpy.context.active_object.data, self.reference_edge, self.get_scale_correction())
if no_processed_count != 0: if no_processed_count != 0:
print("[Flatten UV] {} faces may not be processed correctly because they have problem.".format(no_processed_count)) print("[Flatten UV] {} faces may not be processed correctly because they have problem.".format(no_processed_count))
return {'FINISHED'} return {'FINISHED'}
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
layout.prop(self, "scale_correction")
layout.prop(self, "reference_edge") layout.prop(self, "reference_edge")
def _real_flatten_uv(mesh, reference_edge): def _real_flatten_uv(mesh, reference_edge, scale_correction):
no_processed_count = 0 no_processed_count = 0
if mesh.uv_layers.active is None: if mesh.uv_layers.active is None:
@ -107,9 +124,11 @@ def _real_flatten_uv(mesh, reference_edge):
vec = pp-p1 vec = pp-p1
new_vec = transition_matrix @ vec new_vec = transition_matrix @ vec
# y axis always use 5.0 to scale
# however, x need use custom scale correction.
face.loops[loop_index][uv_lay].uv = ( face.loops[loop_index][uv_lay].uv = (
(new_vec.x if new_vec.x >=0 else -new_vec.x) / 5, (new_vec.x if new_vec.x >=0 else -new_vec.x) / scale_correction,
(new_vec.y) / 5 (new_vec.y) / 5.0
) )
# Show the updates in the viewport # Show the updates in the viewport