#!/usr/bin/env python
# -*- coding: iso-8859-1 -*-

# This is typesetpy, a program form typesetting Python code with LaTeX.
# For more information, see http://www.lysator.liu.se/~jc/typesetpy.html

# Copyright (C) 2005  Jörgen Cederlöf <jc@lysator.liu.se>
# 
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
# 02111-1307, USA.

from __future__ import division
import sys
import re

VERSION = "0.5"

# See /usr/share/a2ps/sheets/python.ssh for inspiration.
# See /usr/share/doc/texmf/latex/fancyvrb/fancyvrb.ps.gz for fancyvrb.

# Put \usepackage{fancyvrb} in the preamble and do \input to the file
# created by this program.

# typesetpy infile outfile <label>

keywords = ['and', 'elif', 'from', 'lambda', 'return', 'break',
            'else', 'global', 'not', 'try', 'class', 'except',
            'if', 'or', 'while', 'continue', 'exec', 'import',
            'pass', 'def', 'finally', 'in', 'print', 'del',
            'is', 'raise', 'for', 'assert', 'as']

options={'commandchars': '§«»',
         'frame': 'single',
         'numbers': 'left',
         'numbersep': '5pt',
         'baselinestretch': '0.8',
         'fontsize': '\\scriptsize'
         }

def getstrings(line, __instring=[None]):
    instring, lastmatch, i = __instring[0], 0, 0
    while i < len(line):
        if   line[i] == '\\':
            i += 1 # Yes, this works for raw strings too.
        elif instring:
            if line[i:i+len(instring)] == instring:
                instring = None
                yield 'string', line[lastmatch:i]
                lastmatch = i
        else:
            if   line[i] == '#':
                yield 'code',    line[lastmatch:i]
                yield 'comment', line[i:]
                i = lastmatch = len(line)
            elif line[i:i+3] == '"""':
                instring = '"""'
                yield 'code', line[lastmatch:i+3]
                lastmatch = i+3
            elif line[i] in ('"', "'"):
                instring = line[i]
                yield 'code', line[lastmatch:i+1]
                lastmatch = i+1
        i += 1
    __instring[0] = instring
    # The backslash isn't really part of the string.
    if instring and line[-1]=='\\':
        yield 'string', line[lastmatch:-1]
        yield 'code', line[-1]
    elif instring:
        yield 'string', line[lastmatch:]
    else:
        yield 'code', line[lastmatch:]

def getkeywords(line):
    tokenre = re.compile("[a-zA-Z_][a-zA-Z_0-9]*")
    lastend = 0
    indef   = False
    for match in tokenre.finditer(line):
        start, end = match.start(), match.end()
        token = line[start:end]
        if indef:
            yield 'code',     line[lastend:start]
            yield 'function', token
            lastend = end
            indef = False
        elif token in keywords:
            yield 'code',     line[lastend:start]
            yield 'keyword',  token
            lastend = end
            if token == 'def':
                indef = True
    yield 'code', line[lastend:]

def gettokens(line):
    for codetype, substring in getstrings(line):
        if   codetype == 'code':
            for x in getkeywords(substring):
                yield x
        else:
            yield codetype, substring

def rewriteline(line):
    l, d = [], []
    for codetype, substring in gettokens(line):
        if not substring:
            continue
        if   codetype == 'keyword':
            l.append("§textbf«%s»" % substring)
        elif codetype == 'string':
            # textcolor only seems to work with {} as commandchars,
            # but color doesn't have that problem.
            l.append("§color«stringcolor»§textit«%s»§normalcolor«»" % substring)
        elif codetype == 'comment':
            l.append("§color«commentcolor»§textsl«%s»§normalcolor«»" % substring)
        elif codetype == 'function':
            l.append(substring)
            d.append(substring)
        elif codetype == 'code':
            l.append(substring)
        else:
            assert False
    return "".join(l), d

def rewritelinehtml(line, commentcolor, stringcolor, counter):
    l, d = [], []
    for codetype, substring in gettokens(line):
        if not substring:
            continue
        # The Python string will be inside a rawhtml environment,
        # which seems to protect it against most dangerous characters
        # like backslash. These need special attention:
        substring = substring.replace(" ", "&nbsp;").replace("%", "\\%")
        if   codetype == 'keyword':
            l.append("<b>%s</b>" % substring)
        elif codetype == 'string':
            l.append("<font color=\"%s\"><i>%s</i></font>" % (stringcolor, substring))
        elif codetype == 'comment':
            l.append("<font color=\"%s\"><i>%s</em></i>" % (commentcolor, substring))
        elif codetype == 'function':
            l.append(substring)
            d.append(substring)
        elif codetype == 'code':
            l.append(substring)
        else:
            assert False
    return "\\the%s&\\begin{rawhtml}<code>%s</code>\\end{rawhtml}\\\\" % (counter, "".join(l)), d

