import collections
import datetime
import sys
import warnings
from typing import Any, Generator, List, Optional, Tuple, Union

import numpy as np
import pyqtgraph as pg
from pyqtgraph import mkBrush, mkPen
from pyqtgraph.graphicsItems.PlotDataItem import PlotDataItem

from qgis.PyQt.QtCore import QRectF
from qgis.PyQt.QtCore import pyqtSignal, QPoint, QPointF, Qt
from qgis.PyQt.QtGui import QColor
from qgis.PyQt.QtWidgets import QAction, QMenu, QSlider, QWidgetAction
from qgis.PyQt.QtWidgets import QApplication
from qgis.PyQt.QtWidgets import QGraphicsRectItem, QGraphicsSceneMouseEvent
from ...plotstyling.plotstyling import PlotStyle, PlotWidgetStyle
from ...unitmodel import datetime64, UnitWrapper
from ...utils import HashablePointF


class SpectralXAxis(pg.AxisItem):

    def __init__(self, *args, **kwds):
        super(SpectralXAxis, self).__init__(*args, **kwds)
        self.setRange(1, 3000)
        self.enableAutoSIPrefix(True)
        self.labelAngle = 0

        self.mDateTimeFormat = '%D'
        self.mUnit: str = ''

    def tickStrings(self, values, scale, spacing):

        if len(values) == 0:
            return []

        if self.mUnit == 'DateTime':
            values64 = datetime64(np.asarray(values))
            v_min, v_max = min(values64), max(values64)
            if v_min < v_max:
                fmt = '%Y'
                for tscale in ['Y', 'M', 'D', 'h', 'm', 's', 'ms']:
                    scale_type = f'datetime64[{tscale}]'
                    rng = v_max.astype(scale_type) - v_min.astype(scale_type)
                    nscale_units = rng.astype(int)
                    if nscale_units > 0:
                        break

                if tscale == 'Y':
                    fmt = '%Y'
                elif tscale == 'M':
                    fmt = '%Y-%m'
                elif tscale == 'D':
                    fmt = '%Y-%m-%d'
                elif tscale == 'h':
                    fmt = '%H:%M'
                elif tscale == 's':
                    fmt = '%H:%M:%S'
                else:
                    fmt = '%S.%f'
                self.mDateTimeFormat = fmt

            strns = []
            for v in values64:
                dt = v.astype(object)
                if isinstance(dt, datetime.datetime):
                    strns.append(dt.strftime(self.mDateTimeFormat))
                else:
                    strns.append('')
            return strns
        else:
            return super(SpectralXAxis, self).tickStrings(values, scale, spacing)

    def setUnit(self, unit: Union[str, UnitWrapper], labelName: str = None):
        """
        Sets the unit of this axis
        :param unit: str
        :param labelName: str, defaults to unit
        """
        if isinstance(unit, UnitWrapper):
            unit = unit.unit
        if not (unit is None or isinstance(unit, str)):
            raise AssertionError
        self.mUnit = unit

        if isinstance(labelName, str):
            self.setLabel(labelName)
        else:
            self.setLabel(unit)

    def unit(self) -> str:
        """
        Returns the unit set for this axis.
        :return:
        """
        return self.mUnit


class SpectralProfilePlotLegend(pg.LegendItem):
    anchorChanged = pyqtSignal(int, int)

    def __init__(self, *args, offset=(-1, 10), max_items=999999, **kwds):
        super().__init__(*args, offset=offset, **kwds)
        self.mMaxItems = int(max_items)

    def anchor(self, *args, **kwds):
        super().anchor(*args, **kwds)

        pt = self.__dict__.get('_GraphicsWidgetAnchor__offset', None)
        if pt:
            self.anchorChanged.emit(int(pt[0]), int(pt[1]))

    def addItem(self, item, name):
        if len(self.items) < self.mMaxItems:
            super().addItem(item, name)


