diff --git a/src/builtins/select.c b/src/builtins/select.c index e70f6de1..2397128e 100644 --- a/src/builtins/select.c +++ b/src/builtins/select.c @@ -63,6 +63,7 @@ B select_c2(B t, B w, B x) { return r; } usz xia = IA(x); + if (xia==0) thrM("โŠ: Indexing out-of-bounds (0โ‰กโ‰ ๐•ฉ)"); u8 xe = TI(x,elType); u8 we = TI(w,elType); #if SINGELI @@ -72,26 +73,46 @@ B select_c2(B t, B w, B x) { // if (we==el_i32 && xe==el_i32) { i32* rp; r = m_i32arrc(&rp, w); if (!avx2_select_i32_32((u8*)i32any_ptr(w), (u8*)i32any_ptr(x), (u8*)rp, wia, xia)) thrM("โŠ: Indexing out-of-bounds"); goto dec_ret; } // if (we==el_i32 && xe==el_f64) { f64* rp; r = m_f64arrc(&rp, w); if (!avx2_select_i32_64((u8*)i32any_ptr(w), (u8*)f64any_ptr(x), (u8*)rp, wia, xia)) thrM("โŠ: Indexing out-of-bounds"); goto dec_ret; } #endif - #define CASE(E,TY) { \ - E* rp = m_tyarrc(&r, sizeof(E), w, TY); \ - E* xp = tyany_ptr(x); \ - for (usz i = 0; i < wia; i++) rp[i] = xp[WRAP(wp[i], xia, thrF("โŠ: Indexing out-of-bounds (%iโˆŠ๐•จ, %sโ‰กโ‰ ๐•ฉ)", wp[i], xia))]; \ - goto dec_ret; \ - } + #define CASE(S,E) case S: for (usz i=i0; i> 63; \ + if (i%64 == 0) { rp[i/64]=b; if (!i) break; } \ + } \ + goto dec_ret; \ + } \ + if (xe!=el_B) { \ + usz xw = elWidth(xe); \ + void* rp = m_tyarrc(&r, xw, w, el2t(xe)); \ + void* xp = tyany_ptr(x); \ + if (sizeof(W) >= 4) { \ + switch(xw) { default:UD; CASEW(1,u8); CASEW(2,u16); CASEW(4,u32); CASEW(8,f64); } \ + } else { \ + W* wt = NULL; \ + for (usz bl=(1<<14)/sizeof(W), i0=0, i1=0; i0wia) i1=wia; \ + W min=wp[i0], max=min; for (usz i=i0+1; imax) max=e; if (e=(i64)xia) thrF("โŠ: Indexing out-of-bounds (%iโˆŠ๐•จ, %sโ‰กโ‰ ๐•ฉ)", max, xia); \ + W* ip=wp; usz off=xia; \ + if (max>=0) { off=0; if (RARE(min<0)) { \ + if (!wt) wt=TALLOCP(W,i1-i0); ip=wt-i0; \ + for (usz i=i0; i=2) { + if (xe==el_bit && wia>=256 && wia/4>=xia && we!=el_bit) { + return taga(cpyBitArr(select_c2(m_f64(0), w, taga(cpyI8Arr(x))))); + } + if (we==el_bit) { + if (xia<2) { + u64* wp=bitarr_ptr(w); + usz i; for (i=0; idata; // +8 so mm is happy +#define TALLOCP(T,AM) ((T*) ((TAlloc*)mm_alloc(TOFF + (AM)*sizeof(T) + 8, t_temp))->data) // +8 so mm is happy +#define TALLOC(T,N,AM) T* N = TALLOCP(T,AM); #define TOBJ(N) (void*)((u8*)(N) - TOFF) #define TFREE(N) mm_free((Value*)TOBJ(N)); #define TREALLOC(N, AM) talloc_realloc(TOBJ(N), AM) @@ -35,4 +36,4 @@ typedef struct TStack { #define TSSIZE(N) (N##_o->size) TStack* ts_e(TStack* o, u32 elsz, u64 am); -#define ARBOBJ(SZ) (TAlloc*)mm_alloc(sizeof(TAlloc)+(SZ), t_arbObj) \ No newline at end of file +#define ARBOBJ(SZ) (TAlloc*)mm_alloc(sizeof(TAlloc)+(SZ), t_arbObj)