include "primitives.alh"
include "conformance_scd.alh"
include "constructors.alh"

Element function allInstances(model : Element, type : Element):
	Element type_mapping
	Element result

	type_mapping = model["type_mapping"]
	result = create_node()

	Integer counter
	counter = 0

	Integer length
	length = read_nr_out(type_mapping)

	Element edge
	while (counter < length):
		edge = read_out(type_mapping, counter)
		if (element_eq(read_edge_dst(edge), type)):
			// Found an element of the specified type
			set_add(result, read_edge_dst(read_out(edge, 0)))
		counter = counter + 1
	
	return result

Element function selectPossibleIncoming(model : Element, target : String, limit_set : Element):
	// Find all possible incoming link types for the target model
	// Should also include those specified on the superclass(es)

	String type
	Element metamodel
	Element elem
	Element result

	result = create_node()
	metamodel = model["metamodel"]

	while (0 < list_len(limit_set)):
		type = set_pop(limit_set)
		elem = metamodel["model"][type]
		if (is_edge(elem)):
			if (is_nominal_instance(model, model["model"][target], read_edge_dst(elem))):
				set_add(result, type)
	
	return result

Element function selectPossibleOutgoing(model : Element, source : String, limit_set : Element):
	// Find all possible outgoing link types for the source model
	// Should also include those specified on the superclass(es)
	String type
	Element metamodel
	Element elem
	Element result

	result = create_node()
	metamodel = model["metamodel"]
	
	while (0 < list_len(limit_set)):
		type = set_pop(limit_set)
		elem = metamodel["model"][type]
		if (is_edge(elem)):
			if (is_nominal_instance(model, model["model"][source], read_edge_src(elem))):
				set_add(result, type)
	
	return result

Element function allOutgoingAssociationInstances(model : Element, source_name : String, assoc_name : String):
	// Read out all outgoing edges of the model and select those that are typed by the specified association
	// TODO for some reason this crashes if allInstances is used!

	Integer nr_out
	Integer i
	Element out
	String out_name
	Element result

	result = create_node()
	nr_out = read_nr_out(model["model"][source_name])
	i = 0
	while (i < nr_out):
		out = read_out(model["model"][source_name], i)
		out_name = reverseKeyLookup(model["model"], out)
		if (is_nominal_instance(model, out, model["metamodel"]["model"][assoc_name])):
			set_add(result, out_name)
		i = i + 1
	return result

Element function allIncomingAssociationInstances(model : Element, target_name : String, assoc_name : String):
	// Read out all outgoing edges of the model and select those that are typed by the specified association
	Integer nr_in
	Integer i
	Element in
	String in_name
	Element result

	result = create_node()
	nr_in = read_nr_in(model["model"][target_name])
	i = 0
	while (i < nr_in):
		in = read_out(model["model"][target_name], i)
		in_name = reverseKeyLookup(model["model"], in)
		if (is_nominal_instance(model, in, model["metamodel"]["model"][assoc_name])):
			set_add(result, in_name)
		i = i + 1
	return result

Element function getAttributeList(model : Element, element : String):
	Element result
	Element keys
	Element type
	Element attr_name
	String attr_type

	result = create_node()
	type = dict_read_node(model["type_mapping"], model["model"][element])
	keys = dict_keys(type)

	// Add our own attributes
	while (0 < list_len(keys)):
		attr_name = set_pop(keys)
		if (is_physical_string(attr_name)):
			attr_type = getName(model["metamodel"], type[attr_name])
			dict_add(result, attr_name, attr_type)

	// Go on to the metalevel
	// TODO

	return result

Element function getInstantiatableAttributes(model : Element, element : String):
	Element result
	result = create_node()
	// Get all outgoing "dictionary" links
	Element set_own
	Element elem
	elem = model["model"][element]
	set_own = dict_keys(element)

	// Filter them
	Element e
	while (0 < read_nr_out(set_own)):
		e = set_pop(set_own)
		if (is_physical_string(e)):
			dict_add(result, e, getName(model, element[e]))

	return result

String function getName(model : Element, element : Element):
	return reverseKeyLookup(model["model"], element)

// TODO Utility functions!
String function reverseKeyLookup(dict : Element, element : Element):
	Element elements
	String name

	elements = dict_keys(dict)
	while (0 < list_len(elements)):
		name = set_pop(elements)
		if (element_eq(dict[name], element)):
			return name

	return string_join(string_join("(unknown: ", cast_e2s(element)), " )")

String function print_dict(dict : Element):
	Element keys
	Element key
	String result

	keys = dict_keys(dict)
	result = ""
	while (0 < list_len(keys)):
		key = set_pop(keys)
		result = result + cast_v2s(key)
		result = result + ": "
		result = result + cast_v2s(dict[key])
		result = result + "\n"
	return result