OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-16 20:24:48 -06:00
Code Issues Packages Projects Releases Wiki Activity

Implement simple constructor for sparse, diagonal matrix.

Browse Source 
This is, effectively, the same as MATLAB's

    spdiags(d, 0)

constructor.
This commit is contained in:
Bård Skaflestad 2013-05-08 14:32:48 +02:00
parent 673c044a6c
commit baab2cdf43
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
AutoDiffHelpers.hpp
View File

@ -247,4 +247,20 @@ subset(const Eigen::Array<Scalar, Eigen::Dynamic, 1>& x,
return (::constructSubsetSparseMatrix<Scalar>(x.size(), indices) * x.matrix()).array();
}
AutoDiff::ForwardBlock<double>::M
spdiag(const AutoDiff::ForwardBlock<double>::V& d)
{
typedef AutoDiff::ForwardBlock<double>::M M;
const int n = d.size();
M mat(n, n);
mat.reserve(Eigen::ArrayXi::Ones(n, 1));
for (M::Index i = 0; i < n; ++i) {
mat.insert(i, i) = d[i];
}
return mat;
}
#endif // OPM_AUTODIFFHELPERS_HEADER_INCLUDED