mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-05-25 15:46:56 +08:00
6edae2d30d
using Sfinae and is_base_of to select correct template which converts to array<Index,NumIndices> user: Gabriel Nützi <gnuetzi@gmx.ch> branch 'default' added unsupported/Eigen/CXX11/src/Tensor/TensorMetaMacros.h added unsupported/test/cxx11_tensor_customIndex.cpp changed unsupported/Eigen/CXX11/Tensor changed unsupported/Eigen/CXX11/src/Tensor/Tensor.h changed unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h changed unsupported/test/CMakeLists.txt
142 lines
3.2 KiB
C++
142 lines
3.2 KiB
C++
// This file is part of Eigen, a lightweight C++ template library
|
|
// for linear algebra.
|
|
//
|
|
// Copyright (C) 2015 Benoit Steiner <benoit.steiner.goog@gmail.com>
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla
|
|
// Public License v. 2.0. If a copy of the MPL was not distributed
|
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
#ifndef EIGEN_CXX11_TENSOR_TENSOR_META_H
|
|
#define EIGEN_CXX11_TENSOR_TENSOR_META_H
|
|
|
|
namespace Eigen {
|
|
|
|
template<bool cond> struct Cond {};
|
|
|
|
template<typename T1, typename T2> EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
|
|
const T1& choose(Cond<true>, const T1& first, const T2&) {
|
|
return first;
|
|
}
|
|
|
|
template<typename T1, typename T2> EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
|
|
const T2& choose(Cond<false>, const T1&, const T2& second) {
|
|
return second;
|
|
}
|
|
|
|
template <size_t n> struct max_n_1 {
|
|
static const size_t size = n;
|
|
};
|
|
template <> struct max_n_1<0> {
|
|
static const size_t size = 1;
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#if defined(EIGEN_HAS_CONSTEXPR)
|
|
#define EIGEN_CONSTEXPR constexpr
|
|
#else
|
|
#define EIGEN_CONSTEXPR
|
|
#endif
|
|
|
|
// Tuple mimics std::pair but works on e.g. nvcc.
|
|
template <typename U, typename V> struct Tuple {
|
|
public:
|
|
U first;
|
|
V second;
|
|
|
|
typedef U first_type;
|
|
typedef V second_type;
|
|
|
|
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
Tuple() : first(), second() {}
|
|
|
|
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
Tuple(const U& f, const V& s) : first(f), second(s) {}
|
|
|
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
Tuple& operator= (const Tuple& rhs) {
|
|
if (&rhs == this) return *this;
|
|
first = rhs.first;
|
|
second = rhs.second;
|
|
return *this;
|
|
}
|
|
|
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
void swap(Tuple& rhs) {
|
|
using numext::swap;
|
|
swap(first, rhs.first);
|
|
swap(second, rhs.second);
|
|
}
|
|
};
|
|
|
|
template <typename U, typename V>
|
|
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
bool operator==(const Tuple<U, V>& x, const Tuple<U, V>& y) {
|
|
return (x.first == y.first && x.second == y.second);
|
|
}
|
|
|
|
template <typename U, typename V>
|
|
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
bool operator!=(const Tuple<U, V>& x, const Tuple<U, V>& y) {
|
|
return !(x == y);
|
|
}
|
|
|
|
|
|
|
|
|
|
namespace internal{
|
|
|
|
template<typename IndexType, Index... Is>
|
|
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
array<Index,sizeof...(Is)> customIndex2Array(const IndexType & idx, numeric_list<Index,Is...>) {
|
|
return { idx(Is)... };
|
|
}
|
|
|
|
/** Make an array (for index/dimensions) out of a custom index */
|
|
template<typename Index, int NumIndices, typename IndexType>
|
|
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
|
array<Index,NumIndices> customIndex2Array(const IndexType & idx) {
|
|
return customIndex2Array(idx, typename gen_numeric_list<Index,NumIndices>::type{});
|
|
}
|
|
|
|
|
|
template <typename B, typename D>
|
|
struct is_base_of
|
|
{
|
|
|
|
typedef char (&yes)[1];
|
|
typedef char (&no)[2];
|
|
|
|
template <typename BB, typename DD>
|
|
struct Host
|
|
{
|
|
operator BB*() const;
|
|
operator DD*();
|
|
};
|
|
|
|
template<typename T>
|
|
static yes check(D*, T);
|
|
static no check(B*, int);
|
|
|
|
static const bool value = sizeof(check(Host<B,D>(), int())) == sizeof(yes);
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#undef EIGEN_CONSTEXPR
|
|
|
|
} // namespace Eigen
|
|
|
|
#endif // EIGEN_CXX11_TENSOR_TENSOR_META_H
|