# piker: trading gear for hackers
# Copyright (C) Tyler Goodlet (in stewardship for piker0)

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 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 Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

'''
Anchor funtions for UI placement of annotions.

'''
from typing import Callable

from PyQt5.QtCore import QPointF
from PyQt5.QtWidgets import QGraphicsPathItem

from ._label import Label


def marker_right_points(

    chart: 'ChartPlotWidget',  # noqa
    marker_size: int = 20,

) -> (float, float, float):
    '''Return x-dimension, y-axis-aware, level-line marker oriented scene values.

    X values correspond to set the end of a level line, end of
    a paried level line marker, and the right most side of the "right"
    axis respectively.

    '''
    # TODO: compute some sensible maximum value here
    # and use a humanized scheme to limit to that length.
    l1_len = chart._max_l1_line_len
    ryaxis = chart.getAxis('right')

    r_axis_x = ryaxis.pos().x()
    up_to_l1_sc = r_axis_x - l1_len - 10

    marker_right = up_to_l1_sc - (1.375 * 2 * marker_size)
    line_end = marker_right - (6/16 * marker_size)

    return line_end, marker_right, r_axis_x


def vbr_left(
    label: Label,

) -> Callable[..., float]:
    """Return a closure which gives the scene x-coordinate for the
    leftmost point of the containing view box.

    """
    return label.vbr().left


def right_axis(

    chart: 'ChartPlotWidget',  # noqa
    label: Label,

    side: str = 'left',
    offset: float = 10,
    avoid_book: bool = True,
    # width: float = None,

) -> Callable[..., float]:
    '''Return a position closure which gives the scene x-coordinate for
    the x point on the right y-axis minus the width of the label given
    it's contents.

    '''
    ryaxis = chart.getAxis('right')

    if side == 'left':

        if avoid_book:
            def right_axis_offset_by_w() -> float:

                # l1 spread graphics x-size
                l1_len = chart._max_l1_line_len

                # sum of all distances "from" the y-axis
                right_offset = l1_len + label.w + offset

                return ryaxis.pos().x() - right_offset

        else:
            def right_axis_offset_by_w() -> float:

                return ryaxis.pos().x() - (label.w + offset)

        return right_axis_offset_by_w

    elif 'right':

        # axis_offset = ryaxis.style['tickTextOffset'][0]

        def on_axis() -> float:

            return ryaxis.pos().x()  # + axis_offset - 2

        return on_axis


def gpath_pin(

    gpath: QGraphicsPathItem,
    label: Label,  # noqa

    location_description: str = 'right-of-path-centered',
    use_right_of_pp_label: bool = False,

) -> QPointF:

    # get actual arrow graphics path
    path_br = gpath.mapToScene(gpath.path()).boundingRect()

    # label.vb.locate(label.txt)  #, children=True)

    if location_description == 'right-of-path-centered':
        return path_br.topRight() - QPointF(label.h/16, label.h / 3)

    if location_description == 'left-of-path-centered':
        return path_br.topLeft() - QPointF(label.w, label.h / 6)

    elif location_description == 'below-path-left-aligned':
        return path_br.bottomLeft() - QPointF(0, label.h / 6)

    elif location_description == 'below-path-right-aligned':
        return path_br.bottomRight() - QPointF(label.w, label.h / 6)



def pp_tight_and_right(
    label: Label

) -> QPointF:
    '''Place *just* right of the pp label.

    '''
    txt = label.txt
    return label.txt.pos() + QPointF(label.w - label.h/3, 0)