TreD/matrix.lua at main · CalebSlot/TreD · 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
145
146
147
148
149
local vector = require("vector")
local quad   = require("quad")

Matrix = {}


Matrix.__index = Matrix

function Matrix:new()
      local matrix = {}
      setmetatable(matrix, Matrix)
      matrix.X = vector:new()
      matrix.Y = vector:new()
      matrix.Z = vector:new()
      return matrix
end
function Matrix:getX()
  return self.X
end
function Matrix:getY()
  return self.Y
end
function Matrix:getZ()
  return self.Z
end
function Matrix:multiplyV(vectorV)

  v = vector:new()
  v:setX(self.X:getX() * vectorV:getX() + self.Y:getX() * vectorV:getY() + self.Z:getX() * vectorV:getZ())
  v:setY(self.X:getY() * vectorV:getX() + self.Y:getY() * vectorV:getY() + self.Z:getY() * vectorV:getZ())
  v:setZ(self.X:getZ() * vectorV:getX() + self.Y:getZ() * vectorV:getY() + self.Z:getZ() * vectorV:getZ())
  return v

end

function Matrix:multiplyQ(quadQ)
  q = quad:new(quadQ:getFirst():getX(),quadQ:getFirst():getY(),quadQ:getSecond():getX(),quadQ:getSecond():getY(),quadQ:getThird():getX(),quadQ:getThird():getY(),quadQ:getFourth():getX(),quadQ:getFourth():getY())
  self:multiplyUpdateQ(q)
  return q
end

function Matrix:identityUpdate()
  self.X:setX(1)
  self.X:setY(0)
  self.X:setZ(0)
  self.Y:setX(0)
  self.Y:setY(1)
  self.Y:setZ(0)
  self.Z:setX(0)
  self.Z:setY(0)
  self.Z:setZ(1)
end
function Matrix:identity()
   m = Matrix:new()
   m:identiyUpdate()
   return m
end
function Matrix:multiplyUpdateQ(quadQ)

  primo   = quadQ:getFirst()
  secondo = quadQ:getSecond()
  terzo   = quadQ:getThird()
  quarto  = quadQ:getFourth()

  quadQ:updateFirst(self.X:getX() * primo:getX() + self.Y:getX() * primo:getY() + self.Z:getX() * primo:getZ(),  self.X:getY() * primo:getX() + self.Y:getY() * primo:getY() + self.Z:getY() * primo:getZ())

 quadQ:updateSecond(self.X:getX() * secondo:getX() + self.Y:getX() * secondo:getY() + self.Z:getX() * secondo:getZ(),  self.X:getY() * secondo:getX() + self.Y:getY() * secondo:getY() + self.Z:getY() * secondo:getZ())

  quadQ:updateThird(self.X:getX() * terzo:getX() + self.Y:getX() * terzo:getY() + self.Z:getX() * terzo:getZ(),  self.X:getY() * terzo:getX() + self.Y:getY() * terzo:getY() + self.Z:getY() * terzo:getZ())

   quadQ:updateFourth(self.X:getX() * quarto:getX() + self.Y:getX() * quarto:getY() + self.Z:getX() * quarto:getZ(),  self.X:getY() * quarto:getX() + self.Y:getY() * quarto:getY() + self.Z:getY() * quarto:getZ())

end

function Matrix:rotationZ(zRot)


  cos = 1
  sin = 0

  if(zRot == 90)
  then
    cos = 0
    sin = 1
  elseif(zRot == 180)
  then
     cos  = -1
     sin  = 0
  elseif(zRot == 270)
  then
     cos  = 0
     sin  = -1
  elseif(zRot == -90)
  then
    cos = 0
    sin = -1
  elseif(zRot == -180)
  then
     cos  = 1
     sin  = 0
  elseif(zRot == -270)
  then
     cos  = 0
     sin  = 1
  else
    zRot = math.rad(zRot)
    cos  = math.cos(zRot)
    sin  = math.sin(zRot)
  end

  self.X:setX(cos)
  self.X:setY(sin)
  self.X:setZ(0)

  self.Y:setX(-sin)
  self.Y:setY(cos)
  self.Y:setZ(0)

  self.Z:setX(0)
  self.Z:setY(0)
  self.Z:setZ(1)

end
  --TODO: build precalculated 90 180 270 matrixes
function Matrix:toString()
  v1 = self:getX()
  v2 = self:getY()
  v3 = self:getZ()
  return "X[" .. v1:toString() .. "]\nY[" .. v2:toString() .. "]\nZ[" .. v3:toString() .. "]"
end

function Matrix:Test()
  print("matrixRot90Z")
  m = Matrix:new()
  print(m:toString())
  m:rotationZ(90)
  print(m:toString())
  v = vector:new()
  v:setX(1)
  v:setY(1)
  v:setZ(1)
  v = m:multiplyV(v)
  print("IDENTITY")
  m:identityUpdate()
  print(m:toString())
end

return Matrix