use gategen::boolvar::*;
use gategen::gatesim::*;
use gategen::intvar::*;
use gatenative::{cpu_build_exec::*, *};
fn mul_add_circuit() -> Circuit<u32> {
call32(|| {
let a = U16Var32::var();
let b = U16Var32::var();
let c = U16Var32::var();
let r = &a * &b + &c;
r.to_translated_circuit(a.concat(b).concat(c).iter())
})
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let b_start = 4782u16;
let c_start = 18941u16;
let param_a = 0x4da0u16;
let param_b = 0x8725u16;
let param_c = 0xe052u16;
let circuit = mul_add_circuit();
let mut builder = CPUBuilder::new(None);
builder.transform_helpers();
builder.add_with_config(
"mul_add",
circuit,
CodeConfig::new()
.arg_inputs(Some(&(16..48).collect::<Vec<_>>()))
.pop_input_len(Some(3))
.pop_input_code(Some(
r##"{
// Get paramareters from input
const uint32_t* params = (const uint32_t*)input;
const uint32_t param_a = params[0];
const uint32_t param_b = params[1];
const uint32_t param_c = params[2];
size_t i;
uint32_t temp[TYPE_LEN];
// generate values of function: (x xor param_a) + (x and param_b) + (x or param_c).
for (i = 0; i < TYPE_LEN; i++) {
const uint32_t id = idx*TYPE_LEN + i;
temp[i] = ((id^param_a) + (id¶m_b) + (id|param_c)) & 0xffff;
}
INPUT_TRANSFORM_B16(i0, i1, i2, i3, i4, i5, i6, i7,
i8, i9, i10, i11, i12, i13, i14, i15, temp);
}"##,
))
.aggr_output_len(Some(1 << 16))
.aggr_output_code(Some(
r##"{
uint32_t* output_u32 = (uint32_t*)output;
OUTPUT_TRANSFORM_B16(output_u32 + TYPE_LEN*idx, o0, o1, o2, o3, o4, o5, o6, o7,
o8, o9, o10, o11, o12, o13, o14, o15);
}"##,
)),
);
let mut execs = builder.build()?;
let input = execs[0].new_data_from_slice(&[param_a.into(), param_b.into(), param_c.into()]);
let output = execs[0].execute(&input, ((c_start as u64) << 16) | (b_start as u64))?;
let output = output.release();
for (i, v) in output.into_iter().enumerate() {
println!("{}: {}", i, v);
}
Ok(())
}