From 01a08ecd040482586b42b8029f25c0650b4c7795 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Thu, 30 Jun 2011 17:18:16 +0200
Subject: [PATCH] use hard-coded TypeInfo instances for primitive types

---
 .../jet/codegen/ExpressionCodegen.java        | 20 ++++++++++++++++++-
 .../codegen/typeInfo/primitiveTypeInfo.jet    | 13 ++++++++++++
 .../jet/codegen/CodegenTestCase.java          |  3 +++
 .../jetbrains/jet/codegen/TypeInfoTest.java   |  4 ++++
 stdlib/src/jet/typeinfo/TypeInfo.java         |  9 +++++++++
 5 files changed, 48 insertions(+), 1 deletion(-)
 create mode 100644 idea/testData/codegen/typeInfo/primitiveTypeInfo.jet

diff --git a/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java b/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
index 6606dc7bcbf..443422e4b2e 100644
--- a/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
+++ b/idea/src/org/jetbrains/jet/codegen/ExpressionCodegen.java
@@ -61,6 +61,18 @@ public class ExpressionCodegen extends JetVisitor {
     private final Stack<Label> myBreakTargets = new Stack<Label>();
     private final Stack<StackValue> myStack = new Stack<StackValue>();
 
+    private static final String[] PRIMITIVE_TYPE_INFO_FIELDS = {
+            null,
+            "BOOL_TYPE_INFO",
+            "CHAR_TYPE_INFO",
+            "BYTE_TYPE_INFO",
+            "SHORT_TYPE_INFO",
+            "INT_TYPE_INFO",
+            "FLOAT_TYPE_INFO",
+            "LONG_TYPE_INFO",
+            "DOUBLE_TYPE_INFO"
+    };
+
     private final InstructionAdapter v;
     private final FrameMap myMap;
     private final JetTypeMapper typeMapper;
@@ -1703,9 +1715,15 @@ public class ExpressionCodegen extends JetVisitor {
             throw new UnsupportedOperationException("don't know what this type parameter resolves to");
         }
 
+        final Type jvmType = typeMapper.mapType(jetType, OwnerKind.INTERFACE);
+        if (jvmType.getSort() <= Type.DOUBLE) {
+            v.getstatic("jet/typeinfo/TypeInfo", PRIMITIVE_TYPE_INFO_FIELDS[jvmType.getSort()], "Ljet/typeinfo/TypeInfo;");
+            return;
+        }
+
         v.anew(JetTypeMapper.TYPE_TYPEINFO);
         v.dup();
-        v.aconst(typeMapper.jvmType((ClassDescriptor) declarationDescriptor, OwnerKind.INTERFACE));
+        v.aconst(jvmType);
         List<TypeProjection> arguments = jetType.getArguments();
         if (arguments.size() > 0) {
             v.iconst(arguments.size());
diff --git a/idea/testData/codegen/typeInfo/primitiveTypeInfo.jet b/idea/testData/codegen/typeInfo/primitiveTypeInfo.jet
new file mode 100644
index 00000000000..fa7f1356aba
--- /dev/null
+++ b/idea/testData/codegen/typeInfo/primitiveTypeInfo.jet
@@ -0,0 +1,13 @@
+class Box<T>(t: T) {
+    var value = t
+}
+
+fun isIntBox(box: Box<out Any?>): Boolean {
+    return box is Box<Int>;
+}
+
+
+fun box(): String {
+  val box = Box<Int>(1)
+  return if (isIntBox(box)) "OK" else "fail"
+}
diff --git a/idea/tests/org/jetbrains/jet/codegen/CodegenTestCase.java b/idea/tests/org/jetbrains/jet/codegen/CodegenTestCase.java
index bb8aed70251..54d0a7947c9 100644
--- a/idea/tests/org/jetbrains/jet/codegen/CodegenTestCase.java
+++ b/idea/tests/org/jetbrains/jet/codegen/CodegenTestCase.java
@@ -75,6 +75,9 @@ public abstract class CodegenTestCase extends LightCodeInsightFixtureTestCase {
             System.out.println(generateToText());
             throw new RuntimeException(e);
         }
+        if (!"OK".equals(actual)) {
+            System.out.println(generateToText());
+        }
         assertEquals("OK", actual);
     }
 
diff --git a/idea/tests/org/jetbrains/jet/codegen/TypeInfoTest.java b/idea/tests/org/jetbrains/jet/codegen/TypeInfoTest.java
index c8d0daea62b..4e831cd2984 100644
--- a/idea/tests/org/jetbrains/jet/codegen/TypeInfoTest.java
+++ b/idea/tests/org/jetbrains/jet/codegen/TypeInfoTest.java
@@ -84,6 +84,10 @@ public class TypeInfoTest extends CodegenTestCase {
         generateFunction();  // assert no exception
     }
 
+    public void testPrimitiveTypeInfo() throws Exception {
+        blackBoxFile("typeInfo/primitiveTypeInfo.jet");
+    }
+
     private Runnable newRunnable() {
         return new Runnable() {
             @Override
diff --git a/stdlib/src/jet/typeinfo/TypeInfo.java b/stdlib/src/jet/typeinfo/TypeInfo.java
index e29ec82bc22..4ff4f9c0c17 100644
--- a/stdlib/src/jet/typeinfo/TypeInfo.java
+++ b/stdlib/src/jet/typeinfo/TypeInfo.java
@@ -79,4 +79,13 @@ public class TypeInfo<T> implements JetObject {
         result = 31 * result + (typeParameters != null ? Arrays.hashCode(typeParameters) : 0);
         return result;
     }
+
+    public static final TypeInfo<Byte> BYTE_TYPE_INFO = new TypeInfo<Byte>(Byte.class);
+    public static final TypeInfo<Short> SHORT_TYPE_INFO = new TypeInfo<Short>(Short.class);
+    public static final TypeInfo<Integer> INT_TYPE_INFO = new TypeInfo<Integer>(Integer.class);
+    public static final TypeInfo<Long> LONG_TYPE_INFO = new TypeInfo<Long>(Long.class);
+    public static final TypeInfo<Character> CHAR_TYPE_INFO = new TypeInfo<Character>(Character.class);
+    public static final TypeInfo<Boolean> BOOL_TYPE_INFO = new TypeInfo<Boolean>(Boolean.class);
+    public static final TypeInfo<Float> FLOAT_TYPE_INFO = new TypeInfo<Float>(Float.class);
+    public static final TypeInfo<Double> DOUBLE_TYPE_INFO = new TypeInfo<Double>(Double.class);
 }