| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.intellij.psi.search.searches; |
| |
| import com.intellij.openapi.extensions.ExtensionPointName; |
| import com.intellij.openapi.util.Condition; |
| import com.intellij.psi.PsiAnonymousClass; |
| import com.intellij.psi.PsiClass; |
| import com.intellij.psi.search.GlobalSearchScope; |
| import com.intellij.psi.search.SearchScope; |
| import com.intellij.psi.util.PsiUtilCore; |
| import com.intellij.util.FilteredQuery; |
| import com.intellij.util.Query; |
| import com.intellij.util.QueryExecutor; |
| |
| /** |
| * @author max |
| */ |
| public class DirectClassInheritorsSearch extends ExtensibleQueryFactory<PsiClass, DirectClassInheritorsSearch.SearchParameters> { |
| public static ExtensionPointName<QueryExecutor> EP_NAME = ExtensionPointName.create("com.intellij.directClassInheritorsSearch"); |
| public static DirectClassInheritorsSearch INSTANCE = new DirectClassInheritorsSearch(); |
| |
| public static class SearchParameters { |
| private final PsiClass myClass; |
| private final SearchScope myScope; |
| private final boolean myIncludeAnonymous; |
| private final boolean myCheckInheritance; |
| |
| public SearchParameters(PsiClass aClass, SearchScope scope, boolean includeAnonymous, boolean checkInheritance) { |
| myClass = aClass; |
| myScope = scope; |
| myIncludeAnonymous = includeAnonymous; |
| myCheckInheritance = checkInheritance; |
| } |
| |
| public SearchParameters(final PsiClass aClass, SearchScope scope, final boolean includeAnonymous) { |
| this(aClass, scope, includeAnonymous, true); |
| } |
| |
| public SearchParameters(final PsiClass aClass, final SearchScope scope) { |
| this(aClass, scope, true); |
| } |
| |
| public PsiClass getClassToProcess() { |
| return myClass; |
| } |
| |
| public SearchScope getScope() { |
| return myScope; |
| } |
| |
| public boolean isCheckInheritance() { |
| return myCheckInheritance; |
| } |
| |
| public boolean includeAnonymous() { |
| return myIncludeAnonymous; |
| } |
| } |
| |
| private DirectClassInheritorsSearch() {} |
| |
| public static Query<PsiClass> search(final PsiClass aClass) { |
| return search(aClass, GlobalSearchScope.allScope(PsiUtilCore.getProjectInReadAction(aClass))); |
| } |
| |
| public static Query<PsiClass> search(final PsiClass aClass, SearchScope scope) { |
| return INSTANCE.createUniqueResultsQuery(new SearchParameters(aClass, scope)); |
| } |
| |
| public static Query<PsiClass> search(final PsiClass aClass, SearchScope scope, boolean includeAnonymous) { |
| return search(aClass, scope, includeAnonymous, true); |
| } |
| |
| public static Query<PsiClass> search(final PsiClass aClass, |
| SearchScope scope, |
| boolean includeAnonymous, |
| final boolean checkInheritance) { |
| final Query<PsiClass> raw = INSTANCE.createUniqueResultsQuery(new SearchParameters(aClass, scope, includeAnonymous, checkInheritance)); |
| |
| if (!includeAnonymous) { |
| return new FilteredQuery<PsiClass>(raw, new Condition<PsiClass>() { |
| @Override |
| public boolean value(final PsiClass psiClass) { |
| return !(psiClass instanceof PsiAnonymousClass); |
| } |
| }); |
| } |
| |
| return raw; |
| } |
| } |