| /* |
| * Copyright 2000-2009 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.slicer; |
| |
| import com.intellij.analysis.AnalysisScope; |
| import com.intellij.openapi.application.ApplicationManager; |
| import com.intellij.openapi.progress.ProgressIndicator; |
| import com.intellij.openapi.progress.ProgressManager; |
| import com.intellij.psi.PsiElement; |
| import com.intellij.psi.PsiSubstitutor; |
| import com.intellij.usageView.UsageInfo; |
| import com.intellij.usages.UsageInfo2UsageAdapter; |
| import com.intellij.util.CommonProcessors; |
| import com.intellij.util.Processor; |
| import gnu.trove.TObjectHashingStrategy; |
| import org.jetbrains.annotations.NotNull; |
| |
| /** |
| * @author cdr |
| */ |
| public class SliceUsage extends UsageInfo2UsageAdapter { |
| private final SliceUsage myParent; |
| public final SliceAnalysisParams params; |
| private final PsiSubstitutor mySubstitutor; |
| |
| public SliceUsage(@NotNull PsiElement element, @NotNull SliceUsage parent, @NotNull PsiSubstitutor substitutor) { |
| super(new UsageInfo(element)); |
| myParent = parent; |
| mySubstitutor = substitutor; |
| params = parent.params; |
| assert params != null; |
| } |
| private SliceUsage(@NotNull PsiElement element, @NotNull SliceAnalysisParams params) { |
| super(new UsageInfo(element)); |
| myParent = null; |
| this.params = params; |
| mySubstitutor = PsiSubstitutor.EMPTY; |
| } |
| |
| public static SliceUsage createRootUsage(@NotNull PsiElement element, @NotNull SliceAnalysisParams params) { |
| return new SliceUsage(element, params); |
| } |
| |
| public void processChildren(@NotNull Processor<SliceUsage> processor) { |
| final PsiElement element = getElement(); |
| ProgressIndicator indicator = ProgressManager.getInstance().getProgressIndicator(); |
| indicator.checkCanceled(); |
| |
| final Processor<SliceUsage> uniqueProcessor = |
| new CommonProcessors.UniqueProcessor<SliceUsage>(processor, new TObjectHashingStrategy<SliceUsage>() { |
| @Override |
| public int computeHashCode(final SliceUsage object) { |
| return object.getUsageInfo().hashCode(); |
| } |
| |
| @Override |
| public boolean equals(final SliceUsage o1, final SliceUsage o2) { |
| return o1.getUsageInfo().equals(o2.getUsageInfo()); |
| } |
| }); |
| |
| ApplicationManager.getApplication().runReadAction(new Runnable() { |
| @Override |
| public void run() { |
| if (params.dataFlowToThis) { |
| SliceUtil.processUsagesFlownDownTo(element, uniqueProcessor, SliceUsage.this, mySubstitutor); |
| } |
| else { |
| SliceForwardUtil.processUsagesFlownFromThe(element, uniqueProcessor, SliceUsage.this); |
| } |
| } |
| }); |
| } |
| |
| public SliceUsage getParent() { |
| return myParent; |
| } |
| |
| @NotNull |
| public AnalysisScope getScope() { |
| return params.scope; |
| } |
| |
| SliceUsage copy() { |
| PsiElement element = getUsageInfo().getElement(); |
| return getParent() == null ? createRootUsage(element, params) : new SliceUsage(element, getParent(),mySubstitutor); |
| } |
| |
| public PsiSubstitutor getSubstitutor() { |
| return mySubstitutor; |
| } |
| |
| } |