Adding full_trace_with_utilities to SAOMechanism

Author: yasserfarouk

src/negmas/inout.py

@@ -750,14 +750,16 @@ def adjust_type(d: dict, base: str = "negmas", domain=None) -> dict:
             base_module="negmas",
             python_class_identifier=python_class_identifier,
         )
-        utils = tuple(
+        utils = [
             deserialize(
                 adjust_type(load(fname), domain=os),
                 python_class_identifier=python_class_identifier,
                 base_module="negmas",
             )
             for fname in ufuns
-        )
+        ]
+        for u in utils:
+            u.outcome_space = os
 
         # d = load(domain)
         # type_ = d.pop("type", "")

src/negmas/sao/mechanism.py

@@ -624,6 +624,16 @@ def __call__(self, state, action=None) -> MechanismStepResult:
         #     ), f"Not all negotiator actions were used in this step: {action}"
         return MechanismStepResult(state, times=times, exceptions=exceptions)
 
+    @property
+    def full_trace_with_utils(self) -> list[tuple]:
+        """Returns the full trace and the utility of the negotiators at each step."""
+        trace = self.full_trace
+        ufuns = [_.ufun for _ in self.negotiators]
+        return [
+            tuple(list(_) + [u(_[-3]) if u else float("nan") for u in ufuns])
+            for _ in trace
+        ]
+
     @property
     def full_trace(self) -> list[TraceElement]:
         """Returns the negotiation history as a list of relative_time/step/negotiator/offer tuples"""
@@ -698,6 +708,65 @@ def not_equal(a, b):
                     response(self._history[-1]),
                 )
             )
+        if (
+            self.state.done
+            and self.agreement is not None
+            and offers
+            and not not_equal(offers[-1].offer, self.agreement)
+        ):
+            acceptances = get_acceptances(self._history[-1])
+            offers[-1] = TraceElement(
+                self._history[-1].time,
+                self._history[-1].relative_time,
+                self._history[-1].step,
+                self._history[-1].current_proposer,
+                self.agreement,
+                {n: ResponseType.ACCEPT_OFFER for n in acceptances},
+                response(self._history[-1]),
+            )
+        if (
+            self.state.done
+            and self.agreement is None
+            and offers
+            and self.state.has_error
+        ):
+            acceptances = get_acceptances(self._history[-1])
+            offers[-1] = TraceElement(
+                self._history[-1].time,
+                self._history[-1].relative_time,
+                self._history[-1].step,
+                self._history[-1].current_proposer,
+                self.agreement,
+                {n: ResponseType.ACCEPT_OFFER for n in acceptances},
+                "error",
+            )
+        elif (
+            self.state.done
+            and self.agreement is None
+            and offers
+            and self.state.timedout
+        ):
+            acceptances = get_acceptances(self._history[-1])
+            offers[-1] = TraceElement(
+                self._history[-1].time,
+                self._history[-1].relative_time,
+                self._history[-1].step,
+                self._history[-1].current_proposer,
+                self.agreement,
+                {n: ResponseType.ACCEPT_OFFER for n in acceptances},
+                "timedout",
+            )
+        elif self.state.done and self.agreement is None and offers:
+            acceptances = get_acceptances(self._history[-1])
+            offers[-1] = TraceElement(
+                self._history[-1].time,
+                self._history[-1].relative_time,
+                self._history[-1].step,
+                self._history[-1].current_proposer,
+                self.agreement,
+                {n: ResponseType.ACCEPT_OFFER for n in acceptances},
+                "broken",
+            )
 
         return offers