tank_island 1.0.0

data/lib/entities/components/ai/tank_fighting_state.rb

@@ -0,0 +1,47 @@
+class TankFightingState < TankMotionState
+  def initialize(object, vision)
+    super
+    @object = object
+    @vision = vision
+  end
+
+  def update
+    change_direction if should_change_direction?
+    if substate_expired?
+      rand > 0.1 ? drive : wait
+    end
+  end
+
+  def change_direction
+    change = case rand(0..100)
+    when 0..20
+      -45
+    when 20..40
+      45
+    when 40..60
+      90
+    when 60..80
+      -90
+    when 80..90
+      135
+    when 90..100
+      -135
+    end
+    @object.physics.change_direction(
+      @object.direction + change)
+    @changed_direction_at = Gosu.milliseconds
+    @will_keep_direction_for = turn_time
+  end
+
+  def wait_time
+    rand(50..300)
+  end
+
+  def drive_time
+    rand(5000..10000)
+  end
+
+  def turn_time
+    rand(300..3000)
+  end
+end

data/lib/entities/components/ai/tank_fleeing_state.rb

@@ -0,0 +1,50 @@
+class TankFleeingState < TankMotionState
+  MAX_FLEE_TIME = 15 * 1000 # 15 seconds
+
+  def initialize(object, vision, gun)
+    super(object, vision)
+    @object = object
+    @vision = vision
+    @gun = gun
+  end
+
+  def can_flee?
+    return true unless @started_fleeing
+    Gosu.milliseconds - @started_fleeing < MAX_FLEE_TIME
+  end
+
+  def enter
+    @started_fleeing ||= Gosu.milliseconds
+  end
+
+  def update
+    change_direction if should_change_direction?
+    drive
+  end
+
+  def change_direction
+    closest_powerup = @vision.closest_powerup(
+      RepairPowerup, HealthPowerup)
+    if closest_powerup
+      angle = Utils.angle_between(
+        @object.x, @object.y,
+        closest_powerup.x, closest_powerup.y)
+      @object.physics.change_direction(
+        angle - angle % 45)
+    else
+      @object.physics.change_direction(
+        180 + @gun.desired_gun_angle -
+        @gun.desired_gun_angle % 45)
+    end
+    @changed_direction_at = Gosu.milliseconds
+    @will_keep_direction_for = turn_time
+  end
+
+  def drive_time
+    10000
+  end
+
+  def turn_time
+    rand(300..600)
+  end
+end

data/lib/entities/components/ai/tank_motion_fsm.rb

@@ -0,0 +1,102 @@
+class TankMotionFSM
+  STATE_CHANGE_DELAY = 500
+  LOCATION_CHECK_DELAY = 5000
+
+  def initialize(object, vision, gun)
+    @object = object
+    @vision = vision
+    @gun = gun
+    @roaming_state = TankRoamingState.new(object, vision)
+    @fighting_state = TankFightingState.new(object, vision)
+    @fleeing_state = TankFleeingState.new(object, vision, gun)
+    @chasing_state = TankChasingState.new(object, vision, gun)
+    @stuck_state = TankStuckState.new(object, vision, gun)
+    @navigating_state = TankNavigatingState.new(object, vision)
+    set_state(@roaming_state)
+  end
+
+  def on_collision(with)
+    @current_state.on_collision(with)
+  end
+
+  def on_damage(amount)
+    if @current_state == @roaming_state
+      set_state(@fighting_state)
+    end
+  end
+
+  def draw(viewport)
+    if $debug
+      @image && @image.draw(
+        @object.x - @image.width / 2,
+        @object.y + @object.graphics.height / 2 -
+        @image.height, 100)
+    end
+  end
+
+  def update
+    choose_state
+    @current_state.update
+  end
+
+  def set_state(state)
+    return unless state
+    return if state == @current_state
+    @last_state_change = Gosu.milliseconds
+    @current_state = state
+    state.enter
+    if $debug
+      @image = Gosu::Image.from_text(
+          $window, state.class.to_s,
+          Gosu.default_font_name, 18)
+    end
+  end
+
+  def choose_state
+    unless @vision.can_go_forward?
+      unless @current_state == @stuck_state
+        set_state(@navigating_state)
+      end
+    end
+    # Keep unstucking itself for a while
+    change_delay = STATE_CHANGE_DELAY
+    if @current_state == @stuck_state
+      change_delay *= 5
+    end
+    now = Gosu.milliseconds
+    return unless now - @last_state_change > change_delay
+    if @last_location_update.nil?
+      @last_location_update = now
+      @last_location = @object.location
+    end
+    if now - @last_location_update > LOCATION_CHECK_DELAY
+      unless @last_location.nil? || @current_state.waiting?
+        if Utils.distance_between(*@last_location, *@object.location) < 20
+          set_state(@stuck_state)
+          @stuck_state.stuck_at = @object.location
+          return
+        end
+      end
+      @last_location_update = now
+      @last_location = @object.location
+    end
+    if @gun.target
+      if @object.health.health > 40
+        if @gun.distance_to_target > BulletPhysics::MAX_DIST
+          new_state = @chasing_state
+        else
+          new_state = @fighting_state
+        end
+      else
+        if @fleeing_state.can_flee?
+          new_state = @fleeing_state
+        else
+          new_state = @fighting_state
+        end
+      end
+    else
+      new_state = @roaming_state
+    end
+    set_state(new_state)
+  end
+end

data/lib/entities/components/ai/tank_motion_state.rb