class SpectralProfilePlotItem(pg.PlotItem):
    sigPopulateContextMenuItems = pyqtSignal(object)

    def __init__(self, *args, **kwds):

        viewBox = kwds.get('viewBox', SpectralViewBox())
        axisItems = kwds.get('axisItems', {'bottom': SpectralXAxis(orientation='bottom')})

        super(SpectralProfilePlotItem, self).__init__(*args, viewBox=viewBox, axisItems=axisItems, **kwds)

        # self.addLegend()
        self.mTempList = []

    def spectralProfilePlotDataItems(self):
        for item in self.listDataItems():
            if isinstance(item, SpectralProfilePlotDataItem):
                yield item

    # def addLegend(self, *args, **kwargs) -> SpectralProfilePlotLegend:
    #
    #     if self.legend is None:
    #         self.legend = SpectralProfilePlotLegend(*args, **kwargs)
    #         self.legend.setParentItem(self.vb)
    #
    #         # add existing items
    #         for item in self.items:
    #             if isinstance(item, SpectralProfilePlotDataItem) and item.name() != '':
    #                 self.legend.addItem(item, name=item.name())
    #
    #     return self.legend
    #
    # def removeLegend(self):
    #     if self.legend and self.legend.parentItem():
    #         self.legend.opts['offset'] = None
    #         self.legend.parentItem().removeItem(self.legend)
    #         self.legend = None

    def getContextMenus(self, event) -> List[QMenu]:
        self.mTempList.clear()
        try:
            self.mTempList.append(self.ctrlMenu)
            self.sigPopulateContextMenuItems.emit(self.mTempList)
        except Exception as ex:
            print(ex)
            pass
        return self.mTempList[:]

    def addItems(self, items: list, *args, **kargs):
        """
        Add a graphics item to the view box.
        If the item has plot data (PlotDataItem, PlotCurveItem, ScatterPlotItem), it may
        be included in analysis performed by the PlotItem.
        """
        if len(items) == 0:
            return

        self.items.extend(items)
        vbargs = {}
        if 'ignoreBounds' in kargs:
            vbargs['ignoreBounds'] = kargs['ignoreBounds']
        self.vb.addItems(items, *args, **vbargs)
        # name = None
        refItem = items[0]
        if hasattr(refItem, 'implements') and refItem.implements('plotData'):
            # name = item.name()
            self.dataItems.extend(items)
            # self.plotChanged()

            for item in items:
                self.itemMeta[item] = kargs.get('params', {})
            self.curves.extend(items)

        if isinstance(refItem, pg.PlotDataItem):
            # configure curve for this plot
            (alpha, auto) = self.alphaState()

            for item in items:
                item.setAlpha(alpha, auto)
                item.setFftMode(self.ctrl.fftCheck.isChecked())
                item.setDownsampling(*self.downsampleMode())
                item.setClipToView(self.clipToViewMode())
                item.setPointMode(self.pointMode())

            # Hide older plots if needed
            self.updateDecimation()

            # Add to average if needed
            self.updateParamList()
            if self.ctrl.averageGroup.isChecked() and 'skipAverage' not in kargs:
                self.addAvgCurve(item)

    def removeItems(self, items):
        """
        Remove an item from the internal ViewBox.
        """
        if len(items) == 0:
            return

        for item in items:
            self.items.remove(item)
            if item in self.dataItems:
                self.dataItems.remove(item)

            # self.vb.removeItem(item)
            """Remove an item from this view."""
            try:
                self.vb.addedItems.remove(item)
            except ValueError:
                pass
            scene = self.vb.scene()
            if scene is not None:
                scene.removeItem(item)
            item.setParentItem(None)

            if item in self.curves:
                self.curves.remove(item)

            if self.legend is not None:
                self.legend.removeItem(item)
        # self.updateDecimation()
        # self.updateParamList()


