summaryrefslogtreecommitdiffhomepage
path: root/docs/cspan_api.md
blob: b9879a6f8920209d021caa3810bec1a4ac78a06c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
# STC [cspan](../include/stc/cspan.h): Multi-dimensional Array View
![Array](pics/array.jpg)

The **cspan** is templated non-owning multi-dimensional view of an array. 

See the c++ class [std::mdspan](https://en.cppreference.com/w/cpp/container/mdspan) for a similar functionality.

## Header file and declaration

```c
#include <stc/cspan.h>
using_cspan(SpanType, ValueType, Rank);     // define SpanType with ValueType elements.
                                            // Rank is number of dimensions (max 4)
// Shorthands:
using_cspan2(S, ValueType);                 // define span types S1, S2 with Ranks 1, 2.
using_cspan3(S, ValueType);                 // define span types S1, S2, S3 with Ranks 1, 2, 3.
using_cspan4(S, ValueType);                 // define span types S1.., S4 with Ranks 1, 2, 3, 4.
```
## Methods
Note that `cspan_make()`, `cmake_from*()`, `cspan_sliceX()` `and cspan_XtoY()` require a (safe) cast to its span-type
on assignment, but not on initialization of a span variable.
```c
SpanType        cspan_make(ValueType* data,  size_t xdim, ...);             // make N-dimensional cspan
SpanType        cspan_from(STCContainer* cnt);                              // create a 1D cspan from a compatible STC container
SpanType        cspan_from_array(ValueType array[]);                        // create a 1D cspan from a C array
SpanType        cspan_from_list(T ValueType, {val0, val1, ...});            // create a 1D cspan from an initializer list
SpanType&       cspan_literal(T SpanType, {val0, val1, ...});               // create a 1D cspan compound literal from init list 

void            cspan_resize(SpanType* self, size_t xdim, ...);             // change the extent of each dimension

size_t          cspan_size(SpanType spn);                                   // return number of elements
unsigned        cspan_rank(SpanType spn);                                   // return number of dimensions
ValueType*      cspan_at(SpanType spn, size_t x, ...);                      // access element
size_t          cspan_index(SpanType spn, size_t x, ...);                   // index of element

SpanType1       cspan_subspan1(SpanType1 spn, size_t offset, size_t count); // get a slice of a 1D cspan
SpanType2       cspan_subspan2(SpanType2 spn, size_t offset, size_t count); // get a slice of a 2D cspan
SpanType3       cspan_subspan3(SpanType3 spn, size_t offset, size_t count); // get a slice of a 3D cspan
SpanType4       cspan_subspan4(SpanType4 spn, size_t offset, size_t count); // get a slice of a 4D cspan

SpanType1       cspan_2to1(SpanType2 spn, size_t x);                        // return a 1D subspan
SpanType1       cspan_3to1(SpanType3 spn, size_t x, size_t y);              // return a 1D subspan
SpanType2       cspan_3to2(SpanType3 spn, size_t x);                        // return a 2D subspan
SpanType1       cspan_4to1(SpanType4 spn, size_t x, size_t y, size_t z);    // return a 1D subspan
SpanType2       cspan_4to2(SpanType4 spn, size_t x, size_t y);              // return a 2D subspan
SpanType3       cspan_4to3(SpanType4 spn, size_t x);                        // return a 3D subspan

SpanType_iter   SpanType_begin(const SpanType* self);
SpanType_iter   SpanType_end(const SpanType* self);
void            SpanType_next(SpanType_iter* it);
```
## Types

| Type name        | Type definition                                      | Used to represent... |
|:-----------------|:-----------------------------------------------------|:---------------------|
| SpanType         | `struct { ValueType *data; uint32_t dim[RANK]; }`    | The SpanType         |
| SpanType`_value` | `ValueType`                                          | The ValueType        |
| SpanType`_iter`  | `struct { ValueType *ref; ... }`                     | Iterator type        |

## Example
```c
#include <stdio.h>
#define i_val float
#include <stc/cstack.h>

#include <stc/cspan.h>
using_cspan3(FS, float); // Shorthand to define span types FS1, FS2, and FS3.

int main()
{
    int xd = 6, yd = 4, zd = 3;
    c_AUTO (cstack_float, vec) {
        c_FORRANGE (i, xd*yd*zd)
            cstack_float_push(&vec, i);

        // define arr[xd][yd][zd] cspan
        FS3 span3 = cspan_make(vec.data, xd, yd, zd);
        *cspan_at(span3, 4, 3, 2) = 3.14f;
        printf("index: %d", (int)cspan_index(span3, 4, 3, 2));

        FS1 span1 = cspan_3to1(span3, 4, 3);
        printf("\niterate span1: ");
        c_FOREACH (i, FS1, span1)
            printf("%g ", *i.ref);

        FS2 span2 = cspan_3to2(span3, 4);
        printf("\niterate span2: ");
        c_FOREACH (i, FS2, span2)
            printf("%g ", *i.ref);

        puts("\niterate span3 by dimensions:");
        c_FORRANGE (i, span3.dim[0]) {
            c_FORRANGE (j, span3.dim[1]) {
                c_FORRANGE (k, span3.dim[2])
                    printf(" %g", *cspan_at(span3, i, j, k));
                printf(" |");
            }
            puts("");
        }

        printf("%g\n", *cspan_at(span3, 4, 3, 2));
        printf("%g\n", *cspan_at(span2, 3, 2));
        printf("%g\n", *cspan_at(span1, 2));
    }
}
```
Output:
```
index: 59
iterate span1: 57 58 3.14
iterate span2: 48 49 50 51 52 53 54 55 56 57 58 3.14
iterate span3 by dimensions:
 0 1 2 | 3 4 5 | 6 7 8 | 9 10 11 |
 12 13 14 | 15 16 17 | 18 19 20 | 21 22 23 |
 24 25 26 | 27 28 29 | 30 31 32 | 33 34 35 |
 36 37 38 | 39 40 41 | 42 43 44 | 45 46 47 |
 48 49 50 | 51 52 53 | 54 55 56 | 57 58 3.14 |
 60 61 62 | 63 64 65 | 66 67 68 | 69 70 71 |
3.14
3.14
3.14
```