def replace_fdef() -> bool:
"""Replace all arguments within the formula definitions."""
changes = False
for k, _fdata in enumerate(function_definitions):
for i in range(len(function_definitions)):
if i != k:
# replace i with k
formula = function_definitions[i]["formula"]
new_formula = xpp_helpers.replace_formula(
formula,
fid=function_definitions[k]["fid"],
old_args=function_definitions[k]["old_args"],
new_args=function_definitions[k]["new_args"],
)
if new_formula != formula:
function_definitions[i]["formula"] = new_formula
function_definitions[i]["new_args"] = list(
sorted(
set(
function_definitions[i]["new_args"]
+ function_definitions[k]["new_args"]
)
)
)
changes = True
return changes
def xpp2sbml(
xpp_file: Path,
sbml_file: Path,
force_lower: bool = False,
validate: bool = True,
debug: bool = False,
):
"""Reads given xpp_file and converts to SBML file.
:param xpp_file: xpp input ode file
:param sbml_file: sbml output file
:param force_lower: force lower case for all lines
:param validate: perform validation on the generated SBML file
:return:
"""
print("-" * 80)
print("xpp2sbml: ", xpp_file, "->", sbml_file)
print("-" * 80)
doc = libsbml.SBMLDocument(3, 1)
model = doc.createModel()
parameters = []
initial_assignments = []
rate_rules = []
assignment_rules = []
functions = [
# definition of min and max
fac.Function("max",
"lambda(x,y, piecewise(x,gt(x,y),y) )",
name="minimum"),
fac.Function("min",
"lambda(x,y, piecewise(x,lt(x,y),y) )",
name="maximum"),
# heav (heavyside)
fac.Function(
"heav",
"lambda(x, piecewise(0,lt(x,0), 0.5, eq(x, 0), 1,gt(x,0), 0))",
name="heavyside",
),
# mod (modulo)
fac.Function("mod", "lambda(x,y, x % y)", name="modulo"),
]
function_definitions = []
events = []
def replace_fdef():
""" Replace all arguments within the formula definitions."""
changes = False
for k, fdata in enumerate(function_definitions):
for i in range(len(function_definitions)):
if i != k:
# replace i with k
formula = function_definitions[i]["formula"]
new_formula = xpp_helpers.replace_formula(
formula,
fid=function_definitions[k]["fid"],
old_args=function_definitions[k]["old_args"],
new_args=function_definitions[k]["new_args"],
)
if new_formula != formula:
function_definitions[i]["formula"] = new_formula
function_definitions[i]["new_args"] = list(
sorted(
set(function_definitions[i]["new_args"] +
function_definitions[k]["new_args"])))
changes = True
return changes
def create_initial_assignment(sid, value):
""" Helper for creating initial assignments """
# check if valid identifier
if "(" in sid:
warnings.warn(
"sid is not valid: {}. Initial assignment is not generated".
format(sid))
return
try:
f_value = float(value)
parameters.append(
fac.Parameter(
sid=sid,
value=f_value,
name="{} = {}".format(sid, value),
constant=False,
))
except ValueError:
"""
Initial data are optional, XPP sets them to zero by default (many xpp model don't write the p(0)=0.
"""
parameters.append(
fac.Parameter(sid=sid, value=0.0, name=sid, constant=False))
initial_assignments.append(
fac.InitialAssignment(sid=sid,
value=value,
name="{} = {}".format(sid, value)))
###########################################################################
# First iteration to parse relevant lines and get the replacement patterns
###########################################################################
parsed_lines = []
# with open(xpp_file, encoding="utf-8") as f:
with open(xpp_file) as f:
lines = f.readlines()
# add info to sbml
text = escape_string("".join(lines))
fac.set_notes(model, NOTES.format(text))
old_line = None
for line in lines:
if force_lower:
line = line.lower()
# clean up the ends
line = line.rstrip("\n").strip()
# handle douple continuation characters in some models
line = line.replace("\\\\", "\\")
# handle semicolons
line = line.rstrip(";")
# join continuation
if old_line:
line = old_line + line
old_line = None
# empty line
if len(line) == 0:
continue
# comment line
if line[0] in XPP_COMMENT_CHARS:
continue
# xpp setting
if line.startswith(XPP_SETTING_CHAR):
continue
# end word
if line == XPP_END_WORD:
continue
# line continuation
if line.endswith(XPP_CONTINUATION_CHAR):
old_line = line.rstrip(XPP_CONTINUATION_CHAR)
continue
# handle the power function
line = line.replace("**", "^")
# handle if(...)then(...)else()
pattern_ite = re.compile(
r"if\s*\((.*)\)\s*then\s*\((.*)\)\s*else\s*\((.*)\)")
pattern_ite_sub = re.compile(
r"if\s*\(.*\)\s*then\s*\(.*\)\s*else\s*\(.*\)")
groups = re.findall(pattern_ite, line)
for group in groups:
condition = group[0]
assignment = group[1]
otherwise = group[2]
f_piecewise = "piecewise({}, {}, {})".format(
assignment, condition, otherwise)
line = re.sub(pattern_ite_sub, f_piecewise, line)
################################
# Function definitions
################################
""" Functions are defined in xpp via fid(arguments) = formula
f(x,y) = x^2/(x^2+y^2)
They can have up to 9 arguments.
The difference to SBML functions is that xpp functions have access to the global parameter values
"""
f_pattern = re.compile(r"(.*)\s*\((.*)\)\s*=\s*(.*)")
groups = re.findall(f_pattern, line)
if groups:
# function definitions found
fid, args, formula = groups[0]
# handles the initial assignments which look like function definitions
if args == "0":
parsed_lines.append(line)
continue
# necessary to find the additional arguments from the ast_node
ast = libsbml.parseL3Formula(formula)
names = set(xpp_helpers.find_names_in_ast(ast))
old_args = [t.strip() for t in args.split(",")]
new_args = [a for a in names if a not in old_args]
# handle special functions
if fid == "power":
warnings.warn(
"power function cannot be added to model, rename function."
)
else:
# store functions with additional arguments
function_definitions.append({
"fid": fid,
"old_args": old_args,
"new_args": new_args,
"formula": formula,
})
# don't append line, function definition has been handeled
continue
parsed_lines.append(line)
if debug:
print("\n\nFUNCTION_DEFINITIONS")
pprint(function_definitions)
# functions can use functions so this also must be replaced
changes = True
while changes:
changes = replace_fdef()
# clean the new arguments
for fdata in function_definitions:
fdata["new_args"] = list(sorted(set(fdata["new_args"])))
if debug:
print("\nREPLACED FUNCTION_DEFINITIONS")
pprint(function_definitions)
# Create function definitions
for k, fdata in enumerate(function_definitions):
fid = fdata["fid"]
formula = fdata["formula"]
arguments = ",".join(fdata["old_args"] + fdata["new_args"])
functions.append(
fac.Function(fid, "lambda({}, {})".format(arguments, formula)), )
###########################################################################
# Second iteration
###########################################################################
if debug:
print("\nPARSED LINES")
pprint(parsed_lines)
print("\n\n")
for line in parsed_lines:
# replace function definitions in lines
new_line = line
for fdata in function_definitions:
new_line = xpp_helpers.replace_formula(new_line, fdata["fid"],
fdata["old_args"],
fdata["new_args"])
if new_line != line:
if False:
print("\nReplaced FD", fdata["fid"], ":", new_line)
print("->", new_line, "\n")
line = new_line
if debug:
# line after function replacements
print("*" * 3, line, "*" * 3)
################################
# Start parsing the given line
################################
# check for the equal sign
tokens = line.split("=")
tokens = [t.strip() for t in tokens]
#######################
# Line without '=' sign
#######################
# wiener
if len(tokens) == 1:
items = [t.strip() for t in tokens[0].split(" ") if len(t) > 0]
# keyword, value
if len(items) == 2:
xid, sid = items[0], items[1]
xpp_type = parse_keyword(xid)
# wiener
if xpp_type == XPP_WIE:
"""Wiener parameters are normally distributed numbers with zero mean
and unit standard deviation. They are useful in stochastic simulations since
they automatically scale with change in the integration time step.
Their names are listed separated by commas or spaces."""
# FIXME: this should be encoded using dist
parameters.append(fac.Parameter(sid=sid, value=0.0))
continue # line finished
else:
warnings.warn("XPP line not parsed: '{}'".format(line))
#####################
# Line with '=' sign
#####################
# parameter, aux, ode, initial assignments
elif len(tokens) >= 2:
left = tokens[0]
items = [t.strip() for t in left.split(" ") if len(t) > 0]
# keyword based information, i.e 2 items are on the left of the first '=' sign
if len(items) == 2:
xid = items[0] # xpp keyword
xpp_type = parse_keyword(xid)
expression = (" ".join(items[1:]) + "=" + "=".join(tokens[1:])
) # full expression after keyword
parts = parts_from_expression(expression)
if False:
print("xid:", xid)
print("expression:", expression)
print("parts:", parts)
# parameter & numbers
if xpp_type in [XPP_PAR, XPP_NUM]:
"""Parameter values are optional; if not they are set to zero.
Number declarations are like parameter declarations, except that they cannot be
changed within the program and do not appear in the parameter window."""
for part in parts:
sid, value = sid_value_from_part(part)
create_initial_assignment(sid, value)
# aux
elif xpp_type == XPP_AUX:
"""Auxiliary quantities are expressions that depend on all of your dynamic
variables which you want to keep track of. Energy is one such example. They are declared
like fixed quantities, but are prefaced by aux ."""
for part in parts:
sid, value = sid_value_from_part(part)
if sid == value:
# avoid circular dependencies (no information in statement)
pass
else:
assignment_rules.append(
fac.AssignmentRule(sid=sid, value=value))
# init
elif xpp_type == XPP_INIT:
for part in parts:
sid, value = sid_value_from_part(part)
create_initial_assignment(sid, value)
# table
elif xpp_type == XPP_TAB:
"""The Table declaration allows the user to specify a function of 1 variable in terms
of a lookup table which uses linear interpolation. The name of the function follows the
declaration and this is followed by (i) a filename (ii) or a function of "t"."""
warnings.warn(
"XPP_TAB not supported: XPP line not parsed: '{}'".
format(line))
else:
warnings.warn("XPP line not parsed: '{}'".format(line))
elif len(items) >= 2:
xid = items[0]
xpp_type = parse_keyword(xid)
# global
if xpp_type == XPP_GLO:
"""Global flags are expressions that signal events when they change sign, from less than
to greater than zero if sign=1 , greater than to less than if sign=-1 or either way
if sign=0. The condition should be delimited by braces {} The events are of the form
variable=expression, are delimited by braces, and separated by semicolons. When the
condition occurs all the variables in the event set are changed possibly discontinuously.
"""
# global sign {condition} {name1 = form1; ...}
pattern_global = re.compile(
r"([+,-]{0,1}\d{1})\s+\{{0,1}(.*)\{{0,1}\s+\{(.*)\}")
groups = re.findall(pattern_global, line)
if groups:
g = groups[0]
sign = int(g[0])
trigger = g[1]
# FIXME: handle sign=-1, sign=0, sign=+1
if sign == -1:
trigger = g[1] + ">= 0"
elif sign == 1:
trigger = g[1] + ">= 0"
elif sign == 0:
trigger = g[1] + ">= 0"
assignment_parts = [t.strip() for t in g[2].split(";")]
assignments = {}
for p in assignment_parts:
key, value = p.split("=")
assignments[key] = value
events.append(
fac.Event(
sid="e{}".format(len(events)),
trigger=trigger,
assignments=assignments,
))
else:
warnings.warn(
"global expression could not be parsed: {}".format(
line))
else:
warnings.warn("XPP line not parsed: '{}'".format(line))
# direct assignments
elif len(items) == 1:
right = tokens[1]
# init
if left.endswith("(0)"):
sid, value = left[0:-3], right
create_initial_assignment(sid, value)
# difference equations
elif left.endswith("(t+1)"):
warnings.warn(
"Difference Equations not supported: XPP line not parsed: '{}'"
.format(line))
# ode
elif left.endswith("'"):
sid = left[0:-1]
rate_rules.append(fac.RateRule(sid=sid, value=right))
elif left.endswith("/dt"):
sid = left[1:-3]
rate_rules.append(fac.RateRule(sid=sid, value=right))
# assignment rules
else:
assignment_rules.append(
fac.AssignmentRule(sid=left, value=right))
else:
warnings.warn("XPP line not parsed: '{}'".format(line))
# add time
assignment_rules.append(
fac.AssignmentRule(sid="t", value="time", name="model time"))
# create SBML objects
objects = (parameters + initial_assignments + functions + rate_rules +
assignment_rules + events)
fac.create_objects(model, obj_iter=objects, debug=False)
"""
Parameter values are optional; if not they are set to zero in xpp.
Many models do not encode the initial zeros.
"""
for p in doc.getModel().getListOfParameters():
if not p.isSetValue():
p.setValue(0.0)
sbml.write_sbml(
doc,
sbml_file,
validate=validate,
program_name="sbmlutils",
program_version=__version__,
units_consistency=False,
)