DYTSrouce/Tool/matlab/include/coder/coder_array/coder_array_mex.h
2024-11-22 23:19:31 +08:00

1360 lines
40 KiB
C++

/* Copyright 2019-2020 The MathWorks, Inc. */
/* Copied from fullfile(matlabroot,'extern','include','coder','coder_array','coder_array_mex.h') */
#pragma once
#ifndef _mw_coder_array_mex_h
#define _mw_coder_array_mex_h
// -------------------------------------------------------------------------------------------------------------------------------------------
// Usage:
//
// coder::array<T, N>: T base type of data, N number of dimensions
//
// coder::array()
// : default constructor
// coder::array(const coder::array &)
// : copy constructor (always make a deep copy of other array)
// coder::array(const T *data, const SizeType *sz)
// : Set data with sizes of this array.
// : (Data is not copied, data is not deleted)
// coder::array::operator = (coder coder::array &)
// : Assign into this array;
// : delete its previous contents (if owning the data.)
// set(const T *data, SizeType sz1, SizeType sz2, ...)
// : Set data with dimensions.
// : (Data is not copied, data is not deleted)
// set_size(SizeType sz1, SizeType sz2, ...)
// : Set sizes of array. Reallocate memory of data if needed.
// bool is_owner() : Return true if the data is owned by the class.
// void no_free()
// : Mark that the class does no longer owns its data;
// : data is no longer deleted.
// SizeType capacity() : How many entries are reserved by memory allocation.
// reshape( SizeType sz1, SizeType sz2, ...)
// : Reshape array to a different ND shape. Do not copy the data.
// : The number of elements must not be changed (numel()==sz1*sz2*...)
// : Return the array with possibly new number of dimensions.
// clear() : Reset array to be empty.
// SizeType numel() : Return the number of elements.
// operator [] (SizeType index) : Extract element at linear index (0 based.)
// size(SizeType dimension) : Size of array of the provided dimension.
// SizeType * size() : Return the pointer to all the sizes of this array.
// SizeType index(SizeType i1, SizeType i2, ...)
// : Compute the linear index from ND index (i1,i2,...)
// at(SizeType i1, SizeType i2, ...) : The element at index (i1,i2,...)
#include <cassert>
#include <cstring>
#include <iterator>
#include <string>
#include <tuple>
#include <vector>
#include "mex.h"
#include "emlrt.h"
#include "tmwtypes.h"
namespace coder {
#ifndef CODER_ARRAY_SIZE_TYPE_DEFINED
typedef int32_T SizeType;
#endif
namespace detail {
#ifndef CODER_ARRAY_DATA_PTR_DEFINED
template <typename T, typename SZ>
class data_ptr {
public:
typedef T value_type;
typedef SZ size_type;
data_ptr()
: mem_(NULL)
, data_(NULL)
, size_(0)
, capacity_(0)
, owner_(false) {
}
data_ptr(const T* _data, SZ _sz)
: mem_(NULL)
, data_(const_cast<T*>(_data))
, size_(_sz)
, capacity_(_sz)
, owner_(false) {
}
data_ptr(const data_ptr& _other)
: mem_(_other.owner_ ? NULL : _other.mem_)
, data_(_other.owner_ ? NULL : _other.data_)
, size_(_other.owner_ ? 0 : _other.size_)
, capacity_(_other.owner_ ? 0 : _other.capacity_)
, owner_(_other.owner_) {
if (owner_) {
resize(_other.size_);
std::copy(_other.data_, _other.data_ + size_, data_);
}
}
~data_ptr() {
deallocate(mem_, data_);
}
std::pair<char*, T*> allocate_primitive(SZ _n) {
char* m = reinterpret_cast<char*>(emlrtMallocMex(_n * sizeof(T)));
T* d = new (m) T[_n];
return std::make_pair(m, d);
}
void deallocate_primitive(char* _m, T* _d) {
if (_m == NULL || !owner_) {
return;
}
emlrtFreeMex(_m);
mem_ = NULL;
}
std::pair<char*, T*> allocate_with_destructor(SZ _n) {
char* m = reinterpret_cast<char*>(emlrtMallocMex(sizeof(SZ) * 2 + _n * sizeof(T)));
*reinterpret_cast<SZ*>(m) = _n;
T* d = new (m + sizeof(SZ)) T[_n];
return std::make_pair(m, d);
}
void deallocate_with_destructor(char* _m, T* _d) {
if (_m == NULL || !owner_) {
return;
}
SZ n = *reinterpret_cast<SZ*>(_m);
for (SZ i = 0; i < n; i++) {
_d[i].~T();
}
emlrtFreeMex(_m);
mem_ = NULL;
}
std::pair<char*, T*> allocate(SZ _n) {
if (std::is_pod<T>::value) {
return allocate_primitive(_n);
} else {
return allocate_with_destructor(_n);
}
}
void deallocate(char* _m, T* _d) {
if (std::is_pod<T>::value) {
deallocate_primitive(_m, _d);
} else {
deallocate_with_destructor(_m, _d);
}
}
void prealloc(SZ _numel) {
size_ = _numel;
capacity_ = _numel;
}
SZ capacity() const {
return capacity_;
}
void reserve(SZ _n, emlrtRTEInfo* _rteInfo = NULL, emlrtConstCTX _ctx = NULL) {
if (_n > capacity_) {
char* new_mem = NULL;
T* new_data = NULL;
std::tie(new_mem, new_data) = allocate(_n);
if (new_data == NULL) {
emlrtHeapAllocationErrorR2012b(_rteInfo, _ctx);
}
if (size_ != 0) {
std::copy(data_, data_ + size_, new_data);
}
deallocate(mem_, data_);
mem_ = new_mem;
data_ = new_data;
capacity_ = _n;
owner_ = true;
}
}
void resize(SZ _n, emlrtRTEInfo* _rteInfo = NULL, emlrtConstCTX _ctx = NULL) {
reserve(_n, _rteInfo, _ctx);
size_ = _n;
}
private:
// Prohibit use of assignment operator to prevent subtle bugs
void operator=(const data_ptr<T, SZ>& _other);
public:
void set(T* _data, const SZ _sz) {
deallocate(mem_, data_);
mem_ = NULL;
data_ = _data;
size_ = _sz;
owner_ = false;
capacity_ = size_;
}
void copy(const T* _data, SZ _size) {
if (data_ == _data) {
size_ = _size;
return;
}
deallocate(mem_, data_);
std::tie(mem_, data_) = allocate(_size);
owner_ = true;
size_ = _size;
capacity_ = size_;
std::copy(_data, _data + _size, data_);
}
void copy(const data_ptr<T, SZ>& _other) {
copy(_other.data_, _other.size_);
}
operator T*() {
return &data_[0];
}
operator const T*() const {
return &data_[0];
}
T& operator[](SZ _index) {
return data_[_index];
}
const T& operator[](SZ _index) const {
return data_[_index];
}
T* operator->() {
return data_;
}
const T* operator->() const {
return data_;
}
bool is_null() const {
return data_ == NULL;
}
void clear() {
deallocate(mem_, data_);
mem_ = NULL;
data_ = NULL;
size_ = 0;
capacity_ = 0;
owner_ = false;
}
bool is_owner() const {
return owner_;
}
void set_owner(bool _b) {
owner_ = _b;
}
private:
char* mem_;
T* data_;
SZ size_;
SZ capacity_;
bool owner_;
};
#endif
} // namespace detail
// Implementing the random access iterator class so coder::array can be
// used in STL iterators.
template <typename T>
class array_iterator : public std::iterator<std::random_access_iterator_tag,
typename T::value_type,
typename T::size_type> {
public:
array_iterator()
: arr_(NULL)
, i_(0) {
}
array_iterator(const array_iterator<T>& other)
: arr_(other.arr_)
, i_(other.i_) {
}
~array_iterator() {
}
typename T::value_type& operator*() const {
return (*arr_)[i_];
}
typename T::value_type* operator->() const {
return &(*arr_)[i_];
}
typename T::value_type& operator[](typename T::size_type _di) const {
return (*arr_)[i_ + _di];
}
array_iterator<T>& operator++() {
++i_;
return *this;
}
array_iterator<T>& operator--() {
--i_;
return *this;
}
array_iterator<T> operator++(int) {
array_iterator<T> cp(*this);
++i_;
return cp;
}
array_iterator<T> operator--(int) {
array_iterator<T> cp(*this);
--i_;
return cp;
}
array_iterator<T>& operator=(const array_iterator<T>& _other) {
this->i_ = _other.i_;
return *this;
}
bool operator==(const array_iterator<T>& _other) const {
return i_ == _other.i_;
}
bool operator!=(const array_iterator<T>& _other) const {
return i_ != _other.i_;
}
bool operator<(const array_iterator<T>& _other) const {
return i_ < _other.i_;
}
bool operator>(const array_iterator<T>& _other) const {
return i_ > _other.i_;
}
bool operator<=(const array_iterator<T>& _other) const {
return i_ <= _other.i_;
}
bool operator>=(const array_iterator<T>& _other) const {
return i_ >= _other.i_;
}
array_iterator<T> operator+(typename T::size_type _add) const {
array_iterator<T> cp(*this);
cp.i_ += _add;
return cp;
}
array_iterator<T>& operator+=(typename T::size_type _add) {
this->i_ += _add;
return *this;
}
array_iterator<T> operator-(typename T::size_type _subtract) const {
array_iterator<T> cp(*this);
cp.i_ -= _subtract;
return cp;
}
array_iterator<T>& operator-=(typename T::size_type _subtract) {
this->i_ -= _subtract;
return *this;
}
typename T::size_type operator-(const array_iterator<T>& _other) const {
return static_cast<typename T::size_type>(this->i_ - _other.i_);
}
array_iterator(T* _arr, typename T::size_type _i)
: arr_(_arr)
, i_(_i) {
}
private:
T* arr_;
typename T::size_type i_;
};
// Const version of the array iterator.
template <typename T>
class const_array_iterator : public std::iterator<std::random_access_iterator_tag,
typename T::value_type,
typename T::size_type> {
public:
const_array_iterator()
: arr_(NULL)
, i_(0) {
}
const_array_iterator(const const_array_iterator<T>& other)
: arr_(other.arr_)
, i_(other.i_) {
}
~const_array_iterator() {
}
const typename T::value_type& operator*() const {
return (*arr_)[i_];
}
const typename T::value_type* operator->() const {
return &(*arr_)[i_];
}
const typename T::value_type& operator[](typename T::size_type _di) const {
return (*arr_)[i_ + _di];
}
const_array_iterator<T>& operator++() {
++i_;
return *this;
}
const_array_iterator<T>& operator--() {
--i_;
return *this;
}
const_array_iterator<T> operator++(int) {
const_array_iterator<T> copy(*this);
++i_;
return copy;
}
const_array_iterator<T> operator--(int) {
const_array_iterator copy(*this);
--i_;
return copy;
}
const_array_iterator<T>& operator=(const const_array_iterator<T>& _other) {
this->i_ = _other.i_;
return *this;
}
bool operator==(const const_array_iterator<T>& _other) const {
return i_ == _other.i_;
}
bool operator!=(const const_array_iterator<T>& _other) const {
return i_ != _other.i_;
}
bool operator<(const const_array_iterator<T>& _other) const {
return i_ < _other.i_;
}
bool operator>(const const_array_iterator<T>& _other) const {
return i_ > _other.i_;
}
bool operator<=(const const_array_iterator<T>& _other) const {
return i_ <= _other.i_;
}
bool operator>=(const const_array_iterator<T>& _other) const {
return i_ >= _other.i_;
}
const_array_iterator<T> operator+(typename T::size_type _add) const {
const_array_iterator<T> cp(*this);
cp.i_ += _add;
return cp;
}
const_array_iterator<T>& operator+=(typename T::size_type _add) {
this->i_ += _add;
return *this;
}
const_array_iterator<T> operator-(typename T::size_type _subtract) const {
const_array_iterator<T> cp(*this);
cp.i_ -= _subtract;
return cp;
}
const_array_iterator<T>& operator-=(typename T::size_type _subtract) {
this->i_ -= _subtract;
return *this;
}
typename T::size_type operator-(const const_array_iterator<T>& _other) const {
return static_cast<typename T::size_type>(this->i_ - _other.i_);
}
const_array_iterator(const T* _arr, typename T::size_type _i)
: arr_(_arr)
, i_(_i) {
}
private:
const T* arr_;
typename T::size_type i_, n_;
};
namespace detail {
// detail::numel<N>: Compile-time product of the given size vector of length N.
template <int N>
class numel {
public:
template <typename SZ>
static SZ compute(SZ _size[]) {
return _size[N - 1] * numel<N - 1>::compute(_size);
}
};
template <>
class numel<0> {
public:
template <typename SZ>
static SZ compute(SZ[]) {
return 1;
}
};
// detail::numel_check<N>
// : Compile-time product of the given size vector of length N with runtime check.
template <int N>
class numel_check {
public:
template <typename SZ>
static SZ compute(emlrtRTEInfo* _rteInfo, emlrtConstCTX _ctx, SZ _size[]) {
return static_cast<SZ>(emlrtSizeMulR2012b(
static_cast<uint32_T>(_size[N - 1]),
static_cast<uint32_T>(numel_check<N - 1>::compute(_rteInfo, _ctx, _size)), _rteInfo,
_ctx));
}
};
template <>
class numel_check<0> {
public:
template <typename SZ>
static SZ compute(emlrtRTEInfo*, emlrtConstCTX, SZ[]) {
return 1;
}
};
// Compute flat index from (column-major) ND size vector and a list of indices.
template <int I>
class index_nd {
public:
template <typename SZ>
static SZ compute(const SZ _size[], const SZ _indices[]) {
const SZ weight = numel<I - 1>::compute(_size);
return weight * _indices[I - 1] + index_nd<I - 1>::compute(_size, _indices);
}
};
template <>
class index_nd<0> {
public:
template <typename SZ>
static SZ compute(SZ[], SZ[]) {
return 0;
}
};
template <bool Cond>
struct match_dimensions {};
template <>
struct match_dimensions<true> {
static void check() {
}
};
template <>
struct match_dimensions<false> {
static void check() {
assert(false && "Method with wrong number of dimensions used.");
}
};
} // namespace detail
// Base class for code::array. SZ is the type used for sizes (currently int32_t.)
// Overloading up to 10 dimensions (not using variadic templates to
// stay compatible with C++98.)
template <typename T, typename SZ, int N>
class array_base {
public:
typedef T value_type;
typedef SZ size_type;
// This is for MEX only. To prevent memory allocation when set_size is called later.
void prealloc(SZ _numel) {
data_.prealloc(_numel);
}
array_base() {
std::memset(size_, 0, sizeof(SZ) * N);
}
array_base(T* _data, const SZ* _sz)
: data_(_data, ::coder::detail::numel<N>::compute(_sz)) {
std::copy(_sz, _sz + N, size_);
}
array_base& operator=(const array_base& _other) {
data_.copy(_other.data_);
std::copy(_other.size_, _other.size_ + N, size_);
return *this;
}
void set(T* _data, SZ _n1) {
::coder::detail::match_dimensions<N == 1>::check();
data_.set(_data, _n1);
size_[0] = _n1;
}
void set(T* _data, SZ _n1, SZ _n2) {
::coder::detail::match_dimensions<N == 2>::check();
data_.set(_data, _n1 * _n2);
size_[0] = _n1;
size_[1] = _n2;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3) {
::coder::detail::match_dimensions<N == 3>::check();
data_.set(_data, _n1 * _n2 * _n3);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4) {
::coder::detail::match_dimensions<N == 4>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5) {
::coder::detail::match_dimensions<N == 5>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4 * _n5);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6) {
::coder::detail::match_dimensions<N == 6>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4 * _n5 * _n6);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7) {
::coder::detail::match_dimensions<N == 7>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4 * _n5 * _n6 * _n7);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8) {
::coder::detail::match_dimensions<N == 8>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4 * _n5 * _n6 * _n7 * _n8);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
}
void set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9) {
::coder::detail::match_dimensions<N == 9>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4 * _n5 * _n6 * _n7 * _n8 * _n9);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
size_[8] = _n9;
}
void
set(T* _data, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9, SZ _n10) {
::coder::detail::match_dimensions<N == 10>::check();
data_.set(_data, _n1 * _n2 * _n3 * _n4 * _n5 * _n6 * _n7 * _n8 * _n9 * _n10);
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
size_[8] = _n9;
size_[9] = _n10;
}
bool is_owner() const {
return data_.is_owner();
}
void no_free() {
data_.set_owner(false);
}
SZ capacity() const {
return data_.capacity();
}
void set_size(emlrtRTEInfo* _rteInfo, emlrtConstCTX _ctx, SZ _n1) {
::coder::detail::match_dimensions<N == 1>::check();
size_[0] = _n1;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo, emlrtConstCTX _ctx, SZ _n1, SZ _n2) {
::coder::detail::match_dimensions<N == 2>::check();
size_[0] = _n1;
size_[1] = _n2;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo, emlrtConstCTX _ctx, SZ _n1, SZ _n2, SZ _n3) {
::coder::detail::match_dimensions<N == 3>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo, emlrtConstCTX _ctx, SZ _n1, SZ _n2, SZ _n3, SZ _n4) {
::coder::detail::match_dimensions<N == 4>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void
set_size(emlrtRTEInfo* _rteInfo, emlrtConstCTX _ctx, SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5) {
::coder::detail::match_dimensions<N == 5>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo,
emlrtConstCTX _ctx,
SZ _n1,
SZ _n2,
SZ _n3,
SZ _n4,
SZ _n5,
SZ _n6) {
::coder::detail::match_dimensions<N == 6>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo,
emlrtConstCTX _ctx,
SZ _n1,
SZ _n2,
SZ _n3,
SZ _n4,
SZ _n5,
SZ _n6,
SZ _n7) {
::coder::detail::match_dimensions<N == 7>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo,
emlrtConstCTX _ctx,
SZ _n1,
SZ _n2,
SZ _n3,
SZ _n4,
SZ _n5,
SZ _n6,
SZ _n7,
SZ _n8) {
::coder::detail::match_dimensions<N == 8>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo,
emlrtConstCTX _ctx,
SZ _n1,
SZ _n2,
SZ _n3,
SZ _n4,
SZ _n5,
SZ _n6,
SZ _n7,
SZ _n8,
SZ _n9) {
::coder::detail::match_dimensions<N == 9>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
size_[8] = _n9;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(emlrtRTEInfo* _rteInfo,
emlrtConstCTX _ctx,
SZ _n1,
SZ _n2,
SZ _n3,
SZ _n4,
SZ _n5,
SZ _n6,
SZ _n7,
SZ _n8,
SZ _n9,
SZ _n10) {
::coder::detail::match_dimensions<N == 10>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
size_[8] = _n9;
size_[9] = _n10;
ensureCapacity(numel_check(_rteInfo, _ctx), _rteInfo, _ctx);
}
void set_size(SZ _n1) {
::coder::detail::match_dimensions<N == 1>::check();
size_[0] = _n1;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2) {
::coder::detail::match_dimensions<N == 2>::check();
size_[0] = _n1;
size_[1] = _n2;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3) {
::coder::detail::match_dimensions<N == 3>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4) {
::coder::detail::match_dimensions<N == 4>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5) {
::coder::detail::match_dimensions<N == 5>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6) {
::coder::detail::match_dimensions<N == 6>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7) {
::coder::detail::match_dimensions<N == 7>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8) {
::coder::detail::match_dimensions<N == 8>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9) {
::coder::detail::match_dimensions<N == 9>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
size_[8] = _n9;
ensureCapacity(numel());
}
void set_size(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9, SZ _n10) {
::coder::detail::match_dimensions<N == 10>::check();
size_[0] = _n1;
size_[1] = _n2;
size_[2] = _n3;
size_[3] = _n4;
size_[4] = _n5;
size_[5] = _n6;
size_[6] = _n7;
size_[7] = _n8;
size_[8] = _n9;
size_[9] = _n10;
ensureCapacity(numel());
}
template <SizeType N1>
array_base<T, SZ, N1> reshape_n(const SZ (&_ns)[N1]) const {
array_base<T, SZ, N1> reshaped(const_cast<T*>(&data_[0]), _ns);
return reshaped;
}
array_base<T, SZ, 1> reshape(SZ _n1) const {
const SZ ns[] = {_n1};
return reshape_n(ns);
}
array_base<T, SZ, 2> reshape(SZ _n1, SZ _n2) const {
const SZ ns[] = {_n1, _n2};
return reshape_n(ns);
}
array_base<T, SZ, 3> reshape(SZ _n1, SZ _n2, SZ _n3) const {
const SZ ns[] = {_n1, _n2, _n3};
return reshape_n(ns);
}
array_base<T, SZ, 4> reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4) const {
const SZ ns[] = {_n1, _n2, _n3, _n4};
return reshape_n(ns);
}
array_base<T, SZ, 5> reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5) const {
const SZ ns[] = {_n1, _n2, _n3, _n4, _n5};
return reshape_n(ns);
}
array_base<T, SZ, 6> reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6) const {
const SZ ns[] = {_n1, _n2, _n3, _n4, _n5, _n6};
return reshape_n(ns);
}
array_base<T, SZ, 7> reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7) const {
const SZ ns[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7};
return reshape_n(ns);
}
array_base<T, SZ, 8> reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8)
const {
const SZ ns[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7, _n8};
return reshape_n(ns);
}
array_base<T, SZ, 9>
reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9) const {
const SZ ns[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7, _n8, _n9};
return reshape_n(ns);
}
array_base<T, SZ, 10>
reshape(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9, SZ _n10) const {
const SZ ns[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7, _n8, _n9, _n10};
return reshape_n(ns);
}
T& operator[](SZ _index) {
return data_[_index];
}
const T& operator[](SZ _index) const {
return data_[_index];
}
void clear() {
data_.clear();
}
T* data() {
return data_;
}
const T* data() const {
return data_;
}
SZ* size() {
return &size_[0];
}
const SZ* size() const {
return &size_[0];
}
SZ size(SZ _index) const {
return size_[_index];
}
SZ numel() const {
return ::coder::detail::numel<N>::compute(size_);
}
SZ numel_check(emlrtRTEInfo* _rteInfo, const void* _ctx) const {
return ::coder::detail::numel_check<N>::compute(_rteInfo, _ctx, size_);
}
SZ index(SZ _n1) const {
::coder::detail::match_dimensions<N == 1>::check();
const SZ indices[] = {_n1};
return ::coder::detail::index_nd<1>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2) const {
::coder::detail::match_dimensions<N == 2>::check();
const SZ indices[] = {_n1, _n2};
return ::coder::detail::index_nd<2>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3) const {
::coder::detail::match_dimensions<N == 3>::check();
const SZ indices[] = {_n1, _n2, _n3};
return ::coder::detail::index_nd<3>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4) const {
::coder::detail::match_dimensions<N == 4>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4};
return ::coder::detail::index_nd<4>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5) const {
::coder::detail::match_dimensions<N == 5>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4, _n5};
return ::coder::detail::index_nd<5>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6) const {
::coder::detail::match_dimensions<N == 6>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4, _n5, _n6};
return ::coder::detail::index_nd<6>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7) const {
::coder::detail::match_dimensions<N == 7>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7};
return ::coder::detail::index_nd<7>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8) const {
::coder::detail::match_dimensions<N == 8>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7, _n8};
return ::coder::detail::index_nd<8>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9) const {
::coder::detail::match_dimensions<N == 9>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7, _n8, _n9};
return ::coder::detail::index_nd<9>::compute(size_, indices);
}
SZ index(SZ _n1, SZ _n2, SZ _n3, SZ _n4, SZ _n5, SZ _n6, SZ _n7, SZ _n8, SZ _n9, SZ _n10)
const {
::coder::detail::match_dimensions<N == 10>::check();
const SZ indices[] = {_n1, _n2, _n3, _n4, _n5, _n6, _n7, _n8, _n9, _n10};
return ::coder::detail::index_nd<10>::compute(size_, indices);
}
T& at(SZ _i1) {
::coder::detail::match_dimensions<N == 1>::check();
return data_[_i1];
}
T& at(SZ _i1, SZ _i2) {
::coder::detail::match_dimensions<N == 2>::check();
return data_[index(_i1, _i2)];
}
T& at(SZ _i1, SZ _i2, SZ _i3) {
::coder::detail::match_dimensions<N == 3>::check();
return data_[index(_i1, _i2, _i3)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4) {
::coder::detail::match_dimensions<N == 4>::check();
return data_[index(_i1, _i2, _i3, _i4)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5) {
::coder::detail::match_dimensions<N == 5>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6) {
::coder::detail::match_dimensions<N == 6>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7) {
::coder::detail::match_dimensions<N == 7>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7, SZ _i8) {
::coder::detail::match_dimensions<N == 8>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7, SZ _i8, SZ _i9) {
::coder::detail::match_dimensions<N == 9>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8, _i9)];
}
T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7, SZ _i8, SZ _i9, SZ _i10) {
::coder::detail::match_dimensions<N == 10>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8, _i9, _i10)];
}
const T& at(SZ _i1) const {
::coder::detail::match_dimensions<N == 1>::check();
return data_[_i1];
}
const T& at(SZ _i1, SZ _i2) const {
::coder::detail::match_dimensions<N == 2>::check();
return data_[index(_i1, _i2)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3) const {
::coder::detail::match_dimensions<N == 3>::check();
return data_[index(_i1, _i2, _i3)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4) const {
::coder::detail::match_dimensions<N == 4>::check();
return data_[index(_i1, _i2, _i3, _i4)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5) const {
::coder::detail::match_dimensions<N == 5>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6) const {
::coder::detail::match_dimensions<N == 6>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7) const {
::coder::detail::match_dimensions<N == 7>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7, SZ _i8) const {
::coder::detail::match_dimensions<N == 8>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7, SZ _i8, SZ _i9) const {
::coder::detail::match_dimensions<N == 9>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8, _i9)];
}
const T& at(SZ _i1, SZ _i2, SZ _i3, SZ _i4, SZ _i5, SZ _i6, SZ _i7, SZ _i8, SZ _i9, SZ _i10)
const {
::coder::detail::match_dimensions<N == 10>::check();
return data_[index(_i1, _i2, _i3, _i4, _i5, _i6, _i7, _i8, _i9, _i10)];
}
array_iterator<array_base<T, SZ, N> > begin() {
return array_iterator<array_base<T, SZ, N> >(this, 0);
}
array_iterator<array_base<T, SZ, N> > end() {
return array_iterator<array_base<T, SZ, N> >(this, this->numel());
}
const_array_iterator<array_base<T, SZ, N> > begin() const {
return const_array_iterator<array_base<T, SZ, N> >(this, 0);
}
const_array_iterator<array_base<T, SZ, N> > end() const {
return const_array_iterator<array_base<T, SZ, N> >(this, this->numel());
}
protected:
::coder::detail::data_ptr<T, SZ> data_;
SZ size_[N];
private:
void ensureCapacity(SZ _newNumel, emlrtRTEInfo* _rteInfo = NULL, emlrtConstCTX _ctx = NULL) {
if (_newNumel > data_.capacity()) {
SZ i = data_.capacity();
if (i < 16) {
i = 16;
}
while (i < _newNumel) {
if (i > 1073741823) {
i = MAX_int32_T;
} else {
i <<= 1;
}
}
data_.reserve(i, _rteInfo, _ctx);
}
data_.resize(_newNumel, _rteInfo, _ctx);
}
};
// The standard coder::array class with base type and number of dimensions.
template <typename T, int N>
class array : public array_base<T, SizeType, N> {
private:
typedef array_base<T, SizeType, N> Base;
public:
array()
: Base() {
}
array(const array<T, N>& _other)
: Base(_other) {
}
array(const Base& _other)
: Base(_other) {
}
array(T* _data, const SizeType* _sz)
: Base(_data, _sz) {
}
};
// Specialize on char_T (row vector) for better support on strings.
template <>
class array<char_T, 2> : public array_base<char_T, SizeType, 2> {
private:
typedef array_base<char_T, SizeType, 2> Base;
public:
array()
: array_base() {
}
array(const array<char_T, 2>& _other)
: Base(_other) {
}
array(const Base& _other)
: Base(_other) {
}
array(const std::string& _str) {
operator=(_str);
}
array(const char_T* _str) {
operator=(_str);
}
array(const std::vector<char_T>& _vec) {
SizeType n = static_cast<SizeType>(_vec.size());
set_size(1, n);
data_.copy(&_vec[0], n);
}
array& operator=(const std::string& _str) {
SizeType n = static_cast<SizeType>(_str.size());
set_size(1, n);
data_.copy(_str.c_str(), n);
return *this;
}
array& operator=(const char_T* _str) {
SizeType n = static_cast<SizeType>(strlen(_str));
set_size(1, n);
data_.copy(_str, n);
return *this;
}
operator std::string() const {
return std::string(static_cast<const char*>(&(*this)[0]), static_cast<int>(size(1)));
}
};
// Specialize on 2 dimensions for better support interactions with
// std::vector and row vectors.
template <typename T>
class array<T, 2> : public array_base<T, SizeType, 2> {
private:
typedef array_base<T, SizeType, 2> Base;
public:
array()
: Base() {
}
array(const array<T, 2>& _other)
: Base(_other) {
}
array(const Base& _other)
: Base(_other) {
}
array(const std::vector<T>& _vec) {
operator=(_vec);
}
array& operator=(const std::vector<T>& _vec) {
SizeType n = static_cast<SizeType>(_vec.size());
Base::set_size(1, n);
Base::data_.copy(&_vec[0], n);
return *this;
}
operator std::vector<T>() const {
const T* p = &Base::data_[0];
return std::vector<T>(p, p + Base::numel());
}
};
// Specialize on 1 dimension for better support with std::vector and
// column vectors.
template <typename T>
class array<T, 1> : public array_base<T, SizeType, 1> {
private:
typedef array_base<T, SizeType, 1> Base;
public:
array()
: Base() {
}
array(const array<T, 1>& _other)
: Base(_other) {
}
array(const Base& _other)
: Base(_other) {
}
array(const std::vector<T>& _vec) {
operator=(_vec);
}
array& operator=(const std::vector<T>& _vec) {
SizeType n = static_cast<SizeType>(_vec.size());
Base::set_size(n);
Base::data_.copy(&_vec[0], n);
return *this;
}
operator std::vector<T>() const {
const T* p = &Base::data_[0];
return std::vector<T>(p, p + Base::numel());
}
};
} // namespace coder
#endif