Add a skeleton of C API for Eigen3.
[blender.git] / extern / Eigen3 / intern / eigenvalues.cc
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2015 Blender Foundation.
19  * All rights reserved.
20  *
21  * Contributor(s): Blender Foundation,
22  *                 Bastien Montagne
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 #ifndef __EIGEN3_EIGENVALUES_C_API_CC__
28 #define __EIGEN3_EIGENVALUES_C_API_CC__
29
30 /* Eigen gives annoying huge amount of warnings here, silence them! */
31 #ifdef __GNUC__
32 #  pragma GCC diagnostic ignored "-Wlogical-op"
33 #endif
34
35 #include <Eigen/Core>
36 #include <Eigen/Eigenvalues>
37
38 #include "eigenvalues.h"
39
40 using Eigen::SelfAdjointEigenSolver;
41
42 using Eigen::MatrixXf;
43 using Eigen::VectorXf;
44 using Eigen::Map;
45
46 using Eigen::Success;
47
48 bool EG3_self_adjoint_eigen_solve(const int size, const float *matrix, float *r_eigen_values, float *r_eigen_vectors)
49 {
50         SelfAdjointEigenSolver<MatrixXf> eigen_solver;
51
52         /* Blender and Eigen matrices are both column-major. */
53         eigen_solver.compute(Map<MatrixXf>((float *)matrix, size, size));
54
55         if (eigen_solver.info() != Success) {
56                 return false;
57         }
58
59         if (r_eigen_values) {
60                 Map<VectorXf>(r_eigen_values, size) = eigen_solver.eigenvalues().transpose();
61         }
62
63         if (r_eigen_vectors) {
64                 Map<MatrixXf>(r_eigen_vectors, size, size) = eigen_solver.eigenvectors();
65         }
66
67         return true;
68 }
69
70 #endif  /* __EIGEN3_EIGENVALUES_C_API_CC__ */