From d31ff7ba504d805067ff90d5dfa56747c558beeb Mon Sep 17 00:00:00 2001
From: Peter Gromov <peter@jetbrains.com>
Date: Tue, 15 Sep 2009 20:29:48 +0400
Subject: [PATCH] CompileContextEx.assignModule()

---
 .../impl/src/com/intellij/compiler/impl/CompileContextExProxy.java    | 4 ++++
 .../impl/src/com/intellij/compiler/impl/CompileContextImpl.java       | 2 +-
 .../impl/src/com/intellij/openapi/compiler/ex/CompileContextEx.java   | 3 +++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextExProxy.java b/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextExProxy.java
index 4a0939da73..b59a148988 100644
--- a/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextExProxy.java
+++ b/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextExProxy.java
@@ -161,4 +161,8 @@ public class CompileContextExProxy implements CompileContextEx {
   public void commitZip(String outputDir) throws IOException {
     myDelegate.commitZip(outputDir);
   }
+
+  public void assignModule(@NotNull VirtualFile root, @NotNull Module module, boolean isTestSource) {
+    myDelegate.assignModule(root, module, isTestSource);
+  }
 }
diff --git a/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextImpl.java b/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextImpl.java
index d7ae492730..09e26573ff 100644
--- a/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextImpl.java
+++ b/java/compiler/impl/src/com/intellij/compiler/impl/CompileContextImpl.java
@@ -286,7 +286,7 @@ public class CompileContextImpl extends UserDataHolderBase implements CompileCon
     return myTask.getIndicator();
   }
 
-  public void assignModule(VirtualFile root, Module module, final boolean isTestSource) {
+  public void assignModule(@NotNull VirtualFile root, @NotNull Module module, final boolean isTestSource) {
     try {
       myRootToModuleMap.put(root, module);
       Set<VirtualFile> set = myModuleToRootsMap.get(module);
diff --git a/java/compiler/impl/src/com/intellij/openapi/compiler/ex/CompileContextEx.java b/java/compiler/impl/src/com/intellij/openapi/compiler/ex/CompileContextEx.java
index 9ff31a704e..de89762d9a 100644
--- a/java/compiler/impl/src/com/intellij/openapi/compiler/ex/CompileContextEx.java
+++ b/java/compiler/impl/src/com/intellij/openapi/compiler/ex/CompileContextEx.java
@@ -5,6 +5,7 @@ import com.intellij.openapi.compiler.CompileContext;
 import com.intellij.openapi.compiler.CompileScope;
 import com.intellij.openapi.compiler.CompilerMessage;
 import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.openapi.module.Module;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -45,4 +46,6 @@ public interface CompileContextEx extends CompileContext {
   void commitZipFiles();
 
   void commitZip(String outputDir) throws IOException;
+
+  void assignModule(@NotNull VirtualFile root, @NotNull Module module, boolean isTestSource);
 }
-- 
2.11.4.GIT