			# 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/>.
			#

			import numpy as np
			from copy import deepcopy as copy
			from rrompy.utilities.base.types import Np1D, Np2D, paramList
			from rrompy.parameter import checkParameterList
			from rrompy.utilities.poly_fitting.polynomial.val import polyval as pvP

			__all__ = ['polyval']

			def polyval(x:paramList, c:Np2D, poles:Np1D,
			basis : str = "MONOMIAL_HEAVISIDE") -> Np2D:
			x = checkParameterList(x, 1)[0]
			poles = poles.flatten()
			if len(c) > len(poles):
			basisp = basis.split("_")[0]
			c0 = pvP(x, c[len(poles) :], basisp)
			else:
			c0 = np.zeros(c.shape[1:] + (len(x),), dtype = c.dtype)
			csh = copy(c0.shape)
			if len(csh) == 1: c0 = c0.reshape(1, -1)
			for j in range(len(x)):
			muDiff = 1. / (x[j] - poles)
			val = muDiff.dot(c[: len(poles)])
			try:
			c0[..., j] += val
			except:
			c0[..., j] += val.flatten()
			if len(csh) == 1: c0 = c0.flatten()
			return c0

RROMPy/rrompy/utilities/poly_fitting/heaviside/val.py
6643b38fd8aemaster

val.py
View Options