Changeset 5196

Timestamp:

Apr 9, 2008, 10:08:44 AM (15 years ago)

Author:

duncan

Message:

Additional functionality. Output stage .csv as well.

File:

• anuga_core/source/anuga/fit_interpolate/interpolate.py (modified) (5 diffs)

anuga_core/source/anuga/fit_interpolate/interpolate.py

@@ -324 +324 @@
                         velocity_x_file,
                         velocity_y_file,
+                        stage_file=None,
                         #quantities = ['depth', 'velocity'],
                         verbose=True,
@@ -353 +354 @@
     velocity_x_writer = writer(file(velocity_x_file, "wb"))
     velocity_y_writer = writer(file(velocity_y_file, "wb"))
+    if stage_file is not None:
+        stage_writer = writer(file(stage_file, "wb"))
     # Write heading
     heading = [str(x[0])+ ':' + str(x[1]) for x in points]
@@ -359 +362 @@
     velocity_x_writer.writerow(heading)
     velocity_y_writer.writerow(heading)
+    if stage_file is not None:
+        stage_writer.writerow(heading)    
     
     for time in callable_sww.get_time():
@@ -364 +369 @@
         velocity_xs = [time]
         velocity_ys = [time]
+        if stage_file is not None:  
+            stages = [time]  
         for point_i, point in enumerate(points):
             quantities = callable_sww(time,point_i)
@@ -391 +398 @@
             velocity_xs.append(velocity_x)
             velocity_ys.append(velocity_y)
+            if stage_file is not None:
+                stages.append(w)
         depth_writer.writerow(depths)
         velocity_x_writer.writerow(velocity_xs)
         velocity_y_writer.writerow(velocity_ys)
+        if stage_file is not None:
+            stage_writer.writerow(stages)