48 lines
1.7 KiB
C
48 lines
1.7 KiB
C
/*
|
|
* This file is part of the micropython-ulab project,
|
|
*
|
|
* https://github.com/v923z/micropython-ulab
|
|
*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2019 Zoltán Vörös
|
|
*/
|
|
|
|
#ifndef _NUMERICAL_
|
|
#define _NUMERICAL_
|
|
|
|
#include "ndarray.h"
|
|
|
|
mp_obj_t numerical_linspace(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_sum(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_mean(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_std(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_min(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_max(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_argmin(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_argmax(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_roll(size_t , const mp_obj_t *, mp_map_t *);
|
|
|
|
// TODO: implement minimum/maximum, flip, and cumsum
|
|
mp_obj_t numerical_minimum(mp_obj_t , mp_obj_t );
|
|
mp_obj_t numerical_maximum(mp_obj_t , mp_obj_t );
|
|
mp_obj_t numerical_cumsum(size_t , const mp_obj_t *, mp_map_t *);
|
|
mp_obj_t numerical_flip(size_t , const mp_obj_t *, mp_map_t *);
|
|
|
|
// this macro could be tighter, if we moved the ifs to the argmin function, assigned <, as well as >
|
|
#define ARG_MIN_LOOP(in, type, start, stop, stride, op) do {\
|
|
type *array = (type *)(in)->array->items;\
|
|
if(((op) == NUMERICAL_MAX) || ((op) == NUMERICAL_ARGMAX)) {\
|
|
for(size_t i=(start)+(stride); i < (stop); i+=(stride)) {\
|
|
if((array)[i] > (array)[best_idx]) {\
|
|
best_idx = i;\
|
|
}\
|
|
}\
|
|
} else{\
|
|
for(size_t i=(start)+(stride); i < (stop); i+=(stride)) {\
|
|
if((array)[i] < (array)[best_idx]) best_idx = i;\
|
|
}\
|
|
}\
|
|
} while(0)
|
|
|
|
#endif
|