PsiElement[] (PsiElement[] sourceElements) { Set<PsiElement> result = new LinkedHashSet<>(); for (PsiElement sourceElement : sourceElements) { result.add(sourceElement instanceof PsiClass ? sourceElement.getContainingFile() : sourceElement); } return PsiUtilCore.toPsiElementArray(result); }