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.resolve.kotlin;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
021    import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
022    import org.jetbrains.jet.lang.descriptors.PackageFragmentDescriptor;
023    import org.jetbrains.jet.lang.resolve.java.JvmAbi;
024    import org.jetbrains.jet.lang.resolve.name.FqName;
025    import org.jetbrains.jet.lang.resolve.name.FqNameUnsafe;
026    import org.jetbrains.jet.lang.resolve.name.Name;
027    import org.jetbrains.jet.lang.resolve.name.SpecialNames;
028    
029    import java.util.ArrayList;
030    import java.util.List;
031    
032    import static org.jetbrains.jet.lang.resolve.name.SpecialNames.isClassObjectName;
033    
034    public class DeserializedResolverUtils {
035        private DeserializedResolverUtils() {
036        }
037    
038        @NotNull
039        public static FqName kotlinFqNameToJavaFqName(@NotNull FqNameUnsafe kotlinFqName) {
040            List<Name> segments = kotlinFqName.pathSegments();
041            List<String> correctedSegments = new ArrayList<String>(segments.size());
042            for (Name segment : segments) {
043                correctedSegments.add(isClassObjectName(segment) ? JvmAbi.CLASS_OBJECT_CLASS_NAME : segment.getIdentifier());
044            }
045            return FqName.fromSegments(correctedSegments);
046        }
047    
048        @NotNull
049        public static FqNameUnsafe javaFqNameToKotlinFqName(@NotNull FqName javaFqName) {
050            if (javaFqName.isRoot()) {
051                return javaFqName.toUnsafe();
052            }
053            List<Name> segments = javaFqName.pathSegments();
054            List<Name> correctedSegments = new ArrayList<Name>(segments.size());
055            correctedSegments.add(segments.get(0));
056            for (int i = 1; i < segments.size(); i++) {
057                Name segment = segments.get(i);
058                boolean isClassObjectName = segment.asString().equals(JvmAbi.CLASS_OBJECT_CLASS_NAME);
059                Name correctedSegment = isClassObjectName ? SpecialNames.getClassObjectName(segments.get(i - 1)) : segment;
060                correctedSegments.add(correctedSegment);
061            }
062            return FqNameUnsafe.fromSegments(correctedSegments);
063        }
064    
065        @NotNull
066        public static FqNameUnsafe naiveKotlinFqName(@NotNull ClassDescriptor descriptor) {
067            DeclarationDescriptor containing = descriptor.getContainingDeclaration();
068            if (containing instanceof ClassDescriptor) {
069                return naiveKotlinFqName((ClassDescriptor) containing).child(descriptor.getName());
070            }
071            else if (containing instanceof PackageFragmentDescriptor) {
072                return ((PackageFragmentDescriptor) containing).getFqName().child(descriptor.getName()).toUnsafe();
073            }
074            else {
075                throw new IllegalArgumentException("Class doesn't have a FQ name: " + descriptor);
076            }
077        }
078    }