use crate::authoring::*;
fn fwd(_op: &Op, _ctx: &dyn Context, operands: &mut dyn CoordinateSet) -> usize {
let n = operands.len();
let mut successes = 0;
for i in 0..n {
let mut o = operands.get_coord(i);
o[0] += 1.;
successes += 1;
operands.set_coord(i, &o);
}
successes
}
fn inv(_op: &Op, _ctx: &dyn Context, operands: &mut dyn CoordinateSet) -> usize {
let n = operands.len();
let mut successes = 0;
for i in 0..n {
let mut o = operands.get_coord(i);
o[0] -= 1.;
successes += 1;
operands.set_coord(i, &o);
}
successes
}
#[rustfmt::skip]
pub const GAMUT: [OpParameter; 1] = [
OpParameter::Flag { key: "inv" },
];
pub fn new(parameters: &RawParameters, _ctx: &dyn Context) -> Result<Op, Error> {
Op::basic(parameters, InnerOp(fwd), Some(InnerOp(inv)), &GAMUT)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn addone() -> Result<(), Error> {
let mut ctx = Minimal::default();
let op = ctx.op("addone")?;
let mut data = crate::test_data::coor2d();
assert_eq!(data[0][0], 55.);
assert_eq!(data[1][0], 59.);
ctx.apply(op, Fwd, &mut data)?;
assert_eq!(data[0][0], 56.);
assert_eq!(data[1][0], 60.);
ctx.apply(op, Inv, &mut data)?;
assert_eq!(data[0][0], 55.);
assert_eq!(data[1][0], 59.);
Ok(())
}
}