moonpool/benchs/raytracer/vec3.ml

type vec3 = { x: float;
              y: float;
              z: float }

let of_floats (e1, e2, e3) =
  {x = e1; y = e2; z = e3}

let add v w =
  {x = v.x +. w.x; 
   y = v.y +. w.y; 
   z = v.z +. w.z}

let sub v w =
  {x = v.x -. w.x;
   y = v.y -. w.y; 
   z = v.z -. w.z}

let neg v =
  sub {x= 0.0; y=0.0; z=0.0} v

let dot v w =
  v.x*.w.x +. v.y*.w.y +.v.z*.w.z

let cross v w =
  {x = v.y*.w.z -. v.z*.w.y;
   y = v.z*.w.x -. v.x*.w.z;
   z = v.x*.w.y -. v.y*.w.x}

let length v =
  sqrt (dot v v)

let unit_vector v =
  let l = length v in
  { x = v.x /. l;
    y = v.y /. l;
    z = v.z /. l }

let mul t v =
  { x = t *. v.x;
    y = t *. v.y;
    z = t *. v.z }

(* pairwise multiplication *)
let pmul v w =
  { x = v.x *. w.x;
    y = v.y *. w.y;
    z = v.z *. w.z; }