94 lines
2.6 KiB
C
94 lines
2.6 KiB
C
|
|
/*
|
|
* This file is part of the micropython-ulab project,
|
|
*
|
|
* https://github.com/v923z/micropython-ulab
|
|
*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2020 Jeff Epler for Adafruit Industries
|
|
* 2020 Zoltán Vörös
|
|
*/
|
|
|
|
#ifndef _NDARRAY_PROPERTIES_
|
|
#define _NDARRAY_PROPERTIES_
|
|
|
|
#include "py/runtime.h"
|
|
#include "py/binary.h"
|
|
#include "py/obj.h"
|
|
#include "py/objarray.h"
|
|
|
|
#include "ulab.h"
|
|
#include "ndarray.h"
|
|
|
|
#if CIRCUITPY
|
|
typedef struct _mp_obj_property_t {
|
|
mp_obj_base_t base;
|
|
mp_obj_t proxy[3]; // getter, setter, deleter
|
|
} mp_obj_property_t;
|
|
|
|
#if NDARRAY_HAS_DTYPE
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_get_dtype_obj, ndarray_dtype);
|
|
STATIC const mp_obj_property_t ndarray_dtype_obj = {
|
|
.base.type = &mp_type_property,
|
|
.proxy = {(mp_obj_t)&ndarray_get_dtype_obj,
|
|
mp_const_none,
|
|
mp_const_none },
|
|
};
|
|
#endif /* NDARRAY_HAS_DTYPE */
|
|
|
|
#if NDARRAY_HAS_ITEMSIZE
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_get_itemsize_obj, ndarray_itemsize);
|
|
STATIC const mp_obj_property_t ndarray_itemsize_obj = {
|
|
.base.type = &mp_type_property,
|
|
.proxy = {(mp_obj_t)&ndarray_get_itemsize_obj,
|
|
mp_const_none,
|
|
mp_const_none },
|
|
};
|
|
#endif /* NDARRAY_HAS_ITEMSIZE */
|
|
|
|
#if NDARRAY_HAS_SHAPE
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_get_shape_obj, ndarray_shape);
|
|
STATIC const mp_obj_property_t ndarray_shape_obj = {
|
|
.base.type = &mp_type_property,
|
|
.proxy = {(mp_obj_t)&ndarray_get_shape_obj,
|
|
mp_const_none,
|
|
mp_const_none },
|
|
};
|
|
#endif /* NDARRAY_HAS_SHAPE */
|
|
|
|
#if NDARRAY_HAS_SIZE
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_get_size_obj, ndarray_size);
|
|
STATIC const mp_obj_property_t ndarray_size_obj = {
|
|
.base.type = &mp_type_property,
|
|
.proxy = {(mp_obj_t)&ndarray_get_size_obj,
|
|
mp_const_none,
|
|
mp_const_none },
|
|
};
|
|
#endif /* NDARRAY_HAS_SIZE */
|
|
|
|
#if NDARRAY_HAS_STRIDES
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_get_strides_obj, ndarray_strides);
|
|
STATIC const mp_obj_property_t ndarray_strides_obj = {
|
|
.base.type = &mp_type_property,
|
|
.proxy = {(mp_obj_t)&ndarray_get_strides_obj,
|
|
mp_const_none,
|
|
mp_const_none },
|
|
};
|
|
#endif /* NDARRAY_HAS_STRIDES */
|
|
|
|
#else
|
|
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_dtype_obj, ndarray_dtype);
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_itemsize_obj, ndarray_itemsize);
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_shape_obj, ndarray_shape);
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_size_obj, ndarray_size);
|
|
MP_DEFINE_CONST_FUN_OBJ_1(ndarray_strides_obj, ndarray_strides);
|
|
|
|
#endif /* CIRCUITPY */
|
|
|
|
#if !CIRCUITPY
|
|
void ndarray_properties_attr(mp_obj_t , qstr , mp_obj_t *);
|
|
#endif
|
|
|
|
#endif
|