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.descriptors.impl;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.annotations.Nullable;
021    import org.jetbrains.kotlin.descriptors.*;
022    import org.jetbrains.kotlin.descriptors.annotations.Annotations;
023    import org.jetbrains.kotlin.name.Name;
024    import org.jetbrains.kotlin.types.JetType;
025    
026    import java.util.Collection;
027    import java.util.Collections;
028    import java.util.List;
029    
030    import static org.jetbrains.kotlin.resolve.descriptorUtil.DescriptorUtilPackage.getBuiltIns;
031    
032    public class PropertySetterDescriptorImpl extends PropertyAccessorDescriptorImpl implements PropertySetterDescriptor {
033    
034        private ValueParameterDescriptor parameter;
035        @NotNull
036        private final PropertySetterDescriptor original;
037    
038        public PropertySetterDescriptorImpl(
039                @NotNull PropertyDescriptor correspondingProperty,
040                @NotNull Annotations annotations,
041                @NotNull Modality modality,
042                @NotNull Visibility visibility,
043                boolean hasBody,
044                boolean isDefault,
045                @NotNull Kind kind,
046                @Nullable PropertySetterDescriptor original,
047                @NotNull SourceElement source
048        ) {
049            super(modality, visibility, correspondingProperty, annotations, Name.special("<set-" + correspondingProperty.getName() + ">"),
050                  hasBody, isDefault, kind, source);
051            this.original = original != null ? original : this;
052        }
053    
054        public void initialize(@NotNull ValueParameterDescriptor parameter) {
055            assert this.parameter == null;
056            this.parameter = parameter;
057        }
058    
059        public void initializeDefault() {
060            assert parameter == null;
061            parameter = createSetterParameter(this, getCorrespondingProperty().getReturnType());
062        }
063    
064        public static ValueParameterDescriptorImpl createSetterParameter(
065                @NotNull PropertySetterDescriptor setterDescriptor,
066                @NotNull JetType type
067        ) {
068            return new ValueParameterDescriptorImpl(
069                    setterDescriptor, null, 0, Annotations.EMPTY, Name.special("<set-?>"), type, false, null, SourceElement.NO_SOURCE
070            );
071        }
072    
073        @NotNull
074        @Override
075        @SuppressWarnings("unchecked")
076        public Collection<? extends PropertySetterDescriptor> getOverriddenDescriptors() {
077            return (Collection) super.getOverriddenDescriptors(false);
078        }
079    
080        @NotNull
081        @Override
082        public List<ValueParameterDescriptor> getValueParameters() {
083            if (parameter == null) {
084                throw new IllegalStateException();
085            }
086            return Collections.singletonList(parameter);
087        }
088    
089        @NotNull
090        @Override
091        public JetType getReturnType() {
092            return getBuiltIns(this).getUnitType();
093        }
094    
095        @Override
096        public <R, D> R accept(DeclarationDescriptorVisitor<R, D> visitor, D data) {
097            return visitor.visitPropertySetterDescriptor(this, data);
098        }
099    
100        @NotNull
101        @Override
102        public PropertySetterDescriptor getOriginal() {
103            return this.original;
104        }
105    }