病毒传播仿真器的实现
现在来谈谈仿真器实现的原理。仿真器使用Python和PyQt5实现。PyQt5是封装了Qt library的跨平台GUI开发库,基于Python语言。 这里主要涉及到仿真器效果绘制,以及如何模拟多个参数。先来说一下绘制市民的状态。绘制的工作通过drawing.py文件的Drawing类来完成。该类是QWidget的子类,这也就意味着Drawing类本身是PyQt5的一个组件。与按钮、标签类似。只是并不需要往Drawing上放置任何子组件。只要在Drawing上绘制各种图形即可。 在PyQt5中,任何一个QWidget的子类,都可以实现一个paintEvent方法,当组件每次刷新时,就会调用paintEvent方法重新绘制组件的内容。Drawing类中paintEvent方法的代码如下:
def paintEvent(self, event):
qp = QPainter() qp.begin(self) # 绘制城市的各种状态的市民 self.drawing(qp) qp.end()
在绘制图像前,需要创建QPainter对象,然后调用QPainter对象的begin方法,结束绘制后,需要调用QPainter对象的end方法。上面代码中的drawing方法用于完成具体的绘制工作。
仿真器可以模拟5000个市民的状态,所以需要用5000个小矩形来表示这5000个市民。也就是在drawing方法中需要绘制这5000个表示市民的小矩形。代码如下:
def drawing(self, event):
... ... # 绘制代表市民的小矩形 persons = Persons().persons if persons == None: return normal_person_count = 0 latency_person_count = 0 confirmed_person_count = 0 freeze_person_count = 0 death_person_count = 0 # 扫描内一个人的状态 for person in persons: if person.state == NORMAL: # 健康人 qp.setPen(Qt.white) normal_person_count += 1 elif person.state == LATENCY: # 潜伏期感染者
qp.setPen(QColor(255,238,0))
latency_person_count += 1 elif person.state == CONFIRMED: # 确诊患者 qp.setPen(Qt.red) confirmed_person_count += 1 elif person.state == FREEZE: # 已隔离者 qp.setPen(QColor(72, 255, 252)) freeze_person_count += 1 elif person.state == DEATH: # 死亡患者 qp.setPen(Qt.black) death_person_count += 1 person.update() # 更新每一个人的状态 bed_half_size = Hospital().bed_size // 2 rect = QRect(person.x - bed_half_size, person.y - bed_half_size,Hospital().bed_size//2, Hospital().bed_size//2) brush = QBrush(Qt.SolidPattern) brush.setColor(qp.pen().color()) qp.setBrush(brush) qp.drawRect(rect)