Reifier hookup

lib/src/codegen/dart_codegen.dart

 // Copyright (c) 2015, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library dev_compiler.src.codegen.dart_codegen;
 
 import 'dart:io' show File;
 
 import 'package:analyzer/analyzer.dart' as analyzer;
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/java_core.dart' as java_core;
 import 'package:analyzer/src/generated/scanner.dart' show Token;
 import 'package:dart_style/dart_style.dart';
 import 'package:logging/logging.dart' as logger;
 import 'package:path/path.dart' as path;
 
 import 'package:dev_compiler/src/info.dart';
 import 'package:dev_compiler/src/checker/rules.dart';
 import 'package:dev_compiler/src/options.dart';
 import 'package:dev_compiler/src/utils.dart' as utils;
 import 'ast_builder.dart';
 import 'code_generator.dart' as codegenerator;
-import 'reify_coercions.dart' show CoercionReifier, NewTypeIdDesc;
+import 'reify_coercions.dart'
+    show CoercionReifier, NewTypeIdDesc, InstrumentedRuntime;
 
 final _log = new logger.Logger('dev_compiler.dart_codegen');
 
-class DevCompilerRuntime {
+class DevCompilerRuntime extends InstrumentedRuntime {
   Identifier _runtimeId = AstBuilder.identifierFromString("DEVC\$RT");
 
   Identifier _castId;
   Identifier _typeToTypeId;
   Identifier _wrapId;
 
   DevCompilerRuntime() {
     _castId = _prefixId(AstBuilder.identifierFromString("cast"));
     _typeToTypeId = _prefixId(AstBuilder.identifierFromString("type"));
     _wrapId = _prefixId(AstBuilder.identifierFromString("wrap"));
@@ skipping 14 matching lines @@
     if (oper.operation == "type") return _typeToTypeId;
     assert(false);
     return null;
   }
 
   Expression runtimeOperation(RuntimeOperation oper) {
     var id = runtimeId(oper);
     var args = oper.arguments;
     return AstBuilder.application(id, args);
   }
+
+  @override
+  Expression wrap(Expression coercion, Expression e, Expression fromType,
+      Expression toType, Expression dartIs, String kind, String location) {
+    var k = AstBuilder.stringLiteral(kind);
+    var key = AstBuilder.multiLineStringLiteral(location);
+    var arguments = <Expression>[coercion, e, fromType, toType, k, key, dartIs];
+    return new RuntimeOperation("wrap", arguments);
+  }
+
+  Expression cast(Expression e, Expression fromType, Expression toType,
+      Expression dartIs, String kind, String location, bool ground) {
+    var k = AstBuilder.stringLiteral(kind);
+    var key = AstBuilder.multiLineStringLiteral(location);
+    var g = AstBuilder.booleanLiteral(ground);
+    var arguments = <Expression>[e, fromType, toType, k, key, dartIs, g];
+    return new RuntimeOperation("cast", arguments);
+  }
+
+  Expression type(Expression witnessFunction) {
+    return new RuntimeOperation("type", <Expression>[witnessFunction]);
+  }
 }
 
 // TODO(leafp) This is kind of a hack, but it works for now.
 class FileWriter extends java_core.PrintStringWriter {
   final CompilerOptions options;
   String _path;
   FileWriter(this.options, this._path);
   int indent = 0;
   int withinInterpolationExpression = 0;
   bool insideForLoop = false;
@@ skipping 151 matching lines @@
   }
 }
 
 // TODO(leafp) Not sure if this is the right way to generate
 // Dart source going forward, but it's a quick way to get started.
 class UnitGenerator extends UnitGeneratorCommon with ConversionVisitor<Object> {
   CompilationUnit unit;
   final java_core.PrintWriter _out;
   final String outDir;
   Set<LibraryElement> _extraImports;
-  final _runtime = new DevCompilerRuntime();
+  final _runtime;
   bool _qualifyNames = true;
   Map<Identifier, NewTypeIdDesc> _newIdentifiers;
 
   UnitGenerator(this.unit, java_core.PrintWriter out, String this.outDir,
-      this._extraImports, this._newIdentifiers)
+      this._extraImports, this._newIdentifiers, this._runtime)
       : _out = out,
         super(out);
 
   void output(String s) => _out.print(s);
   void outputln(String s) => _out.println(s);
 
   // Choose a canonical prefix for a library that we are adding.
   // Currently just chooses something unlikely to conflict with a user
   // prefix.
   // TODO(leafp): Make this robust.
@@ skipping 107 matching lines @@
   }
 
   void generate() {
     visitCompilationUnit(unit);
   }
 }
 
 class DartGenerator extends codegenerator.CodeGenerator {
   final CompilerOptions options;
   TypeRules _rules;
+  final DevCompilerRuntime _runtime = new DevCompilerRuntime();
 
   DartGenerator(String outDir, Uri root, TypeRules rules, this.options)
       : _rules = rules,
         super(outDir, root, rules);
 
   Set<LibraryElement> computeExtraImports(Map<Identifier, NewTypeIdDesc> ids) {
     var imports = new Set<LibraryElement>();
     void process(Identifier id, NewTypeIdDesc desc) {
       if (_identifierNeedsQualification(id, desc)) {
         var library = desc.importedFrom;
         if (utils.isDartPrivateLibrary(library)) {
           _log.severe("Dropping import of private library ${library}\n");
           return;
         }
         imports.add(library);
       }
     }
     ids.forEach(process);
     return imports;
   }
 
   String generateLibrary(LibraryUnit library, LibraryInfo info) {
-    var r = new CoercionReifier(library, rules);
+    var r = new CoercionReifier(library, rules, options, _runtime);
     var ids = r.reify();
     var extraImports = computeExtraImports(ids);
 
     for (var unit in library.partsThenLibrary) {
       var libraryDir = makeOutputDirectory(info, unit);
       var uri = unit.element.source.uri;
       _log.fine("Generating unit $uri");
       FileWriter out = new FileWriter(
           options, path.join(libraryDir, '${uri.pathSegments.last}'));
-      var unitGen = new UnitGenerator(unit, out, outDir, extraImports, ids);
+      var unitGen =
+          new UnitGenerator(unit, out, outDir, extraImports, ids, _runtime);
       unitGen.generate();
       out.finalize();
     }
 
     return null;
   }
 }
 
 class EmptyUnitGenerator extends UnitGeneratorCommon {
   final java_core.PrintWriter _out;
@@ skipping 29 matching lines @@
   void generateUnit(CompilationUnit unit, LibraryInfo info, String libraryDir) {
     var uri = unit.element.source.uri;
     _log.fine("Emitting original unit " + uri.toString());
     FileWriter out = new FileWriter(
         options, path.join(libraryDir, '${uri.pathSegments.last}'));
     var unitGen = new EmptyUnitGenerator(unit, out);
     unitGen.generate();
     out.finalize();
   }
 }