From c5d8aae5152e7e3838dd7d609cc491ded0b30513 Mon Sep 17 00:00:00 2001
From: Maxim Shafirov <max@maxbook.labs.intellij.net>
Date: Fri, 25 Feb 2011 14:49:50 +0300
Subject: [PATCH] codegen, initial

---
 .idea/dictionaries/max.xml                    |   7 +
 .../jet/codegen/ClassVisitorFactory.java      |  17 ++
 .../jet/codegen/ExpressionCodegen.java        | 256 ++++++++++++++++++
 .../jet/codegen/FunctionCodegen.java          |  25 ++
 .../jetbrains/jet/codegen/MethodCodeGen.java  |   8 +
 .../jet/codegen/NamespaceCodegen.java         |  45 +++
 .../jet/codegen/PropertyCodegen.java          |  25 ++
 .../jet/lang/psi/JetConstantExpression.java   |  43 +++
 .../jetbrains/jet/lang/psi/JetNamespace.java  |   4 +
 .../jet/codegen/ExpressionGenTest.java        |  77 ++++++
 10 files changed, 507 insertions(+)
 create mode 100644 .idea/dictionaries/max.xml
 create mode 100644 idea/src/org/jetbrains/jet/codegen/ClassVisitorFactory.java
 create mode 100644 idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
 create mode 100644 idea/src/org/jetbrains/jet/codegen/FunctionCodegen.java
 create mode 100644 idea/src/org/jetbrains/jet/codegen/MethodCodeGen.java
 create mode 100644 idea/src/org/jetbrains/jet/codegen/NamespaceCodegen.java
 create mode 100644 idea/src/org/jetbrains/jet/codegen/PropertyCodegen.java
 create mode 100644 idea/tests/org/jetbrains/jet/codegen/ExpressionGenTest.java

