pret-haskell.lua

   1 -- filename : type-haskell.lua
   2 -- comment  : Pretty printing of haskell programs. Currently, this is just
   3 ---           verbatim printing with the option of applying strikethrough
   4 -- author   : Matthijs Kooijman, Universiteit Twente, NL
   5 -- copyright: Matthijs Kooijman
   6 -- license  : None
   7
   8 local utf = unicode.utf8
   9
  10 local visualizer = buffers.newvisualizer('haskell')
  11
  12 -- Print a string or table of strings with the catcodes set to the default
  13 -- context ones. Use this function to print a string that contains context
  14 -- macro's that should be interpreted. Does not insert any automatic
  15 -- linebreaking.
  16 local function ctxsprint(str)
  17     tex.sprint(tex.ctxcatcodes, str)
  18 end
  19
  20 -- Print a string or table of strings almost literal. Each character in the
  21 -- string will be printed with the "character" catcode, causing it to show up
  22 -- literal in the output. Only the space character is printed with the "space"
  23 -- catcode.
  24 local function texwrite(str)
  25     tex.write(str)
  26 end
  27
  28 function visualizer.flush_line(str,nested)
  29     while str ~= '' do
  30         if utf.match(str, '^ ') then
  31             ctxsprint('\\obs ')
  32             -- Eat the first character
  33             str = utf.sub(str, 2)
  34         else
  35             local text, rest = utf.match(str, "^%-%-(.-)%-%-(.*)")
  36             if text then
  37                 ctxsprint('\\strikethrough{')
  38                 -- Recursively call ourselves to handle spaces gracefully.
  39                 visualizer.flush_line(text)
  40                 ctxsprint('}')
  41                 -- Eat the processed characters
  42                 str = rest
  43             elseif utf.match(str, "^%-%-") then
  44                 ctxsprint('{\\italic{--')
  45                 -- Recursively call ourselves to handle spaces gracefully.
  46                 visualizer.flush_line(utf.sub(str, 3))
  47                 ctxsprint('}}')
  48                 -- Done with this line
  49                 str = ''
  50             else
  51                 -- Write the first character
  52                 texwrite(utf.sub(str, 1, 1))
  53                 -- Eat the first character
  54                 str = utf.sub(str, 2)
  55             end
  56         end
  57     end
  58 end
  59
  60 -- vim: set sw=4 sts=4 expandtab ai: