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.codegen.inline;
018    
019    import java.util.HashMap;
020    import java.util.Map;
021    
022    public class NameGenerator {
023    
024        private final String ownerMethod;
025    
026        private int nextLambdaIndex = 1;
027        private int nextWhenIndex = 1;
028    
029        private final Map<String, NameGenerator> subGenerators = new HashMap<String, NameGenerator>();
030    
031        public NameGenerator(String onwerMethod) {
032            this.ownerMethod = onwerMethod;
033        }
034    
035        public String genLambdaClassName() {
036            return ownerMethod + "$" + nextLambdaIndex++;
037        }
038    
039        public String genWhenClassNamePrefix() {
040            return ownerMethod + "$" + nextWhenIndex++;
041        }
042    
043        public NameGenerator subGenerator(String inliningMethod) {
044            NameGenerator generator = subGenerators.get(inliningMethod);
045            if (generator == null) {
046                generator = new NameGenerator(ownerMethod+ "$" + inliningMethod);
047                subGenerators.put(inliningMethod, generator);
048            }
049            return generator;
050        }
051    }