#include "../core.h" #include "../utils/mut.h" #include "../builtins.h" #include #define F64_MIN -INFINITY #define F64_MAX INFINITY #if !USE_VALGRIND static u64 vg_rand(u64 x) { return x; } #endif #if SINGELI_X86_64 #define SINGELI_FILE scan #include "../utils/includeSingeli.h" #if __PCLMUL__ #define SINGELI_FILE neq #include "../utils/includeSingeli.h" #endif #endif B scan_ne(B x, u64 p, u64 ia) { // consumes x u64* xp = bitarr_ptr(x); u64* rp; B r=m_bitarrv(&rp,ia); #if SINGELI_X86_64 && __PCLMUL__ clmul_scan_ne(p, xp, rp, BIT_N(ia)); #else for (usz i = 0; i < BIT_N(ia); i++) { u64 c = xp[i]; u64 r = c ^ (c<<1); r^= r<< 2; r^= r<< 4; r^= r<<8; r^= r<<16; r^= r<<32; r^= p; rp[i] = r; p = -(r>>63); // repeat sign bit } #endif decG(x); return r; } static B scan_or(B x, u64 ia) { // consumes x u64* xp = bitarr_ptr(x); u64* rp; B r=m_bitarrv(&rp,ia); usz n=BIT_N(ia); u64 xi; usz i=0; while (i,MIN,or ,asc) } #undef MINMAX // Initialized: try to convert 𝕨 to type of 𝕩 // (could do better for out-of-range floats) B shape_c2(B, B, B); #define MM2_ICASE(T,N,C,I) \ case el_##T : { \ if (wv!=(T)wv) { if (wv C 0) { r=C2(shape,m_f64(ia),w); break; } else wv=I; } \ T* xp=T##any_ptr(x); T* rp; r=m_##T##arrv(&rp, ia); MINMAX_SCAN(T,N,C,wv); \ break; } #define MINMAX2(NAME,C,INIT,BIT,BI,ORD) \ i32 wv=0; if (q_i32(w)) { wv=w.f; } else { x=taga(cpyF64Arr(x)); xe=el_f64; } \ B r; switch (xe) { default:UD; \ case el_bit: if (wv C BI) r=C2(shape,m_f64(ia),w); else return scan_##BIT(x, ia); break; \ MM2_ICASE(i8 ,NAME,C,I8_##INIT ) \ MM2_ICASE(i16,NAME,C,I16_##INIT) \ MM_CASE(i32,NAME,C,wv) \ MM_CASE(f64,NAME,C,w.f) \ } \ decG(x); return FL_SET(r, fl_##ORD); static B scan2_min_num(B w, B x, u8 xe, usz ia) { MINMAX2(min,<,MAX,and,1,dsc) } static B scan2_max_num(B w, B x, u8 xe, usz ia) { MINMAX2(max,>,MIN,or ,0,asc) } #undef MINMAX2 #undef MM_CASE #undef MM_CASE2 #undef MINMAX_SCAN B scan_c1(Md1D* d, B x) { B f = d->f; if (isAtm(x) || RNK(x)==0) thrM("`: Argument cannot have rank 0"); ur xr = RNK(x); usz ia = IA(x); if (ia==0) return x; B xf = getFillQ(x); u8 xe = TI(x,elType); if (xr==1 && xe<=el_f64 && isFun(f) && v(f)->flags) { u8 rtid = v(f)->flags-1; if (xe==el_bit) { u64* xp=bitarr_ptr(x); if (rtid==n_add ) return scan_add_bool(x, ia); if (rtid==n_or | rtid==n_ceil ) return scan_or(x, ia); if (rtid==n_and | rtid==n_mul | rtid==n_floor) return scan_and(x, ia); if (rtid==n_ne ) return scan_ne(x, 0, ia); if (rtid==n_lt) { u64* rp; B r=m_bitarrv(&rp,ia); usz n=BIT_N(ia); u64 m10 = 0x5555555555555555; u64 p = 0; for (usz i=0; i>63); rp[i] = p = x & (m10 ^ (x + c)); } decG(x); return r; } goto base; } if (rtid==n_add) { // + if (xe==el_i8 ) { i8* xp=i8any_ptr (x); i32* rp; B r=m_i32arrv(&rp, ia); i32 c=0; for (usz i=0; if; if (isAtm(x) || RNK(x)==0) thrM("`: 𝕩 cannot have rank 0"); ur xr = RNK(x); usz* xsh = SH(x); usz ia = IA(x); B wf = getFillQ(w); u8 xe = TI(x,elType); if (xr==1 && xe<=el_f64 && isFun(f) && v(f)->flags) { u8 rtid = v(f)->flags-1; if (rtid==n_floor) return scan2_min_num(w, x, xe, ia); // ⌊ if (rtid==n_ceil ) return scan2_max_num(w, x, xe, ia); // ⌈ if (!q_i32(w)) goto base; i32 wv = o2iG(w); if (xe==el_bit) { u64* xp=bitarr_ptr(x); if (rtid==n_add) { i32* rp; B r=m_i32arrv(&rp, ia); i64 c=wv; for (usz i=0; i