# This file was generated with the assistance of an AI coding tool.

from __future__ import annotations

from typing import Any

import ifcopenshell

import ifcopenshell.util.cost as cost_util

def _cost_item_to_dict(item: ifcopenshell.entity_instance, max_depth: int | None, depth: int) -> dict[str, Any]:

raw_values = cost_util.get_cost_values(item)

values = [{"formula": v.get("label", ""), "category": v.get("category")} for v in raw_values]

if max_depth is not None and depth >= max_depth:

child_count = len(cost_util.get_nested_cost_items(item))

subitems = {"truncated": True, "count": child_count} if child_count else []

else:

subitems = [_cost_item_to_dict(sub, max_depth, depth + 1) for sub in cost_util.get_nested_cost_items(item)]

return {

"id": item.id(),

"name": getattr(item, "Name", None),

"values": values,

"subitems": subitems,

}

def cost(model: ifcopenshell.file, max_depth: int | None = None) -> list[dict[str, Any]]:

"""Return a list of IfcCostSchedule entries with nested cost item trees.

max_depth limits how many levels of subitems are expanded (None = unlimited).

At the cutoff level, subitems is replaced with {"truncated": True, "count": N}.

"""

result = []

for cost_schedule in model.by_type("IfcCostSchedule"):

items = [_cost_item_to_dict(i, max_depth, depth=1) for i in cost_util.get_root_cost_items(cost_schedule)]

result.append(

{

"id": cost_schedule.id(),

"name": getattr(cost_schedule, "Name", None),

"predefined_type": getattr(cost_schedule, "PredefinedType", None),

"items": items,

}

)

return result