i-mic-rts/lib/entity.rb

class IMICRTS
  class Entity
    Stub = Struct.new(:name, :type, :cost, :description, :setup)
    @entities = {}

    def self.get(name)
      @entities.dig(name)
    end

    def self.define_entity(name, type, cost, description, &block)
      if entity = get(name)
        raise "#{name.inspect} is already defined!"
      else
        @entities[name] = Stub.new(name, type, cost, description, block)
      end
    end

    attr_reader :player, :id, :name, :type
    attr_accessor :position, :angle, :radius, :target, :state,
                  :movement, :health, :max_health,
                  :turret
    def initialize(name:, player:, id:, position:, angle:, director:)
      @player = player
      @id = id
      @position = position
      @angle = angle

      @radius = 32 / 2
      @target = nil
      @state  = :idle

      if entity = Entity.get(name)
        @name = entity.name
        @type = entity.type

        entity.setup.call(self, director)
      else
        raise "Failed to find entity #{name.inspect} definition"
      end

      @goal_color   = Gosu::Color.argb(175, 25, 200, 25)
      @target_color = Gosu::Color.argb(175, 200, 25, 25)
    end

    def serialize
    end

    def deserialize
    end

    def body_image=(image)
      @body_image = Gosu::Image.new("#{IMICRTS::ASSETS_PATH}/#{image}", retro: true)
    end

    def shell_image=(image)
      @shell_image = Gosu::Image.new("#{IMICRTS::ASSETS_PATH}/#{image}", retro: true)
    end

    def turret_body_image=(image)
      @turret_body_image = Gosu::Image.new("#{IMICRTS::ASSETS_PATH}/#{image}", retro: true)
    end

    def turret_shell_image=(image)
      @turret_shell_image = Gosu::Image.new("#{IMICRTS::ASSETS_PATH}/#{image}", retro: true)
    end

    def target=(entity)
      @target = entity
    end

    def hit?(x_or_vector, y = nil)
      vector = nil
      if x_or_vector.is_a?(CyberarmEngine::Vector)
        vector = x_or_vector
      else
        raise "Y cannot be nil!" if y.nil?
        vector = CyberarmEngine::Vector.new(x_or_vector, y)
      end

      @position.distance(vector) < @radius + 1
    end

    def draw
      @body_image.draw_rot(@position.x, @position.y, @position.z, @angle)
      @shell_image.draw_rot(@position.x, @position.y, @position.z, @angle, 0.5, 0.5, 1, 1, @player.color)
    end

    def update
      rotate_towards(@target) if @target && @movement
    end

    def tick(tick_id)
      @on_tick.call if @on_tick
    end

    def on_tick(&block)
      @on_tick = block
    end

    def selected_draw
      draw_radius
      draw_gizmos
    end

    def draw_radius
      Gosu.draw_circle(@position.x, @position.y, @radius, ZOrder::ENTITY_RADIUS, @player.color)
    end

    def draw_gizmos
      Gosu.draw_rect(@position.x - @radius, @position.y - (@radius + 2), @radius * 2, 2, Gosu::Color::GREEN, ZOrder::ENTITY_GIZMOS)

      if @target.is_a?(IMICRTS::Entity)
        Gosu.draw_line(@position.x, @position.y, @target_color, @target.position.x, @target.position.y, @target_color, ZOrder::ENTITY_GIZMOS) if @target
      else
        Gosu.draw_line(@position.x, @position.y, @goal_color, @target.x, @target.y, @goal_color, ZOrder::ENTITY_GIZMOS) if @target
      end
    end

    def rotate_towards(vector)
      _angle = Gosu.angle(@position.x, @position.y, vector.x, vector.y)
      a = (360.0 + (_angle - @angle)) % 360.0

      # Fails if vector is directly behind entity
      if @angle.between?(_angle - 3, _angle + 3)
        @angle = _angle
      elsif a < 180
        @angle -= 1.0
      else
        @angle += 1.0
      end

      @angle %= 360.0
    end
  end
end

Dir.glob("#{IMICRTS::GAME_ROOT_PATH}/lib/entities/*.rb").each do |entity|
  require_relative entity
end