diff --git a/.idea/dictionaries/max.xml b/.idea/dictionaries/max.xml
new file mode 100644
index 00000000000..59b317d1177
--- /dev/null
+++ b/.idea/dictionaries/max.xml
@@ -0,0 +1,7 @@
+<component name="ProjectDictionaryState">
+  <dictionary name="max">
+    <words>
+      <w>codegen</w>
+    </words>
+  </dictionary>
+</component>
\ No newline at end of file
diff --git a/idea/src/org/jetbrains/jet/codegen/ClassVisitorFactory.java b/idea/src/org/jetbrains/jet/codegen/ClassVisitorFactory.java
new file mode 100644
index 00000000000..40ff59f7c38
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/codegen/ClassVisitorFactory.java
@@ -0,0 +1,17 @@
+package org.jetbrains.jet.codegen;
+
+import org.jetbrains.jet.lang.psi.JetNamespace;
+import org.objectweb.asm.ClassVisitor;
+import org.objectweb.asm.util.TraceClassVisitor;
+
+import java.io.PrintWriter;
+import java.io.StringWriter;
+
+/**
+ * @author max
+ */
+public class ClassVisitorFactory {
+    public ClassVisitor visitorForClassIn(JetNamespace namespace) {
+        return new TraceClassVisitor(new PrintWriter(new StringWriter()));
+    }
+}
diff --git a/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java b/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
new file mode 100644
index 00000000000..7a928630a67
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
@@ -0,0 +1,256 @@
+package org.jetbrains.jet.codegen;
+
+import gnu.trove.TObjectIntHashMap;
+import org.jetbrains.jet.lang.psi.*;
+import org.objectweb.asm.Label;
+import org.objectweb.asm.MethodVisitor;
+import org.objectweb.asm.Type;
+import org.objectweb.asm.commons.InstructionAdapter;
+
+import java.util.List;
+import java.util.Stack;
+
+/**
+ * @author max
+ */
+public class ExpressionCodegen extends JetVisitor {
+    private final Stack<Label> myLoopStarts = new Stack<Label>();
+    private final Stack<Label> myLoopEnds = new Stack<Label>();
+
+    private final InstructionAdapter v;
+    private final TObjectIntHashMap<JetProperty> myVarIndex = new TObjectIntHashMap<JetProperty>();
+    private       int myMaxVarIndex = 0;
+
+    public ExpressionCodegen(MethodVisitor v) {
+        this.v = new InstructionAdapter(v);
+    }
+
+    private void gen(JetElement expr) {
+        if (expr == null) throw new CompilationException();
+        expr.accept(this);
+    }
+
+    @Override
+    public void visitExpression(JetExpression expression) {
+        throw new UnsupportedOperationException("Codegen for " + expression + " is not yet implemented");
+    }
+
+    @Override
+    public void visitParenthesizedExpression(JetParenthesizedExpression expression) {
+        gen(expression.getExpression());
+    }
+
+    @Override
+    public void visitAnnotatedExpression(JetAnnotatedExpression expression) {
+        gen(expression.getBaseExpression());
+    }
+
+    @Override
+    public void visitIfExpression(JetIfExpression expression) {
+        gen(expression.getCondition());
+        unboxBoolean();
+
+        JetExpression thenExpression = expression.getThen();
+        JetExpression elseExpression = expression.getElse();
+
+        if (thenExpression == null && elseExpression == null) {
+            throw new CompilationException();
+        }
+
+        if (thenExpression == null) {
+            generateSingleBranchIf(elseExpression, false);
+            return;
+        }
+
+        if (elseExpression == null) {
+            generateSingleBranchIf(thenExpression, true);
+            return;
+        }
+
+
+        Label elseLabel = new Label();
+        v.ifeq(elseLabel); // == 0, i.e. false
+
+        gen(thenExpression);
+
+        Label endLabel = new Label();
+        v.goTo(endLabel);
+        v.mark(elseLabel);
+
+        gen(elseExpression);
+
+        v.mark(endLabel);
+    }
+
+    @Override
+    public void visitWhileExpression(JetWhileExpression expression) {
+        Label condition = new Label();
+        myLoopStarts.push(condition);
+        v.mark(condition);
+
+        Label end = new Label();
+        myLoopEnds.push(end);
+
+        gen(expression.getCondition());
+        unboxBoolean();
+        v.ifeq(end);
+
+        genToUnit(expression.getBody());
+        v.goTo(condition);
+
+        v.mark(end);
+        myLoopEnds.pop();
+        myLoopStarts.pop();
+    }
+
+    @Override
+    public void visitDoWhileExpression(JetDoWhileExpression expression) {
+        Label condition = new Label();
+        v.mark(condition);
+        myLoopStarts.push(condition);
+
+        Label end = new Label();
+        myLoopEnds.push(end);
+
+        genToUnit(expression.getBody());
+
+        gen(expression.getCondition());
+        unboxBoolean();
+        v.ifne(condition);
+
+        v.mark(end);
+
+        myLoopEnds.pop();
+        myLoopStarts.pop();
+    }
+
+    @Override
+    public void visitBreakExpression(JetBreakExpression expression) {
+        String labelName = expression.getLabelName();
+
+        Label label = labelName == null ? myLoopEnds.peek() : null; // TODO:
+
+        v.goTo(label);
+    }
+
+    @Override
+    public void visitContinueExpression(JetContinueExpression expression) {
+        String labelName = expression.getLabelName();
+
+        Label label = labelName == null ? myLoopStarts.peek() : null; // TODO:
+
+        v.goTo(label);
+    }
+
+    private void unboxBoolean() {
+        v.invokevirtual("java/lang/Boolean", "booleanValue", "()Z");
+    }
+
+    private void generateSingleBranchIf(JetExpression expression, boolean inverse) {
+        Label endLabel = new Label();
+
+        if (inverse) {
+            v.ifeq(endLabel);
+        }
+        else {
+            v.ifne(endLabel);
+        }
+
+        genToUnit(expression);
+
+        v.mark(endLabel);
+    }
+
+    private void genToUnit(JetExpression expression) {
+        gen(expression);
+
+        if (!isUnitType(expression)) {
+            v.pop();
+        }
+    }
+
+    @Override
+    public void visitConstantExpression(JetConstantExpression expression) {
+        Object value = expression.getValue();
+        v.aconst(value);
+
+        if (value instanceof Integer) {
+            v.invokestatic("java/lang/Integer", "valueOf", "(I)Ljava/lang/Integer;");
+        }
+        else if (value instanceof Boolean) {
+            v.invokestatic("java/lang/Boolean", "valueOf", "(Z)Ljava/lang/Boolean;");
+        }
+        else if (value instanceof Character) {
+            v.invokestatic("java/lang/Character", "valueOf", "(C)Ljava/lang/Character;");
+        }
+        else if (value instanceof Short) {
+            v.invokestatic("java/lang/Short", "valueOf", "(S)Ljava/lang/Short;");
+        }
+        else if (value instanceof Long) {
+            v.invokestatic("java/lang/Long", "valueOf", "(J)Ljava/lang/Long;");
+        }
+        else if (value instanceof Byte) {
+            v.invokestatic("java/lang/Byte", "valueOf", "(B)Ljava/lang/Byte;");
+        }
+        else if (value instanceof Float) {
+            v.invokestatic("java/lang/Float", "valueOf", "(F)Ljava/lang/Float;");
+        }
+        else if (value instanceof Double) {
+            v.invokestatic("java/lang/Double", "valueOf", "(D)Ljava/lang/Double;");
+        }
+    }
+
+    @Override
+    public void visitBlockExpression(JetBlockExpression expression) {
+        Label blockStart = new Label();
+        v.mark(blockStart);
+
+        List<JetElement> statements = expression.getStatements();
+        for (JetElement statement : statements) {
+            if (statement instanceof JetProperty) {
+                myVarIndex.put((JetProperty) statement, myMaxVarIndex++);
+            }
+        }
+
+        for (JetElement statement : statements) {
+            gen(statement);
+        }
+
+        Label blockEnd = new Label();
+        v.mark(blockEnd);
+
+        for (JetElement statement : statements) {
+            if (statement instanceof JetProperty) {
+                JetProperty var = (JetProperty) statement;
+                int index = myVarIndex.remove(var);
+                v.visitLocalVariable(var.getName(), getType(var).getDescriptor(), null, blockStart, blockEnd, index);
+                myMaxVarIndex--;
+            }
+        }
+
+    }
+
+    private Type getType(JetProperty var) {
+        return InstructionAdapter.OBJECT_TYPE;  // TODO:
+    }
+
+    private boolean isUnitType(JetExpression e) {
+        return false; // TODO:!!!
+    }
+
+    @Override
+    public void visitProperty(JetProperty property) {
+        int index = myVarIndex.get(property);
+
+        assert index >= 0;
+
+        JetExpression initializer = property.getInitializer();
+        if (initializer != null) {
+            gen(initializer);
+            v.store(index, getType(property));
+        }
+    }
+
+    private static class CompilationException extends RuntimeException {
+    }
+}
diff --git a/idea/src/org/jetbrains/jet/codegen/FunctionCodegen.java b/idea/src/org/jetbrains/jet/codegen/FunctionCodegen.java
new file mode 100644
index 00000000000..773085e52fb
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/codegen/FunctionCodegen.java
@@ -0,0 +1,25 @@
+package org.jetbrains.jet.codegen;
+
+import org.jetbrains.jet.lang.psi.JetClass;
+import org.jetbrains.jet.lang.psi.JetFunction;
+import org.jetbrains.jet.lang.psi.JetNamespace;
+import org.objectweb.asm.ClassVisitor;
+
+/**
+ * @author max
+ */
+public class FunctionCodegen {
+    private final ClassVisitor v;
+
+    public FunctionCodegen(ClassVisitor v) {
+        this.v = v;
+    }
+
+    public void gen(JetFunction f, JetNamespace owner) {
+
+    }
+
+    public void gen(JetFunction f, JetClass owner) {
+
+    }
+}
diff --git a/idea/src/org/jetbrains/jet/codegen/MethodCodeGen.java b/idea/src/org/jetbrains/jet/codegen/MethodCodeGen.java
new file mode 100644
index 00000000000..d9d657ac822
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/codegen/MethodCodeGen.java
@@ -0,0 +1,8 @@
+package org.jetbrains.jet.codegen;
+
+/**
+ * @author max
+ */
+public class MethodCodeGen {
+
+}
diff --git a/idea/src/org/jetbrains/jet/codegen/NamespaceCodegen.java b/idea/src/org/jetbrains/jet/codegen/NamespaceCodegen.java
new file mode 100644
index 00000000000..894d6a47f76
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/codegen/NamespaceCodegen.java
@@ -0,0 +1,45 @@
+package org.jetbrains.jet.codegen;
+
+import org.jetbrains.jet.lang.psi.JetDeclaration;
+import org.jetbrains.jet.lang.psi.JetFunction;
+import org.jetbrains.jet.lang.psi.JetNamespace;
+import org.jetbrains.jet.lang.psi.JetProperty;
+import org.objectweb.asm.ClassVisitor;
+import org.objectweb.asm.Opcodes;
+
+/**
+ * @author max
+ */
+public class NamespaceCodegen {
+    private final ClassVisitorFactory factory;
+
+    public NamespaceCodegen(ClassVisitorFactory factory) {
+        this.factory = factory;
+    }
+
+    public void generate(JetNamespace namespace) {
+        final ClassVisitor v = factory.visitorForClassIn(namespace);
+
+        final PropertyCodegen propertyCodegen = new PropertyCodegen(v);
+        final FunctionCodegen functionCodegen = new FunctionCodegen(v);
+        v.visit(Opcodes.V1_6,
+                Opcodes.ACC_PUBLIC,
+                namespace.getFQName().replace('.', '/') + "/namespace",
+                null,
+                "jet/lang/Namespace",
+                new String[0]
+                );
+
+        for (JetDeclaration declaration : namespace.getDeclarations()) {
+            if (declaration instanceof JetProperty) {
+                propertyCodegen.gen((JetProperty) declaration, namespace);
+            }
+            else if (declaration instanceof JetFunction) {
+                functionCodegen.gen((JetFunction) declaration, namespace);
+            }
+        }
+
+
+        v.visitEnd();
+    }
+}
diff --git a/idea/src/org/jetbrains/jet/codegen/PropertyCodegen.java b/idea/src/org/jetbrains/jet/codegen/PropertyCodegen.java
new file mode 100644
index 00000000000..f94b378e9fb
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/codegen/PropertyCodegen.java
@@ -0,0 +1,25 @@
+package org.jetbrains.jet.codegen;
+
+import org.jetbrains.jet.lang.psi.JetClass;
+import org.jetbrains.jet.lang.psi.JetNamespace;
+import org.jetbrains.jet.lang.psi.JetProperty;
+import org.objectweb.asm.ClassVisitor;
+
+/**
+ * @author max
+ */
+public class PropertyCodegen {
+    private final ClassVisitor v;
+
+    public PropertyCodegen(ClassVisitor v) {
+        this.v = v;
+    }
+
+    public void gen(JetProperty p, JetNamespace owner) {
+
+    }
+
+    public void gen(JetProperty p, JetClass owner) {
+
+    }
+}
diff --git a/idea/src/org/jetbrains/jet/lang/psi/JetConstantExpression.java b/idea/src/org/jetbrains/jet/lang/psi/JetConstantExpression.java
index cb4d1e0ce9f..41bc1fa5557 100644
--- a/idea/src/org/jetbrains/jet/lang/psi/JetConstantExpression.java
+++ b/idea/src/org/jetbrains/jet/lang/psi/JetConstantExpression.java
@@ -1,7 +1,9 @@
 package org.jetbrains.jet.lang.psi;
 
 import com.intellij.lang.ASTNode;
