Homec4science
Diffusion RationalROMPy (develop) Browse

RROMPy/rrompy/reduction_methods/base/fit_utils.py
88e97ff5f796develop

Last Change
Raw File

fit_utils.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 numpy import pi, polynomial as po
from rrompy.utilities.base.types import DictAny
from rrompy.utilities.warning_manager import warn
__all__ = ['setupFitCallables']
def setupFitCallables(kind:str) -> DictAny:
"""Setup main callables and name for data fitting."""
kind = kind.upper()
if kind == "CHEBYSHEV":
val = po.chebyshev.chebval
vander = po.chebyshev.chebvander
fitname = "chebfit"
roots = po.chebyshev.chebroots
domcoeff = lambda n: 2. ** (n - 1) if n > 0 else 1.
elif kind == "LEGENDRE":
val = po.legendre.legval
vander = po.legendre.legvander
fitname = "legfit"
roots = po.legendre.legroots
from scipy.special import binom
domcoeff = lambda n: (2. ** n * (pi * n) ** -.5 if n > 10
else .5 ** n * binom(2 * n, n))
else:
if kind != "MONOMIAL":
warn("Fitting basis not recognized. Overriding to 'MONOMIAL'.")
val = po.polynomial.polyval
vander = po.polynomial.polyvander
fitname = "polyfit"
roots = po.polynomial.polyroots
domcoeff = lambda n: 1.
return {"val":val, "vander":vander, "fitname":fitname,
"roots":roots, "domcoeff":domcoeff}
c4science · Help