class SpectralViewBox(pg.ViewBox):
    """
    Subclass of PyQgtGraph ViewBox

    """
    sigRectDrawn = pyqtSignal(QRectF, QGraphicsSceneMouseEvent)

    def __init__(self, parent=None):
        """
        Constructor of the CustomViewBox
        """
        super().__init__(parent, enableMenu=True)

        self._selecting = False
        self._p0 = None

        self._rect_item = QGraphicsRectItem()
        self.setSelectionRectangleBrush('select')

    def setSelectionRectangleBrush(self, mode: str = 'select'):

        pen = mkPen('yellow')

        c = QColor('yellow')
        c.setAlpha(140)
        brush = mkBrush(c)

        if mode == 'deselect':
            pen.setStyle(Qt.PenStyle.DashLine)
            # brush.setStyle(Qt.Dense4Pattern)
        self._rect_item.setPen(pen)
        self._rect_item.setBrush(brush)

    def addItems(self, pdis: list, ignoreBounds=False):
        """
        Add multiple QGraphicsItem to this view. The view will include this item when determining how to set its range
        automatically unless *ignoreBounds* is True.
        """
        for i, item in enumerate(pdis):
            if item.zValue() < self.zValue():
                item.setZValue(self.zValue() + 1 + i)

        scene = self.scene()
        if scene is not None and scene is not item.scene():
            for item in pdis:
                scene.addItem(item)  # Necessary due to Qt bug: https://bugreports.qt-project.org/browse/QTBUG-18616
                item.setParentItem(self.childGroup)
        if not ignoreBounds:
            self.addedItems.extend(pdis)
        # self.updateAutoRange()

    def mousePressEvent(self, ev):
        has_shift = ev.modifiers() & Qt.KeyboardModifier.ShiftModifier
        has_ctrl = ev.modifiers() & Qt.Modifier.CTRL
        if ev.button() == Qt.MouseButton.LeftButton and (has_shift or has_ctrl):
            self._selecting = True
            self._p0 = self.mapSceneToView(ev.scenePos())
            rect = QRectF(self._p0, self._p0)

            self._rect_item.setRect(rect)
            if has_shift:
                self.setSelectionRectangleBrush('select')
            elif has_ctrl:
                self.setSelectionRectangleBrush('deselect')
            self.addItem(self._rect_item, ignoreBounds=True)
            QApplication.setOverrideCursor(Qt.CursorShape.CrossCursor)
            ev.accept()

        else:
            super().mousePressEvent(ev)

    def mouseMoveEvent(self, ev):
        if self._selecting and self._rect_item is not None:
            p1 = self._p0
            p2 = self.mapSceneToView(ev.scenePos())
            rect = QRectF(p1, p2).normalized()
            self._rect_item.setRect(rect)
            ev.accept()
        else:
            super().mouseMoveEvent(ev)

    def mouseReleaseEvent(self, ev: QGraphicsSceneMouseEvent):
        if self._selecting and ev.button() == Qt.MouseButton.LeftButton:
            rect = self._rect_item.rect()

            tl = self.mapViewToScene(rect.topLeft())
            lr = self.mapViewToScene(rect.bottomRight())
            srect = QRectF(tl, lr).normalized()

            self.sigRectDrawn.emit(srect, ev)
            # clean up
            self.removeItem(self._rect_item)
            self._selecting = False
            QApplication.restoreOverrideCursor()
            ev.accept()
        else:
            super().mouseReleaseEvent(ev)


MouseClickData = collections.namedtuple('MouseClickData', ['idx', 'xValue', 'yValue', 'pxDistance', 'pdi'])

FEATURE_ID = int
FIELD_INDEX = int
MODEL_NAME = str
X_UNIT = str
PLOT_DATA_KEY = Tuple[FEATURE_ID, FIELD_INDEX, X_UNIT]


def default_selection_style(style: PlotStyle) -> PlotStyle:
    style2: PlotStyle = style.clone()
    style2.setLineWidth(style.lineWidth() + 2)
    return style2


