Changeset 4189

Timestamp:

Jan 23, 2007, 5:31:18 PM (17 years ago)

Author:

duncan

Message:

can select the maximum time

File:

• anuga_work/development/friction_UA_flume_2006/calc_norm.py (modified) (3 diffs)

anuga_work/development/friction_UA_flume_2006/calc_norm.py

@@ -62 +62 @@
               is_trial_run=False,
               outputdir_name=None,
-              pro_instance=None):    
+              pro_instance=None,
+              max_time=None):    
     if is_trial_run is True:
         outputdir_name += '_test'
@@ -76 +77 @@
     flume_results, _ = load_csv(flume_results)
     #print "flume_results",flume_results #.keys()
-
+    if max_time is not None:
+        for i,time in enumerate(flume_results['time']):
+            if float(time) > float(max_time):
+                break
+        for keys in flume_results.iterkeys():
+            flume_results[keys] = flume_results[keys][0:i]
+    
     #print "flume_results['time']", flume_results['time']
+    import sys; sys.exit() 
     #Get a list of the files
     files = os.listdir(pro_instance.outputdir)
@@ -134 +142 @@
 if __name__ == "__main__":
     calc_norm(
-        #'ensemble_average_h_smooth.csv',
-        'ensemble_average_h_rough.csv',
-              is_trial_run = False,
-              outputdir_name='friction_UA')
-    
+        'ensemble_average_h_smooth.csv',
+        #'ensemble_average_h_rough.csv',
+        is_trial_run = True,   # adds _test to outputdir_name
+        outputdir_name='friction_UA',
+        max_time=None)
+