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.cli.jvm.compiler;
018    
019    import com.google.common.collect.Maps;
020    import kotlin.modules.Module;
021    import org.jetbrains.annotations.NotNull;
022    import org.jetbrains.annotations.Nullable;
023    
024    import java.io.File;
025    import java.util.Collections;
026    import java.util.List;
027    import java.util.Map;
028    
029    public class ModuleChunk {
030    
031        public static final ModuleChunk EMPTY = new ModuleChunk(Collections.<Module>emptyList());
032    
033        private final List<Module> modules;
034        private final Map<File, Module> sourceFileToModule = Maps.newHashMap();
035    
036        public ModuleChunk(@NotNull List<Module> modules) {
037            this.modules = modules;
038            for (Module module : modules) {
039                for (String file : module.getSourceFiles()) {
040                    sourceFileToModule.put(new File(file).getAbsoluteFile(), module);
041                }
042            }
043        }
044    
045        @NotNull
046        public List<Module> getModules() {
047            return modules;
048        }
049    
050        @Nullable
051        public Module findModuleBySourceFile(@NotNull File sourceFile) {
052            return sourceFileToModule.get(sourceFile.getAbsoluteFile());
053        }
054    }