roots.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 numpy import polynomial as po
	from .base import polybases
	from .marginalize import polymarginalize
	from rrompy.utilities.base.types import Np1D, Np2D
	from rrompy.utilities.base import freepar as fp
	from rrompy.utilities.exception_manager import RROMPyException

	__all__ = ['polyroots']

	def polyroots(c:Np2D, basis:str, marginalVals : Np1D = [fp]) -> Np1D:
	if basis.upper() not in polybases:
	raise RROMPyException("Polynomial basis not recognized.")
	rootsbase = {"CHEBYSHEV" : po.chebyshev.chebroots,
	"LEGENDRE" : po.legendre.legroots,
	"MONOMIAL" : po.polynomial.polyroots}[basis.upper()]
	return rootsbase(polymarginalize(c, basis, marginalVals, 1))