Set<PsiClass> (PsiNewExpression expression) { if (expression.isArrayCreation()) return null; PsiAnonymousClass aClass = expression.getAnonymousClass(); if (aClass != null) return Collections.singleton(aClass); PsiJavaCodeReferenceElement reference = expression.getClassReference(); if (reference != null) { PsiClass psiClass = ObjectUtils.tryCast(reference.resolve(), PsiClass.class); if (psiClass != null) { return Collections.singleton(psiClass); } } return null; }