Function get_texture_for_object

data_generation/object_texture_mapping.py:283–319

Get appropriate texture path for an object, or None if it should use color. Args: obj_name: Name of the object Returns: str: Path to texture directory, or None if object should use solid color

(obj_name)

def get_texture_for_object(obj_name):
    """
    Get appropriate texture path for an object, or None if it should use color.

    Args:
        obj_name: Name of the object

    Returns:
        str: Path to texture directory, or None if object should use solid color
    """
    # Check if object should use texture
    if obj_name not in TEXTURED_OBJECTS:
        return None

    # Get possible texture categories for this object
    possible_categories = TEXTURED_OBJECTS[obj_name]

    # Randomly select one category from the possibilities
    category = random.choice(possible_categories)

    # Get available textures in this category
    if category not in TEXTURE_CATEGORIES:
        print(f"  Warning: Unknown texture category '{category}' for {obj_name}")
        return None

    available_textures = TEXTURE_CATEGORIES[category]

    # Select random texture from category
    texture_name = random.choice(available_textures)
    texture_path = os.path.join(TEXTURES_BASE, texture_name)

    # Verify texture exists
    if not os.path.exists(texture_path):
        print(f"  Warning: Texture path not found: {texture_path}")
        return None

    return texture_path


def get_color_for_object(obj_name, character_name=None):