typed ⌾(a⊸⊏)

This commit is contained in:
dzaima 2021-09-11 23:39:05 +03:00
parent 9b779726b5
commit 83011df536

View File

@ -994,25 +994,54 @@ B select_ucw(B t, B o, B w, B x) {
#define EQ(F) #define EQ(F)
#define FREE_CHECK #define FREE_CHECK
#endif #endif
if (TI(w,elType)==el_i32) {
u8 we = TI(w,elType);
u8 xe = TI(x,elType);
u8 re = TI(rep,elType);
if (we==el_i32) {
i32* wp = i32any_ptr(w); i32* wp = i32any_ptr(w);
if (reusable(x) && TI(x,elType)==TI(rep,elType)) { if (re<el_f64 && xe<el_f64) {
if (v(x)->type==t_i32arr) { u8 me = xe>re?xe:re;
i32* xp = i32arr_ptr(x); bool re = reusable(x);
i32* rp = i32any_ptr(rep); if (me==el_i32) {
I32Arr* xn = re? toI32Arr(x) : cpyI32Arr(x);
rep = toI32Any(rep); i32* rp = i32any_ptr(rep);
for (usz i = 0; i < wia; i++) { for (usz i = 0; i < wia; i++) {
i64 cw = wp[i]; if (cw<0) cw+= (i64)xia; // already checked i64 cw = wp[i]; if (RARE(cw<0)) cw+= (i64)xia; // we're free to assume w is valid
i32 cr = rp[i]; i32 cr = rp[i];
EQ(cr!=xp[cw]); EQ(cr != xn->a[cw]);
xp[cw] = cr; xn->a[cw] = cr;
} }
dec(w); dec(rep); FREE_CHECK; dec(w); dec(rep); FREE_CHECK; return taga(xn);
return x; } else if (me==el_i16) {
} else if (v(x)->type==t_harr) { I16Arr* xn = re? toI16Arr(x) : cpyI16Arr(x);
rep = toI16Any(rep); i16* rp = i16any_ptr(rep);
for (usz i = 0; i < wia; i++) {
i64 cw = wp[i]; if (RARE(cw<0)) cw+= (i64)xia;
i16 cr = rp[i];
EQ(cr != xn->a[cw]);
xn->a[cw] = cr;
}
dec(w); dec(rep); FREE_CHECK; return taga(xn);
} else if (me==el_i8) {
I8Arr* xn = re? toI8Arr(x) : cpyI8Arr(x);
rep = toI8Any(rep); i8* rp = i8any_ptr(rep);
for (usz i = 0; i < wia; i++) {
i64 cw = wp[i]; if (RARE(cw<0)) cw+= (i64)xia;
i8 cr = rp[i];
EQ(cr != xn->a[cw]);
xn->a[cw] = cr;
}
dec(w); dec(rep); FREE_CHECK; return taga(xn);
} else UD;
}
if (reusable(x) && xe==re) {
if (v(x)->type==t_harr) {
B* xp = harr_ptr(x); B* xp = harr_ptr(x);
SGet(rep) SGet(rep)
for (usz i = 0; i < wia; i++) { for (usz i = 0; i < wia; i++) {
i64 cw = wp[i]; if (cw<0) cw+= (i64)xia; i64 cw = wp[i]; if (RARE(cw<0)) cw+= (i64)xia;
B cr = Get(rep, i); B cr = Get(rep, i);
EQ(!equal(cr,xp[cw])); EQ(!equal(cr,xp[cw]));
dec(xp[cw]); dec(xp[cw]);
@ -1022,11 +1051,11 @@ B select_ucw(B t, B o, B w, B x) {
return x; return x;
} }
} }
MAKE_MUT(r, xia); mut_init(r, el_or(TI(x,elType), TI(rep,elType))); MAKE_MUT(r, xia); mut_init(r, el_or(xe, re));
mut_copyG(r, 0, x, 0, xia); mut_copyG(r, 0, x, 0, xia);
SGet(rep) SGet(rep)
for (usz i = 0; i < wia; i++) { for (usz i = 0; i < wia; i++) {
i64 cw = wp[i]; if (cw<0) cw+= (i64)xia; i64 cw = wp[i]; if (RARE(cw<0)) cw+= (i64)xia;
B cr = Get(rep, i); B cr = Get(rep, i);
EQ(!equal(mut_getU(r, cw), cr)); EQ(!equal(mut_getU(r, cw), cr));
mut_rm(r, cw); mut_rm(r, cw);
@ -1035,11 +1064,11 @@ B select_ucw(B t, B o, B w, B x) {
dec(w); dec(rep); FREE_CHECK; dec(w); dec(rep); FREE_CHECK;
return mut_fcd(r, x); return mut_fcd(r, x);
} }
MAKE_MUT(r, xia); mut_init(r, el_or(TI(x,elType), TI(rep,elType))); MAKE_MUT(r, xia); mut_init(r, el_or(xe, re));
mut_copyG(r, 0, x, 0, xia); mut_copyG(r, 0, x, 0, xia);
SGet(rep) SGet(rep)
for (usz i = 0; i < wia; i++) { for (usz i = 0; i < wia; i++) {
i64 cw = o2i64u(GetU(w, i)); if (cw<0) cw+= (i64)xia; i64 cw = o2i64u(GetU(w, i)); if (RARE(cw<0)) cw+= (i64)xia;
B cr = Get(rep, i); B cr = Get(rep, i);
EQ(!equal(mut_getU(r, cw), cr)); EQ(!equal(mut_getU(r, cw), cr));
mut_rm(r, cw); mut_rm(r, cw);