Changeset 806

Timestamp:

Jan 28, 2005, 5:53:59 PM (20 years ago)

Author:

prow

Message:

 

File:

• inundation/ga/storm_surge/pmesh/pmesh.py (modified) (6 diffs)

inundation/ga/storm_surge/pmesh/pmesh.py

@@ -17 +17 @@
 TRIANGLE_COLOUR = 'green'
 APPLICATION_NAME = 'Pmesh'
+SET_COLOUR = 'red'
 
 NO_SELECTION = 0
@@ -199 +200 @@
                                statushelp='' )
 
+
+
+    def createSetTools(self):
+        """
+        Add set tool buttons to the top of the GUI
+        """
+        ToolBarButton(self, self.toolbar, 'sep', 'sep.gif', width=10,
+                      state='disabled')
+        for key, func, balloon in [
+                ('threshold', self.threshold, 'threshold the set'),
+                ('Courant_threshold', self.Courant_threshold, 'Courant_threshold the set'),
+                ('polyset', self.triangles_to_polySet, 'make a poly set out of selected triangles'),      
+                ('refineSet', self.refineSet, 'Refine the set')]:
+            ToolBarButton(self, self.toolbar, key, '%s.gif' %key,
+                          command=func, balloonhelp=balloon,
+                               statushelp='' )
+
+    def createSetIcons(self):
+        """
+        Add Edit buttons to the top of the GUI
+        """
+        ToolBarButton(self, self.toolbar, 'sep', 'sep.gif', width=10,
+                      state='disabled')
+        for key, func, balloon in [
+                ('selectAllTriangles', self.selectAllTriangles, 'select all'),
+                ('none', self.clearSelection, 'clear selection')]:
+            ToolBarButton(self, self.toolbar, key, '%s.gif' %key,
+                          command=func, balloonhelp=balloon,
+                               statushelp='' )
+
+
+
+    def refineSet(self,parent):
+        self.mesh.refineSet(self.selSet)
+        self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+    def setStructureNumber(self,parent):
+        dialog =  setStructureNumberDialog(self.canvas)
+        if dialog.numberOK:
+            self.structureSize = dialog.number
+
+    def erode(self, parent):
+#Not implimented
+        self.mesh.undrawSet(self.canvas,self.selSet,self.SCALE)
+        self.mesh.erode(self.selSet,structureSize=self.structureSize)
+        self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+    def dilate(self, parent):
+#Not implimented
+        self.mesh.undrawSet(self.canvas,self.selSet,self.SCALE)
+        self.mesh.dilate(self.selSet,structureSize=self.structureSize)
+        self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+
+    def threshold(self, parent):
+        """
+        add a vertex using a window and entering x y values.
+
+        the parent attribute isn't used by this function.
+        need to userstand toolbarbutton.py to know how to
+        get rid of it.
+        """
+        if self.selSet == 'None':
+            self.selectAllTriangles(parent)
+
+        dialog = ThresholdDialog(self.canvas)
+        if dialog.minmaxValuesOk:
+            self.mesh.undrawSet(self.canvas,self.selSet,self.SCALE)
+            self.mesh.threshold(self.selSet,min=dialog.min,max=dialog.max)
+            self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+    def Courant_threshold(self, parent):
+        """
+        add a vertex using a window and entering x y values.
+
+        the parent attribute isn't used by this function.
+        need to userstand toolbarbutton.py to know how to
+        get rid of it.
+        """
+        if self.selSet == 'None':
+            self.selectAllTriangles(parent)
+
+        dialog = Courant_ThresholdDialog(self.canvas)
+        if dialog.minmaxValuesOk:
+            self.mesh.undrawSet(self.canvas,self.selSet,self.SCALE)
+            self.mesh.Courant_threshold(self.selSet,min=dialog.min,max=dialog.max)
+            self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+    def triangles_to_polySet(self,parent):
+        vertices, segments = self.mesh.triangles_to_polySet(self.selSet)
+        event = None
+        for v in vertices.keys():
+            x = v.x*self.SCALE
+            y = v.y*self.SCALE
+            vertices[v]=self.drawVertex(x,y,event)
+        for s in segments:
+            v0 = vertices[s[0]]
+            v1 = vertices[s[1]]
+            self.drawSegment(v0,v1)
+
+    def selectTriangles(self,setName):
+        """
+        """
+        self.mesh.undrawSet(self.canvas,self.selSet,self.SCALE)
+        self.selSet = setName
+        self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+    def selectAllTriangles(self,parent):
+        """
+        selected all triangles in the mesh
+        """
+        self.selSet = self.mesh.selectAllTriangles()
+        self.mesh.drawSet(self.canvas,self.selSet,self.SCALE,colour=SET_COLOUR)
+
+    def clearSelection(self,parent):
+        """
+        """
+        self.mesh.undrawSet(self.canvas,self.selSet,self.SCALE)
+        self.selSet = self.mesh.clearSelection()
+
+
     def createMesh(self):
         """
@@ -212 +334 @@
         
         self.Triangles = visualmesh.vTriangles(mesh.Triangle)
+        self.selSet='None'
 
 
@@ -1081 +1204 @@
     def close(self):
         self.quit()
-
     def createInterface(self):
         """
