summaryrefslogtreecommitdiff
path: root/Master/texmf-dist/tex/latex/prettytok/prettytok.lua
blob: a0aba0724352b4e02ffb20fd4dcbc49fd552adf6 (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
do
local function looks_like_token(t)
	return type(t)=="userdata" or (type(t)=="table" and t.cmdname~=nil and t.tok~=nil)
end

-- public API :: once the package is loaded, this holds the .tok value of the frozen relax token
prettyprint_frozenrelaxtok=nil

local function prettyprint_one_arg(tokenlist)

	if looks_like_token(tokenlist) then
		return prettyprint_one_arg({tokenlist})
	end

	local s=""

	local function printstring(t)
		-- this is just for debugging/convenience purpose. If the item is a string instead of a token...
		t=tostring(t)
		for _, c in utf8.codes(t) do
			if c==32 then
				s=s.."token("..c..',"A"),'
			else
				s=s.."token("..c..',"C"),'
			end
		end
	end

	if type(tokenlist)~="table" then
		-- user give a string (or something similar). Print it as detokenized characters
		printstring(tokenlist)
	
	else

		-- normal print of tokenlist
		for i=1, #tokenlist do
			local t=tokenlist[i]
			if not looks_like_token(t) then
				printstring(t)
			elseif t.csname==nil then
				s=s.."token("..t.mode..',"'..(({left_brace=1, right_brace=2, math_shift=3, tab_mark=4, mac_param=6, sup_mark=7, sub_mark=8, spacer="A", letter="B", other_char="C"})[t.cmdname])..'"),'
			else
				if t.active then
					s=s.."token("..utf8.codepoint(t.csname)..',"D"),'
				elseif t.tok==prettyprint_frozenrelaxtok then
					s=s.."csfrozenrelax(),"
				else
					s=s.."cs("
					for j=1, #t.csname do
						s=s..t.csname:byte(j)..","
					end
					s=s.."),"
				end
			end
		end
		
	end
	return s
end

local function check_file_opened()
	-- if it's \relax then get_macro() also returns nil
	if token.get_macro("pretty_check_already_init:") == nil then
		error("Output file not initialized!")
	end
end

local function get_file_number()
	check_file_opened()
	return token.get_mode(token.create("__prettyh_file"))
end

function prettyprint(...)
	local prettyfilenumber=get_file_number()

	local s="print_tl("
	local args={...}
	for i=1, select("#", ...) do
		s=s..prettyprint_one_arg(args[i])
	end

	texio.write(prettyfilenumber, s..")//</script><script>\n")
end

function prettyprintw()
	check_file_opened()
	local extra=token.scan_toks()
	local s={}
	while true do
		local n=token.get_next()
		s[#s+1]=n
		if n.csname=="prettystop" then break end
	end

	prettyprint(extra, s)
	token.put_next(s)
end
end