更新figure class：在重新绘制line的过程中，只移除之前绘制了的line；更新了配色方案

1 files changed, 28 insertions, 12 deletions

diff --git a/vprtw_aco_figure.py b/vprtw_aco_figure.py
index 01dbc80..6fa8765 100644
--- a/vprtw_aco_figure.py
+++ b/vprtw_aco_figure.py
@@ -18,42 +18,58 @@ class VrptwAcoFigure:
         self.figure = plt.figure(figsize=(10, 10))
         self.figure_ax = self.figure.add_subplot(1, 1, 1)
         self.path_queue = path_queue
-        self._dot_color = 'k'
-        self._line_color_list = ['r', 'y', 'g', 'c', 'b', 'm']
+        self._depot_color = 'k'
+        self._customer_color = 'steelblue'
+        self._line_color = 'darksalmon'
 
     def _draw_point(self):
-        # 先画出图中的点
-        self.figure_ax.plot(list(node.x for node in self.nodes),
-                            list(node.y for node in self.nodes), '%s.' % self._dot_color)
-        self.figure.show()
+        # 画出depot
+        self.figure_ax.scatter([self.nodes[0].x], [self.nodes[0].y], c=self._depot_color, label='depot', s=40)
+
+        # 画出customer
+        self.figure_ax.scatter(list(node.x for node in self.nodes[1:]),
+                               list(node.y for node in self.nodes[1:]), c=self._customer_color, label='customer', s=20)
         plt.pause(0.5)
 
     def run(self):
+        # 先绘制出各个结点
         self._draw_point()
+        self.figure.show()
 
         # 从队列中读取新的path，进行绘制
         while True:
             if not self.path_queue.empty():
+                # 取队列中最新的一个path，其他的path丢弃
                 info = self.path_queue.get()
                 while not self.path_queue.empty():
                     info = self.path_queue.get()
-                path, distance, used_vehicle_num = info.get_path_info()
 
+                path, distance, used_vehicle_num = info.get_path_info()
                 if path is None:
+                    print('[draw figure]: exit')
                     break
 
-                self.figure_ax.clear()
-                self._draw_point()
-                self.figure.canvas.draw()
+                # 需要先记录要移除的line，不能直接在第一个循环中进行remove，
+                # 不然self.figure_ax.lines会在循环的过程中改变，导致部分line无法成功remove
+                remove_obj = []
+                for line in self.figure_ax.lines:
+                    if line._label == 'line':
+                        remove_obj.append(line)
+
+                for line in remove_obj:
+                    self.figure_ax.lines.remove(line)
+                remove_obj.clear()
                 plt.pause(1)
 
+                # 重新绘制line
+                self.figure_ax.set_title('current path travel distance: %f' % distance)
                 self._draw_line(path)
             plt.pause(1)
 
     def _draw_line(self, path):
-        # 根据path中index进行路劲的绘制
+        # 根据path中index进行路径的绘制
         for i in range(1, len(path)):
             x_list = [self.nodes[path[i - 1]].x, self.nodes[path[i]].x]
             y_list = [self.nodes[path[i - 1]].y, self.nodes[path[i]].y]
-            self.figure_ax.plot(x_list, y_list, '%s-' % self._line_color_list[0])
+            self.figure_ax.plot(x_list, y_list, color=self._line_color, linewidth=1.5, label='line')
             plt.pause(0.05)