class SpectralProfilePlotDataItem(PlotDataItem):
    """
    A pyqtgraph.PlotDataItem to plot a SpectralProfile
    """

    def __init__(self, *args, **kwds):
        super().__init__(*args, **kwds)

        self.setAcceptedMouseButtons(Qt.MouseButton.LeftButton | Qt.MouseButton.RightButton)

        self.mDefaultStyle: PlotStyle = PlotStyle()
        self.mSelectedStyle = default_selection_style

        self.mIsSelected: bool = False
        self.mVisID: Optional[str] = None
        self.mLayerID: Optional[str] = None
        self.mFeatureID: Optional[int] = None
        self.mField: Optional[str] = None

    def curveIsSelected(self) -> bool:
        return self.mIsSelected

    def setCurveIsSelected(self, b: bool = True):

        if self.mIsSelected == b:
            return

        self.mIsSelected = b
        if b:
            if callable(self.mSelectedStyle):
                selectedStyle = self.mSelectedStyle(self.mDefaultStyle)
            else:
                selectedStyle = self.mSelectedStyle
            if not (isinstance(selectedStyle, PlotStyle)):
                raise AssertionError
            self.setPlotStyle(selectedStyle)
        else:
            self.setPlotStyle(self.mDefaultStyle)

    def selectPoints(self, point_indices):
        pass

    def selectedPoints(self) -> list:

        return []

    def layerID(self) -> Optional[str]:
        return self.mLayerID

    def featureID(self) -> Optional[int]:
        return self.mFeatureID

    def field(self) -> Optional[str]:
        return self.mField

    # On right-click, raise a context menu
    def mouseClickEvent(self, ev):
        if ev.button() == Qt.MouseButton.RightButton:
            if self.raiseContextMenu(ev):
                ev.accept()

    def setProfileData(self,
                       plot_data: dict,
                       plot_style: PlotStyle,
                       showBadBands: bool = True,
                       sortBands: bool = False,
                       zValue: int = None,
                       label: str = None):

        self.mDefaultStyle = plot_style
        y = plot_data.get('y')

        if y is None:
            self.clear()
            return

        x = plot_data.get('x', list(range(len(y))))

        linePen = pg.mkPen(plot_style.linePen)
        symbolPen = pg.mkPen(plot_style.markerPen)
        symbolBrush = pg.mkBrush(plot_style.markerBrush)

        symbol = plot_style.markerSymbol
        symbolSize = plot_style.markerSize

        if isinstance(x[0], (datetime.date, datetime.datetime)):
            x = np.asarray(x, dtype=np.datetime64)

        # replace None by NaN
        x = np.asarray(x, dtype=float)
        y = np.asarray(y, dtype=float)

        if sortBands:
            idx = np.argsort(x)
            x = x[idx]
            y = y[idx]

        if not showBadBands and 'bbl' in plot_data.keys():
            valid = np.array(plot_data['bbl'], dtype=float) > 0
            valid = valid & np.isfinite(valid)
            y = np.where(valid, y, np.nan)

        connect = np.isfinite(x) & np.isfinite(y)

        self.setData(x=x, y=y, z=zValue,
                     name=label,
                     connect=connect,
                     pen=linePen,
                     symbol=symbol,
                     symbolPen=symbolPen,
                     symbolBrush=symbolBrush,
                     symbolSize=symbolSize)

    def setPlotStyle(self, plotStyle: PlotStyle):
        """

        :param plotStyle:
        :return:
        """
        self.opts['pen'] = pg.mkPen(plotStyle.linePen)
        self.opts['symbol'] = plotStyle.markerSymbol
        self.opts['symbolPen'] = pg.mkPen(plotStyle.markerPen)
        self.opts['symbolBrush'] = pg.mkBrush(plotStyle.markerBrush)
        self.opts['symbolSize'] = plotStyle.markerSize
        self.updateItems(styleUpdate=True)

        # if isinstance(pts, pg.ScatterPlotItem):
        #    pdi = pts.parentItem()
        #    if isinstance(pdi, SpectralProfilePlotDataItem):
        #        pt = pts.ptsClicked[0]
        #        i = pt.index()
        #        data = MouseClickData(idx=i, xValue=pdi.xData[i], yValue=pdi.yData[i], pxDistance=0, pdi=self)
        #        self.sigProfileClicked.emit(data)

    def closestDataPoint(self, pos) -> Tuple[int, float, float, float]:
        x = pos.x()
        y = pos.y()
        pw = self.pixelWidth()
        ph = self.pixelHeight()
        _ = []
        dataX, dataY = self.getData()
        distX = np.abs(dataX - x) / pw
        distY = np.abs(dataY - y) / ph

        dist = np.sqrt(distX ** 2 + distY ** 2)
        idx = np.nanargmin(dist)
        return idx, dataX[idx], dataY[idx], dist[idx]

    def plot(self) -> pg.PlotWidget:
        """
        Opens a PlotWindow and plots this SpectralProfilePlotDataItem to
        :return:
        :rtype:
        """
        pw = pg.plot(title=self.name())
        pw.getPlotItem().addItem(self)
        return pw

    def updateItems(self, *args, **kwds):
        if not self.signalsBlocked():
            super().updateItems(*args, **kwds)

    def viewRangeChanged(self, *args, **kwds):
        if not self.signalsBlocked():
            super().viewRangeChanged()

    def setClickable(self, b: bool, width=None):
        """
        :param b:
        :param width:
        :return:
        """
        if not (isinstance(b, bool)):
            raise AssertionError
        self.curve.setClickable(b, width=width)

    def populateContextMenu(self, menu: QMenu):
        pass

    def raiseContextMenu(self, ev):
        menu = self.contextMenu()

        # Let the scene add on to the end of our context menu
        # (this is optional)
        menu = self.scene().addParentContextMenus(self, menu, ev)

        pos = ev.screenPos()
        menu.popup(QPoint(pos.x(), pos.y()))
        return True

    # This method will be called when this item's _children_ want to raise
    # a context menu that includes their parents' menus.
    def contextMenu(self, event=None):

        self.menu = QMenu()
        self.menu.setTitle(self.name + " options..")

        green = QAction("Turn green", self.menu)
        green.triggered.connect(self.setGreen)
        self.menu.addAction(green)
        self.menu.green = green

        blue = QAction("Turn blue", self.menu)
        blue.triggered.connect(self.setBlue)
        self.menu.addAction(blue)
        self.menu.green = blue

        alpha = QWidgetAction(self.menu)
        alphaSlider = QSlider()
        alphaSlider.setOrientation(Qt.Orientation.Horizontal)
        alphaSlider.setMaximum(255)
        alphaSlider.setValue(255)
        alphaSlider.valueChanged.connect(self.setAlpha)
        alpha.setDefaultWidget(alphaSlider)
        self.menu.addAction(alpha)
        self.menu.alpha = alpha
        self.menu.alphaSlider = alphaSlider
        return self.menu


