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.lang.resolve.lazy.data;
018
019import org.jetbrains.annotations.NotNull;
020import org.jetbrains.jet.lang.descriptors.ClassKind;
021import org.jetbrains.jet.lang.psi.*;
022
023import java.util.List;
024
025public class JetClassInfo extends JetClassOrObjectInfo<JetClass> {
026
027    @NotNull
028    private final ClassKind kind;
029
030    protected JetClassInfo(@NotNull JetClass classOrObject) {
031        super(classOrObject);
032        if (element instanceof JetEnumEntry) {
033            this.kind = ClassKind.ENUM_ENTRY;
034        }
035        else if (element.isTrait()) {
036            this.kind = ClassKind.TRAIT;
037        }
038        else if (element.isAnnotation()) {
039            this.kind = ClassKind.ANNOTATION_CLASS;
040        }
041        else if (element.isEnum()) {
042            this.kind = ClassKind.ENUM_CLASS;
043        }
044        else {
045            this.kind = ClassKind.CLASS;
046        }
047    }
048
049    @Override
050    public JetClassObject getClassObject() {
051        return element.getClassObject();
052    }
053
054    @NotNull
055    @Override
056    public List<JetTypeParameter> getTypeParameters() {
057        return element.getTypeParameters();
058    }
059
060    @NotNull
061    @Override
062    public List<? extends JetParameter> getPrimaryConstructorParameters() {
063        return element.getPrimaryConstructorParameters();
064    }
065
066    @NotNull
067    @Override
068    public ClassKind getClassKind() {
069        return kind;
070    }
071}