From 665a2f56726e8cdfee9200a28424b596204fbfcf Mon Sep 17 00:00:00 2001
From: Edward Amsden <edward@blackriversoft.com>
Date: Wed, 1 Feb 2023 18:35:05 -0600
Subject: [PATCH] [ares] Source code for cue-and-then-jam benchmark

---
 rust/ares/benches/cue_pill.rs | 70 +++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 rust/ares/benches/cue_pill.rs

diff --git a/rust/ares/benches/cue_pill.rs b/rust/ares/benches/cue_pill.rs
new file mode 100644
index 0000000..6e0cc0f
--- /dev/null
+++ b/rust/ares/benches/cue_pill.rs
@@ -0,0 +1,70 @@
+use memmap;
+use ares::serialization::{cue,jam};
+use ares::mem::NockStack;
+use ares::noun::{IndirectAtom,DirectAtom};
+use std::env;
+use std::fs::{File,OpenOptions};
+use std::io;
+use std::mem;
+use std::ptr::{copy_nonoverlapping,write_bytes};
+use std::time::SystemTime;
+
+fn main() -> io::Result<()> {
+    let filename = env::args().nth(1).expect("Must provide input filename");
+    let output_filename = format!("{}.out", filename.clone());
+    let f = File::open(filename)?;
+    let in_len = f.metadata()?.len();
+    let mut stack = NockStack::new(1 << 10 << 10 << 10, 0);
+    let jammed_input = unsafe {
+        let in_map = memmap::Mmap::map(&f)?;
+        let word_len = (in_len + 7) >> 3;
+        let (mut atom, dest) = IndirectAtom::new_raw_mut(&mut stack, word_len as usize);
+        write_bytes(dest.add(word_len as usize - 1), 0, 8);
+        copy_nonoverlapping(in_map.as_ptr(), dest as *mut u8, in_len as usize);
+        mem::drop(in_map);
+        atom.normalize_as_atom()
+    };
+
+    let now = SystemTime::now();
+
+    let mut i = 0;
+    let mut input = unsafe { DirectAtom::new_unchecked(0).as_atom().as_noun() };
+    loop {
+      if (i >= 1) { break; };
+      i += 1;
+      input = cue(&mut stack, jammed_input);
+    };
+
+    match now.elapsed() {
+        Ok(elapse) => {
+            println!("{}", elapse.as_secs_f64());
+        },
+        Err(_) => println!("NO TIME FOR YOU!"),
+    };
+
+    let nuw = SystemTime::now();
+
+    let jammed_output = jam(&mut stack, input);
+
+    match nuw.elapsed() {
+        Ok(elapse) => {
+            println!("Jam: {}", elapse.as_secs_f64());
+        },
+        Err(_) => println!("NO TIME FOR YOU!"),
+    };
+
+    let f_out = OpenOptions::new().read(true).write(true).create(true).open(output_filename)?;
+    f_out.set_len((jammed_output.size() << 3) as u64)?;
+    unsafe {
+        let mut out_map = memmap::MmapMut::map_mut(&f_out)?;
+        copy_nonoverlapping(
+            jammed_output.data_pointer() as *mut u8,
+            out_map.as_mut_ptr(),
+            jammed_output.size() << 3,
+        );
+        out_map.flush();
+    };
+        
+      
+    Ok(())
+}