path: root/samples/expert-tankspank.rb

# Tankspank
# kevin conner
# connerk@gmail.com
# version 3, 13 March 2008
# this code is free, do what you like with it!

$width, $height = 700, 500
$camera_tightness = 0.1

module Collisions
	def contains? x, y
		not (x < west or x > east or y < north or y > south)
	end
	
	def intersects? other
		not (other.east < west or other.west > east or
			other.south < north or other.north > south)
	end
end

class Building
	include Collisions
	
	attr_reader :west, :east, :north, :south
	
	def initialize(west, east, north, south)
		@west, @east, @north, @south = west, east, north, south
		@top, @bottom = 1.1 + rand(3) * 0.15, 1.0
		
		color = (1..3).collect { 0.2 + 0.4 * rand }
		color << 0.9
		@stroke = $app.rgb *color
		color[-1] = 0.3
		@fill = $app.rgb *color
	end
	
	def draw
		$app.stroke @stroke
		$app.fill @fill
		Opp.draw_opp_box(@west, @east, @north, @south, @top, @bottom)
	end
end

module Guidance
	def guidance_system x, y, dest_x, dest_y, angle
		vx, vy = dest_x - x, dest_y - y
		if vx.abs < 0.1 and vy.abs <= 0.1
			yield 0, 0
		else
			length = Math.sqrt(vx * vx + vy * vy)
			vx /= length
			vy /= length
			ax, ay = Math.cos(angle), Math.sin(angle)
			cos_between = vx * ax + vy * ay
			sin_between = vx * -ay + vy * ax
			yield sin_between, cos_between
		end
	end
end

module Life
	attr_reader :health
	def dead?
		@health == 0
	end
	def hurt damage
		@health = [@health - damage, 0].max
	end
end

class Tank
	include Collisions
	include Guidance
	include Life
	# ^ sounds like insurance
	
	@@collide_size = 15
	def west; @x - @@collide_size; end
	def east; @x + @@collide_size; end
	def north; @y - @@collide_size; end
	def south; @y + @@collide_size; end
	
	attr_reader :x, :y
	
	def initialize
		@x, @y = 0, -125
		@last_x, @last_y = @x, @y
		@tank_angle = 0.0
		@dest_x, @dest_y = 0, 0
		@acceleration = 0.0
		@speed = 0.0
		@moving = false
		
		@aim_angle = 0.0
		@target_x, @target_y = 0, 0
		@aimed = false
		
		@health = 100
	end
	
	def set_destination
		@dest_x, @dest_y = @target_x, @target_y
		@moving = true
	end
	
	def fire
		Opp.add_shell Shell.new(@x + 30 * Math.cos(@aim_angle),
			@y + 30 * Math.sin(@aim_angle), @aim_angle)
	end
	
	def update button, mouse_x, mouse_y
		@target_x, @target_y = mouse_x, mouse_y
		
		if @moving
			guidance_system @x, @y, @dest_x, @dest_y, @tank_angle do |direction, on_target|
				turn direction
				@acceleration = on_target * 0.25
			end
			
			distance = Math.sqrt((@dest_x - @x) ** 2 + (@dest_y - @y) ** 2)
			@moving = false if distance < 50
		else
			@acceleration = 0.0
		end
		
		guidance_system @x, @y, @target_x, @target_y, @aim_angle do |direction, on_target|
			aim direction
			@aimed = on_target > 0.98
		end
		
		integrity = @health / 100.0 # the more hurt you are, the slower you go
		@speed = [[@speed + @acceleration, 5.0 * integrity].min, -3.0 * integrity].max
		@speed *= 0.9 if !@moving
		
		@last_x, @last_y = @x, @y
		@x += @speed * Math.cos(@tank_angle)
		@y += @speed * Math.sin(@tank_angle)
	end
	
	def collide_and_stop
		@x, @y = @last_x, @last_y
		hurt @speed.abs * 3 + 5
		@speed = 0
		@moving = false
	end
	
	def turn direction
		@tank_angle += [[-0.03, direction].max, 0.03].min
	end
	
	def aim direction
		@aim_angle += [[-0.1, direction].max, 0.1].min
	end
	
	def draw
		$app.stroke $app.blue
		$app.fill $app.blue(0.4)
		Opp.draw_opp_rect @x - 20, @x + 20, @y - 15, @y + 15, 1.05, @tank_angle
		#Opp.draw_opp_box @x - 20, @x + 20, @y - 20, @y + 20, 1.03, 1.0
		Opp.draw_opp_rect @x - 10, @x + 10, @y - 7, @y + 7, 1.05, @aim_angle
		x, unused1, y, unused2 = Opp.project(@x, 0, @y, 0, 1.05)
		$app.line x, y, x + 25 * Math.cos(@aim_angle), y + 25 * Math.sin(@aim_angle)
		
		$app.stroke $app.red
		$app.fill $app.red(@aimed ? 0.4 : 0.1)
		Opp.draw_opp_oval @target_x - 10, @target_x + 10, @target_y - 10, @target_y + 10, 1.00
		
		if @moving
			$app.stroke $app.green
			$app.fill $app.green(0.2)
			Opp.draw_opp_oval @dest_x - 20, @dest_x + 20, @dest_y - 20, @dest_y + 20, 1.00
		end
	end
