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.psi.stubs.elements;
018
019 import com.intellij.psi.stubs.IndexSink;
020 import com.intellij.psi.stubs.StubElement;
021 import com.intellij.psi.stubs.StubInputStream;
022 import com.intellij.psi.stubs.StubOutputStream;
023 import com.intellij.util.io.StringRef;
024 import org.jetbrains.annotations.NonNls;
025 import org.jetbrains.annotations.NotNull;
026 import org.jetbrains.kotlin.name.FqName;
027 import org.jetbrains.kotlin.psi.KtObjectDeclaration;
028 import org.jetbrains.kotlin.psi.psiUtil.KtPsiUtilKt;
029 import org.jetbrains.kotlin.psi.stubs.KotlinObjectStub;
030 import org.jetbrains.kotlin.psi.stubs.impl.KotlinObjectStubImpl;
031 import org.jetbrains.kotlin.psi.stubs.impl.Utils;
032 import org.jetbrains.kotlin.resolve.lazy.ResolveSessionUtils;
033
034 import java.io.IOException;
035 import java.util.List;
036
037 public class KtObjectElementType extends KtStubElementType<KotlinObjectStub, KtObjectDeclaration> {
038 public KtObjectElementType(@NotNull @NonNls String debugName) {
039 super(debugName, KtObjectDeclaration.class, KotlinObjectStub.class);
040 }
041
042 @Override
043 public KotlinObjectStub createStub(@NotNull KtObjectDeclaration psi, StubElement parentStub) {
044 String name = psi.getName();
045 FqName fqName = ResolveSessionUtils.safeFqNameForLazyResolve(psi);
046 List<String> superNames = KtPsiUtilKt.getSuperNames(psi);
047 return new KotlinObjectStubImpl(parentStub, StringRef.fromString(name), fqName, Utils.INSTANCE.wrapStrings(superNames),
048 psi.isTopLevel(), psi.isCompanion(), psi.isLocal(), psi.isObjectLiteral());
049 }
050
051 @Override
052 public void serialize(@NotNull KotlinObjectStub stub, @NotNull StubOutputStream dataStream) throws IOException {
053 dataStream.writeName(stub.getName());
054
055 FqName fqName = stub.getFqName();
056 dataStream.writeName(fqName != null ? fqName.toString() : null);
057
058 dataStream.writeBoolean(stub.isTopLevel());
059 dataStream.writeBoolean(stub.isCompanion());
060 dataStream.writeBoolean(stub.isLocal());
061 dataStream.writeBoolean(stub.isObjectLiteral());
062
063 List<String> superNames = stub.getSuperNames();
064 dataStream.writeVarInt(superNames.size());
065 for (String name : superNames) {
066 dataStream.writeName(name);
067 }
068 }
069
070 @NotNull
071 @Override
072 public KotlinObjectStub deserialize(@NotNull StubInputStream dataStream, StubElement parentStub) throws IOException {
073 StringRef name = dataStream.readName();
074 StringRef fqNameStr = dataStream.readName();
075 FqName fqName = fqNameStr != null ? new FqName(fqNameStr.toString()) : null;
076
077 boolean isTopLevel = dataStream.readBoolean();
078 boolean isCompanion = dataStream.readBoolean();
079 boolean isLocal = dataStream.readBoolean();
080 boolean isObjectLiteral = dataStream.readBoolean();
081
082 int superCount = dataStream.readVarInt();
083 StringRef[] superNames = StringRef.createArray(superCount);
084 for (int i = 0; i < superCount; i++) {
085 superNames[i] = dataStream.readName();
086 }
087
088 return new KotlinObjectStubImpl(parentStub, name, fqName, superNames, isTopLevel, isCompanion, isLocal, isObjectLiteral);
089 }
090
091 @Override
092 public void indexStub(@NotNull KotlinObjectStub stub, @NotNull IndexSink sink) {
093 StubIndexService.getInstance().indexObject(stub, sink);
094 }
095 }