# -----------------------------------------------------------------------------
#
#     P A G E B O T  E X A M P L E S
#
#     Copyright (c) 2017 <NAME> <https://github.com/thomgb>
#     www.pagebot.io
#     Licensed under MIT conditions
#
#     Supporting DrawBot, www.drawbot.com
#     Supporting Flat, xxyxyz.org/flat
# -----------------------------------------------------------------------------
#
#     conditionsHierarchy.py
#

import sys, inspect
from pagebot import getContext
from pagebot.toolbox.units import *
from pagebot.toolbox.color import Color, blackColor, blueColor, greenColor
from pagebot.fonttoolbox.objects.font import findFont
from pagebot.conditions import *

context = getContext()

X0 = 100
Y0 = 100
WIDTH = 1600
HEIGHT = 1400
HBOX = 34
WBOX = 170
GAP = 20
HGAP = 60
P = 15
TEXTSIZE = pt(12)
OFFSET = 9

titleFont = findFont('BungeeInline-Regular')
font = findFont('Roboto-Regular')
boldFont = findFont('BungeeOutline-Regular')

def drawClassHierarchy(obj, colorRange, i):
    
    previous = None
    y = Y0
    x = X0
    
    for c in list(obj.__mro__)[::-1]:
        current = c.__name__
        #if current == 'object':
        #    continue

        if i >= len(colorRange):
            i = 0
            
        drawClass(current, x, y, colorRange[i])
        
        if previous is not None:
            drawConnection(current, previous)

        previous = current
        y += HGAP
        i += 1
    return i

def drawConnection(current, previous):
    if sorted([current, previous]) in connections:
        return
        
    pos0 = drawnclasses[current]
    p0x, p0y = pos0
    pos1 = drawnclasses[previous]
    p1x, p1y = pos1
    context.stroke(blueColor)
    
    # Determines box entry / exit points.
    if p0y > p1y:
        #print('%s > %s' % (current, previous))
        p0x += WBOX / 2
        p1x += WBOX / 2
        p1y += HBOX
    elif p0y < p1y:
        # Never happens?
        p0x + WBOX / 2
        p1x + WBOX / 2
        p0y += HBOX
    elif p0y == p1y:
        p0y += HBOX / 2
        p1y += HBOX / 2
        if p1x > p0x:
            p0x += WBOX
        elif p1x < p0x:
            p1x += WBOX
    
    # TODO: draw only once for any location.
    context.circle(p0x, p0y, 3)
    context.circle(p1x, p1y, 3)

    # Curve.

    path = context.newPath()
    context.moveTo((p0x, p0y))

    cp0x = p0x - OFFSET
    cp0y = p0y - (p0y - p1y) / 3
    context.stroke(greenColor)
    #context.fill(None)
    #context.circle(cp0x, cp0y, 3)

    cp1x = p1x + OFFSET
    cp1y = p1y + (p0y - p1y) / 3
    context.stroke(greenColor)
    #context.fill(None)
    #context.circle(cp1x, cp1y, 3)
    context.fill(None)
    context.stroke((1, 0, 1, 0.5))
    context.curveTo((cp0x, cp0y), (cp1x, cp1y), (p1x, p1y))
    drawPath(path)
    
    #cp1x = p1x
    #path.moveTo((p0x, p0y))
    connections.append(sorted([current, previous]))
    #print(connections)

def drawClass(name, x, y, color):
    if name in drawnclasses:
        return
        
    pos = (x, y)
    
    while pos in positions:
        px, py = pos
        newx = px + GAP + WBOX
        if newx >= WIDTH - WBOX - GAP:
            newx = X0
            py += HGAP# / 2
        pos = (newx, py)
        
    context.fill(blackColor)
    context.fontSize(TEXTSIZE)

    boxx, boxy = pos
    textx = boxx + P
    texty = boxy + P
    context.stroke(None)
    #color = Color(0.6, 1, 0.6)
    context.fill(color)
    context.roundedRect(boxx, boxy, WBOX, HBOX)
    context.fill(blackColor)

    style = dict(font=font.path, fontSize=TEXTSIZE, textFill=0.1)
    bs = context.newString(name, style=style)
    context.text(bs, (textx, texty))
    drawnclasses[name] = pos
    positions.append(pos)
    
def getColorRange(l):
    colorRange = []
    for i in range(l):
        v = i * 1.0 / l
        c = Color(0.7, 0.7, v)
        colorRange.append(c)
    return colorRange

def drawClasses(inspected):
    classes = []
    
    for _, obj in inspected:
        if inspect.isclass(obj):
            classes.append(obj)

    l = len(classes)
    colorRange = getColorRange(l)
    
    i = 0
    
    for o in classes:
        i = drawClassHierarchy(o, colorRange, i)

context.newPage(pt(WIDTH), pt(HEIGHT))
connections = []
drawnclasses = {}
positions = []
classes = []    
classes.extend(inspect.getmembers(sys.modules['pagebot.conditions']))
drawClasses(classes)

context.fill(0)
context.stroke(None)
context.fontSize(42)
msg = 'PageBot Alignment Conditions '
msg1 = 'Object Hierarchy'
style = dict(font=titleFont.path, fontSize=36, textFill=0.5)
boldStyle = dict(font=boldFont.path, fontSize=36, textFill=0)
bs = context.newString(msg, style=style)
bs += context.newString(msg1, style=boldStyle)
context.text(bs, (100, HEIGHT - 100))
context.saveDrawing('_export/conditionObjectHierarchy.png')
context.saveDrawing('_export/conditionObjectHierarchy.pdf')
