Ejemplo n.º 1
0
    def run_ai(self):
        if self.micro_orders == []:
            cmd, pos, target = self.current_order
        else:
            cmd, pos, target = self.micro_orders[0]

        self._passive_ai()
        self._attack_ai()
        self._help_ai()

        if cmd == "stop" or cmd == "hold position":
            self.velocity = [0, 0, 0]

            if target == 0:
                self.next_order()

        elif cmd == "move":
            self._move_ai(pos)

            if vectors.distance(self.pos, pos) <= vectors.total_velocity(self.velocity):
                self.pos = pos
                self.velocity = [0, 0, 0]
                self.next_order()

        elif cmd == "attack":
            target = self.get_first_target()

            # If we have a target, lets move closer to it
            if target != None:
                # First, are we within optimum range of our target?
                # If not then we need to get closer
                target_distance = vectors.distance(self.pos, target.pos)
                if target_distance > self.optimum_attack_range:
                    attack_pos = vectors.get_midpoint(self.pos, target.pos, self.optimum_attack_range)
                    self._move_ai(attack_pos)
                else:
                    # If we are close enough then we can slow down
                    self._decelerate_ai()

        elif cmd == "aid":
            if target == None:
                target = self.get_first_ally()

            # If we have a target, lets move closer to it
            if target != None:
                dist = vectors.distance(self.pos, target.pos)

                if dist > self.optimum_heal_range:
                    target_pos = vectors.get_midpoint(self.pos, target.pos, self.optimum_heal_range)
                    self._move_ai(target_pos)
                else:
                    self._decelerate_ai()

        else:
            raise Exception("No handler for cmd %s (pos: %s, target: %s)" % (cmd, pos, target))

        # Do we have something to build?
        if self.build_queue != []:
            pass
Ejemplo n.º 2
0
    def test_midpoint(self):
        vals = (
            ([10, 10, 0], [15, 5, 0], 1, [10.707, 9.292, 0]),
            ([10, 10, 0], [15, 5, 0], 5, [13.535, 6.464, 0]),
            ([0, 0, 0], [100, 100, 0], 20, [14.14, 14.14, 0]),

            # Now 3D
            ([10, 10, 10], [15, 5, 0], 5, [13.535, 6.464, 14.082]),
        )

        for pos1, pos2, distance, expected in vals:
            x, y, z = vectors.get_midpoint(pos1, pos2, distance)

            try:
                self.assertAlmostEqual(expected[0], x, places=2)
                self.assertAlmostEqual(expected[1], y, places=2)
                self.assertAlmostEqual(expected[2], z, places=2)
            except Exception as e:
                print("\n\nTrying to midpoint({}, {}, {})\n\n".format(
                    pos1, pos2, distance))
                raise
Ejemplo n.º 3
0
 def test_midpoint(self):
     vals = (
         ([10,10,0], [15,5,0], 1, [10.707, 9.292, 0]),
         ([10,10,0], [15,5,0], 5, [13.535, 6.464, 0]),
         
         ([0,0,0], [100,100,0], 20, [14.14, 14.14, 0]),
         
         # Now 3D
         ([10,10,10], [15,5,0], 5, [13.535, 6.464, 14.082]),
     )
     
     for pos1, pos2, distance, expected in vals:
         x,y,z = vectors.get_midpoint(pos1, pos2, distance)
         
         try:
             self.assertAlmostEqual(expected[0], x, places=2)
             self.assertAlmostEqual(expected[1], y, places=2)
             self.assertAlmostEqual(expected[2], z, places=2)
         except Exception as e:
             print("\n\nTrying to midpoint({}, {}, {})\n\n".format(pos1, pos2, distance))
             raise
Ejemplo n.º 4
0
    def run_ai(self):
        if self.micro_orders == []:
            cmd, pos, target = self.current_order
        else:
            cmd, pos, target = self.micro_orders[0]

        self._passive_ai()
        self._attack_ai()
        self._help_ai()

        if cmd == "stop" or cmd == "hold position":
            self.velocity = [0, 0, 0]

            if target == 0:
                self.next_order()

        elif cmd == "move":
            self._move_ai(pos)

            if vectors.distance(self.pos, pos) <= vectors.total_velocity(
                    self.velocity):
                self.pos = pos
                self.velocity = [0, 0, 0]
                self.next_order()

        elif cmd == "attack":
            target = self.get_first_target()

            # If we have a target, lets move closer to it
            if target != None:
                # First, are we within optimum range of our target?
                # If not then we need to get closer
                target_distance = vectors.distance(self.pos, target.pos)
                if target_distance > self.optimum_attack_range:
                    attack_pos = vectors.get_midpoint(
                        self.pos, target.pos, self.optimum_attack_range)
                    self._move_ai(attack_pos)
                else:
                    # If we are close enough then we can slow down
                    self._decelerate_ai()

        elif cmd == "aid":
            if target == None:
                target = self.get_first_ally()

            # If we have a target, lets move closer to it
            if target != None:
                dist = vectors.distance(self.pos, target.pos)

                if dist > self.optimum_heal_range:
                    target_pos = vectors.get_midpoint(self.pos, target.pos,
                                                      self.optimum_heal_range)
                    self._move_ai(target_pos)
                else:
                    self._decelerate_ai()

        else:
            raise Exception("No handler for cmd %s (pos: %s, target: %s)" %
                            (cmd, pos, target))

        # Do we have something to build?
        if self.build_queue != []:
            pass