Changeset 8578 for trunk/anuga_core/source/anuga/pmesh/mesh_quadtree.py

Timestamp:

Sep 14, 2012, 9:56:39 PM (11 years ago)

Author:

steve

Message:

Added extra unit tests for set_stage and set_elevation operators

File:

• trunk/anuga_core/source/anuga/pmesh/mesh_quadtree.py (modified) (5 diffs)

trunk/anuga_core/source/anuga/pmesh/mesh_quadtree.py

@@ -24 +24 @@
 
 import numpy as num
+import sys
  
 
@@ -41 +42 @@
         It contains optimisations and search patterns specific to meshes.
     """
-    def __init__(self, mesh):
+    def __init__(self, mesh, verbose=False):
         """Build quad tree for mesh.
 
@@ -60 +61 @@
         
         normals = mesh.get_normals()
-        
+
+        if verbose :
+            print '['+60*' '+']',
+            sys.stdout.flush()
+
+        M = N/60
+
         # Check each triangle
-        for i in range(N):
+        for i in xrange(N):
+
+            if verbose and i%M == 0 :
+                #restart_line()
+                print '\r['+(i/M)*'.'+(60-(i/M))*' ' +']',
+                sys.stdout.flush()
+
             i3 = 3*i
             x0, y0 = V[i3, :]
@@ -68 +81 @@
             x2, y2 = V[i3+2, :]
 
+            #FIXME SR: Should save memory by just using triangle id!
             node_data = [i, V[i3:i3+3, :], normals[i, :]]
 
@@ -74 +88 @@
                              min([y0, y1, y2]), max([y0, y1, y2])), \
                              node_data))
+
+        if verbose:
+            print ''
 
     def search_fast(self, point):