Changeset 2402 for inundation/pmesh/test_mesh_interface.py

Timestamp:

Feb 14, 2006, 3:29:10 PM (19 years ago)

Author:

duncan

Message:

added auto geo-reffing in create_mesh_from_regions

File:

• inundation/pmesh/test_mesh_interface.py (modified) (2 diffs)

inundation/pmesh/test_mesh_interface.py

@@ -5 +5 @@
 from mesh_interface import *
 from load_mesh.loadASCII import *
-from coordinate_transforms.geo_reference import Geo_reference
 from utilities.polygon import inside_polygon
+from coordinate_transforms.geo_reference import Geo_reference,DEFAULT_ZONE
 
 class TestCase(unittest.TestCase):
@@ -107 +107 @@
                                        closed = False),
                         'FAILED!')
+
+
+        
+    def test_create_mesh_from_regions2(self):
+
+        # These are the absolute values
+        min_x = 10 
+        min_y = 88
+        polygon_absolute = [[min_x,min_y],[1000,100],[1000,1000],[100,1000]]
+        
+        x_p = -10
+        y_p = -40
+        zone = 808
+        geo_ref_poly = Geo_reference(zone, x_p, y_p)
+        polygon = geo_ref_poly.change_points_geo_ref(polygon_absolute)
+
+        boundary_tags = {'walls':[0,1],'bom':[2]}
+        
+        inner1_polygon_absolute = [[10,10],[20,10],[20,20],[10,20]]
+        inner1_polygon = geo_ref_poly. \
+                         change_points_geo_ref(inner1_polygon_absolute)
+
+        inner2_polygon_absolute = [[30,30],[40,30],[40,40],[30,40]]
+        inner2_polygon = geo_ref_poly. \
+                         change_points_geo_ref(inner2_polygon_absolute)
+        
+        interior_regions = [(inner1_polygon, 5),(inner2_polygon, 0.2)]
+        m = create_mesh_from_regions(polygon,
+                                     boundary_tags,
+                                     10000000,
+                                     interior_regions=interior_regions,
+                                     poly_geo_reference=geo_ref_poly)
+        
+
+        # Test the mesh instance
+        self.failUnless(len(m.regions)==3,
+                        'FAILED!')
+        segs = m.getUserSegments()
+        self.failUnless(len(segs)==12,
+                        'FAILED!')
+        self.failUnless(len(m.userVertices)==12,
+                        'FAILED!') 
+        self.failUnless(segs[0].tag=='walls',
+                        'FAILED!')  
+        self.failUnless(segs[1].tag=='walls',
+                        'FAILED!') 
+         
+        self.failUnless(segs[2].tag=='bom',
+                        'FAILED!') 
+        self.failUnless(segs[3].tag=='',
+                        'FAILED!')
+        
+        self.failUnless(m.geo_reference.get_zone()==zone,
+                        'FAILED!')
+        self.failUnless(m.geo_reference.get_xllcorner()==min_x,
+                        'FAILED!')
+        self.failUnless(m.geo_reference.get_yllcorner()==min_y,
+                        'FAILED!')
+
+        
+    def test_create_mesh_from_regions3(self):
+
+        # These are the absolute values
+        min_x = 10 
+        min_y = 88
+        polygon = [[min_x,min_y],[1000,100],[1000,1000],[100,1000]]
+        
+
+        x_p = -10
+        y_p = -40
+        geo_ref_poly = Geo_reference(56, x_p, y_p)
+        
+        boundary_tags = {'walls':[0,1],'bom':[2]}
+        
+        inner1_polygon_absolute = [[10,10],[20,10],[20,20],[10,20]]
+        inner1_polygon = geo_ref_poly. \
+                         change_points_geo_ref(inner1_polygon_absolute)
+
+        inner2_polygon_absolute = [[30,30],[40,30],[40,40],[30,40]]
+        inner2_polygon = geo_ref_poly. \
+                         change_points_geo_ref(inner2_polygon_absolute)
+        
+        interior_regions = [(inner1_polygon, 5),(inner2_polygon, 0.2)]
+        m = create_mesh_from_regions(polygon,
+                                     boundary_tags,
+                                     10000000,
+                                     interior_regions=interior_regions)
+        
+
+        # Test the mesh instance
+        self.failUnless(len(m.regions)==3,
+                        'FAILED!')
+        segs = m.getUserSegments()
+        self.failUnless(len(segs)==12,
+                        'FAILED!')
+        self.failUnless(len(m.userVertices)==12,
+                        'FAILED!') 
+        self.failUnless(segs[0].tag=='walls',
+                        'FAILED!')  
+        self.failUnless(segs[1].tag=='walls',
+                        'FAILED!') 
+         
+        self.failUnless(segs[2].tag=='bom',
+                        'FAILED!') 
+        self.failUnless(segs[3].tag=='',
+                        'FAILED!')
+        
+        self.failUnless(m.geo_reference.get_zone()==DEFAULT_ZONE,
+                        'FAILED!')
+        self.failUnless(m.geo_reference.get_xllcorner()==min_x,
+                        'FAILED!')
+        self.failUnless(m.geo_reference.get_yllcorner()==min_y,
+                        'FAILED!')
+
 #-------------------------------------------------------------
 if __name__ == "__main__":