Coverage for larch/math/__init__.py: 100%
14 statements
« prev ^ index » next coverage.py v7.6.0, created at 2024-10-16 21:04 +0000
« prev ^ index » next coverage.py v7.6.0, created at 2024-10-16 21:04 +0000
1__DOC__ = """Mathematical functions for Larch"""
3import numpy as np
5from .utils import (linregress, realimag, as_ndarray,
6 complex_phase, deriv, interp, interp1d, safe_log,
7 remove_dups, remove_nans, remove_nans2, index_of,
8 index_nearest, savitzky_golay, smooth, boxcar, polyfit)
10from .lineshapes import (gaussian, lorentzian, voigt, pvoigt, hypermet,
11 pearson7, lognormal, gammaln,
12 breit_wigner, damped_oscillator,
13 expgaussian, doniach, skewed_voigt,
14 students_t, logistic, erf, erfc, wofz)
16from .peaks import peak_indices
17from .fitpeak import fit_peak
18from .convolution1D import glinbroad
19from .lincombo_fitting import lincombo_fit, lincombo_fitall, groups2matrix
20from .pca import pca_train, pca_fit, nmf_train, save_pca_model, read_pca_model
21from .learn_regress import pls_train, pls_predict, lasso_train, lasso_predict
22from .gridxyz import gridxyz
23from .spline import spline_rep, spline_eval
24from . import transformations as trans
26_larch_builtins = {'_math': dict(linregress=linregress, polyfit=polyfit,
27 realimag=realimag, as_ndarray=as_ndarray,
28 complex_phase=complex_phase, deriv=deriv,
29 interp=interp, interp1d=interp1d,
30 remove_dups=remove_dups,
31 remove_nans2=remove_nans2,
32 safe_log=safe_log,
33 index_of=index_of,
34 index_nearest=index_nearest,
35 savitzky_golay=savitzky_golay,
36 smooth=smooth, boxcar=boxcar,
37 glinbroad=glinbroad, gridxyz=gridxyz,
38 pca_train=pca_train,
39 pca_fit=pca_fit,
40 save_pca_model=save_pca_model,
41 read_pca_model=read_pca_model,
42 nmf_train=nmf_train,
43 pls_train=pls_train,
44 pls_predict=pls_predict,
45 lasso_train=lasso_train,
46 lasso_predict=lasso_predict,
47 groups2matrix=groups2matrix,
48 fit_peak=fit_peak,
49 lincombo_fit=lincombo_fit,
50 lincombo_fitall=lincombo_fitall,
51 spline_rep=spline_rep,
52 spline_eval=spline_eval,
53 gaussian=gaussian,
54 lorentzian=lorentzian, voigt=voigt,
55 pvoigt=pvoigt, hypermet=hypermet,
56 pearson7=pearson7, lognormal=lognormal,
57 gammaln=gammaln,
58 breit_wigner=breit_wigner,
59 damped_oscillator=damped_oscillator,
60 expgaussian=expgaussian,
61 doniach=doniach,
62 skewed_voigt=skewed_voigt,
63 students_t=students_t, logistic=logistic,
64 erf=erf, erfc=erfc, wofz=wofz),
65 '_math.transforms': {# 'doc': trans.__doc__,
66 'identity_matrix': trans.identity_matrix,
67 'translation_matrix': trans.translation_matrix,
68 'translation_from_matrix': trans.translation_from_matrix,
69 'reflection_matrix': trans.reflection_matrix,
70 'reflection_from_matrix': trans.reflection_from_matrix,
71 'rotation_matrix': trans.rotation_matrix,
72 'rotation_from_matrix': trans.rotation_from_matrix,
73 'scale_matrix': trans.scale_matrix,
74 'scale_from_matrix': trans.scale_from_matrix,
75 'projection_matrix': trans.projection_matrix,
76 'projection_from_matrix': trans.projection_from_matrix,
77 'clip_matrix': trans.clip_matrix,
78 'shear_matrix': trans.shear_matrix,
79 'shear_from_matrix': trans.shear_from_matrix,
80 'decompose_matrix': trans.decompose_matrix,
81 'compose_matrix': trans.compose_matrix,
82 'orthogonalization_matrix': trans.orthogonalization_matrix,
83 'affine_matrix_from_points': trans.affine_matrix_from_points,
84 'superimposition_matrix': trans.superimposition_matrix,
85 'euler_matrix': trans.euler_matrix,
86 'euler_from_matrix': trans.euler_from_matrix,
87 'euler_from_quaternion': trans.euler_from_quaternion,
88 'quaternion_from_euler': trans.quaternion_from_euler,
89 'quaternion_about_axis': trans.quaternion_about_axis,
90 'quaternion_matrix': trans.quaternion_matrix,
91 'quaternion_from_matrix': trans.quaternion_from_matrix,
92 'quaternion_multiply': trans.quaternion_multiply,
93 'quaternion_conjugate': trans.quaternion_conjugate,
94 'quaternion_inverse': trans.quaternion_inverse,
95 'quaternion_real': trans.quaternion_real,
96 'quaternion_imag': trans.quaternion_imag,
97 'quaternion_slerp': trans.quaternion_slerp,
98 'vector_norm': trans.vector_norm,
99 'unit_vector': trans.unit_vector,
100 'vector_product': trans.vector_product,
101 'angle_between_vectors': trans.angle_between_vectors,
102 'inverse_matrix': trans.inverse_matrix}}