Changeset 1295 for inundation/ga/storm_surge/pyvolution/realtime_visualisation.py

Timestamp:

May 8, 2005, 11:50:32 PM (20 years ago)

Author:

steve

Message:

Playing with coloured surfaces

File:

• inundation/ga/storm_surge/pyvolution/realtime_visualisation.py (modified) (22 diffs)

inundation/ga/storm_surge/pyvolution/realtime_visualisation.py

@@ -11 +11 @@
         #n = vector(0,0,0)
 
-    return n    
+    return n
 
 
@@ -20 +20 @@
 
         #Outward normal
-        
+
         self.normal=normal(v1-v0, v2-v0)
 
         self.panel = faces(frame = frame)
-        self.backpanel = faces(frame = frame)        
+        self.backpanel = faces(frame = frame)
         for v in (v0,v1,v2):
             self.panel.append(pos=v, normal = self.normal, color=color)
 
         for v in (v2,v1,v0):
-            self.backpanel.append(pos=v, normal = -self.normal, color=color)            
-
-            
+            self.backpanel.append(pos=v, normal = -self.normal, color=color)
+
+
     def move(self,v):
         """Move panel in direction given by vector v
@@ -37 +37 @@
 
         for vertex in self.panel.pos:
-            vertex += v            
+            vertex += v
 
     def update_height(self,d,fix_baseline=False):
@@ -47 +47 @@
         v0 = self.panel.pos[0]
         v1 = self.panel.pos[1]
-        v2 = self.panel.pos[2]        
-        
+        v2 = self.panel.pos[2]
+
         n = normal(v1-v0, self.normal)
 
@@ -54 +54 @@
             self.panel.pos[2] -= d*n
         else:
-            self.panel.pos[:2] += d*n        
+            self.panel.pos[:2] += d*n
 
     def update_color(self, c):
@@ -61 +61 @@
 
         self.panel.color = c
-        
+
 
     def set_vertexheights(self, heights, floor_heights = None):
@@ -70 +70 @@
             floor_heights = zeros(heights.shape, Float)
 
-        all_vertices_below_threshold = True    
+        all_vertices_below_threshold = True
         for k in range(3):
             w = heights[k]
@@ -77 +77 @@
             if w-z >= hmin:
                 all_vertices_below_threshold = False
-            
+
             vertex = self.panel.pos[k]
             vertex[2] = w
@@ -85 +85 @@
 
 
-        #Update color to visualise dry areas 
-        if all_vertices_below_threshold: 
+        #Update color to visualise dry areas
+        if all_vertices_below_threshold:
             self.panel.color = self.bottom_color
-            self.backpanel.color = self.bottom_color                
+            self.backpanel.color = self.bottom_color
         else:
             self.panel.color = self.top_color
@@ -94 +94 @@
 
 
-        #update normal    
+        #update normal
         v0 = self.panel.pos[0]
         v1 = self.panel.pos[1]
@@ -104 +104 @@
         self.backpanel.normal=-n
 
-        
+
 
 
@@ -120 +120 @@
     except:
         bed = False
-        
+
     N = Q.vertex_values.shape[0]
 
@@ -128 +128 @@
         z0 = Q.vertex_values[i, 0]
         z1 = Q.vertex_values[i, 1]
-        z2 = Q.vertex_values[i, 2]        
+        z2 = Q.vertex_values[i, 2]
 
         x0, y0 = domain.get_vertex_coordinate(i,0)
         x1, y1 = domain.get_vertex_coordinate(i,1)
-        x2, y2 = domain.get_vertex_coordinate(i,2)        
-        
+        x2, y2 = domain.get_vertex_coordinate(i,2)
+
         V0 = vector(x0, y0, z0)
         V1 = vector(x1, y1, z1)
@@ -149 +149 @@
 
         col = (c0,c1,c2)
-            
+
         visual_top = Triangle(V0,V1,V2,color=col,frame=fr)
         visual_top.top_color = col
 
-    
+
         #Bottom surface
         #v0, v1, v2 = volume.vertices
@@ -168 +168 @@
         V0 = vector(x0, y0, z0)
         V1 = vector(x1, y1, z1)
-        V2 = vector(x2, y2, z2)        
+        V2 = vector(x2, y2, z2)
 
         c0 = 0.3
@@ -176 +176 @@
             c2 = 0.4*c2   #To show triangles in slightly different shades
 
-        col = (c0,c1,c2)   
+        col = (c0,c1,c2)
         visual_bottom = Triangle(V0, V1, V2, color=col,frame=fr)
         visual_top.bottom_color=col
-        
+
         domain.visuals.append( (visual_top, visual_bottom) )
-        
+
     update(domain)
-    #print 'Scale', scene.scale    
-
-
-def update(domain):    
+    #print 'Scale', scene.scale
+
+
+def update(domain):
     """Update vertex heights.
-    The argument index refers to which conserved quantity to visualise.    
+    The argument index refers to which conserved quantity to visualise.
     If domain.smooth is set True, vertex values will be averaged
     yielding a smoother surface.
@@ -198 +198 @@
     Q = domain.quantities['stage']
     N = Q.vertex_values.shape[0]
-    
+
     #print scene.forward
     #FIXME: Use smoother from pyvolution instead
@@ -218 +218 @@
 
 
-    for k in range(N):    
+    for k in range(N):
         if domain.smooth:
             #The averages
             x = zeros(3, Float)
-            for i in range(3):            
+            for i in range(3):
                 vertex = domain.triangles[k, i]
                 A = array(vertex_heights[vertex])
                 x[i] = sum(A)/len(A)
-        else:        
+        else:
             #The true values
             x = [Q.vertex_values[k, 0],
@@ -232 +232 @@
                  Q.vertex_values[k, 2]]
 
-            
+
         #Do it
         floor_heights = array([pos[2] for pos in domain.visuals[k][1].panel.pos])
-        
+
         domain.visuals[k][0].set_vertexheights(x, floor_heights)
 
@@ -246 +246 @@
 #Original
 scene.center = (0.5,0.5,0)
-scene.forward = vector(-0.1, 0.5, -0.5)
+scene.forward = vector(0.0, 0.5, -0.5)
 
 #Temporary (for bedslope)
-#scene.forward = vector(0.0006, 0.7, -0.03)    
+#scene.forward = vector(0.0006, 0.7, -0.03)
 
 
@@ -258 +258 @@
 #Temporary for hackett - end
 
-    
+
 scene.ambient = 0.4
 scene.lights = [(0.6, 0.3, 0.2), (0.1, -0.5, 0.4), (-0.1, 0.1, -0.4),
@@ -266 +266 @@
 
 
-