builtin/common/vector.lua

   1
   2 vector = {}
   3
   4 function vector.new(a, b, c)
   5         if type(a) == "table" then
   6                 assert(a.x and a.y and a.z, "Invalid vector passed to vector.new()")
   7                 return {x=a.x, y=a.y, z=a.z}
   8         elseif a then
   9                 assert(b and c, "Invalid arguments for vector.new()")
  10                 return {x=a, y=b, z=c}
  11         end
  12         return {x=0, y=0, z=0}
  13 end
  14
  15 function vector.equals(a, b)
  16         return a.x == b.x and
  17                a.y == b.y and
  18                a.z == b.z
  19 end
  20
  21 function vector.length(v)
  22         return math.hypot(v.x, math.hypot(v.y, v.z))
  23 end
  24
  25 function vector.normalize(v)
  26         local len = vector.length(v)
  27         if len == 0 then
  28                 return {x=0, y=0, z=0}
  29         else
  30                 return vector.divide(v, len)
  31         end
  32 end
  33
  34 function vector.round(v)
  35         return {
  36                 x = math.floor(v.x + 0.5),
  37                 y = math.floor(v.y + 0.5),
  38                 z = math.floor(v.z + 0.5)
  39         }
  40 end
  41
  42 function vector.distance(a, b)
  43         local x = a.x - b.x
  44         local y = a.y - b.y
  45         local z = a.z - b.z
  46         return math.hypot(x, math.hypot(y, z))
  47 end
  48
  49 function vector.direction(pos1, pos2)
  50         local x_raw = pos2.x - pos1.x
  51         local y_raw = pos2.y - pos1.y
  52         local z_raw = pos2.z - pos1.z
  53         local x_abs = math.abs(x_raw)
  54         local y_abs = math.abs(y_raw)
  55         local z_abs = math.abs(z_raw)
  56         if x_abs >= y_abs and
  57            x_abs >= z_abs then
  58                 y_raw = y_raw * (1 / x_abs)
  59                 z_raw = z_raw * (1 / x_abs)
  60                 x_raw = x_raw / x_abs
  61         end
  62         if y_abs >= x_abs and
  63            y_abs >= z_abs then
  64                 x_raw = x_raw * (1 / y_abs)
  65                 z_raw = z_raw * (1 / y_abs)
  66                 y_raw = y_raw / y_abs
  67         end
  68         if z_abs >= y_abs and
  69            z_abs >= x_abs then
  70                 x_raw = x_raw * (1 / z_abs)
  71                 y_raw = y_raw * (1 / z_abs)
  72                 z_raw = z_raw / z_abs
  73         end
  74         return {x=x_raw, y=y_raw, z=z_raw}
  75 end
  76
  77
  78 function vector.add(a, b)
  79         if type(b) == "table" then
  80                 return {x = a.x + b.x,
  81                         y = a.y + b.y,
  82                         z = a.z + b.z}
  83         else
  84                 return {x = a.x + b,
  85                         y = a.y + b,
  86                         z = a.z + b}
  87         end
  88 end
  89
  90 function vector.subtract(a, b)
  91         if type(b) == "table" then
  92                 return {x = a.x - b.x,
  93                         y = a.y - b.y,
  94                         z = a.z - b.z}
  95         else
  96                 return {x = a.x - b,
  97                         y = a.y - b,
  98                         z = a.z - b}
  99         end
 100 end
 101
 102 function vector.multiply(a, b)
 103         if type(b) == "table" then
 104                 return {x = a.x * b.x,
 105                         y = a.y * b.y,
 106                         z = a.z * b.z}
 107         else
 108                 return {x = a.x * b,
 109                         y = a.y * b,
 110                         z = a.z * b}
 111         end
 112 end
 113
 114 function vector.divide(a, b)
 115         if type(b) == "table" then
 116                 return {x = a.x / b.x,
 117                         y = a.y / b.y,
 118                         z = a.z / b.z}
 119         else
 120                 return {x = a.x / b,
 121                         y = a.y / b,
 122                         z = a.z / b}
 123         end
 124 end
 125