muMLE/examples/cbd/models.py
2024-11-13 10:07:16 +01:00

41 lines
1.4 KiB
Python

from util import loader
import os
THIS_DIR = os.path.dirname(__file__)
# get file contents as string
def read_file(filename):
with open(THIS_DIR+'/'+filename) as file:
return file.read()
def load_metamodels(state, scd_mmm):
mm_cs = read_file('models/mm_design.od')
mm_rt_cs = mm_cs + read_file('models/mm_runtime.od')
mm = loader.parse_and_check(state, mm_cs, scd_mmm, "Design meta-model")
mm_rt = loader.parse_and_check(state, mm_rt_cs, scd_mmm, "Runtime meta-model")
return (mm, mm_rt)
def load_fibonacci(state, scd_mmm):
mm, mm_rt = load_metamodels(state, scd_mmm)
m_cs = read_file('models/m_fibonacci.od')
m_rt_initial_cs = m_cs + read_file('models/m_fibonacci_initial.od')
m = loader.parse_and_check(state, m_cs, mm, "Fibonacci model")
m_rt_initial = loader.parse_and_check(state, m_rt_initial_cs, mm_rt, "Fibonacci initial state")
return (mm, mm_rt, m, m_rt_initial)
RULES0 = ["delay_in", "delay_out", "function_out"] # high priority
RULES1 = ["advance_time"] # low priority
def load_rules(state, mm_rt_ramified):
get_filename = lambda rule_name, kind: f"{THIS_DIR}/models/r_{rule_name}_{kind}.od"
rules0 = loader.load_rules(state, get_filename, mm_rt_ramified, RULES0)
rules1 = loader.load_rules(state, get_filename, mm_rt_ramified, RULES1)
return rules0, rules1