Changeset 8506 for trunk/anuga_core/source/anuga/structures/inlet_operator.py

Timestamp:

Aug 13, 2012, 3:47:51 PM (13 years ago)

Author:

steve

Message:

Adding in default argument

File:

• trunk/anuga_core/source/anuga/structures/inlet_operator.py (modified) (5 diffs)

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

@@ -3 +3 @@
 import inlet
 
+from warnings import warn
 
 class Inlet_operator(anuga.Operator):
@@ -19 +20 @@
                  line,
                  Q = 0.0,
+                 default = None,
                  description = None,
                  label = None,
@@ -40 +42 @@
         self.applied_Q = 0.0
 
+        self.set_default(default)
+
+
     def __call__(self):
 
@@ -71 +76 @@
         """Allowing local modifications of Q
         """
+        from anuga.fit_interpolate.interpolate import Modeltime_too_early, Modeltime_too_late
         
         if callable(self.Q):
-            Q = self.Q(t)[0]
+            try:
+                Q = self.Q(t)[0]
+            except Modeltime_too_early, e:
+                raise Modeltime_too_early(e)
+            except Modeltime_too_late, e:
+                Q = self.get_default(t,err_msg=e)
         else:
             Q = self.Q
@@ -122 +133 @@
 
 
+    def set_default(self, default=None):
+        
+        """ Either leave default as None or change it into a function"""
+
+        if default is not None:
+            # If it is a constant, make it a function
+            if not callable(default):
+                tmp = default
+                default = lambda t: tmp
+
+            # Check that default_rate is a function of one argument
+            try:
+                default(0.0)
+            except:
+                raise Exception(msg)
+
+        self.default = default
+        self.default_invoked = False
+
+
+
+    def get_default(self,t, err_msg=' '):
+        """ Call get_default only if exception
+        Modeltime_too_late(msg) has been raised
+        """
+
+        from anuga.fit_interpolate.interpolate import Modeltime_too_early, Modeltime_too_late
+
+
+        if self.default is None:
+            msg = '%s: ANUGA is trying to run longer than specified data.\n' %str(err_msg)
+            msg += 'You can specify keyword argument default in the '
+            msg += 'operator to tell it what to do in the absence of time data.'
+            raise Modeltime_too_late(msg)
+        else:
+            # Pass control to default rate function
+            value = self.default(t)
+
+            if self.default_invoked is False:
+                # Issue warning the first time
+                msg = ('%s\n'
+                       'Instead I will use the default rate: %s\n'
+                       'Note: Further warnings will be supressed'
+                       % (str(err_msg), str(self.default(t))))
+                warn(msg)
+
+                # FIXME (Ole): Replace this crude flag with
+                # Python's ability to print warnings only once.
+                # See http://docs.python.org/lib/warning-filter.html
+                self.default_invoked = True
+
+        return value
+
+
     def set_Q(self, Q):