Fix ulab, math and template.

This commit is contained in:
Scott Shawcroft 2020-05-14 15:57:35 -07:00
parent b477c4812d
commit afc84c2fd1
No known key found for this signature in database
GPG Key ID: 9349BC7E64B1921E
15 changed files with 739 additions and 689 deletions

2
.gitignore vendored
View File

@ -52,7 +52,7 @@ _build
# Generated rst files
######################
genrst/
autoapi/
/autoapi/
# ctags and similar
###################

View File

@ -67,6 +67,7 @@ help:
clean:
rm -rf $(BUILDDIR)/*
rm -rf autoapi
rm -rf $(STUBDIR) $(DISTDIR) *.egg-info
html: stubs
@ -213,10 +214,10 @@ check-translate: locale/circuitpython.pot $(wildcard locale/*.po)
$(PYTHON) tools/check_translations.py $^
stubs:
mkdir -p circuitpython-stubs
python tools/extract_pyi.py shared-bindings/ $(STUBDIR)
python tools/extract_pyi.py ports/atmel-samd/bindings $(STUBDIR)
python setup.py sdist
@mkdir -p circuitpython-stubs
@$(PYTHON) tools/extract_pyi.py shared-bindings/ $(STUBDIR)
@$(PYTHON) tools/extract_pyi.py ports/atmel-samd/bindings $(STUBDIR)
@$(PYTHON) setup.py -q sdist
update-frozen-libraries:
@echo "Updating all frozen libraries to latest tagged version."

View File

@ -71,11 +71,13 @@ extensions.append('autoapi.extension')
autoapi_type = 'python'
# Uncomment this if debugging autoapi
# autoapi_keep_files = True
autoapi_dirs = ['circuitpython-stubs']
autoapi_keep_files = True
autoapi_dirs = [os.path.join('circuitpython-stubs', x) for x in os.listdir('circuitpython-stubs')]
print(autoapi_dirs)
autoapi_add_toctree_entry = False
autoapi_options = ['members', 'undoc-members', 'private-members', 'show-inheritance', 'special-members', 'show-module-summary']
autoapi_template_dir = 'docs/autoapi/templates'
autoapi_python_use_implicit_namespaces = True
# The encoding of source files.
#source_encoding = 'utf-8-sig'

View File

@ -0,0 +1,93 @@
{% if not obj.display %}
:orphan:
{% endif %}
:mod:`{{ obj.name }}`
======={{ "=" * obj.name|length }}
.. py:module:: {{ obj.name }}
{% if obj.docstring %}
.. autoapi-nested-parse::
{{ obj.docstring|prepare_docstring|indent(3) }}
{% endif %}
{% block subpackages %}
{% set visible_subpackages = obj.subpackages|selectattr("display")|list %}
{% if visible_subpackages %}
.. toctree::
:titlesonly:
:maxdepth: 3
{% for subpackage in visible_subpackages %}
{{ subpackage.short_name }}/index.rst
{% endfor %}
{% endif %}
{% endblock %}
{% block submodules %}
{% set visible_submodules = obj.submodules|selectattr("display")|list %}
{% if visible_submodules %}
.. toctree::
:titlesonly:
:maxdepth: 1
{% for submodule in visible_submodules %}
{{ submodule.short_name }}/index.rst
{% endfor %}
{% endif %}
{% endblock %}
{% block content %}
{% if obj.all is not none %}
{% set visible_children = obj.children|selectattr("short_name", "in", obj.all)|list %}
{% elif obj.type is equalto("package") %}
{% set visible_children = obj.children|selectattr("display")|list %}
{% else %}
{% set visible_children = obj.children|selectattr("display")|rejectattr("imported")|list %}
{% endif %}
{% if visible_children %}
{% set visible_classes = visible_children|selectattr("type", "equalto", "class")|list %}
{% set visible_functions = visible_children|selectattr("type", "equalto", "function")|list %}
{% if "show-module-summary" in autoapi_options and (visible_classes or visible_functions) %}
{% block classes %}
{% if visible_classes %}
Classes
~~~~~~~
.. autoapisummary::
{% for klass in visible_classes %}
{{ klass.id }}
{% endfor %}
{% endif %}
{% endblock %}
{% block functions %}
{% if visible_functions %}
Functions
~~~~~~~~~
.. autoapisummary::
{% for function in visible_functions %}
{{ function.id }}
{% endfor %}
{% endif %}
{% endblock %}
{% endif %}
{% for obj_item in visible_children %}
{{ obj_item.rendered|indent(0) }}
{% endfor %}
{% endif %}
{% endblock %}

View File

@ -78,9 +78,7 @@ STATIC NORETURN void math_error(void) {
// 1.442695040888963407354163704 is 1/_M_LN2
#define log2(x) (log(x) * 1.442695040888963407354163704)
#endif
//| Constants
//| ---------
//|
//| e: Any = ...
//| """base of the natural logarithm"""
//|
@ -88,9 +86,6 @@ STATIC NORETURN void math_error(void) {
//| """the ratio of a circle's circumference to its diameter"""
//|
//| Functions
//| ---------
//|
//| def acos(x: Any) -> Any:
//| """Return the inverse cosine of ``x``."""
//| ...
@ -181,7 +176,7 @@ STATIC NORETURN void math_error(void) {
//| """Returns the square root of ``x``."""
//| ...
//|
//| def tan(x: Any) -> Any: ...
//| def tan(x: Any) -> Any:
//| """Return the tangent of ``x``."""
//| ...
//|
@ -195,61 +190,58 @@ MATH_FUN_2(pow, pow)
MATH_FUN_1(exp, exp)
#if MICROPY_PY_MATH_SPECIAL_FUNCTIONS
// Special functions
// -----------------
//
// .. function:: expm1(x)
//
// Return ``exp(x) - 1``.
//
//| def expm1(x):
//| """Return ``exp(x) - 1``."""
//| ...
//|
MATH_FUN_1(expm1, expm1)
// .. function:: log2(x)
//
// Return the base-2 logarithm of ``x``.
//
//| def log2(x):
//| """Return the base-2 logarithm of ``x``."""
//| ...
//|
MATH_FUN_1_ERRCOND(log2, log2, (x <= (mp_float_t)0.0))
// .. function:: log10(x)
//
// Return the base-10 logarithm of ``x``.
//
//| def log10(x):
//| """Return the base-10 logarithm of ``x``."""
//| ...
//|
MATH_FUN_1_ERRCOND(log10, log10, (x <= (mp_float_t)0.0))
// .. function:: cosh(x)
//
// Return the hyperbolic cosine of ``x``.
//
//| def cosh(x):
//| """Return the hyperbolic cosine of ``x``."""
//| ...
//|
MATH_FUN_1(cosh, cosh)
// .. function:: sinh(x)
//
// Return the hyperbolic sine of ``x``.
//
//| def sinh(x):
//| """Return the hyperbolic sine of ``x``."""
//| ...
//|
MATH_FUN_1(sinh, sinh)
// .. function:: tanh(x)
//
// Return the hyperbolic tangent of ``x``.
//
//| def tanh(x):
//| """Return the hyperbolic tangent of ``x``."""
//| ...
//|
MATH_FUN_1(tanh, tanh)
// .. function:: acosh(x)
//
// Return the inverse hyperbolic cosine of ``x``.
//
//| def acosh(x):
//| """Return the inverse hyperbolic cosine of ``x``."""
//| ...
//|
MATH_FUN_1(acosh, acosh)
// .. function:: asinh(x)
//
// Return the inverse hyperbolic sine of ``x``.
//
//| def asinh(x):
//| """Return the inverse hyperbolic sine of ``x``."""
//| ...
//|
MATH_FUN_1(asinh, asinh)
// .. function:: atanh(x)
//
// Return the inverse hyperbolic tangent of ``x``.
//
//| def atanh(x):
//| """Return the inverse hyperbolic tangent of ``x``."""
//| ...
//|
MATH_FUN_1(atanh, atanh)
#endif
@ -288,28 +280,28 @@ MATH_FUN_1_TO_INT(trunc, trunc)
MATH_FUN_2(ldexp, ldexp)
#if MICROPY_PY_MATH_SPECIAL_FUNCTIONS
// .. function:: erf(x)
//
// Return the error function of ``x``.
//
//| def erf(x):
//| """Return the error function of ``x``."""
//| ...
//|
MATH_FUN_1(erf, erf)
// .. function:: erfc(x)
//
// Return the complementary error function of ``x``.
//
//| def erfc(x):
//| """Return the complementary error function of ``x``."""
//| ...
//|
MATH_FUN_1(erfc, erfc)
// .. function:: gamma(x)
//
// Return the gamma function of ``x``.
//
//| def gamma(x):
//| """Return the gamma function of ``x``."""
//| ...
//|
MATH_FUN_1(gamma, tgamma)
// .. function:: lgamma(x)
//
// Return the natural logarithm of the gamma function of ``x``.
//
//| def lgamma(x):
//| """Return the natural logarithm of the gamma function of ``x``."""
//| ...
//|
MATH_FUN_1(lgamma, lgamma)
#endif
//TODO: factorial, fsum

View File

@ -1,9 +1,4 @@
:mod:`ulab` --- Manipulate numeric data similar to numpy
========================================================
.. module:: ulab
:synopsis: Manipulate numeric data similar to numpy
"""Manipulate numeric data similar to numpy
`ulab` is a numpy-like module for micropython, meant to simplify and
speed up common mathematical operations on arrays. The primary goal was to
@ -17,20 +12,13 @@ https://micropython-ulab.readthedocs.io/en/latest/
`ulab` is modeled after numpy, and aims to be a compatible subset where
possible. Numpy's documentation can be found at
https://docs.scipy.org/doc/numpy/index.html
https://docs.scipy.org/doc/numpy/index.html"""
.. contents::
.. attribute:: __version__
The closest corresponding version of micropython-ulab
ulab.array -- 1- and 2- dimensional array
-----------------------------------------
.. class:: ulab.array(values, \*, dtype=float)
:param sequence values: Sequence giving the initial content of the array.
class array:
"""1- and 2- dimensional array"""
def __init__(self, values, *, dtype=float):
""":param sequence values: Sequence giving the initial content of the array.
:param dtype: The type of array values, ``int8``, ``uint8``, ``int16``, ``uint16``, or ``float``
The `values` sequence can either be another ~ulab.array, sequence of numbers
@ -45,127 +33,116 @@ ulab.array -- 1- and 2- dimensional array
like `zeros` or `linspace`.
`ulab.array` implements the buffer protocol, so it can be used in many
places an `array.array` can be used.
places an `array.array` can be used."""
...
.. attribute:: shape
shape: tuple = ...
"""The size of the array, a tuple of length 1 or 2"""
The size of the array, a tuple of length 1 or 2
size: int = ...
"""The number of elements in the array"""
.. attribute:: size
itemsize: int = ...
"""The number of elements in the array"""
The number of elements in the array
.. attribute:: itemsize
The number of elements in the array
.. method:: flatten(\*, order='C')
:param order: Whether to flatten by rows ('C') or columns ('F')
def flatten(self, *, order='C'):
""":param order: Whether to flatten by rows ('C') or columns ('F')
Returns a new `ulab.array` object which is always 1 dimensional.
If order is 'C' (the default", then the data is ordered in rows;
If it is 'F', then the data is ordered in columns. "C" and "F" refer
to the typical storage organization of the C and Fortran languages.
to the typical storage organization of the C and Fortran languages."""
...
.. method:: sort(\*, axis=1)
def sort(self, *, axis=1):
""":param axis: Whether to sort elements within rows (0), columns (1), or elements (None)"""
...
:param axis: Whether to sort elements within rows (0), columns (1), or elements (None)
def transpose(self):
"""Swap the rows and columns of a 2-dimensional array"""
...
.. method:: transpose()
def __add__(self):
"""Adds corresponding elements of the two arrays, or adds a number to all
elements of the array. If both arguments are arrays, their sizes must match."""
...
Swap the rows and columns of a 2-dimensional array
def __sub__(self):
"""Subtracts corresponding elements of the two arrays, or adds a number to all
elements of the array. If both arguments are arrays, their sizes must match."""
...
.. method:: __add__()
Adds corresponding elements of the two arrays, or adds a number to all
elements of the array. If both arguments are arrays, their sizes must match.
.. method:: __sub__()
Subtracts corresponding elements of the two arrays, or adds a number to all
elements of the array. If both arguments are arrays, their sizes must match.
.. method:: __mul__()
Multiplies corresponding elements of the two arrays, or multiplies
def __mul__(self):
"""Multiplies corresponding elements of the two arrays, or multiplies
all elements of the array by a number. If both arguments are arrays,
their sizes must match.
their sizes must match."""
...
.. method:: __div__()
Multiplies corresponding elements of the two arrays, or divides
def __div__(self):
"""Multiplies corresponding elements of the two arrays, or divides
all elements of the array by a number. If both arguments are arrays,
their sizes must match.
their sizes must match."""
...
.. method:: __pow__()
Computes the power (x**y) of corresponding elements of the the two arrays,
def __pow__():
"""Computes the power (x**y) of corresponding elements of the the two arrays,
or one number and one array. If both arguments are arrays, their sizes
must match.
must match."""
...
.. method:: __getitem__()
def __getitem__():
"""Retrieve an element of the array."""
...
Retrieve an element of the array.
def __setitem__():
"""Set an element of the array."""
...
.. method:: __setitem__()
int8 = ...
"""Type code for signed integers in the range -128 .. 127 inclusive, like the 'b' typecode of `array.array`"""
Set an element of the array.
int16 = ...
"""Type code for signed integers in the range -32768 .. 32767 inclusive, like the 'h' typecode of `array.array`"""
Array type codes
----------------
.. attribute:: int8
float = ...
"""Type code for floating point values, like the 'f' typecode of `array.array`"""
Type code for signed integers in the range -128 .. 127 inclusive, like the 'b' typecode of `array.array`
uint8 = ...
"""Type code for unsigned integers in the range 0 .. 255 inclusive, like the 'H' typecode of `array.array`"""
.. attribute:: int16
Type code for signed integers in the range -32768 .. 32767 inclusive, like the 'h' typecode of `array.array`
.. attribute:: float
Type code for floating point values, like the 'f' typecode of `array.array`
.. attribute:: uint8
Type code for unsigned integers in the range 0 .. 255 inclusive, like the 'H' typecode of `array.array`
.. attribute:: uint16
Type code for unsigned integers in the range 0 .. 65535 inclusive, like the 'h' typecode of `array.array`
Basic Array defining functions
------------------------------
.. method:: ones(shape, \*, dtype=float)
uint16 = ...
"""Type code for unsigned integers in the range 0 .. 65535 inclusive, like the 'h' typecode of `array.array`"""
def ones(shape, *, dtype=float):
"""
.. param: shape
Shape of the array, either an integer (for a 1-D array) or a tuple of 2 integers (for a 2-D array)
.. param: dtype
Type of values in the array
Return a new array of the given shape with all elements set to 1.
.. method:: zeros
Return a new array of the given shape with all elements set to 1."""
...
def zeros(shape, *, dtype):
"""
.. param: shape
Shape of the array, either an integer (for a 1-D array) or a tuple of 2 integers (for a 2-D array)
.. param: dtype
Type of values in the array
Return a new array of the given shape with all elements set to 0.
Return a new array of the given shape with all elements set to 0."""
...
.. method:: eye(size, \*, dtype=float)
Return a new square array of size, with the diagonal elements set to 1
and the other elements set to 0.
.. method:: linspace(start, stop, \*, dtype=float, num=50, endpoint=True)
def eye(size, *, dtype=float):
"""Return a new square array of size, with the diagonal elements set to 1
and the other elements set to 0."""
...
def linspace(start, stop, *, dtype=float, num=50, endpoint=True):
"""
.. param: start
First value in the array
@ -188,336 +165,5 @@ Basic Array defining functions
endpoint=True, the exact ``stop`` value may not be included due to the
inaccuracy of floating point arithmetic.
Return a new 1-D array with ``num`` elements ranging from ``start`` to ``stop`` linearly.
:mod:`ulab.compare` --- Comparison functions
============================================
.. module::ulab.compare
.. method:: clip(x1, x2, x3)
Constrain the values from ``x1`` to be between ``x2`` and ``x3``.
``x2`` is assumed to be less than or equal to ``x3``.
Arguments may be ulab arrays or numbers. All array arguments
must be the same size. If the inputs are all scalars, a 1-element
array is returned.
Shorthand for ``ulab.maximum(x2, ulab.minimum(x1, x3))``
.. method:: maximum(x1, x2)
Compute the element by element maximum of the arguments.
Arguments may be ulab arrays or numbers. All array arguments
must be the same size. If the inputs are both scalars, a number is
returned
.. method:: minimum(x1, x2)
Compute the element by element minimum of the arguments.
Arguments may be ulab arrays or numbers. All array arguments
must be the same size. If the inputs are both scalars, a number is
returned
:mod:`ulab.vector` --- Element-by-element functions
===================================================
.. module:: ulab.vector
These functions can operate on numbers, 1-D arrays, or 2-D arrays by
applying the function to every element in the array. This is typically
much more efficient than expressing the same operation as a Python loop.
.. method:: acos
Computes the inverse cosine function
.. method:: acosh
Computes the inverse hyperbolic cosine function
.. method:: asin
Computes the inverse sine function
.. method:: asinh
Computes the inverse hyperbolic sine function
.. method:: around(a, \*, decimals)
Returns a new float array in which each element is rounded to
``decimals`` places.
.. method:: atan
Computes the inverse tangent function; the return values are in the
range [-pi/2,pi/2].
.. method:: atan2(y,x)
Computes the inverse tangent function of y/x; the return values are in
the range [-pi, pi].
.. method:: atanh
Computes the inverse hyperbolic tangent function
.. method:: ceil
Rounds numbers up to the next whole number
.. method:: cos
Computes the cosine function
.. method:: erf
Computes the error function, which has applications in statistics
.. method:: erfc
Computes the complementary error function, which has applications in statistics
.. method:: exp
Computes the exponent function.
.. method:: expm1
Computes $e^x-1$. In certain applications, using this function preserves numeric accuracy better than the `exp` function.
.. method:: floor
Rounds numbers up to the next whole number
.. method:: gamma
Computes the gamma function
.. method:: lgamma
Computes the natural log of the gamma function
.. method:: log
Computes the natural log
.. method:: log10
Computes the log base 10
.. method:: log2
Computes the log base 2
.. method:: sin
Computes the sine
.. method:: sinh
Computes the hyperbolic sine
.. method:: sqrt
Computes the square root
.. method:: tan
Computes the tangent
.. method:: tanh
Computes the hyperbolic tangent
:mod:`ulab.linalg` - Linear algebra functions
=============================================
.. module:: ulab.linalg
.. method:: cholesky(A)
:param ~ulab.array A: a positive definite, symmetric square matrix
:return ~ulab.array L: a square root matrix in the lower triangular form
:raises ValueError: If the input does not fulfill the necessary conditions
The returned matrix satisfies the equation m=LL*
.. method:: det
:param: m, a square matrix
:return float: The determinant of the matrix
Computes the eigenvalues and eigenvectors of a square matrix
.. method:: dot(m1, m2)
:param ~ulab.array m1: a matrix
:param ~ulab.array m2: a matrix
Computes the matrix product of two matrices
**WARNING:** Unlike ``numpy``, this function cannot be used to compute the dot product of two vectors
.. method:: eig(m)
:param m: a square matrix
:return tuple (eigenvectors, eigenvalues):
Computes the eigenvalues and eigenvectors of a square matrix
.. method:: inv(m)
:param ~ulab.array m: a square matrix
:return: The inverse of the matrix, if it exists
:raises ValueError: if the matrix is not invertible
Computes the inverse of a square matrix
.. method:: size(array)
Return the total number of elements in the array, as an integer.
.. method:: trace(m)
:param m: a square matrix
Compute the trace of the matrix, the sum of its diagonal elements.
:mod:`ulab.filter` --- Filtering functions
==========================================
.. module:: ulab.filter
.. method:: convolve(r, c=None)
:param ulab.array a:
:param ulab.array v:
Returns the discrete, linear convolution of two one-dimensional sequences.
The result is always an array of float. Only the ``full`` mode is supported,
and the ``mode`` named parameter of numpy is not accepted. Note that all other
modes can be had by slicing a ``full`` result.
Convolution filters can implement high pass, low pass, band pass, etc.,
filtering operations. Convolution filters are typically constructed ahead
of time. This can be done using desktop python with scipy, or on web pages
such as https://fiiir.com/
Convolution is most time-efficient when both inputs are of float type.
:mod:`ulab.fft` --- Frequency-domain functions
==============================================
.. module:: ulab.fft
.. method:: fft(r, c=None)
:param ulab.array r: A 1-dimension array of values whose size is a power of 2
:param ulab.array c: An optional 1-dimension array of values whose size is a power of 2, giving the complex part of the value
:return tuple (r, c): The real and complex parts of the FFT
Perform a Fast Fourier Transform from the time domain into the frequency domain
See also ~ulab.extras.spectrum, which computes the magnitude of the fft,
rather than separately returning its real and imaginary parts.
.. method:: ifft(r, c=None)
:param ulab.array r: A 1-dimension array of values whose size is a power of 2
:param ulab.array c: An optional 1-dimension array of values whose size is a power of 2, giving the complex part of the value
:return tuple (r, c): The real and complex parts of the inverse FFT
Perform an Inverse Fast Fourier Transform from the frequeny domain into the time domain
:mod:`ulab.numerical` --- Numerical and Statistical functions
=============================================================
.. module:: ulab.numerical
Most of these functions take an "axis" argument, which indicates whether to
operate over the flattened array (None), rows (0), or columns (1).
.. method:: argmax(array, \*, axis=None)
Return the index of the maximum element of the 1D array
.. method:: argmin(array, \*, axis=None)
Return the index of the minimum element of the 1D array
.. method:: argsort(array, \*, axis=None)
Returns an array which gives indices into the input array from least to greatest.
.. method:: diff(array, \*, axis=1)
Return the numerical derivative of successive elements of the array, as
an array. axis=None is not supported.
.. method:: flip(array, \*, axis=None)
Returns a new array that reverses the order of the elements along the
given axis, or along all axes if axis is None.
.. method:: max(array, \*, axis=None)
Return the maximum element of the 1D array
.. method:: mean(array, \*, axis=None)
Return the mean element of the 1D array, as a number if axis is None, otherwise as an array.
.. method:: min(array, \*, axis=None)
Return the minimum element of the 1D array
.. method:: roll(array, distance, \*, axis=None)
Shift the content of a vector by the positions given as the second
argument. If the ``axis`` keyword is supplied, the shift is applied to
the given axis. The array is modified in place.
.. method:: std(array, \*, axis=None)
Return the standard deviation of the array, as a number if axis is None, otherwise as an array.
.. method:: sum(array, \*, axis=None)
Return the sum of the array, as a number if axis is None, otherwise as an array.
.. method:: sort(array, \*, axis=0)
Sort the array along the given axis, or along all axes if axis is None.
The array is modified in place.
:mod:`ulab.poly` --- Polynomial functions
=========================================
.. module:: ulab.poly
.. method:: polyfit([x, ] y, degree)
Return a polynomial of given degree that approximates the function
f(x)=y. If x is not supplied, it is the range(len(y)).
.. method:: polyval(p, x)
Evaluate the polynomial p at the points x. x must be an array.
:mod:`ulab.extras` --- Additional functions not in numpy
========================================================
.. method:: spectrum(r):
:param ulab.array r: A 1-dimension array of values whose size is a power of 2
Computes the spectrum of the input signal. This is the absolute value of the (complex-valued) fft of the signal.
This function is similar to scipy's ``scipy.signal.spectrogram``.
Return a new 1-D array with ``num`` elements ranging from ``start`` to ``stop`` linearly."""
...

View File

@ -0,0 +1,30 @@
"""Comparison functions"""
def clip(x1, x2, x3):
"""
Constrain the values from ``x1`` to be between ``x2`` and ``x3``.
``x2`` is assumed to be less than or equal to ``x3``.
Arguments may be ulab arrays or numbers. All array arguments
must be the same size. If the inputs are all scalars, a 1-element
array is returned.
Shorthand for ``ulab.maximum(x2, ulab.minimum(x1, x3))``"""
...
def maximum(x1, x2):
"""
Compute the element by element maximum of the arguments.
Arguments may be ulab arrays or numbers. All array arguments
must be the same size. If the inputs are both scalars, a number is
returned"""
...
def minimum(x1, x2):
"""Compute the element by element minimum of the arguments.
Arguments may be ulab arrays or numbers. All array arguments
must be the same size. If the inputs are both scalars, a number is
returned"""
...

View File

@ -0,0 +1,10 @@
"""Additional functions not in numpy"""
def spectrum(r):
"""
:param ulab.array r: A 1-dimension array of values whose size is a power of 2
Computes the spectrum of the input signal. This is the absolute value of the (complex-valued) fft of the signal.
This function is similar to scipy's ``scipy.signal.spectrogram``."""
...

View File

@ -0,0 +1,22 @@
"""Frequency-domain functions"""
def fft(r, c=None):
"""
:param ulab.array r: A 1-dimension array of values whose size is a power of 2
:param ulab.array c: An optional 1-dimension array of values whose size is a power of 2, giving the complex part of the value
:return tuple (r, c): The real and complex parts of the FFT
Perform a Fast Fourier Transform from the time domain into the frequency domain
See also ~ulab.extras.spectrum, which computes the magnitude of the fft,
rather than separately returning its real and imaginary parts."""
...
def ifft(r, c=None):
"""
:param ulab.array r: A 1-dimension array of values whose size is a power of 2
:param ulab.array c: An optional 1-dimension array of values whose size is a power of 2, giving the complex part of the value
:return tuple (r, c): The real and complex parts of the inverse FFT
Perform an Inverse Fast Fourier Transform from the frequeny domain into the time domain"""
...

View File

@ -0,0 +1,19 @@
"""Filtering functions"""
def convolve(r, c=None):
"""
:param ulab.array a:
:param ulab.array v:
Returns the discrete, linear convolution of two one-dimensional sequences.
The result is always an array of float. Only the ``full`` mode is supported,
and the ``mode`` named parameter of numpy is not accepted. Note that all other
modes can be had by slicing a ``full`` result.
Convolution filters can implement high pass, low pass, band pass, etc.,
filtering operations. Convolution filters are typically constructed ahead
of time. This can be done using desktop python with scipy, or on web pages
such as https://fiiir.com/
Convolution is most time-efficient when both inputs are of float type."""
...

View File

@ -0,0 +1,57 @@
"""Linear algebra functions"""
def cholesky(A):
"""
:param ~ulab.array A: a positive definite, symmetric square matrix
:return ~ulab.array L: a square root matrix in the lower triangular form
:raises ValueError: If the input does not fulfill the necessary conditions
The returned matrix satisfies the equation m=LL*"""
...
def det():
"""
:param: m, a square matrix
:return float: The determinant of the matrix
Computes the eigenvalues and eigenvectors of a square matrix"""
...
def dot(m1, m2):
"""
:param ~ulab.array m1: a matrix
:param ~ulab.array m2: a matrix
Computes the matrix product of two matrices
**WARNING:** Unlike ``numpy``, this function cannot be used to compute the dot product of two vectors"""
...
def eig(m):
"""
:param m: a square matrix
:return tuple (eigenvectors, eigenvalues):
Computes the eigenvalues and eigenvectors of a square matrix"""
...
def inv(m):
"""
:param ~ulab.array m: a square matrix
:return: The inverse of the matrix, if it exists
:raises ValueError: if the matrix is not invertible
Computes the inverse of a square matrix"""
...
def size(array):
"""Return the total number of elements in the array, as an integer."""
...
def trace(m):
"""
:param m: a square matrix
Compute the trace of the matrix, the sum of its diagonal elements."""
...

View File

@ -0,0 +1,57 @@
"""Numerical and Statistical functions
Most of these functions take an "axis" argument, which indicates whether to
operate over the flattened array (None), rows (0), or columns (1)."""
def argmax(array, *, axis=None):
"""Return the index of the maximum element of the 1D array"""
...
def argmin(array, *, axis=None):
"""Return the index of the minimum element of the 1D array"""
...
def argsort(array, *, axis=None):
"""Returns an array which gives indices into the input array from least to greatest."""
...
def diff(array, *, axis=1):
"""Return the numerical derivative of successive elements of the array, as
an array. axis=None is not supported."""
...
def flip(array, *, axis=None):
"""Returns a new array that reverses the order of the elements along the
given axis, or along all axes if axis is None."""
...
def max(array, *, axis=None):
"""Return the maximum element of the 1D array"""
...
def mean(array, *, axis=None):
"""Return the mean element of the 1D array, as a number if axis is None, otherwise as an array."""
...
def min(array, *, axis=None):
"""Return the minimum element of the 1D array"""
...
def roll(array, distance, *, axis=None):
"""Shift the content of a vector by the positions given as the second
argument. If the ``axis`` keyword is supplied, the shift is applied to
the given axis. The array is modified in place."""
...
def std(array, *, axis=None):
"""Return the standard deviation of the array, as a number if axis is None, otherwise as an array."""
...
def sum(array, *, axis=None):
"""Return the sum of the array, as a number if axis is None, otherwise as an array."""
...
def sort(array, *, axis=0):
"""Sort the array along the given axis, or along all axes if axis is None.
The array is modified in place."""
...

View File

@ -0,0 +1,10 @@
"""Polynomial functions"""
def polyfit(x, y, degree):
"""Return a polynomial of given degree that approximates the function
f(x)=y. If x is not supplied, it is the range(len(y))."""
...
def polyval(p, x):
"""Evaluate the polynomial p at the points x. x must be an array."""
...

View File

@ -0,0 +1,108 @@
"""Element-by-element functions
These functions can operate on numbers, 1-D arrays, or 2-D arrays by
applying the function to every element in the array. This is typically
much more efficient than expressing the same operation as a Python loop."""
def acos():
"""Computes the inverse cosine function"""
...
def acosh():
"""Computes the inverse hyperbolic cosine function"""
...
def asin():
"""Computes the inverse sine function"""
...
def asinh():
"""Computes the inverse hyperbolic sine function"""
...
def around(a, *, decimals):
"""Returns a new float array in which each element is rounded to
``decimals`` places."""
...
def atan():
"""Computes the inverse tangent function; the return values are in the
range [-pi/2,pi/2]."""
...
def atan2(y,x):
"""Computes the inverse tangent function of y/x; the return values are in
the range [-pi, pi]."""
...
def atanh():
"""Computes the inverse hyperbolic tangent function"""
...
def ceil():
"""Rounds numbers up to the next whole number"""
...
def cos():
"""Computes the cosine function"""
...
def erf():
"""Computes the error function, which has applications in statistics"""
...
def erfc():
"""Computes the complementary error function, which has applications in statistics"""
...
def exp():
"""Computes the exponent function."""
...
def expm1():
"""Computes $e^x-1$. In certain applications, using this function preserves numeric accuracy better than the `exp` function."""
...
def floor():
"""Rounds numbers up to the next whole number"""
...
def gamma():
"""Computes the gamma function"""
...
def lgamma():
"""Computes the natural log of the gamma function"""
...
def log():
"""Computes the natural log"""
...
def log10():
"""Computes the log base 10"""
...
def log2():
"""Computes the log base 2"""
...
def sin():
"""Computes the sine"""
...
def sinh():
"""Computes the hyperbolic sine"""
...
def sqrt():
"""Computes the square root"""
...
def tan():
"""Computes the tangent"""
...
def tanh():
"""Computes the hyperbolic tangent"""
...

View File

@ -6,27 +6,21 @@ import traceback
top_level = sys.argv[1].strip("/")
stub_directory = sys.argv[2]
if top_level.count("/") == 1:
top_level, module = top_level.split("/")
modules = [module]
else:
modules = os.listdir(top_level)
modules = sorted(modules)
def convert_folder(top_level, stub_directory):
print(top_level, stub_directory)
ok = 0
total = 0
for module in modules:
module_path = os.path.join(top_level, module)
if not os.path.isdir(module_path):
continue
filenames = sorted(os.listdir(top_level))
pyi_lines = []
classes = os.listdir(module_path)
classes = [x for x in sorted(classes) if x.endswith(".c")]
if classes and classes[-1] == "__init__.c":
classes.insert(0, classes.pop())
for class_file in classes:
class_path = os.path.join(module_path, class_file)
with open(class_path, "r") as f:
for filename in filenames:
full_path = os.path.join(top_level, filename)
file_lines = []
if os.path.isdir(full_path):
mok, mtotal = convert_folder(full_path, os.path.join(stub_directory, filename))
ok += mok
total += mtotal
elif filename.endswith(".c"):
with open(full_path, "r") as f:
for line in f:
if line.startswith("//|"):
if line[3] == " ":
@ -35,18 +29,24 @@ for module in modules:
line = line[3:]
else:
continue
pyi_lines.append(line)
file_lines.append(line)
elif filename.endswith(".pyi"):
with open(full_path, "r") as f:
file_lines.extend(f.readlines())
raw_stubs = [x for x in sorted(classes) if x.endswith(".pyi")]
if raw_stubs and raw_stubs[-1] == "__init__.pyi":
raw_stubs.insert(0, raw_stubs.pop())
for raw_stub in raw_stubs:
raw_stub_path = os.path.join(module_path, raw_stub)
with open(raw_stub_path, "r") as f:
pyi_lines.extend(f.readlines())
stub_filename = os.path.join(stub_directory, module + ".pyi")
# Always put the contents from an __init__ first.
if filename.startswith("__init__."):
pyi_lines = file_lines + pyi_lines
else:
pyi_lines.extend(file_lines)
if not pyi_lines:
return ok, total
stub_filename = os.path.join(stub_directory, "__init__.pyi")
print(stub_filename)
stub_contents = "".join(pyi_lines)
os.makedirs(stub_directory, exist_ok=True)
with open(stub_filename, "w") as f:
f.write(stub_contents)
@ -59,6 +59,9 @@ for module in modules:
e = e.__cause__
traceback.print_exception(type(e), e, e.__traceback__)
print()
return ok, total
ok, total = convert_folder(top_level, stub_directory)
print(f"{ok} ok out of {total}")