Optimize top level class searching in LazyJavaPackageScope

Do not try to search something, that is known not to exist

compiler/cli/src/org/jetbrains/kotlin/cli/jvm/compiler/JvmDependenciesIndex.kt

@@ -20,9 +20,7 @@ import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.util.containers.IntArrayList
 import org.jetbrains.kotlin.name.ClassId
 import org.jetbrains.kotlin.name.FqName
-import java.util.ArrayList
-import java.util.EnumSet
-import java.util.HashMap
+import java.util.*
 
 data class JavaRoot(val file: VirtualFile, val type: JavaRoot.RootType, val prefixFqName: FqName? = null) {
     enum class RootType {
@@ -99,6 +97,24 @@ class JvmDependenciesIndex(_roots: List<JavaRoot>) {
         }
     }
 
+    fun collectKnownClassNamesInPackage(
+            packageFqName: FqName
+    ): Set<String> {
+        var result = hashSetOf<String>()
+        traverseDirectoriesInPackage(packageFqName, continueSearch = {
+            dir, rootType ->
+
+            for (child in dir.children) {
+                if (child.extension != "class" && child.extension != "java") continue
+                result.add(child.nameWithoutExtension)
+            }
+
+            true
+        })
+
+        return result
+    }
+
     private data class HandleResult<T : Any>(val result: T?, val continueSearch: Boolean)
 
     private fun <T : Any> search(

compiler/cli/src/org/jetbrains/kotlin/cli/jvm/compiler/KotlinCliJavaFileManagerImpl.kt

@@ -120,6 +120,8 @@ class KotlinCliJavaFileManagerImpl(private val myPsiManager: PsiManager)
         return findClassInPsiFile(classNameWithInnerClasses, file)
     }
 
+    override fun knownClassNamesInPackage(packageFqName: FqName) = index.collectKnownClassNamesInPackage(packageFqName)
+
     companion object {
         private val LOG = Logger.getInstance(KotlinCliJavaFileManagerImpl::class.java)
 

compiler/frontend.java/src/org/jetbrains/kotlin/load/java/JavaClassFinderImpl.java

@@ -36,6 +36,7 @@ import org.jetbrains.kotlin.resolve.jvm.KotlinJavaPsiFacade;
 
 import javax.annotation.PostConstruct;
 import javax.inject.Inject;
+import java.util.Set;
 
 public class JavaClassFinderImpl implements JavaClassFinder {
     private Project project;
@@ -108,4 +109,10 @@ public class JavaClassFinderImpl implements JavaClassFinder {
         PsiPackage psiPackage = javaFacade.findPackage(fqName.asString(), javaSearchScope);
         return psiPackage == null ? null : new JavaPackageImpl(psiPackage, javaSearchScope);
     }
+
+    @Nullable
+    @Override
+    public Set<String> knownClassNamesInPackage(@NotNull FqName packageFqName) {
+        return javaFacade.knownClassNamesInPackage(packageFqName);
+    }
 }

compiler/frontend.java/src/org/jetbrains/kotlin/resolve/jvm/KotlinCliJavaFileManager.kt

@@ -20,7 +20,10 @@ import com.intellij.psi.PsiClass
 import com.intellij.psi.impl.file.impl.JavaFileManager
 import com.intellij.psi.search.GlobalSearchScope
 import org.jetbrains.kotlin.name.ClassId
+import org.jetbrains.kotlin.name.FqName
 
 interface KotlinCliJavaFileManager : JavaFileManager {
     fun findClass(classId: ClassId, searchScope: GlobalSearchScope): PsiClass?
+
+    fun knownClassNamesInPackage(packageFqName: FqName): Set<String>?
 }

compiler/frontend.java/src/org/jetbrains/kotlin/resolve/jvm/KotlinJavaPsiFacade.java

@@ -40,12 +40,15 @@ import kotlin.collections.ArraysKt;
 import kotlin.collections.CollectionsKt;
 import kotlin.jvm.functions.Function1;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.jetbrains.kotlin.name.FqName;
 import org.jetbrains.kotlin.progress.ProgressIndicatorAndCompilationCanceledStatus;
 import org.jetbrains.kotlin.name.ClassId;
 
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Set;
 import java.util.concurrent.ConcurrentMap;
 
 public class KotlinJavaPsiFacade {
@@ -112,6 +115,17 @@ public class KotlinJavaPsiFacade {
         return null;
     }
 
+    @Nullable
+    public Set<String> knownClassNamesInPackage(@NotNull FqName packageFqName) {
+        KotlinPsiElementFinderWrapper[] finders = finders();
+
+        if (finders.length == 1) {
+            return ((KotlinPsiElementFinderImpl) finders[0]).knownClassNamesInPackage(packageFqName);
+        }
+
+        return null;
+    }
+
     @NotNull
     private PsiClass[] findClassesInDumbMode(@NotNull String qualifiedName, @NotNull GlobalSearchScope scope) {
         String packageName = StringUtil.getPackageName(qualifiedName);
@@ -328,6 +342,15 @@ public class KotlinJavaPsiFacade {
             return findClass(classId.asSingleFqName().asString(), scope);
         }
 
+        @Nullable
+        public Set<String> knownClassNamesInPackage(@NotNull FqName packageFqName) {
+            if (isCliFileManager) {
+                return ((KotlinCliJavaFileManager) javaFileManager).knownClassNamesInPackage(packageFqName);
+            }
+
+            return null;
+        }
+
         @Override
         public PsiPackage findPackage(@NotNull String qualifiedName, @NotNull GlobalSearchScope scope) {
             if (isCliFileManager) {