Changeset 4200 for anuga_core/source/anuga/abstract_2d_finite_volumes/domain.py

Timestamp:

Jan 31, 2007, 12:07:32 PM (18 years ago)

Author:

ole

Message:

Added diagnosticts about which triangle is responsible for the largest computed speed.

File:

• anuga_core/source/anuga/abstract_2d_finite_volumes/domain.py (modified) (4 diffs)

anuga_core/source/anuga/abstract_2d_finite_volumes/domain.py

@@ -4 +4 @@
 
    Copyright 2004
-   Ole Nielsen, Stephen Roberts, Duncan Gray, Christopher Zoppou
+   Ole Nielsen, Stephen Roberts, Duncan Gray
    Geoscience Australia
 """
 
-from Numeric import allclose
+from Numeric import allclose, argmax
 from anuga.config import epsilon
 
@@ -216 +216 @@
         self.already_computed_flux = zeros((N, 3), Int)
 
+        # Storage for maximal speeds computed for each triangle by compute_fluxes
+        # This is used for diagnostics only
+        self.max_speed = zeros(N, Float)
+
         if mesh_filename is not None:
             # If the mesh file passed any quantity values
@@ -546 +550 @@
         ##assert hasattr(self, 'boundary_objects')
 
-    def write_time(self):
-        print self.timestepping_statistics()
-
-
-    def timestepping_statistics(self):
+    def write_time(self, track_location=False):
+        print self.timestepping_statistics(track_location)
+
+
+    def timestepping_statistics(self, track_location=False):
         """Return string with time stepping statistics for printing or logging
+
+        Optional boolean keyword track_location decides whether to report location of smallest timestep. 
         """
 
@@ -568 +574 @@
                      self.max_timestep, self.number_of_steps,
                      self.number_of_first_order_steps)
+
+        if track_location is True:
+            msg += '\n'
+            # Find index of largest computed flux speed
+            i = argmax(self.max_speed)
+
+            x, y = self.get_centroid_coordinates()[i]
+
+            s = 'Triangle #%d with centroid (%.4f, %.4f) ' %(i, x, y)
+            s += 'had the largest computed speed: %.4f m/s' %(self.max_speed[i])
+
+            # FIXME (Ole): Maybe add all quantity values at vertices
+
+            msg += s
 
         return msg