git.oblomov.eu Git - ohcount/blob - test/expected_dir/lua1.lua

   1 lua     comment -- random code from http://lua-users.org/wiki/TextJustification
   2 lua     comment -- easy queue implementation ------------------------------------------
   3 lua     blank
   4 lua     code    function q_create()
   5 lua     code        local q = {}
   6 lua     code        q.first = 0
   7 lua     code        q.last = 0
   8 lua     code        return q
   9 lua     code    end
  10 lua     blank
  11 lua     code    function q_insert(q, s)
  12 lua     code        q[q.last] = s
  13 lua     code        q.last = q.last + 1
  14 lua     code    end
  15 lua     blank
  16 lua     code    function q_empty(q)
  17 lua     code        return q.first >= q.last
  18 lua     code    end
  19 lua     blank
  20 lua     code    function q_remove(q)
  21 lua     code        if q_empty(q) then
  22 lua     code            return nil
  23 lua     code        end
  24 lua     code        local s = q[q.first]
  25 lua     code        q[q.first] = nil
  26 lua     code        q.first = q.first+1
  27 lua     code        return s
  28 lua     code    end
  29 lua     blank
  30 lua     code    function q_card(q)
  31 lua     code        return q.last - q.first
  32 lua     code    end
  33 lua     blank
  34 lua     code    function q_length(q, f)
  35 lua     code        local l, i = 0, q.first
  36 lua     code        while i < q.last do
  37 lua     code            l = l + strlen(q[i])
  38 lua     code            i = i + 1
  39 lua     code        end
  40 lua     code        return l
  41 lua     code    end
  42 lua     blank
  43 lua     comment -- line creation routines ------------------------------------------
  44 lua     blank
  45 lua     comment -- justifies one line to fit into MAX columns
  46 lua     code    function justify(q)
  47 lua     code        local blanks = MAX - q_length(q)
  48 lua     code        local skips = q_card(q) - 1
  49 lua     code        local line = q_remove(q)
  50 lua     code        local quotient = floor(blanks/skips)
  51 lua     code        local reminder = blanks/skips - quotient
  52 lua     code        local error = 0
  53 lua     code        while skips > 0 do
  54 lua     code            error = error + reminder
  55 lua     code            if error >= 0.5 then
  56 lua     code                error = error - 1
  57 lua     code                line = line .. strrep(" ", quotient+1)
  58 lua     code            else
  59 lua     code                line = line .. strrep(" ", quotient)
  60 lua     code            end
  61 lua     code            line = line .. q_remove(q)
  62 lua     code            skips = skips - 1
  63 lua     code        end
  64 lua     code        return line or ""
  65 lua     code    end
  66 lua     blank
  67 lua     comment -- join all words with one space between them
  68 lua     code    function catenate(q)
  69 lua     code        local line = q_remove(q)
  70 lua     code        while not q_empty(q) do
  71 lua     code            line = line .. " " .. q_remove(q)
  72 lua     code        end
  73 lua     code        return line or ""
  74 lua     code    end
  75 lua     blank
  76 lua     comment -- main program -----------------------------------------------------
  77 lua     code    DEFMAX = 72
  78 lua     comment -- tries to get MAX from command-line
  79 lua     code    if not arg or getn(arg) < 1 then
  80 lua     code        MAX = DEFMAX
  81 lua     code    else
  82 lua     code        MAX = tonumber(arg[1])
  83 lua     code        if not MAX or MAX < 0 then
  84 lua     code            MAX = DEFMAX
  85 lua     code        end
  86 lua     code    end
  87 lua     blank
  88 lua     comment -- collects all text from stdin
  89 lua     code    text = q_create()
  90 lua     code    line = read()
  91 lua     code    while line do
  92 lua     code        _, n = gsub(line, "(%S+)", function (s) q_insert(%text, s) end)
  93 lua     code        if n == 0 then
  94 lua     code            q_insert(text, "\n")
  95 lua     code        end
  96 lua     code        line = read()
  97 lua     code    end
  98 lua     blank
  99 lua     comment -- justify paragraphs
 100 lua     code    line = q_create()
 101 lua     code    word = q_remove(text)
 102 lua     code    size = 0
 103 lua     code    while word do
 104 lua     code        if word == "\n" then
 105 lua     code            if not q_empty(line) then
 106 lua     code                write(catenate(line), "\n\n")
 107 lua     code            else
 108 lua     code                write("\n")
 109 lua     code            end
 110 lua     code            size = 0
 111 lua     code        elseif size + strlen(word) > MAX then
 112 lua     code            write(justify(line), "\n")
 113 lua     code            size = 0
 114 lua     code        end
 115 lua     code        if word ~= "\n" then
 116 lua     code            q_insert(line, word)
 117 lua     code            size = size + strlen(word) + 1
 118 lua     code        end
 119 lua     code        word = q_remove(text)
 120 lua     code    end
 121 lua     code    write(catenate(line), "\n")