tests/testthat/test-matrix4.R

In ravetools: Signal and Image Processing Toolbox for Analyzing Intracranial Electroencephalography Data

test_that("Matrix4", {

  library(testthat)
  m <- ravetools:::new_matrix4()
  expect_true(m$is_matrix4)

  expect_equal(m$to_array(), diag(1, 4))
  expect_equal(m$elements, diag(1, 4))

  m$set(matrix(1:16, 4, 4))
  expect_equal(m$to_array(), matrix(1:16, 4, 4))
  expect_equal(m[], matrix(1:16, 4, 4))

  m$set(1:16)
  m0 <- matrix(1:16, 4, 4, byrow = TRUE)
  expect_equal(m[], m0)

  m2 <- m$clone2()
  expect_equal(m2$identity()[], diag(1, 4))
  expect_equal(m[], m0)

  m2$copy(m)
  expect_equal(m2[], m0)

  expect_equal(m2$identity()$copy_position(m)[], matrix(c(1,0,0,4,
                                                          0,1,0,8,
                                                          0,0,1,12,
                                                          0,0,0,1), 4, byrow = TRUE))
  x <- ravetools:::new_vector3()
  y <- ravetools:::new_vector3()
  z <- ravetools:::new_vector3()
  m2$copy(m)$extract_basis(x,y,z)
  expect_equal(x[], m0[1:3,1,drop = FALSE])
  expect_equal(y[], m0[1:3,2,drop = FALSE])
  expect_equal(z[], m0[1:3,3,drop = FALSE])


  m00 <- diag(1, 4)
  m00[1:3,1] <- x[]
  m00[1:3,2] <- y[]
  m00[1:3,3] <- 0
  expect_equal(m2$identity()$make_basis(x, y)[], m00)
  m00[1:3,3] <- z[]
  expect_equal(m2$identity()$make_basis(x, y, z)[], m00)

  m00[1:3,1] <- y[]
  m00[1:3,2] <- z[]
  m00[1:3,3] <- x[]
  expect_equal(m2$identity()$make_basis(y[], z[], x[])[], m00)

  m00 <- m2$extract_rotation(m)[]
  expect_equal(m0[1:3,1] / m00[1:3,1], rep(norm(m0[1:3,1], "2"), 3))
  expect_equal(m0[1:3,2] / m00[1:3,2], rep(norm(m0[1:3,2], "2"), 3))
  expect_equal(m0[1:3,3] / m00[1:3,3], rep(norm(m0[1:3,3], "2"), 3))

  ## JS code:
  ## new THREE.Matrix4().lookAt(new THREE.Vector3(1,2,3), new THREE.Vector3(4,5,6), new THREE.Vector3(9,8,6)).elements.join(",")
  ex <- matrix(
    c(-0.5345224838248486,0.8017837257372731,-0.2672612419124245,0,
      0.6172133998483676,0.15430334996209175,-0.7715167498104594,0,
      -0.5773502691896257,-0.5773502691896257,-0.5773502691896257,0,
      0,0,0,1), 4
  )
  expect_equal(m2$identity()$look_at(new_vector3(1,2,3), new_vector3(4,5,6), new_vector3(9,8,6))[], ex)


  expect_equal(m2$set(t(m0))$multiply(m)[], t(m0) %*% m0)
  expect_equal(m2$set(t(m0))$premultiply(m)[], m0 %*% t(m0))
  expect_equal(m2$set(m0)$multiply_matrices(m, m2)[], m0 %*% m0)
  expect_equal(m2$set(m0)$multiply_scalar(1.4)[], m0 * 1.4)

  m00 <- matrix(c(
    1, 0, 0, 128,
    -1,1, 0, 128,
    0,-1, 1, 128,
    0, 0, 4, 1
  ), 4, 4, byrow = TRUE)
  expect_equal(m2$set(m00)$determinant(), det(m00))

  m00 <- t(m0)
  m00[1:3,4] <- 21:23
  expect_equal(m2$set(m0)$transpose()$set_position(21:23)[], m00)

  m00 <- matrix(c(
    1, 0, 0, 128,
    -1,1, 0, 128,
    0,-1, 1, 128,
    0, 0, 4, 1
  ), 4, 4, byrow = TRUE)
  expect_equal(m2$set(m00)$invert()[], solve(m00))
  # microbenchmark::microbenchmark(
  #   m2$set(m00)$invert(),
  #   solve(m00)
  # )
  expect_equal(m2$set(m00)$scale(2,3,4)[], m00 %*% diag(c(2,3,4,1)))

  expect_equal(m2$set(m00)$get_max_scale_on_axis(), sqrt(2))

  m00 <- diag(1, 4)
  m00[1:3,4] <- c(1,2,3)
  expect_equal(m2$make_translation(1,2,3)[], m00)

  theta <- pi / 6
  expect_equal(m2$make_rotation_x(theta)[], matrix(c(
    1, 0, 0, 0,
    0, cos(theta), -sin(theta), 0,
    0, sin(theta), cos(theta), 0,
    0, 0, 0, 1
  ), 4, 4, byrow = TRUE))

  expect_equal(m2$make_rotation_y(theta)[], matrix(c(
    cos(theta), 0, sin(theta), 0,
    0, 1, 0, 0,
    -sin(theta), 0, cos(theta), 0,
    0, 0, 0, 1
  ), 4, 4, byrow = TRUE))

  expect_equal(m2$make_rotation_z(theta)[], matrix(c(
    cos(theta), -sin(theta), 0, 0,
    sin(theta), cos(theta), 0, 0,
    0, 0, 1, 0,
    0, 0, 0, 1
  ), 4, 4, byrow = TRUE))

  ## JS Code:
  # new THREE.Matrix4().makeRotationAxis(new THREE.Vector3(1,2,3), Math.PI / 6).elements.join(",")
  expect_equal(m2$make_rotation_axis(c(1,2,3), theta)[], matrix(c(
    1,1.7679491924311224,-0.598076211353316,0,
    -1.2320508075688772,1.401923788646684,1.3038475772933678,0,
    1.4019237886466838,0.3038475772933678,2.0717967697244903,0,
    0,0,0,1
  ), 4, 4, byrow = FALSE))

  expect_equal(m2$make_scale(1,2,3)[], diag(c(1,2,3,1)))
  expect_equal(m2$make_scale(x)[], diag(c(x[],1)))

  # xy, xz, yx, yz, zx, zy
  expect_equal(m2$make_shear(1,2,3,4,5,6)[], matrix(c(
    1, 3, 5, 0,
    1, 1, 6, 0,
    2, 4, 1, 0,
    0, 0, 0, 1
  ), 4, 4, byrow = TRUE))

  ## JS Code:
  # new THREE.Matrix4().makePerspective(1,2,3,4,5,6).elements.join(",")
  expect_equal(m2$make_perpective(1,2,3,4,5,6)[], matrix(c(
    10,0,0,0,0,-10,0,0,3,-7,-11,-1,0,0,-60,0
  ), 4, 4, byrow = FALSE))

  ## JS Code:
  # new THREE.Matrix4().makeOrthographic(1,2,3,4,5,6).elements.join(",")
  expect_equal(m2$make_orthographic(1,2,3,4,5,6)[], matrix(c(
    2,0,0,0,0,-2,0,0,0,0,-2,0,-3,7,-11,1
  ), 4, 4, byrow = FALSE))

})