001 /*
002 * Copyright 2010-2015 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.kotlin.codegen.intrinsics;
018
019 import com.google.common.collect.ImmutableList;
020 import kotlin.text.StringsKt;
021 import org.jetbrains.annotations.NotNull;
022 import org.jetbrains.annotations.Nullable;
023 import org.jetbrains.kotlin.builtins.KotlinBuiltIns;
024 import org.jetbrains.kotlin.builtins.PrimitiveType;
025 import org.jetbrains.kotlin.descriptors.CallableMemberDescriptor;
026 import org.jetbrains.kotlin.name.FqName;
027 import org.jetbrains.kotlin.name.FqNameUnsafe;
028 import org.jetbrains.kotlin.name.Name;
029 import org.jetbrains.kotlin.resolve.jvm.JvmPrimitiveType;
030 import org.jetbrains.kotlin.types.expressions.OperatorConventions;
031
032 import static org.jetbrains.kotlin.builtins.KotlinBuiltIns.*;
033 import static org.jetbrains.org.objectweb.asm.Opcodes.*;
034
035 public class IntrinsicMethods {
036 public static final String INTRINSICS_CLASS_NAME = "kotlin/jvm/internal/Intrinsics";
037
038 private static final FqName KOTLIN_JVM = new FqName("kotlin.jvm");
039 /* package */ static final FqNameUnsafe RECEIVER_PARAMETER_FQ_NAME = new FqNameUnsafe("T");
040
041 private static final IntrinsicMethod UNARY_MINUS = new UnaryMinus();
042 private static final IntrinsicMethod UNARY_PLUS = new UnaryPlus();
043 private static final IntrinsicMethod NUMBER_CAST = new NumberCast();
044 private static final IntrinsicMethod INV = new Inv();
045 private static final IntrinsicMethod RANGE_TO = new RangeTo();
046 private static final IntrinsicMethod INC = new Increment(1);
047 private static final IntrinsicMethod DEC = new Increment(-1);
048 private static final IntrinsicMethod HASH_CODE = new HashCode();
049
050 private static final IntrinsicMethod ARRAY_SIZE = new ArraySize();
051 private static final Equals EQUALS = new Equals();
052 private static final IteratorNext ITERATOR_NEXT = new IteratorNext();
053 private static final ArraySet ARRAY_SET = new ArraySet();
054 private static final ArrayGet ARRAY_GET = new ArrayGet();
055 private static final StringPlus STRING_PLUS = new StringPlus();
056 private static final ToString TO_STRING = new ToString();
057 private static final Clone CLONE = new Clone();
058
059 private static final IntrinsicMethod ARRAY_ITERATOR = new ArrayIterator();
060 private final IntrinsicsMap intrinsicsMap = new IntrinsicsMap();
061
062 public IntrinsicMethods() {
063 intrinsicsMap.registerIntrinsic(KOTLIN_JVM, RECEIVER_PARAMETER_FQ_NAME, "javaClass", -1, new JavaClassProperty());
064 intrinsicsMap.registerIntrinsic(KOTLIN_JVM, KotlinBuiltIns.FQ_NAMES.kClass, "java", -1, new KClassJavaProperty());
065 intrinsicsMap.registerIntrinsic(new FqName("kotlin.jvm.internal.unsafe"), null, "monitorEnter", 1, MonitorInstruction.MONITOR_ENTER);
066 intrinsicsMap.registerIntrinsic(new FqName("kotlin.jvm.internal.unsafe"), null, "monitorExit", 1, MonitorInstruction.MONITOR_EXIT);
067 intrinsicsMap.registerIntrinsic(KOTLIN_JVM, KotlinBuiltIns.FQ_NAMES.array, "isArrayOf", 0, new IsArrayOf());
068
069 intrinsicsMap.registerIntrinsic(BUILT_INS_PACKAGE_FQ_NAME, null, "arrayOf", 1, new ArrayOf());
070
071 ImmutableList<Name> primitiveCastMethods = OperatorConventions.NUMBER_CONVERSIONS.asList();
072 for (Name method : primitiveCastMethods) {
073 String methodName = method.asString();
074 declareIntrinsicFunction(FQ_NAMES.number, methodName, 0, NUMBER_CAST);
075 for (PrimitiveType type : PrimitiveType.NUMBER_TYPES) {
076 declareIntrinsicFunction(type.getTypeFqName(), methodName, 0, NUMBER_CAST);
077 }
078 }
079
080 for (PrimitiveType type : PrimitiveType.NUMBER_TYPES) {
081 FqName typeFqName = type.getTypeFqName();
082 declareIntrinsicFunction(typeFqName, "plus", 0, UNARY_PLUS);
083 declareIntrinsicFunction(typeFqName, "unaryPlus", 0, UNARY_PLUS);
084 declareIntrinsicFunction(typeFqName, "minus", 0, UNARY_MINUS);
085 declareIntrinsicFunction(typeFqName, "unaryMinus", 0, UNARY_MINUS);
086 declareIntrinsicFunction(typeFqName, "inv", 0, INV);
087 declareIntrinsicFunction(typeFqName, "rangeTo", 1, RANGE_TO);
088 declareIntrinsicFunction(typeFqName, "inc", 0, INC);
089 declareIntrinsicFunction(typeFqName, "dec", 0, DEC);
090 }
091
092 for (PrimitiveType type : PrimitiveType.values()) {
093 FqName typeFqName = type.getTypeFqName();
094 declareIntrinsicFunction(typeFqName, "equals", 1, EQUALS);
095 declareIntrinsicFunction(typeFqName, "hashCode", 0, HASH_CODE);
096 declareIntrinsicFunction(typeFqName, "toString", 0, TO_STRING);
097
098 intrinsicsMap.registerIntrinsic(
099 BUILT_INS_PACKAGE_FQ_NAME, null, StringsKt.decapitalize(type.getArrayTypeName().asString()) + "Of", 1, new ArrayOf()
100 );
101 }
102
103 declareBinaryOp("plus", IADD);
104 declareBinaryOp("minus", ISUB);
105 declareBinaryOp("times", IMUL);
106 declareBinaryOp("div", IDIV);
107 declareBinaryOp("mod", IREM);
108 declareBinaryOp("shl", ISHL);
109 declareBinaryOp("shr", ISHR);
110 declareBinaryOp("ushr", IUSHR);
111 declareBinaryOp("and", IAND);
112 declareBinaryOp("or", IOR);
113 declareBinaryOp("xor", IXOR);
114
115 declareIntrinsicFunction(FQ_NAMES._boolean, "not", 0, new Not());
116
117 declareIntrinsicFunction(FQ_NAMES.string, "plus", 1, new Concat());
118 declareIntrinsicFunction(FQ_NAMES.string, "get", 1, new StringGetChar());
119
120 declareIntrinsicFunction(FQ_NAMES.cloneable, "clone", 0, CLONE);
121
122 intrinsicsMap.registerIntrinsic(BUILT_INS_PACKAGE_FQ_NAME, KotlinBuiltIns.FQ_NAMES.any, "toString", 0, TO_STRING);
123 intrinsicsMap.registerIntrinsic(BUILT_INS_PACKAGE_FQ_NAME, KotlinBuiltIns.FQ_NAMES.string, "plus", 1, STRING_PLUS);
124 intrinsicsMap.registerIntrinsic(BUILT_INS_PACKAGE_FQ_NAME, null, "arrayOfNulls", 1, new NewArray());
125
126 for (PrimitiveType type : PrimitiveType.values()) {
127 declareIntrinsicFunction(type.getTypeFqName(), "compareTo", 1, new CompareTo());
128 declareIntrinsicFunction(COLLECTIONS_PACKAGE_FQ_NAME.child(Name.identifier(type.getTypeName().asString() + "Iterator")), "next", 0, ITERATOR_NEXT);
129 }
130
131 declareArrayMethods();
132 }
133
134 private void declareArrayMethods() {
135 for (JvmPrimitiveType jvmPrimitiveType : JvmPrimitiveType.values()) {
136 declareArrayMethods(jvmPrimitiveType.getPrimitiveType().getArrayTypeFqName());
137 }
138 declareArrayMethods(FQ_NAMES.array.toSafe());
139 }
140
141 private void declareArrayMethods(@NotNull FqName arrayTypeFqName) {
142 declareIntrinsicFunction(arrayTypeFqName, "size", -1, ARRAY_SIZE);
143 declareIntrinsicFunction(arrayTypeFqName, "set", 2, ARRAY_SET);
144 declareIntrinsicFunction(arrayTypeFqName, "get", 1, ARRAY_GET);
145 declareIntrinsicFunction(arrayTypeFqName, "clone", 0, CLONE);
146 declareIntrinsicFunction(arrayTypeFqName, "iterator", 0, ARRAY_ITERATOR);
147 declareIntrinsicFunction(arrayTypeFqName, "<init>", 2, ArrayConstructor.INSTANCE);
148 }
149
150 private void declareBinaryOp(@NotNull String methodName, int opcode) {
151 BinaryOp op = new BinaryOp(opcode);
152 for (PrimitiveType type : PrimitiveType.values()) {
153 declareIntrinsicFunction(type.getTypeFqName(), methodName, 1, op);
154 }
155 }
156
157 private void declareIntrinsicFunction(
158 @NotNull FqName classFqName,
159 @NotNull String methodName,
160 int arity,
161 @NotNull IntrinsicMethod implementation
162 ) {
163 intrinsicsMap.registerIntrinsic(classFqName, null, methodName, arity, implementation);
164 }
165
166 private void declareIntrinsicFunction(
167 @NotNull FqNameUnsafe classFqName,
168 @NotNull String methodName,
169 int arity,
170 @NotNull IntrinsicMethod implementation
171 ) {
172 intrinsicsMap.registerIntrinsic(classFqName.toSafe(), null, methodName, arity, implementation);
173 }
174
175 @Nullable
176 public IntrinsicMethod getIntrinsic(@NotNull CallableMemberDescriptor descriptor) {
177 return intrinsicsMap.getIntrinsic(descriptor);
178 }
179 }