Changeset 715 for inundation/ga/storm_surge/pyvolution/region.py

Timestamp:

Dec 22, 2004, 2:20:59 PM (20 years ago)

Author:

duncan

Message:

added getting and setting of unique vertex quantities, with a subset of triangles

File:

• inundation/ga/storm_surge/pyvolution/region.py (modified) (4 diffs)

inundation/ga/storm_surge/pyvolution/region.py

@@ -12 +12 @@
     """   
 
-    def __init__(self):
-        pass
+    def __init__(self, location='vertices'):
+        self.location = location
         
     def __call__(self, tag, elements, domain):
@@ -20 +20 @@
 
 
-class Set_Region(Region):
+    def build_indexes(self, elements, domain):
+        """
+        Return a list of triangle_id or vertex_id, depending on the location
+        """
+        if self.location == 'unique vertices':
+            return domain.get_unique_vertices(elements)
+        else:
+            return elements
+                
+class Set_region(Region):
     
     def __init__(self, tag, quantity, X, location='vertices'):
@@ -43 +52 @@
         """    
         if tag == self.tag:
-            domain.set_quantity(self.quantity, self.X, indexes = elements)
+            domain.set_quantity(self.quantity,
+                                self.X,
+                                location=self.location,
+                                indexes=self.build_indexes(elements, domain))
 
-
-class Add_Value_To_Region(Region):
+        
+class Add_value_to_region(Region):
     """
     Will add a value to the current quantity value.
     """
     
-    def __init__(self, tag, quantity, X, location='vertex'):
-        Region.__init__(self)
+    def __init__(self, tag, quantity, X, location='vertices', initial_quantity=None):
+        #I have to get this going!
+        #Region.__init__(self)
         self.tag = tag 
-        self.quantity = quantity
+        self.quantity_answer = quantity
         self.location = location
         self.X = X
+        if initial_quantity is None:
+            self.quantity_initial_value = quantity
+        else:
+            self.quantity_initial_value = initial_quantity
         if callable(X):
             raise 'This class does not work with functions'  
@@ -67 +84 @@
         """    
         if tag == self.tag:
-            new_values = domain.get_quantity(self.quantity,
-                                             indexes = elements) + self.X
-            domain.set_quantity(self.quantity, new_values, indexes = elements)
+            new_values = domain.get_quantity(self.quantity_initial_value,
+                          indexes=self.build_indexes(elements, domain),
+                          location=self.location) + self.X
+            domain.set_quantity(self.quantity_answer, new_values,
+                                indexes=self.build_indexes(elements, domain),
+                                location=self.location)
+
+
+class Stage_no_less_than_elevation(Region):
+    """
+    Will set the stage to not be less than the elevation.
+    This would be good, but it's not region dependent.
+    Wait for it to become a default for pyvolution.
+    """
+    
+    def __init__(self):
+        pass