implement yielding

Author: pufferfish101007

README.md

@@ -28,7 +28,7 @@ If you experience runtime stack overflow errors in debug mode, try using the `-s
 To add a new block named `category_opcode`, if it cannot be reduced to simpler blocks:
 1. create `src/instructions/category/opcode.rs`. Make sure to `use super::super::prelude::*` and create the relevant `pub` items:
 - (optional) `pub struct Fields` (must be `Debug` and `Clone`)
-- `pub fn wasm(func: &StepFunc, inputs: Rc<[IrType]>, (fields: &Fields)?) -> HQResult<Vec<Instruction<'static>>>;`
+- `pub fn wasm(func: &StepFunc, inputs: Rc<[IrType]>, (fields: &Fields)?) -> HQResult<Vec<InternalInstruction>>;`
 - - wasm is generated using the `wasm_gen::wasm` macro. See [its README](./wasm-gen/README.md) for usage instructions, or e.g. [say.rs](./src/instructions/looks/say.rs) for an example.
 - `pub fn acceptable_inputs() -> Rc<[IrType]>;`
 - - these should really be base types (see BASE_TYPES in [types.rs](./src/ir/types.rs))

build.rs

@@ -38,7 +38,14 @@ fn main() {
                         let fields_name =
                             format!("{}_{}_fields", category, opcode).to_case(Case::Pascal);
                         paths.push((
-                            format!("{}::{}", category, opcode),
+                            format!(
+                                "{}::{}",
+                                category,
+                                match opcode {
+                                    "yield" | "loop" => format!("r#{opcode}"),
+                                    _ => opcode.to_string(),
+                                }
+                            ),
                             format!("{}_{}", category, opcode,),
                             fields,
                             fields_name,
@@ -116,21 +123,28 @@ pub enum IrOpcode {{
 
 impl IrOpcode {{
     /// maps an opcode to its acceptable input types
-     pub fn acceptable_inputs(&self) -> Rc<[crate::ir::Type]> {{
+    pub fn acceptable_inputs(&self) -> Rc<[crate::ir::Type]> {{
         match self {{
             {}
         }}
     }}
 
     /// maps an opcode to its WASM instructions
-     pub fn wasm(&self, step_func: &crate::wasm::StepFunc, inputs: Rc<[crate::ir::Type]>) -> HQResult<Vec<wasm_encoder::Instruction<'static>>> {{
+    pub fn wasm(&self, step_func: &crate::wasm::StepFunc, inputs: Rc<[crate::ir::Type]>) -> HQResult<Vec<crate::wasm::InternalInstruction>> {{
         match self {{
             {}
         }}
     }}
     
     /// maps an opcode to its output type
-     pub fn output_type(&self, inputs: Rc<[crate::ir::Type]>) -> HQResult<Option<crate::ir::Type>> {{
+    pub fn output_type(&self, inputs: Rc<[crate::ir::Type]>) -> HQResult<Option<crate::ir::Type>> {{
+        match self {{
+            {}
+        }}
+    }}
+
+    /// does this opcode request a screen refresh (and by extension yields)?
+    pub const fn yields(&self) -> bool {{
         match self {{
             {}
         }}
@@ -170,6 +184,13 @@ pub use fields::*;
                     format!("IrOpcode::{} => {}::output_type(inputs),", id, path)
                 }
             }).collect::<Vec<_>>().join("\n\t\t\t"),
+            paths.iter().map(|(path, id, fields, _)| {
+                if *fields {
+                    format!("IrOpcode::{}(_) => {}::YIELDS,", id, path)
+                } else {
+                    format!("IrOpcode::{} => {}::YIELDS,", id, path)
+                }
+            }).collect::<Vec<_>>().join("\n\t\t\t"),
             paths.iter().filter(|(_, _, fields, _)| *fields)
             .map(|(path, _, _, fields_name)|
                 format!("pub use {}::Fields as {};", path, fields_name)

playground/views/HomeView.vue

@@ -50,6 +50,11 @@ import ProjectInput from '../components/ProjectInput.vue'
       id: 1050568991,
       name: 'fibonacci benchmark',
       author: 'pufferfish_test',
+    },
+    {
+      id: 1116438829,
+      name: 'HQ test project 2.0',
+      author: 'pufferfish_test'
     }
   ];
 </script>

src/instructions.rs

@@ -18,13 +18,13 @@ include!(concat!(env!("OUT_DIR"), "/ir-opcodes.rs"));
 mod input_switcher;
 pub use input_switcher::wrap_instruction;
 
-pub use hq::_yield::YieldMode;
+pub use hq::r#yield::YieldMode;
 
 mod prelude {
     pub(crate) use crate::ir::Type as IrType;
     pub use crate::prelude::*;
-    pub(crate) use crate::wasm::StepFunc;
-    pub use wasm_encoder::{Instruction, RefType, ValType};
+    pub(crate) use crate::wasm::{InternalInstruction, StepFunc};
+    pub use wasm_encoder::{RefType, ValType};
     pub use wasm_gen::wasm;
 
     /// Canonical NaN + bit 33, + string pointer in bits 1-32

src/instructions/control.rs

@@ -1,2 +1,2 @@
-pub mod _if;
-pub mod repeat;
+pub mod if_else;
+pub mod r#loop;

src/instructions/control/_if.rs

@@ -1,2 +1,3 @@
 pub mod setvariableto;
+pub mod teevariable;
 pub mod variable;

src/instructions/control/repeat.rs

@@ -8,7 +8,7 @@ pub fn wasm(
     func: &StepFunc,
     inputs: Rc<[IrType]>,
     Fields(variable): &Fields,
-) -> HQResult<Vec<Instruction<'static>>> {
+) -> HQResult<Vec<InternalInstruction>> {
     let global_index: u32 = func
         .registries()
         .variables()
@@ -36,6 +36,8 @@ pub fn output_type(_inputs: Rc<[IrType]>, _fields: &Fields) -> HQResult<Option<I
     Ok(None)
 }
 
+pub const YIELDS: bool = false;
+
 crate::instructions_test!(
     any;
     data_setvariableto;

src/instructions/data.rs

@@ -8,7 +8,7 @@ pub fn wasm(
     func: &StepFunc,
     _inputs: Rc<[IrType]>,
     Fields(variable): &Fields,
-) -> HQResult<Vec<Instruction<'static>>> {
+) -> HQResult<Vec<InternalInstruction>> {
     let global_index: u32 = func
         .registries()
         .variables()
@@ -28,6 +28,8 @@ pub fn output_type(_inputs: Rc<[IrType]>, Fields(rcvar): &Fields) -> HQResult<Op
     }))
 }
 
+pub const YIELDS: bool = false;
+
 crate::instructions_test!(
     any;
     data_variable;

src/instructions/data/setvariableto.rs

@@ -1,5 +1,6 @@
-pub mod _yield;
 pub mod cast;
+pub mod drop;
 pub mod float;
 pub mod integer;
 pub mod text;
+pub mod r#yield;

src/instructions/data/variable.rs

@@ -32,7 +32,7 @@ pub fn wasm(
     func: &StepFunc,
     inputs: Rc<[IrType]>,
     &Fields(to): &Fields,
-) -> HQResult<Vec<Instruction<'static>>> {
+) -> HQResult<Vec<InternalInstruction>> {
     let from = inputs[0];
 
     let target = best_cast_candidate(from, to)?;
@@ -104,6 +104,8 @@ pub fn output_type(inputs: Rc<[IrType]>, &Fields(to): &Fields) -> HQResult<Optio
     ))
 }
 
+pub const YIELDS: bool = false;
+
 crate::instructions_test! {float; hq_cast; t @ super::Fields(IrType::Float)}
 crate::instructions_test! {string; hq_cast; t @ super::Fields(IrType::String)}
 crate::instructions_test! {int; hq_cast; t @ super::Fields(IrType::QuasiInt)}

src/instructions/hq.rs

@@ -7,7 +7,7 @@ pub fn wasm(
     _func: &StepFunc,
     _inputs: Rc<[IrType]>,
     fields: &Fields,
-) -> HQResult<Vec<Instruction<'static>>> {
+) -> HQResult<Vec<InternalInstruction>> {
     Ok(wasm![F64Const(fields.0)])
 }
 
@@ -29,4 +29,6 @@ pub fn output_type(_inputs: Rc<[IrType]>, &Fields(val): &Fields) -> HQResult<Opt
     }))
 }
 
