from gui_qt.Qt import QtWidgets, QtCore
from gui_qt._py.tnmh_dialog import Ui_Dialog
from gui_qt.lib.pg_objects import PlotItem


class TgCalculator(QtWidgets.QDialog, Ui_Dialog):
    def __init__(self, management, parent=None):
        super().__init__(parent=parent)

        self.setupUi(self)
        self._management = management

        self._plots = {}

    def __call__(self):
        self.clear()
        self.add_sets()
        # s.valuesChanged.connect(self.management.shift_scale)

    def clear(self):
        self.listWidget.clear()
        for k, v in self._plots.items():
            self.graphicsView.removeItem(v)

    def add_sets(self):
        for key, name in self._management.active_sets:
            data = self._management.data[key]

            item = QtWidgets.QListWidgetItem(name)
            item.setData(QtCore.Qt.UserRole, key)
            plot = PlotItem(x=data.x, y=data.y)
            self.graphicsView.addItem(plot)
            self._plots[key] = plot


if __name__ == '__main__':
    app = QtWidgets.QApplication([])
    w = TgCalculator(None)
    w.show()
    app.exec()