forked from MightyPork/crsn
add rpush and rpop stack instructions
This commit is contained in:
@@ -5,4 +5,6 @@ pub enum StackOp {
|
|||||||
NewStack { dst: Wr },
|
NewStack { dst: Wr },
|
||||||
Push { obj: RdObj, src: Rd },
|
Push { obj: RdObj, src: Rd },
|
||||||
Pop { dst: Wr, obj: RdObj },
|
Pop { dst: Wr, obj: RdObj },
|
||||||
|
ReversePush { obj: RdObj, src: Rd },
|
||||||
|
ReversePop { dst: Wr, obj: RdObj },
|
||||||
}
|
}
|
||||||
|
|||||||
+57
-32
@@ -10,6 +10,7 @@ use crate::defs::StackOp;
|
|||||||
use crsn::sexp::Sexp;
|
use crsn::sexp::Sexp;
|
||||||
use crsn::sexp;
|
use crsn::sexp;
|
||||||
use crsn::utils::A;
|
use crsn::utils::A;
|
||||||
|
use crsn::asm::data::{RdObj, Rd, Wr};
|
||||||
|
|
||||||
#[derive(Debug, Default)]
|
#[derive(Debug, Default)]
|
||||||
struct Stacks {
|
struct Stacks {
|
||||||
@@ -28,40 +29,20 @@ impl OpTrait for StackOp {
|
|||||||
state.write(*dst, id)?;
|
state.write(*dst, id)?;
|
||||||
}
|
}
|
||||||
StackOp::Push { obj, src } => {
|
StackOp::Push { obj, src } => {
|
||||||
state.clear_status();
|
prepare_push(state, obj, src,
|
||||||
let handle = state.read_obj(*obj)?;
|
|stack, val| stack.push_back(val))?;
|
||||||
let val = state.read(*src)?;
|
}
|
||||||
|
StackOp::ReversePush { obj, src } => {
|
||||||
let stacks: &mut Stacks = state.ext_mut();
|
prepare_push(state, obj, src,
|
||||||
stacks.store.get_mut(&handle)
|
|stack, val| stack.push_front(val))?;
|
||||||
.ok_or_else(|| Fault::ObjectNotExist(handle))?
|
|
||||||
.push_back(val);
|
|
||||||
}
|
}
|
||||||
StackOp::Pop { dst, obj } => {
|
StackOp::Pop { dst, obj } => {
|
||||||
state.clear_status();
|
prepare_pop(state, dst, obj,
|
||||||
let handle = state.read_obj(*obj)?;
|
|stack| stack.pop_back())?;
|
||||||
|
}
|
||||||
let stacks: &mut Stacks = state.ext_mut();
|
StackOp::ReversePop { dst, obj } => {
|
||||||
let stack = stacks.store.get_mut(&handle)
|
prepare_pop(state, dst, obj,
|
||||||
.ok_or_else(|| Fault::ObjectNotExist(handle))?;
|
|stack| stack.pop_front())?;
|
||||||
|
|
||||||
let val = stack.pop_back();
|
|
||||||
|
|
||||||
if stack.is_empty() {
|
|
||||||
state.set_flag(Cond::Zero, true);
|
|
||||||
}
|
|
||||||
|
|
||||||
let val = match val {
|
|
||||||
None => {
|
|
||||||
state.set_flag(Cond::Overflow, true);
|
|
||||||
0
|
|
||||||
}
|
|
||||||
Some(val) => {
|
|
||||||
val
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
state.write(*dst, val)?;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -73,11 +54,55 @@ impl OpTrait for StackOp {
|
|||||||
StackOp::NewStack { dst } => sexp::list(&[A("stack"), A(dst)]),
|
StackOp::NewStack { dst } => sexp::list(&[A("stack"), A(dst)]),
|
||||||
StackOp::Push { obj, src } => sexp::list(&[A("push"), A(obj), A(src)]),
|
StackOp::Push { obj, src } => sexp::list(&[A("push"), A(obj), A(src)]),
|
||||||
StackOp::Pop { dst, obj } => sexp::list(&[A("pop"), A(dst), A(obj)]),
|
StackOp::Pop { dst, obj } => sexp::list(&[A("pop"), A(dst), A(obj)]),
|
||||||
|
StackOp::ReversePush { obj, src } => sexp::list(&[A("rpush"), A(obj), A(src)]),
|
||||||
|
StackOp::ReversePop { dst, obj } => sexp::list(&[A("rpop"), A(dst), A(obj)]),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
pub(crate) fn drop_obj(state: &mut RunState, handle: Value) -> Result<Option<()>, Fault> {
|
pub(crate) fn drop_obj(state: &mut RunState, handle: Value) -> Result<Option<()>, Fault> {
|
||||||
let stacks: &mut Stacks = state.ext_mut();
|
let stacks: &mut Stacks = state.ext_mut();
|
||||||
Ok(stacks.store.remove(&handle).map(|_| ()))
|
Ok(stacks.store.remove(&handle).map(|_| ()))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn prepare_push(mut state: &mut RunState, obj: &RdObj, src: &Rd, pushfn : impl FnOnce(&mut VecDeque<Value>, Value) -> ()) -> Result<(), Fault> {
|
||||||
|
state.clear_status();
|
||||||
|
let handle = state.read_obj(*obj)?;
|
||||||
|
let val = state.read(*src)?;
|
||||||
|
|
||||||
|
let stacks: &mut Stacks = state.ext_mut();
|
||||||
|
let mutable = stacks.store.get_mut(&handle)
|
||||||
|
.ok_or_else(|| Fault::ObjectNotExist(handle))?;
|
||||||
|
|
||||||
|
pushfn(mutable, val);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn prepare_pop(state: &mut RunState, dst: &Wr, obj: &RdObj, popfn : impl FnOnce(&mut VecDeque<Value>) -> Option<Value>) -> Result<(), Fault> {
|
||||||
|
state.clear_status();
|
||||||
|
let handle = state.read_obj(*obj)?;
|
||||||
|
|
||||||
|
let stacks: &mut Stacks = state.ext_mut();
|
||||||
|
let stack = stacks.store.get_mut(&handle)
|
||||||
|
.ok_or_else(|| Fault::ObjectNotExist(handle))?;
|
||||||
|
|
||||||
|
let val = popfn(stack);
|
||||||
|
|
||||||
|
if stack.is_empty() {
|
||||||
|
state.set_flag(Cond::Zero, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
let val = match val {
|
||||||
|
None => {
|
||||||
|
state.set_flag(Cond::Overflow, true);
|
||||||
|
0
|
||||||
|
}
|
||||||
|
Some(val) => {
|
||||||
|
val
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
state.write(*dst, val)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|||||||
@@ -27,6 +27,20 @@ pub(crate) fn parse<'a>(keyword: &str, mut args: TokenParser<'a>) -> Result<Pars
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
"rpush" => {
|
||||||
|
StackOp::ReversePush {
|
||||||
|
obj: args.next_rdobj()?,
|
||||||
|
src: args.next_rd()?,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
"rpop" => {
|
||||||
|
StackOp::ReversePop {
|
||||||
|
dst: args.next_wr()?,
|
||||||
|
obj: args.next_rdobj()?,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
_other => {
|
_other => {
|
||||||
return Ok(ParseRes::Unknown(args));
|
return Ok(ParseRes::Unknown(args));
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,5 +10,27 @@
|
|||||||
(pop r1 @r0)
|
(pop r1 @r0)
|
||||||
(cmp r1 1 (ne? (fault)))
|
(cmp r1 1 (ne? (fault)))
|
||||||
|
|
||||||
|
; test reverse push
|
||||||
|
(push @r0 1)
|
||||||
|
(push @r0 7)
|
||||||
|
(rpush @r0 8)
|
||||||
|
|
||||||
|
(pop r1 @r0)
|
||||||
|
(cmp r1 7 (ne? (fault)))
|
||||||
|
|
||||||
|
(pop r1 @r0)
|
||||||
|
(cmp r1 1 (ne? (fault)))
|
||||||
|
|
||||||
|
(pop r1 @r0)
|
||||||
|
(cmp r1 8 (ne? (fault)))
|
||||||
|
|
||||||
|
; test reverse pop
|
||||||
|
|
||||||
|
(push @r0 1000)
|
||||||
|
(push @r0 2000)
|
||||||
|
|
||||||
|
(rpop r1 @r0)
|
||||||
|
(cmp r1 1000 (ne? (fault)))
|
||||||
|
|
||||||
(drop @r0)
|
(drop @r0)
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user