diff options
Diffstat (limited to 'Python/pywarpx/Constants.py')
| -rw-r--r-- | Python/pywarpx/Constants.py | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/Python/pywarpx/Constants.py b/Python/pywarpx/Constants.py index 93c45c92e..1ae1dda0d 100644 --- a/Python/pywarpx/Constants.py +++ b/Python/pywarpx/Constants.py @@ -4,6 +4,8 @@ # # License: BSD-3-Clause-LBNL +import re + from .Bucket import Bucket class Constants(Bucket): @@ -19,4 +21,33 @@ class Constants(Bucket): assert self.argvattrs[name] == value, Exception('Inconsistent values given for user defined constants') Bucket.__setattr__(self, name, value) + def add_keywords(self, kwdict): + mangle_dict = {} + for k,v in kwdict.items(): + # Check if keyword has already been defined + # WarpX has a single global dictionary of expression variables so each + # variable must be unique + if k in self.argvattrs: + # if so, mangle the name by appending a numerical suffix + mangle_number = 1 + k_mangled = f'{k}{mangle_number}' + while k_mangled in self.argvattrs: + # make sure that the mangled name has also not already been defined + mangle_number += 1 + k_mangled = f'{k}{mangle_number}' + mangle_dict[k] = k_mangled + k = k_mangled + setattr(self, k, v) + return mangle_dict + + def mangle_expression(self, expression, mangle_dict): + if expression is None: + return None + # For each key in mangle_dict, modify the expression replacing + # the key with its value, the mangled version of key + for k,v in mangle_dict.items(): + expression = re.sub(r'\b%s\b'%k, v, expression) + return expression + + my_constants = Constants() |