+pub const YIELDS: bool = false;
+
 crate::instructions_test! {tests; hq_float; @ super::Fields(0.0)}

src/instructions/hq/_yield.rs

@@ -7,7 +7,7 @@ pub fn wasm(
     _func: &StepFunc,
     _inputs: Rc<[IrType]>,
     fields: &Fields,
-) -> HQResult<Vec<Instruction<'static>>> {
+) -> HQResult<Vec<InternalInstruction>> {
     Ok(wasm![I32Const(fields.0)])
 }
 
@@ -24,4 +24,6 @@ pub fn output_type(_inputs: Rc<[IrType]>, &Fields(val): &Fields) -> HQResult<Opt
     }))
 }
 
+pub const YIELDS: bool = false;
+
 crate::instructions_test! {tests; hq_integer; @ super::Fields(0)}

src/instructions/hq/cast.rs

@@ -8,7 +8,7 @@ pub fn wasm(
     func: &StepFunc,
     _inputs: Rc<[IrType]>,
     fields: &Fields,
-) -> HQResult<Vec<Instruction<'static>>> {
+) -> HQResult<Vec<InternalInstruction>> {
     let string_idx = func
         .registries()
         .strings()
@@ -43,4 +43,6 @@ pub fn output_type(_inputs: Rc<[IrType]>, Fields(val): &Fields) -> HQResult<Opti
     }))
 }
 
+pub const YIELDS: bool = false;
+
 crate::instructions_test! {tests; hq_text; @ super::Fields("hello, world!".into())}