Spaces:
Running
Running
File size: 1,291 Bytes
8f3f8db |
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 |
inlining
exitVar: exitVar label: exitLabel
"Replace each return statement in this method with an assignment to the exit variable followed by a goto to the given label. Return true if a goto was generated."
"Optimization: If exitVar is nil, the return value of the inlined method is not being used, so don't add the assignment statement."
| newStmts labelUsed |
labelUsed := false.
parseTree nodesDo: [ :node |
node isStmtList ifTrue: [
newStmts := OrderedCollection new: 100.
node statements do: [ :stmt |
(stmt isReturn) ifTrue: [
exitVar = nil ifTrue: [
stmt expression isLeaf ifFalse: [
"evaluate return expression even though value isn't used"
newStmts add: stmt expression.
].
] ifFalse: [
"assign return expression to exit variable"
newStmts add:
(TAssignmentNode new
setVariable: (TVariableNode new setName: exitVar)
expression: stmt expression).
].
(stmt == parseTree statements last) ifFalse: [
"generate a goto (this return is NOT the last statement in the method)"
newStmts add: (TGoToNode new setLabel: exitLabel).
labelUsed := true.
].
] ifFalse: [
newStmts addLast: stmt.
].
].
node setStatements: newStmts asArray.
].
].
^labelUsed |