use tree colors in fit result for sub-funcs (#253)

closes #201

Co-authored-by: Dominik Demuth <dominik.demuth@physik.tu-darmstadt.de>
Reviewed-on: #253

src/gui_qt/main/management.py

@@ -85,7 +85,7 @@ class UpperManagement(QtCore.QObject):
     newData = QtCore.pyqtSignal([list, str], [list, str, bool])
     deleteData = QtCore.pyqtSignal(list)
     dataChanged = QtCore.pyqtSignal(str)
-    fitFinished = QtCore.pyqtSignal(list)
+    fitFinished = QtCore.pyqtSignal(list, dict)
     stopFit = QtCore.pyqtSignal()
     properties_collected = QtCore.pyqtSignal(dict)
     unset_state = QtCore.pyqtSignal(list)
@@ -586,13 +586,23 @@ class UpperManagement(QtCore.QObject):
     def end_fit(self, result: list, success: bool):
         if success:
             logger.info('Successful fit')
-            self.fitFinished.emit(result)
+
+            sub_colors = {}
+            for k, v in self.__fit_options[0].items():
+                sub_colors.update({set_id: v['color'] for set_id in v['data_parameter']})
+
+            self.fitFinished.emit(result, sub_colors)
+
         else:
             e = result[0]
             logger.exception(e, exc_info=True)
-            QtWidgets.QMessageBox.warning(QtWidgets.QWidget(), 'Fit failed',
-                                          f'Fit kaput with exception: \n\n{e!r}')
-            self.fitFinished.emit([])
+            QtWidgets.QMessageBox.warning(
+                QtWidgets.QWidget(),
+                'Fit failed',
+                f'Fit kaput with exception: \n\n{e!r}'
+            )
+            self.fitFinished.emit([], {})
+
         self._fit_active = False
 
     @QtCore.pyqtSlot(dict)