mirror of
https://github.com/AxioDL/metaforce.git
synced 2025-12-08 18:24:55 +00:00
DCLN cooking and various bug fixes
This commit is contained in:
Jack Andersen
174
gmm/gmm_dense_sylvester.h
Normal file
174
gmm/gmm_dense_sylvester.h
Normal file
@@ -0,0 +1,174 @@
|
||||
/* -*- c++ -*- (enables emacs c++ mode) */
|
||||
/*===========================================================================
|
||||
|
||||
Copyright (C) 2003-2017 Yves Renard
|
||||
|
||||
This file is a part of GetFEM++
|
||||
|
||||
GetFEM++ is free software; you can redistribute it and/or modify it
|
||||
under the terms of the GNU Lesser General Public License as published
|
||||
by the Free Software Foundation; either version 3 of the License, or
|
||||
(at your option) any later version along with the GCC Runtime Library
|
||||
Exception either version 3.1 or (at your option) any later version.
|
||||
This program is distributed in the hope that it will be useful, but
|
||||
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||
or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
|
||||
License and GCC Runtime Library Exception for more details.
|
||||
You should have received a copy of the GNU Lesser General Public License
|
||||
along with this program; if not, write to the Free Software Foundation,
|
||||
Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
|
||||
As a special exception, you may use this file as it is a part of a free
|
||||
software library without restriction. Specifically, if other files
|
||||
instantiate templates or use macros or inline functions from this file,
|
||||
or you compile this file and link it with other files to produce an
|
||||
executable, this file does not by itself cause the resulting executable
|
||||
to be covered by the GNU Lesser General Public License. This exception
|
||||
does not however invalidate any other reasons why the executable file
|
||||
might be covered by the GNU Lesser General Public License.
|
||||
|
||||
===========================================================================*/
|
||||
|
||||
/** @file gmm_dense_sylvester.h
|
||||
@author Yves Renard <Yves.Renard@insa-lyon.fr>
|
||||
@date June 5, 2003.
|
||||
@brief Sylvester equation solver.
|
||||
*/
|
||||
#ifndef GMM_DENSE_SYLVESTER_H
|
||||
#define GMM_DENSE_SYLVESTER_H
|
||||
|
||||
#include "gmm_kernel.h"
|
||||
|
||||
namespace gmm {
|
||||
|
||||
/* ********************************************************************* */
|
||||
/* Kronecker system matrix. */
|
||||
/* ********************************************************************* */
|
||||
template <typename MAT1, typename MAT2, typename MAT3>
|
||||
void kron(const MAT1 &m1, const MAT2 &m2, const MAT3 &m3_,
|
||||
bool init = true) {
|
||||
MAT3 &m3 = const_cast<MAT3 &>(m3_);
|
||||
size_type m = mat_nrows(m1), n = mat_ncols(m1);
|
||||
size_type l = mat_nrows(m2), k = mat_ncols(m2);
|
||||
|
||||
GMM_ASSERT2(mat_nrows(m3) == m*l && mat_ncols(m3) == n*k,
|
||||
"dimensions mismatch");
|
||||
|
||||
for (size_type i = 0; i < m; ++i)
|
||||
for (size_type j = 0; j < m; ++j)
|
||||
if (init)
|
||||
gmm::copy(gmm::scaled(m2, m1(i,j)),
|
||||
gmm::sub_matrix(m3, sub_interval(l*i, l),
|
||||
sub_interval(k*j, k)));
|
||||
else
|
||||
gmm::add(gmm::scaled(m2, m1(i,j)),
|
||||
gmm::sub_matrix(m3, sub_interval(l*i, l),
|
||||
sub_interval(k*j, k)));
|
||||
}
|
||||
|
||||
|
||||
/* ********************************************************************* */
|
||||
/* Copy a matrix into a vector. */
|
||||
/* ********************************************************************* */
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
colmatrix_to_vector(const MAT &A, VECT &v, col_major) {
|
||||
size_type m = mat_nrows(A), n = mat_ncols(A);
|
||||
GMM_ASSERT2(m*n == vect_size(v), "dimensions mismatch");
|
||||
for (size_type i = 0; i < n; ++i)
|
||||
gmm::copy(mat_col(A, i), sub_vector(v, sub_interval(i*m, m)));
|
||||
}
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
colmatrix_to_vector(const MAT &A, VECT &v, row_and_col)
|
||||
{ colmatrix_to_vector(A, v, col_major()); }
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
colmatrix_to_vector(const MAT &A, VECT &v, col_and_row)
|
||||
{ colmatrix_to_vector(A, v, col_major()); }
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
colmatrix_to_vector(const MAT &A, VECT &v, row_major) {
|
||||
size_type m = mat_nrows(mat), n = mat_ncols(A);
|
||||
GMM_ASSERT2(m*n == vect_size(v), "dimensions mismatch");
|
||||
for (size_type i = 0; i < m; ++i)
|
||||
gmm::copy(mat_row(A, i), sub_vector(v, sub_slice(i, n, m)));
|
||||
}
|
||||
|
||||
template <typename MAT, typename VECT> inline
|
||||
colmatrix_to_vector(const MAT &A, const VECT &v_) {
|
||||
VECT &v = const_cast<VECT &>(v_);
|
||||
colmatrix_to_vector(A, v, typename linalg_traits<MAT>::sub_orientation());
|
||||
}
|
||||
|
||||
|
||||
/* ********************************************************************* */
|
||||
/* Copy a vector into a matrix. */
|
||||
/* ********************************************************************* */
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
vector_to_colmatrix(const VECT &v, MAT &A, col_major) {
|
||||
size_type m = mat_nrows(A), n = mat_ncols(A);
|
||||
GMM_ASSERT2(m*n == vect_size(v), "dimensions mismatch");
|
||||
for (size_type i = 0; i < n; ++i)
|
||||
gmm::copy(sub_vector(v, sub_interval(i*m, m)), mat_col(A, i));
|
||||
}
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
vector_to_colmatrix(const VECT &v, MAT &A, row_and_col)
|
||||
{ vector_to_colmatrix(v, A, col_major()); }
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
vector_to_colmatrix(const VECT &v, MAT &A, col_and_row)
|
||||
{ vector_to_colmatrix(v, A, col_major()); }
|
||||
|
||||
template <typename MAT, typename VECT>
|
||||
vector_to_colmatrix(const VECT &v, MAT &A, row_major) {
|
||||
size_type m = mat_nrows(mat), n = mat_ncols(A);
|
||||
GMM_ASSERT2(m*n == vect_size(v), "dimensions mismatch");
|
||||
for (size_type i = 0; i < m; ++i)
|
||||
gmm::copy(sub_vector(v, sub_slice(i, n, m)), mat_row(A, i));
|
||||
}
|
||||
|
||||
template <typename MAT, typename VECT> inline
|
||||
vector_to_colmatrix(const VECT &v, const MAT &A_) {
|
||||
MAT &A = const_cast<MAT &>(A_);
|
||||
vector_to_colmatrix(v, A, typename linalg_traits<MAT>::sub_orientation());
|
||||
}
|
||||
|
||||
/* ********************************************************************* */
|
||||
/* Solve sylvester equation. */
|
||||
/* ********************************************************************* */
|
||||
|
||||
// very prohibitive solver, to be replaced ...
|
||||
template <typename MAT1, typename MAT2, typename MAT3, typename MAT4 >
|
||||
void sylvester(const MAT1 &m1, const MAT2 &m2, const MAT3 &m3,
|
||||
const MAT4 &m4_) {
|
||||
typedef typename linalg_traits<Mat>::value_type T;
|
||||
|
||||
MAT3 &m4 = const_cast<MAT4 &>(m4_);
|
||||
size_type m = mat_nrows(m1), n = mat_ncols(m1);
|
||||
size_type l = mat_nrows(m2), k = mat_ncols(m2);
|
||||
|
||||
GMM_ASSERT2(m == n && l == k && m == mat_nrows(m3) &&
|
||||
l == mat_ncols(m3) && m == mat_nrows(m4) && l == mat_ncols(m4),
|
||||
"dimensions mismatch");
|
||||
|
||||
gmm::dense_matrix<T> akronb(m*l, m*l);
|
||||
gmm::dense_matrix<T> idm(m, m), idl(l,l);
|
||||
gmm::copy(identity_matrix(), idm);
|
||||
gmm::copy(identity_matrix(), idl);
|
||||
std::vector<T> x(m*l), c(m*l);
|
||||
|
||||
kron(idl, m1, akronb);
|
||||
kron(gmm::transposed(m2), idm, akronb, false);
|
||||
|
||||
colmatrix_to_vector(m3, c);
|
||||
lu_solve(akronb, c, x);
|
||||
vector_to_colmatrix(x, m4);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
Reference in New Issue
Block a user