muMLE/transformation/rewriter.py

158 lines
8.1 KiB
Python

# Things you can do:
# - Create/delete objects, associations, attributes
# - Change attribute values
# - ? that's it?
from uuid import UUID
from services.bottom.V0 import Bottom
from transformation import ramify
from services import od
from services.primitives.string_type import String
from services.primitives.integer_type import Integer
def process_rule(state, lhs: UUID, rhs: UUID):
bottom = Bottom(state)
to_delete = { name for name in bottom.read_keys(lhs) if name not in bottom.read_keys(rhs) }
to_create = { name for name in bottom.read_keys(rhs) if name not in bottom.read_keys(lhs) }
common = { name for name in bottom.read_keys(lhs) if name in bottom.read_keys(rhs) }
# print("to_delete:", to_delete)
# print("to_create:", to_create)
return to_delete, to_create, common
# Rewrite is performed in-place (modifying `host_m`)
# Also updates the `mapping` in-place, to become RHS -> host
def rewrite(state, lhs_m: UUID, rhs_m: UUID, pattern_mm: UUID, name_mapping: dict, host_m: UUID, mm: UUID):
bottom = Bottom(state)
scd_metamodel_id = state.read_dict(state.read_root(), "SCD")
scd_metamodel = UUID(state.read_value(scd_metamodel_id))
class_type = od.get_scd_mm_class_node(bottom)
attr_link_type = od.get_scd_mm_attributelink_node(bottom)
assoc_type = od.get_scd_mm_assoc_node(bottom)
modelref_type = od.get_scd_mm_modelref_node(bottom)
m_od = od.OD(mm, host_m, bottom.state)
rhs_od = od.OD(pattern_mm, rhs_m, bottom.state)
to_delete, to_create, common = process_rule(state, lhs_m, rhs_m)
# Perform deletions
for pattern_name_to_delete in to_delete:
# For every name in `to_delete`, look up the name of the matched element in the host graph
model_el_name_to_delete = name_mapping[pattern_name_to_delete]
# print('deleting', model_el_name_to_delete)
# Look up the matched element in the host graph
el_to_delete, = bottom.read_outgoing_elements(host_m, model_el_name_to_delete)
# Delete
bottom.delete_element(el_to_delete)
# Remove from mapping
del name_mapping[pattern_name_to_delete]
# extended_mapping = dict(name_mapping) # will be extended with created elements
edges_to_create = [] # postpone creation of edges after creation of nodes
# Perform creations
for pattern_name_to_create in to_create:
# print('creating', pattern_name_to_create)
# We have to come up with a name for the element-to-create in the host graph
i = 0
while True:
model_el_name_to_create = pattern_name_to_create + str(i) # use the label of the element in the RHS as a basis
if len(bottom.read_outgoing_elements(host_m, model_el_name_to_create)) == 0:
break # found an available name
# Determine the type of the thing to create
rhs_el_to_create, = bottom.read_outgoing_elements(rhs_m, pattern_name_to_create)
rhs_type = od.get_type(bottom, rhs_el_to_create)
original_type = ramify.get_original_type(bottom, rhs_type)
if original_type != None:
# Now get the type of the type
if od.is_typed_by(bottom, original_type, class_type):
# It's type is typed by Class -> it's an object
# print(' -> creating object')
o = m_od._create_object(model_el_name_to_create, original_type)
name_mapping[pattern_name_to_create] = model_el_name_to_create
elif od.is_typed_by(bottom, original_type, attr_link_type):
# print(' -> postpone (is attribute link)')
edges_to_create.append((pattern_name_to_create, rhs_el_to_create, original_type, 'attribute link', model_el_name_to_create))
elif od.is_typed_by(bottom, original_type, assoc_type):
# print(' -> postpone (is link)')
edges_to_create.append((pattern_name_to_create, rhs_el_to_create, original_type, 'link', model_el_name_to_create))
else:
original_type_name = od.get_object_name(bottom, mm, original_type)
print(" -> warning: don't know about", original_type_name)
else:
# print(" -> no original (un-RAMified) type")
# assume the type of the object is already the original type
# this is because primitive types (e.g., Integer) are not RAMified
type_name = od.get_object_name(bottom, pattern_mm, rhs_type)
if type_name == "String":
# Assume the string is a Python expression to evaluate
python_expr = String(UUID(bottom.read_value(rhs_el_to_create)), bottom.state).read()
result = eval(python_expr, {}, {})
# Write the result into the host model.
# This will be the *value* of an attribute. The attribute-link (connecting an object to the attribute) will be created as an edge later.
if isinstance(result, int):
m_od.create_integer_value(model_el_name_to_create, result)
elif isinstance(result, str):
m_od.create_string_value(model_el_name_to_create, result)
name_mapping[pattern_name_to_create] = model_el_name_to_create
else:
raise Exception(f"RHS element '{pattern_name_to_create}' needs to be created in host, but has no un-RAMified type, and I don't know what to do with it. It's type is", type_name)
# print("create edges....")
for pattern_name_to_create, rhs_el_to_create, original_type, original_type_name, model_el_name_to_create in edges_to_create:
# print('creating', pattern_name_to_create)
if original_type_name == 'attribute link':
# print(' -> creating attribute link')
src = bottom.read_edge_source(rhs_el_to_create)
src_name = od.get_object_name(bottom, rhs_m, src)
tgt = bottom.read_edge_target(rhs_el_to_create)
tgt_name = od.get_object_name(bottom, rhs_m, tgt)
obj_name = name_mapping[src_name] # name of object in host graph to create slot for
orig_attr_name = od.get_attr_name(bottom, original_type)
m_od.create_slot(orig_attr_name, obj_name, name_mapping[tgt_name])
elif original_type_name == 'link':
# print(' -> creating link')
src = bottom.read_edge_source(rhs_el_to_create)
src_name = od.get_object_name(bottom, rhs_m, src)
tgt = bottom.read_edge_target(rhs_el_to_create)
tgt_name = od.get_object_name(bottom, rhs_m, tgt)
obj_name = name_mapping[src_name] # name of object in host graph to create slot for
attr_name = od.get_object_name(bottom, mm, original_type)
m_od.create_link(model_el_name_to_create, attr_name, obj_name, name_mapping[tgt_name])
# Perform updates (only on values)
for pattern_el_name in common:
model_el_name = name_mapping[pattern_el_name]
# print('updating', model_el_name)
model_el, = bottom.read_outgoing_elements(host_m, model_el_name)
host_type = od.get_type(bottom, model_el)
if od.is_typed_by(bottom, host_type, class_type):
# print(' -> is classs')
# nothing to do
pass
elif od.is_typed_by(bottom, host_type, attr_link_type):
# print(' -> is attr link')
# nothing to do
pass
elif od.is_typed_by(bottom, host_type, modelref_type):
# print(' -> is modelref')
old_value = od.read_primitive_value(bottom, model_el, mm)
rhs_el, = bottom.read_outgoing_elements(rhs_m, pattern_el_name)
expr = od.read_primitive_value(bottom, rhs_el, pattern_mm)
result = eval(expr, {}, {'v': old_value})
# print('eval result=', result)
if isinstance(result, int):
# overwrite the old value, in-place
referred_model_id = UUID(bottom.read_value(model_el))
Integer(referred_model_id, state).create(result)
else:
raise Exception("Unimplemented type. Value:", result)
else:
raise Exception("Don't know what to do with element of type", host_type)