MAINT: Return more info in W1 computations

pmgbergen · Jan 12, 2025 · c257c92 · c257c92
1 parent 98b8797
commit c257c92
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/darsia/measure/wasserstein.py b/src/darsia/measure/wasserstein.py
@@ -2072,8 +2072,14 @@ def _solve(self, flat_mass_diff: np.ndarray) -> tuple[float, np.ndarray, dict]:
         total_timings = self._analyze_timings(convergence_history["timing"])
         peak_memory_consumption = tracemalloc.get_traced_memory()[1] / 10**9
 
+        # Compute l1 norm of the flux
+        unweighted_transport_density = self.transport_density(flux, weighted=False)
+        flux_l1_norm = self.mass_matrix_cells.dot(unweighted_transport_density).sum()
+
         # Define performance metric
         info = {
+            "distance": new_distance,  # includes weight
+            "flux_l1_norm": flux_l1_norm,  # without weight
             "converged": iter < num_iter - 1,
             "number_iterations": iter,
             "convergence_history": convergence_history,