예제 #1
0
 def __init__(self):
     main = MainBase()
     main.pos = vec2d(300, 300)
     main.target = vec2d(main.pos[0] + 10, main.pos[1])
     out = Outpost()
     out.pos = vec2d(200, 200)
     out.target = vec2d(out.pos[0] + 10, out.pos[1])
     self.buildings = [main, out]
예제 #2
0
    def __init__(self):
        self.w, self.h = 800, 600
        PygameHelper.__init__(self, size=(self.w, self.h), fill=((255,255,255)))
        
        self.hud = Hud()
        
        self.drones = []
        
        for i in range(4): 
            tempagent = Drone()
            tempagent.pos = vec2d(int(uniform(0, self.w - 20)), int(uniform(0, 400)))
            target = vec2d(int(uniform(0, self.w - 20)), int(uniform(0, 400)))
            tempagent.target.append(target)
            self.drones.append(tempagent)

        self.selected = self.drones[0]
        self.builder = self.drones[0]
        self.project = vec2d(0, 0)
        self.buildtype = 0
        self.buildpos = vec2d(0, 0)
        
        self.buildings = []
        self.ores = []
        
        count = 0
        while count < 15:
            tempore = Ore()
            tempore.pos = vec2d(int(uniform(0, 1500)), int(uniform(0, 1500)))
            if not len(self.ores) == 0:
                flag = 1
                for orecheck in self.ores:
                    dist = orecheck.pos.get_distance(tempore.pos)
                    if dist < 120:
                        flag = 0
                if flag:
                    self.ores.append(tempore)
                    count += 1
            else:
                self.ores.append(tempore)
                count += 1
        
        firstbuild = MainBase()
        firstbuild.pos = vec2d(300, 300)
        firstbuild.target = vec2d(firstbuild.pos[0] + 10, firstbuild.pos[1])
        self.buildings.append(firstbuild)
예제 #3
0
 def test_init_main_base(self):
     firstbuild = MainBase()
     firstbuild.pos = vec2d(300, 300)
     firstbuild.target = vec2d(firstbuild.pos[0] + 10, firstbuild.pos[1])
     self.assertEqual(firstbuild.pos[0], 300)
     self.assertEqual(firstbuild.pos[1], 300)