From 4f40617568d1c3a9fadff855571b66d1db0f5c02 Mon Sep 17 00:00:00 2001
From: Antonio De Lucreziis <antonio.delucreziis@gmail.com>
Date: Sat, 25 May 2024 14:38:10 +0200
Subject: [PATCH] status info

---
 examples/graphs-1/src/main.rs | 152 +++++++++++++++++++---------------
 1 file changed, 84 insertions(+), 68 deletions(-)

diff --git a/examples/graphs-1/src/main.rs b/examples/graphs-1/src/main.rs
index 7987d8c..d2abcee 100644
--- a/examples/graphs-1/src/main.rs
+++ b/examples/graphs-1/src/main.rs
@@ -1,9 +1,9 @@
 use std::{collections::HashMap, env, ops::AddAssign, time::Instant};
 
 use asd::{gfa::Entry, parser};
-use macroquad::{prelude::*, rand};
+use macroquad::{prelude::*, rand, ui::root_ui};
 use nalgebra::{Point2, SVector};
-use petgraph::{algo::dijkstra, stable_graph::StableGraph};
+use petgraph::{algo::dijkstra, graph::NodeIndex, stable_graph::StableGraph};
 
 use rayon::prelude::*;
 
@@ -24,88 +24,104 @@ async fn main() {
     loop {
         // Update
 
-        for _ in 1..2 {
-            let forces = graph
-                .node_indices()
-                .par_bridge()
-                .map(|idx| {
-                    (
-                        idx,
-                        desired_distance_matrix
-                            .get(&idx)
-                            .unwrap()
-                            .par_iter()
-                            .map(|(&other_idx, &distance)| {
-                                let pos = graph.node_weight(idx).unwrap().1;
-                                let other_pos = graph.node_weight(other_idx).unwrap().1;
-                                let delta = other_pos - pos;
-                                let dist = delta.norm_squared();
-                                let correction = dist - (distance * distance);
-
-                                // println!("correction: {:?}", correction);
-
-                                if distance > 0.0 && dist > 1e-6 {
-                                    0.01 * delta.normalize() * correction.atan()
-                                } else {
-                                    SVector::<f32, 2>::zeros()
-                                }
-                            })
-                            .sum::<SVector<f32, 2>>()
-                            + {
-                                let pos = graph.node_weight(idx).unwrap().1;
-
-                                let delta = pos - Point2::new(0.0, 0.0);
+        let start_update = Instant::now();
 
-                                let dist = delta.norm_squared();
+        update(&mut graph, &desired_distance_matrix);
 
-                                if dist > 1e-6 {
-                                    -0.01 * delta.normalize() / dist.max(1.0)
-                                } else {
-                                    SVector::<f32, 2>::zeros()
-                                }
-                            }
-                            + graph
-                                .node_indices()
-                                .par_bridge()
-                                .filter(|&other_idx| other_idx != idx)
-                                .map(|other_idx| {
-                                    let pos = graph.node_weight(idx).unwrap().1;
-                                    let other_pos = graph.node_weight(other_idx).unwrap().1;
-                                    let delta = other_pos - pos;
-                                    let dist = delta.norm();
-                                    if dist > 1e-3 {
-                                        -0.001 * delta.normalize() / (dist * dist)
-                                    } else {
-                                        SVector::<f32, 2>::zeros()
-                                    }
-                                })
-                                .sum::<SVector<f32, 2>>(),
-                    )
-                })
-                .collect::<HashMap<_, _>>();
-
-            forces.iter().for_each(|(idx, force)| {
-                let (_, pos) = graph.node_weight_mut(*idx).unwrap();
-                pos.add_assign(force);
-            });
-        }
+        let update_elapsed = start_update.elapsed();
 
         // Render
 
-        let now = Instant::now();
+        let start_render = Instant::now();
 
         clear_background(WHITE);
 
         draw_graph(&graph);
 
-        let elapsed = now.elapsed();
+        let render_elapsed = start_render.elapsed();
 
-        println!("frame: {:?}", elapsed);
+        root_ui().label(
+            None,
+            format!("update: {:?}, render: {:?}", update_elapsed, render_elapsed).as_str(),
+        );
+
+        // println!("update: {:?}, render: {:?}", update_elapsed, render_elapsed);
 
         next_frame().await
     }
 }
 
+fn update(
+    graph: &mut StableGraph<(String, Point2<f32>), ()>,
+    desired_distance_matrix: &HashMap<NodeIndex, HashMap<NodeIndex, f32>>,
+) {
+    for _ in 1..2 {
+        let forces = graph
+            .node_indices()
+            .par_bridge()
+            .map(|idx| {
+                (
+                    idx,
+                    desired_distance_matrix
+                        .get(&idx)
+                        .unwrap()
+                        .par_iter()
+                        .map(|(&other_idx, &distance)| {
+                            let pos = graph.node_weight(idx).unwrap().1;
+                            let other_pos = graph.node_weight(other_idx).unwrap().1;
+                            let delta = other_pos - pos;
+                            let dist = delta.norm_squared();
+                            let correction = dist - (distance * distance);
+
+                            // println!("correction: {:?}", correction);
+
+                            if distance > 0.0 && dist > 1e-6 {
+                                0.01 * delta.normalize() * correction.atan()
+                            } else {
+                                SVector::<f32, 2>::zeros()
+                            }
+                        })
+                        .sum::<SVector<f32, 2>>()
+                        + {
+                            let pos = graph.node_weight(idx).unwrap().1;
+
+                            let delta = pos - Point2::new(0.0, 0.0);
+
+                            let dist = delta.norm_squared();
+
+                            if dist > 1e-6 {
+                                -0.01 * delta.normalize() / dist.max(1.0)
+                            } else {
+                                SVector::<f32, 2>::zeros()
+                            }
+                        }
+                        + graph
+                            .node_indices()
+                            .par_bridge()
+                            .filter(|&other_idx| other_idx != idx)
+                            .map(|other_idx| {
+                                let pos = graph.node_weight(idx).unwrap().1;
+                                let other_pos = graph.node_weight(other_idx).unwrap().1;
+                                let delta = other_pos - pos;
+                                let dist = delta.norm();
+                                if dist > 1e-3 {
+                                    -0.001 * delta.normalize() / (dist * dist)
+                                } else {
+                                    SVector::<f32, 2>::zeros()
+                                }
+                            })
+                            .sum::<SVector<f32, 2>>(),
+                )
+            })
+            .collect::<HashMap<_, _>>();
+
+        forces.iter().for_each(|(idx, force)| {
+            let (_, pos) = graph.node_weight_mut(*idx).unwrap();
+            pos.add_assign(force);
+        });
+    }
+}
+
 fn load_graph() -> StableGraph<(String, Point2<f32>), ()> {
     println!("Loading graph");