summaryrefslogtreecommitdiff
path: root/lib/coderay/helpers/scanner_helper.rb
blob: 9f48b8330a488d185787ac594833b8e1a1d994aa (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
module CodeRay
	module Scanners

		class Scanner

			# A WordList is a Hash with some additional features.
			# It is intended to be used for keyword recognition.
			#
			# WordList is highly optimized to be used in Scanners,
			# typically to decide whether a given ident is a keyword.
			#
			# Example:
			# 
			#  # define word arrays
			#  RESERVED_WORDS = %w[
			#    asm break case continue default do else
			#    ...		
			#  ]
			#  
			#  PREDEFINED_TYPES = %w[
			#    int long short char void
			#    ...
			#  ]
			#  
			#  PREDEFINED_CONSTANTS = %w[
			#    EOF NULL ...
			#  ]
			#  
			#  # make a WordList
			#  IDENT_KIND = Scanner::WordList.new(:ident).
			#    add(RESERVED_WORDS, :reserved).
			#    add(PREDEFINED_TYPES, :pre_type).
			#    add(PREDEFINED_CONSTANTS, :pre_constant)
			#
			#  ...
			#
			#  def scan_tokens tokens, options
			#    ...
			#    
			#    elsif match = scan(/ [A-Za-z_][A-Za-z_0-9]* /x)
			#      # use it
			#      kind = IDENT_KIND[match]
			#      ...
			#  
			class WordList < Hash

				# Creates a new WordList with +default+ as default value.
				# case_mode controls how keys are compared;
				# :case_match is faster.
				def initialize default = false, case_mode = :case_match
					@case_ignore =
						case case_mode
						when :case_match then false
						when :case_ignore then true
						else raise ArgumentError,
							":case_ignore or :case_match expected, but #{case_mode} given"
						end

					if @case_ignore
						super() do |h, k|
							h[k] = h.fetch k.downcase, default
						end
					else
						super default
					end
				end

				# Checks if a word is included.
				def include? word
					self[word] if @case_ignore
					has_key? word
				end

				# Add words to the list and associate them with
				# +kind+.
				def add words, kind = true
					words.each do |word|
						self[mind_case(word)] = kind
					end
					self
				end

				alias words keys

				# Returns whether key comparing is done case insensitive.
				def case_ignore?
					@case_mode
				end

			private
				# helper method for key 
				def mind_case word
					if @case_ignore
						word.downcase
					else
						word.dup
					end
				end

			end		

		end

	end
end

# vim:sw=2:ts=2:noet:tw=78