node = hou.pwd()
stage = node.editableStage()

# Add code to modify the stage.
# Use drop down menu to select examples.


yournode =  hou.node('/stage/null5').stage().GetPrimAtPath('/arcsmoke')

def count_prims(node):
    count = 0
    for child in node.GetChildren():
        count += 1
        count += count_prims(child)
    return count
    
prim_count = count_prims(yournode)
print("The node has", prim_count, "prims.")

from pxr import Usd
print(dir(Usd))

node = hou.pwd()
stage = node.editableStage()


def list_module_contents(module, prefix="", visited=None):
    contents = []

    # Initialize the visited set if it's not provided
    if visited is None:
        visited = set()

    # Prevent infinite recursion by checking if the module was already visited
    if id(module) in visited:
        return contents

    visited.add(id(module))

    for name in dir(module):
        # Check if the attribute is readable
        if not hasattr(module, name):
            continue

        attr = getattr(module, name)
        full_name = f"{prefix}{name}"

        if isinstance(attr, type):
            # If the attribute is a class, recursively list its contents
            contents.extend(list_module_contents(attr, f"{full_name}.", visited))
        else:
            contents.append(full_name)

    return contents




from pxr import Usd

# List everything in the Usd module recursively
contents = [i for i in list_module_contents(Usd) if '__' not in i]

print(contents)

node = hou.pwd()
stage = node.editableStage()
from pxr import Usd
# Add code to modify the stage.
# Use drop down menu to select examples.

xs = [x for x in stage.Traverse()]
y = xs[0]
for prim in xs:
    children = [c for c in prim.GetAllChildren()]
    models = [m for m in children if m.IsModel()]
    if prim.IsLoaded():
        print(prim, prim.GetTypeName(), models)


#print(dir(y))

from pxr import Usd

node = hou.pwd()
stage = node.editableStage()


user_node_path = hou.parm('nodepath').eval()
user_node = stage.GetPrimAtPath(user_node_path)

child_counts = {}

for child in user_node.GetChildren():
    child_type = child.GetTypeName()

    if child_type in child_counts:
        child_counts[child_type] += 1
    else:
        child_counts[child_type] = 1


print(f"USD Prims child count by type for node: {user_node_path}")
for child_type, count in child_counts.items():
    print(f"  {child_type}: {count}")
print("=============================")

hou.node("/stage/your_node").stagePrimStats()["Mesh:Total"]

# or even this one to get polygon count

hou.node("/stage/your_node").stagePrimStats(do_geometry_counts=True)["Mesh (Polygons):Total"]