Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F64186329
Allocate.hpp
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sat, May 25, 04:17
Size
4 KB
Mime Type
text/x-c++
Expires
Mon, May 27, 04:17 (2 d)
Engine
blob
Format
Raw Data
Handle
17865115
Attached To
rGOOSEFEM GooseFEM
Allocate.hpp
View Options
/**
Implementation of Allocate.h
\file Allocate.hpp
\copyright Copyright 2017. Tom de Geus. All rights reserved.
\license This project is released under the GNU Public License (GPLv3).
*/
#ifndef GOOSEFEM_ALLOCATE_HPP
#define GOOSEFEM_ALLOCATE_HPP
#include "Allocate.h"
namespace GooseFEM {
namespace detail {
template <class T, class S>
inline bool has_shape_begin(const T& t, const S& s)
{
return s.dimension() >= t.dimension() &&
std::equal(t.shape().cbegin(), t.shape().cend(), s.shape().begin());
}
} // namespace detail
template <size_t dim, size_t rank, class T>
inline void asTensor(const xt::xtensor<T, dim>& arg, xt::xtensor<T, dim + rank>& ret)
{
using strides_type = typename xt::xtensor<T, dim>::strides_type::value_type;
GOOSEFEM_ASSERT(detail::has_shape_begin(arg, ret));
std::array<strides_type, dim + rank> ret_strides;
std::copy(arg.strides().begin(), arg.strides().end(), ret_strides.begin());
std::fill(ret_strides.begin() + dim, ret_strides.end(), 0);
ret = xt::strided_view(arg, ret.shape(), std::move(ret_strides), 0ul, xt::layout_type::dynamic);
}
template <size_t dim, size_t rank, class T>
inline xt::xtensor<T, dim + rank> AsTensor(
const xt::xtensor<T, dim>& arg,
const std::array<size_t, rank>& shape)
{
std::array<size_t, dim + rank> ret_shape;
std::copy(arg.shape().begin(), arg.shape().end(), ret_shape.begin());
std::copy(shape.begin(), shape.end(), ret_shape.begin() + dim);
xt::xtensor<T, dim + rank> ret = xt::empty<T>(ret_shape);
GooseFEM::asTensor<dim, rank>(arg, ret);
return ret;
}
template <size_t dim, size_t rank, class T>
inline xt::xtensor<T, dim + rank> AsTensor(const xt::xtensor<T, dim>& arg, size_t n)
{
std::array<size_t, dim + rank> ret_shape;
std::copy(arg.shape().begin(), arg.shape().end(), ret_shape.begin());
std::fill(ret_shape.begin() + dim, ret_shape.end(), n);
xt::xtensor<T, dim + rank> ret = xt::empty<T>(ret_shape);
GooseFEM::asTensor<dim, rank>(arg, ret);
return ret;
}
template <class T>
inline xt::xarray<typename T::value_type> AsTensor(size_t rank, const T& arg, const std::vector<size_t>& shape)
{
using value_type = typename T::value_type;
GOOSEFEM_ASSERT(rank == shape.size());
size_t dim = arg.dimension();
std::vector<size_t> ret_shape(dim + rank);
xt::dynamic_shape<ptrdiff_t> ret_strides(dim + rank);
std::copy(arg.shape().begin(), arg.shape().end(), ret_shape.begin());
std::copy(arg.strides().begin(), arg.strides().end(), ret_strides.begin());
std::copy(shape.begin(), shape.end(), ret_shape.begin() + dim);
std::fill(ret_strides.begin() + dim, ret_strides.end(), 0);
xt::xarray<value_type> ret = xt::empty<value_type>(ret_shape);
ret = xt::strided_view(arg, ret.shape(), std::move(ret_strides), 0ul, xt::layout_type::dynamic);
return ret;
}
template <class T>
inline xt::xarray<typename T::value_type> AsTensor(size_t rank, const T& arg, size_t n)
{
using value_type = typename T::value_type;
size_t dim = arg.dimension();
using strides_type = typename T::strides_type::value_type;
std::vector<size_t> ret_shape(dim + rank);
xt::svector<strides_type> ret_strides(dim + rank);
std::copy(arg.shape().begin(), arg.shape().end(), ret_shape.begin());
std::copy(arg.strides().begin(), arg.strides().end(), ret_strides.begin());
std::fill(ret_shape.begin() + dim, ret_shape.end(), n);
std::fill(ret_strides.begin() + dim, ret_strides.end(), 0);
xt::xarray<value_type> ret = xt::empty<value_type>(ret_shape);
ret = xt::strided_view(arg, ret.shape(), std::move(ret_strides), 0ul, xt::layout_type::dynamic);
return ret;
}
template <class T>
inline xt::xtensor<T, 2> as3d(const xt::xtensor<T, 2>& data)
{
GOOSEFEM_ASSERT(data.shape(1) > 0 && data.shape(1) < 4)
if (data.shape(1) == 3ul) {
return data;
}
xt::xtensor<T, 2> ret = xt::zeros<T>(std::array<size_t, 2>{data.shape(0), 3ul});
if (data.shape(1) == 2ul) {
xt::view(ret, xt::all(), xt::keep(0, 1)) = data;
}
if (data.shape(1) == 1ul) {
xt::view(ret, xt::all(), xt::keep(0)) = data;
}
return ret;
}
} // namespace GooseFEM
#endif
Event Timeline
Log In to Comment