make SameDirectoryWeigher and SameLogicalRootWeigher work with completion file copy

This commit is contained in:
peter
2016-05-10 08:41:17 +02:00
parent 2e82ca06f9
commit 665ca5d1fb
2 changed files with 2 additions and 2 deletions
@@ -39,7 +39,7 @@ public class SameDirectoryWeigher extends ProximityWeigher {
private static PsiDirectory getParentDirectory(PsiElement element) {
PsiFile file = InjectedLanguageUtil.getTopLevelFile(element);
if (file != null) {
element = file;
element = file.getOriginalFile();
}
while (element != null && !(element instanceof PsiDirectory)) {
element = element.getParent();
@@ -57,7 +57,7 @@ public class SameLogicalRootWeigher extends ProximityWeigher {
final PsiFile psiFile = element.getContainingFile();
if (psiFile == null) return null;
final VirtualFile file = psiFile.getVirtualFile();
final VirtualFile file = psiFile.getOriginalFile().getVirtualFile();
if (file == null) return null;
return LogicalRootsManager.getLogicalRootsManager(element.getProject()).findLogicalRoot(file);