$ // This file is part of NIT ( http://www.nitlanguage.org ).
$ //
$ // Copyright 2008 Jean Privat <jean@pryen.org>
$ // Based on algorithms developped for ( http://www.sablecc.org/ ).
$ //
$ // Licensed under the Apache License, Version 2.0 (the "License");
$ // you may not use this file except in compliance with the License.
$ // You may obtain a copy of the License at
$ //
$ //     http://www.apache.org/licenses/LICENSE-2.0
$ //
$ // Unless required by applicable law or agreed to in writing, software
$ // distributed under the License is distributed on an "AS IS" BASIS,
$ // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
$ // See the License for the specific language governing permissions and
$ // limitations under the License.

$ template make_parser()

# State of the parser automata as stored in the parser stack.
private class State
	# The internal state number
	readable writable var _state: Int

	# The node stored with the state in the stack
	readable writable var _nodes: nullable Object

	init(state: Int, nodes: nullable Object)
	do
		_state = state
		_nodes = nodes
	end
end

class Parser
special ParserTable
	# Associated lexer
	var _lexer: Lexer

	# Stack of pushed states and productions
	var _stack: Array[State]

	# Position in the stack
	var _stack_pos: Int

	# Create a new parser based on a given lexer
	init(lexer: Lexer)
	do
		_lexer = lexer
		_stack = new Array[State]
		_stack_pos = -1
		build_goto_table
		build_action_table
		build_reduce_table
	end

	# Do a transition in the automata
	private fun go_to(index: Int): Int
	do
		var state = state
		var table = _goto_table[index]
		var low = 1
		var high = table.length/2 - 1

		while low <= high do
			var middle = (low + high) / 2
			var subindex = middle * 2

			if state < table[subindex] then
				high = middle - 1
			else if state > table[subindex] then
				low = middle + 1
			else
				return table[subindex + 1]
			end
		end

		return table[1] # Default value
	end

	# Push someting in the state stack
	private fun push(numstate: Int, list_node: nullable Object)
	do
		var pos = _stack_pos + 1
		_stack_pos = pos
		if pos < _stack.length then
			var state = _stack[pos]
			state.state = numstate
			state.nodes = list_node
		else
			_stack.push(new State(numstate, list_node))
		end
	end

	# The current state
	private fun state: Int
	do
		return _stack[_stack_pos].state
	end

	# Pop something from the stack state
	private fun pop: nullable Object
	do
		var res = _stack[_stack_pos].nodes
		_stack_pos = _stack_pos -1
		return res
	end

	# Build and return a full AST.
	fun parse: Start
	do
		push(0, null)

		var lexer = _lexer
		while true do
			var token = lexer.peek
			var last_pos = token.location.column_start
			var last_line = token.location.line_start

			if token isa PError then
				return new Start(null, token)
			end

			var index = token.parser_index
			var table = _action_table[state]
			var action_type = table[1]
			var action_value = table[2]

			var low = 1
			var high = table.length/3 - 1

			while low <= high do
				var middle = (low + high) / 2
				var subindex = middle * 3

				if index < table[subindex] then
					high = middle - 1
				else if index > table[subindex] then
					low = middle + 1
				else
					action_type = table[subindex + 1]
					action_value = table[subindex + 2]
					high = low -1 # break
				end
			end

			if action_type == 0 then # SHIFT
				push(action_value, lexer.next)
			else if action_type == 1 then # REDUCE
				_reduce_table[action_value].action(self)
			else if action_type == 2 then # ACCEPT
				var node2 = lexer.next
				assert node2 isa EOF
				var node1 = pop
				assert node1 isa ${/parser/prods/prod/@ename}
				var node = new Start(node1, node2)
				(new SearchTokensVisitor).enter_visit(node)
				return node
			else if action_type == 3 then # ERROR
				var location = new Location(lexer.filename, last_line, last_line, last_pos, last_pos)
				var node2 = new PError.init_error(error_messages[errors[action_value]],location)
				var node = new Start(null, node2)
				return node
			end
		end
		abort
	end

	var _reduce_table: Array[ReduceAction]
	private fun build_reduce_table
	do
		_reduce_table = new Array[ReduceAction].with_items(
$ foreach {rules/rule}
			new ReduceAction@index[-sep ','-]
$ end foreach
		)
	end
end