+import com.intellij.psi.tree.IElementType;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.JetNodeTypes;
 
 /**
  * @author max
@@ -15,4 +17,45 @@ public class JetConstantExpression extends JetExpression {
     public void accept(JetVisitor visitor) {
         visitor.visitConstantExpression(this);
     }
+
+    public Object getValue() {
+        IElementType elementType = getNode().getElementType();
+        String nodeText = getNode().getText();
+        if (elementType == JetNodeTypes.INTEGER_CONSTANT) {
+            return Integer.parseInt(nodeText);
+        }
+        else if (elementType == JetNodeTypes.LONG_CONSTANT) {
+            return Long.parseLong(nodeText);
+        }
+        else if (elementType == JetNodeTypes.FLOAT_CONSTANT) {
+            assert nodeText.length() > 0;
+            char lastChar = nodeText.charAt(nodeText.length() - 1);
+            if (lastChar == 'f' || lastChar == 'F') {
+                return Float.parseFloat(nodeText.substring(0, nodeText.length() - 1));
+            }
+            else if (lastChar == 'd' || lastChar == 'D') {
+                return Double.parseDouble(nodeText.substring(0, nodeText.length() - 1));
+            }
+            else {
+                return Double.parseDouble(nodeText);
+            }
+        }
+        else if (elementType == JetNodeTypes.BOOLEAN_CONSTANT) {
+            return Boolean.parseBoolean(nodeText);
+        }
+        else if (elementType == JetNodeTypes.CHARACTER_CONSTANT) {
+            return nodeText.charAt(1);
+        }
+        else if (elementType == JetNodeTypes.STRING_CONSTANT) {
+            int tail = nodeText.length();
+            if (nodeText.endsWith("\"")) tail--;
+            return nodeText.substring(1, tail);
+        }
+        else if (elementType == JetNodeTypes.NULL) {
+            return null;
+        }
+        else {
+            throw new IllegalArgumentException("Unsupported constant: " + this);
+        }
+    }
 }
diff --git a/idea/src/org/jetbrains/jet/lang/psi/JetNamespace.java b/idea/src/org/jetbrains/jet/lang/psi/JetNamespace.java
index cdb1d626c60..75fb1051c83 100644
--- a/idea/src/org/jetbrains/jet/lang/psi/JetNamespace.java
+++ b/idea/src/org/jetbrains/jet/lang/psi/JetNamespace.java
@@ -35,4 +35,8 @@ public class JetNamespace extends JetDeclaration {
         PsiElement nameNode = findChildByType(JetNodeTypes.NAMESPACE_NAME);
         return nameNode != null ? nameNode.getText() : "";
     }
+
+    public String getFQName() {
+        return getName(); // TODO: Must include container namespace names, module root namespace
+    }
 }
diff --git a/idea/tests/org/jetbrains/jet/codegen/ExpressionGenTest.java b/idea/tests/org/jetbrains/jet/codegen/ExpressionGenTest.java
new file mode 100644
index 00000000000..a78b024860b
--- /dev/null
+++ b/idea/tests/org/jetbrains/jet/codegen/ExpressionGenTest.java
@@ -0,0 +1,77 @@
+package org.jetbrains.jet.codegen;
+
+import com.intellij.codeInsight.daemon.LightDaemonAnalyzerTestCase;
+import com.intellij.psi.util.PsiTreeUtil;
+import org.jetbrains.jet.lang.psi.JetProperty;
+import org.objectweb.asm.util.TraceMethodVisitor;
+
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.io.StringWriter;
+
+/**
+ * @author max
+ */
+public class ExpressionGenTest extends LightDaemonAnalyzerTestCase {
+    public void testIntegerZeroExpression() throws Exception {
+        checkCode("0",
+                "LDC 0\n" +
+                "INVOKESTATIC java/lang/Integer.valueOf (I)Ljava/lang/Integer;");
+    }
+
+    public void testIf() throws Exception {
+        checkCode("if (false) 15 else 20",
+                "LDC false\n" +
+                "INVOKESTATIC java/lang/Boolean.valueOf (Z)Ljava/lang/Boolean;\n" +
+                "INVOKEVIRTUAL java/lang/Boolean.booleanValue ()Z\n" +
+                "IFEQ L0\n" +
+                "LDC 15\n" +
+                "INVOKESTATIC java/lang/Integer.valueOf (I)Ljava/lang/Integer;\n" +
+                "GOTO L1\n" +
+                "L0\n" +
+                "LDC 20\n" +
+                "INVOKESTATIC java/lang/Integer.valueOf (I)Ljava/lang/Integer;\n" +
+                "L1");
+
+        checkCode("if (false) 15",
+                "LDC false\n" +
+                "INVOKESTATIC java/lang/Boolean.valueOf (Z)Ljava/lang/Boolean;\n" +
+                "INVOKEVIRTUAL java/lang/Boolean.booleanValue ()Z\n" +
+                "IFEQ L0\n" +
+                "LDC 15\n" +
+                "INVOKESTATIC java/lang/Integer.valueOf (I)Ljava/lang/Integer;\n" +
+                "POP\n" +
+                "L0");
+
+        checkCode("if (false) else 20",
+                "LDC false\n" +
+                "INVOKESTATIC java/lang/Boolean.valueOf (Z)Ljava/lang/Boolean;\n" +
+                "INVOKEVIRTUAL java/lang/Boolean.booleanValue ()Z\n" +
+                "IFNE L0\n" +
+                "LDC 20\n" +
+                "INVOKESTATIC java/lang/Integer.valueOf (I)Ljava/lang/Integer;\n" +
+                "POP\n" +
+                "L0");
+    }
+
+    private void checkCode(String expr, String expectation) throws IOException {
+        configureFromFileText("test.jet", "val x = " + expr);
+
+        JetProperty p = PsiTreeUtil.getParentOfType(getFile().findElementAt(0), JetProperty.class);
+
+        TraceMethodVisitor trace = new TraceMethodVisitor();
+        p.getInitializer().accept(new ExpressionCodegen(trace));
+
+        StringWriter out = new StringWriter();
+        trace.print(new PrintWriter(out));
+        assertEquals(trimLines(expectation), trimLines(out.getBuffer().toString()));
+    }
+
+    private static String trimLines(String s) {
+        StringBuilder b = new StringBuilder();
+        for (String line : s.split("\\n")) {
+            b.append(line.trim()).append('\n');
+        }
+        return b.toString();
+    }
+}