From f47be08d9849abc027009e6c102d983745016649 Mon Sep 17 00:00:00 2001
From: Aaron Fenyes <aaron.fenyes@fareycircles.ooo>
Date: Sun, 15 Sep 2024 11:31:22 -0700
Subject: [PATCH] Display: get the assembly from the app state

---
 app-proto/sketch-outline/src/display.rs | 35 +++++++++++--------------
 app-proto/sketch-outline/src/main.rs    |  2 +-
 2 files changed, 16 insertions(+), 21 deletions(-)

diff --git a/app-proto/sketch-outline/src/display.rs b/app-proto/sketch-outline/src/display.rs
index b1bf6ea..59a72c9 100644
--- a/app-proto/sketch-outline/src/display.rs
+++ b/app-proto/sketch-outline/src/display.rs
@@ -1,5 +1,5 @@
 use core::array;
-use nalgebra::{DMatrix, DVector, Rotation3, Vector3};
+use nalgebra::{DMatrix, Rotation3, Vector3};
 use sycamore::{prelude::*, motion::create_raf};
 use web_sys::{
     console,
@@ -11,6 +11,8 @@ use web_sys::{
     wasm_bindgen::{JsCast, JsValue}
 };
 
+use crate::AppState;
+
 fn compile_shader(
     context: &WebGl2RenderingContext,
     shader_type: u32,
@@ -81,6 +83,8 @@ fn bind_vertex_attrib(
 
 #[component]
 pub fn Display() -> View {
+    let state = use_context::<AppState>();
+    
     // canvas
     let display = create_node_ref();
     
@@ -104,12 +108,6 @@ pub fn Display() -> View {
     let mean_frame_interval = create_signal(0.0);
     
     on_mount(move || {
-        /* SCAFFOLDING */
-        /* create list of construction elements */
-        const SPHERE_MAX: usize = 200;
-        let mut sphere_vec = Vec::<DVector<f64>>::new();
-        let mut color_vec = Vec::<[f32; 3]>::new();
-        
         // timing
         let mut last_time = 0.0;
         
@@ -183,6 +181,7 @@ pub fn Display() -> View {
         );
         
         // find indices of vertex attributes and uniforms
+        const SPHERE_MAX: usize = 200;
         let position_index = ctx.get_attrib_location(&program, "position") as u32;
         let sphere_cnt_loc = ctx.get_uniform_location(&program, "sphere_cnt");
         let sphere_sp_locs = get_uniform_array_locations::<SPHERE_MAX>(
@@ -280,15 +279,11 @@ pub fn Display() -> View {
                 };
                 let construction_to_world = &location * &orientation;
                 
-                // update the construction
-                sphere_vec.clear();
-                sphere_vec.push(&construction_to_world * DVector::<f64>::from_column_slice(&[0.5, 0.5, 0.0, 0.5, -0.25]));
-                sphere_vec.push(&construction_to_world * DVector::<f64>::from_column_slice(&[-0.5, -0.5, 0.0, 0.5, -0.25]));
-                sphere_vec.push(&construction_to_world * DVector::<f64>::from_column_slice(&[0.0, 0.0, 0.0, 0.4, -0.625]));
-                color_vec.clear();
-                color_vec.push([1.00_f32, 0.25_f32, 0.00_f32]);
-                color_vec.push([0.00_f32, 0.25_f32, 1.00_f32]);
-                color_vec.push([0.75_f32, 0.75_f32, 0.75_f32]);
+                // get the construction
+                let elements = state.assembly.elements.get_clone();
+                let element_iter = (&elements).into_iter();
+                let reps_world: Vec<_> = element_iter.clone().map(|elt| &construction_to_world * &elt.rep).collect();
+                let colors: Vec<_> = element_iter.map(|elt| elt.color).collect();
                 
                 // set the resolution
                 let width = canvas.width() as f32;
@@ -297,9 +292,9 @@ pub fn Display() -> View {
                 ctx.uniform1f(shortdim_loc.as_ref(), width.min(height));
                 
                 // pass the construction
-                ctx.uniform1i(sphere_cnt_loc.as_ref(), sphere_vec.len() as i32);
-                for n in 0..sphere_vec.len() {
-                    let v = &sphere_vec[n];
+                ctx.uniform1i(sphere_cnt_loc.as_ref(), elements.len() as i32);
+                for n in 0..reps_world.len() {
+                    let v = &reps_world[n];
                     ctx.uniform3f(
                         sphere_sp_locs[n].as_ref(),
                         v[0] as f32, v[1] as f32, v[2] as f32
@@ -310,7 +305,7 @@ pub fn Display() -> View {
                     );
                     ctx.uniform3fv_with_f32_array(
                         color_locs[n].as_ref(),
-                        &color_vec[n]
+                        &colors[n]
                     );
                 }
                 
diff --git a/app-proto/sketch-outline/src/main.rs b/app-proto/sketch-outline/src/main.rs
index 6c6608f..ee7682a 100644
--- a/app-proto/sketch-outline/src/main.rs
+++ b/app-proto/sketch-outline/src/main.rs
@@ -36,7 +36,7 @@ fn main() {
                             id: String::from("central"),
                             label: String::from("Central"),
                             color: [0.75_f32, 0.75_f32, 0.75_f32],
-                            rep: DVector::<f64>::from_column_slice(&[0.0, 0.0, 0.0, 0.25, 1.0])
+                            rep: DVector::<f64>::from_column_slice(&[0.0, 0.0, 0.0, 0.4, -0.625])
                         }
                     ])
                 }