Changeset 7793 for anuga_work/development/anuga_1d/domain.py

Timestamp:

Jun 6, 2010, 10:48:41 PM (13 years ago)

Author:

steve

Message:

Setting up test function for Quantity

File:

• anuga_work/development/anuga_1d/domain.py (modified) (2 diffs)

anuga_work/development/anuga_1d/domain.py

@@ -112 +112 @@
         self.build_tagged_elements_dictionary(tagged_elements)
         
-        from quantity import Quantity, Conserved_quantity
-        #from quantity_domain import Quantity, Conserved_quantity
+        from quantity import Quantity
+
         
         #List of quantity names entering
@@ -592 +592 @@
             return self.vertices[elem_id,vertex]
 
-    def get_area(self, elem_id):
-        """Return area of element id
-        """
-
-        return self.areas[elem_id]
+    def get_area(self, elem_id=None):
+        """Return total domain area or area of element id
+        """
+
+        if elem_id is None:
+            return sum(self.areas)
+        else:
+            return self.areas[elem_id]
 
     def get_quantity(self, name, location='vertices', indices = None):