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

geometry.py@5411a25cfca7 (annotated)

geometry.py

Sun, 10 Dec 2017 15:45:50 +0200

author

Santeri Piippo

date

Sun, 10 Dec 2017 15:45:50 +0200

changeset 1

5411a25cfca7

parent 0

55b4c97d44c5

child 3

1dc58f44d556

permissions

-rw-r--r--

Parsing function complete

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
1 5411a25cfca7 Parsing function complete Santeri Piippo parents: 0 diff changeset	21	class Polygon:
5411a25cfca7 Parsing function complete Santeri Piippo parents: 0 diff changeset	22	def __init__(self, vertices):
5411a25cfca7 Parsing function complete Santeri Piippo parents: 0 diff changeset	23	self.vertices = vertices
5411a25cfca7 Parsing function complete Santeri Piippo parents: 0 diff changeset	24	def __repr__(self):
5411a25cfca7 Parsing function complete Santeri Piippo parents: 0 diff changeset	25	return str.format('Polygon({!r})', self.vertices)
5411a25cfca7 Parsing function complete Santeri Piippo parents: 0 diff changeset	26
0 55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	27	def is_real(number):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	28	return isinstance(number, int) or isinstance(number, float)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	29
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	30	class TransformationMatrix:
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	31	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	32	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	33	matrix.
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	34	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	35	def __init__(self, values):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	36	assert(all(is_real(x) for x in values))
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	37	assert len(values) == 9
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	38	self.values = values
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	39	def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	40	return str.format('TransformationMatrix({!r})', self.values)
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	41	def __getitem__(self, index):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	42	return self.values[index]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	43
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	44	def complete_matrix(matrix, anchor):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	45	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	46	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	47	transformation matrix.
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	return [
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	50	matrix[0], matrix[1], matrix[2], anchor.x,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	51	matrix[3], matrix[4], matrix[5], anchor.y,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	52	matrix[6], matrix[7], matrix[8], anchor.z,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	53	0, 0, 0, 1,
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	54	]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	55
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	56	def transform(vertex, transformation_matrix):
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	57	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	58	Transforms a vertex by a 4×4 transformation matrix.
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	59	'''
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	60	u = transformation_matrix[0] * vertex.x \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	61	+ transformation_matrix[1] * vertex.y \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	62	+ transformation_matrix[2] * vertex.z \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	63	+ transformation_matrix[3]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	64	v = transformation_matrix[4] * vertex.x \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	65	+ transformation_matrix[5] * vertex.y \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	66	+ transformation_matrix[6] * vertex.z \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	67	+ transformation_matrix[7]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	68	w = transformation_matrix[8] * vertex.x \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	69	+ transformation_matrix[9] * vertex.y \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	70	+ transformation_matrix[10] * vertex.z \
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	71	+ transformation_matrix[11]
55b4c97d44c5 Initial commit with half-done parsing function Santeri Piippo parents: diff changeset	72	return Vertex(u, v, w)