@@ -0,0 +1,84 @@
+class TankMotionState
+  def initialize(object, vision)
+    @object = object
+    @vision = vision
+  end
+
+  def enter
+    # Override if necessary
+  end
+
+  def change_direction
+    # Override
+  end
+
+  def wait_time
+    # Override and return a number
+  end
+
+  def drive_time
+    # Override and return a number
+  end
+
+  def turn_time
+    # Override and return a number
+  end
+
+  def update
+    # Override
+  end
+
+  def wait
+    @sub_state = :waiting
+    @started_waiting = Gosu.milliseconds
+    @will_wait_for = wait_time
+    @object.throttle_down = false
+  end
+
+  def waiting?
+    @sub_state == :waiting
+  end
+
+  def drive
+    @sub_state = :driving
+    @started_driving = Gosu.milliseconds
+    @will_drive_for = drive_time
+    @object.throttle_down = true
+  end
+
+  def should_change_direction?
+    return true unless @vision.can_go_forward?
+    return true unless @changed_direction_at
+    Gosu.milliseconds - @changed_direction_at >
+      @will_keep_direction_for
+  end
+
+  def substate_expired?
+    now = Gosu.milliseconds
+    case @sub_state
+    when :waiting
+      true if now - @started_waiting > @will_wait_for
+    when :driving
+      true if now - @started_driving > @will_drive_for
+    else
+      true
+    end
+  end
+
+  def on_collision(with)
+    change = case rand(0..100)
+    when 0..30
+      -90
+    when 30..60
+      90
+    when 60..70
+      135
+    when 80..90
+      -135
+    else
+      180
+    end
+    @object.physics.change_direction(
+      @object.direction + change)
+  end
+end

data/lib/entities/components/ai/tank_navigating_state.rb

@@ -0,0 +1,34 @@
+class TankNavigatingState < TankMotionState
+  def initialize(object, vision)
+    @object = object
+    @vision = vision
+  end
+
+  def update
+    change_direction if should_change_direction?
+    drive
+  end
+
+  def change_direction
+    closest_free_path = @vision.closest_free_path
+    if closest_free_path
+      @object.physics.change_direction(
+        Utils.angle_between(
+          @object.x, @object.y, *closest_free_path))
+    end
+    @changed_direction_at = Gosu.milliseconds
+    @will_keep_direction_for = turn_time
+  end
+
+  def wait_time
+    rand(10..100)
+  end
+
+  def drive_time
+    rand(1000..2000)
+  end
+
+  def turn_time
+    rand(300..1000)
+  end
+end

data/lib/entities/components/ai/tank_roaming_state.rb

@@ -0,0 +1,83 @@
+class TankRoamingState < TankMotionState
+  def initialize(object, vision)
+    super
+    @object = object
+    @vision = vision
+  end
+
+  def update
+    change_direction if should_change_direction?
+    if substate_expired?
+      rand > 0.2 ? drive : wait
+    end
+  end
+
+  def required_powerups
+    required = []
+    health = @object.health.health
+    if @object.fire_rate_modifier < 2 && health > 50
+      required << FireRatePowerup
+    end
+    if @object.speed_modifier < 1.5 && health > 50
+      required << TankSpeedPowerup
+    end
+    if health < 100
+      required << RepairPowerup
+    end
+    if health < 190
+      required << HealthPowerup
+    end
+    required
+  end
+
+  def change_direction
+    closest_powerup = @vision.closest_powerup(
+      *required_powerups)
+    if closest_powerup
+      @seeking_powerup = true
+      angle = Utils.angle_between(
+        @object.x, @object.y,
+        closest_powerup.x, closest_powerup.y)
+      @object.physics.change_direction(
+        angle - angle % 45)
+      # Go away from inaccessable powerup
+      unless @vision.can_go_forward?
+        @seeking_powerup = false
+        @object.physics.change_direction(
+          @object.direction + 180)
+      end
+    else
+      @seeking_powerup = false
+      change = case rand(0..100)
+      when 0..30
+        -45
+      when 30..60
+        45
+      when 60..80
+        90
+      when 80..100
+        -90
+      end
+      @object.physics.change_direction(
+        @object.direction + change)
+    end
+    @changed_direction_at = Gosu.milliseconds
+    @will_keep_direction_for = turn_time
+  end
+
+  def wait_time
+    rand(50..400)
+  end
+
+  def drive_time
+    rand(1000..3000)
+  end
+
+  def turn_time
+    if @seeking_powerup
+      rand(100..300)
+    else
+      rand(1000..3000)
+    end
+  end
+end

data/lib/entities/components/ai/tank_stuck_state.rb

@@ -0,0 +1,45 @@
+class TankStuckState < TankMotionState
+  attr_accessor :stuck_at
+  def initialize(object, vision, gun)
+    super(object, vision)
+    @object = object
+    @vision = vision
+    @gun = gun
+  end
+
+  def update
+    change_direction if should_change_direction?
+    drive
+  end
+
+  def change_direction
+    closest_free_path = @vision.closest_free_path_away_from(
+      @stuck_at)
+    if closest_free_path
+      @object.physics.change_direction(
+        Utils.angle_between(
+          @object.x, @object.y, *closest_free_path))
+    else
+      if @object.health.health > 50 && rand > 0.9
+        @object.shoot(*Utils.point_at_distance(
+          *@object.location,
+          @object.gun_angle,
+          150))
+      end
+    end
+    @changed_direction_at = Gosu.milliseconds
+    @will_keep_direction_for = turn_time
+  end
+
+  def wait_time
+    rand(10..100)
+  end
+
+  def drive_time
+    rand(1000..2000)
+  end
+
+  def turn_time
+    rand(1000..2000)
+  end
+end