def printtex(infilename, outfilename, label, latex2html=False, standalone=False):
    i = open(infilename)
    o = open(outfilename, 'w')
    commentout = (not standalone) and "%" or ""
    options['label'] = label
    print >>o, "%% Created with typesetpy version %s" % VERSION
    if latex2html:
        print >>o, "% This file will only work with latex2html, not the normal latex."
        print >>o, "% The generated HTML won't be pretty, but the output from latex2html never is."
        print >>o, "% Pass \"-html_version 4.01,table\" to latex2html, or the table will look very strange."
        print >>o, "% Due to some strange table handling in latex2html, you will probably"
        print >>o, "% need to run the HTML through sed 's/cols=\"2\"//g' or the line number"
        print >>o, "% column will be too wide."
    else:
        print >>o, "% This file will not work with latex2html."
    if standalone:
        print >>o, "% This is a standalone LaTeX document."
    else:
        print >>o, "% This is a a LaTeX file ment to be included in another document.",
        print >>o, "% Include with something like:"
        print >>o, "% "
        print >>o, "%   \\documentclass{book}"
        print >>o, "%   \\usepackage{fancyvrb}"
        print >>o, "%   \\usepackage{color}"
        if latex2html:
            print >>o, "%   \\usepackage{html}"
        print >>o, "%   \\begin{document}"
        print >>o,"%%     \\input{%s}" % outfilename
        print >>o, "%   \\end{document}"
        print >>o, "% "
    print >>o
    print >>o, commentout + "\\documentclass{book}"
    if not latex2html:
        print >>o, commentout + "\\usepackage{fancyvrb}"
        print >>o, commentout + "\\usepackage{color}"
    else:
        print >>o, commentout + "\\usepackage{html}"
    print >>o, commentout + "\\begin{document}"
    print >>o, "%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%"
    if not latex2html:
        print >>o, "\\definecolor{commentcolor}{rgb}{0.1,0.1,0.8}"
        print >>o, "\\definecolor{stringcolor}{rgb}{0,0.4,0}"
        # Default is:
        # \def\theFancyVerbLine{\rmfamily\tiny\arabic{FancyVerbLine}}
        print >>o, "\\renewcommand{\\theFancyVerbLine}{%"
        print >>o, "   \\raisebox{0.6pt}{\\rmfamily\\tiny\\arabic{FancyVerbLine}}}"
        print >>o, "%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%"
        print >>o, "\\begin{Verbatim}["
        for key in options:
            print >>o, "   %s=%s," % (key, options[key])
        print >>o, "        ]"
    else:
        linecounter = "%slinenumber" % label.replace(".", "")
        print >>o, "\\newcounter{%s}" % linecounter
        print >>o, "\\begin{tabular}{r|l}"
        print >>o, "  \\multicolumn{2}{c}{%s} \\\\" % label
        print >>o, "  \\hline \\\\"
        commentcolor = "#1a1acd"
        stringcolor  = "#006700"
    oldfunctions = []
    for line in i:
        if not latex2html:
            rewrittenline, functions = rewriteline(line.strip('\n'))
        else:
            print >>o, "\\refstepcounter{%s}" % linecounter,
            rewrittenline, functions = rewritelinehtml(line.strip('\n'),
                                         commentcolor, stringcolor, linecounter)
        for function in functions:
            if function in oldfunctions:
                origfunction, i = function, 1
                while function in oldfunctions:
                    function = "%s:%d" % (origfunction, i)
                    i += 1
            if not latex2html:
                print >>o, '§label«functiondef:%s:%s»' % (label, function),
            else:
                print >>o, '\label{functiondef:%s:%s}' % (label, function),
            oldfunctions.append(function)
        print >>o, rewrittenline
    if not latex2html:
        print >>o, "\\end{Verbatim}"
    else:
        print >>o, "\\end{tabular}"
    print >>o, "%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%"
    print >>o, commentout + "\\end{document}"


def main(argv):
    def usage(e=1):
        f = e and sys.stderr or sys.stdout
        print >>f, "This is typesetpy version %s." % VERSION
        print >>f, "Copyright (C) 2005 Jörgen Cederlöf <jc@lysator.liu.se>"
        print >>f
        print >>f, "Usage: %s [--standalone] [--latex2html] input.py output.tex <label>" % argv[0]
        print >>f, "  --standalone creates a standalone LaTeX document."
        print >>f
        print >>f, "  The name of the python file will be used as label if label is not"
        print >>f, "  given on the command line."
        print >>f
        print >>f, "  Due to incompabilities between latex and latex2html, two modes are"
        print >>f, "  supported. With --latex2html a latex file suitable for latex2html"
        print >>f, "  is generated, but it won't work with normal latex."
        print >>f
        print >>f, "  Reference function foo when label is bar with:"
        print >>f, "        \\ref{functiondef:bar:foo}"
        print >>f, "  For further information on how to use the generated files,"
        print >>f, "  see the comments in their preambles."
        print >>f
        print >>f, "  For more information on typesetpy, see: "
        print >>f, "        http://www.lysator.liu.se/~jc/typesetpy.html"
        sys.exit(e)
    if len(argv) < 2: usage()
    if argv[1] in ('-h', '--help'):
        usage(0)
    if argv[1] == '--standalone':
        standalone = True
        del argv[1]
    else:
        standalone = False
    if len(argv) < 2: usage()
    if argv[1] == '--latex2html':
        latex2html = True
        del argv[1]
    else:
        latex2html = False
    if len(argv) not in (3,4): usage()
    infile, outfile = argv[1], argv[2]
    if len(argv) == 4:
        label = argv[3]
    else:
        label = infile
    printtex(infile, outfile, label, latex2html, standalone)

if __name__ == "__main__":
    main(sys.argv)