[jets] add cat jet

This commit is contained in:
Philip Monk 2023-02-14 21:18:02 -07:00
parent 50c90381ef
commit 7f4a411e7e
2 changed files with 45 additions and 0 deletions

View File

@ -46,6 +46,7 @@ pub fn get_jet(jet_name: Noun) -> Result<Jet, ()> {
tas!(b"dis") => Ok(jet_dis),
tas!(b"mix") => Ok(jet_mix),
tas!(b"end") => Ok(jet_end),
tas!(b"cat") => Ok(jet_cat),
tas!(b"cut") => Ok(jet_cut),
tas!(b"met") => Ok(jet_met),
tas!(b"mug") => Ok(jet_mug),

View File

@ -440,6 +440,30 @@ pub fn jet_end(stack: &mut NockStack, subject: Noun) -> Result<Noun, JetErr> {
}
}
pub fn jet_cat(stack: &mut NockStack, subject: Noun) -> Result<Noun, JetErr> {
let arg = raw_slot(subject, 6);
let bloq = raw_slot(arg, 2).as_direct()?.data() as usize;
if bloq >= 64 {
return Err(Deterministic);
}
let a = raw_slot(arg, 6).as_atom()?;
let b = raw_slot(arg, 7).as_atom()?;
let new_size = a.size() + b.size();
if new_size == 0 {
Ok(a.as_noun())
} else {
unsafe {
let len_a = met(bloq, a);
let len_b = met(bloq, b);
let (mut new_indirect, new_slice) = IndirectAtom::new_raw_mut_bitslice(stack, new_size);
chop(bloq, 0, len_a, 0, new_slice, a.as_bitslice())?;
chop(bloq, 0, len_b, len_a, new_slice, b.as_bitslice())?;
Ok(new_indirect.normalize_as_atom().as_noun())
}
}
}
pub fn jet_cut(stack: &mut NockStack, subject: Noun) -> Result<Noun, JetErr> {
let arg = raw_slot(subject, 6);
let bloq = raw_slot(arg, 2).as_direct()?.data() as usize;
@ -1016,6 +1040,26 @@ mod tests {
assert_jet(s, jet_end, sam, a96);
}
#[test]
fn test_cat() {
let ref mut s = init();
let (a0, a24, a63, _a96, a128) = atoms(s);
let sam = T(s, &[a0, a0, a0]);
assert_jet(s, jet_cat, sam, D(0));
let sam = T(s, &[a0, a24, a128]);
let res = A(s, &ubig!(_0xdeadbeef12345678fedcba9876543210876543));
assert_jet(s, jet_cat, sam, res);
let sam = T(s, &[a0, a63, a24]);
let res = A(s, &ubig!(0x43b2a1ffffffffffffffff));
assert_jet(s, jet_cat, sam, res);
let sam = T(s, &[D(1), a63, a24]);
let res = A(s, &ubig!(0x8765437fffffffffffffff));
assert_jet(s, jet_cat, sam, res);
let sam = T(s, &[D(4), a24, a128]);
let res = A(s, &ubig!(_0xdeadbeef12345678fedcba987654321000876543));
assert_jet(s, jet_cat, sam, res);
}
#[test]
fn test_cut() {
let ref mut s = init();