Remove graphics in the Experiment tab when all the generations are cut

* #49

Author: Ezetowers

MLC/GUI/Experiment/ExperimentWindow.py

@@ -458,6 +458,8 @@ def on_gen_start_over_button_clicked(self):
                      'Executing on_gen_start_over_button_clicked function'
                      .format(self._experiment_name))
 
+################################### VIEW AND MODEL METHODS #########################################
+
     def _config_table_edited(self, left, right):
         config_table = self._autogenerated_object.config_table
         table_model = config_table.model()
@@ -667,18 +669,12 @@ def _update_experiment_info(self):
         else:
             self._current_gen = 0
             gen_count_group.setTitle("Generation: 0/0")
+            db_view = self._autogenerated_object.db_view
+            db_view.setModel(None)
 
     def _update_individuals_figure(self):
+        chart_layout = self._autogenerated_object.chart_layout
         if self._current_gen != 0:
-            chart_layout = self._autogenerated_object.chart_layout
-
-            # Matplotlib graphs
-            # indiv_canvas = CostPerIndividualCanvas(indiv_graph_frame, width=5, height=4, dpi=85)
-            # indiv_canvas.set_costs(current_generation.get_costs())
-            # indiv_canvas.set_xlabel('Individuals')
-            # indiv_canvas.set_ylabel('Costs')
-            # indiv_canvas.set_title('Cost Per Individual')
-            # indiv_canvas.compute_initial_figure(self._autogenerated_object.log_check.isChecked())
 
             # Draw current generation
             current_generation = self._mlc_local.get_generation(self._experiment_name, self._current_gen)
@@ -705,13 +701,19 @@ def _update_individuals_figure(self):
                     costs = gen.get_costs()
                     indiv_chart.add_data(samples, costs, color=Qt.blue, line_width=1)
 
-            indiv_canvas = indiv_chart.get_widget()
-
             # Remove all previous widgets before rendering again
             for i in reversed(range(chart_layout.count())):
                 chart_layout.itemAt(i).widget().setParent(None)
+
             # Add the Indiv Canvas
+            indiv_canvas = indiv_chart.get_widget()
             chart_layout.addWidget(indiv_canvas)
+        else:
+            # Remove all previous widgets before rendering again
+            for i in reversed(range(chart_layout.count())):
+                chart_layout.itemAt(i).widget().setParent(None)
+            chart_layout.addWidget(self._autogenerated_object.textEdit_2)
+
 
     def _reload_experiment_config(self, some_string):
         logger.debug('[EXPERIMENT {0}] [EXPERIMENT_CONF_FILE_CHANGED] - '