view .svn/pristine/7a/7acfd8a3d508663aeabae0309fddb7dc899a74cf.svn-base @ 1384:b51b5ae3734c luisf

Merge
author luisf <luis.figueira@eecs.qmul.ac.uk>
date Fri, 20 Sep 2013 19:04:25 +0100
parents cbb26bc654de
children
line wrap: on
line source
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
      super
      
      @count = 0
    end
    
    def finish options
      output @count
    end
    
  public
    
    def text_token text, kind
      @count += 1
    end
    
    def begin_group kind
      @count += 1
    end
    alias end_group begin_group
    alias begin_line begin_group
    alias end_line begin_group
    
  end
  
end
end