| OLD | NEW |
| 1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 library cps_ir.optimization.remove_refinements; | 5 library cps_ir.optimization.remove_refinements; |
| 6 | 6 |
| 7 import 'optimizers.dart' show Pass; | 7 import 'optimizers.dart' show Pass; |
| 8 import 'shrinking_reductions.dart' show ParentVisitor; | 8 import 'shrinking_reductions.dart' show ParentVisitor; |
| 9 import 'cps_ir_nodes.dart'; | 9 import 'cps_ir_nodes.dart'; |
| 10 | 10 |
| 11 /// Removes all [Refinement] nodes from the IR. | 11 /// Removes all [Refinement] nodes from the IR. |
| 12 /// | 12 /// |
| 13 /// This simplifies subsequent passes that don't rely on path-sensitive | 13 /// This simplifies subsequent passes that don't rely on path-sensitive |
| 14 /// type information but depend on equality between primitives. | 14 /// type information but depend on equality between primitives. |
| 15 class RemoveRefinements extends RecursiveVisitor implements Pass { | 15 class RemoveRefinements extends RecursiveVisitor implements Pass { |
| 16 String get passName => 'Remove refinement nodes'; | 16 String get passName => 'Remove refinement nodes'; |
| 17 | 17 |
| 18 void rewrite(FunctionDefinition node) { | 18 void rewrite(FunctionDefinition node) { |
| 19 new ParentVisitor().visit(node); | 19 new ParentVisitor().visit(node); |
| 20 visit(node); | 20 visit(node); |
| 21 } | 21 } |
| 22 | 22 |
| 23 @override | 23 @override |
| 24 Expression traverseLetPrim(LetPrim node) { | 24 Expression traverseLetPrim(LetPrim node) { |
| 25 Expression next = node.body; |
| 25 if (node.primitive is Refinement) { | 26 if (node.primitive is Refinement) { |
| 26 Refinement refinement = node.primitive; | 27 Refinement refinement = node.primitive; |
| 27 refinement.value.definition.substituteFor(refinement); | 28 refinement.value.definition.substituteFor(refinement); |
| 28 refinement.value.unlink(); | 29 refinement.destroy(); |
| 29 InteriorNode parent = node.parent; | 30 node.remove(); |
| 30 parent.body = node.body; | |
| 31 node.body.parent = parent; | |
| 32 } | 31 } |
| 33 return node.body; | 32 return next; |
| 34 } | 33 } |
| 35 } | 34 } |
| OLD | NEW |