Changeset 2200 for inundation/pmesh/mesh.py

Timestamp:

Jan 11, 2006, 5:31:49 PM (18 years ago)

Author:

duncan

Message:

in pmesh.mesh, can now add a region specified as a polygon. (This isn't fully implemented yet)

File:

• inundation/pmesh/mesh.py (modified) (8 diffs)

inundation/pmesh/mesh.py

@@ -28 +28 @@
 import load_mesh
 from coordinate_transforms.geo_reference import Geo_reference,DEFAULT_ZONE
+from utilities.polygon import point_in_polygon 
 
 SET_COLOUR='red'
@@ -185 +186 @@
     Defined by a point in the hole enclosed by segments.
     """
+
     HOLECORNERLENGTH = 6
+    
     def draw(self, canvas, tags, colour = 'purple',scale = 1, xoffset = 0, yoffset =0, ):
         x =  scale*(self.x + xoffset)
@@ -396 +399 @@
         Null represents no neighbor
         """
-        self.attribute = attribute
+        self.attribute = attribute #this is a string
         
     def __repr__(self):
@@ -501 +504 @@
         self.meshVertices=[]
 
-        #Peters stuff
-        # FIXME (DSG) Sets of what?  
         self.setID={}
         #a dictionary of names.
@@ -667 +668 @@
         self.regions.append(h)
         return h
-    
+
+    #FIXME(DSG-DSG) remove EN, have a relative flag. 
     def addRegionEN(self, x,y):
         h=Region(x-self.geo_reference.xllcorner,
@@ -673 +675 @@
         self.regions.append(h)
         return h
-    
+    def addRegionFromPolygon(self, polygon, tags=None,
+                             maxArea=None, Geo_ref=None):
+        #take into account georef
+        #create points, segs and tags
+        region_dict = {}
+        region_dict['points'] = polygon
+
+        #Create segments
+        #E.g. [[0,1], [1,2], [2,3], [3,0]]
+        segments = []
+        N = len(polygon)
+        for i in range(N):
+            lo = i
+            hi = (lo + 1) % N
+            segments.append( [lo, hi] ) 
+        region_dict['segments'] = segments
+
+
+        #Create tags
+        #E.g. ['wall', 'wall', 'ocean', 'wall']
+
+        segment_tags = ['']*N
+        if tags is not None:
+            for key in tags:
+                indices = tags[key]
+                for i in indices:
+                    segment_tags[i] = key
+        region_dict['segment_tags'] = segment_tags
+    
+        self.addVertsSegs(region_dict) #this is assuming absolute geos
+    
+        #get inner point
+        inner_point = point_in_polygon(polygon)
+        inner = self.addRegionEN(inner_point[0], inner_point[1])
+
+        if maxArea is not None:
+            inner.setMaxArea(maxArea)
+        
     def getUserVertices(self):
         return self.userVertices
@@ -1312 +1351 @@
         segs2delete = self.alphaUserSegments
         #FIXME(DSG-DSG) this algorithm needs comments
-        #FIXME(DSG-DSG) can it be sped up?  It's slow
         new_segs = {}
         #alpha_segs = []
@@ -1603 +1641 @@
         yoffset = HEIGHT - b
         return [SCALE, xoffset, yoffset]
-            
-    def plotMeshTriangle(self,tag = 0,WIDTH = 400,HEIGHT = 400):
-        """
-        Plots all node connections.
-        tag = 0 (no node numbers), tag = 1 (node numbers)
-        """
-
-        try:
-            from Tkinter import Tk, Frame, Button, Canvas, BOTTOM, Label
-
-            [SCALE, xoffset, yoffset] = self.scaleoffset( WIDTH, HEIGHT)
-            
-            root = Tk()
-            frame = Frame(root)
-            frame.pack()
-            button = Button(frame, text="OK", fg="red", command=frame.quit)
-            button.pack(side=BOTTOM)
-            canvas = Canvas(frame,bg="white", width=WIDTH, height=HEIGHT)
-            canvas.pack()
-            text = Label(frame, width=20, height=10, text='triangle mesh')
-            text.pack()
-
-            #print self.meshTriangles
-            for triangle in self.meshTriangles:
-                triangle.draw(canvas,1,
-                              scale = SCALE,
-                              xoffset = xoffset,
-                              yoffset = yoffset )
-                
-            root.mainloop()
-
-        except:
-            print "Unexpected error:", sys.exc_info()[0]
-            raise
-
-            #print """
-            #node::plot Failed.
-            #Most probably, the Tkinter module is not available.
-            #"""
-
-    def plotUserSegments(self,tag = 0,WIDTH = 400,HEIGHT = 400):
-        """
-        Plots all node connections.
-        tag = 0 (no node numbers), tag = 1 (node numbers)
-        """
-
-        try:
-            from Tkinter import Tk, Frame, Button, Canvas, BOTTOM, Label
-            
-            [SCALE, xoffset, yoffset] = self.scaleoffset( WIDTH, HEIGHT)
-
-            root = Tk()
-            frame = Frame(root)
-            frame.pack()
-            button = Button(frame, text="OK", fg="red", command=frame.quit)
-            button.pack(side=BOTTOM)
-            canvas = Canvas(frame, bg="white", width=WIDTH, height=HEIGHT)
-            canvas.pack()
-            text = Label(frame, width=20, height=10, text='user segments')
-            text.pack()
-           
-            for segment in self.getUserSegments():
-                segment.draw(canvas,SCALE, xoffset, yoffset )
-                
-            root.mainloop()
-
-        except:
-            print "Unexpected error:", sys.exc_info()[0]
-            raise
-
-            #print """
-            #node::plot Failed.
-            #Most probably, the Tkinter module is not available.
-            #"""
-            
-
+
+         
     def exportASCIIobj(self,ofile):
         """