bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/compilation/SCLCompiler.java

   1 package org.simantics.scl.compiler.compilation;
   2
   3 import java.io.IOException;
   4 import java.io.Reader;
   5 import java.util.ArrayList;
   6 import java.util.Map;
   7
   8 import org.simantics.scl.compiler.environment.EnvironmentFactory;
   9 import org.simantics.scl.compiler.errors.ErrorLog;
  10 import org.simantics.scl.compiler.internal.codegen.types.JavaReferenceValidatorFactory;
  11 import org.simantics.scl.compiler.internal.header.ModuleHeader;
  12 import org.simantics.scl.compiler.internal.parsing.declarations.DeclarationAst;
  13 import org.simantics.scl.compiler.internal.parsing.exceptions.SCLSyntaxErrorException;
  14 import org.simantics.scl.compiler.internal.parsing.parser.SCLParserImpl;
  15 import org.simantics.scl.compiler.internal.parsing.parser.SCLParserOptions;
  16 import org.simantics.scl.compiler.module.ConcreteModule;
  17 import org.simantics.scl.compiler.module.options.ModuleCompilationOptions;
  18 import org.simantics.scl.compiler.top.ModuleInitializer;
  19 import org.simantics.scl.compiler.top.SCLCompilerConfiguration;
  20 import org.simantics.scl.compiler.top.StandardModuleInitializer;
  21
  22 public class SCLCompiler {
  23     CompilationContext compilationContext = new CompilationContext();
  24     DeclarationClassification declarations = new DeclarationClassification(compilationContext);
  25
  26     // publishable results
  27     Map<String, byte[]> classes;
  28     ConcreteModule module;
  29     ModuleInitializer moduleInitializer;
  30
  31     private CompilationTimer timer;
  32     private ModuleCompilationOptions options;
  33
  34     JavaReferenceValidatorFactory jrvFactory;
  35
  36     public SCLCompiler(ModuleCompilationOptions options, JavaReferenceValidatorFactory jrvFactory) {
  37         this.options = options == null ? ModuleCompilationOptions.STANDARD_OPTIONS : options;
  38         this.jrvFactory = jrvFactory;
  39     }
  40
  41     @SuppressWarnings("unchecked")
  42     public void addSource(Reader sourceReader) {
  43         if(SCLCompilerConfiguration.ENABLE_TIMING) initializeTiming();
  44         try {
  45             SCLParserImpl parser = new SCLParserImpl(sourceReader);
  46             parser.setParserOptions(SCLParserOptions.MODULE_DEFAULT);
  47             parser.setCompilationContext(compilationContext);
  48             if(!parser.isEmpty())
  49             for(DeclarationAst declaration : (ArrayList<DeclarationAst>)parser.parseModule())
  50                 declarations.handle(declaration);
  51         } catch(SCLSyntaxErrorException e) {
  52             compilationContext.errorLog.log(e.location, e.getMessage());
  53         } catch(Exception e) {
  54             compilationContext.errorLog.log(e);
  55         } finally {
  56             try {
  57                 sourceReader.close();
  58             } catch (IOException e) {
  59                 e.printStackTrace();
  60             }
  61         }
  62         if(SCLCompilerConfiguration.ENABLE_TIMING) phaseFinished("Parsing");
  63     }
  64
  65     private boolean hasErrors() {
  66         return !compilationContext.errorLog.hasNoErrors();
  67     }
  68
  69     public void compile(
  70             EnvironmentFactory localEnvironmentFactory,
  71             String moduleName) {
  72         try {
  73             if(hasErrors()) return;
  74             Elaboration elaboration = new Elaboration(compilationContext,
  75                     timer,
  76                     localEnvironmentFactory,
  77                     moduleName,
  78                     compilationContext.header,
  79                     declarations.importsAst,
  80                     jrvFactory,
  81                     declarations.valueDefinitionsAst,
  82                     declarations.relationDefinitionsAst);
  83             if(options.computeCoverage)
  84                 elaboration.addCoverageBranchPoints();
  85             // Elaboration
  86             if(hasErrors()) return;
  87             elaboration.addTypesToEnvironment(
  88                     declarations.dataTypesAst,
  89                     declarations.typeAliasesAst,
  90                     declarations.effectsAst,
  91                     declarations.rulesetsAst);
  92             if(hasErrors()) return;
  93             elaboration.processTypeAliases(declarations.typeAliasesAst);
  94             if(hasErrors()) return;
  95             elaboration.processDataTypes(declarations.dataTypesAst);
  96             if(hasErrors()) return;
  97             elaboration.processTypeClasses(declarations.typeClassesAst);
  98             if(hasErrors()) return;
  99             elaboration.processDerivingInstances(declarations.derivingInstancesAst, declarations.instancesAst);
 100             if(hasErrors()) return;
 101             elaboration.processInstances(declarations.instancesAst);
 102             if(hasErrors()) return;
 103             elaboration.processJavaMethods(declarations.javaMethodDeclarations);
 104             if(hasErrors()) return;
 105             elaboration.processRulesets(declarations.rulesetsAst);
 106             if(hasErrors()) return;
 107             elaboration.addDataTypesToEnvironment();
 108             elaboration.addTypeClassesToEnvironment();
 109             elaboration.preprocessValueDefinitions(declarations.typeAnnotationsAst);
 110             elaboration.processMappingRelations(declarations.mappingRelationsAst);
 111             elaboration.addFixityToEnvironment(declarations.fixityAst);
 112             elaboration.addValueDefinitionsToEnvironment(declarations.typeAnnotationsAst);
 113             elaboration.processRules(declarations.rulesAst);
 114             elaboration.addSupplementedTypeAnnotationsToEnvironment();
 115             if(SCLCompilerConfiguration.ENABLE_TIMING) phaseFinished("Elaboration");
 116
 117             // Type checking
 118             if(hasErrors()) return;
 119             //new TypeChecking(errorLog, elaboration.environment, elaboration.module).typeCheck();
 120             new TypeChecking(compilationContext, elaboration.module).typeCheck();
 121
 122             if(SCLCompilerConfiguration.ENABLE_TIMING) phaseFinished("Type checking");
 123
 124             // FIXME HACK for testing
 125             /*if(!elaboration.module.getRules().isEmpty()) {
 126                 TransformationBuilder tb = new TransformationBuilder(elaboration.module.getRules());
 127                 tb.compileRules(elaboration.environment);
 128                 System.out.println(tb.getCompiledExpression(elaboration.environment));
 129             }*/
 130
 131             // Documentation generation
 132             new DocumentationGeneration(
 133                     declarations.valueDocumentation,
 134                     declarations.typeDocumentation,
 135                     declarations.classDocumentation,
 136                     elaboration.module
 137                     )
 138                 .generateDocumentation();
 139             this.declarations = null;
 140
 141             // Code generation
 142             if(hasErrors()) return;
 143             CodeGeneration codeGeneration = new CodeGeneration(
 144                     compilationContext,
 145                     elaboration.javaReferenceValidator,
 146                     elaboration.module);
 147             codeGeneration.simplifyValues();
 148             if(hasErrors()) return;
 149             codeGeneration.convertToSSA();
 150             if(hasErrors()) return;
 151             codeGeneration.optimizeSSA();
 152
 153             if(SCLCompilerConfiguration.ENABLE_TIMING) phaseFinished("SSA conversion and optimization");
 154
 155             if(hasErrors()) return;
 156             codeGeneration.generateCode();
 157             if(hasErrors()) return;
 158             codeGeneration.generateDataTypes(elaboration.dataTypes);
 159             if(hasErrors()) return;
 160             codeGeneration.generateTypeClasses();
 161             if(hasErrors()) return;
 162             codeGeneration.generateTypeClassInstances();
 163             if(hasErrors()) return;
 164
 165             classes = codeGeneration.classes;
 166             module =  codeGeneration.module;
 167             moduleInitializer = StandardModuleInitializer.create(
 168                     compilationContext.namingPolicy.getModuleClassName(),
 169                     codeGeneration.externalConstants);
 170
 171             module.setClasses(classes);
 172             module.setParentClassLoader(elaboration.javaReferenceValidator.getClassLoader());
 173             module.setModuleInitializer(moduleInitializer);
 174             module.setBranchPoints(elaboration.branchPoints);
 175             if(compilationContext.errorLog.hasErrorsOrWarnings())
 176                 module.setWarnings(compilationContext.errorLog.getErrors());
 177
 178             if(SCLCompilerConfiguration.ENABLE_TIMING) {
 179                 phaseFinished("Code generation");
 180                 reportTiming(moduleName);
 181             }
 182         } catch(Exception e) {
 183             compilationContext.errorLog.log(e);
 184         }
 185     }
 186
 187     public ErrorLog getErrorLog() {
 188         return compilationContext.errorLog;
 189     }
 190
 191     public Map<String, byte[]> getClasses() {
 192         return classes;
 193     }
 194
 195     public ConcreteModule getModule() {
 196         return module;
 197     }
 198
 199     public ModuleInitializer getModuleInitializer() {
 200         return moduleInitializer;
 201     }
 202
 203     private void initializeTiming() {
 204         timer = new CompilationTimer();
 205     }
 206
 207     private void phaseFinished(String phaseName) {
 208         timer.phaseFinished(phaseName);
 209     }
 210
 211     private void reportTiming(String moduleName) {
 212         timer.report(moduleName);
 213     }
 214 }