Function run_quantify

src/ifcedit/ifcedit/quantify.py:16–37

Run quantity take-off on the model using the named rule. Modifies the model in-place by adding/updating IfcElementQuantity psets. Returns a summary dict with ok, rule, and elements_quantified.

(model: ifcopenshell.file, rule: str, selector: str | None = None)

def run_quantify(model: ifcopenshell.file, rule: str, selector: str | None = None) -> dict[str, Any]:
    """Run quantity take-off on the model using the named rule.

    Modifies the model in-place by adding/updating IfcElementQuantity psets.
    Returns a summary dict with ok, rule, and elements_quantified.
    """
    from ifc5d.qto import edit_qtos, quantify
    from ifc5d.qto import rules as rule_sets

    if rule not in rule_sets:
        return {"ok": False, "error": f"Unknown rule: {rule}. Available: {list(rule_sets.keys())}"}

    import ifcopenshell.util.selector

    if selector:
        elements = set(ifcopenshell.util.selector.filter_elements(model, selector))
    else:
        elements = set(model.by_type("IfcElement"))

    results = quantify(model, elements, rule_sets[rule])
    edit_qtos(model, results)
    return {"ok": True, "rule": rule, "elements_quantified": len(results)}