Using JetClsMethod interface to recognize method compiled from kotlin

This commit is contained in:
Nikolay Krasko
2013-05-20 13:23:46 +04:00
parent 3a2a932e2f
commit 3b35810c9a
2 changed files with 6 additions and 9 deletions
@@ -23,7 +23,6 @@ import com.intellij.openapi.util.Comparing;
import com.intellij.openapi.util.SystemInfo;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiCompiledElement;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.impl.java.stubs.PsiClassStub;
@@ -161,12 +160,10 @@ public class LightClassUtil {
List<PsiMethod> methods = new SmartList<PsiMethod>();
for (PsiMethod method : psiClass.getMethods()) {
try {
if (method instanceof JetClsMethod) {
if (method instanceof PsiCompiledElement && ((PsiCompiledElement) method).getMirror() == declaration) {
methods.add(method);
if (!collectAll) {
return methods;
}
if (method instanceof JetClsMethod && ((JetClsMethod) method).getOrigin() == declaration) {
methods.add(method);
if (!collectAll) {
return methods;
}
}
}