// fields_new.cpp: RcppArmadillo unit test code for field types
//
// Copyright (C) 2021         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/>.

#define RCPP_ARMADILLO_FIX_Field 1

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>

// [[Rcpp::export]]
arma::field<arma::mat> field1m22n() {
    arma::mat A = arma::randn(2,2);

    arma::field<arma::mat> F(1);
    F(0) = A;

    return F;
}


// [[Rcpp::export]]
arma::field<arma::mat> field11m22n() {
    arma::mat A = arma::randn(2,2);

    arma::field<arma::mat> F(1,1);
    F(0,0) = A;

    return F;
}

// [[Rcpp::export]]
arma::field<arma::mat> field12m22n() {
    arma::mat A = arma::randn(2,2);
    arma::mat B = arma::randn(2,2);

    arma::field<arma::mat> F(1,2);
    F(0,0) = A;
    F(0,1) = B;
    //F.print("12m22xx");
    return F;
}

// [[Rcpp::export]]
arma::field<arma::mat> field21m22n() {
    arma::mat A = arma::randn(2,2);
    arma::mat B = arma::randn(2,2);

    arma::field<arma::mat> F(2,1);
    F(0,0) = A;
    F(1,0) = B;

    return F;
}

// [[Rcpp::export]]
arma::field<arma::mat> field22m2233n() {
    arma::mat A = arma::randn(2,2);
    arma::mat B = arma::randn(3,3);

    arma::field<arma::mat> F(2,2);
    F(0,1) = A;
    F(1,0) = B;

    return F;
}

// [[Rcpp::export]]
arma::field<arma::mat> field222m223344n() {
    arma::mat A = arma::randn(2,2);
    arma::mat B = arma::randn(3,3);
    arma::mat C = arma::randn(4,4);

    arma::field<arma::mat> F(2,2,2);
    F(0,1,0) = A;
    F(1,0,0) = B;
    F(0,0,1) = C;

    return F;
}





// [[Rcpp::export]]
arma::ivec infield1m22n(arma::field<arma::mat> F) {
    arma::uvec v = { F.n_rows, F.n_cols, F.n_slices };
    return arma::conv_to<arma::ivec>::from(v);
}

// [[Rcpp::export]]
arma::ivec infield11m22n(arma::field<arma::mat> F) {
    arma::uvec v = { F.n_rows, F.n_cols, F.n_slices };
    return arma::conv_to<arma::ivec>::from(v);
}

// [[Rcpp::export]]
arma::ivec infield12m22n(arma::field<arma::mat> F) {
    arma::uvec v = { F.n_rows, F.n_cols, F.n_slices };
    return arma::conv_to<arma::ivec>::from(v);
}

// [[Rcpp::export]]
arma::ivec infield21m22n(arma::field<arma::mat> F) {
    arma::uvec v = { F.n_rows, F.n_cols, F.n_slices };
    return arma::conv_to<arma::ivec>::from(v);
}

// [[Rcpp::export]]
arma::ivec infield22m2233n(arma::field<arma::mat> F) {
    arma::uvec v = { F.n_rows, F.n_cols, F.n_slices };
    return arma::conv_to<arma::ivec>::from(v);
}

// [[Rcpp::export]]
arma::ivec infield222m223344n(arma::field<arma::mat> F) {
    arma::uvec v = { F.n_rows, F.n_cols, F.n_slices };
    return arma::conv_to<arma::ivec>::from(v);
}