Module: CodeModels::NavigationExtensions

Included in:
CodeModelsAstNode
Defined in:
lib/codemodels/navigation.rb

Instance Method Summary collapse

Instance Method Details

#all_children(flag = nil) ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/codemodels/navigation.rb', line 5

def all_children(flag=nil)
	also_foreign = (flag==:also_foreign)
	arr = []
	ecore = self.class.ecore
	# Awful hack to forbid the same reference is visited twice when
	# two references with the same name are found
	already_used_references = []
	ecore.eAllReferences.select {|r| r.containment}.each do |ref|
		#raise "Too many features with name #{ref.name}. Count: #{features_by_name(ref.name).count}" if features_by_name(ref.name).count!=1
		unless already_used_references.include?(ref.name)
			res = self.send(ref.name.to_sym)
			if ref.many
				d = arr.count
				res.each do |el|
					arr << el unless res==nil
				end
			elsif res!=nil
				d = arr.count
				arr << res
			end
			already_used_references << ref.name
		end
	end
	if also_foreign
		arr.concat(self.foreign_asts)
	end
	arr
end

#all_children_also_foreign ⇒ Object 



107
108
109
 
# File 'lib/codemodels/navigation.rb', line 107

def all_children_also_foreign
	all_children(:also_foreign)
end

#all_children_deep(flag = nil) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
 
# File 'lib/codemodels/navigation.rb', line 34

def all_children_deep(flag=nil)
	arr = []
	all_children(flag).each do |c|
		arr << c
		c.all_children_deep(flag).each do |cc|
			arr << cc
		end
	end			
	arr
end

#all_children_deep_also_foreign ⇒ Object 



111
112
113
 
# File 'lib/codemodels/navigation.rb', line 111

def all_children_deep_also_foreign
	all_children_deep(:also_foreign)
end

#all_children_deep_of_type(flag = nil, type) ⇒ Object 



81
82
83
 
# File 'lib/codemodels/navigation.rb', line 81

def all_children_deep_of_type(flag=nil,type)
	all_children_deep(flag).select {|c| c and c.is_a?(type)}
end

#all_children_of_type(flag = nil, type) ⇒ Object 



77
78
79
 
# File 'lib/codemodels/navigation.rb', line 77

def all_children_of_type(flag=nil,type)
	all_children(flag).select {|c| c and c.is_a?(type)}
end

#collect_values_with_count ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/codemodels/navigation.rb', line 52

def collect_values_with_count
	values = Hash.new {|h,k| h[k]=0}
	self.class.ecore.eAllAttributes.each do |a|
		v = self.send(:"#{a.name}")
		if v!=nil
			if a.many
				v.each {|el| values[el]+=1}
			else
				values[v]+=1
			end
		end
	end
	values			
end

#collect_values_with_count_subtree(flag = nil) ⇒ Object 



67
68
69
70
71
72
73
74
75
 
# File 'lib/codemodels/navigation.rb', line 67

def collect_values_with_count_subtree(flag=nil)
	values = collect_values_with_count
	all_children_deep(flag).each do |c|
		c.collect_values_with_count.each do |k,v|
			values[k]+=v
		end
	end
	values
end

#container(flag = nil) ⇒ Object 



97
98
99
100
101
102
103
104
105
 
# File 'lib/codemodels/navigation.rb', line 97

def container(flag=nil)
	also_foreign = (flag==:also_foreign)
	return self.eContainer if self.eContainer
	if also_foreign
		self.foreign_container
	else
		nil
	end
end

#only_child_deep_of_type(flag = nil, type) ⇒ Object 



91
92
93
94
95
 
# File 'lib/codemodels/navigation.rb', line 91

def only_child_deep_of_type(flag=nil,type)
	selected = all_children_deep_of_type(flag,type)
	raise "Exactly one child of type #{type} expected, #{selected.count} found on #{self}" unless selected.count==1
	selected[0]
end

#only_child_of_type(flag = nil, type) ⇒ Object 



85
86
87
88
89
 
# File 'lib/codemodels/navigation.rb', line 85

def only_child_of_type(flag=nil,type)
	selected = all_children_of_type(flag,type)
	raise "Exactly one child of type #{type} expected, #{selected.count} found on #{self}" unless selected.count==1
	selected[0]
end

#traverse(flag = nil, &op) ⇒ Object 



45
46
47
48
49
50
 
# File 'lib/codemodels/navigation.rb', line 45

def traverse(flag=nil,&op)
	op.call(self)
	all_children_deep(flag).each do |c|
		op.call(c)
	end
end

#traverse_also_foreign(&block) ⇒ Object 



115
116
117
 
# File 'lib/codemodels/navigation.rb', line 115

def traverse_also_foreign(&block)
	traverse(:also_foreign,block)
end