# # ##### 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 LICENSE BLOCK ##### __author__ = "imdjs, Nutti " __status__ = "production" __version__ = "5.1" __date__ = "24 Feb 2018" import math from math import atan2, sin, cos import bpy import bmesh from bpy.props import ( StringProperty, BoolProperty, IntProperty, EnumProperty, ) from mathutils import Vector from .. import common class MUV_CPUVCopyUV(bpy.types.Operator): """ Operation class: Copy UV coordinate """ bl_idname = "uv.muv_cpuv_copy_uv" bl_label = "Copy UV (Operation)" bl_description = "Copy UV coordinate (Operation)" bl_options = {'REGISTER', 'UNDO'} uv_map = StringProperty(options={'HIDDEN'}) def execute(self, context): props = context.scene.muv_props.cpuv if self.uv_map == "": self.report({'INFO'}, "Copy UV coordinate") else: self.report( {'INFO'}, "Copy UV coordinate (UV map:%s)" % (self.uv_map)) obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) if common.check_version(2, 73, 0) >= 0: bm.faces.ensure_lookup_table() # get UV layer if self.uv_map == "": if not bm.loops.layers.uv: self.report( {'WARNING'}, "Object must have more than one UV map") return {'CANCELLED'} uv_layer = bm.loops.layers.uv.verify() else: uv_layer = bm.loops.layers.uv[self.uv_map] # get selected face props.src_uvs = [] props.src_pin_uvs = [] props.src_seams = [] for face in bm.faces: if face.select: uvs = [l[uv_layer].uv.copy() for l in face.loops] pin_uvs = [l[uv_layer].pin_uv for l in face.loops] seams = [l.edge.seam for l in face.loops] props.src_uvs.append(uvs) props.src_pin_uvs.append(pin_uvs) props.src_seams.append(seams) if not props.src_uvs or not props.src_pin_uvs: self.report({'WARNING'}, "No faces are selected") return {'CANCELLED'} self.report({'INFO'}, "%d face(s) are selected" % len(props.src_uvs)) return {'FINISHED'} class MUV_CPUVCopyUVMenu(bpy.types.Menu): """ Menu class: Copy UV coordinate """ bl_idname = "uv.muv_cpuv_copy_uv_menu" bl_label = "Copy UV" bl_description = "Copy UV coordinate" def draw(self, context): layout = self.layout # create sub menu obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) uv_maps = bm.loops.layers.uv.keys() layout.operator( MUV_CPUVCopyUV.bl_idname, text="[Default]", icon="IMAGE_COL" ).uv_map = "" for m in uv_maps: layout.operator( MUV_CPUVCopyUV.bl_idname, text=m, icon="IMAGE_COL" ).uv_map = m class MUV_CPUVPasteUV(bpy.types.Operator): """ Operation class: Paste UV coordinate """ bl_idname = "uv.muv_cpuv_paste_uv" bl_label = "Paste UV (Operation)" bl_description = "Paste UV coordinate (Operation)" bl_options = {'REGISTER', 'UNDO'} uv_map = StringProperty(options={'HIDDEN'}) strategy = EnumProperty( name="Strategy", description="Paste Strategy", items=[ ('N_N', 'N:N', 'Number of faces must be equal to source'), ('N_M', 'N:M', 'Number of faces must not be equal to source') ], default="N_M" ) flip_copied_uv = BoolProperty( name="Flip Copied UV", description="Flip Copied UV...", default=False ) rotate_copied_uv = IntProperty( default=0, name="Rotate Copied UV", min=0, max=30 ) copy_seams = BoolProperty( name="Copy Seams", description="Copy Seams", default=True ) def execute(self, context): props = context.scene.muv_props.cpuv if not props.src_uvs or not props.src_pin_uvs: self.report({'WARNING'}, "Need copy UV at first") return {'CANCELLED'} if self.uv_map == "": self.report({'INFO'}, "Paste UV coordinate") else: self.report( {'INFO'}, "Paste UV coordinate (UV map:%s)" % (self.uv_map)) obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) if common.check_version(2, 73, 0) >= 0: bm.faces.ensure_lookup_table() # get UV layer if self.uv_map == "": if not bm.loops.layers.uv: self.report( {'WARNING'}, "Object must have more than one UV map") return {'CANCELLED'} uv_layer = bm.loops.layers.uv.verify() else: uv_layer = bm.loops.layers.uv[self.uv_map] # get selected face dest_uvs = [] dest_pin_uvs = [] dest_seams = [] dest_face_indices = [] for face in bm.faces: if face.select: dest_face_indices.append(face.index) uvs = [l[uv_layer].uv.copy() for l in face.loops] pin_uvs = [l[uv_layer].pin_uv for l in face.loops] seams = [l.edge.seam for l in face.loops] dest_uvs.append(uvs) dest_pin_uvs.append(pin_uvs) dest_seams.append(seams) if not dest_uvs or not dest_pin_uvs: self.report({'WARNING'}, "No faces are selected") return {'CANCELLED'} if self.strategy == 'N_N' and len(props.src_uvs) != len(dest_uvs): self.report( {'WARNING'}, "Number of selected faces is different from copied" + "(src:%d, dest:%d)" % (len(props.src_uvs), len(dest_uvs))) return {'CANCELLED'} # paste for i, idx in enumerate(dest_face_indices): suv = None spuv = None ss = None duv = None if self.strategy == 'N_N': suv = props.src_uvs[i] spuv = props.src_pin_uvs[i] ss = props.src_seams[i] duv = dest_uvs[i] elif self.strategy == 'N_M': suv = props.src_uvs[i % len(props.src_uvs)] spuv = props.src_pin_uvs[i % len(props.src_pin_uvs)] ss = props.src_seams[i % len(props.src_seams)] duv = dest_uvs[i] if len(suv) != len(duv): self.report({'WARNING'}, "Some faces are different size") return {'CANCELLED'} suvs_fr = [uv for uv in suv] spuvs_fr = [pin_uv for pin_uv in spuv] ss_fr = [s for s in ss] # flip UVs if self.flip_copied_uv is True: suvs_fr.reverse() spuvs_fr.reverse() ss_fr.reverse() # rotate UVs for _ in range(self.rotate_copied_uv): uv = suvs_fr.pop() pin_uv = spuvs_fr.pop() s = ss_fr.pop() suvs_fr.insert(0, uv) spuvs_fr.insert(0, pin_uv) ss_fr.insert(0, s) # paste UVs for l, suv, spuv, ss in zip(bm.faces[idx].loops, suvs_fr, spuvs_fr, ss_fr): l[uv_layer].uv = suv l[uv_layer].pin_uv = spuv if self.copy_seams is True: l.edge.seam = ss self.report({'INFO'}, "%d face(s) are copied" % len(dest_uvs)) bmesh.update_edit_mesh(obj.data) if self.copy_seams is True: obj.data.show_edge_seams = True return {'FINISHED'} class MUV_CPUVPasteUVMenu(bpy.types.Menu): """ Menu class: Paste UV coordinate """ bl_idname = "uv.muv_cpuv_paste_uv_menu" bl_label = "Paste UV" bl_description = "Paste UV coordinate" def draw(self, context): sc = context.scene layout = self.layout # create sub menu obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) uv_maps = bm.loops.layers.uv.keys() ops = layout.operator(MUV_CPUVPasteUV.bl_idname, text="[Default]") ops.uv_map = "" ops.copy_seams = sc.muv_cpuv_copy_seams ops.strategy = sc.muv_cpuv_strategy for m in uv_maps: ops = layout.operator(MUV_CPUVPasteUV.bl_idname, text=m) ops.uv_map = m ops.copy_seams = sc.muv_cpuv_copy_seams ops.strategy = sc.muv_cpuv_strategy class MUV_CPUVIECopyUV(bpy.types.Operator): """ Operation class: Copy UV coordinate on UV/Image Editor """ bl_idname = "uv.muv_cpuv_ie_copy_uv" bl_label = "Copy UV" bl_description = "Copy UV coordinate (only selected in UV/Image Editor)" bl_options = {'REGISTER', 'UNDO'} @classmethod def poll(cls, context): return context.mode == 'EDIT_MESH' def execute(self, context): props = context.scene.muv_props.cpuv obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) uv_layer = bm.loops.layers.uv.verify() if common.check_version(2, 73, 0) >= 0: bm.faces.ensure_lookup_table() for face in bm.faces: if not face.select: continue skip = False for l in face.loops: if not l[uv_layer].select: skip = True break if skip: continue props.src_uvs.append([l[uv_layer].uv.copy() for l in face.loops]) return {'FINISHED'} class MUV_CPUVIEPasteUV(bpy.types.Operator): """ Operation class: Paste UV coordinate on UV/Image Editor """ bl_idname = "uv.muv_cpuv_ie_paste_uv" bl_label = "Paste UV" bl_description = "Paste UV coordinate (only selected in UV/Image Editor)" bl_options = {'REGISTER', 'UNDO'} @classmethod def poll(cls, context): return context.mode == 'EDIT_MESH' def execute(self, context): props = context.scene.muv_props.cpuv obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) uv_layer = bm.loops.layers.uv.verify() if common.check_version(2, 73, 0) >= 0: bm.faces.ensure_lookup_table() dest_uvs = [] dest_face_indices = [] for face in bm.faces: if not face.select: continue skip = False for l in face.loops: if not l[uv_layer].select: skip = True break if skip: continue dest_face_indices.append(face.index) uvs = [l[uv_layer].uv.copy() for l in face.loops] dest_uvs.append(uvs) for suvs, duvs in zip(props.src_uvs, dest_uvs): src_diff = suvs[1] - suvs[0] dest_diff = duvs[1] - duvs[0] src_base = suvs[0] dest_base = duvs[0] src_rad = atan2(src_diff.y, src_diff.x) dest_rad = atan2(dest_diff.y, dest_diff.x) if src_rad < dest_rad: radian = dest_rad - src_rad elif src_rad > dest_rad: radian = math.pi * 2 - (src_rad - dest_rad) else: # src_rad == dest_rad radian = 0.0 ratio = dest_diff.length / src_diff.length break for suvs, fidx in zip(props.src_uvs, dest_face_indices): for l, suv in zip(bm.faces[fidx].loops, suvs): base = suv - src_base radian_ref = atan2(base.y, base.x) radian_fin = (radian + radian_ref) length = base.length turn = Vector((length * cos(radian_fin), length * sin(radian_fin))) target_uv = Vector((turn.x * ratio, turn.y * ratio)) + \ dest_base l[uv_layer].uv = target_uv bmesh.update_edit_mesh(obj.data) return {'FINISHED'} class MUV_CPUVSelSeqCopyUV(bpy.types.Operator): """ Operation class: Copy UV coordinate by selection sequence """ bl_idname = "uv.muv_cpuv_selseq_copy_uv" bl_label = "Copy UV (Selection Sequence) (Operation)" bl_description = "Copy UV data by selection sequence (Operation)" bl_options = {'REGISTER', 'UNDO'} uv_map = StringProperty(options={'HIDDEN'}) def execute(self, context): props = context.scene.muv_props.cpuv_selseq if self.uv_map == "": self.report({'INFO'}, "Copy UV coordinate (selection sequence)") else: self.report( {'INFO'}, "Copy UV coordinate (selection sequence) (UV map:%s)" % (self.uv_map)) obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) if common.check_version(2, 73, 0) >= 0: bm.faces.ensure_lookup_table() # get UV layer if self.uv_map == "": if not bm.loops.layers.uv: self.report( {'WARNING'}, "Object must have more than one UV map") return {'CANCELLED'} uv_layer = bm.loops.layers.uv.verify() else: uv_layer = bm.loops.layers.uv[self.uv_map] # get selected face props.src_uvs = [] props.src_pin_uvs = [] props.src_seams = [] for hist in bm.select_history: if isinstance(hist, bmesh.types.BMFace) and hist.select: uvs = [l[uv_layer].uv.copy() for l in hist.loops] pin_uvs = [l[uv_layer].pin_uv for l in hist.loops] seams = [l.edge.seam for l in hist.loops] props.src_uvs.append(uvs) props.src_pin_uvs.append(pin_uvs) props.src_seams.append(seams) if not props.src_uvs or not props.src_pin_uvs: self.report({'WARNING'}, "No faces are selected") return {'CANCELLED'} self.report({'INFO'}, "%d face(s) are selected" % len(props.src_uvs)) return {'FINISHED'} class MUV_CPUVSelSeqCopyUVMenu(bpy.types.Menu): """ Menu class: Copy UV coordinate by selection sequence """ bl_idname = "uv.muv_cpuv_selseq_copy_uv_menu" bl_label = "Copy UV (Selection Sequence)" bl_description = "Copy UV coordinate by selection sequence" def draw(self, context): layout = self.layout obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) uv_maps = bm.loops.layers.uv.keys() layout.operator( MUV_CPUVSelSeqCopyUV.bl_idname, text="[Default]", icon="IMAGE_COL").uv_map = "" for m in uv_maps: layout.operator( MUV_CPUVSelSeqCopyUV.bl_idname, text=m, icon="IMAGE_COL").uv_map = m class MUV_CPUVSelSeqPasteUV(bpy.types.Operator): """ Operation class: Paste UV coordinate by selection sequence """ bl_idname = "uv.muv_cpuv_selseq_paste_uv" bl_label = "Paste UV (Selection Sequence) (Operation)" bl_description = "Paste UV coordinate by selection sequence (Operation)" bl_options = {'REGISTER', 'UNDO'} uv_map = StringProperty(options={'HIDDEN'}) strategy = EnumProperty( name="Strategy", description="Paste Strategy", items=[ ('N_N', 'N:N', 'Number of faces must be equal to source'), ('N_M', 'N:M', 'Number of faces must not be equal to source') ], default="N_M" ) flip_copied_uv = BoolProperty( name="Flip Copied UV", description="Flip Copied UV...", default=False ) rotate_copied_uv = IntProperty( default=0, name="Rotate Copied UV", min=0, max=30 ) copy_seams = BoolProperty( name="Copy Seams", description="Copy Seams", default=True ) def execute(self, context): props = context.scene.muv_props.cpuv_selseq if not props.src_uvs or not props.src_pin_uvs: self.report({'WARNING'}, "Need copy UV at first") return {'CANCELLED'} if self.uv_map == "": self.report({'INFO'}, "Paste UV coordinate (selection sequence)") else: self.report( {'INFO'}, "Paste UV coordinate (selection sequence) (UV map:%s)" % (self.uv_map)) obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) if common.check_version(2, 73, 0) >= 0: bm.faces.ensure_lookup_table() # get UV layer if self.uv_map == "": if not bm.loops.layers.uv: self.report( {'WARNING'}, "Object must have more than one UV map") return {'CANCELLED'} uv_layer = bm.loops.layers.uv.verify() else: uv_layer = bm.loops.layers.uv[self.uv_map] # get selected face dest_uvs = [] dest_pin_uvs = [] dest_seams = [] dest_face_indices = [] for hist in bm.select_history: if isinstance(hist, bmesh.types.BMFace) and hist.select: dest_face_indices.append(hist.index) uvs = [l[uv_layer].uv.copy() for l in hist.loops] pin_uvs = [l[uv_layer].pin_uv for l in hist.loops] seams = [l.edge.seam for l in hist.loops] dest_uvs.append(uvs) dest_pin_uvs.append(pin_uvs) dest_seams.append(seams) if not dest_uvs or not dest_pin_uvs: self.report({'WARNING'}, "No faces are selected") return {'CANCELLED'} if self.strategy == 'N_N' and len(props.src_uvs) != len(dest_uvs): self.report( {'WARNING'}, "Number of selected faces is different from copied faces " + "(src:%d, dest:%d)" % (len(props.src_uvs), len(dest_uvs))) return {'CANCELLED'} # paste for i, idx in enumerate(dest_face_indices): suv = None spuv = None ss = None duv = None if self.strategy == 'N_N': suv = props.src_uvs[i] spuv = props.src_pin_uvs[i] ss = props.src_seams[i] duv = dest_uvs[i] elif self.strategy == 'N_M': suv = props.src_uvs[i % len(props.src_uvs)] spuv = props.src_pin_uvs[i % len(props.src_pin_uvs)] ss = props.src_seams[i % len(props.src_seams)] duv = dest_uvs[i] if len(suv) != len(duv): self.report({'WARNING'}, "Some faces are different size") return {'CANCELLED'} suvs_fr = [uv for uv in suv] spuvs_fr = [pin_uv for pin_uv in spuv] ss_fr = [s for s in ss] # flip UVs if self.flip_copied_uv is True: suvs_fr.reverse() spuvs_fr.reverse() ss_fr.reverse() # rotate UVs for _ in range(self.rotate_copied_uv): uv = suvs_fr.pop() pin_uv = spuvs_fr.pop() s = ss_fr.pop() suvs_fr.insert(0, uv) spuvs_fr.insert(0, pin_uv) ss_fr.insert(0, s) # paste UVs for l, suv, spuv, ss in zip(bm.faces[idx].loops, suvs_fr, spuvs_fr, ss_fr): l[uv_layer].uv = suv l[uv_layer].pin_uv = spuv if self.copy_seams is True: l.edge.seam = ss self.report({'INFO'}, "%d face(s) are copied" % len(dest_uvs)) bmesh.update_edit_mesh(obj.data) if self.copy_seams is True: obj.data.show_edge_seams = True return {'FINISHED'} class MUV_CPUVSelSeqPasteUVMenu(bpy.types.Menu): """ Menu class: Paste UV coordinate by selection sequence """ bl_idname = "uv.muv_cpuv_selseq_paste_uv_menu" bl_label = "Paste UV (Selection Sequence)" bl_description = "Paste UV coordinate by selection sequence" def draw(self, context): sc = context.scene layout = self.layout # create sub menu obj = context.active_object bm = bmesh.from_edit_mesh(obj.data) uv_maps = bm.loops.layers.uv.keys() ops = layout.operator(MUV_CPUVSelSeqPasteUV.bl_idname, text="[Default]") ops.uv_map = "" ops.copy_seams = sc.muv_cpuv_copy_seams ops.strategy = sc.muv_cpuv_strategy for m in uv_maps: ops = layout.operator(MUV_CPUVSelSeqPasteUV.bl_idname, text=m) ops.uv_map = m ops.copy_seams = sc.muv_cpuv_copy_seams ops.strategy = sc.muv_cpuv_strategy