# Copyright (C) 2021 - 2022  Dirk Eddelbuettel
#
# This file is part of RcppArmadillo.
#
# RcppArmadillo is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
#
# RcppArmadillo is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with RcppArmadillo.  If not, see <http://www.gnu.org/licenses/>.

library(RcppArmadillo)

Rcpp::sourceCpp("cpp/fields.cpp")

.onWindows <- .Platform$OS.type == "windows"

f1m22 <- field1m22()
expect_true(inherits(f1m22, "array"))
expect_true(inherits(f1m22[[1]], "matrix"))
if (!has_old_field_behavior()) expect_equal(dim(f1m22), c(1,1,1))
expect_equal(dim(f1m22[[1]]), c(2,2))

f11m22 <- field11m22()
expect_true(inherits(f11m22, "array"))
expect_true(inherits(f11m22[[1]], "matrix"))
if (!has_old_field_behavior()) expect_equal(dim(f11m22), c(1,1,1))
expect_equal(dim(f11m22[[1]]), c(2,2))

f12m22 <- field12m22()
expect_true(inherits(f12m22, "array"))
expect_true(inherits(f12m22[[1]], "matrix"))
if (!has_old_field_behavior()) expect_equal(dim(f12m22), c(1,2,1))
expect_equal(dim(f12m22[[1]]), c(2,2))
expect_equal(dim(f12m22[[2]]), c(2,2))

f21m22 <- field21m22()
expect_true(inherits(f21m22, "array"))
expect_true(inherits(f21m22[[1]], "matrix"))
if (!has_old_field_behavior()) expect_equal(dim(f21m22), c(2,1,1))
expect_equal(dim(f21m22[[1]]), c(2,2))
expect_equal(dim(f21m22[[2]]), c(2,2))

f22m2233 <- field22m2233()
expect_true(inherits(f22m2233, "array"))
expect_true(inherits(f22m2233[[1]], "matrix"))
if (!has_old_field_behavior()) expect_equal(dim(f22m2233), c(2,2,1))
expect_equal(dim(f22m2233[[2]]), c(3,3))
expect_equal(dim(f22m2233[[3]]), c(2,2))

if (!has_old_field_behavior()) {
    f222m223344 <- field222m223344()
    expect_true(inherits(f222m223344, "array"))
    expect_true(inherits(f222m223344[[1]], "matrix"))
    expect_equal(dim(f222m223344), c(2,2,2))
    expect_equal(dim(f222m223344[[2]]), c(3,3))
    expect_equal(dim(f222m223344[[3]]), c(2,2))
}

v <- infield1m22( field1m22() )
expect_equal(v, matrix(c(1L, 1L, 1L),3,1))

v <- infield11m22( field11m22() )
expect_equal(v, matrix(c(1L, 1L, 1L),3,1))

v <- infield12m22( field12m22() )
expect_equal(v, matrix(c(2L, 1L, 1L),3,1))  # should be 1,2,1 ?

v <- infield21m22( field21m22() )
expect_equal(v, matrix(c(2L, 1L, 1L),3,1))

v <- infield22m2233( field22m2233() )
expect_equal(v, matrix(c(4L, 1L, 1L),3,1))  # should 2,2,1 ?

#v <- infield222m223344( field222m223344() )
#expect_equal(v, matrix(c(4L, 1L, 1L),3,1))  # should 2,2,1 ?
#print(v)