File Metadata

val.py
View Options

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