class PlotUpdateBlocker(object):
    """
    A blocker for plot updates
    """

    def __init__(self, plot: pg.PlotWidget):
        isinstance(plot, pg.PlotWidget)
        self.mPlotWidget: pg.PlotWidget = plot

    def __enter__(self):
        plotItem = self.mPlotWidget.getPlotItem()
        legend: pg.LegendItem = plotItem.legend

        if isinstance(legend, pg.LegendItem):
            legend.size = 'dummy'
        plotItem.getViewBox()._updatingRange = True

    def __exit__(self, exc_type, exc_value, tb):

        plotItem = self.mPlotWidget.getPlotItem()
        legend: pg.LegendItem = plotItem.legend

        vb = plotItem.getViewBox()
        vb._updatingRange = False
        vb.updateAutoRange()

        if isinstance(legend, pg.LegendItem):
            legend.size = None
            legend.updateSize()


class SpectralProfilePlotWidget(pg.GraphicsLayoutWidget):
    """
    A widget to plot SpectralProfiles
    """
    sigPlotDataItemSelected = pyqtSignal(SpectralProfilePlotDataItem, Qt.Modifier)

    def __init__(self, *args, **kwargs):

        super().__init__(*args, *kwargs)

        pi1 = SpectralProfilePlotItem()
        pi2 = SpectralProfilePlotItem()
        pi2.vb.setXLink(pi1)
        self.addItem(pi1)
        self.nextRow()
        self.nextRow()
        self.addItem(pi2)
        # vb2 = self.addViewBox()
        # vb2.addItem(self.pi2)

        # self.l1 = l1
        # self.l2 = l2
        self.plotItem = pi1
        self.plotItem1 = pi1
        self.plotItem2 = pi2

        layout = self.ci.layout
        layout.setRowStretchFactor(0, 2)
        layout.setContentsMargins(2, 2, 2, 2)

        self.mCurrentMousePosition: QPointF = QPointF()
        self.setAntialiasing(True)
        self.setAcceptDrops(True)

        self.mCrosshairLineV = pg.InfiniteLine(angle=90, movable=False)
        self.mCrosshairLineH = pg.InfiniteLine(angle=0, movable=False)

        self.mInfoLabelCursor = pg.TextItem(text='<cursor position>', anchor=(1.0, 0.0))
        self.mInfoHover = pg.TextItem(text='', anchor=QPointF(0.0, 0.0))

        self.mCrosshairLineH.pen.setWidth(2)
        self.mCrosshairLineV.pen.setWidth(2)
        self.mCrosshairLineH.setZValue(9999999)
        self.mCrosshairLineV.setZValue(9999999)
        self.mInfoLabelCursor.setZValue(9999999)
        self.mInfoHover.setZValue(9999999)

        self.scene().addItem(self.mInfoLabelCursor)
        self.scene().addItem(self.mInfoHover)
        self.mInfoHover.setParentItem(pi1)
        self.mInfoHover.setPos(50, 0)
        self.mInfoLabelCursor.setParentItem(pi1)

        self.mLegendItem1 = pg.LegendItem(offset=(50, 30))
        self.mLegendItem1.setParentItem(self.plotItem1.getViewBox())

        self.mLegendItem2 = pg.LegendItem(offset=(50, 30))
        self.mLegendItem2.setParentItem(self.plotItem2.getViewBox())

        pi1.addItem(self.mCrosshairLineV, ignoreBounds=True)
        pi1.addItem(self.mCrosshairLineH, ignoreBounds=True)
        # pi1.addItem(self.mInfoScatterPoints)
        self.proxy2D = pg.SignalProxy(self.scene().sigMouseMoved, rateLimit=100, slot=self.onMouseMoved2D)

        # self.mUpdateTimer = QTimer()
        # self.mUpdateTimer.setInterval(500)
        # self.mUpdateTimer.setSingleShot(False)

        self.mMaxInfoLength: int = 30
        self.mShowCrosshair: bool = True
        self.mShowCursorInfo: bool = True

        # activate option "Visible Data Only" for y-axis to ignore a y-value, when the x-value is nan
        # self.setAutoVisible(y=True)

    def legend(self) -> pg.LegendItem:
        return self.mLegendItem1

    def spectralProfilePlotDataItems(self,
                                     is_selected: Optional[bool] = None) \
            -> Generator[SpectralProfilePlotDataItem, Any, None]:
        """
        Returns a generator of SpectralProfilePlotDataItems
        :return:
        """
        if isinstance(is_selected, bool):
            for item in self.plotItem1.spectralProfilePlotDataItems():
                if isinstance(item, SpectralProfilePlotDataItem):
                    if item.curveIsSelected() == is_selected:
                        yield item
        else:
            for item in self.plotItem1.spectralProfilePlotDataItems():
                if isinstance(item, SpectralProfilePlotDataItem):
                    yield item

    def existingInfoScatterPoints(self) -> List[HashablePointF]:
        return [HashablePointF(p.pos()) for p in self.mInfoScatterPoints.points()]

    def setShowCrosshair(self, b: bool):
        if not (isinstance(b, bool)):
            raise AssertionError
        self.mShowCrosshair = b

    def setBackgroundColor(self, color: Union[str, QColor]):
        c = QColor(color)
        self.setBackground(c)

        bgBrush = mkBrush(c)
        c2 = bgBrush.color()
        c2.setAlpha(128)
        bgBrush.setColor(c2)
        self.mInfoHover.fill = bgBrush
        self.mInfoLabelCursor.fill = bgBrush

    def plotItems(self) -> List:
        return [self.plotItem1, self.plotItem2]

    def setForegroundColor(self, color: Union[str, QColor]):
        c = QColor(color)

        # set Foreground color
        self.mInfoLabelCursor.setColor(c)
        for plotItem in self.plotItems():
            for axis in plotItem.axes.values():
                ai: pg.AxisItem = axis['item']
                if isinstance(ai, pg.AxisItem):
                    ai.setPen(c)
                    ai.setTextPen(c)
                    ai.label.setDefaultTextColor(c)

    def setCrosshairColor(self, color: Union[str, QColor]):
        self.mCrosshairLineH.pen.setColor(QColor(color))
        self.mCrosshairLineV.pen.setColor(QColor(color))

    def setSelectionColor(self, color: Union[str, QColor]):
        c = QColor(color)
        # self.mInfoScatterPoints.opts['pen'].setColor(c)
        # self.mInfoScatterPoints.opts['brush'].setColor(c)

        # selection color = hover color
        self.mInfoHover.setColor(c)
        for item in self.spectralProfilePlotDataItems():
            item.scatter.opts['hoverPen'] = mkPen(c)
            item.scatter.opts['hoverBrush'] = mkBrush(c)

    def setInfoColor(self, color: Union[str, QColor]):
        self.mInfoLabelCursor.setColor(QColor(color))

    def setShowCursorInfo(self, b: bool):
        if not (isinstance(b, bool)):
            raise AssertionError
        self.mShowCursorInfo = b

    def xAxis(self) -> SpectralXAxis:
        return self.plotItem1.getAxis('bottom')

    def yAxis(self) -> pg.AxisItem:
        return self.plotItem1.getAxis('left')

    def viewBox(self) -> SpectralViewBox:
        return self.plotItem1.getViewBox()

    def getPlotItem(self):
        return self.plotItem1

    def updatePositionInfo(self):
        x, y = self.mCurrentMousePosition.x(), self.mCurrentMousePosition.y()
        positionInfoHtml = '<html><body>'
        if self.xAxis().mUnit == 'DateTime':
            positionInfoHtml += 'x:{}\ny:{:0.5f}'.format(datetime64(x), y)
        elif self.xAxis().mUnit == 'DOY':
            positionInfoHtml += 'x:{}\ny:{:0.5f}'.format(int(x), y)
        else:
            positionInfoHtml += 'x:{:0.5f}\ny:{:0.5f}'.format(x, y)

        # for pt, v in self.mInfoScatterPointHtml.items():
        #    positionInfoHtml += f'{v}'
        positionInfoHtml += '</body></html>'
        self.mInfoLabelCursor.setHtml(positionInfoHtml)

    def setWidgetStyle(self, style: PlotWidgetStyle):
        warnings.warn(DeprecationWarning(), stacklevel=2)
        self.mInfoLabelCursor.setColor(style.textColor)
        # self.mInfoScatterPoints.opts['pen'].setColor(QColor(style.selectionColor))
        # self.mInfoScatterPoints.opts['brush'].setColor(QColor(style.selectionColor))
        self.mCrosshairLineH.pen.setColor(style.crosshairColor)
        self.mCrosshairLineV.pen.setColor(style.crosshairColor)
        self.setBackground(style.backgroundColor)

        # set Foreground color
        for axis in self.plotItem.axes.values():
            ai: pg.AxisItem = axis['item']
            if isinstance(ai, pg.AxisItem):
                ai.setPen(style.foregroundColor)
                ai.setTextPen(style.foregroundColor)
                ai.label.setDefaultTextColor(style.foregroundColor)

    def leaveEvent(self, ev):
        super().leaveEvent(ev)

        # disable mouse-position related plot items
        self.mCrosshairLineH.setVisible(False)
        self.mCrosshairLineV.setVisible(False)
        self.mInfoLabelCursor.setVisible(False)

    def onMouseMoved2D(self, evt):
        pos = evt[0]  # using signal proxy turns original arguments into a tuple

        plotItem = self.plotItem1
        if not (isinstance(plotItem, SpectralProfilePlotItem)):
            raise AssertionError
        vb = plotItem.vb
        if not (isinstance(vb, SpectralViewBox)):
            raise AssertionError

        if plotItem.sceneBoundingRect().contains(pos) and self.underMouse():
            mousePoint = vb.mapSceneToView(pos)
            self.mCurrentMousePosition = mousePoint

            _ = None
            _ = -1
            _ = sys.float_info.max
            # sx, sy = self.mInfoScatterPoints.getData()

            self.updatePositionInfo()

            s = self.size()
            pos = QPointF(s.width(), 0)
            self.mInfoLabelCursor.setVisible(self.mShowCursorInfo)
            self.mInfoLabelCursor.setPos(pos)

            self.mCrosshairLineH.setVisible(self.mShowCrosshair)
            self.mCrosshairLineV.setVisible(self.mShowCrosshair)
            self.mCrosshairLineV.setPos(mousePoint.x())
            self.mCrosshairLineH.setPos(mousePoint.y())
        else:
            vb.setToolTip('')
            self.mCrosshairLineH.setVisible(False)
            self.mCrosshairLineV.setVisible(False)
            self.mInfoLabelCursor.setVisible(False)
