"""
/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
"""
import os

from qgis.PyQt.uic import loadUiType
from qgis.PyQt.QtCore import Qt, QTimer
from qgis.PyQt.QtGui import QIcon
from qgis.PyQt.QtWidgets import QDockWidget, QAbstractItemView, QTableWidget, QTableWidgetItem
from qgis.core import Qgis, QgsMapLayerProxyModel, QgsFieldProxyModel, QgsWkbTypes, QgsFeatureRequest, QgsCoordinateTransform, QgsProject, QgsRectangle, QgsPoint, QgsPointXY, QgsGeometry
from qgis.gui import QgsRubberBand
from qgis.utils import isPluginLoaded, plugins
from .settings import settings
import traceback

MAX_LIST_SIZE = 5000

FORM_CLASS, _ = loadUiType(os.path.join(
    os.path.dirname(__file__), 'ui/density.ui'))


class HeatmapAnalysis(QDockWidget, FORM_CLASS):
    selected_layer = None
    density_layer = None
    selected_score_field = None

    def __init__(self, iface, parent):
        super(HeatmapAnalysis, self).__init__(parent)
        self.setupUi(self)
        self.canvas = iface.mapCanvas()
        self.iface = iface
        self.clearButton.setIcon(QIcon(':/images/themes/default/mIconClearText.svg'))
        self.dataComboBox.setFilters(QgsMapLayerProxyModel.Filter.PointLayer)
        self.dataComboBox.layerChanged.connect(self.layerChanged)
        self.densityHeatmapComboBox.setFilters(QgsMapLayerProxyModel.Filter.PolygonLayer)
        self.densityHeatmapComboBox.layerChanged.connect(self.layerChanged)
        self.idComboBox.setFilters(QgsFieldProxyModel.Filter.Int | QgsFieldProxyModel.Filter.LongLong)
        self.idComboBox.fieldChanged.connect(self.fieldChanged)
        self.countComboBox.setFilters(QgsFieldProxyModel.Filter.Numeric)
        self.countComboBox.fieldChanged.connect(self.fieldChanged)
        self.zoomComboBox.currentIndexChanged.connect(self.zoomModeChanged)
        self.resultsTable.setEditTriggers(QTableWidget.EditTrigger.NoEditTriggers)
        self.resultsTable.setColumnCount(2)
        self.resultsTable.setSortingEnabled(False)
        self.resultsTable.setHorizontalHeaderLabels(['ID','Score'])
        self.resultsTable.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
        self.resultsTable.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
        self.resultsTable.itemSelectionChanged.connect(self.select_feature)
        # self.resultsTable.itemClicked.connect(self.select_feature)
        self.rb = QgsRubberBand(self.canvas, QgsWkbTypes.GeometryType.LineGeometry)
        self.rb.setColor(settings.line_flash_color)
        self.rb.setWidth(settings.line_flash_width)

    def showEvent(self, e):
        self.layerChanged()

    def closeEvent(self, e):
        layer = self.densityHeatmapComboBox.currentLayer()
        if layer:
            layer.setSubsetString('')
        QDockWidget.closeEvent(self, e)

    def fieldChanged(self, fieldName):
        self.layerChanged()

    def zoomModeChanged(self, index):
        if index == 2:
            self.resultsTable.setSelectionMode(QAbstractItemView.SelectionMode.SingleSelection)
            layer = self.densityHeatmapComboBox.currentLayer()
            if layer:
                layer.setSubsetString('')
        else:
            self.resultsTable.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)

    def layerChanged(self):
        if not self.isVisible():
            return
        reset_results = False
        layer = self.dataComboBox.currentLayer()
        if layer != self.selected_layer:
            reset_results = True
            self.selected_layer = layer
        density_layer = self.densityHeatmapComboBox.currentLayer()
        if density_layer != self.density_layer:
            if density_layer:
                density_layer.setSubsetString('')
            self.idComboBox.blockSignals(True)
            self.idComboBox.setLayer(density_layer)
            self.idComboBox.setField('id')
            self.idComboBox.blockSignals(False)
            self.countComboBox.blockSignals(True)
            self.countComboBox.setLayer(density_layer)
            self.countComboBox.setField('NUMPOINTS')
            self.countComboBox.blockSignals(False)
            reset_results = True
            self.density_layer = density_layer
        
        score_field = self.countComboBox.currentField()
        if score_field != self.selected_score_field:
            reset_results = True
            self.selected_score_field = score_field

        if reset_results:
            self.resultsTable.setRowCount(0)

    def select_feature(self):
        density_layer = self.densityHeatmapComboBox.currentLayer()
        selectedItems = self.resultsTable.selectedItems()
        auto_zoom = self.zoomComboBox.currentIndex()
        if len(selectedItems) == 0:
            if density_layer:
                density_layer.setSubsetString('')
            return
        if auto_zoom != 2:
            id_field_name = self.idComboBox.currentField()
            ids = set()
            for item in selectedItems:
                selectedRow = item.row()
                ids.add(self.resultsTable.item(selectedRow, 0).text())
            ids_str = ",".join(ids)
            exp = '"{}" IN ({})'.format(id_field_name, ids_str)
            density_layer.setSubsetString(exp)
        
        if auto_zoom:
            density_crs = density_layer.crs()
            canvas_crs = self.canvas.mapSettings().destinationCrs()
            xform = QgsCoordinateTransform(density_crs, canvas_crs, QgsProject.instance())
            # center = xform.transform(density_layer.extent()).center()
            # rect = QgsRectangle(center.x(), center.y(), center.x(), center.y())
            if auto_zoom == 1:  # Auto pan to center of selected features
                rect = xform.transform(density_layer.extent())
                center = rect.center()
                rect = QgsRectangle(center.x(), center.y(), center.x(), center.y())
                self.canvas.setExtent(rect)
            elif auto_zoom == 2:  # Pan and flash point
                pt = self.resultsTable.item(selectedItems[0].row(), 0).data(Qt.ItemDataRole.UserRole)
                pt = xform.transform(pt.x(), pt.y())
                rect = QgsRectangle(pt.x(), pt.y(), pt.x(), pt.y())
                self.canvas.setExtent(rect)
                self.highlight(pt)
                self.canvas.refresh()
            else:  # Zoom to selected features
                rect = xform.transform(density_layer.extent())
                self.canvas.setExtent(rect)
                

    def on_applyButton_pressed(self):
        self.resultsTable.setRowCount(0)
        density_layer = self.densityHeatmapComboBox.currentLayer()
        score_field = self.countComboBox.currentField()
        id_field = self.idComboBox.currentField()
        if not id_field or not score_field:
            return

        request=QgsFeatureRequest().addOrderBy(score_field, ascending=False)
        iter = density_layer.getFeatures(request)
        for i, f in enumerate(iter):
            score = f[score_field]
            id = f[id_field]
            self.resultsTable.insertRow(i)
            item = QTableWidgetItem('{}'.format(id))
            item.setData(Qt.ItemDataRole.UserRole, f.geometry().centroid().asPoint())
            self.resultsTable.setItem(i, 0, item)
            try:
                item = QTableWidgetItem('{}'.format(score))
            except:
                item = QTableWidgetItem('')
            self.resultsTable.setItem(i, 1, item)
            if i >= MAX_LIST_SIZE - 1:
                break

    def on_clearButton_pressed(self):
        """"
            This deselects all selected tracks. It doesn't remove them.
        """
        self.resultsTable.clearSelection()

    def highlight(self, point):
        currExt = self.canvas.extent()

        leftPt = QgsPoint(currExt.xMinimum(), point.y())
        rightPt = QgsPoint(currExt.xMaximum(), point.y())

        topPt = QgsPoint(point.x(), currExt.yMaximum())
        bottomPt = QgsPoint(point.x(), currExt.yMinimum())

        horizLine = QgsGeometry.fromPolyline([leftPt, rightPt])
        vertLine = QgsGeometry.fromPolyline([topPt, bottomPt])

        self.rb.reset(QgsWkbTypes.GeometryType.LineGeometry)
        self.rb.setColor(settings.line_flash_color)
        self.rb.setWidth(settings.line_flash_width)
        self.rb.addGeometry(horizLine, None)
        self.rb.addGeometry(vertLine, None)

        QTimer.singleShot(1000, self.resetRubberbands)

    def resetRubberbands(self):
        self.rb.reset()