end

class Shell
	attr_reader :x, :y
	
	def initialize x, y, angle
		@x, @y, @angle = x, y, angle
		@speed = 10.0
	end
	
	def update
		@x += @speed * Math.cos(@angle)
		@y += @speed * Math.sin(@angle)
	end
	
	def draw
		$app.stroke $app.red
		$app.fill $app.red(0.1)
		Opp.draw_opp_box @x - 2, @x + 2, @y - 2, @y + 2, 1.05, 1.04
	end
end

class Opp
	def self.new_game
		@offset_x, @offset_y = 0, 0
		@buildings = [
			[-1000, -750, -750, -250],
			[-500, 250, -750, -250],
			[500, 1000, -750, -500],
			[750, 1250, -250, 0],
			[750, 1250, 250, 750],
			[250, 500, 0, 750],
			[-250, 0, 0, 500],
			[-500, 0, 750, 1000],
			[-1000, -500, 0, 500],
			[400, 600, -350, -150]
		].collect { |p| Building.new *p }
		@shells = []
		@boundary = [-1250, 1500, -1250, 1250]
		@tank = Tank.new
		@center_x, @center_y = $app.width / 2, $app.height / 2
	end
	
	def self.tank
		@tank
	end
	
	def self.read_input
		@input = $app.mouse
	end
	
	def self.update_scene
		button, x, y = @input
		x += @offset_x - @center_x
		y += @offset_y - @center_y
		
		@tank.update(button, x, y) if !@tank.dead?
		@buildings.each do |b|
			@tank.collide_and_stop if b.intersects? @tank
		end
		
		@shells.each { |s| s.update }
		@buildings.each do |b|
			@shells.reject! do |s|
				b.contains?(s.x, s.y)
			end
		end
		#collide shells with tanks -- don't need this until there are enemy tanks
		#@shells.reject! do |s|
		#	@tank.contains?(s.x, s.y)
		#end
		
		$app.clear do
			@offset_x += $camera_tightness * (@tank.x - @offset_x)
			@offset_y += $camera_tightness * (@tank.y - @offset_y)
			
			$app.background $app.black
			@center_x, @center_y = $app.width / 2, $app.height / 2
			
			$app.stroke $app.red(0.9)
			$app.nofill
			draw_opp_box *(@boundary + [1.1, 1.0, false])
			
			@tank.draw
			@shells.each { |s| s.draw }
			@buildings.each { |b| b.draw }
		end
	end
	
	def self.add_shell shell
		@shells << shell
		@shells.shift if @shells.size > 10
	end
	
	def self.project left, right, top, bottom, depth
		[left, right].collect { |x| @center_x + depth * (x - @offset_x) } +
			[top, bottom].collect { |y| @center_y + depth * (y - @offset_y) }
	end
	
	# here "front" and "back" push the rect into and out of the window.
	# 1.0 means your x and y units are pixels on the surface.
	# greater than that brings the box closer.  less pushes it back.  0.0 => infinity.
	# the front will be filled but the rest is wireframe only.
	def self.draw_opp_box left, right, top, bottom, front, back, occlude = true
		near_left, near_right, near_top, near_bottom = project(left, right, top, bottom, front)
		far_left, far_right, far_top, far_bottom = project(left, right, top, bottom, back)
		
		# determine which sides of the box are visible
		if occlude
			draw_left = @center_x < near_left
			draw_right = near_right < @center_x
			draw_top = @center_y < near_top
			draw_bottom = near_bottom < @center_y
		else
			draw_left, draw_right, draw_top, draw_bottom = [true] * 4
		end
		
		# draw lines for the back edges
		$app.line far_left, far_top, far_right, far_top if draw_top
		$app.line far_left, far_bottom, far_right, far_bottom if draw_bottom
		$app.line far_left, far_top, far_left, far_bottom if draw_left
		$app.line far_right, far_top, far_right, far_bottom if draw_right
		
		# draw lines to connect the front and back
		$app.line near_left, near_top, far_left, far_top if draw_left or draw_top
		$app.line near_right, near_top, far_right, far_top if draw_right or draw_top
		$app.line near_left, near_bottom, far_left, far_bottom if draw_left or draw_bottom
		$app.line near_right, near_bottom, far_right, far_bottom if draw_right or draw_bottom
		
		# draw the front, filled
		$app.rect near_left, near_top, near_right - near_left, near_bottom - near_top
	end
	
	def self.draw_opp_rect left, right, top, bottom, depth, angle, with_x = false
		pl, pr, pt, pb = project(left, right, top, bottom, depth)
		cos = Math.cos(angle)
		sin = Math.sin(angle)
		cx, cy = (pr + pl) / 2.0, (pb + pt) / 2.0
		points = [[pl, pt], [pr, pt], [pr, pb], [pl, pb]].collect do |x, y|
			[cx + (x - cx) * cos - (y - cy) * sin,
				cy + (x - cx) * sin + (y - cy) * cos]
		end
		
		$app.line *(points[0] + points[1])
		$app.line *(points[1] + points[2])
		$app.line *(points[2] + points[3])
		$app.line *(points[3] + points[0])
	end
	
	def self.draw_opp_oval left, right, top, bottom, depth
		pl, pr, pt, pb = project(left, right, top, bottom, depth)
		$app.oval(pl, pt, pr - pl, pb - pt)
	end
	
	def self.draw_opp_plane x1, y1, x2, y2, front, back, stroke_color
		near_x1, near_x2, near_y1, near_y2 = project(x1, x2, y1, y2, front)
		far_x1, far_x2, far_y1, far_y2 = project(x1, x2, y1, y2, back)
		
		$app.stroke stroke_color
		
		$app.line far_x1, far_y1, far_x2, far_y2
		$app.line far_x1, far_y1, near_x1, near_y1
		$app.line far_x2, far_y2, near_x2, near_y2
		$app.line near_x1, near_y1, near_x2, near_y2
	end
end

Shoes.app :width => $width, :height => $height do
	$app = self
	
	Opp.new_game
	@playing = true
	
	keypress do |key|
		if @playing
			if key == "1" or key == "z"
				Opp.tank.set_destination
			elsif key == "2" or key == "x" or key == " "
				Opp.tank.fire
			end
		else
			if key == "n"
				Opp.new_game
				@playing = true
			end
		end
	end
	
	click do |button, x, y|
		if @playing
			if button == 1
				Opp.tank.set_destination
			else
				Opp.tank.fire
			end
		end
	end
	
	game_over_count = -1
	animate(60) do
		Opp.read_input if @playing
		Opp.update_scene
		
		@playing = false if Opp.tank.dead?
		if !@playing
			stack do
				banner "Game Over", :stroke => white, :margin => 10
				caption "learn to drive!", :stroke => white, :margin => 20
			end
		end
	end
end