simulation.py 文件源码

def draw_termites(self):
        """Draw termites on simulation.

        Args:
            None.
        Returns:
            None.
        """
        for termite in self.termites:
            cv2.circle(self.background, termite.trail[self.current_step],
                       self.params['termite_radius'], termite.color, 2)
            cv2.circle(self.background, termite.trail[self.current_step],
                       2, termite.color, -1)
            cv2.putText(self.background, termite.number, (termite.trail[self.current_step][0] - 4,
                        termite.trail[self.current_step][1] - self.params['termite_radius'] - 5), 2,
                        color=termite.color, fontScale=0.4)
            cv2.circle(self.video_source.current_frame, termite.trail[self.current_step],
                       self.params['termite_radius'], termite.color, 2)
            cv2.circle(self.video_source.current_frame, termite.trail[self.current_step],
                       2, termite.color, -1)
            cv2.putText(self.video_source.current_frame, termite.number, (termite.trail[self.current_step][0] - 4,
                        termite.trail[self.current_step][1] - self.params['termite_radius'] - 5), 2,
                        color=termite.color, fontScale=0.4)