polydat 0.1.0

Polydat — generation kernel for deterministic variate generation in nb-rs (formerly nbrs-variates)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

// Copyright 2024-2026 Jonathan Shook
// SPDX-License-Identifier: Apache-2.0

//! Programmatic kernel construction via the assembler API.

use polydat::assembly::{GkAssembler, WireRef};
use polydat::nodes::hash::Hash64;
use polydat::nodes::arithmetic::{ModU64, MixedRadix};

fn main() {
    let mut asm = GkAssembler::new(vec!["cycle".into()]);

    // Decompose cycle into (region, device, reading) coordinates
    asm.add_node("decompose", Box::new(MixedRadix::new(vec![100, 1000, 0])),
        vec![WireRef::input("cycle")]);

    // Hash region for a deterministic region code
    asm.add_node("region_h", Box::new(Hash64::new()),
        vec![WireRef::node_port("decompose", 0)]);
    asm.add_node("region_code", Box::new(ModU64::new(10000)),
        vec![WireRef::node("region_h")]);

    // Declare outputs
    asm.add_output("region", WireRef::node_port("decompose", 0));
    asm.add_output("device", WireRef::node_port("decompose", 1));
    asm.add_output("reading", WireRef::node_port("decompose", 2));
    asm.add_output("region_code", WireRef::node("region_code"));

    let mut kernel = asm.compile().expect("assembly failed");

    println!("cycle     region  device  reading  region_code");
    println!("--------  ------  ------  -------  -----------");
    for cycle in [0, 1, 50, 100, 10000, 100000] {
        kernel.set_inputs(&[cycle]);
        println!("{cycle:>8}  {:>6}  {:>6}  {:>7}  {:>11}",
            kernel.pull("region").as_u64(),
            kernel.pull("device").as_u64(),
            kernel.pull("reading").as_u64(),
            kernel.pull("region_code").as_u64(),
        );
    }
}