Got it, let's sort out this selection detection for Maya! Your current code only grabs the first selected item but doesn't distinguish between faces, vertices, or full objects. Here's a straightforward implementation that matches the numeric return values you want (0 for faces, 1 for vertices, 3 for objects):

Working Code #

import maya.cmds as cmds

def get_selection_type():
    # Grab all selected items, flatten to handle individual components properly
    selected_items = cmds.ls(sl=True, flatten=True)
    
    # Handle empty selection case
    if not selected_items:
        return -1  # Adjust this default value if needed
    
    # Check the first selected item to determine its type
    first_selection = selected_items[0]
    
    # Maya uses unique suffixes for components: .f[...] for faces, .vtx[...] for vertices
    if '.f[' in first_selection:
        return 0
    elif '.vtx[' in first_selection:
        return 1
    # No component suffix means it's a full object
    else:
        return 3

# Test the function and implement your conditional logic
selection_code = get_selection_type()
print(f"Selection code: {selection_code}")

if selection_code == 0:
    # Replace with your face-specific actions
    print("Processing selected faces...")
elif selection_code == 1:
    # Replace with your vertex-specific actions
    print("Processing selected vertices...")
elif selection_code == 3:
    # Replace with your object-specific actions
    print("Processing selected objects...")
else:
    print("Nothing selected, or unknown selection type!")

How This Works #

• flatten=True: This flag ensures we get individual components (like each face or vertex) instead of parent objects wrapped in component groups—critical for accurately detecting selection types.
• String Pattern Checks: Maya appends distinct suffixes to component names. We check for .f[X] (faces) and .vtx[X] (vertices) to identify the selection category.
• Empty Selection Handling: The function returns -1 when nothing is selected; you can tweak this default if you need a different fallback value.

Quick Note #

If you need to handle mixed selections (e.g., selecting both faces and vertices at once), you'd need to add extra logic to detect that scenario. But this code assumes you're selecting one type at a time, which aligns with your original conditional flow.

内容的提问来源于stack exchange，提问作者JokerMartini