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.asJava;
018    
019    import com.intellij.lang.Language;
020    import com.intellij.psi.*;
021    import com.intellij.psi.impl.light.LightElement;
022    import org.jetbrains.annotations.NotNull;
023    
024    import java.util.ArrayList;
025    import java.util.List;
026    
027    // Copy of com.intellij.psi.impl.light.LightParameterListBuilder
028    public class LightParameterListBuilder extends LightElement implements PsiParameterList {
029        private final List<PsiParameter> myParameters = new ArrayList<PsiParameter>();
030        private final KtLightMethod parent;
031        private PsiParameter[] myCachedParameters;
032    
033        public LightParameterListBuilder(PsiManager manager, Language language, KtLightMethod parent) {
034            super(manager, language);
035            this.parent = parent;
036        }
037    
038        public void addParameter(PsiParameter parameter) {
039            myParameters.add(parameter);
040            myCachedParameters = null;
041        }
042    
043        @Override
044        public KtLightMethod getParent() {
045            return parent;
046        }
047    
048        @Override
049        public String toString() {
050            return "Light parameter list";
051        }
052    
053        @NotNull
054        @Override
055        public PsiParameter[] getParameters() {
056            if (myCachedParameters == null) {
057                if (myParameters.isEmpty()) {
058                    myCachedParameters = PsiParameter.EMPTY_ARRAY;
059                }
060                else {
061                    myCachedParameters = myParameters.toArray(new PsiParameter[myParameters.size()]);
062                }
063            }
064    
065            return myCachedParameters;
066        }
067    
068        @Override
069        public int getParameterIndex(PsiParameter parameter) {
070            return myParameters.indexOf(parameter);
071        }
072    
073        @Override
074        public int getParametersCount() {
075            return myParameters.size();
076        }
077    
078        @Override
079        public void accept(@NotNull PsiElementVisitor visitor) {
080            if (visitor instanceof JavaElementVisitor) {
081                ((JavaElementVisitor) visitor).visitParameterList(this);
082            }
083        }
084    }