Search in sources :

Example 1 with UsageIndexValue

use of com.intellij.compiler.classFilesIndex.impl.UsageIndexValue in project intellij-community by JetBrains.

the class CachedRelevantStaticMethodSearcher method getRelevantStaticMethods.

@NotNull
public List<ContextRelevantStaticMethod> getRelevantStaticMethods(final String resultQualifiedClassName, final int minOccurrence) {
    if (resultQualifiedClassName == null || ChainCompletionStringUtil.isPrimitiveOrArrayOfPrimitives(resultQualifiedClassName) || myCompletionContext.getTarget().getClassQName().equals(resultQualifiedClassName)) {
        return Collections.emptyList();
    }
    final TreeSet<UsageIndexValue> indexValues = myIndexReader.getMethods(resultQualifiedClassName);
    if (!indexValues.isEmpty()) {
        int occurrences = 0;
        final List<ContextRelevantStaticMethod> relevantMethods = new ArrayList<>();
        for (final UsageIndexValue indexValue : extractStaticMethods(indexValues)) {
            final MethodIncompleteSignature methodInvocation = indexValue.getMethodIncompleteSignature();
            final PsiMethod method;
            if (myCachedResolveResults.containsKey(methodInvocation)) {
                method = myCachedResolveResults.get(methodInvocation);
            } else {
                final PsiMethod[] methods = myCompletionContext.resolveNotDeprecated(methodInvocation);
                method = MethodChainsSearchUtil.getMethodWithMinNotPrimitiveParameters(methods, Collections.singleton(myCompletionContext.getTarget().getClassQName()));
                myCachedResolveResults.put(methodInvocation, method);
                if (method == null) {
                    return Collections.emptyList();
                }
            }
            if (method == null) {
                return Collections.emptyList();
            }
            if (method.hasModifierProperty(PsiModifier.PUBLIC)) {
                if (isMethodValid(method, myCompletionContext, resultQualifiedClassName)) {
                    occurrences += indexValue.getOccurrences();
                    if (myCompletionContext.getResolveScope().contains(method.getContainingFile().getVirtualFile())) {
                        relevantMethods.add(new ContextRelevantStaticMethod(method, null));
                    }
                    if (occurrences >= minOccurrence) {
                        return relevantMethods;
                    }
                }
            }
        }
    }
    return Collections.emptyList();
}
Also used : MethodIncompleteSignature(com.intellij.compiler.classFilesIndex.impl.MethodIncompleteSignature) UsageIndexValue(com.intellij.compiler.classFilesIndex.impl.UsageIndexValue) ContextRelevantStaticMethod(com.intellij.compiler.classFilesIndex.chainsSearch.context.ContextRelevantStaticMethod) NotNull(org.jetbrains.annotations.NotNull)

Example 2 with UsageIndexValue

use of com.intellij.compiler.classFilesIndex.impl.UsageIndexValue in project intellij-community by JetBrains.

the class ChainsSearcher method search.

