vector3.py

#!/usr/bin/env python

# Vector3 class - super minimal; just adding things as I need them
import math

class Vector3:
	def __init__(self, x=0, y=0, z=0):
		self.x = x
		self.y = y
		self.z = z

	# normalizes self and returns the length
	def normalize(self):
		length2 = self.x * self.x + self.y * self.y + self.z * self.z
		length = math.sqrt(length2)
		if length != 0: #if our vector is a zero vector
			self.x /= length
			self.y /= length
			self.z /= length
		return length

	def length(self):
		length2 = self.x * self.x + self.y * self.y + self.z * self.z
		return math.sqrt(length2)

	def cross(a,b):
		return Vector3(a.y*b.z - a.z*b.y, a.z*b.x - a.x*b.z, a.x*b.y-a.y*b.x)

	def dot(a,b):
		return a.x*b.x + a.y*b.y + a.z*b.z

	def __eq__(self,vector):
		return (self.x == vector.x and self.y == vector.y and self.z == vector.z)

	def __ne__(self,vector):
		return (self.x != vector.x or self.y != vector.y or self.z != vector.z)

	def __add__(self, vector):
		return Vector3(self.x + vector.x, self.y + vector.y, self.z + vector.z)

	def __sub__(self, vector):
		return Vector3(self.x - vector.x, self.y - vector.y, self.z - vector.z)

	def __neg__(self):
		return Vector3(-self.x, -self.y, -self.z)

	def __mul__(self, scalar):
		return Vector3(self.x * scalar, self.y * scalar, self.z * scalar)

	def __rmul__(self, scalar):
		return Vector3(scalar * self.x, scalar * self.y, scalar * self.z)

	def __div__(self, scalar):
		return Vector3(self.x/scalar, self.y/scalar, self.z/scalar)

	def __repr__(self):
		return "<%f,%f,%f>" % (self.x,self.y,self.z)

	def __str__(self):
		return "<%f,%f,%f>" % (self.x,self.y,self.z)

	def __iter__(self):
		return iter([self.x,self.y,self.z])