예제 #1
0
 def community_init(self, group):
     infected_id = random.sample(
         range(self.size), int(round(self.size * setting.INITIAL_CHANCE,
                                     0)))
     self.get_empty_big_pos()
     # infected_id = random.sample(range(self.size), 1)
     for num in range(self.size):
         ent = Entities.Entity()
         _id = num + 1
         ent.entity_init(setting.RESOLUTION, _id)
         # ent.entity_init(world_size=self.world_size, id=_id)
         ent.is_dist_kept = self.keep_dist_generator()
         self.entity_pool[f'{ent.id}'] = ent
         if _id in infected_id:
             ent.status.is_infected = 1
             self.infected_id_pool.append((f'{_id}'))
         self.big_pos_getter(ent)
         group.add(ent)