Changeset 3177 for inundation/pmesh

Timestamp:

Jun 19, 2006, 10:30:57 AM (19 years ago)

Author:

duncan

Message:

added function to pmesh.mesh for adding vertices by passing in a geospatial instance.

Location:

inundation/pmesh

Files:

• mesh.py (modified) (4 diffs)
• test_mesh.py (modified) (2 diffs)

inundation/pmesh/mesh.py

@@ -13 +13 @@
    Geoscience Australia
 """
-import load_mesh.loadASCII
-import alpha_shape.alpha_shape
 
 import sys
@@ -29 +27 @@
 from coordinate_transforms.geo_reference import Geo_reference,DEFAULT_ZONE
 from utilities.polygon import point_in_polygon 
+import load_mesh.loadASCII
+import alpha_shape.alpha_shape
+from geospatial_data.geospatial_data import Geospatial_data, ensure_geospatial
 
 SET_COLOUR='red'
@@ -609 +610 @@
 
     
-    
+    def add_vertices(self, data):
+        data = ensure_geospatial(data)
+
+        # get points relative to the mesh geo_ref
+        points = data.get_data_points(geo_reference=self.geo_reference)
+        #FIXME finish....
+    
+        for point in points:
+            v=Vertex(point[0], point[1])
+            self.userVertices.append(v)
+            
     def add_hole_from_polygon(self, polygon, tags=None, geo_reference=None):
         """
@@ -853 +864 @@
         
     def getUserVertices(self):
+        """
+        Note: The x,y values will be relative to the mesh geo_reference
+        This returns a list of vertex objects
+        """
         return self.userVertices
+
+    def get_user_vertices(self, absolute=True):
+        """
+        This returns a list of (x,y) values
+        (maybe it should return a geospatical object?
+        It makes it a bit confusing though.)
+        """
+        pointlist=[]
+        for vertex in self.userVertices:
+            pointlist.append([vertex.x,vertex.y])
+        spat = Geospatial_data(pointlist, geo_reference=self.geo_reference)
+        return spat.get_data_points(absolute=absolute)
     
     def getUserSegments(self):

inundation/pmesh/test_mesh.py

@@ -11 +11 @@
 from load_mesh.loadASCII import *
 from coordinate_transforms.geo_reference import Geo_reference
+from geospatial_data.geospatial_data import Geospatial_data
 from utilities.polygon import  is_inside_polygon ### inside_polygon
 
@@ -2052 +2053 @@
         self.failUnless(4==len(vert),
                         'FAILED!')
+   
+    def test_add_vertices(self):
+        points_ab = [[0.1,1],[0.4,.2],[7,5],[10,5]]
+        geo =  Geo_reference(56,23,21)
+        points = geo.change_points_geo_ref(points_ab)
+        spat = Geospatial_data(points, geo_reference=geo)
+        
+        geo_mesh =  Geo_reference(56,100,200)
+        m = Mesh(geo_reference=geo_mesh)
+        m.add_vertices(spat)
+
+        vert = m.getUserVertices()
+        #print "vert",vert 
+        self.failUnless(4==len(vert),
+                        'FAILED!')
+        vert= m.get_user_vertices(absolute=True)
+        self.failUnless(vert==points_ab,
+                        'FAILED!')
+        
+     
+    def test_get_user_vertices(self):
+        points_ab = [[0.1,1],[0.4,.2],[7,5],[10,5]]
+        geo =  Geo_reference(56,23,21)
+        points = geo.change_points_geo_ref(points_ab)
+        spat = Geospatial_data(points, geo_reference=geo)
+        
+        geo_mesh =  Geo_reference(56,100,200)
+        m = Mesh(geo_reference=geo_mesh)
+        m.add_vertices(spat)
+
+        vert = m.getUserVertices()
+        #print "vert",vert 
+        self.failUnless(4==len(vert),
+                        'FAILED!')
+        vert= m.get_user_vertices(absolute=True)
+        self.failUnless(vert==points_ab,
+                        'FAILED!')
+        vert= m.get_user_vertices(absolute=False)
+        points_new = m.geo_reference.get_absolute(vert)
+        
+        self.failUnless(points_ab==points_new,
+                        'FAILED!')
+        
         
 def list_comp(A,B):