pret-lam.lua

   1 -- filename : type-lam.lua
   2 -- comment  : Pretty printing of (extended) lambda calculus
   3 -- author   : Matthijs Kooijman, Universiteit Twente, NL
   4 -- copyright: Matthijs Kooijman
   5 -- license  : None
   6
   7 local utf = unicode.utf8
   8
   9 if not buffers                 then buffers                 = { } end
  10 if not buffers.visualizers     then buffers.visualizers     = { } end
  11 if not buffers.visualizers.lam then buffers.visualizers.lam = { } end
  12
  13 buffers.visualizers.lam.colors = {
  14     "prettytwo",
  15     "prettyone",
  16     "prettythree",
  17     "prettyfour"
  18 }
  19
  20 buffers.visualizers.lam.tokens = {
  21     [' '] = {repr = '\\obs '},
  22     ['_'] = {repr = '\\_'},
  23     ['->'] = {repr = '\\rightarrow'},
  24     ['case'] = {style = 'bold'},
  25     ['of'] = {style = 'bold'},
  26     ['let'] = {style = 'bold'},
  27 }
  28
  29 function buffers.visualizers.lam.flush_line(str,nested)
  30     local result, state = { }, 0
  31     local finish, change = buffers.finish_state, buffers.change_state
  32     -- Set the colorscheme, which is used by finish_state and change_state
  33     buffers.currentcolors = buffers.visualizers.lam.colors
  34
  35     while str ~= "" do
  36         found = false
  37         for tok,props in pairs(buffers.visualizers.lam.tokens) do
  38             -- Try to remove tok from the start of str
  39             str, count = utf.gsub(str, "^" .. tok, "")
  40             if count ~= 0 then
  41                 -- Replace the token with the repr, or just itself
  42                 result[#result+1] = props.repr or tok
  43                 -- Enclose the token in {\style .. }
  44                 if props.style then
  45                     result[#result] = "{\\" .. props.style ..
  46                                         " " .. result[#result] ..
  47                                         "}"
  48                 end
  49                 -- We found a token, now start over
  50                 found = true
  51                 break
  52             end
  53         end
  54         if not found then
  55             result[#result+1] = utf.sub(str, 1, 1)
  56             str = utf.sub(str, 2)
  57         end
  58     end
  59     state = finish(state, result)
  60     buffers.flush_result(result,nested)
  61 end
  62
  63 -- vim: set sw=4 sts=4 expandtab ai: