OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Added superset() functions.

Browse Source 
The reverse of subset(), superset() takes a sparse representation of a vector
and makes a full representation of it.

Also added a little documentation.
This commit is contained in:
Atgeirr Flø Rasmussen 2013-05-15 08:50:22 +02:00
parent 38259f43bf
commit 959f7cbcac
1 changed files with 38 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
AutoDiffHelpers.hpp
View File

@ -227,9 +227,18 @@ namespace {
sub.setFromTriplets(triplets.begin(), triplets.end()); sub.setFromTriplets(triplets.begin(), triplets.end());
return sub; return sub;
} }
template <typename Scalar, class IntVec>
Eigen::SparseMatrix<Scalar>
constructSupersetSparseMatrix(const int full_size, const IntVec& indices)
{
return constructSubsetSparseMatrix<Scalar>(full_size, indices).transpose();
}
} // anon namespace } // anon namespace
/// Returns x(indices).
template <typename Scalar, class IntVec> template <typename Scalar, class IntVec>
AutoDiff::ForwardBlock<Scalar> AutoDiff::ForwardBlock<Scalar>
subset(const AutoDiff::ForwardBlock<Scalar>& x, subset(const AutoDiff::ForwardBlock<Scalar>& x,
@ -239,6 +248,8 @@ subset(const AutoDiff::ForwardBlock<Scalar>& x,
} }
/// Returns x(indices).
template <typename Scalar, class IntVec> template <typename Scalar, class IntVec>
Eigen::Array<Scalar, Eigen::Dynamic, 1> Eigen::Array<Scalar, Eigen::Dynamic, 1>
subset(const Eigen::Array<Scalar, Eigen::Dynamic, 1>& x, subset(const Eigen::Array<Scalar, Eigen::Dynamic, 1>& x,
@ -248,6 +259,33 @@ subset(const Eigen::Array<Scalar, Eigen::Dynamic, 1>& x,
} }
/// Returns v where v(indices) == x, v(!indices) == 0 and v.size() == n.
template <typename Scalar, class IntVec>
AutoDiff::ForwardBlock<Scalar>
superset(const AutoDiff::ForwardBlock<Scalar>& x,
const IntVec& indices,
const int n)
{
return ::constructSupersetSparseMatrix<Scalar>(n, indices) * x;
}
/// Returns v where v(indices) == x, v(!indices) == 0 and v.size() == n.
template <typename Scalar, class IntVec>
Eigen::Array<Scalar, Eigen::Dynamic, 1>
superset(const Eigen::Array<Scalar, Eigen::Dynamic, 1>& x,
const IntVec& indices,
const int n)
{
return ::constructSupersetSparseMatrix<Scalar>(n, indices) * x;
}
/// Construct square sparse matrix with the
/// elements of d on the diagonal.
AutoDiff::ForwardBlock<double>::M AutoDiff::ForwardBlock<double>::M
spdiag(const AutoDiff::ForwardBlock<double>::V& d) spdiag(const AutoDiff::ForwardBlock<double>::V& d)
{ {