sources/scalac/transformer/matching/CollectVariableTraverser.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

package scalac.transformer.matching ;

import scalac.util.Name ;
import scalac.ast.Tree ;
import scalac.symtab.Symbol ;

import java.util.Vector;

class CollectVariableTraverser extends VariableTraverser {

      protected Vector vars;

      boolean isVariableName( Name name ) {
            return ( name.toString().indexOf("$") == -1 )
                  && super.isVariableName( name );
      }


      void handleVariableSymbol( Symbol sym ) {
            vars.add( sym );
      }

      public CollectVariableTraverser() {
            this.vars = new Vector();
      }

      static boolean containsBinding( Tree  pat ) {

            CollectVariableTraverser cvt = new CollectVariableTraverser();
            cvt.traverse( pat );
            return !cvt.vars.isEmpty();

      }

    static Vector collectVars( Tree pat ) {
	CollectVariableTraverser cvt = new CollectVariableTraverser();
	cvt.traverse( pat );
	return cvt.vars;
    }

}