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
017 package org.jetbrains.jet.lang.psi.stubs.elements;
018
019 import com.intellij.lang.ASTNode;
020 import com.intellij.psi.PsiElement;
021 import com.intellij.psi.stubs.IndexSink;
022 import com.intellij.psi.stubs.StubElement;
023 import com.intellij.psi.stubs.StubInputStream;
024 import com.intellij.psi.stubs.StubOutputStream;
025 import com.intellij.util.io.StringRef;
026 import org.jetbrains.annotations.NonNls;
027 import org.jetbrains.annotations.NotNull;
028 import org.jetbrains.jet.lang.psi.JetExpression;
029 import org.jetbrains.jet.lang.psi.JetParameter;
030 import org.jetbrains.jet.lang.psi.JetTypeReference;
031 import org.jetbrains.jet.lang.psi.stubs.PsiJetParameterStub;
032 import org.jetbrains.jet.lang.psi.stubs.impl.PsiJetParameterStubImpl;
033 import org.jetbrains.jet.lang.resolve.name.FqName;
034
035 import java.io.IOException;
036
037 public class JetParameterElementType extends JetStubElementType<PsiJetParameterStub, JetParameter> {
038 public JetParameterElementType(@NotNull @NonNls String debugName) {
039 super(debugName);
040 }
041
042 @Override
043 public JetParameter createPsiFromAst(@NotNull ASTNode node) {
044 return new JetParameter(node);
045 }
046
047 @Override
048 public JetParameter createPsi(@NotNull PsiJetParameterStub stub) {
049 return new JetParameter(stub);
050 }
051
052 @Override
053 public PsiJetParameterStub createStub(@NotNull JetParameter psi, StubElement parentStub) {
054 JetTypeReference typeReference = psi.getTypeReference();
055 JetExpression defaultValue = psi.getDefaultValue();
056
057 return new PsiJetParameterStubImpl(parentStub, psi.getFqName(),
058 psi.getName(), psi.isMutable(), psi.isVarArg(),
059 typeReference != null ? typeReference.getText() : null,
060 defaultValue != null ? defaultValue.getText() : null);
061 }
062
063 @Override
064 public boolean shouldCreateStub(ASTNode node) {
065 if (!super.shouldCreateStub(node)) {
066 return false;
067 }
068 PsiElement psi = node.getPsi();
069 return psi instanceof JetParameter && !((JetParameter) psi).isLoopParameter();
070 }
071
072 @Override
073 public void serialize(@NotNull PsiJetParameterStub stub, @NotNull StubOutputStream dataStream) throws IOException {
074 dataStream.writeName(stub.getName());
075 dataStream.writeBoolean(stub.isMutable());
076 dataStream.writeBoolean(stub.isVarArg());
077 dataStream.writeName(stub.getTypeText());
078 dataStream.writeName(stub.getDefaultValueText());
079 FqName name = stub.getFqName();
080 dataStream.writeName(name != null ? name.asString() : null);
081 }
082
083 @NotNull
084 @Override
085 public PsiJetParameterStub deserialize(@NotNull StubInputStream dataStream, StubElement parentStub) throws IOException {
086 StringRef name = dataStream.readName();
087 boolean isMutable = dataStream.readBoolean();
088 boolean isVarArg = dataStream.readBoolean();
089 StringRef typeText = dataStream.readName();
090 StringRef defaultValueText = dataStream.readName();
091 StringRef fqNameAsString = dataStream.readName();
092 FqName fqName = fqNameAsString != null ? new FqName(fqNameAsString.toString()) : null;
093
094 return new PsiJetParameterStubImpl(parentStub, fqName, name, isMutable, isVarArg,
095 typeText, defaultValueText);
096 }
097
098 @Override
099 public void indexStub(@NotNull PsiJetParameterStub stub, @NotNull IndexSink sink) {
100 // No index
101 }
102 }