fenics_matrices.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	# Copyright (C) 2018 by the RROMPy authors
	#
	# This file is part of RROMPy.
	#
	# RROMPy is free software: you can redistribute it and/or modify
	# it under the terms of the GNU Lesser General Public License as published by
	# the Free Software Foundation, either version 3 of the License, or
	# (at your option) any later version.
	#
	# RROMPy 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 Lesser General Public License for more details.
	#
	# You should have received a copy of the GNU Lesser General Public License
	# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
	#

	from scipy.sparse import csr_matrix
	import fenics as fen
	from rrompy.utilities.base.types import Np2D, FenBC, FenExpr

	__all__ = ['fenics2Sparse']

	def fenics2Sparse(expr:FenExpr, formCompPars : dict = {}, DBC : FenBC = None,
	zero : bool = False) -> Np2D:
	assembled = fen.assemble(expr, form_compiler_parameters = formCompPars)
	if zero:
	DBC.zero(assembled)
	else:
	DBC.apply(assembled)
	emat = fen.as_backend_type(assembled).mat()
	er, ec, ev = emat.getValuesCSR()
	return csr_matrix((ev, ec, er), shape = emat.size)