File Metadata

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

	from itertools import product
	from .kernel import kernels
	from rrompy.utilities.exception_manager import RROMPyException
	from rrompy.utilities.poly_fitting.polynomial.base import (polybases as pbP,
	polyfitname as pfnP,
	polydomcoeff as polydomcoeffB)

	__all__ = ['rbbases', 'polybases', 'polyfitname', 'polydomcoeff']

	rbbases = list(kernels.keys())

	polybases = [x + "_" + y for x, y in product(pbP, rbbases)]

	def polyfitname(basis:str) -> str:
	basissp = basis.split("_")
	if len(basissp) != 2 or basissp[1].upper() not in rbbases:
	raise RROMPyException("Polynomial-radial basis combination not "
	"recognized.")
	return pfnP(basissp[0]) + "_" + basissp[1].lower()

	def polydomcoeff(n:int, basis:str) -> float:
	return polydomcoeffB(n, basis.split("_")[0])