feat: added preprocessor macros to manually unroll loops

1 files changed, 37 insertions, 0 deletions

diff --git a/sys/libmath/loop.h b/sys/libmath/loop.h
new file mode 100644
index 0000000..9a425a7
--- /dev/null
+++ b/sys/libmath/loop.h
@@ -0,0 +1,37 @@
+#pragma once
+
+#define ROUNDBY(x, n) ((x) & ~((n)-1))
+
+/* loop unrolling (vertical) */
+#define LOOP1(I,STMT,...)  STMT(I,__VA_ARGS__) 
+#define LOOP2(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP1((I+1),STMT,__VA_ARGS__)
+#define LOOP3(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP2((I+1),STMT,__VA_ARGS__)
+#define LOOP4(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP3((I+1),STMT,__VA_ARGS__)
+#define LOOP5(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP4((I+1),STMT,__VA_ARGS__)
+#define LOOP6(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP5((I+1),STMT,__VA_ARGS__)
+#define LOOP7(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP6((I+1),STMT,__VA_ARGS__)
+#define LOOP8(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP7((I+1),STMT,__VA_ARGS__)
+#define LOOP9(I,STMT,...)  STMT(I,__VA_ARGS__) LOOP8((I+1),STMT,__VA_ARGS__)
+#define LOOP10(I,STMT,...) STMT(I,__VA_ARGS__) LOOP9((I+1),STMT,__VA_ARGS__)
+#define LOOP11(I,STMT,...) STMT(I,__VA_ARGS__) LOOP10((I+1),STMT,__VA_ARGS__)
+#define LOOP12(I,STMT,...) STMT(I,__VA_ARGS__) LOOP11((I+1),STMT,__VA_ARGS__)
+#define LOOP13(I,STMT,...) STMT(I,__VA_ARGS__) LOOP12((I+1),STMT,__VA_ARGS__)
+#define LOOP14(I,STMT,...) STMT(I,__VA_ARGS__) LOOP13((I+1),STMT,__VA_ARGS__)
+#define LOOP15(I,STMT,...) STMT(I,__VA_ARGS__) LOOP14((I+1),STMT,__VA_ARGS__)
+#define LOOP16(I,STMT,...) STMT(I,__VA_ARGS__) LOOP15((I+1),STMT,__VA_ARGS__)
+
+#define _LOOP_(n,I,STMT,...) LOOP##n(I,STMT,__VA_ARGS__) 
+#define LOOP(n,I,STMT,...) _LOOP_(n,I,STMT,__VA_ARGS__)
+
+/* loop expansion (horizontal) */
+#define EXPAND1(I,TERM,OP,...) TERM(I,__VA_ARGS__)
+#define EXPAND2(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND1((I+1),TERM,OP,__VA_ARGS__)
+#define EXPAND3(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND2((I+1),TERM,OP,__VA_ARGS__)
+#define EXPAND4(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND3((I+1),TERM,OP,__VA_ARGS__)
+#define EXPAND5(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND4((I+1),TERM,OP,__VA_ARGS__)
+#define EXPAND6(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND5((I+1),TERM,OP,__VA_ARGS__)
+#define EXPAND7(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND6((I+1),TERM,OP,__VA_ARGS__)
+#define EXPAND8(I,TERM,OP,...) TERM(I,__VA_ARGS__) OP EXPAND7((I+1),TERM,OP,__VA_ARGS__)
+
+#define _EXPAND_(n,I,TERM,OP,...) EXPAND##n(I,TERM,OP,__VA_ARGS__) 
+#define EXPAND(n,I,TERM,OP, ...) _EXPAND_(n,I,TERM,OP,__VA_ARGS__)