micropython-samples/parse2d/README.md

140 wiersze
4.7 KiB
Markdown

# Access a list or array as a 2D array
The `parse2d` module provides a generator function `do_args`. This enables a
generator to be instantiated which maps one or two dimensional address args
onto index values. These can be used to address any object which can be
represented as a one dimensional array including arrays, bytearrays, lists or
random access files. The aim is to simplify writing classes that implement 2D
arrays of objects - specifically writing `__getitem__` and `__setitem__`
methods whose addressing modes conform with Python conventions.
Addressing modes include slice objects; sets of elements can be accessed or
populated without explicit iteration. Thus, if `demo` is an instance of a user
defined class, the following access modes might be permitted:
```python
demo = MyClass(10, 10) # Create an object that can be viewed as 10 rows * 10 cols
demo[8, 8] = 42 # Populate a single cell
demo[0:, 0] = iter((66, 77, 88)) # Populate a column
demo[2:5, 2:5] = iter(range(50, 60)) # Populate a rectangular region.
print(sum(demo[0, 0:])) # Sum a row
for row, _ in enumerate(demo[0:, 0]):
print(*demo[row, 0:]) # Size- and shape-agnostic print
```
The object can also be accessed as a 1D list. An application can mix and match
1D and 2D addressing as required:
```python
demo[-1] = 999 # Set last element
demo[-10: -1] = 7 # Set previous 9 elements
```
The focus of this module is convenience, minimising iteration and avoiding
error-prone address calculations. It is not a high performance solution. The
module resulted from guidance provided in
[this discussion](https://github.com/orgs/micropython/discussions/11611).
###### [Main README](../README.md)
# The do_args generator function
This takes the following args:
* `args` A 1- or 2-tuple. In the case of a 1-tuple (1D access) the element is
an int or slice object. In the 2-tuple (2D) case each element can be an int or
a slice.
* `nrows` No. of rows in the array.
* `ncols` No. of columns.
This facilitates the design of `__getitem__` and `__setitem__`, e.g.
```python
def __getitem__(self, *args):
indices = do_args(args, self.nrows, self.ncols)
for i in indices:
yield self.cells[i]
```
The generator is agnostic of the meaning of the first and second args: the
mathematical `[x, y]` or the graphics `[row, col]` conventions may be applied.
Index values are `row * ncols + col` or `x * ncols + y` as shown by the
following which must be run on CPython:
```python
>>> g = do_args(((1, slice(0, 9)),), 10, 10)
>>> for index in g: print(f"{index} ", end = "")
10 11 12 13 14 15 16 17 18 >>>
```
Three argument slices are supported:
```python
>>> g = do_args(((1, slice(8, 0, -2)),), 10, 10)
>>> for index in g: print(f"{index} ", end = "")
...
18 16 14 12 >>>
```
# Addressing
The module aims to conform with Python rules. Thus, if `demo` is an instance of
a class representing a 10x10 array,
```python
print(demo[0, 10])
```
will produce an `IndexError: Index out of range`. By contrast
```python
print(demo[0, 0:100])
```
will print row 0 (columns 0-9)without error. This is analogous to Python's
behaviour with list addressing:
```python
>>> l = [0] * 10
>>> l[10]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
IndexError: list index out of range
>>> l[0:100]
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
>>>
```
# Class design
## __getitem__()
RAM usage can be minimised by having this return an iterator. This enables
usage such as `sum(arr[0:, 1])` and minimises RAM allocation.
## __setitem__()
The semantics of the right hand side of assignment expressions is defined in
the user class. The following uses a `list` as the addressable object.
```python
from parse2d import do_args
class MyIntegerArray:
def __init__(self, nrows, ncols):
self.nrows = nrows
self.ncols = ncols
self.cells = [0] * nrows * ncols
def __getitem__(self, *args):
indices = do_args(args, self.nrows, self.ncols)
for i in indices:
yield self.cells[i]
def __setitem__(self, *args):
value = args[1]
indices = do_args(args[: -1], self.nrows, self.ncols)
for i in indices:
self.cells[i] = value
```
The `__setitem__` method is minimal. In a practical class `value` might be a
`list`, `tuple` or an object supporting the iterator protocol.
# The int2D demo
RHS semantics differ from Python `list` practice in that you can populate an
entire row with
```python
demo[0:, 0] = iter((66, 77, 88))
```
If the iterator runs out of data, the last item is repeated. Equally you could
have
```python
demo[0:, 0] = 4
```
The demo also illustrates the case where `__getitem__` returns an iterator.
###### [Main README](../README.md)