#see license.txt for license details

#history http://www.reportlab.co.uk/cgi-bin/viewcvs.cgi/public/reportlab/trunk/reportlab/tools/docco/rltemplate.py

# doc template for RL manuals.  Currently YAML is hard-coded

#to use this, which is wrong.

from reportlab.platypus import PageTemplate, \

     BaseDocTemplate, Frame, Paragraph

from reportlab.lib.units import inch, cm

from reportlab.lib.sequencer import Sequencer

from reportlab.rl_config import defaultPageSize

class FrontCoverTemplate(PageTemplate):

    def __init__(self, id, pageSize=defaultPageSize):

        self.pageWidth = pageSize[0]

        self.pageHeight = pageSize[1]

        frame1 = Frame(inch,

                       3*inch,

                       self.pageWidth - 2*inch,

                       self.pageHeight - 518, id='cover')

        PageTemplate.__init__(self, id, [frame1])  # note lack of onPage

    def afterDrawPage(self, canvas, doc):

        canvas.saveState()

        canvas.drawImage('../images/replogo.gif',2*inch, 8*inch)

        canvas.setFont('Times-Roman', 10)

        canvas.line(inch, 120, self.pageWidth - inch, 120)

        canvas.drawString(inch, 100, 'Thornton House')

        canvas.drawString(inch, 88, 'Thornton Road')

        canvas.drawString(inch, 76, 'Wimbledon')

        canvas.drawString(inch, 64, 'London SW19 4NG, UK')

        canvas.restoreState()

class OneColumnTemplate(PageTemplate):

    def __init__(self, id, pageSize=defaultPageSize):

        self.pageWidth = pageSize[0]

        self.pageHeight = pageSize[1]

        frame1 = Frame(inch,

                       inch,

                       self.pageWidth - 2*inch,

                       self.pageHeight - 2*inch,

                       id='normal')

        PageTemplate.__init__(self, id, [frame1])  # note lack of onPage

    def afterDrawPage(self, canvas, doc):

        y = self.pageHeight - 50

        canvas.saveState()

        canvas.setFont('Times-Roman', 10)

        canvas.drawString(inch, y+8, doc.title)

        canvas.drawRightString(self.pageWidth - inch, y+8, doc.chapter)

        canvas.line(inch, y, self.pageWidth - inch, y)

        canvas.drawCentredString(doc.pagesize[0] / 2, 0.75*inch, 'Page %d' % canvas.getPageNumber())

        canvas.restoreState()

class TOCTemplate(PageTemplate):

    def __init__(self, id, pageSize=defaultPageSize):

        self.pageWidth = pageSize[0]

        self.pageHeight = pageSize[1]

        frame1 = Frame(inch,

                       inch,

                       self.pageWidth - 2*inch,

                       self.pageHeight - 2*inch,

                       id='normal')

        PageTemplate.__init__(self, id, [frame1])  # note lack of onPage

    def afterDrawPage(self, canvas, doc):

        y = self.pageHeight - 50

        canvas.saveState()

        canvas.setFont('Times-Roman', 10)

        canvas.drawString(inch, y+8, doc.title)

        canvas.drawRightString(self.pageWidth - inch, y+8, 'Table of contents')

        canvas.line(inch, y, self.pageWidth - inch, y)

        canvas.drawCentredString(doc.pagesize[0] / 2, 0.75*inch, 'Page %d' % canvas.getPageNumber())

        canvas.restoreState()

class TwoColumnTemplate(PageTemplate):

    def __init__(self, id, pageSize=defaultPageSize):

        self.pageWidth = pageSize[0]

        self.pageHeight = pageSize[1]

        colWidth = 0.5 * (self.pageWidth - 2.25*inch)

        frame1 = Frame(inch,

                       inch,

                       colWidth,

                       self.pageHeight - 2*inch,

                       id='leftCol')

        frame2 = Frame(0.5 * self.pageWidth + 0.125,

                       inch,

                       colWidth,

                       self.pageHeight - 2*inch,

                       id='rightCol')

        PageTemplate.__init__(self, id, [frame1, frame2])  # note lack of onPage

    def afterDrawPage(self, canvas, doc):

        y = self.pageHeight - 50

        canvas.saveState()

        canvas.setFont('Times-Roman', 10)

        canvas.drawString(inch, y+8, doc.title)

        canvas.drawRightString(self.pageWidth - inch, y+8, doc.chapter)

        canvas.line(inch, y, self.pageWidth - inch, y*inch)

        canvas.drawCentredString(doc.pagesize[0] / 2, 0.75*inch, 'Page %d' % canvas.getPageNumber())

        canvas.restoreState()

class RLDocTemplate(BaseDocTemplate):

    def afterInit(self):

        self.addPageTemplates(FrontCoverTemplate('Cover', self.pagesize))

        self.addPageTemplates(TOCTemplate('TOC', self.pagesize))

        self.addPageTemplates(OneColumnTemplate('Normal', self.pagesize))

        self.addPageTemplates(TwoColumnTemplate('TwoColumn', self.pagesize))

        self.seq = Sequencer()

    def beforeDocument(self):

        self.canv.showOutline()

        self.title = "(Document Title Goes Here)"

        self.chapter = "(No chapter yet)"

        self.seq.reset('section')

        self.seq.reset('chapter')

    def afterFlowable(self, flowable):

        """Detect Level 1 and 2 headings, build outline,

        and track chapter title."""

        if isinstance(flowable, Paragraph):

            style = flowable.style.name

            txt = flowable.getPlainText()

            if style == 'Title':

                self.title = txt

            elif style == 'Heading1':

                self.chapter = txt 

                key = 'ch%s' % self.seq.nextf('chapter')

                self.canv.bookmarkPage(key)

                self.canv.addOutlineEntry(txt, key, 0, 0)

                self.seq.reset("section")

                self.notify('TOCEntry', (0, txt, self.page, key))

            elif style == 'Heading2':

                self.section = flowable.text

                key = 'ch%ss%s' % (self.seq.thisf("chapter"), self.seq.nextf("section"))

                self.canv.bookmarkPage(key)

                self.canv.addOutlineEntry(txt, key, 1, 0)

                self.notify('TOCEntry', (1, txt, self.page, key))