channel07/Vector.lua at master · digitalcreature/channel07 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
require "oop"
require "util"

Vector = class() do

	local base = Vector

	function Vector.one() return 1, 1, 1 end
	function Vector.zero() return 0, 0, 0 end
	function Vector.north() return 0, -1, 0 end
	function Vector.south() return 0, 1, 0 end
	function Vector.east() return 1, 0, 0 end
	function Vector.west() return -1, 0, 0 end
	function Vector.up() return 0, 0, 1 end
	function Vector.down() return 0, 0, -1 end

	function Vector.random()
		return util.calln(3, math.random, -1, 1)
	end

	local function args(x, y, z)
		if type(x) == "table" then
			return x.x, x.y, x.z
		else
			return x, y, z
		end
	end

	function base:init(x, y, z)
		self:set(x, y, z)
	end

	function base:set(x, y, z)
		x, y, z = args(x, y, z)
		self.x = x or self.x or 0
		self.y = y or self.y or 0
		self.z = z or self.z or 0
		return self
	end

	function base:xy(z)
		return self.x, self.y, z
	end

	function base:xyz()
		return self.x, self.y, self.z
	end

	function base:add(x, y, z)
		local x, y, z = args(x, y, z)
		self.x = self.x + x
		self.y = self.y + y
		self.z = self.z + z
		return self
	end

	function base:sub(x, y, z)
		local x, y, z = args(x, y, z)
		self.x = self.x - x
		self.y = self.y - y
		self.z = self.z - z
		return self
	end

	function base:scale(x, y, z)
		local x, y, z = args(x, y, z)
		y = y or x
		z = z or x
		self.x = self.x * x
		self.y = self.y * y
		self.z = self.z * z
		return self
	end

	function base:dist2(x, y, z)
		local x, y, z = args(x, y, z)
		x = x - self.x
		y = y - self.y
		z = z - self.z
		return x * x + y * y + z * z
	end

	function base:dist(x, y, z)
		local x, y, z = args(x, y, z)
		return math.sqrt(self:dist2(x, y, z))
	end

	function base:dot(x, y, z)
		local x, y, z = args(x, y, z)
		return self.x * x + self.y * y + self.z * z
	end

	function base:cross(x, y, z)
		local x, y, z = args(x, y, z)
		return
			self.y * z - self.z * y,
			self.z * x - self.x * z,
			self.x * y - self.y * x
	end

	function base:len2()
		return self:dot(self)
	end

	function base:len()
		return math.sqrt(self:len2())
	end

	function base:__tostring()
		local x = math.floor(self.x * 100) / 100
		local y = math.floor(self.y * 100) / 100
		local z = math.floor(self.z * 100) / 100
		return "<"..x..", "..y..", "..z..">"
	end

	function base.__eq(a, b)
		return
			a.x == b.x and
			a.y == b.y and
			a.z == b.z
	end

	function base:rotate2d(r)
		local x, y = self.x, self.y
		self.x = x * math.cos(-r) - y * math.sin(-r)
		self.y = x * math.sin(-r) + y * math.cos(-r)
		return self
	end

	function base:angle2d()
		return - math.atan2(self.y, self.x)
	end

	function base:norm()
		return self:scale(1 / self:len())
	end

	function base:projectscalar(x, y, z)
		local x, y, z = args(x, y, z)
		local len = math.sqrt(x * x + y * y + z * z)
		return self:dot(x, y, z) / len
	end

end