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

1__DOC__ = """Mathematical functions for Larch""" 

2 

3import numpy as np 

4 

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) 

9 

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) 

15 

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 

25 

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