hardops / operators / meshtools / multi_tool / join.py

from .utils import *
from .... utility.shader import point_uniform


class Join:
    def setup(self):
        self.start = False
        self.first_vert = None
        self.second_vert = None

        # Overrides
        self.override_release_confirm = False


    def update(self, context, event, data, op):
        # Cancel
        if event.type == 'C' and event.value == "PRESS":
            self.setup()

        # (1) First click : Get vert under mouse
        if self.start == False:
            if event.type == 'LEFTMOUSE' and event.value == 'PRESS':
                self.first_vert = get_vert_under_mouse(context, event, data)
                if self.first_vert != None:
                    self.start = True

        # (2) Mouse move : Show line to next vert under mouse
        if self.start == True:
            if event.type == "MOUSEMOVE":
                self.second_vert = get_vert_under_mouse(context, event, data)
                if self.first_vert == self.second_vert:
                    self.second_vert = None

        # (3) Second click : Join the (first vert, second vert) and start over
        if self.start == True:

            confirm = True if event.type == 'LEFTMOUSE' and event.value == 'PRESS' else False
            if self.override_release_confirm:
                confirm = True if event.type == 'LEFTMOUSE' and event.value == 'RELEASE' else confirm

            if confirm and self.first_vert and self.second_vert:
                bpy.ops.mesh.select_all(action='DESELECT')
                self.first_vert.select = True
                self.second_vert.select = True
                bpy.ops.mesh.vert_connect_path()
                # bmesh.ops.connect_vert_pair(data.bm, verts=verts)
                self.setup()
                data.save()


    def help(self):
        return [
            ("C",     "Cancel selection"),
            ("Click", "Click 2 verts to join at last"),
            ("", "________JOIN________")]


    def draw_2d(self, context, data, op):
        pass


    def draw_3d(self, context, data, op):

        if self.first_vert != None:

            shader = point_uniform.compile()
            batch = batch_for_shader(shader, 'POINTS', {'pos': [data.obj.matrix_world @ self.first_vert.co]})
            shader.bind()
            shader.uniform_float('color', (1,0,0,1))
            shader.uniform_float('size', 6)
            gpu.state.blend_set('ALPHA')
            gpu.state.program_point_size_set(True)
            batch.draw(shader)
            gpu.state.program_point_size_set(False)
            del shader
            del batch

        if self.second_vert != None:

            first_pos = data.obj.matrix_world @ self.first_vert.co
            second_pos = data.obj.matrix_world @ self.second_vert.co

            verts = [first_pos, second_pos]
            indices = [(0,1)]

            built_in_shader = 'UNIFORM_COLOR' if bpy.app.version[0] >=4 else '3D_UNIFORM_COLOR'
            shader = gpu.shader.from_builtin(built_in_shader)
            batch = batch_for_shader(shader, 'LINES', {'pos': verts}, indices=indices)
            shader.bind()
            shader.uniform_float('color', (0,0,0,1))
            #Enable(GL_LINE_SMOOTH)
            gpu.state.blend_set('ALPHA')
            gpu.state.line_width_set(3)
            batch.draw(shader)
            del shader
            del batch

        if self.first_vert != None and self.second_vert != None:

            shader = point_uniform.compile()
            batch = batch_for_shader(shader, 'POINTS', {'pos': [data.obj.matrix_world @ self.second_vert.co]})
            shader.bind()
            shader.uniform_float('color', (1,0,0,1))
            shader.uniform_float('size', 6)
            gpu.state.blend_set('ALPHA')
            gpu.state.program_point_size_set(True)
            batch.draw(shader)
            gpu.state.program_point_size_set(False)
            del shader
            del batch