Changeset 7975 for trunk/anuga_core/source/anuga/structures/inlet.py

Timestamp:

Aug 25, 2010, 4:35:40 PM (14 years ago)

Author:

steve

Message:

Got a working culvert!

File:

• trunk/anuga_core/source/anuga/structures/inlet.py (moved) (moved from trunk/anuga_core/source/anuga/structures/Inlet.py) (4 diffs)

trunk/anuga_core/source/anuga/structures/inlet.py

@@ +1 @@
+
 from anuga.geometry.polygon import inside_polygon, is_inside_polygon
-from anuga.config import velocity_protection 
+from anuga.config import velocity_protection, g
 import math
 
@@ -117 +118 @@
     def get_total_water_volume(self):
        
-       return num.sum(self.get_heights*self.get_areas())
+       return num.sum(self.get_heights()*self.get_areas())
     
     
@@ -134 +135 @@
             
             
-    def get_average_velocity(self):
+    def get_average_speed(self):
  
             u, v = self.get_velocities()
@@ -141 +142 @@
             average_v = num.sum(v)/self.triangle_indices.size
             
-            return math.sqrt(average_u**2 + average_v**2) 
+            return math.sqrt(average_u**2 + average_v**2)
 
 
-    def set_total_energy(self, total_energy):
+
+    def get_average_velocity_head(self):
+
+        return 0.5*self.get_average_speed()**2/g
+
+
+    def get_average_total_energy(self):
         
-        self.total_energy = total_energy
+        return self.get_average_velocity_head() + self.get_average_stage()
         
+    
+    def get_average_specific_energy(self):
         
-    def get_total_energy(self):
-        
-        return self.total_energy
-        
-        
-    def set_specific_energy(self, specific_energy):
-        
-        self.specific_energy = specific_energy
+        return self.get_average_velocity_head() + self.get_average_height()
 
-    
-    def get_specific_energy(self):
-        
-        return self.specific_energy
-        
-        
+
+
+    def set_heights(self,height):
+
+        self.domain.quantities['stage'].centroid_values.put(self.triangle_indices, self.get_elevations() + height)
+
+
+    def set_stages(self,stage):
+
+        self.domain.quantities['stage'].centroid_values.put(self.triangle_indices, stage)
+
+    def set_xmoms(self,xmom):
+
+        self.xmoms=self.domain.quantities['xmomentum'].centroid_values.put(self.triangle_indices, xmom)
+
+
+    def set_ymoms(self,ymom):
+
+        self.xmoms=self.domain.quantities['ymomentum'].centroid_values.put(self.triangle_indices, ymom)
+
+
+    def set_elevations(self,elevation):
+
+        self.domain.quantities['elevation'].centroid_values.put(self.triangle_indices, elevation)