From df9bd0308642f973c4990bcc8bb19b7b133c7413 Mon Sep 17 00:00:00 2001
From: Roman Shevchenko <roman.shevchenko@jetbrains.com>
Date: Wed, 13 Jul 2016 19:01:17 +0200
Subject: [PATCH] [java] stub-based module lookup

---
 .../src/com/intellij/psi/impl/source/PsiJavaFileImpl.java   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiJavaFileImpl.java b/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiJavaFileImpl.java
index c6956a5715a4..f38b9673640e 100644
--- a/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiJavaFileImpl.java
+++ b/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiJavaFileImpl.java
@@ -19,6 +19,7 @@ import com.intellij.ide.highlighter.JavaFileType;
 import com.intellij.openapi.fileTypes.FileType;
 import com.intellij.psi.*;
 import com.intellij.psi.impl.java.stubs.JavaStubElementTypes;
+import com.intellij.psi.impl.java.stubs.PsiJavaFileStub;
 import com.intellij.psi.util.PsiTreeUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -37,6 +38,11 @@ public class PsiJavaFileImpl extends PsiJavaFileBaseImpl {
   @Nullable
   @Override
   public PsiJavaModule getModuleDeclaration() {
+    PsiJavaFileStub stub = (PsiJavaFileStub)getStub();
+    if (stub != null) {
+      return stub.getModule();
+    }
+
     PsiElement element = getFirstChild();
     if (element instanceof PsiWhiteSpace || element instanceof PsiComment) {
       element = PsiTreeUtil.skipSiblingsForward(element, PsiWhiteSpace.class, PsiComment.class);