@NotNull
private static List<MethodsChain> search(final MethodsUsageIndexReader indexReader, final SearchInitializer initializer, final Set<String> toSet, final int pathMaximalLength, final int maxResultSize, final String targetQName, final ChainCompletionContext context) {
    final Set<String> allExcludedNames = MethodChainsSearchUtil.joinToHashSet(context.getExcludedQNames(), targetQName);
    final SearchInitializer.InitResult initResult = initializer.init(Collections.<String>emptySet());
    final Map<MethodIncompleteSignature, MethodsChain> knownDistance = initResult.getChains();
    final List<WeightAware<MethodIncompleteSignature>> allInitialVertexes = initResult.getVertexes();
    final LinkedList<WeightAware<Pair<MethodIncompleteSignature, MethodsChain>>> q = new LinkedList<>(ContainerUtil.map(allInitialVertexes, methodIncompleteSignatureWeightAware -> {
        final MethodIncompleteSignature underlying = methodIncompleteSignatureWeightAware.getUnderlying();
        return new WeightAware<>(Pair.create(underlying, new MethodsChain(context.resolveNotDeprecated(underlying), methodIncompleteSignatureWeightAware.getWeight(), underlying.getOwner())), methodIncompleteSignatureWeightAware.getWeight());
    }));
    int maxWeight = 0;
    for (final MethodsChain methodsChain : knownDistance.values()) {
        if (methodsChain.getChainWeight() > maxWeight) {
            maxWeight = methodsChain.getChainWeight();
        }
    }
    final ResultHolder result = new ResultHolder(context.getPsiManager());
    while (!q.isEmpty()) {
        ProgressManager.checkCanceled();
        final WeightAware<Pair<MethodIncompleteSignature, MethodsChain>> currentVertex = q.poll();
        final int currentVertexDistance = currentVertex.getWeight();
        final Pair<MethodIncompleteSignature, MethodsChain> currentVertexUnderlying = currentVertex.getUnderlying();
        final MethodsChain currentVertexMethodsChain = knownDistance.get(currentVertexUnderlying.getFirst());
        if (currentVertexDistance != currentVertexMethodsChain.getChainWeight()) {
            continue;
        }
        if (currentVertex.getUnderlying().getFirst().isStatic() || toSet.contains(currentVertex.getUnderlying().getFirst().getOwner())) {
            result.add(currentVertex.getUnderlying().getSecond());
            continue;
        }
        final String currentReturnType = currentVertexUnderlying.getFirst().getOwner();
        final SortedSet<UsageIndexValue> nextMethods = indexReader.getMethods(currentReturnType);
        final MaxSizeTreeSet<WeightAware<MethodIncompleteSignature>> currentSignatures = new MaxSizeTreeSet<>(maxResultSize);
        for (final UsageIndexValue indexValue : nextMethods) {
            final MethodIncompleteSignature vertex = indexValue.getMethodIncompleteSignature();
            final int occurrences = indexValue.getOccurrences();
            if (vertex.isStatic() || !vertex.getOwner().equals(targetQName)) {
                final int vertexDistance = Math.min(currentVertexDistance, occurrences);
                final MethodsChain knownVertexMethodsChain = knownDistance.get(vertex);
                if ((knownVertexMethodsChain == null || knownVertexMethodsChain.getChainWeight() < vertexDistance)) {
                    if (currentSignatures.isEmpty() || currentSignatures.last().getWeight() < vertexDistance) {
                        if (currentVertexMethodsChain.size() < pathMaximalLength - 1) {
                            final MethodIncompleteSignature methodInvocation = indexValue.getMethodIncompleteSignature();
                            final PsiMethod[] psiMethods = context.resolveNotDeprecated(methodInvocation);
                            if (psiMethods.length != 0 && MethodChainsSearchUtil.checkParametersForTypesQNames(psiMethods, allExcludedNames)) {
                                final MethodsChain newBestMethodsChain = currentVertexMethodsChain.addEdge(psiMethods, indexValue.getMethodIncompleteSignature().getOwner(), vertexDistance);
                                currentSignatures.add(new WeightAware<>(indexValue.getMethodIncompleteSignature(), vertexDistance));
                                knownDistance.put(vertex, newBestMethodsChain);
                            }
                        }
                    }
                } else {
                    break;
                }
            }
        }
        boolean updated = false;
        if (!currentSignatures.isEmpty()) {
            boolean isBreak = false;
            for (final WeightAware<MethodIncompleteSignature> sign : currentSignatures) {
                final PsiMethod[] resolved = context.resolveNotDeprecated(sign.getUnderlying());
                if (!isBreak) {
                    if (sign.getWeight() * NEXT_METHOD_IN_CHAIN_RATIO > currentVertex.getWeight()) {
                        final boolean stopChain = sign.getUnderlying().isStatic() || toSet.contains(sign.getUnderlying().getOwner());
                        if (stopChain) {
                            updated = true;
                            result.add(currentVertex.getUnderlying().getSecond().addEdge(resolved, sign.getUnderlying().getOwner(), sign.getWeight()));
                            continue;
                        } else {
                            updated = true;
                            final MethodsChain methodsChain = currentVertexUnderlying.second.addEdge(resolved, sign.getUnderlying().getOwner(), sign.getWeight());
                            q.add(new WeightAware<>(Pair.create(sign.getUnderlying(), methodsChain), sign.getWeight()));
                            continue;
                        }
                    }
                }
                final MethodsChain methodsChain = currentVertexUnderlying.second.addEdge(resolved, sign.getUnderlying().getOwner(), sign.getWeight());
                final ParametersMatcher.MatchResult parametersMatchResult = ParametersMatcher.matchParameters(methodsChain, context);
                if (parametersMatchResult.noUnmatchedAndHasMatched() && parametersMatchResult.hasTarget()) {
                    updated = true;
                    q.addFirst(new WeightAware<>(Pair.create(sign.getUnderlying(), methodsChain), sign.getWeight()));
                }
                isBreak = true;
            }
        }
        if (!updated && (currentVertex.getUnderlying().getFirst().isStatic() || !targetQName.equals(currentVertex.getUnderlying().getFirst().getOwner()))) {
            result.add(currentVertex.getUnderlying().getSecond());
        }
        if (result.size() > maxResultSize) {
            return result.getResult();
        }
    }
    return result.getResult();
}
Also used : ProgressManager(com.intellij.openapi.progress.ProgressManager) java.util(java.util) PsiMethod(com.intellij.psi.PsiMethod) MethodsUsageIndexReader(com.intellij.compiler.classFilesIndex.impl.MethodsUsageIndexReader) ContainerUtil(com.intellij.util.containers.ContainerUtil) UsageIndexValue(com.intellij.compiler.classFilesIndex.impl.UsageIndexValue) MethodIncompleteSignature(com.intellij.compiler.classFilesIndex.impl.MethodIncompleteSignature) PsiManager(com.intellij.psi.PsiManager) PsiModifier(com.intellij.psi.PsiModifier) Nullable(org.jetbrains.annotations.Nullable) PsiClass(com.intellij.psi.PsiClass) ChainCompletionContext(com.intellij.compiler.classFilesIndex.chainsSearch.context.ChainCompletionContext) TargetType(com.intellij.compiler.classFilesIndex.chainsSearch.context.TargetType) Function(com.intellij.util.Function) Pair(com.intellij.openapi.util.Pair) Logger(com.intellij.openapi.diagnostic.Logger) NotNull(org.jetbrains.annotations.NotNull) PsiMethod(com.intellij.psi.PsiMethod) UsageIndexValue(com.intellij.compiler.classFilesIndex.impl.UsageIndexValue) Pair(com.intellij.openapi.util.Pair) MethodIncompleteSignature(com.intellij.compiler.classFilesIndex.impl.MethodIncompleteSignature) NotNull(org.jetbrains.annotations.NotNull)

