luavec3.lua

-- by MonstersGoBoom 
--[[
Copyright (c) 2017 MonstersGoBoom
All rights reserved.

Redistribution and use in source and binary forms are permitted
provided that the above copyright notice and this paragraph are
duplicated in all such forms and that any documentation,
advertising materials, and other materials related to such
distribution and use acknowledge that the software was developed
by the <organization>. The name of the
<organization> may not be used to endorse or promote products derived
from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
]]--
local math_sin = math.sin
local math_cos = math.cos
local math_rad = math.rad 
local math_min = math.min
local math_max = math.max
local math_abs = math.abs
local math_sqrt = math.sqrt
local math_atan2 = math.atan2

local matrix = {}
setmetatable(matrix, matrix);
matrix.type = matrix
local XX,YX,ZX,WX,XY,YY,ZY,WY,XZ,YZ,ZZ,WZ,XW,YW,ZW,WW = 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16
function matrix:__call()local a={}a.m={1,0,0,0,0,1,0,0,0,0,1,0,0,0,0,1}matrix.__index=self;setmetatable(a,self)return a end
function matrix:__tostring()local a=string.format("\n[%.3f],[%.3f],[%.3f],[%.3f]\n",self.m[1],self.m[2],self.m[3],self.m[4])a=a..string.format("[%.3f],[%.3f],[%.3f],[%.3f]\n",self.m[5],self.m[6],self.m[7],self.m[8])a=a..string.format("[%.3f],[%.3f],[%.3f],[%.3f]\n",self.m[9],self.m[10],self.m[11],self.m[12])a=a..string.format("[%.3f],[%.3f],[%.3f],[%.3f]\n",self.m[13],self.m[14],self.m[15],self.m[16])return a end
function matrix:__add(a)local b=matrix()if a.type==matrix then for c=1,16 do b.m[c]=self.m[c]+a.m[c]end end;return b end
function matrix:__sub(a)local b=matrix()if a.type==matrix then for c=1,16 do b.m[c]=self.m[c]-a.m[c]end end;return b end
function matrix:__mul(a)local b=matrix()if a.type==matrix then local c=self;local d=a;b.m[XX]=c.m[XX]*d.m[XX]+c.m[XY]*d.m[YX]+c.m[XZ]*d.m[ZX]b.m[XY]=c.m[XX]*d.m[XY]+c.m[XY]*d.m[YY]+c.m[XZ]*d.m[ZY]b.m[XZ]=c.m[XX]*d.m[XZ]+c.m[XY]*d.m[YZ]+c.m[XZ]*d.m[ZZ]b.m[XW]=c.m[XX]*d.m[XW]+c.m[XY]*d.m[YW]+c.m[XZ]*d.m[ZW]+c.m[XW]b.m[YX]=c.m[YX]*d.m[XX]+c.m[YY]*d.m[YX]+c.m[YZ]*d.m[ZX]b.m[YY]=c.m[YX]*d.m[XY]+c.m[YY]*d.m[YY]+c.m[YZ]*d.m[ZY]b.m[YZ]=c.m[YX]*d.m[XZ]+c.m[YY]*d.m[YZ]+c.m[YZ]*d.m[ZZ]b.m[YW]=c.m[YX]*d.m[XW]+c.m[YY]*d.m[YW]+c.m[YZ]*d.m[ZW]+c.m[YW]b.m[ZX]=c.m[ZX]*d.m[XX]+c.m[ZY]*d.m[YX]+c.m[ZZ]*d.m[ZX]b.m[ZY]=c.m[ZX]*d.m[XY]+c.m[ZY]*d.m[YY]+c.m[ZZ]*d.m[ZY]b.m[ZZ]=c.m[ZX]*d.m[XZ]+c.m[ZY]*d.m[YZ]+c.m[ZZ]*d.m[ZZ]b.m[ZW]=c.m[ZX]*d.m[XW]+c.m[ZY]*d.m[YW]+c.m[ZZ]*d.m[ZW]+c.m[ZW]return b end end
-- returns a matrix, don't pass one in
function matrix.rx(a)local b=matrix()local c=math_sin(a)local d=math_cos(a)b.m[YY]=d;b.m[YZ]=c;b.m[ZY]=-c;b.m[ZZ]=d;return b end
function matrix.ry(a)local b=matrix()local c=math_sin(a)local d=math_cos(a)b.m[XX]=d;b.m[XZ]=-c;b.m[ZX]=c;b.m[ZZ]=d;return b end
function matrix.rz(a)local b=matrix()local c=math_sin(a)local d=math_cos(a)b.m[XX]=d;b.m[XY]=c;b.m[YX]=-c;b.m[YY]=d;return b end
local vec3 = {}
setmetatable(vec3, vec3);
vec3.type = vec3
function vec3:__call(a,b,c)local d={}vec3.__index=self;setmetatable(d,self)d.x=a or 0;d.y=b or 0;d.z=c or 0;return d end
function vec3:__tostring()return string.format("(%03f,%03f,%03f)",self.x,self.y,self.z)end
function vec3:__len() return self:length() end
function vec3:__add(a)return vec3(self.x+a.x,self.y+a.y,self.z+a.z)end
function vec3:__sub(right) return vec3(self.x - right.x,self.y - right.y,self.z - right.z)end
function vec3:__div(right) return vec3(self.x / right.x,self.y / right.y,self.z / right.z)end
function vec3:length() return math_sqrt(self.x * self.x + self.y * self.y + self.z * self.z) end
function vec3:distance(to) a = self - to return #a end
function vec3:dot(right) local mag = 1 / #self * #right return (self.x * right.x + self.y * right.y + self.z * right.z) * mag end
function vec3:cross(right) local c = vec3() c.x = self.y * right.z - self.z * right.y	 c.y = self.z * right.x - self.x * right.z c.z = self.x * right.y - self.y * right.x return c end
function vec3:lerp(a,b,amount) local c = vec3() c.x = a.x + (b.x - a.x) * amount c.y = a.y + (b.y - a.y) * amount c.z = a.z + (b.z - a.z) * amount return c end
function vec3:normalize()local a=#self;if a==0 then self.x=0;self.y=0;self.z=0;return end;a=1.0/a;self.x=self.x*a;self.y=self.y*a;self.z=self.z*a end
function vec3:__mul(right)
	if type(right)=="number" then 
		return vec3(self.x * right,self.y * right,self.z * right)
	elseif right.type == vec3 then 
		return vec3(self.x * right.x,self.y * right.y,self.z * right.z)
	elseif right.type == matrix then 
		local a = right
		temp = vec3()		
		temp.x = self.x*a.m[XX] + self.y*a.m[XY] + self.z*a.m[XZ] + a.m[XW];
		temp.y = self.x*a.m[YX] + self.y*a.m[YY] + self.z*a.m[YZ] + a.m[YW];
		temp.z = self.x*a.m[ZX] + self.y*a.m[ZY] + self.z*a.m[ZZ] + a.m[ZW];
		return temp
	else
		trace("not a vec or number")
	end
end
local mat = matrix()
local mat2 = matrix.rx(32)
trace("mat")
trace(mat)
trace("mat2")
trace(mat2)
trace("mat * mat2")
local m3 = mat * mat2
trace(m3)
local va = vec3(32,0,64)
local vb = vec3(1,2,3)
trace("va ---")
trace(va)
trace("vb ---")
trace(vb)
local vc = va + vb * 3
trace("---")
trace(vc)
vc = vc * mat2
trace(vc)