java/structuralsearch-java/src/com/intellij/structuralsearch/impl/matcher/filters/TypedSymbolNodeFilter.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

package com.intellij.structuralsearch.impl.matcher.filters;

import com.intellij.dupLocator.util.NodeFilter;
import com.intellij.psi.*;

/**
 * Filter for typed symbols
 */
public class TypedSymbolNodeFilter extends JavaElementVisitor implements NodeFilter {
  private boolean result;

  @Override public void visitMethod(PsiMethod psiMethod) {
    result = psiMethod.hasTypeParameters();
  }

  @Override public void visitClass(PsiClass psiClass) {
    result = psiClass.hasTypeParameters();
  }

  @Override public void visitReferenceElement(PsiJavaCodeReferenceElement psiJavaCodeReferenceElement) {
    result = psiJavaCodeReferenceElement.getParameterList().getTypeParameterElements().length > 0;
  }

  @Override public void visitTypeParameter(PsiTypeParameter parameter) {
    // we need this since TypeParameter instanceof PsiClass (?)
  }

  private static class NodeFilterHolder {
    private static final NodeFilter instance = new TypedSymbolNodeFilter();
  }

  public static NodeFilter getInstance() {
    return NodeFilterHolder.instance;
  }

  private TypedSymbolNodeFilter() {
  }

  public boolean accepts(PsiElement element) {
    result = false;
    if (element!=null) element.accept(this);
    return result;
  }
}