micropython/ports/secure_iot/ndarray_properties.c at master · Mindgrove-Technologies/micropython · GitHub

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

/*
 * This file is part of the micropython-ulab project,
 *
 * https://github.com/v923z/micropython-ulab
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2021-2025 Zoltán Vörös
 *
*/

#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "py/obj.h"
#include "py/runtime.h"

#include "ulab.h"
#include "ndarray.h"
#include "numpy/ndarray/ndarray_iter.h"
#if ULAB_SUPPORTS_COMPLEX
#include "numpy/carray/carray.h"
#endif

void ndarray_properties_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
    if (dest[0] == MP_OBJ_NULL) {
        switch(attr) {
            #if NDARRAY_HAS_DTYPE
            case MP_QSTR_dtype:
                dest[0] = ndarray_dtype(self_in);
                break;
            #endif
            #if NDARRAY_HAS_FLATITER
            case MP_QSTR_flat:
                dest[0] = ndarray_flatiter_make_new(self_in);
                break;
            #endif
            #if NDARRAY_HAS_ITEMSIZE
            case MP_QSTR_itemsize:
                dest[0] = ndarray_itemsize(self_in);
                break;
            #endif
            #if NDARRAY_HAS_NDIM
            case MP_QSTR_ndim:
                dest[0] = ndarray_ndim(self_in);
                break;
            #endif
            #if NDARRAY_HAS_SHAPE
            case MP_QSTR_shape:
                dest[0] = ndarray_shape(self_in);
                break;
            #endif
            #if NDARRAY_HAS_SIZE
            case MP_QSTR_size:
                dest[0] = ndarray_size(self_in);
                break;
            #endif
            #if NDARRAY_HAS_STRIDES
            case MP_QSTR_strides:
                dest[0] = ndarray_strides(self_in);
                break;
            #endif
            #if NDARRAY_HAS_TRANSPOSE
            case MP_QSTR_T:
                dest[0] = ndarray_transpose(self_in);
                break;
            #endif
            #if ULAB_SUPPORTS_COMPLEX
            #if ULAB_NUMPY_HAS_IMAG
            case MP_QSTR_imag:
                dest[0] = carray_imag(self_in);
                break;
            #endif
            #if ULAB_NUMPY_HAS_IMAG
            case MP_QSTR_real:
                dest[0] = carray_real(self_in);
                break;
            #endif
            #endif /* ULAB_SUPPORTS_COMPLEX */
            default:
                // forward to locals dict
                dest[1] = MP_OBJ_SENTINEL;
                break;
        }
    } else {
        if(dest[1]) {
            switch(attr) {
                #if ULAB_MAX_DIMS > 1
                #if NDARRAY_HAS_RESHAPE
                case MP_QSTR_shape:
                    ndarray_reshape_core(self_in, dest[1], 1);
                    break;
                #endif
                #endif
                default:
                    return;
                    break;
            }
            dest[0] = MP_OBJ_NULL;
        }
    }
}