5#ifndef DUNE_ISTL_UMFPACK_HH
6#define DUNE_ISTL_UMFPACK_HH
8#if HAVE_SUITESPARSE_UMFPACK || defined DOXYGEN
18#include<dune/istl/bccsmatrixinitializer.hh>
22#include <dune/istl/solverfactory.hh>
39 template<
class M,
class T,
class TM,
class TD,
class TA>
40 class SeqOverlappingSchwarz;
42 template<
class T,
bool tag>
43 struct SeqOverlappingSchwarzAssemblerHelper;
48 struct UMFPackMethodChooser
50 static constexpr bool valid = false ;
54 struct UMFPackMethodChooser<double>
56 static constexpr bool valid = true ;
58 template<
typename... A>
59 static void defaults(A... args)
61 umfpack_dl_defaults(args...);
63 template<
typename... A>
64 static void free_numeric(A... args)
66 umfpack_dl_free_numeric(args...);
68 template<
typename... A>
69 static void free_symbolic(A... args)
71 umfpack_dl_free_symbolic(args...);
73 template<
typename... A>
74 static int load_numeric(A... args)
76 return umfpack_dl_load_numeric(args...);
78 template<
typename... A>
79 static void numeric(A... args)
81 umfpack_dl_numeric(args...);
83 template<
typename... A>
84 static void report_info(A... args)
86 umfpack_dl_report_info(args...);
88 template<
typename... A>
89 static void report_status(A... args)
91 umfpack_dl_report_status(args...);
93 template<
typename... A>
94 static int save_numeric(A... args)
96 return umfpack_dl_save_numeric(args...);
98 template<
typename... A>
99 static void solve(A... args)
101 umfpack_dl_solve(args...);
103 template<
typename... A>
104 static void symbolic(A... args)
106 umfpack_dl_symbolic(args...);
111 struct UMFPackMethodChooser<
std::complex<double> >
113 static constexpr bool valid = true ;
115 template<
typename... A>
116 static void defaults(A... args)
118 umfpack_zl_defaults(args...);
120 template<
typename... A>
121 static void free_numeric(A... args)
123 umfpack_zl_free_numeric(args...);
125 template<
typename... A>
126 static void free_symbolic(A... args)
128 umfpack_zl_free_symbolic(args...);
130 template<
typename... A>
131 static int load_numeric(A... args)
133 return umfpack_zl_load_numeric(args...);
135 template<
typename... A>
136 static void numeric(
const long int* cs,
const long int* ri,
const double* val, A... args)
138 umfpack_zl_numeric(cs,ri,val,NULL,args...);
140 template<
typename... A>
141 static void report_info(A... args)
143 umfpack_zl_report_info(args...);
145 template<
typename... A>
146 static void report_status(A... args)
148 umfpack_zl_report_status(args...);
150 template<
typename... A>
151 static int save_numeric(A... args)
153 return umfpack_zl_save_numeric(args...);
155 template<
typename... A>
156 static void solve(
long int m,
const long int* cs,
const long int* ri, std::complex<double>* val,
double* x,
const double* b,A... args)
158 const double* cval =
reinterpret_cast<const double*
>(val);
159 umfpack_zl_solve(m,cs,ri,cval,NULL,x,NULL,b,NULL,args...);
161 template<
typename... A>
162 static void symbolic(
long int m,
long int n,
const long int* cs,
const long int* ri,
const double* val, A... args)
164 umfpack_zl_symbolic(m,n,cs,ri,val,NULL,args...);
171 struct UMFPackVectorChooser
174 template<
typename T,
typename A,
int n,
int m>
175 struct UMFPackVectorChooser<BCRSMatrix<FieldMatrix<T,n,m>,A > >
178 using domain_type = BlockVector<
180 typename std::allocator_traits<A>::template rebind_alloc<FieldVector<T,m> > >;
182 using range_type = BlockVector<
184 typename std::allocator_traits<A>::template rebind_alloc<FieldVector<T,n> > >;
187 template<
typename T,
typename A>
188 struct UMFPackVectorChooser<BCRSMatrix<T,A> >
191 using domain_type = BlockVector<T, A>;
193 using range_type = BlockVector<T, A>;
213 typename Impl::UMFPackVectorChooser<M>::domain_type,
214 typename Impl::UMFPackVectorChooser<M>::range_type >
216 using T =
typename M::field_type;
221 using matrix_type = M;
223 typedef ISTL::Impl::BCCSMatrix<typename Matrix::field_type, long int>
UMFPackMatrix;
227 using domain_type =
typename Impl::UMFPackVectorChooser<M>::domain_type;
229 using range_type =
typename Impl::UMFPackVectorChooser<M>::range_type;
234 return SolverCategory::Category::sequential;
248 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
249 "Unsupported Type in UMFPack (only double and std::complex<double> supported)");
250 Caller::defaults(UMF_Control);
266 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
267 "Unsupported Type in UMFPack (only double and std::complex<double> supported)");
268 Caller::defaults(UMF_Control);
283 :
UMFPack(mat_, config.get<int>(
"verbose", 0))
288 UMFPack() : matrixIsLoaded_(false), verbosity_(0)
291 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
292 "Unsupported Type in UMFPack (only double and std::complex<double> supported)");
293 Caller::defaults(UMF_Control);
309 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
310 "Unsupported Type in UMFPack (only double and std::complex<double> supported)");
311 Caller::defaults(UMF_Control);
313 int errcode = Caller::load_numeric(&UMF_Numeric,
const_cast<char*
>(file));
314 if ((errcode == UMFPACK_ERROR_out_of_memory) || (errcode == UMFPACK_ERROR_file_IO))
316 matrixIsLoaded_ =
false;
322 matrixIsLoaded_ =
true;
323 std::cout <<
"UMFPack decomposition successfully loaded from " << file << std::endl;
336 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
337 "Unsupported Type in UMFPack (only double and std::complex<double> supported)");
338 Caller::defaults(UMF_Control);
339 int errcode = Caller::load_numeric(&UMF_Numeric,
const_cast<char*
>(file));
340 if (errcode == UMFPACK_ERROR_out_of_memory)
342 if (errcode == UMFPACK_ERROR_file_IO)
344 matrixIsLoaded_ =
true;
345 std::cout <<
"UMFPack decomposition successfully loaded from " << file << std::endl;
351 if ((umfpackMatrix_.N() + umfpackMatrix_.M() > 0) || matrixIsLoaded_)
360 if (umfpackMatrix_.N() != b.dim())
362 if (umfpackMatrix_.M() != x.dim())
367 double UMF_Apply_Info[UMFPACK_INFO];
368 Caller::solve(UMFPACK_A,
369 umfpackMatrix_.getColStart(),
370 umfpackMatrix_.getRowIndex(),
371 umfpackMatrix_.getValues(),
372 reinterpret_cast<double*
>(&x[0]),
373 reinterpret_cast<double*
>(&b[0]),
381 res.
elapsed = UMF_Apply_Info[UMFPACK_SOLVE_WALLTIME];
383 printOnApply(UMF_Apply_Info);
401 double UMF_Apply_Info[UMFPACK_INFO];
402 Caller::solve(UMFPACK_A,
403 umfpackMatrix_.getColStart(),
404 umfpackMatrix_.getRowIndex(),
405 umfpackMatrix_.getValues(),
411 printOnApply(UMF_Apply_Info);
427 if (option >= UMFPACK_CONTROL)
430 UMF_Control[option] = value;
438 int errcode = Caller::save_numeric(UMF_Numeric,
const_cast<char*
>(file));
439 if (errcode != UMFPACK_OK)
446 if ((umfpackMatrix_.N() + umfpackMatrix_.M() > 0) || matrixIsLoaded_)
448 if (matrix.N() == 0 or matrix.M() == 0)
451 if (umfpackMatrix_.N() + umfpackMatrix_.M() + umfpackMatrix_.nonzeroes() != 0)
452 umfpackMatrix_.free();
453 umfpackMatrix_.setSize(MatrixDimension<Matrix>::rowdim(matrix),
454 MatrixDimension<Matrix>::coldim(matrix));
455 ISTL::Impl::BCCSMatrixInitializer<Matrix, long int> initializer(umfpackMatrix_);
457 copyToBCCSMatrix(initializer, matrix);
463 void setSubMatrix(
const Matrix& _mat,
const S& rowIndexSet)
465 if ((umfpackMatrix_.N() + umfpackMatrix_.M() > 0) || matrixIsLoaded_)
468 if (umfpackMatrix_.N() + umfpackMatrix_.M() + umfpackMatrix_.nonzeroes() != 0)
469 umfpackMatrix_.free();
471 umfpackMatrix_.setSize(rowIndexSet.size()*MatrixDimension<Matrix>::rowdim(_mat) / _mat.
N(),
472 rowIndexSet.size()*MatrixDimension<Matrix>::coldim(_mat) / _mat.
M());
473 ISTL::Impl::BCCSMatrixInitializer<Matrix, long int> initializer(umfpackMatrix_);
475 copyToBCCSMatrix(initializer, ISTL::Impl::MatrixRowSubset<
Matrix,std::set<std::size_t> >(_mat,rowIndexSet));
492 UMF_Control[UMFPACK_PRL] = 1;
494 UMF_Control[UMFPACK_PRL] = 2;
496 UMF_Control[UMFPACK_PRL] = 4;
514 return umfpackMatrix_;
523 if (!matrixIsLoaded_)
525 Caller::free_symbolic(&UMF_Symbolic);
526 umfpackMatrix_.free();
528 Caller::free_numeric(&UMF_Numeric);
529 matrixIsLoaded_ =
false;
532 const char* name() {
return "UMFPACK"; }
535 typedef typename Dune::UMFPackMethodChooser<T> Caller;
537 template<
class Mat,
class X,
class TM,
class TD,
class T1>
538 friend class SeqOverlappingSchwarz;
539 friend struct SeqOverlappingSchwarzAssemblerHelper<
UMFPack<
Matrix>,true>;
544 double UMF_Decomposition_Info[UMFPACK_INFO];
545 Caller::symbolic(
static_cast<int>(umfpackMatrix_.N()),
546 static_cast<int>(umfpackMatrix_.N()),
547 umfpackMatrix_.getColStart(),
548 umfpackMatrix_.getRowIndex(),
549 reinterpret_cast<double*
>(umfpackMatrix_.getValues()),
552 UMF_Decomposition_Info);
553 Caller::numeric(umfpackMatrix_.getColStart(),
554 umfpackMatrix_.getRowIndex(),
555 reinterpret_cast<double*
>(umfpackMatrix_.getValues()),
559 UMF_Decomposition_Info);
560 Caller::report_status(UMF_Control,UMF_Decomposition_Info[UMFPACK_STATUS]);
563 std::cout <<
"[UMFPack Decomposition]" << std::endl;
564 std::cout <<
"Wallclock Time taken: " << UMF_Decomposition_Info[UMFPACK_NUMERIC_WALLTIME] <<
" (CPU Time: " << UMF_Decomposition_Info[UMFPACK_NUMERIC_TIME] <<
")" << std::endl;
565 std::cout <<
"Flops taken: " << UMF_Decomposition_Info[UMFPACK_FLOPS] << std::endl;
566 std::cout <<
"Peak Memory Usage: " << UMF_Decomposition_Info[UMFPACK_PEAK_MEMORY]*UMF_Decomposition_Info[UMFPACK_SIZE_OF_UNIT] <<
" bytes" << std::endl;
567 std::cout <<
"Condition number estimate: " << 1./UMF_Decomposition_Info[UMFPACK_RCOND] << std::endl;
568 std::cout <<
"Numbers of non-zeroes in decomposition: L: " << UMF_Decomposition_Info[UMFPACK_LNZ] <<
" U: " << UMF_Decomposition_Info[UMFPACK_UNZ] << std::endl;
572 Caller::report_info(UMF_Control,UMF_Decomposition_Info);
576 void printOnApply(
double* UMF_Info)
578 Caller::report_status(UMF_Control,UMF_Info[UMFPACK_STATUS]);
581 std::cout <<
"[UMFPack Solve]" << std::endl;
582 std::cout <<
"Wallclock Time: " << UMF_Info[UMFPACK_SOLVE_WALLTIME] <<
" (CPU Time: " << UMF_Info[UMFPACK_SOLVE_TIME] <<
")" << std::endl;
583 std::cout <<
"Flops Taken: " << UMF_Info[UMFPACK_SOLVE_FLOPS] << std::endl;
584 std::cout <<
"Iterative Refinement steps taken: " << UMF_Info[UMFPACK_IR_TAKEN] << std::endl;
585 std::cout <<
"Error Estimate: " << UMF_Info[UMFPACK_OMEGA1] <<
" resp. " << UMF_Info[UMFPACK_OMEGA2] << std::endl;
590 bool matrixIsLoaded_;
594 double UMF_Control[UMFPACK_CONTROL];
597 template<
typename T,
typename A,
int n,
int m>
598 struct IsDirectSolver<UMFPack<BCRSMatrix<FieldMatrix<T,n,m>,A> > >
603 template<
typename T,
typename A>
604 struct StoresColumnCompressed<UMFPack<BCRSMatrix<T,A> > >
606 enum { value =
true };
609 struct UMFPackCreator {
610 template<
class F,
class=
void>
struct isValidBlock : std::false_type{};
611 template<
class B>
struct isValidBlock<B,
std::enable_if_t<std::is_same<typename FieldTraits<B>::real_type,double>::value>> : std::true_type {};
613 template<
typename TL,
typename M>
614 std::shared_ptr<Dune::InverseOperator<typename Dune::TypeListElement<1, TL>::type,
615 typename Dune::TypeListElement<2, TL>::type>>
618 isValidBlock<
typename Dune::TypeListElement<1, TL>::type::block_type>::value,
int> = 0)
const
620 int verbose = config.
get(
"verbose", 0);
621 return std::make_shared<Dune::UMFPack<M>>(mat,verbose);
625 template<
typename TL,
typename M>
626 std::shared_ptr<Dune::InverseOperator<typename Dune::TypeListElement<1, TL>::type,
627 typename Dune::TypeListElement<2, TL>::type>>
630 !isValidBlock<
typename Dune::TypeListElement<1, TL>::type::block_type>::value,
int> = 0)
const
633 "Unsupported Type in UMFPack (only double and std::complex<double> supported)");
636 DUNE_REGISTER_DIRECT_SOLVER(
"umfpack",Dune::UMFPackCreator());
Implementation of the BCRSMatrix class.
Base class for Dune-Exceptions.
Definition: exceptions.hh:96
derive error class from the base class in common
Definition: istlexception.hh:19
Abstract base class for all solvers.
Definition: solver.hh:99
A generic dynamic dense matrix.
Definition: matrix.hh:561
size_type M() const
Return the number of columns.
Definition: matrix.hh:700
size_type N() const
Return the number of rows.
Definition: matrix.hh:695
Hierarchical structure of string parameters.
Definition: parametertree.hh:37
std::string get(const std::string &key, const std::string &defaultValue) const
get value as string
Definition: parametertree.cc:185
Default exception class for range errors.
Definition: exceptions.hh:254
The UMFPack direct sparse solver.
Definition: umfpack.hh:215
A few common exception classes.
Implements a matrix constructed from a given type representing a field and compile-time given number ...
Implements a vector constructed from a given type representing a field and a compile-time given size.
#define DUNE_THROW(E, m)
Definition: exceptions.hh:218
void free()
free allocated space.
Definition: umfpack.hh:521
virtual void apply(domain_type &x, range_type &b, InverseOperatorResult &res)
Apply inverse operator,.
Definition: umfpack.hh:358
virtual SolverCategory::Category category() const
Category of the solver (see SolverCategory::Category)
Definition: umfpack.hh:232
void setMatrix(const Matrix &matrix)
Initialize data from given matrix.
Definition: umfpack.hh:444
UMFPack(const Matrix &mat_, const ParameterTree &config)
Construct a solver object from a matrix.
Definition: umfpack.hh:282
ISTL::Impl::BCCSMatrix< typename Matrix::field_type, long int > UMFPackMatrix
The corresponding UMFPack matrix type.
Definition: umfpack.hh:223
UMFPack(const Matrix &mat_, const char *file, int verbose=0)
Try loading a decomposition from file and do a decomposition if unsuccessful.
Definition: umfpack.hh:306
typename Impl::UMFPackVectorChooser< M >::range_type range_type
The type of the range of the solver.
Definition: umfpack.hh:229
UMFPack()
default constructor
Definition: umfpack.hh:288
ISTL::Impl::BCCSMatrixInitializer< M, long int > MatrixInitializer
Type of an associated initializer class.
Definition: umfpack.hh:225
void apply(T *x, T *b)
additional apply method with c-arrays in analogy to superlu
Definition: umfpack.hh:399
void setVerbosity(int v)
sets the verbosity level for the UMFPack solver
Definition: umfpack.hh:487
UMFPack(const char *file, int verbose=0)
try loading a decomposition from file
Definition: umfpack.hh:333
void saveDecomposition(const char *file)
saves a decomposition to a file
Definition: umfpack.hh:436
UMFPackMatrix & getInternalMatrix()
Return the column compress matrix from UMFPack.
Definition: umfpack.hh:512
typename Impl::UMFPackVectorChooser< M >::domain_type domain_type
The type of the domain of the solver.
Definition: umfpack.hh:227
UMFPack(const Matrix &matrix, int verbose=0)
Construct a solver object from a matrix.
Definition: umfpack.hh:245
virtual void apply(domain_type &x, range_type &b, double reduction, InverseOperatorResult &res)
apply inverse operator, with given convergence criteria.
Definition: umfpack.hh:389
void setOption(unsigned int option, double value)
Set UMFPack-specific options.
Definition: umfpack.hh:425
M Matrix
The matrix type.
Definition: umfpack.hh:220
UMFPack(const Matrix &matrix, int verbose, bool)
Constructor for compatibility with SuperLU standard constructor.
Definition: umfpack.hh:263
void * getFactorization()
Return the matrix factorization.
Definition: umfpack.hh:503
Dune namespace.
Definition: alignedallocator.hh:13
Implementations of the inverse operator interface.
Templates characterizing the type of a solver.
Statistics about the application of an inverse operator.
Definition: solver.hh:48
double elapsed
Elapsed time in seconds.
Definition: solver.hh:82
int iterations
Number of iterations.
Definition: solver.hh:67
bool converged
True if convergence criterion has been met.
Definition: solver.hh:73
Category
Definition: solvercategory.hh:23