Codebase list ohcount / 85104d5a-b71d-4620-9315-33d4daf34104/main test / expected_dir / lua1.lua
85104d5a-b71d-4620-9315-33d4daf34104/main

Tree @85104d5a-b71d-4620-9315-33d4daf34104/main (Download .tar.gz)

lua1.lua @85104d5a-b71d-4620-9315-33d4daf34104/mainraw · history · blame

lua	comment	-- random code from http://lua-users.org/wiki/TextJustification
lua	comment	-- easy queue implementation ------------------------------------------
lua	blank	
lua	code	function q_create()
lua	code	    local q = {}
lua	code	    q.first = 0
lua	code	    q.last = 0
lua	code	    return q
lua	code	end
lua	blank	
lua	code	function q_insert(q, s)
lua	code	    q[q.last] = s
lua	code	    q.last = q.last + 1
lua	code	end
lua	blank	
lua	code	function q_empty(q)
lua	code	    return q.first >= q.last
lua	code	end
lua	blank	
lua	code	function q_remove(q)
lua	code	    if q_empty(q) then
lua	code	        return nil
lua	code	    end
lua	code	    local s = q[q.first]
lua	code	    q[q.first] = nil
lua	code	    q.first = q.first+1
lua	code	    return s
lua	code	end
lua	blank	
lua	code	function q_card(q)
lua	code	    return q.last - q.first
lua	code	end
lua	blank	
lua	code	function q_length(q, f)
lua	code	    local l, i = 0, q.first
lua	code	    while i < q.last do
lua	code	        l = l + strlen(q[i])
lua	code	        i = i + 1
lua	code	    end
lua	code	    return l
lua	code	end
lua	blank	
lua	comment	-- line creation routines ------------------------------------------
lua	blank	
lua	comment	-- justifies one line to fit into MAX columns 
lua	code	function justify(q)
lua	code	    local blanks = MAX - q_length(q)
lua	code	    local skips = q_card(q) - 1
lua	code	    local line = q_remove(q)
lua	code	    local quotient = floor(blanks/skips)
lua	code	    local reminder = blanks/skips - quotient
lua	code	    local error = 0
lua	code	    while skips > 0 do
lua	code	        error = error + reminder
lua	code	        if error >= 0.5 then
lua	code	            error = error - 1
lua	code	            line = line .. strrep(" ", quotient+1)
lua	code	        else
lua	code	            line = line .. strrep(" ", quotient)
lua	code	        end
lua	code	        line = line .. q_remove(q)
lua	code	        skips = skips - 1
lua	code	    end
lua	code	    return line or ""
lua	code	end
lua	blank	
lua	comment	-- join all words with one space between them
lua	code	function catenate(q)
lua	code	    local line = q_remove(q)
lua	code	    while not q_empty(q) do
lua	code	        line = line .. " " .. q_remove(q)
lua	code	    end
lua	code	    return line or ""
lua	code	end
lua	blank	
lua	comment	-- main program -----------------------------------------------------
lua	code	DEFMAX = 72
lua	comment	-- tries to get MAX from command-line
lua	code	if not arg or getn(arg) < 1 then
lua	code	    MAX = DEFMAX
lua	code	else
lua	code	    MAX = tonumber(arg[1])
lua	code	    if not MAX or MAX < 0 then
lua	code	        MAX = DEFMAX
lua	code	    end
lua	code	end
lua	blank	
lua	comment	-- collects all text from stdin
lua	code	text = q_create()
lua	code	line = read()
lua	code	while line do
lua	code	    _, n = gsub(line, "(%S+)", function (s) q_insert(%text, s) end)
lua	code	    if n == 0 then
lua	code	        q_insert(text, "\n")
lua	code	    end
lua	code	    line = read()
lua	code	end
lua	blank	
lua	comment	-- justify paragraphs
lua	code	line = q_create()
lua	code	word = q_remove(text)
lua	code	size = 0
lua	code	while word do
lua	code	    if word == "\n" then
lua	code	        if not q_empty(line) then
lua	code	            write(catenate(line), "\n\n")
lua	code	        else
lua	code	            write("\n")
lua	code	        end
lua	code	        size = 0
lua	code	    elseif size + strlen(word) > MAX then
lua	code	        write(justify(line), "\n")
lua	code	        size = 0
lua	code	    end
lua	code	    if word ~= "\n" then
lua	code	        q_insert(line, word)
lua	code	        size = size + strlen(word) + 1
lua	code	    end
lua	code	    word = q_remove(text)
lua	code	end
lua	code	write(catenate(line), "\n")