#see license.txt for license details

"""

Tests for the text Label class.

"""

from reportlab.lib.testutils import setOutDir,setOutDir,makeSuiteForClasses, outputfile, printLocation

setOutDir(__name__)

import os, sys, copy

from os.path import join, basename, splitext

import unittest

from reportlab.lib import colors

from reportlab.lib.units import cm

from reportlab.lib.pagesizes import A4

from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle

from reportlab.pdfgen.canvas import Canvas

from reportlab.graphics.shapes import *

from reportlab.graphics.charts.textlabels import Label

from reportlab.platypus.flowables import Spacer, PageBreak

from reportlab.platypus.paragraph import Paragraph

from reportlab.platypus.xpreformatted import XPreformatted

from reportlab.platypus.frames import Frame

from reportlab.platypus.doctemplate import PageTemplate, BaseDocTemplate

def myMainPageFrame(canvas, doc):

    "The page frame used for all PDF documents."

    canvas.saveState()

    #canvas.rect(2.5*cm, 2.5*cm, 15*cm, 25*cm)

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

    pageNumber = canvas.getPageNumber()

    canvas.drawString(10*cm, cm, str(pageNumber))

    canvas.restoreState()

class MyDocTemplate(BaseDocTemplate):

    "The document template used for all PDF documents."

    _invalidInitArgs = ('pageTemplates',)

    def __init__(self, filename, **kw):

        frame1 = Frame(2.5*cm, 2.5*cm, 15*cm, 25*cm, id='F1')

        self.allowSplitting = 0

        BaseDocTemplate.__init__(self, filename, **kw)

        template = PageTemplate('normal', [frame1], myMainPageFrame)

        self.addPageTemplates(template)

class LabelTestCase(unittest.TestCase):

    "Test Label class."

    def _test0(self):

        "Perform original test function."

        pdfPath = outputfile('test_charts_textlabels.pdf')

        c = Canvas(pdfPath)

        label = Label()

        demoLabel = label.demo()

        demoLabel.drawOn(c, 0, 0)

        c.save()

    def _makeProtoLabel(self):

        "Return a label prototype for further modification."

        protoLabel = Label()

        protoLabel.dx = 0

        protoLabel.dy = 0

        protoLabel.boxStrokeWidth = 0.1

        protoLabel.boxStrokeColor = colors.black

        protoLabel.boxFillColor = colors.yellow

        # protoLabel.text = 'Hello World!' # Does not work as expected.

        return protoLabel

    def _makeDrawings(self, protoLabel, text=None):

        # Set drawing dimensions.

        w, h = drawWidth, drawHeight = 400, 100

        drawings = []

        for boxAnchors in ('sw se nw ne', 'w e n s', 'c'):

            boxAnchors = boxAnchors.split(' ')

            # Create drawing.

            d = Drawing(w, h)

            d.add(Line(0, h*0.5, w, h*0.5, strokeColor=colors.gray, strokeWidth=0.5))

            d.add(Line(w*0.5 ,0, w*0.5, h, strokeColor=colors.gray, strokeWidth=0.5))

            labels = []

            for boxAnchor in boxAnchors:

                # Modify label, put it on a drawing.

                label = copy.deepcopy(protoLabel)

                label.boxAnchor = boxAnchor

                args = {'ba':boxAnchor, 'text':text or 'Hello World!'}

                label.setText('(%(ba)s) %(text)s (%(ba)s)' % args)

                labels.append(label)

            for label in labels:

                d.add(label)

            drawings.append(d)

        return drawings

    def test1(self):

        "Test all different box anchors."

        # Build story.

        story = []

        styleSheet = getSampleStyleSheet()

        bt = styleSheet['BodyText']

        h1 = styleSheet['Heading1']

        h2 = styleSheet['Heading2']

        h3 = styleSheet['Heading3']

        story.append(Paragraph('Tests for class <i>Label</i>', h1))

        story.append(Paragraph('Testing box anchors', h2))

        story.append(Paragraph("""This should display "Hello World" labels

written as black text on a yellow box relative to the origin of the crosshair

axes. The labels indicate their relative position being one of the nine

canonical points of a box: sw, se, nw, ne, w, e, n, s or c (standing for

<i>southwest</i>, <i>southeast</i>... and <i>center</i>).""", bt))

        story.append(Spacer(0, 0.5*cm))

        # Round 1a

        story.append(Paragraph('Helvetica 10pt', h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.textAnchor = 'start'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 10

        drawings = self._makeDrawings(protoLabel)

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        # Round 1b

        story.append(Paragraph('Helvetica 18pt', h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.textAnchor = 'start'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 18

        drawings = self._makeDrawings(protoLabel)

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        # Round 1c

        story.append(Paragraph('Helvetica 18pt, multi-line', h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.textAnchor = 'start'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 18

        drawings = self._makeDrawings(protoLabel, text='Hello\nWorld!')

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        story.append(Paragraph('Testing text (and box) anchors', h2))

        story.append(Paragraph("""This should display labels as before,

but now with a fixes size and showing some effect of setting the

textAnchor attribute.""", bt))

        story.append(Spacer(0, 0.5*cm))

        # Round 2a

        story.append(Paragraph("Helvetica 10pt, textAnchor='start'", h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.width = 4*cm

        protoLabel.height = 1.5*cm

        protoLabel.textAnchor = 'start'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 10

        drawings = self._makeDrawings(protoLabel)

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        # Round 2b

        story.append(Paragraph("Helvetica 10pt, textAnchor='middle'", h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.width = 4*cm

        protoLabel.height = 1.5*cm

        protoLabel.textAnchor = 'middle'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 10

        drawings = self._makeDrawings(protoLabel)

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        # Round 2c

        story.append(Paragraph("Helvetica 10pt, textAnchor='end'", h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.width = 4*cm

        protoLabel.height = 1.5*cm

        protoLabel.textAnchor = 'end'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 10

        drawings = self._makeDrawings(protoLabel)

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        # Round 2d

        story.append(Paragraph("Helvetica 10pt, multi-line, textAnchor='start'", h3))

        story.append(Spacer(0, 0.5*cm))

        w, h = drawWidth, drawHeight = 400, 100

        protoLabel = self._makeProtoLabel()

        protoLabel.setOrigin(drawWidth*0.5, drawHeight*0.5)

        protoLabel.width = 4*cm

        protoLabel.height = 1.5*cm

        protoLabel.textAnchor = 'start'

        protoLabel.fontName = 'Helvetica'

        protoLabel.fontSize = 10

        drawings = self._makeDrawings(protoLabel, text='Hello\nWorld!')

        for d in drawings:

            story.append(d)

            story.append(Spacer(0, 1*cm))

        story.append(PageBreak())

        path = outputfile('test_charts_textlabels.pdf')

        doc = MyDocTemplate(path)

        doc.multiBuild(story)

def makeSuite():

    return makeSuiteForClasses(LabelTestCase)

#noruntests

if __name__ == "__main__":

    unittest.TextTestRunner().run(makeSuite())

    printLocation()