From 706b844eb35fe8b0dc6cbacca457df3683d964c8 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Fri, 15 Apr 2011 13:44:31 +0200
Subject: [PATCH] working array read, non-working array write

---
 .../jet/codegen/ExpressionCodegen.java        | 43 ++++++++++++++++++-
 .../jet/codegen/NamespaceGenTest.java         | 14 ++++++
 2 files changed, 56 insertions(+), 1 deletion(-)

diff --git a/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java b/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
index 9808125e3c8..3e481a3bb57 100644
--- a/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
+++ b/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
@@ -682,7 +682,21 @@ public class ExpressionCodegen extends JetVisitor {
     }
 
     private void generateAssignmentExpression(JetBinaryExpression expression) {
-        if (expression.getLeft() instanceof JetReferenceExpression) {
+        if (expression.getLeft() instanceof JetArrayAccessExpression) {
+            final JetArrayAccessExpression arrayAccess = (JetArrayAccessExpression) expression.getLeft();
+            final Type arrayType = expressionType(arrayAccess.getArrayExpression());
+            if (arrayType.getSort() == Type.ARRAY) {
+                gen(arrayAccess.getArrayExpression(), arrayType);
+                generateArrayIndex(arrayAccess);
+                final Type elementType = arrayType.getElementType();
+                gen(expression.getRight(), elementType);
+                v.astore(elementType);
+            }
+            else {
+                throw new UnsupportedOperationException("Don't know how to generate assignment to non-Java array " + expression.getLeft().getText());
+            }
+        }
+        else if (expression.getLeft() instanceof JetReferenceExpression) {
             final JetReferenceExpression lhs = (JetReferenceExpression) expression.getLeft();
             final int index = indexOfLocal(lhs);
             final Type type = typeMapper.mapType(bindingContext.getExpressionType(lhs));
@@ -882,6 +896,33 @@ public class ExpressionCodegen extends JetVisitor {
         throw new UnsupportedOperationException("don't know how to generate this new expression");
     }
 
+    @Override
+    public void visitArrayAccessExpression(JetArrayAccessExpression expression) {
+        final JetExpression array = expression.getArrayExpression();
+        final Type arrayType = expressionType(array);
+        if (arrayType.getSort() == Type.ARRAY) {
+            gen(array, arrayType);
+            generateArrayIndex(expression);
+            final Type elementType = arrayType.getElementType();
+            v.aload(elementType);
+            myStack.push(StackValue.onStack(elementType));
+        }
+        else {
+            throw new UnsupportedOperationException("array access to non-Java arrays is not supported");
+        }
+    }
+
+    private void generateArrayIndex(JetArrayAccessExpression expression) {
+        final List<JetExpression> indices = expression.getIndexExpressions();
+        if (indices.size() != 1) {
+            throw new UnsupportedOperationException("array access with more than one index is not supported");
+        }
+        if (!expressionType(indices.get(0)).equals(Type.INT_TYPE)) {
+            throw new UnsupportedOperationException("array access with non-integer is not supported");
+        }
+        gen(indices.get(0), Type.INT_TYPE);
+    }
+
     private static class CompilationException extends RuntimeException {
     }
 }
diff --git a/idea/tests/org/jetbrains/jet/codegen/NamespaceGenTest.java b/idea/tests/org/jetbrains/jet/codegen/NamespaceGenTest.java
index 5bae6bab106..7e8e395fb82 100644
--- a/idea/tests/org/jetbrains/jet/codegen/NamespaceGenTest.java
+++ b/idea/tests/org/jetbrains/jet/codegen/NamespaceGenTest.java
@@ -290,4 +290,18 @@ public class NamespaceGenTest extends CodegenTestCase {
         main.invoke(null, c);
         assertEquals(239, c.gridx);
     }
+
+    public void testArrayRead() throws Exception {
+        loadText("fun foo(c: Array<String>) = c[0]");
+        final Method main = generateFunction();
+        assertEquals("main", main.invoke(null, new Object[] { new String[] { "main" } }));
+    }
+
+    public void testArrayWrite() throws Exception {
+        loadText("fun foo(c: Array<String>) { c[0] = \"jet\"; }");
+        final Method main = generateFunction();
+        String[] array = new String[] { null };
+        main.invoke(null, new Object[] { array });
+        assertEquals("jet", array[0]);
+    }
 }