Changeset 7582

Timestamp:

Dec 7, 2009, 9:46:15 PM (15 years ago)

Author:

ole

Message:

Work in progress

File:

• anuga_work/development/classroom/ripcurrent.py (modified) (2 diffs)

anuga_work/development/classroom/ripcurrent.py

@@ -11 +11 @@
 from anuga.shallow_water.shallow_water_domain import Reflective_boundary
 from anuga.shallow_water.shallow_water_domain import Time_boundary
+from anuga.shallow_water.data_manager import get_mesh_and_quantities_from_file
+from anuga.fit_interpolate.interpolate import Interpolation_function
 from pylab import figure, quiver, show, cos, sin, pi
 import numpy
@@ -175 +177 @@
 #------------------------------------------------------------------------------
 # Post processing
-#------------------------------------------------------------------------------    
-
-# Use this
-
-#     from anuga.fit_interpolate.interpolate import Interpolation_function
-
-#     # Get mesh and quantities from sww file
-#     X = get_mesh_and_quantities_from_file(filename,
-#                                           quantities=quantity_names,
-#                                           verbose=verbose)
-#     mesh, quantities, time = X
-
-#     # Find all intersections and associated triangles.
-#     segments = mesh.get_intersecting_segments(polyline, verbose=verbose)
-
-#     # Get midpoints
-#     interpolation_points = segment_midpoints(segments)
-
-#     # Interpolate
-#     if verbose:
-#         log.critical('Interpolating - total number of interpolation points = %d'
-#                      % len(interpolation_points))
-
-#     I = Interpolation_function(time,
-#                                quantities,
-#                                quantity_names=quantity_names,
-#                                vertex_coordinates=mesh.nodes,
-#                                triangles=mesh.triangles,
-#                                interpolation_points=interpolation_points,
-#                                verbose=verbose)
-
-#     return segments, I
-
-
-
-
+#------------------------------------------------------------------------------
+
+# Get mesh and quantities from sww file
+quantity_names = ['elevation'] + domain.conserved_quantities
+
+X = get_mesh_and_quantities_from_file(domain.get_name() + '.sww',
+                                      quantities=quantity_names,
+                                      verbose=True)
+
+mesh, quantities, timesteps = X
+
+
+# Interpolate quantities to gauges
+I = Interpolation_function(timesteps,
+                           quantities,
+                           quantity_names=quantity_names,
+                           vertex_coordinates=mesh.nodes,
+                           triangles=mesh.triangles,
+                           interpolation_points=gauges,
+                           verbose=True)
+
+
+
+# Compute velocity field
+#uu = numpy.array(number_of_gauges)
+#vv = numpy.array(number_of_gauges)
+#for t in timesteps:
+#    for i in range(number_of_gauges):
+#        elevation, stage, uh, vh = I(t, point_id=i)#
+#
+#        depth = stage-elevation
+#        uu[i] += uh/depth
+#        vv[i] += vh/depth