# IfcOpenShell - IFC toolkit and geometry engine
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
#
# This file is part of IfcOpenShell.
#
# IfcOpenShell is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# IfcOpenShell is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with IfcOpenShell.  If not, see <http://www.gnu.org/licenses/>.

import ifcopenshell.api
import ifcopenshell.util.date


class Usecase:
    def __init__(self, file, cost_item=None):
        """Calculates the total cost of all resources associated with a cost item

        A cost item may have construction resources (e.g. equipment, material,
        etc) assigned to it. Construction resources may be assigned directly to
        the cost item, or assigned first to a task, and the task is then
        assigned to the cost item.

        The cost of a resource is calculated by the total sum of all of its base
        costs. If no quantity is provided, that sum is considered to be the
        total cost. Otherwise, it is considered to be a unit cost, and is then
        multiplied by the resource quantity. The quantity is either stored as a
        base quantity (such as a volume) for a things like material resources,
        or as a duration as a daily rate for labour resources.

        The final calculated cost is set as the cost item's value. Any
        previously existing values are removed.

        :param cost_item: The IfcCostItem to calculate
        :type cost_item: ifccopenshell.entity_instance.entity_instance
        :return: None
        :rtype: None

        Example:

        .. code:: python

            # First, we need a cost schedule and item
            schedule = ifcopenshell.api.run("cost.add_cost_schedule", model)
            item = ifcopenshell.api.run("cost.add_cost_item", model, cost_schedule=schedule)

            # Let's imagine we have our own formworking crew
            crew = ifcopenshell.api.run("resource.add_resource", model, ifc_class="IfcCrewResource")

            # ... and they need concrete
            concrete = ifcopenshell.api.run("resource.add_resource", model,
                ifc_class="IfcConstructionMaterialResource", parent_resource=crew)
            ifcopenshell.api.run("control.assign_control", model,
                relating_control=item, related_object=concrete)
            # ... which has a unit price of 42.0 per m3
            value = ifcopenshell.api.run("cost.add_cost_value", model, parent=concrete)
            ifcopenshell.api.run("cost.edit_cost_value", model, cost_value=value,
                attributes={"AppliedValue": 42.0})
            # ... and a volume of 200m3
            quantity = ifcopenshell.api.run("resource.add_resource_quantity", model,
                resource=concrete, ifc_class="IfcQuantityVolume")
            ifcopenshell.api.run("resource.edit_resource_quantity", model,
                physical_quantity=quantity, "attributes": {"VolumeValue": 200.0})

            # Let's say they also need some equipment
            equipment = ifcopenshell.api.run("resource.add_resource", model,
                ifc_class="IfcConstructionEquipmentResource", parent_resource=crew)
            ifcopenshell.api.run("control.assign_control", model,
                relating_control=item, related_object=equipment)
            # ... with a fixed price of 50,000
            value = ifcopenshell.api.run("cost.add_cost_value", model, parent=concrete)
            ifcopenshell.api.run("cost.edit_cost_value", model, cost_value=value,
                attributes={"AppliedValue": 42.0})

            # (42 * 200) + 50000 = 58400 is our calculated cost
            ifcopenshell.api.run("cost.calculate_cost_item_resource_value", model, cost_item=item)
        """
        self.file = file
        self.settings = {"cost_item": cost_item}

    def execute(self):
        for cost_value in self.settings["cost_item"].CostValues or []:
            ifcopenshell.api.run(
                "cost.remove_cost_value", self.file, parent=self.settings["cost_item"], cost_value=cost_value
            )

        resources = []
        for rel in self.settings["cost_item"].Controls or []:
            for related_object in rel.RelatedObjects:
                if related_object.is_a("IfcConstructionResource"):
                    resources.append(related_object)
                elif related_object.is_a("IfcTask"):
                    for rel2 in related_object.OperatesOn or []:
                        for related_object2 in rel2.RelatedObjects:
                            if related_object2.is_a("IfcConstructionResource"):
                                resources.append(related_object2)

        total_cost = 0
        for resource in resources:
            cost = self.get_cost(resource)
            quantity = self.get_quantity(resource)
            if not cost or not quantity:
                continue
            total_cost += cost * quantity

        if total_cost:
            cost_value = ifcopenshell.api.run("cost.add_cost_value", self.file, parent=self.settings["cost_item"])
            cost_value.AppliedValue = self.file.createIfcMonetaryMeasure(total_cost)

    def get_cost(self, resource):
        total = 0
        for cost_value in resource.BaseCosts or []:
            total += cost_value.AppliedValue.wrappedValue if cost_value.AppliedValue else 0
        return total

    def get_quantity(self, resource):
        total = 0
        if resource.BaseQuantity:
            return resource.BaseQuantity[3]
        if resource.Usage and resource.Usage.ScheduleWork:
            # For now we assume either hourly or daily depending on how duration is stored
            duration = ifcopenshell.util.date.ifc2datetime(resource.Usage.ScheduleWork)
            if duration.days:
                return duration.days
            return duration.seconds / 60 / 60