Aggregations

MethodIncompleteSignature (com.intellij.compiler.classFilesIndex.impl.MethodIncompleteSignature)2 UsageIndexValue (com.intellij.compiler.classFilesIndex.impl.UsageIndexValue)2 NotNull (org.jetbrains.annotations.NotNull)2 ChainCompletionContext (com.intellij.compiler.classFilesIndex.chainsSearch.context.ChainCompletionContext)1 ContextRelevantStaticMethod (com.intellij.compiler.classFilesIndex.chainsSearch.context.ContextRelevantStaticMethod)1 TargetType (com.intellij.compiler.classFilesIndex.chainsSearch.context.TargetType)1 MethodsUsageIndexReader (com.intellij.compiler.classFilesIndex.impl.MethodsUsageIndexReader)1 Logger (com.intellij.openapi.diagnostic.Logger)1 ProgressManager (com.intellij.openapi.progress.ProgressManager)1 Pair (com.intellij.openapi.util.Pair)1 PsiClass (com.intellij.psi.PsiClass)1 PsiManager (com.intellij.psi.PsiManager)1 PsiMethod (com.intellij.psi.PsiMethod)1 PsiModifier (com.intellij.psi.PsiModifier)1 Function (com.intellij.util.Function)1 ContainerUtil (com.intellij.util.containers.ContainerUtil)1 java.util (java.util)1 Nullable (org.jetbrains.annotations.Nullable)1