Changeset 1071

Timestamp:

Mar 14, 2005, 10:11:37 AM (20 years ago)

Author:

duncan

Message:

new functions

File:

• inundation/ga/storm_surge/pmesh/load_mesh/loadASCII.py (modified) (2 diffs)

inundation/ga/storm_surge/pmesh/load_mesh/loadASCII.py

@@ -51 +51 @@
 
 from string import  find, rfind
-from Numeric import array, Float, Int16, Int32, Character,reshape, concatenate
+from Numeric import array, Float, Int16, Int32, Character,reshape, concatenate, take
 from os.path import splitext
 
@@ -1182 +1182 @@
     return xya_dict
 
+#FIXME(dsg), turn this dict plus methods into a class?
+def take_points(dict,indices_to_keep):
+    dict = point_atts2array(dict)
+    #FIXME maybe the points data structure should become a class?
+    dict['pointlist'] = take(dict['pointlist'],indices_to_keep)
+
+    for key in dict['attributelist'].keys():
+        dict['attributelist'][key]= take(dict['attributelist'][key],
+                                         indices_to_keep)
+    return dict
+    
+def add_point_dictionaries (dict1, dict2):
+    """
+    """
+    dict1 = point_atts2array(dict1)
+    dict2 = point_atts2array(dict2)
+    
+    combined = {}  
+    combined['pointlist'] = concatenate((dict2['pointlist'],
+                                         dict1['pointlist']),axis=0)   
+    atts = {}   
+    for key in dict2['attributelist'].keys():
+        atts[key]= concatenate((dict2['attributelist'][key],
+                                dict1['attributelist'][key]), axis=0)
+    combined['attributelist']=atts
+    combined['geo_reference'] = dict1['geo_reference']
+    return combined
+    
 #FIXME (DSG) need an export_points_file method..                
 def export_xya_file( file_name, xya_dict, title, delimiter = ','):