// -*- mode: C++ -*-

//

// RcppArrayFireWrap.cpp: Rcpp/ArrayFire glue, support for wrap

//

// Copyright (C) 2015 Kazuki Fukui

// Copyright (C) 2017 Ralf Stubner (R Institute GmbH)

//

// This file is part of RcppArrayFire.

//

// RcppArrayFire 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.

//

// RcppArrayFire 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 RcppArrayFire. If not, see <http://www.gnu.org/licenses/>.

#include <RcppArrayFireWrap.h>

namespace RcppArrayFire{

SEXP wrap_af_impl( const af::array& object ){

::Rcpp::RObject x;

switch(object.type()){

case f64:

x = wrap_array<double>( object ) ;

break;

case c64:

x = wrap_array<std::complex<double>>( object ) ;

break;

case f32:

x = wrap_array<float>( object ) ;

break;

case c32:

x = wrap_array<std::complex<float>>( object ) ;

break;

case s32:

x = wrap_array<int>( object ) ;

break;

case u32:

x = wrap_array<unsigned int>( object ) ;

break;

default:

Rcpp::stop("Unsopprted data type");

break;

}

return x;

}

} /* namespace RcppArrayFire */

namespace Rcpp{

/* wrap */

template <> SEXP wrap (const af::dim4& dims) {

std::vector<dim_t> res(dims.ndims());

for(int i = 0; i < dims.ndims(); ++i)

res[i] = dims[i];

return wrap(res);

}

template <> SEXP wrap (const af::array& data ){

::Rcpp::RObject x = ::RcppArrayFire::wrap_af_impl( data );

if (data.issparse() == false && data.numdims() > 1)

x.attr("dim") = wrap(data.dims());

return x;

}

}