corridor.py

#!/usr/bin/env python

from geom import vec3
from geom import vec2
import math
import geom

class Corridor(object):
	def __init__(self, c, ci, p1, p2):
#		self.room_indexes = (r1, r2)
		self.end_points = (p1, p2) # actual end points of connection
		self.geometry = [] # geometry to use or draw ( a quadrilateral )
		self.connection = c # connection formed by this corridor
		self.ends = [] # actual ends of corridor ( often shorter than connection )
		self.corridor_index = ci
		self.start_door = None
		self.end_door = None
		return
	
	def make_walls_and_doors(self, width):
		line_start = self.ends[0]
		line_end   = self.ends[1]
		self.start_door = geom.line_end_perp2d(line_start, line_end, width) # magic number == corridor width
		self.end_door   = geom.line_end_perp2d(line_end, line_start, width)
		self.geometry = (  self.start_door[0].x, self.start_door[0].y, # start door
				   self.start_door[1].x, self.start_door[1].y, #				   

				   self.start_door[1].x, self.start_door[1].y, # side
				    self.end_door[0].x, self.end_door[0].y,     #

				    self.end_door[0].x, self.end_door[0].y,     # end door
				    self.end_door[1].x, self.end_door[1].y,     #
				    
				    self.end_door[1].x, self.end_door[1].y,     # side
				    self.start_door[0].x, self.start_door[0].y  ) 
		return
	
	def make_geometry(self, width):
		self.ends = [ vec2(self.end_points[0].x, self.end_points[0].y),
			      vec2(self.end_points[1].x, self.end_points[1].y) ]
		self.make_walls_and_doors(width)
		return

	def shorten_corridor(self, amount, end, width):
		other_end = 1-end
		old_line = self.ends[end] - self.ends[other_end]
		old_line_length = old_line.length()
		delta = old_line / old_line.length()
		self.ends[end] = self.ends[other_end] + delta * old_line_length * ( 1.0 - amount )
		self.make_walls_and_doors(width)
		return

	def door_geometry(self, end):
		if (end == 0):
			return ( self.start_door[0], self.start_door[1] )
		if (end == 1):
			return ( self.end_door[0], self.end_door[1] )