You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
1.6 KiB
30 lines
1.6 KiB
#include "src/adapters/EigenAdapter.h"
|
|
|
|
namespace storm {
|
|
namespace adapters {
|
|
|
|
template<typename ValueType>
|
|
std::unique_ptr<Eigen::SparseMatrix<ValueType>> EigenAdapter::toEigenSparseMatrix(storm::storage::SparseMatrix<ValueType> const& matrix) {
|
|
// Build a list of triplets and let Eigen care about the insertion.
|
|
std::vector<Eigen::Triplet<ValueType>> triplets;
|
|
triplets.reserve(matrix.getNonzeroEntryCount());
|
|
|
|
for (uint64_t row = 0; row < matrix.getRowCount(); ++row) {
|
|
for (auto const& element : matrix.getRow(row)) {
|
|
triplets.emplace_back(row, element.getColumn(), element.getValue());
|
|
}
|
|
}
|
|
|
|
std::unique_ptr<Eigen::SparseMatrix<ValueType>> result = std::make_unique<Eigen::SparseMatrix<ValueType>>(matrix.getRowCount(), matrix.getColumnCount());
|
|
result->setFromTriplets(triplets.begin(), triplets.end());
|
|
return result;
|
|
}
|
|
|
|
template std::unique_ptr<Eigen::SparseMatrix<double>> EigenAdapter::toEigenSparseMatrix(storm::storage::SparseMatrix<double> const& matrix);
|
|
|
|
#ifdef STORM_HAVE_CARL
|
|
template std::unique_ptr<Eigen::SparseMatrix<storm::RationalNumber>> EigenAdapter::toEigenSparseMatrix(storm::storage::SparseMatrix<storm::RationalNumber> const& matrix);
|
|
template std::unique_ptr<Eigen::SparseMatrix<storm::RationalFunction>> EigenAdapter::toEigenSparseMatrix(storm::storage::SparseMatrix<storm::RationalFunction> const& matrix);
|
|
#endif
|
|
}
|
|
}
|