admin/mlib
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ALGO_REDUCE macro

Browse Source
This commit is contained in:
Patrick Pelissier committed 2017-03-03 21:30:38 +01:00
1 parent 2af4cb2fc2
commit 85596c1a7a
2 files changed
+74 -9
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

m-algo.h
+62 -9
View File
@@ -46,6 +46,13 @@
(ALGOI_EXTRACT(contD, contDop, contS, contSop, __VA_ARGS__), \ (ALGOI_EXTRACT(contD, contDop, contS, contSop, __VA_ARGS__), \
ALGOI_EXTRACT_ARG(contD, contDop, contS, contSop, __VA_ARGS__ )) ALGOI_EXTRACT_ARG(contD, contDop, contS, contSop, __VA_ARGS__ ))
#define ALGO_REDUCE(dest, cont, contOp, ...) \
M_IF_NARGS_EQ1(__VA_ARGS__) \
(ALGOI_REDUCE(dest, cont, contOp, __VA_ARGS__), \
M_IF_NARGS_EQ2(__VA_ARGS__) \
(ALGOI_REDUCE_MAP(dest, cont, contOp, __VA_ARGS__), \
ALGOI_REDUCE_MAP_ARG(dest, cont, contOp, __VA_ARGS__) ) )
/********************************** INTERNAL ************************************/ /********************************** INTERNAL ************************************/
#define ALGOI_DEF(name, container_t, cont_oplist, type_t, type_oplist, it_t) \ #define ALGOI_DEF(name, container_t, cont_oplist, type_t, type_oplist, it_t) \
@@ -207,17 +214,19 @@
//TODO: const_iterator & CM_EACH missing... //TODO: const_iterator & CM_EACH missing...
//TODO: Algorithm missing //TODO: Algorithm missing
// nth_element, reduce, find_last, find_if, count_if, ... // nth_element, average, find_last, find_if, count_if, ...
#define ALGOI_MAP(container, cont_oplist, func) \ #define ALGOI_MAP(container, cont_oplist, func) do { \
for M_EACH(item, l, cont_oplist) { \ for M_EACH(item, container, cont_oplist) { \
func(*item); \ func(*item); \
} \ } \
} while (0)
#define ALGOI_MAP_ARG(container, cont_oplist, func, ...) \ #define ALGOI_MAP_ARG(container, cont_oplist, func, ...) do { \
for M_EACH(item, l, cont_oplist) { \ for M_EACH(item, container, cont_oplist) { \
func(__VA_ARGS__, *item); \ func(__VA_ARGS__, *item); \
} \ } \
} while (0)
#define ALGOI_EXTRACT(contDst, contDstOplist, \ #define ALGOI_EXTRACT(contDst, contDstOplist, \
contSrc, contSrcOplist, \ contSrc, contSrcOplist, \
@@ -241,4 +250,48 @@
M_IF_METHOD(REVERSE, contDstOplist) (M_GET_REVERSE contDstOplist (contDstOplist);, ) \ M_IF_METHOD(REVERSE, contDstOplist) (M_GET_REVERSE contDstOplist (contDstOplist);, ) \
} while (0) } while (0)
#define ALGOI_REDUCE(dest, cont, cont_oplist, reduceFunc) do { \
bool m_init_done = false; \
for M_EACH(item, cont, cont_oplist) { \
if (m_init_done) { \
reduceFunc(dest, *item); \
} else { \
M_GET_SET M_GET_OPLIST cont_oplist (dest, *item); \
m_init_done = true; \
} \
} \
} while (0)
#define ALGOI_REDUCE_MAP(dest, cont, cont_oplist, reduceFunc, mapFunc) do { \
bool m_init_done = false; \
M_GET_SUBTYPE cont_oplist m_tmp; \
M_GET_INIT M_GET_OPLIST cont_oplist (m_tmp); \
for M_EACH(item, cont, cont_oplist) { \
mapFunc(m_tmp, *item); \
if (m_init_done) { \
reduceFunc(dest, m_tmp); \
} else { \
M_GET_SET M_GET_OPLIST cont_oplist (dest, m_tmp); \
m_init_done = true; \
} \
} \
M_GET_CLEAR M_GET_OPLIST cont_oplist (m_tmp); \
} while (0)
#define ALGOI_REDUCE_MAP_ARG(dest, cont, cont_oplist, reduceFunc, mapFunc, ...) do { \
bool m_init_done = false; \
M_GET_SUBTYPE cont_oplist m_tmp; \
M_GET_INIT M_GET_OPLIST cont_oplist (m_tmp); \
for M_EACH(item, cont, cont_oplist) { \
mapFunc(m_tmp, __VA_ARGS__, *item); \
if (m_init_done) { \
reduceFunc(dest, m_tmp); \
} else { \
M_GET_SET M_GET_OPLIST cont_oplist (dest, m_tmp); \
m_init_done = true; \
} \
} \
M_GET_CLEAR M_GET_OPLIST cont_oplist (m_tmp); \
} while (0)
#endif #endif
tests/test-malgo.c
+12
View File
@@ -87,6 +87,18 @@ static void test_extract(void)
#define cond2(c, d) ((d) > (c)) #define cond2(c, d) ((d) > (c))
ALGO_EXTRACT(a, ARRAY_OPLIST(int), l, LIST_OPLIST(int), cond2, 10); ALGO_EXTRACT(a, ARRAY_OPLIST(int), l, LIST_OPLIST(int), cond2, 10);
assert(array_int_size(a) == 89); assert(array_int_size(a) == 89);
int dst = 0;
#define inc(d, c) (d) += (c)
ALGO_REDUCE(dst, a, ARRAY_OPLIST(int), inc);
assert (dst == 100*99/2-10*11/2);
#define sqr(d, c) (d) = (c)*(c)
ALGO_REDUCE(dst, a, ARRAY_OPLIST(int), inc, sqr);
assert (dst == 327965);
#define sqr2(d, f, c) (d) = (f) * (c)
ALGO_REDUCE(dst, a, ARRAY_OPLIST(int), inc, sqr2, 4);
assert (dst == (100*99/2-10*11/2) *4 );
array_int_clear(a); array_int_clear(a);
list_int_clear(l); list_int_clear(l);
} }