Skip to main content
aboutsummaryrefslogtreecommitdiffstats
blob: c307273d1a0c8bb35fbc426019ce383928b5cb4f (plain) (blame)
1
2
3
4
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
/*******************************************************************************
 * Copyright (c) 2011 protos software gmbh (http://www.protos.de).
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * CONTRIBUTORS:
 * 		Henrik Rentz-Reichert (initial contribution)
 * 
 *******************************************************************************/

package org.eclipse.etrice.dctools.ast.internal

import org.eclipse.etrice.core.room.Attribute
import org.eclipse.etrice.core.room.InterfaceItem
import org.eclipse.etrice.core.room.Message
import org.eclipse.etrice.core.room.Operation
import org.eclipse.etrice.dctools.ast.ITranslationProvider
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstArrayAccessNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstBracketNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstBracketNode.BracketType
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstIdentifierNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstOperationCallNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstOtherNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstPeriodNode
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstWhitespaceNode
import org.eclipse.etrice.dctools.fsm.ast.util.DCAstPrinter
import org.eclipse.etrice.dctools.fsm.ast.util.IDCAstNodeVisitor
import org.eclipse.etrice.dctools.fsm.ast.nodes.DCAstFeatureCallNode
import org.eclipse.etrice.core.room.Port
import org.eclipse.etrice.core.room.VarDecl

class DCTranslatorVisitor implements IDCAstNodeVisitor {

	ITranslationProvider translationProvider
	
	val sb = new StringBuilder
	var currentSB = sb
	
	new(ITranslationProvider translationProvider) {
		this.translationProvider = translationProvider
	}
	
	def String getTranslatedCode() {
		sb.toString
	}
	
	
	override visitBegin(DCAstNode node) {
		
		if (node instanceof DCAstBracketNode) {
			if (node.shouldCollectArguments) {
				currentSB = new StringBuilder
				node.linkedData = currentSB
			}
			else {
				currentSB.append(node.left)
			}
		}
		else {
			if (!node.skipOutput) {
				switch node {
					DCAstIdentifierNode: currentSB.append(node.id)
					DCAstPeriodNode: currentSB.append(".")
					DCAstWhitespaceNode: currentSB.append(node.text)
					DCAstOtherNode: currentSB.append(node.text)
				}
			}
		}
		
		return true
	}
	
	override visitEnd(DCAstNode node) {
		switch node {
			DCAstBracketNode: {
				if (node.linkedData!==null) {
					// leave the nested StringBuilder and switch to the enclosing (or top) one
					currentSB = node.enclosingSB
				}
				else if (node.closed) {
					currentSB.append(node.right)
				}
			}
			
			DCAstFeatureCallNode case node.canTranslate: node.translate
		}
	}
	
	private def canTranslate(DCAstFeatureCallNode node) {
		// if all even children are linked then we can translate the whole
		for (var i=0; i<node.children.size; i=i+2) {
			if (node.c(i).linkedObject===null) {
				return false
			}
		}
		
		// make sure the last child is not a period
		val lastNode = node.children.last
		if (lastNode instanceof DCAstPeriodNode) {
			return false
		}
		
		return true
	}
	
	private def translate(DCAstFeatureCallNode node) {
		val lastNode = node.children.last
		var linkedObject = lastNode.linkedObject
		
		// TODO: handle getter for data driven ports: port.message - i.e. no () after message
		
		if (lastNode instanceof DCAstOperationCallNode) {
			if (linkedObject instanceof Message) {
				handleSendMessage(lastNode, linkedObject)
				return
			}
		}
		
		node.handleCallChain
	}
	
	def handleCallChain(DCAstFeatureCallNode node) {
		var String instance = null
		for (var i=0; i<node.children.size; i=i+2) {
			val childNode = node.c(i)
			val linkedObject = childNode.linkedObject
			switch linkedObject {
				Attribute: {
					val index = if (childNode instanceof DCAstArrayAccessNode) {
						(childNode.bracketNode.linkedData as StringBuilder).toString
					}
					instance = translationProvider.getAttributeGetter(linkedObject, instance, index, childNode.originalText)
				}
				
				VarDecl: {
					instance = linkedObject.name
				}
				
				Operation: {
					val args = if (childNode instanceof DCAstOperationCallNode) {
						(childNode.bracketNode.linkedData as StringBuilder).toString
					}
					instance = translationProvider.getOperationText(linkedObject, instance, args, childNode.originalText)
				}
			}
		}
		currentSB.append(instance)
	}
	
	private def handleSendMessage(DCAstOperationCallNode msgNode, Message msg) {
		val args = (msgNode.bracketNode.linkedData as StringBuilder).toString
		// look back to interface item
		val ifItemNode = msgNode.parent.c(0)
		val ifItem = ifItemNode.linkedObject as InterfaceItem
		
		val translated =
			if (ifItemNode instanceof DCAstIdentifierNode) {
				// accessed as scalar (means broadcast for replicated port)
				val inst = ifItemNode.id
				translationProvider.getInterfaceItemMessageText(ifItem, msg, inst, args, null, msgNode.originalText)
			}
			else if (ifItemNode instanceof DCAstArrayAccessNode) {
				// accessed as array
				val inst = ifItemNode.id
				val index = (ifItemNode.bracketNode.linkedData as StringBuilder).toString
				translationProvider.getInterfaceItemMessageText(ifItem, msg, inst, args, index, msgNode.originalText)
			}
			else {
				// TODO: assert(false)
				"internal translator error"
			}
		currentSB.append(translated)
	}
	
	private def skipOutput(DCAstNode node) {
		val surroundingContext = node.getEnclosingBracketOrFeatureCall
		
		// check whether we are inside a bracket without a feature call
		if (surroundingContext instanceof DCAstBracketNode) {
			return false
		}
		
		if (surroundingContext instanceof DCAstFeatureCallNode) {
			return surroundingContext.canTranslate
		}
		
		return false
	}
	
	private def getEnclosingBracketOrFeatureCall(DCAstNode node) {
		var parent = node.parent
		while (parent!==null) {
			if (parent instanceof DCAstFeatureCallNode || parent instanceof DCAstBracketNode) {
				return parent
			}
			parent = parent.parent
		}
		
		return null
	}
	
	private def getOriginalText(DCAstNode node) {
		val printer = new DCAstPrinter
		node.visit(printer)
		return printer.text
	}
	
	private def getEnclosingSB(DCAstBracketNode node) {
		var parent = node.parent
		while (parent!==null) {
			if (parent instanceof DCAstBracketNode && parent.linkedData instanceof StringBuilder) {
				return parent.linkedData as StringBuilder
			}
			parent = parent.parent
		}
		
		// return top level
		return sb
	}
	
	private def shouldCollectArguments(DCAstBracketNode node) {
		val linkedObject = node.parent.linkedObject
		
		if (node.type==BracketType.CURLY) {
			false
		}
		else if (linkedObject instanceof Operation) {
			true
		}
		else if (linkedObject instanceof Message) {
			true
		}
		else if (linkedObject instanceof Attribute) {
			true
		}
		else if (linkedObject instanceof Port) {
			true
		}
		else {
			false
		}
	}
}

Back to the top