@@ -1095 +1217 @@
         self.createZooms()
         self.createEdits()
+        self.createSetTools()
+        self.createSetIcons()
         #self.addVertsAndSegs() # !!!DSG start pmesh with a triangle
         self.selectFunc('pointer')
@@ -1321 +1445 @@
             showerror('Bad mesh generation values',
                                    'Values are out of range.')
+
+
+class  ThresholdDialog(Dialog):
+    """
+    Dialog box for thresholding a set by entering minimum 
+    and maximum values
+    """
+    def body(self, master):
+        """
+        GUI description
+        """
+        self.title("Threshold selected set")
+        
+        Label(master, text='minimum attribute:').grid(row=0, sticky=W)
+        Label(master, text='maximum attribute:').grid(row=1, sticky=W)
+
+        self.minstr   = Entry(master, width = 16, name ="entry")
+        self.maxstr   = Entry(master, width = 16)
+       
+        self.minstr.grid(row=0, column=1, sticky=W)
+        self.maxstr.grid(row=1, column=1, sticky=W)
+        self.minstr.focus_force()
+        self.min  = 0
+        self.max  = 0
+        self.minmaxValuesOk = False
+        
+    def apply(self):
+        self.minmaxValuesOk = True
+        try:
+            self.min = float(self.minstr.get())
+            self.max = float(self.maxstr.get())
+        except ValueError:
+            self.ValuesOk = False
+            showerror('Bad mesh generation values',
+                                   ' Values are not numbers.')
+
+
+class  Courant_ThresholdDialog(Dialog):
+    """
+    Dialog box for Courant_thresholding a set by entering minimum 
+    and maximum values
+    """
+    def body(self, master):
+        """
+        GUI description
+        """
+        self.title("Courant_Threshold selected set")
+        
+        Label(master, text='minimum attribute:').grid(row=0, sticky=W)
+        Label(master, text='maximum attribute:').grid(row=1, sticky=W)
+
+        self.minstr   = Entry(master, width = 16, name ="entry")
+        self.maxstr   = Entry(master, width = 16)
+       
+        self.minstr.grid(row=0, column=1, sticky=W)
+        self.maxstr.grid(row=1, column=1, sticky=W)
+        self.minstr.focus_force()
+        self.min  = 0
+        self.max  = 0
+        self.minmaxValuesOk = False
+        
+    def apply(self):
+        self.minmaxValuesOk = True
+        try:
+            self.min = float(self.minstr.get())
+            self.max = float(self.maxstr.get())
+        except ValueError:
+            self.ValuesOk = False
+            showerror('Bad mesh generation values',
+                                   ' Values are not numbers.')
+
+
+class  setStructureNumberDialog(Dialog):
+    """
+    Dialog box for setting the number of triangles
+    used to make up dilation or erosion
+    """
+    def body(self, master):
+        """
+        GUI description
+        """
+        self.title("Set number of elements effected by morphing sets")
+        
+        Label(master, text='number:').grid(row=0, sticky=W)
+
+        self.number = Entry(master, width = 16, name ="entry")
+       
+        self.number.grid(row=0, column=1, sticky=W)
+        self.number.focus_force()
+        self.number = 0
+        self.numberOk = False
+        
+    def apply(self):
+        self.numberOk = True
+        try:
+            self.number = int(self.number.get())
+        except ValueError:
+            self.numberOK = False
+            showerror('Bad mesh generation values',
+                                   ' Values are not numbers.')
+
+
       
 if __name__ == '__main__':