3
0
mirror of https://github.com/triqs/dft_tools synced 2024-12-27 06:43:40 +01:00
dft_tools/triqs/arrays/blas_lapack/getrf.hpp
Olivier Parcollet 072b45ac1c arrays: Remove dim0, dim1, .shape in various matrix object.
Not in the concept, not needed, just an annoyance.
replaced by free functions :
first_dim(A), second_dim(A), get_shape(A) and so on...
2013-08-27 13:43:58 +02:00

65 lines
2.5 KiB
C++

/*******************************************************************************
*
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
*
* Copyright (C) 2012 by O. Parcollet
*
* TRIQS is free software: you can redistribute it and/or modify it under the
* terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later
* version.
*
* TRIQS 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 General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License along with
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
*
******************************************************************************/
#ifndef TRIQS_ARRAYS_BLAS_LAPACK_GETRF_H
#define TRIQS_ARRAYS_BLAS_LAPACK_GETRF_H
#include <complex>
#include "./tools.hpp"
#include "./qcache.hpp"
namespace triqs { namespace arrays { namespace lapack {
using namespace blas_lapack_tools;
namespace f77 { // overload
typedef std::complex<double> dcomplex;
extern "C" {
void TRIQS_FORTRAN_MANGLING(dgetrf) (const int &,const int &,double*, const int & ,int *,int &);
void TRIQS_FORTRAN_MANGLING(zgetrf) (const int &,const int &,std::complex<double>*, const int & ,int *,int &);
}
inline void getrf (const int & M, const int & N, double* A, const int & LDA, int * ipiv, int & info) {
TRIQS_FORTRAN_MANGLING(dgetrf)(M,N,A,LDA,ipiv,info);
}
inline void getrf (const int & M, const int & N, std::complex<double>* A, const int & LDA, int * ipiv, int & info) {
TRIQS_FORTRAN_MANGLING(zgetrf)(M,N,A,LDA,ipiv,info);
}
}
/**
* Calls getrf on a matrix or view
* Takes care of making temporary copies if necessary
*/
template<typename MT>
typename std::enable_if< is_blas_lapack_type<typename MT::value_type>::value, int >::type
getrf (MT & A, arrays::vector<int> & ipiv, bool assert_fortran_order = false ) {
if (assert_fortran_order && A.memory_layout_is_c()) TRIQS_RUNTIME_ERROR<< "matrix passed to getrf is not in Fortran order";
reflexive_qcache<MT> Ca(A);
auto dm = std::min(first_dim(Ca()), second_dim(Ca()));
if (ipiv.size() < dm) ipiv.resize(dm);
int info;
f77::getrf ( get_n_rows(Ca()), get_n_cols(Ca()), Ca().data_start(), get_ld(Ca()), ipiv.data_start(), info);
return info;
}
}}}// namespace
#endif