summaryrefslogtreecommitdiff
path: root/sources/scalac/transformer/matching/CollectVariableTraverser.java
blob: 207eeea41969452ef07a5314718c6d30b950488b (plain) (blame)
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
45
46
package scalac.transformer.matching ;

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


import java.util.HashSet;

class CollectVariableTraverser extends VariableTraverser {

    protected HashSet nogeneratedVars;
    protected HashSet vars;
    /*
      boolean isVariableName( Name name ) {
            return ( name.toString().indexOf("$") == -1 )
                  && super.isVariableName( name );
      }
    */
      void handleVariableSymbol( Symbol sym ) {
            vars.add( sym );
            if( sym.name.toString().indexOf("$") == -1 ) {
                nogeneratedVars.add( sym );
            }
      }

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

      static boolean containsBinding( Tree  pat ) {

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

      }

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

}