diff options
Diffstat (limited to 'lib/coderay/encoders/count.rb')
| -rw-r--r-- | lib/coderay/encoders/count.rb | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/lib/coderay/encoders/count.rb b/lib/coderay/encoders/count.rb index 52d4bdd..98a427e 100644 --- a/lib/coderay/encoders/count.rb +++ b/lib/coderay/encoders/count.rb @@ -11,17 +11,23 @@ module Encoders protected def setup options - @out = 0 + super + + @count = 0 + end + + def finish options + output @count end public def text_token text, kind - @out += 1 + @count += 1 end def begin_group kind - @out += 1 + @count += 1 end alias end_group begin_group alias begin_line begin_group |