[import maya.cmds as cmds
import maya.mel as mel

def remove_isolated_vertices(mesh):
    count = 0
    try:
        total_verts = cmds.polyEvaluate(mesh, vertex=True)
        verts_to_delete = []

        for i in range(total_verts):
            vert = "{}.vtx[{}]".format(mesh, i)
            connected_edges = cmds.polyListComponentConversion(
                vert,
                fromVertex=True,
                toEdge=True
            )
            if not connected_edges:
                verts_to_delete.append(vert)
                count += 1

        if verts_to_delete:
            cmds.delete(verts_to_delete)
            print(">>> DA XOA: {} isolated vertices".format(count))

    except Exception as e:
        print("LOI:", e)

    return count


def remove_nonmanifold_vertices(mesh):
    count = 0
    try:
        cmds.select(mesh, replace=True)
        cmds.selectMode(component=True)
        cmds.selectType(vertex=True)

        mel.eval('polySelectConstraint -mode 3 -type 0x0001 -topology 2')
        nonmanifold_verts = cmds.ls(selection=True, flatten=True)
        mel.eval('polySelectConstraint -mode 0')

        if nonmanifold_verts:
            count = len(nonmanifold_verts)
            cmds.delete(nonmanifold_verts)
            print(">>> DA XOA: {} non-manifold vertices".format(count))

        cmds.selectMode(object=True)

    except Exception as e:
        print("LOI:", e)
        cmds.selectMode(object=True)

    return count


def remove_zero_length_edges_vertices(mesh):
    count = 0
    tolerance = 0.0001

    try:
        total_edges = cmds.polyEvaluate(mesh, edge=True)
        edges_to_delete = []

        for i in range(total_edges):
            edge = "{}.e[{}]".format(mesh, i)
            info = cmds.polyInfo(edge, edgeLength=True)

            if info:
                length = float(info[0].split(':')[-1].strip())
                if length < tolerance:
                    edges_to_delete.append(edge)

        if edges_to_delete:
            count = len(edges_to_delete)
            cmds.delete(edges_to_delete)
            print(">>> DA XOA: {} zero-length edges".format(count))

    except Exception as e:
        print("LOI:", e)

    return count


def merge_duplicate_vertices(mesh, distance=0.001):
    count = 0
    try:
        initial_verts = cmds.polyEvaluate(mesh, vertex=True)
        cmds.select(mesh, replace=True)

        cmds.polyMergeVertex(
            mesh,
            distance=distance,
            alwaysMergeTwoVertices=False,
            constructionHistory=False
        )

        final_verts = cmds.polyEvaluate(mesh, vertex=True)
        count = initial_verts - final_verts

        if count > 0:
            print(">>> DA MERGE: {} vertices".format(count))

    except Exception as e:
        print("LOI:", e)

    return count


def full_vertex_cleanup(mesh):
    print("\nCLEANUP:", mesh)

    results = {
        'merged': 0,
        'zero_edges': 0,
        'isolated': 0,
        'nonmanifold': 0
    }

    try:
        results['merged'] = merge_duplicate_vertices(mesh, 0.001)
        results['zero_edges'] = remove_zero_length_edges_vertices(mesh)
        results['isolated'] = remove_isolated_vertices(mesh)
        results['nonmanifold'] = remove_nonmanifold_vertices(mesh)

    except Exception as e:
        print("LOI:", e)

    return results


def main():
    selected = cmds.ls(selection=True, transforms=True)

    if not selected:
        all_transforms = cmds.ls(type='transform')
        selected = []
        for obj in all_transforms:
            shapes = cmds.listRelatives(obj, shapes=True, noIntermediate=True)
            if shapes and cmds.nodeType(shapes[0]) == 'mesh':
                selected.append(obj)

    if not selected:
        cmds.confirmDialog(
            title='Loi',
            message='Khong tim thay mesh',
            button=['OK']
        )
        return

    total_results = {
        'merged': 0,
        'zero_edges': 0,
        'isolated': 0,
        'nonmanifold': 0
    }

    for mesh in selected:
        shapes = cmds.listRelatives(mesh, shapes=True, noIntermediate=True)
        if not shapes or cmds.nodeType(shapes[0]) != 'mesh':
            continue

        results = full_vertex_cleanup(mesh)

        for k in total_results:
            total_results[k] += results[k]

    total_fixed = sum(total_results.values())

    if total_fixed > 0:
        cmds.confirmDialog(
            title='Hoan tat',
            message='DA SUA {} vertices'.format(total_fixed),
            button=['OK'],
            icon='information'
        )
    else:
        cmds.confirmDialog(
            title='Ket qua',
            message='Mesh sach se',
            button=['OK'],
            icon='information'
        )


main()
]