# Find first and last tokens of production nodes
private class SearchTokensVisitor
special Visitor
	var _untokenned_nodes: Array[Prod]
	var _last_token: nullable Token = null
	redef fun visit(n: nullable PNode)
	do
		if n == null then
			return
		else if n isa Token then
			_last_token = n
			for no in _untokenned_nodes do
				no.first_token = n
			end
			_untokenned_nodes.clear
		else
			assert n isa Prod
			_untokenned_nodes.add(n)
			n.visit_all(self)
			n.last_token = _last_token

			if n.first_token != null then
				var start_location = n.first_token.location
				var end_location = _last_token.location

				if start_location != null and end_location != null then
					var file = end_location.file
					var line_start = start_location.line_start
					var line_end = end_location.line_end
					var column_start = start_location.column_start
					var column_end = end_location.column_end
					n.location = new Location(file, line_start, line_end, column_start, column_end)
				end
			end
		end
	end
	init
	do
		_untokenned_nodes = new Array[Prod]
	end
end

# Each reduca action has its own class, this one is the root of the hierarchy.
private abstract class ReduceAction
	fun action(p: Parser) is abstract
end

$ foreach {rules/rule}
private class ReduceAction@index
special ReduceAction
	redef fun action(p: Parser)
	do
					var node_list: nullable Object = null
$   foreach {action}
$   choose
$     when {@cmd='POP'}
					var ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} = p.pop
$     end
$     when {@cmd='FETCHLIST'}
					var ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} = ${translate(@from,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}
					assert ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} isa Array[Object]
$     end
$     when {@cmd='FETCHNODE'}
					var ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} = ${translate(@from,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}
					assert ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} isa nullable @etype
$     end
$     when {@cmd='ADDNODE'}
					if ${translate(@node,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} != null then
						${translate(@tolist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}.add(${translate(@node,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")})
					end
$     end
$     when {@cmd='ADDLIST'}
#					if ${translate(@fromlist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} != null then
						if ${translate(@tolist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}.is_empty then
							${translate(@tolist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} = ${translate(@fromlist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}
						else
							${translate(@tolist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}.append(${translate(@fromlist,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")})
						end
#					end
$     end
$     when {@cmd='MAKELIST'}
					var ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")} = new Array[Object]
$     end
$     when {@cmd='MAKENODE'}
					var ${translate(@result,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}: nullable @etype = new @etype.init_${translate(@etype,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}(
$       foreach {arg}
$           if @null
						null[-sep ','-]
$           else
						${translate(.,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}[-sep ','-]
$           end
$       end foreach
					)
$     end
$     when {@cmd='RETURNNODE'}
$       if @null
					node_list = null
$       else
					node_list = ${translate(@node,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}
$       end
$     end
$     when {@cmd='RETURNLIST'}
					node_list = ${translate(@list,"ABCDEFGHIJKLMNOPQRSTUVWXYZ","abcdefghijklmnopqrstuvwxyz")}
$     end
$   end choose
$   end foreach
					p.push(p.go_to(@leftside), node_list)
	end
init do end
end
$ end foreach
$ end template

$ template make_parser_tables()
# Parser that build a full AST
abstract class ParserTable
	var _action_table: Array[Array[Int]]
	private fun build_action_table
	do
		_action_table = once [
$ foreach {parser_data/action_table/row}
			action_table_row${position()}[-sep ','-]
$ end foreach
	        ]
	end

$ foreach {parser_data/action_table/row}
	private fun action_table_row${position()}: Array[Int]
	do
		return [
$   foreach {action}
				@from, @action, @to[-sep ','-]
$   end foreach
			]
	end
$ end foreach

	var _goto_table: Array[Array[Int]]
	private fun build_goto_table
	do
		_goto_table = once [
$ foreach {parser_data/goto_table/row}
			[
$   foreach {goto}
				@from, @to[-sep ','-]
$   end foreach
			][-sep ','-]
$ end foreach
		]
	end

	private fun error_messages: Array[String]
	do
		return once [
$ foreach {parser_data/error_messages/msg}
			"${sablecc:string2escaped_unicode(.)}"[-sep ','-]
$ end
		]
	end

	private fun errors: Array[Int]
	do
		return once [
			[-foreach {parser_data/errors/i}-]${.}[-sep ','-][-end-]
		]
	end

	init do end
end
$ end template