# -*- coding: utf-8 -*-

"""
***************************************************************************
    classificationscheme.py

    Methods and Objects to describe raster classifications
    ---------------------
    Date                 : Juli 2017
    Copyright            : (C) 2017 by Benjamin Jakimow
    Email                : benjamin.jakimow@geo.hu-berlin.de
***************************************************************************
    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 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 General Public License for more details.

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

import csv
import json
import os
import pathlib
import re
import sys
import warnings
from pathlib import Path
from typing import Any, List, Optional, Union

import numpy as np
from osgeo import gdal

from qgis.PyQt.QtCore import (
    NULL, pyqtSignal, QAbstractListModel, QAbstractTableModel, QByteArray, QItemSelectionModel,
    QMimeData, QModelIndex, QObject, QSize, Qt, QMetaType)
from qgis.PyQt.QtGui import QBrush, QColor, QIcon, QPixmap
from qgis.PyQt.QtWidgets import (
    QAction, QApplication, QColorDialog, QComboBox, QDialog, QDialogButtonBox, QFileDialog,
    QHBoxLayout, QInputDialog, QMenu, QMessageBox, QPushButton, QTableView, QToolButton, QVBoxLayout, QWidget)
from qgis.PyQt.QtXml import QDomDocument, QDomImplementation
from qgis.core import (
    Qgis, QgsCategorizedSymbolRenderer, QgsField, QgsFillSymbol, QgsLineSymbol, QgsMapLayer,
    QgsMarkerSymbol, QgsPalettedRasterRenderer, QgsProject, QgsProviderRegistry, QgsRasterLayer, QgsRasterRenderer,
    QgsReadWriteContext, QgsRendererCategory, QgsVectorLayer)
from qgis.gui import (
    QgsDialog, QgsEditorConfigWidget, QgsEditorWidgetFactory, QgsEditorWidgetWrapper, QgsGui,
    QgsMapLayerComboBox)
from ..utils import gdalDataset, loadUi, nextColor, registeredMapLayers, stringFromByteArray, stringToByteArray

DEFAULT_UNCLASSIFIEDCOLOR = QColor('black')
DEFAULT_FIRST_COLOR = QColor('#a6cee3')

MIMEDATA_KEY = 'hub-classscheme'
MIMEDATA_KEY_TEXT = 'text/plain'
MIMEDATA_INTERNAL_IDs = 'classinfo_ids'
MIMEDATA_KEY_QGIS_STYLE = QGSCLIPBOARD_STYLE_MIME = 'application/qgis.style'
MAX_UNIQUE_CLASSES = 100
_CLASS_SCHEME_EDITOR_WIDGET_FACTORY = None


def findMapLayersWithClassInfo() -> list:
    """
    Returns QgsMapLayers from which a ClassificationScheme can be derived.
    Searches in all QgsMapLayerStores known to classification.MAP_LAYER_STORES
    :return: [list-of-QgsMapLayer]
    """

    results = []
    for lyr in registeredMapLayers():
        if isinstance(lyr, QgsVectorLayer) and isinstance(lyr.renderer(), QgsCategorizedSymbolRenderer):
            results.append(lyr)
        elif isinstance(lyr, QgsRasterLayer) and isinstance(lyr.renderer(), QgsPalettedRasterRenderer):
            results.append(lyr)
    return results


def hasClassification(pathOrDataset):
    """
    This function tests if a gdal-readable raster data set contains
    categorical information that can be used to retrieve a ClassificationScheme
    :param pathOrDataset: string | gdal.Dataset
    :return: True | False
    """
    ds = None
    try:
        if isinstance(pathOrDataset, gdal.Dataset):
            ds = pathOrDataset
        elif isinstance(pathOrDataset, str):
            ds = gdal.Open(pathOrDataset)
        elif isinstance(pathOrDataset, QgsRasterLayer):
            ds = gdal.Open(pathOrDataset.source())
    except Exception:
        ds = None

    if not isinstance(ds, gdal.Dataset):
        return False

    for b in range(ds.RasterCount):
        band = ds.GetRasterBand(b + 1)
        if band.GetCategoryNames() or band.GetColorTable():
            return True
    return False


def getTextColorWithContrast(c: Union[str, QColor]) -> QColor:
    """
    Returns a QColor with good contrast to c
    :param c: QColor
    :return: QColor
    """
    if not isinstance(c, QColor):
        c = QColor(c)
    if c.lightness() < 0.5:
        return QColor('white')
    else:
        return QColor('black')


class ClassInfo(QObject):
    sigSettingsChanged = pyqtSignal()

    def __init__(self, label=0, name=None, color=None, parent=None):
        super(ClassInfo, self).__init__(parent)

        if name is None:
            name = 'Unclassified' if label == 0 else 'Class {}'.format(label)

        if color is None:
            if label == 0:
                color = DEFAULT_UNCLASSIFIEDCOLOR
            else:
                color = DEFAULT_FIRST_COLOR

        self.mName = name
        self.mLabel = label
        self.mColor = None
        if color:
            self.setColor(color)

    def setLabel(self, label: int):
        """
        Sets the label value.
        :param label: int, must be >= 0
        """
        if not isinstance(label, int) and label >= 0:
            raise Exception('label must be int >= 0')
        self.mLabel = label
        self.sigSettingsChanged.emit()

    def label(self) -> int:
        """
        Returns the class label values
        :return: int
        """
        return self.mLabel

    def color(self) -> QColor:
        """
        Returns the class color.
        :return: QColor
        """
        return QColor(self.mColor)

    def name(self) -> str:
        """
        Returns the class name
        :return: str
        """
        return self.mName

    def setColor(self, color: QColor):
        """
        Sets the class color.
        :param color: QColor
        """
        if not isinstance(color, QColor):
            color = QColor(color)
        if not isinstance(color, QColor):
            raise Exception('color must be a QColor')
        self.mColor = color
        self.sigSettingsChanged.emit()

    def setName(self, name: str):
        """
        Sets thes class name
        :param name: str
        """
        self.mName = str(name)
        self.sigSettingsChanged.emit()

    def pixmap(self, *args) -> QPixmap:
        """
        Returns a QPixmap. Default size is 20x20px
        :param args: QPixmap arguments.
        :return: QPixmap
        """
        if len(args) == 0:
            args = (QSize(20, 20),)

        pm = QPixmap(*args)
        pm.fill(self.mColor)
        return pm

    def icon(self, *args) -> QIcon:
        """
        Returns the class color as QIcon
        :param args: QPixmap arguments
        :return: QIcon
        """
        return QIcon(self.pixmap(*args))

    def clone(self):
        """
        Create a copy of this ClassInfo
        :return: ClassInfo
        """
        return ClassInfo(name=self.mName,
                         color=self.mColor,
                         label=self.mLabel)

    def __hash__(self):
        return hash(id(self))

    def __ne__(self, other):
        return not self.__eq__(other)

    def __eq__(self, other):
        if not isinstance(other, ClassInfo):
            return False
        return other.mName == self.mName \
            and other.mLabel == self.mLabel \
            and other.mColor.getRgb() == self.mColor.getRgb()

    def __repr__(self):
        return 'ClassInfo' + self.__str__()

    def __str__(self):
        return '{} "{}" ({})'.format(self.mLabel, self.mName, self.mColor.name())

    def asMap(self) -> dict:

        return {'label': self.label(),
                'name': self.name(),
                'color': self.color().name()}

    @classmethod
    def fromMap(cls, data: dict) -> 'ClassInfo':
        if not isinstance(data, dict):
            raise Exception('data must be a dict')

        return ClassInfo(label=data.get('label'),
                         name=data.get('name'),
                         color=data.get('color'))

    def json(self) -> str:
        return json.dumps(self.asMap(), ensure_ascii=False)

    def fromJSON(self, jsonString: str):
        try:
            data = json.loads(jsonString)
            return ClassInfo.fromMap(data)
        except (TypeError, ValueError, json.decoder.JSONDecodeError):
            return None


class ClassificationScheme(QAbstractTableModel):
    sigClassesRemoved = pyqtSignal(list)
    # sigClassRemoved = pyqtSignal(ClassInfo, int)
    # sigClassAdded = pyqtSignal(ClassInfo, int)
    sigClassesAdded = pyqtSignal(list)
    sigNameChanged = pyqtSignal(str)

    sigIsEditableChanged = pyqtSignal(bool)

    cLabel = 0
    cName = 1
    cColor = 2

    def __init__(self, name: str = 'Classification', zero_based: bool = False):
        super(ClassificationScheme, self).__init__()
        self.mClasses: List[ClassInfo] = []
        self.mName = name
        self.mIsEditable = True

        self.mZeroBased: bool = zero_based

        self.mColNames = {
            self.cColor: 'Color',
            self.cName: 'Name',
            self.cLabel: 'Label',
        }

    def setIsEditable(self, b: bool):
        """
        Sets if class names and colors can be changed
        :param b: bool
        """
        if b != self.mIsEditable:
            self.mIsEditable = b
            self.dataChanged.emit(self.createIndex(0, 0),
                                  self.createIndex(self.rowCount() - 1, self.columnCount() - 1))
            self.sigIsEditableChanged.emit(self.mIsEditable)

    def isEditable(self) -> bool:
        """
        Returns if class names and colors can be changed.
        :return: bool
        """
        return self.mIsEditable

    def dropMimeData(self,
                     mimeData: QMimeData,
                     action: Qt.DropAction,
                     row: int, column: int,
                     parent: QModelIndex, **kwargs):
        if row == -1:
            row = parent.row()
        if action == Qt.DropAction.MoveAction:
            if MIMEDATA_INTERNAL_IDs in mimeData.formats():
                ids = json.loads(stringFromByteArray(mimeData.data(MIMEDATA_INTERNAL_IDs)))

                classesToBeMoved = [c for c in self if id(c) in ids]
                self.beginResetModel()
                for c in reversed(classesToBeMoved):
                    idx = self.classInfo2index(c)

                    # self.beginMoveRows(QModelIndex(), idx.row(), idx.row(), QModelIndex(), row)
                    del self.mClasses[idx.row()]
                    self.mClasses.insert(row, c)
                    # self.endMoveRows()
                self.endResetModel()
                self._updateLabels()
                return True
        elif action == Qt.DropAction.CopyAction:
            if MIMEDATA_KEY in mimeData.formats():
                cs = ClassificationScheme.fromQByteArray(mimeData.data(MIMEDATA_KEY))
                self.insertClasses(cs[:], row)

        return False

    def mimeData(self, indexes, vector_layer: bool = True, vector_attr: str = None) -> QMimeData:
        """
        Returns class infos as QMimeData.
        :param indexes:
        :return:
        """

        if indexes is None:
            indexes = [self.createIndex(r, 0) for r in range(len(self))]

        classes = [self[idx.row()] for idx in indexes]
        cs = ClassificationScheme()
        cs.insertClasses(classes)
        mimeData = QMimeData()
        mimeData.setData(MIMEDATA_KEY, cs.qByteArray())
        mimeData.setData(
            MIMEDATA_INTERNAL_IDs,
            stringToByteArray(json.dumps([id(c) for c in classes], ensure_ascii=False))
        )
        mimeData.setText(cs.toString())

        featureRenderer = self.featureRenderer()
        rasterRenderer = self.rasterRenderer()

        docType = QDomImplementation().createDocumentType('qgis', 'http://mrcc.com/qgis.dtd', 'SYSTEM')
        doc = QDomDocument(docType)
        root = doc.createElement('qgis')
        root.setAttribute('version', Qgis.QGIS_VERSION)
        doc.appendChild(root)

        if vector_layer:
            root.setAttribute('styleCategories', "Symbology")

            node = featureRenderer.save(doc, QgsReadWriteContext())
            if isinstance(vector_attr, str):
                node.setAttribute('attr', vector_attr)
            root.appendChild(node)
        else:
            pipe = doc.createElement('pipe')
            root.appendChild(pipe)
            rasterRenderer.writeXml(doc, pipe)

        mimeData.setData(MIMEDATA_KEY_QGIS_STYLE, doc.toByteArray())
        mimeData.setText(doc.toString())
        return mimeData

    def mimeTypes(self) -> list:
        """
        Returns a list of supported mimeTypes.
        :return: [list-of-str]
        """
        return [MIMEDATA_KEY, MIMEDATA_INTERNAL_IDs, MIMEDATA_KEY_TEXT]

    def rowCount(self, parent: QModelIndex = None, **kwargs) -> int:
        """
        Returns the number of row / ClassInfos.
        :param parent: QModelIndex
        :return: int
        """
        return len(self.mClasses)

    def columnCount(self, parent: QModelIndex = None, **kwargs) -> int:
        return len(self.mColNames)

    def index2ClassInfo(self, index) -> ClassInfo:
        if isinstance(index, QModelIndex):
            index = index.row()
        return self.mClasses[index]

    def classInfo2index(self, classInfo: ClassInfo) -> QModelIndex:
        row = self.mClasses.index(classInfo)
        return self.createIndex(row, 0)

    def classInfos(self) -> List[ClassInfo]:
        return self.mClasses[:]

    def classInfo(self, label: Any = None, name: str = None) -> ClassInfo:
        """
        Returns the 1st ClassInfo instance that matches a given label or class name
        :param label: the class label to match with
        :param name: the class name to match with
        """
        if label:
            for c in self.classInfos():
                if c.label() == label:
                    return c
        elif name:
            for c in self.classInfos():
                if c.name() == name:
                    return c
        return None

    def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):
        if not index.isValid():
            return None

        col = index.column()
        row = index.row()
        classInfo = self.index2ClassInfo(row)

        if role == Qt.ItemDataRole.DisplayRole:
            if col == self.cLabel:
                return classInfo.label()
            if col == self.cName:
                return classInfo.name()
            if col == self.cColor:
                return classInfo.color().name()

        if role == Qt.ItemDataRole.ForegroundRole:
            if col == self.cColor:
                return QBrush(getTextColorWithContrast(classInfo.color()))

        if role == Qt.ItemDataRole.BackgroundRole:
            if col == self.cColor:
                return QBrush(classInfo.color())

        if role == Qt.ItemDataRole.AccessibleTextRole:
            if col == self.cLabel:
                return str(classInfo.label())
            if col == self.cName:
                return classInfo.name()
            if col == self.cColor:
                return classInfo.color().name()

        if role == Qt.ItemDataRole.ToolTipRole:
            if col == self.cLabel:
                return 'Class label "{}"'.format(classInfo.label())
            if col == self.cName:
                return 'Class name "{}"'.format(classInfo.name())
            if col == self.cColor:
                return 'Class color "{}"'.format(classInfo.color().name())

        if role == Qt.ItemDataRole.EditRole:
            if col == self.cLabel:
                return classInfo.label()
            if col == self.cName:
                return classInfo.name()
            if col == self.cColor:
                return classInfo.color()

        if role == Qt.ItemDataRole.UserRole:
            return classInfo

        return None

    def supportedDragActions(self):
        return Qt.DropAction.MoveAction

    def supportedDropActions(self):
        return Qt.DropAction.MoveAction | Qt.DropAction.CopyAction

    def setData(self, index: QModelIndex, value, role: int):
        if not index.isValid():
            return False

        col = index.column()
        row = index.row()
        classInfo = self.index2ClassInfo(row)
        b = False
        if role == Qt.ItemDataRole.EditRole:
            if col == self.cLabel:
                classInfo.setLabel(int(value))
                b = True
            if col == self.cName:
                classInfo.setName(value)
                b = True
            if col == self.cColor:
                classInfo.setColor(value)
                b = True
        if b:
            self.dataChanged.emit(index, index, [role])
        return False

    def flags(self, index: QModelIndex):
        if not index.isValid():
            return Qt.ItemFlag.NoItemFlags
        col = index.column()

        flags = Qt.ItemFlag.ItemIsSelectable | Qt.ItemFlag.ItemIsEnabled
        if self.mIsEditable:
            flags |= Qt.ItemFlag.ItemIsDragEnabled | Qt.ItemFlag.ItemIsDropEnabled
            if self.isEditable():
                if col == self.cLabel and not self.mZeroBased:
                    flags |= Qt.ItemFlag.ItemIsEditable
                elif col == self.cName:
                    flags |= Qt.ItemFlag.ItemIsEditable
        return flags

    def headerData(self, section: int, orientation: Qt.Orientation, role: int = Qt.ItemDataRole.DisplayRole):

        if role == Qt.ItemDataRole.DisplayRole:
            if orientation == Qt.Orientation.Horizontal:
                return self.mColNames[section]

        return super(ClassificationScheme, self).headerData(section, orientation, role)

    def setName(self, name: str = '') -> str:
        """
        Sets ClassificationScheme name
        :param name: str
        :return: str, the name
        """
        b = name != self.mName
        self.mName = name
        if b:
            self.sigNameChanged.emit(self.mName)
        return self.mName

    def name(self) -> str:
        """
        Returns the ClassificationScheme name
        :return:
        """
        return self.mName

    @classmethod
    def fromMap(cls, data: dict) -> Optional['ClassificationScheme']:

        try:
            cs = ClassificationScheme(name=data['name'])
            classes = []
            for classData in data['classes']:
                c = ClassInfo.fromMap(classData)
                classes.append(c)
            cs.insertClasses(classes)
            return cs
        except Exception as ex:
            print(ex, file=sys.stderr)
            return None

    def asMap(self) -> dict:

        data = {'name': self.mName,
                'classes': [c.asMap() for c in self]
                }
        return data

    def json(self) -> str:
        """
        Returns a JSON string of this ClassificationScheme which can be deserialized
        with ClassificationScheme.fromJSON()
        :return: str, JSON string
        """
        return json.dumps(self.asMap(), ensure_ascii=False)

    def qByteArray(self) -> QByteArray:
        """
        Serializes this ClassicationScheme as QByteArray.
        Can be deserialized with ClassificationScheme.fromQByteArray()
        :return: QByteArray
        """
        return stringToByteArray(self.json())

    @classmethod
    def fromQByteArray(cls, array: QByteArray):
        txt = stringFromByteArray(array)
        return cls.fromJson(txt)

    @classmethod
    def fromFile(cls, p: str):
        try:
            if os.path.isfile(p):
                if p.endswith('.json'):
                    jsonStr = None
                    with open(p, 'r') as f:
                        jsonStr = f.read()
                    return cls.fromJson(jsonStr)
                elif p.endswith('.csv'):
                    return cls.fromCsv(p)
                elif p.endswith('.qml'):
                    return cls.fromQml(p)
        except Exception as ex:
            print(ex, file=sys.stderr)
        return None

    @classmethod
    def fromJson(cls, jsonStr: str) -> Optional['ClassificationScheme']:
        try:
            data = json.loads(jsonStr)
            return cls.fromMap(data)
        except Exception as ex:
            print(ex, file=sys.stderr)
            return None

    def rasterRenderer(self, band: int = 1) -> QgsPalettedRasterRenderer:
        """
        Returns the ClassificationScheme as QgsPalettedRasterRenderer
        :return: ClassificationScheme
        """
        classes = []
        for classInfo in self:
            qgsClass = QgsPalettedRasterRenderer.Class(
                int(classInfo.label()),
                color=classInfo.color(),
                label=classInfo.name())
            classes.append(qgsClass)
        renderer = QgsPalettedRasterRenderer(None, band, classes)
        return renderer

    @classmethod
    def fromRasterRenderer(cls, renderer: QgsRasterRenderer):
        """
        Extracts a ClassificatonScheme from a QgsRasterRenderer
        :param renderer: QgsRasterRenderer
        :return: ClassificationScheme
        """
        if not isinstance(renderer, QgsPalettedRasterRenderer):
            return None

        classes = []
        for qgsClass in renderer.classes():
            classInfo = ClassInfo(label=qgsClass.value,
                                  name=qgsClass.label,
                                  color=QColor(qgsClass.color))
            classes.append(classInfo)

        if len(classes) == 0:
            return None

        cs = ClassificationScheme()
        cs.insertClasses(classes)

        return cs

    def featureRenderer(self,
                        symbolType: Union[QgsMarkerSymbol, QgsFillSymbol, QgsLineSymbol] = QgsFillSymbol
                        ) -> QgsCategorizedSymbolRenderer:
        """
        Returns the ClassificationScheme as QgsCategorizedSymbolRenderer
        :return: ClassificationScheme
        """

        r = QgsCategorizedSymbolRenderer(self.name(), [])

        for c in self:
            c: ClassInfo
            try:
                symbol = symbolType()
            except TypeError:
                symbol = symbolType(None)
            symbol.setColor(QColor(c.color()))
            cat = QgsRendererCategory()
            cat.setValue(c.label())
            cat.setSymbol(symbol)
            cat.setLabel(c.name())
            cat.setRenderState(True)
            r.addCategory(cat)
        return r

    @classmethod
    def fromFeatureRenderer(cls, renderer: QgsCategorizedSymbolRenderer):
        """
        Extracts a ClassificationScheme from a QgsCategorizedSymbolRenderer
        :param renderer: QgsCategorizedSymbolRenderer
        :return: ClassificationScheme
        """
        if not isinstance(renderer, QgsCategorizedSymbolRenderer):
            return None

        # move a None element to first position

        no_data_class = None
        classes = []
        class_values = []
        for cat in renderer.categories():
            cat: QgsRendererCategory
            name = cat.label()
            value = cat.value()
            color = QColor(cat.symbol().color())
            class_values.append(value)
            if name.lower().strip() in [None, '', 'unclassified'] and value in [None, '', 0]:
                if no_data_class is None:
                    no_data_class = ClassInfo(0, name, color)
            else:
                c = ClassInfo(value, name, color)
                classes.append(c)

        if len(classes) == 0 and no_data_class is None:
            return None

        if no_data_class is None:
            no_data_class = ClassInfo(0, 'Unclassified', QColor('black'))

        classes.insert(0, no_data_class)
        cs = ClassificationScheme()
        cs.insertClasses(classes)
        return cs

    def clear(self):
        """
        Removes all ClassInfos
        """
        self.beginRemoveColumns(QModelIndex(), 0, self.rowCount() - 1)
        removed = self.mClasses[:]
        del self.mClasses[:]
        self.endRemoveRows()
        self.sigClassesRemoved.emit(removed)

    def clone(self):
        return self.copy()

    def copy(self):
        """
        Create a copy of this ClassificationScheme
        :return:
        """
        cs = ClassificationScheme()
        classes = [c.clone() for c in self.mClasses]
        cs.insertClasses(classes, 0)
        return cs

    def __getitem__(self, slice):
        return self.mClasses[slice]

    def __delitem__(self, slice):
        classes = self[slice]
        self.removeClasses(classes)

    def __contains__(self, item):
        return item in self.mClasses

    def __len__(self):
        return len(self.mClasses)

    def __iter__(self):
        return iter(self.mClasses)

    def __ne__(self, other):
        return not self.__eq__(other)

    def __eq__(self, other):
        if not (isinstance(other, ClassificationScheme) and len(self) == len(other)):
            return False
        return all(self[i] == other[i] for i in range(len(self)))

    def __str__(self):
        return self.__repr__() + '{} classes'.format(len(self))

    def range(self):
        """
        Returns the class label range (min,max).
        """
        labels = self.classLabels()
        return min(labels), max(labels)

    def classNames(self):
        """
        Returns all class names.
        :return: [list-of-class-names (str)]
        """
        return [c.name() for c in self.mClasses]

    def classColors(self):
        """
        Returns all class color.
        :return: [list-of-class-colors (QColor)]
        """
        return [QColor(c.color()) for c in self.mClasses]

    def classLabels(self) -> list:
        """
        Returns the list of class labels [0,...,n-1]
        :return: [list-of-int]
        """
        return [c.label() for c in self.mClasses]

    def classColorArray(self) -> np.ndarray:
        """
        Returns the RGBA class-colors as array
        :return: numpy.ndarray([nClasses,4])
        """
        return np.asarray([c.color().getRgb() for c in self])

    def gdalColorTable(self) -> gdal.ColorTable:
        """
        Returns the class colors as GDAL Color Table
        :return: gdal.Colortable
        """
        ct = gdal.ColorTable()
        for i, c in enumerate(self):
            c: ClassInfo
            ct.SetColorEntry(i, c.mColor.getRgb())
        return ct

    def _updateLabels(self):
        """
        Assigns class labels according to the ClassInfo position
        in case the ClassificationScheme is zero-based
        """
        if not self.mZeroBased:
            return

        for i, c in enumerate(self.mClasses):
            c.mLabel = i
        self.dataChanged.emit(self.createIndex(0, 0),
                              self.createIndex(self.rowCount() - 1, 0),
                              [Qt.ItemDataRole.DisplayRole, Qt.ItemDataRole.ToolTipRole])

    def removeClasses(self, classes):
        """
        Removes as single ClassInfo or a list of ClassInfos.
        :param classes: ClassInfo or [list-of-ClassInfo-to-remove]
        :returns: [list-of-removed-ClassInfo]
        """
        if isinstance(classes, ClassInfo):
            classes = [classes]

        removedIndices = []
        for c in classes:
            if c in self.mClasses:
                removedIndices.append(self.mClasses.index(c))

        removedIndices = list(reversed(sorted(removedIndices)))
        removedClasses = []
        for i in removedIndices:
            c = self.mClasses[i]
            self.beginRemoveRows(QModelIndex(), i, i)
            self.mClasses.remove(c)
            removedClasses.append(c)
            self.endRemoveRows()
        self._updateLabels()
        self.sigClassesRemoved.emit(removedClasses)

    def createClasses(self, n: int):
        """
        Creates n new classes with default an default initialization.
        Can be used to populate the ClassificationScheme.
        :param n: int, number of classes to add.
        """
        if not isinstance(n, int) and n >= 0:
            raise Exception('n must be >=0 ')

        classes = []

        if len(self) > 0:
            nextCol = nextColor(self[-1].color())
        else:
            nextCol = DEFAULT_FIRST_COLOR

        for i in range(n):
            j = len(self) + i
            if j == 0:
                color = QColor('black')
                name = 'Unclassified'
            else:
                color = QColor(nextCol)
                nextCol = nextColor(nextCol)
                name = 'Class {}'.format(j)
            classes.append(ClassInfo(name=name, color=color, label=j))
        self.insertClasses(classes)

    def addClasses(self, classes, index=None):
        warnings.warn('use insertClasses()', DeprecationWarning, stacklevel=2)
        self.insertClasses(classes, index=index)

    def insertClasses(self, classes: Union[List[ClassInfo], ClassInfo], index=None):
        """
        Adds / inserts a list of ClassInfos
        :param classes: [list-of-ClassInfo]
        :param index: int, index to insert the first of the new classes.
                           defaults to len(ClassificationScheme)

        """
        if isinstance(classes, ClassInfo):
            classes = [ClassInfo]
        if not isinstance(classes, list):
            raise AssertionError('classes must be a list of ClassInfo')
        if len(classes) == 0:
            return

        for c in classes:
            if not isinstance(c, ClassInfo):
                raise AssertionError(f'Not a ClassInfo {c}')

            if id(c) in [id(c) for c in self.mClasses]:
                raise AssertionError(
                    'You cannot add the same ClassInfo instance to a ClassificationScheme twice. Create a copy first.')

        if index is None:
            # default: add new classes to end of list
            index = len(self.mClasses)
        # negative index? insert to beginning
        index = max(index, 0)

        self.beginInsertRows(QModelIndex(), index, index + len(classes) - 1)
        for i, c in enumerate(classes):
            index = index + i
            # c.sigSettingsChanged.connect(self.onClassInfoSettingChanged)
            self.mClasses.insert(index, c)
        self.endInsertRows()
        self._updateLabels()
        self.sigClassesAdded.emit(classes)

    def classIndexFromValue(self, value, matchSimilarity=False) -> int:
        """
        Get a values and returns the index of ClassInfo that matches best to.
        :param value: any
        :return: int
        """
        classNames = self.classNames()
        i = -1

        # 1. match on identity
        if isinstance(value, (int, float)):
            i = int(value)

        elif isinstance(value, str):
            if value in classNames:
                i = classNames.index(value)

        # 2. not found? match on similarity
        if i == -1 and matchSimilarity is True:
            if isinstance(value, (int, float)):
                pass

            elif isinstance(value, str):
                if value in classNames:
                    i = classNames.index(value)
            pass
        return i

    def classFromValue(self, value, matchSimilarity=False) -> Optional[ClassInfo]:
        i = self.classIndexFromValue(value, matchSimilarity=matchSimilarity)
        if i != -1:
            return self[i]
        else:
            return None

    def addClass(self, c, index=None):
        warnings.warn('Use insert class', DeprecationWarning, stacklevel=2)

    def insertClass(self, c: ClassInfo, index=None):
        """
        Adds a ClassInfo
        :param c: ClassInfo
        :param index: int, index to add the ClassInfo. Defaults to the end.
        """
        self.insertClasses([c], index=index)

    def saveToRasterBand(self, band: gdal.Band):
        """
        Saves the ClassificationScheme to the gdal.Band.
        ClassInfo names are stored by gdal.Band.SetCategoryNames and colors as gdal.ColorTable.
        :param band: gdal.Band
        """

        ct = gdal.ColorTable()
        cat = []

        for i, classInfo in enumerate(self.mClasses):
            c: QColor = classInfo.mColor
            cat.append(classInfo.mName)
            rgba = (c.red(), c.green(), c.blue(), c.alpha())
            ct.SetColorEntry(i, rgba)
        if len(cat) == 0:
            ct = None

        def equalColorTables(ct1: gdal.ColorTable, ct2: gdal.ColorTable) -> bool:
            if ct1 is None and ct2 is None:
                return True
            elif isinstance(ct1, gdal.ColorTable) and isinstance(ct2, gdal.ColorTable):
                if ct1.GetCount() != ct2.GetCount():
                    return False
                for c in range(ct1.GetCount()):
                    if ct1.GetColorEntry(c) != ct2.GetColorEntry(c):
                        return False
                return True
            else:
                return False

        try:
            band.SetCategoryNames(cat)
        except Exception as ex:
            print(ex, file=sys.stderr)

        try:
            if not equalColorTables(ct, band.GetColorTable()):
                band.SetColorTable(ct)
        except Exception as ex:
            print(ex, file=sys.stderr)

    def saveToRaster(self, raster: Union[str, gdal.Dataset, QgsRasterLayer], bandIndex=0):
        """
        Saves this ClassificationScheme to an raster image
        :param raster: path (str) of raster image or gdal.Dataset instance
        :param bandIndex: band index of raster band to set this ClassificationScheme.
                          Defaults to 0 = the first band
        """
        from ..utils import gdalDataset
        ds = gdalDataset(raster)
        if not (isinstance(ds, gdal.Dataset) and ds.RasterCount > bandIndex):
            raise Exception('Unable to retrieve gdal.Dataset')
        band = ds.GetRasterBand(bandIndex + 1)
        self.saveToRasterBand(band)
        ds.FlushCache()

    def toString(self, sep=';') -> str:
        """
        A quick dump of all ClassInfos
        :param sep: value separator, ';' by default
        :return: str
        """
        lines = ['ClassificationScheme("{}")'.format(self.name())]
        lines += [sep.join(['label', 'name', 'color'])]
        for classInfo in self.mClasses:
            c = classInfo.color()
            info = [classInfo.label(), classInfo.name(), c.name()]
            info = ['{}'.format(v) for v in info]
            lines.append(sep.join(info))
        return '\n'.join(lines)

    def saveToCsv(self, path: str, sep: str = ';', mode: str = None) -> str:
        """
        Saves the ClassificationScheme as CSV table.
        :param path: str, path of CSV file
        :param sep: separator (';' by default)
        :returns: the path of written file (if something was written)
        """
        if mode is None:
            lines = self.toString(sep=sep)
            with open(path, 'w') as f:
                f.write(lines)

            return path

        return None

    def saveToJson(self, path: str, mode: str = None) -> str:
        """
        Save the ClassificationScheme as JSON file.
        :param path: str, path of JSON file
        :return: path of written file
        """
        if mode is None:
            lines = self.json()
            with open(path, 'w') as f:
                f.write(lines)
            return path

        return None

    @classmethod
    def create(cls, n):
        """
        Create a ClassificationScheme with n classes (including 'Unclassified' with label = 0)
        :param n: number of classes including 'Unclassified'
        :return: ClassificationScheme
        """
        s = ClassificationScheme()
        s.createClasses(n)
        return s

    @classmethod
    def fromMimeData(cls, mimeData: QMimeData):

        if not isinstance(mimeData, QMimeData):
            return None

        if MIMEDATA_KEY in mimeData.formats():
            ba = cls.fromQByteArray(mimeData.data(MIMEDATA_KEY))
            if isinstance(ba, ClassificationScheme):
                return ba

        if MIMEDATA_KEY_QGIS_STYLE in mimeData.formats():

            ba = mimeData.data(MIMEDATA_KEY_QGIS_STYLE)
            doc = QDomDocument()
            doc.setContent(ba)
            # print(doc.toString())
            rasterRenderers = doc.elementsByTagName('rasterrenderer')
            for i in range(rasterRenderers.count()):
                node = rasterRenderers.at(i).toElement()
                if node.attribute('type') == 'paletted':
                    return ClassificationScheme.fromRasterRenderer(QgsPalettedRasterRenderer.create(node, None))

            vectorRenderers = doc.elementsByTagName('renderer-v2')
            for i in range(vectorRenderers.count()):
                node = vectorRenderers.at(i).toElement()
                if node.attribute('type') == 'categorizedSymbol':
                    context = QgsReadWriteContext()
                    return ClassificationScheme.fromFeatureRenderer(QgsCategorizedSymbolRenderer.create(node, context))

        if MIMEDATA_KEY_TEXT in mimeData.formats():
            ba = cls.fromQByteArray(mimeData.data(MIMEDATA_KEY_TEXT))
            if isinstance(ba, ClassificationScheme):
                return ba

        return None

    @classmethod
    def fromUniqueFieldValues(cls, layer: QgsVectorLayer, fieldIndex):
        scheme = None

        if not isinstance(layer, QgsVectorLayer):
            return scheme

        if isinstance(fieldIndex, str):
            fieldIndex = layer.fields().indexFromName(fieldIndex)
        elif isinstance(fieldIndex, QgsField):
            fieldIndex = layer.fields().indexFromName(fieldIndex.name())

        if not isinstance(fieldIndex, int) and fieldIndex >= 0 and fieldIndex < layer.fields().count():
            return scheme

        field = layer.fields().at(fieldIndex)

        values = sorted(layer.uniqueValues(fieldIndex, limit=MAX_UNIQUE_CLASSES))
        values = [v for v in values if v not in [None, NULL]]
        if len(values) > 0:
            scheme = ClassificationScheme()
            scheme.insertClass(ClassInfo(0, 'unclassified'))
            if field.isNumeric():
                for v in values:
                    scheme.insertClass(ClassInfo(int(v), name=str(v)))
            else:
                for i, v in enumerate(values):
                    scheme.insertClass(ClassInfo(i + 1, name=str(v)))

        return scheme

    @classmethod
    def fromMapLayer(cls, layer: QgsMapLayer):
        """
        :param layer:
        :return:
        """
        scheme = None
        if not (isinstance(layer, QgsMapLayer) and layer.isValid()):
            return scheme

        if isinstance(layer, QgsRasterLayer):
            scheme = cls.fromRasterRenderer(layer.renderer())
            if not isinstance(scheme, ClassificationScheme):
                if layer.dataProvider().name() == 'gdal':
                    scheme = cls.fromRasterImage(layer.source())

        if isinstance(layer, QgsVectorLayer):
            scheme = cls.fromFeatureRenderer(layer.renderer())

        return scheme

    @classmethod
    def fromRasterBand(cls, band: gdal.Band):
        """
        Reads the ClassificationScheme of a gdal.Band
        :param band: gdal.Band
        :return: ClassificationScheme, None if classes are undefined.
        """
        ct = band.GetColorTable()
        cat = band.GetCategoryNames()
        if not isinstance(cat, list) or len(cat) == 0:
            return None
        scheme = ClassificationScheme()
        classes = []
        for i, catName in enumerate(cat):
            classInfo: ClassInfo = ClassInfo(name=catName, label=i)
            if ct is not None:
                gdal_color = ct.GetColorEntry(i)
                if gdal_color in [None, NULL]:
                    gdal_color = (0, 0, 0, 0)
                classInfo.setColor(QColor(*gdal_color))
            classes.append(classInfo)

        if len(classes) == 0:
            return None

        scheme.insertClasses(classes)
        return scheme

    @classmethod
    def fromRasterImage(cls, path, bandIndex=None):
        """
        Reads a ClassificationScheme from a gdal.Dataset
        :param path: str with path to gdal.Dataset or gdal.Dataset instances
        :param bandIndex: int with band index
        :return: ClassificationScheme
        """
        ds = gdalDataset(path)
        if not isinstance(ds, gdal.Dataset):
            return None

        if bandIndex is None:
            for b in range(ds.RasterCount):
                band = ds.GetRasterBand(b + 1)
                cat = band.GetCategoryNames()

                if cat is not None:
                    bandIndex = b
                    break

            if bandIndex is None:
                return None

        band = ds.GetRasterBand(bandIndex + 1)
        return ClassificationScheme.fromRasterBand(band)

    @classmethod
    def fromCsv(cls, pathCSV: str, mode: str = None):
        """
        Read the ClassificationScheme from a CSV table
        :param path: str, path of CSV file
        :return: ClassificationScheme
        """
        text = None
        with open(pathCSV) as f:
            text = f.read()
        if not isinstance(text, str):
            raise Exception('Unable to read {}'.format(pathCSV))

        lines = text.splitlines()
        lines = [line.strip() for line in lines]
        lines = [line for line in lines if len(line) > 0]
        if len(lines) <= 1:
            raise Exception('CSV does not contain enough values')

        matches = re.search(r'ClassificationScheme\("(.*)"\)', text)
        if matches:
            name = re.search(r'ClassificationScheme\("(.*)"\)', text).group(1)
        else:
            name = 'Classification'

        # b = False
        # columnNames = None
        delimiter = ';'
        for i, line in enumerate(lines):
            matches = re.search(r'^[ ]*(?P<label>label)[ ]*[;\t,][ ]*(?P<name>name)[ ]*([;\t,][ ]*(?P<color>color))?',
                                line, re.IGNORECASE)
            if matches:
                delimiter = re.search(r'[;\t,]', line).group()
                break

        if not matches:
            raise Exception('Missing column header "label;name:color"')

        # cName = matches.group('name')
        # cColor = matches.group('color')
        fieldnames = [matches.group('label'), matches.group('name'), matches.group('color')]

        cs = ClassificationScheme()
        cs.setName(name)
        # read CSV data
        reader = csv.DictReader(lines[i:], delimiter=delimiter)

        iLabel = None
        iName = None
        iColor = None
        for i, name in enumerate(reader.fieldnames):
            if iName is None and re.search(r'name', name, re.I):
                iName = i
            if iColor is None and re.search(r'color', name, re.I):
                iColor = i
            if iLabel is None and re.search(r'label', name, re.I):
                iLabel = i
        rows = [row for row in reader]

        nc = len(rows)
        if nc == 0:
            return None

        cs = cls.create(nc)
        for i, row in enumerate(rows):
            c: ClassInfo = cs[i]
            if iLabel is not None:
                c.setLabel(int(row[fieldnames[iLabel]]))
            if iName is not None:
                c.setName(row[fieldnames[iName]])
            if iColor is not None:
                colorValue = row[fieldnames[iColor]].strip()

                matches = re.search(r'^(?P<R>\d+),(?P<G>\d+),(?P<B>\d+)(,(?P<A>\d+))?$', colorValue)
                if matches:
                    R = int(matches.group('R'))
                    G = int(matches.group('G'))
                    B = int(matches.group('B'))
                    A = matches.group('B')
                    if A:
                        A = int(A)
                    c.setColor(QColor(R, G, B, A))
                else:
                    c.setColor(QColor(colorValue))

        return cs

    def saveToQml(self, path):
        """
        Saves the class infos into a QML file
        :param path: str, path of QML file
        """
        raise NotImplementedError()

    @classmethod
    def fromQml(cls, path: Union[Path, str]):
        """
        Reads a ClassificationScheme from a QML file.
        :param path: str, path to QML file
        :return: ClassificationScheme
        """
        path = Path(path)
        raise NotImplementedError()


class ClassificationSchemeComboBoxModel(QAbstractListModel):

    def __init__(self):
        super(ClassificationSchemeComboBoxModel, self).__init__()

        self.mClassScheme = None
        self.mAllowEmptyField = False

    def setAllowEmptyField(self, b: bool):
        changed = self.mAllowEmptyField != b

        if changed:
            if b:
                self.beginInsertRows(QModelIndex(), 0, 0)
                self.mAllowEmptyField = b
                self.endInsertRows()
            else:
                self.beginRemoveRows(QModelIndex(), 0, 0)
                self.mAllowEmptyField = b
                self.endRemoveRows()

    def allowEmptyField(self) -> bool:
        return self.mAllowEmptyField

    def setClassificationScheme(self, classScheme: ClassificationScheme):
        self.beginResetModel()
        self.mClassScheme = classScheme
        self.endResetModel()

    def classificationScheme(self) -> ClassificationScheme:
        return self.mClassScheme

    def rowCount(self, parent) -> int:
        if not isinstance(self.mClassScheme, ClassificationScheme):
            return 0

        n = len(self.mClassScheme)
        if self.allowEmptyField():
            n += 1
        return n

    def columnCount(self, parent: QModelIndex):
        return 1

    def idx2csIdx(self, index: QModelIndex):

        if not isinstance(self.mClassScheme, ClassificationScheme):
            return QModelIndex()

        if self.allowEmptyField():
            return self.mClassScheme.createIndex(index.row() - 1, index.column())
        else:
            return self.mClassScheme.createIndex(index.row(), index.column())

    def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):

        if not index.isValid():
            return None

        if self.allowEmptyField() and index.row() == 0:
            if role == Qt.ItemDataRole.DisplayRole:
                return ''

        idxCs = self.idx2csIdx(index)
        if not idxCs.isValid():
            return None
        else:

            classInfo = self.mClassScheme.data(idxCs, role=Qt.ItemDataRole.UserRole)
            if not isinstance(classInfo, ClassInfo):
                raise Exception(f'wrong type: {classInfo}')
            if role == Qt.ItemDataRole.UserRole:
                return classInfo
            nCols = self.mClassScheme.columnCount(idxCs)
            if role in [Qt.ItemDataRole.DisplayRole, Qt.ItemDataRole.ToolTipRole, Qt.ItemDataRole.WhatsThisRole]:
                infos = []
                for col in range(nCols):
                    idx = self.mClassScheme.createIndex(idxCs.row(), col)
                    infos.append(str(self.mClassScheme.data(idx, role=role)))
                if role == Qt.ItemDataRole.DisplayRole:
                    return ' '.join(infos[0:2])
                if role == Qt.ItemDataRole.ToolTipRole:
                    return '\n'.join(infos)
                if role == Qt.ItemDataRole.WhatsThisRole:
                    return '\n'.join(infos)

            elif role == Qt.ItemDataRole.DecorationRole:
                return classInfo.icon()

        return None

    def flags(self, index: QModelIndex):
        if not index.isValid():
            return None
        return Qt.ItemFlag.ItemIsEnabled | Qt.ItemFlag.ItemIsSelectable


class ClassificationMapLayerComboBox(QgsMapLayerComboBox):
    """
    A QgsMapLayerComboBox that shows map layers from which a ClassificationScheme can be derived.
    """

    def __init__(self, parent=None):
        super(ClassificationMapLayerComboBox, self).__init__(parent)

        self.updateExceptedLayerList()

    def updateExceptedLayerList(self):

        toExclude = [lyr for lyr in QgsProject.instance().mapLayers().values()
                     if ClassificationScheme.fromMapLayer(lyr) is None
                     ]
        self.setExceptedLayerList(toExclude)

    def currentClassification(self) -> ClassificationScheme:
        lyr = self.currentLayer()
        if isinstance(lyr, QgsMapLayer):
            return ClassificationScheme.fromMapLayer(self.currentLayer())
        else:
            return None


class ClassificationSchemeComboBox(QComboBox):

    def __init__(self, parent=None, classification: ClassificationScheme = None):
        super(ClassificationSchemeComboBox, self).__init__(parent)
        if not isinstance(classification, ClassificationScheme):
            classification = ClassificationScheme()
        self.view().setMinimumWidth(200)
        model = ClassificationSchemeComboBoxModel()
        model.setClassificationScheme(classification)
        self.mModel = None
        self.setModel(model)

    def classIndexFromValue(self, value) -> int:
        """
        Returns the index
        :param value:
        :return:
        """
        cs = self.classificationScheme()
        i = cs.classIndexFromValue(value)

        if isinstance(self.mModel, ClassificationSchemeComboBoxModel) and self.mModel.allowEmptyField():
            i += 1
        return i

    def setModel(self, model: ClassificationSchemeComboBoxModel):
        """
        Sets the combobox model. Must be of type `ClassificationSchemeComboBoxModel`.
        :param model: ClassificationSchemeComboBoxModel
        """
        super(ClassificationSchemeComboBox, self).setModel(model)
        self.mModel = model

    def classificationScheme(self) -> ClassificationScheme:
        """
        Returns the ClassificationScheme
        :return: ClassificationScheme
        """
        return self.mModel.classificationScheme()

    def setClassificationScheme(self, classificationScheme: ClassificationScheme):
        """
        Specifies the ClassificationScheme which is represented by this mode.
        :param classificationScheme: ClassificationScheme
        """
        self.mModel.setClassificationScheme(classificationScheme)

    def currentClassInfo(self) -> ClassInfo:
        """
        Returns the currently selected ClassInfo
        :return: ClassInfo
        """
        i = self.currentIndex()
        classInfo = None
        if i >= 0 and i < self.count():
            classInfo = self.itemData(i, role=Qt.ItemDataRole.UserRole)
        return classInfo

    def setCurrentClassInfo(self, classInfo: ClassInfo) -> bool:
        """
        Sets the current ClassInfo
        :param classInfo: ClassInfo
        :return: bool, True, if class was found and set.
        """
        for i in range(self.count()):
            if self.itemData(i, role=Qt.ItemDataRole.UserRole) == classInfo:
                self.setCurrentIndex(i)
                return True
        return False


class ClassificationSchemeWidget(QWidget):
    sigValuesChanged = pyqtSignal()

    def __init__(self, parent=None, classificationScheme: ClassificationScheme = None):
        super(ClassificationSchemeWidget, self).__init__(parent)
        pathUi = pathlib.Path(__file__).parent / 'classificationscheme.ui'
        loadUi(pathUi, self)

        self.mScheme = ClassificationScheme()
        self.mScheme.sigClassesAdded.connect(self.validateButtons)
        self.mScheme.sigClassesRemoved.connect(self.validateButtons)
        if classificationScheme is not None:
            self.setClassificationScheme(classificationScheme)
        self.mScheme.sigIsEditableChanged.connect(self.onIsEditableChanged)
        self.tableClassificationScheme: QTableView
        self.tableClassificationScheme.setModel(self.mScheme)
        self.tableClassificationScheme.doubleClicked.connect(self.onTableDoubleClick)
        self.tableClassificationScheme.resizeColumnsToContents()
        self.selectionModel = QItemSelectionModel(self.mScheme)
        self.selectionModel.selectionChanged.connect(self.validateButtons)

        self.tableClassificationScheme.setSelectionModel(self.selectionModel)

        self.initActions()
        self.setIsEditable(True)
        self.validateButtons()  # enable/disable widgets depending on a selection and state of classification scheme

    def isEditable(self) -> bool:
        return self.mScheme.isEditable()

    def setIsEditable(self, b: bool):
        self.mScheme.setIsEditable(b)

    def onIsEditableChanged(self, b: bool):

        # hide/show buttons to edit the scheme
        setE = True
        setV = True

        buttons = [self.btnAddClasses, self.btnLoadClasses, self.btnPasteClasses, self.btnRemoveClasses]

        if setE:
            for btn in buttons:
                btn.setEnabled(b)

                a = btn.defaultAction()
                if isinstance(a, QAction):
                    a.setEnabled(b)

        if setV:
            for btn in buttons:
                btn.setVisible(b)

    def copyClassesToClipboard(self, vector_layer: bool = False):
        classes = self.selectedClasses(allIfNone=True)
        if len(classes) == 0:
            return
        cs = ClassificationScheme()
        cs.insertClasses(classes)
        cb = QApplication.clipboard()
        cb.setMimeData(cs.mimeData(None, vector_layer=vector_layer))

    def onPasteClasses(self):
        mimeData = QApplication.clipboard().mimeData()

        cs = ClassificationScheme.fromMimeData(mimeData)
        if isinstance(cs, ClassificationScheme):
            self.mScheme.insertClasses(cs[:])

    def onSaveClasses(self):
        classes = self.selectedClasses(allIfNone=True)
        if len(classes) == 0:
            return

        cs = ClassificationScheme()
        cs.insertClasses(classes)

        filter = "CSV (*.csv *.txt);;JSON (*.json)"
        path, filter = QFileDialog.getSaveFileName(self, "Save classes to file",
                                                   "/home", filter)
        if isinstance(path, str) and len(path) > 0:

            if path.endswith('.json'):

                pass

            elif path.endswith('.csv'):

                cs.saveToCsv(path)

            if filter == 'csv':
                pass

    def onLoadClassesFromRenderer(self, layer):
        cs = ClassificationScheme.fromMapLayer(layer)
        if isinstance(cs, ClassificationScheme):
            self.mScheme.insertClasses(cs[:])

    def onLoadClassesFromField(self, layer, field):

        if field is None:
            raise NotImplementedError()

        cs = ClassificationScheme.fromUniqueFieldValues(layer, field)
        if isinstance(cs, ClassificationScheme):
            self.mScheme.insertClasses(cs[:])
        pass

    def onLoadClasses(self, mode: str):
        """
        Opens a dialog to add ClassInfos from other sources, like raster images, text files and QgsMapLayers.
        :param mode: 'raster', 'layer', 'textfile'
        """
        if mode == 'raster':
            filter = QgsProviderRegistry.instance().fileRasterFilters()
            path, filter = QFileDialog.getOpenFileName(self,
                                                       "Read classes from raster image",
                                                       "/home", filter)
            if isinstance(path, str) and os.path.isfile(path):
                cs = ClassificationScheme.fromRasterImage(path)
                if isinstance(cs, ClassificationScheme):
                    self.mScheme.insertClasses(cs[:])

        if mode == 'layer':
            possibleLayers = findMapLayersWithClassInfo()
            if len(possibleLayers) == 0:
                QMessageBox.information(self, 'Load classes from layer',
                                        'No layers with categorical render styles available.')
            else:
                choices = ['{} ({})'.format(lyr.name(), lyr.source()) for lyr in possibleLayers]

                dialog = QInputDialog(parent=self)
                dialog.setWindowTitle('Load classes from layer')
                dialog.setTextValue('Select map layer')
                dialog.setComboBoxItems(choices)
                dialog.setOption(QInputDialog.InputDialogOption.UseListViewForComboBoxItems)
                if dialog.exec() == QDialog.DialogCode.Accepted:
                    selection = dialog.textValue()
                    i = choices.index(selection)
                    layer = possibleLayers[i]
                    if isinstance(layer, QgsVectorLayer):
                        cs = ClassificationScheme.fromFeatureRenderer(layer.renderer())
                    elif isinstance(layer, QgsRasterLayer):
                        cs = ClassificationScheme.fromRasterRenderer(layer.renderer())
                    if isinstance(cs, ClassificationScheme):
                        self.mScheme.insertClasses(cs[:])
            pass

        if mode == 'textfile':

            filter = "CSV (*.csv *.txt);;JSON (*.json);;QML (*.qml)"
            path, filter = QFileDialog.getOpenFileName(self,
                                                       "Read classes from text file",
                                                       "/home", filter)
            if isinstance(path, str) and os.path.isfile(path):
                cs = ClassificationScheme.fromFile(path)
                if isinstance(cs, ClassificationScheme):
                    self.mScheme.insertClasses(cs[:])

    def initActions(self):

        m = QMenu('Load classes')
        m.setToolTip('Load classes ...')
        self.actionLoadFromRasterFile.triggered.connect(lambda: self.onLoadClasses('raster'))
        m.addAction(self.actionLoadFromRasterFile)

        self.actionLoadFromMapLayer.triggered.connect(lambda: self.onLoadClasses('layer'))
        m.addAction(self.actionLoadFromMapLayer)

        self.actionLoadfromTextFile.triggered.connect(lambda: self.onLoadClasses('textfile'))
        m.addAction(self.actionLoadfromTextFile)

        parent = self.parent()
        if isinstance(parent, ClassificationSchemeEditorConfigWidget):
            layer = parent.layer()
            idx = parent.field()

            if isinstance(layer, QgsVectorLayer) and idx >= 0:
                field = layer.fields().at(idx)
                m.addSeparator()
                a = m.addAction('Unique values "{}"'.format(field.name()))
                a.triggered.connect(lambda _, lyr=layer, f=idx: self.onLoadClassesFromField(lyr, idx))

                if isinstance(layer.renderer(), QgsCategorizedSymbolRenderer):
                    a = m.addAction('Current Renderer {}'.format(layer.name()))
                    a.triggered.connect(lambda _, lyr=layer: self.onLoadClassesFromRenderer(lyr))

        self.btnLoadClasses.setMenu(m)

        self.actionRemoveClasses.triggered.connect(self.removeSelectedClasses)
        self.actionAddClasses.triggered.connect(lambda: self.createClasses(1))

        self.actionSaveClasses.setIcon(QIcon(r'://images/themes/default/mActionFileSaveAs.svg'))
        self.actionSaveClasses.triggered.connect(self.onSaveClasses)

        QApplication.clipboard().dataChanged.connect(self.onClipboard)
        self.actionPasteClasses.setIcon(QIcon(r'://images/themes/default/mActionEditPaste.svg'))
        self.actionPasteClasses.triggered.connect(self.onPasteClasses)

        self.btnSaveClasses.setDefaultAction(self.actionSaveClasses)
        self.btnRemoveClasses.setDefaultAction(self.actionRemoveClasses)
        self.btnAddClasses.setDefaultAction(self.actionAddClasses)

        self.actionCopyClasses.triggered.connect(lambda *args: self.copyClassesToClipboard(vector_layer=False))
        self.actionCopyToVectorLayer.triggered.connect(lambda *args: self.copyClassesToClipboard(vector_layer=True))
        self.actionCopyToRasterLayer.triggered.connect(lambda *args: self.copyClassesToClipboard(vector_layer=False))
        m = QMenu('Copy classes')
        m.addAction(self.actionCopyToVectorLayer)
        m.addAction(self.actionCopyToRasterLayer)

        self.btnCopyClasses.setMenu(m)
        self.btnCopyClasses.setDefaultAction(self.actionCopyClasses)
        self.btnPasteClasses.setDefaultAction(self.actionPasteClasses)

        self.onClipboard()

    def onClipboard(self, *args):
        mimeData = QApplication.clipboard().mimeData()
        b = isinstance(mimeData, QMimeData) and (
            MIMEDATA_KEY_TEXT in mimeData.formats() or MIMEDATA_KEY_QGIS_STYLE in mimeData.formats()
        )

        self.actionPasteClasses.setEnabled(b)

    def onTableDoubleClick(self, idx):
        model = self.tableClassificationScheme.model()
        classInfo = model.index2ClassInfo(idx)
        if idx.column() == ClassificationScheme.cColor and model.isEditable():
            c = QColorDialog.getColor(classInfo.mColor, self.tableClassificationScheme,
                                      'Set color for "{}"'.format(classInfo.name()))
            model.setData(idx, c, role=Qt.ItemDataRole.EditRole)

    def validateButtons(self, *args):
        n = len(self.selectionModel.selectedRows())

        self.actionRemoveClasses.setEnabled(n > 0)

        infix = '{} selected class(es)'.format(n) if n > 0 else 'all classes'

        self.actionRemoveClasses.setToolTip('Remove {}'.format(infix))
        self.actionCopyClasses.setToolTip('Copy {}'.format(infix))
        self.actionSaveClasses.setToolTip('Save {}'.format(infix))

        b = len(self.mScheme) > 0
        self.actionRemoveClasses.setEnabled(b)
        self.actionCopyClasses.setEnabled(b)
        self.actionSaveClasses.setEnabled(b)

    def createClasses(self, n) -> List[ClassInfo]:
        self.mScheme.createClasses(n)

    def selectedClasses(self, allIfNone: bool = False) -> List[ClassInfo]:
        """
        Returns the list of selected ClassInfos
        :return: [list-of-ClassInfo]
        :param allIfNone: if True, returns all classes in case none is selected.
        :return: [list-of-ClassInfo]
        :rtype:
        """
        indices = list(reversed(self.selectionModel.selectedRows()))
        if len(indices) == 0 and allIfNone:
            return self.mScheme[:]
        else:
            return [self.mScheme.index2ClassInfo(idx) for idx in indices]

    def removeSelectedClasses(self):
        classes = self.selectedClasses()
        if len(classes) > 0:
            self.mScheme.removeClasses(classes)

    def loadClasses(self, *args):

        defDir = None
        path, _ = QFileDialog.getOpenFileName(self, 'Select Raster File', directory=defDir)
        if os.path.exists(path):

            try:
                scheme = ClassificationScheme.fromRasterImage(path)
                if scheme is not None:
                    self.appendClassificationScheme(scheme)
            except Exception as ex:
                QMessageBox.critical(self, "Unable to load class info", str(ex))

    def appendClassificationScheme(self, classificationScheme):
        if not isinstance(classificationScheme, ClassificationScheme):
            raise Exception(f'wrong type: {classificationScheme}')
        self.mScheme.insertClasses([c for c in classificationScheme])

    def setClassificationScheme(self, classificationScheme: ClassificationScheme):
        if not isinstance(classificationScheme, ClassificationScheme):
            raise Exception(f'wrong type: {classificationScheme}')
        self.mScheme.clear()
        self.appendClassificationScheme(classificationScheme)

    def classificationScheme(self):
        return self.mScheme


class ClassificationSchemeDialog(QgsDialog):
    @classmethod
    def getClassificationScheme(cls, *args, **kwds):
        """
        Opens a dialog to edit a ClassificationScheme
        :param args:
        :param kwds:
        :return: None | ClassificationScheme
        """
        d = cls(*args, **kwds)
        d.exec()

        if d.result() == QDialog.DialogCode.Accepted:
            return d.classificationScheme()
        else:
            return None

    def __init__(self, parent=None, classificationScheme=None, title='Specify Classification Scheme'):
        super(ClassificationSchemeDialog, self).__init__(
            parent=parent,
            buttons=QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel
        )
        self.w = ClassificationSchemeWidget(parent=self, classificationScheme=classificationScheme)
        self.setWindowTitle(title)
        self.btOk = QPushButton('Ok')
        self.btCancel = QPushButton('Cancel')
        buttonBar = QHBoxLayout()
        # buttonBar.addWidget(self.btCancel)
        # buttonBar.addWidget(self.btOk)
        layout = self.layout()
        layout.addWidget(self.w)
        layout.addLayout(buttonBar)
        # self.setLayout(l)

        if isinstance(classificationScheme, ClassificationScheme):
            self.setClassificationScheme(classificationScheme)

    def classificationScheme(self):
        return self.w.classificationScheme()

    def setClassificationScheme(self, classificationScheme):
        self.w.setClassificationScheme(classificationScheme)


class ClassificationSchemeEditorWidgetWrapper(QgsEditorWidgetWrapper):

    def __init__(self, vl: QgsVectorLayer, fieldIdx: int, editor: QWidget, parent: QWidget):
        super(ClassificationSchemeEditorWidgetWrapper, self).__init__(vl, fieldIdx, editor, parent)

        self.mComboBox = None
        self.mDefaultValue = None

    def createWidget(self, parent: QWidget):
        # log('createWidget')
        w = ClassificationSchemeComboBox(parent)
        w.model().setAllowEmptyField(True)
        w.setVisible(True)
        return w

    def initWidget(self, editor: QWidget):
        # log(' initWidget')
        conf = self.config()

        if isinstance(editor, ClassificationSchemeComboBox):
            self.mComboBox = editor
            self.mComboBox.setClassificationScheme(classSchemeFromConfig(conf))
            self.mComboBox.currentIndexChanged.connect(self.onValueChanged)

    def onValueChanged(self, *args):
        self.valueChanged.emit(self.value())

    def valid(self, *args, **kwargs) -> bool:
        return isinstance(self.mComboBox, ClassificationSchemeComboBox)

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

        value = None
        if isinstance(self.mComboBox, ClassificationSchemeComboBox):
            classInfo = self.mComboBox.currentClassInfo()
            if isinstance(classInfo, ClassInfo):

                typeCode = self.field().type()
                if typeCode == QMetaType.Type.QString:
                    value = classInfo.name()
                elif typeCode in [QMetaType.Type.Int, QMetaType.Type.Double]:
                    value = classInfo.label()

        return value

    def setEnabled(self, enabled: bool):

        if isinstance(self.mComboBox, ClassificationSchemeComboBox):
            self.mComboBox.setEnabled(enabled)

    def setValue(self, value):

        if isinstance(self.mComboBox, ClassificationSchemeComboBox):
            i = self.mComboBox.classIndexFromValue(value)
            self.mComboBox.setCurrentIndex(i)


class ClassificationSchemeEditorConfigWidget(QgsEditorConfigWidget):

    def __init__(self, vl: QgsVectorLayer, fieldIdx: int, parent: QWidget):
        super(ClassificationSchemeEditorConfigWidget, self).__init__(vl, fieldIdx, parent)
        # self.setupUi(self)
        if not isinstance(vl, QgsVectorLayer):
            raise Exception('vl is not a QgsVectorLayer')
        self.mSchemeWidget = ClassificationSchemeWidget(parent=self)
        self.mSchemeWidget.classificationScheme().sigClassesAdded.connect(self.onClassesAdded)
        self.mSchemeWidget.classificationScheme().sigClassesRemoved.connect(self.onClassesRemoved)

        self.mActionImport = QAction('Import from layer')
        self.mActionImport.setIcon(QIcon(':/qps/ui/icons/import_symbology.svg'))
        self.mActionImport.triggered.connect(self.onImportFromLayer)

        self.mActionExport = QAction('Set to layer')
        self.mActionExport.triggered.connect(self.onExportToLayer)
        self.mActionExport.setIcon(QIcon(':/qps/ui/icons/export_symbology.svg'))

        self.btnImport = QToolButton(parent=self.mSchemeWidget)
        self.btnExport = QToolButton(parent=self.mSchemeWidget)
        self.btnImport.setDefaultAction(self.mActionImport)
        self.btnExport.setDefaultAction(self.mActionExport)
        self.btnImport.setAutoRaise(True)
        self.btnExport.setAutoRaise(True)

        l: QVBoxLayout = self.mSchemeWidget.btnBarLayout
        l.addWidget(self.btnImport)
        l.addWidget(self.btnExport)

        self.setLayout(QVBoxLayout())
        self.layout().addWidget(self.mSchemeWidget)
        vl.rendererChanged.connect(self.updateButtons)
        self.mLastConfig = {}

    def onImportFromLayer(self):

        lyr: QgsVectorLayer = self.layer()

        if isinstance(lyr, QgsVectorLayer):
            r = lyr.renderer()
            if isinstance(r, QgsCategorizedSymbolRenderer):
                cs = ClassificationScheme.fromFeatureRenderer(r)
                if len(cs) > 0:
                    self.classificationScheme().clear()
                    self.classificationScheme().insertClasses(cs[:])

    def onExportToLayer(self):
        cs = self.classificationScheme()
        if isinstance(cs, ClassificationScheme) and len(cs) > 0:
            renderer = cs.featureRenderer()
            renderer.setClassAttribute(self.layer().fields().at(self.field()).name())

            self.layer().setRenderer(renderer)

    def onClassesAdded(self):
        self.updateButtons()

    def onClassesRemoved(self):
        self.updateButtons()

    def classificationScheme(self) -> ClassificationScheme:
        return self.mSchemeWidget.classificationScheme()

    def updateButtons(self):

        self.mActionExport.setEnabled(len(self.classificationScheme()) > 0)
        lyr: QgsVectorLayer = self.layer()

        has_classes = False
        if isinstance(lyr, QgsVectorLayer):
            r = lyr.renderer()
            if isinstance(r, QgsCategorizedSymbolRenderer):
                has_classes = r.classAttribute() in self.layer().fields().names()

        self.mActionImport.setEnabled(has_classes)

    def config(self, *args, **kwargs) -> dict:
        return classSchemeToConfig(self.mSchemeWidget.classificationScheme())

    def setConfig(self, config: dict):
        self.mLastConfig = config
        cs = classSchemeFromConfig(config)
        cs.setName(self.layer().fields()[self.field()].name())
        self.mSchemeWidget.setClassificationScheme(cs)

    def resetClassificationScheme(self):
        self.setConfig(self.mLastConfig)


def classSchemeToConfig(classScheme: ClassificationScheme) -> dict:
    """Converts a ClassificationScheme into a dictionary that can be used in an QgsEditorWidgetSetup"""
    config = {'classes': classScheme.json()}
    return config


def classSchemeFromConfig(conf: dict) -> ClassificationScheme:
    """
    Converts a configuration dictionary into a ClassificationScheme.
    :param conf: dict
    :return: ClassificationScheme
    """
    cs = None
    if 'classes' in conf.keys():
        cs = ClassificationScheme.fromJson(conf['classes'])
    if not isinstance(cs, ClassificationScheme):
        return ClassificationScheme()
    else:
        return cs


class ClassificationSchemeWidgetFactory(QgsEditorWidgetFactory):

    def __init__(self, name: str):

        super(ClassificationSchemeWidgetFactory, self).__init__(name)

        self.mConfigurations = {}

    def configWidget(self, layer: QgsVectorLayer, fieldIdx: int,
                     parent=QWidget) -> ClassificationSchemeEditorConfigWidget:
        """
        Returns a SpectralProfileEditorConfigWidget
        :param layer: QgsVectorLayer
        :param fieldIdx: int
        :param parent: QWidget
        :return: SpectralProfileEditorConfigWidget
        """

        w = ClassificationSchemeEditorConfigWidget(layer, fieldIdx, parent)

        key = self.configKey(layer, fieldIdx)

        initialConfig = layer.editorWidgetSetup(fieldIdx).config()
        self.writeConfig(key, initialConfig)
        w.setConfig(self.readConfig(key))
        w.changed.connect(lambda: self.writeConfig(key, w.config()))
        return w

    def configKey(self, layer: QgsVectorLayer, fieldIdx: int):
        """
        Returns a tuple to be used as dictionary key to identify a layer field configuration.
        :param layer: QgsVectorLayer
        :param fieldIdx: int
        :return: (str, int)
        """
        return (layer.id(), fieldIdx)

    def create(self, layer: QgsVectorLayer, fieldIdx: int, editor: QWidget,
               parent: QWidget) -> ClassificationSchemeEditorWidgetWrapper:
        """
        Create a ClassificationSchemeEditorWidgetWrapper
        :param layer: QgsVectorLayer
        :param fieldIdx: int
        :param editor: QWidget
        :param parent: QWidget
        :return: ClassificationSchemeEditorWidgetWrapper
        """
        w = ClassificationSchemeEditorWidgetWrapper(layer, fieldIdx, editor, parent)
        return w

    def writeConfig(self, key: tuple, config: dict):
        """
        :param key: tuple (str, int), as created with .configKey(layer, fieldIdx)
        :param config: dict with config values
        """
        self.mConfigurations[key] = config

    def readConfig(self, key: tuple):
        """
        :param key: tuple (str, int), as created with .configKey(layer, fieldIdx)
        :return: {}
        """
        if key in self.mConfigurations.keys():
            conf = self.mConfigurations[key]
        else:
            # return the very default "empty" configuration
            conf = {}
        return conf

    def fieldScore(self, vl: QgsVectorLayer, fieldIdx: int) -> int:
        """
        This method allows disabling this editor widget type for a certain field.
        0: not supported: none String fields
        5: maybe support String fields with length <= 400
        20: specialized support: String fields with length > 400

        :param vl: QgsVectorLayer
        :param fieldIdx: int
        :return: int
        """
        # log(' fieldScore()')
        if fieldIdx < 0:
            return 0
        field = vl.fields().at(fieldIdx)
        if re.search('(int|float|double|text|string)', field.typeName(), re.I):
            if re.search('class', field.name(), re.I):
                return 5  # should we return 10 for showing specialized support?
            else:
                return 5
        else:
            return 0  # no support

    def supportsField(self, vl: QgsVectorLayer, idx: int):
        field = vl.fields().at(idx)
        if isinstance(field, QgsField) and re.search('(int|float|double|text|string)', field.typeName(), re.I):
            return True
        return False


EDITOR_WIDGET_REGISTRY_KEY = 'Raster Classification'


def classificationSchemeEditorWidgetFactory(register: bool = True) -> ClassificationSchemeWidgetFactory:
    global _CLASS_SCHEME_EDITOR_WIDGET_FACTORY
    if not isinstance(_CLASS_SCHEME_EDITOR_WIDGET_FACTORY, ClassificationSchemeWidgetFactory):
        _CLASS_SCHEME_EDITOR_WIDGET_FACTORY = ClassificationSchemeWidgetFactory(EDITOR_WIDGET_REGISTRY_KEY)
    reg = QgsGui.editorWidgetRegistry()
    if register and EDITOR_WIDGET_REGISTRY_KEY not in reg.factories().keys():
        reg.registerWidget(EDITOR_WIDGET_REGISTRY_KEY, _CLASS_SCHEME_EDITOR_WIDGET_FACTORY)
    return reg.factory(EDITOR_WIDGET_REGISTRY_KEY)


def registerClassificationSchemeEditorWidget():
    warnings.warn(DeprecationWarning('Use classificationscheme.classificationSchemeEditorWidgetFactory(register=True)'),
                  stacklevel=2)
    classificationSchemeEditorWidgetFactory(True)
