001/*
002 * Copyright 2010-2013 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
017package org.jetbrains.jet.codegen.intrinsics;
018
019import com.intellij.psi.PsiElement;
020import org.jetbrains.annotations.NotNull;
021import org.jetbrains.asm4.Type;
022import org.jetbrains.asm4.commons.InstructionAdapter;
023import org.jetbrains.jet.lang.resolve.java.AsmTypeConstants;
024import org.jetbrains.jet.codegen.ExpressionCodegen;
025import org.jetbrains.jet.codegen.StackValue;
026import org.jetbrains.jet.codegen.state.GenerationState;
027import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
028import org.jetbrains.jet.lang.descriptors.FunctionDescriptor;
029import org.jetbrains.jet.lang.psi.JetCallExpression;
030import org.jetbrains.jet.lang.psi.JetExpression;
031import org.jetbrains.jet.lang.psi.JetSimpleNameExpression;
032import org.jetbrains.jet.lang.resolve.BindingContext;
033import org.jetbrains.jet.lang.resolve.java.JvmPrimitiveType;
034import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns;
035import org.jetbrains.jet.lang.types.lang.PrimitiveType;
036
037import java.util.List;
038
039public class ArrayIterator implements IntrinsicMethod {
040    @Override
041    public StackValue generate(
042            ExpressionCodegen codegen,
043            InstructionAdapter v,
044            @NotNull Type expectedType,
045            PsiElement element,
046            List<JetExpression> arguments,
047            StackValue receiver,
048            @NotNull GenerationState state
049    ) {
050        receiver.put(receiver.type, v);
051        JetCallExpression call = (JetCallExpression) element;
052        FunctionDescriptor funDescriptor = (FunctionDescriptor) codegen.getBindingContext()
053                .get(BindingContext.REFERENCE_TARGET, (JetSimpleNameExpression) call.getCalleeExpression());
054        assert funDescriptor != null;
055        ClassDescriptor containingDeclaration = (ClassDescriptor) funDescriptor.getContainingDeclaration().getOriginal();
056        if (containingDeclaration.equals(KotlinBuiltIns.getInstance().getArray())) {
057            v.invokestatic("jet/runtime/ArrayIterator", "iterator", "([Ljava/lang/Object;)Ljava/util/Iterator;");
058            return StackValue.onStack(AsmTypeConstants.JET_ITERATOR_TYPE);
059        }
060        else {
061            for (JvmPrimitiveType jvmPrimitiveType : JvmPrimitiveType.values()) {
062                PrimitiveType primitiveType = jvmPrimitiveType.getPrimitiveType();
063                ClassDescriptor arrayClass = KotlinBuiltIns.getInstance().getPrimitiveArrayClassDescriptor(primitiveType);
064                if (containingDeclaration.equals(arrayClass)) {
065                    String methodSignature = "([" + jvmPrimitiveType.getJvmLetter() + ")" + jvmPrimitiveType.getIterator().getDescriptor();
066                    v.invokestatic("jet/runtime/ArrayIterator", "iterator", methodSignature);
067                    return StackValue.onStack(jvmPrimitiveType.getIterator().getAsmType());
068                }
069            }
070            throw new UnsupportedOperationException(containingDeclaration.toString());
071        }
072    }
073}