xref: /aosp_15_r20/external/eigen/Eigen/src/Core/util/ReshapedHelper.h (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2017 Gael Guennebaud <[email protected]>
5 //
6 // This Source Code Form is subject to the terms of the Mozilla
7 // Public License v. 2.0. If a copy of the MPL was not distributed
8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9 
10 
11 #ifndef EIGEN_RESHAPED_HELPER_H
12 #define EIGEN_RESHAPED_HELPER_H
13 
14 namespace Eigen {
15 
16 enum AutoSize_t   { AutoSize };
17 const int AutoOrder = 2;
18 
19 namespace internal {
20 
21 template<typename SizeType,typename OtherSize, int TotalSize>
22 struct get_compiletime_reshape_size {
23   enum { value = get_fixed_value<SizeType>::value };
24 };
25 
26 template<typename SizeType>
get_runtime_reshape_size(SizeType size,Index,Index)27 Index get_runtime_reshape_size(SizeType size, Index /*other*/, Index /*total*/) {
28   return internal::get_runtime_value(size);
29 }
30 
31 template<typename OtherSize, int TotalSize>
32 struct get_compiletime_reshape_size<AutoSize_t,OtherSize,TotalSize> {
33   enum {
34     other_size = get_fixed_value<OtherSize>::value,
35     value = (TotalSize==Dynamic || other_size==Dynamic) ? Dynamic : TotalSize / other_size };
36 };
37 
38 inline Index get_runtime_reshape_size(AutoSize_t /*size*/, Index other, Index total) {
39   return total/other;
40 }
41 
42 template<int Flags, int Order>
43 struct get_compiletime_reshape_order {
44   enum { value = Order == AutoOrder ? Flags & RowMajorBit : Order };
45 };
46 
47 }
48 
49 } // end namespace Eigen
50 
51 #endif // EIGEN_RESHAPED_HELPER_H
52