summaryrefslogtreecommitdiff
path: root/lib/coderay/encoders/count.rb
blob: 52d4bdda84979d24c14c9a41ac6dcd8a59e89e1f (plain)
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
module CodeRay
module Encoders
  
  # Returns the number of tokens.
  # 
  # Text and block tokens are counted.
  class Count < Encoder
    
    register_for :count
    
  protected
    
    def setup options
      @out = 0
    end
    
  public
    
    def text_token text, kind
      @out += 1
    end
    
    def begin_group kind
      @out += 1
    end
    alias end_group begin_group
    alias begin_line begin_group
    alias end_line begin_group
    
  end
  
end
end