• file
• latest
• revisions
• annotate
• diff
• comparison
• raw

geometry.py@55b4c97d44c5 (annotated)

geometry.py

Sun, 10 Dec 2017 15:37:26 +0200

author

Santeri Piippo

date

Sun, 10 Dec 2017 15:37:26 +0200

changeset 0

55b4c97d44c5

child 1

5411a25cfca7

permissions

-rw-r--r--

Initial commit with half-done parsing function

0 55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	1	class Vertex:
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	2	def __init__(self, x, y, z):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	3	self.x, self.y, self.z = x, y, z
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	4	def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	5	return str.format('Vertex({!r}, {!r}, {!r})', self.x, self.y, self.z)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	6	def distance_to(self, other):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	7	# can't use hypot because of 3 arguments
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	8	from math import sqrt
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	9	return sqrt(
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	10	(self.x - other.x) ** 2 +
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	11	(self.y - other.y) ** 2 +
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	12	(self.z - other.z) ** 2
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	13	)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	14
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	15	class LineSegment:
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	16	def __init__(self, v1, v2):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	17	self.v1, self.v2 = v1, v2
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	18	def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	19	return str.format('LineSegment({!r}, {!r})', self.v1, self.v2)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	20
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	21	def is_real(number):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	22	return isinstance(number, int) or isinstance(number, float)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	23
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	24	class TransformationMatrix:
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	25	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	26	A 3×3 matrix forming the top-left portion of a full 4×4 transformation
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	27	matrix.
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	28	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	29	def __init__(self, values):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	30	assert(all(is_real(x) for x in values))
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	31	assert len(values) == 9
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	32	self.values = values
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	33	def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	34	return str.format('TransformationMatrix({!r})', self.values)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	35	def __getitem__(self, index):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	36	return self.values[index]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	37
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	38	def complete_matrix(matrix, anchor):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	39	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	40	Combines a 3×3 matrix and an anchor vertex into a full 4×4
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	41	transformation matrix.
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	42	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	43	return [
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	44	matrix[0], matrix[1], matrix[2], anchor.x,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	45	matrix[3], matrix[4], matrix[5], anchor.y,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	46	matrix[6], matrix[7], matrix[8], anchor.z,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	47	0, 0, 0, 1,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	48	]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	49
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	50	def transform(vertex, transformation_matrix):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	51	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	52	Transforms a vertex by a 4×4 transformation matrix.
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	53	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	54	u = transformation_matrix[0] * vertex.x \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	55	+ transformation_matrix[1] * vertex.y \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	56	+ transformation_matrix[2] * vertex.z \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	57	+ transformation_matrix[3]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	58	v = transformation_matrix[4] * vertex.x \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	59	+ transformation_matrix[5] * vertex.y \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	60	+ transformation_matrix[6] * vertex.z \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	61	+ transformation_matrix[7]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	62	w = transformation_matrix[8] * vertex.x \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	63	+ transformation_matrix[9] * vertex.y \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	64	+ transformation_matrix[10] * vertex.z \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	65	+ transformation_matrix[11]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	66	return Vertex(u, v, w)