mirror of https://github.com/aziis98/asd-2024.git
added another example sub-project
parent
64cdb36292
commit
42fe73d0ae
@ -0,0 +1,18 @@
|
|||||||
|
[package]
|
||||||
|
name = "fdg-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
nalgebra = { version = "0.32.3", features = ["rand"] }
|
||||||
|
petgraph = { version = "0.6.4", features = [
|
||||||
|
"stable_graph",
|
||||||
|
], default-features = false }
|
||||||
|
num-traits = "0.2.17"
|
||||||
|
rand = "0.8.5"
|
||||||
|
rustc-hash = "1.1.0"
|
||||||
|
rayon = { version = "1.8.0", optional = true }
|
||||||
|
macroquad = "0.4.4"
|
||||||
|
petgraph-gen = "0.1.3"
|
||||||
|
fdg = { git = "https://github.com/grantshandy/fdg" }
|
||||||
|
asd = { path = "../../" }
|
||||||
@ -0,0 +1,147 @@
|
|||||||
|
use std::{collections::HashMap, env};
|
||||||
|
|
||||||
|
use asd::{
|
||||||
|
gfa::{Entry, Orientation},
|
||||||
|
parser,
|
||||||
|
};
|
||||||
|
use fdg::{
|
||||||
|
fruchterman_reingold::{FruchtermanReingold, FruchtermanReingoldConfiguration},
|
||||||
|
nalgebra::Rotation2,
|
||||||
|
petgraph::Graph,
|
||||||
|
simple::Center,
|
||||||
|
Force, ForceGraph,
|
||||||
|
};
|
||||||
|
|
||||||
|
use macroquad::prelude::*;
|
||||||
|
|
||||||
|
#[macroquad::main("fdg demo")]
|
||||||
|
async fn main() {
|
||||||
|
let mut graph = Graph::<(String, Orientation), ()>::new();
|
||||||
|
|
||||||
|
let file = std::fs::File::open(env::args().nth(1).expect("missing gfa file argument")).unwrap();
|
||||||
|
let entries = parser::parse_source(file).unwrap();
|
||||||
|
|
||||||
|
let mut index_map = HashMap::new();
|
||||||
|
|
||||||
|
for entry in entries {
|
||||||
|
println!("{:?}", entry);
|
||||||
|
|
||||||
|
if let Entry::Link {
|
||||||
|
from,
|
||||||
|
from_orient,
|
||||||
|
to,
|
||||||
|
to_orient,
|
||||||
|
} = entry
|
||||||
|
{
|
||||||
|
// add first node if not present
|
||||||
|
let a = index_map
|
||||||
|
.entry(from.clone())
|
||||||
|
.or_insert_with(|| graph.add_node((from.clone(), from_orient)))
|
||||||
|
.to_owned();
|
||||||
|
|
||||||
|
// add second node if not present
|
||||||
|
let b = index_map
|
||||||
|
.entry(to.clone())
|
||||||
|
.or_insert_with(|| graph.add_node((to.clone(), to_orient)))
|
||||||
|
.to_owned();
|
||||||
|
|
||||||
|
graph.add_edge(a, b, ());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut force_graph: ForceGraph<f32, 2, (String, Orientation), ()> =
|
||||||
|
fdg::init_force_graph_uniform(graph, 200.0);
|
||||||
|
|
||||||
|
// custom closure force which rotates each node
|
||||||
|
// let mut rotate = |graph: &mut ForceGraph<f32, 2, (String, Orientation), ()>| {
|
||||||
|
// graph
|
||||||
|
// .node_weights_mut()
|
||||||
|
// .for_each(|(_, p)| *p = Rotation2::new(0.005).transform_point(p))
|
||||||
|
// };
|
||||||
|
let mut force = FruchtermanReingold {
|
||||||
|
conf: FruchtermanReingoldConfiguration {
|
||||||
|
scale: 400.0,
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
loop {
|
||||||
|
println!("frame");
|
||||||
|
force.apply_many(&mut force_graph, 1);
|
||||||
|
|
||||||
|
Center::default().apply(&mut force_graph);
|
||||||
|
// rotate.apply(&mut force_graph);
|
||||||
|
|
||||||
|
let scale = calculate_scale(&force_graph);
|
||||||
|
|
||||||
|
clear_background(WHITE);
|
||||||
|
|
||||||
|
for idx in force_graph.edge_indices() {
|
||||||
|
let ((_, source), (_, target)) = force_graph
|
||||||
|
.edge_endpoints(idx)
|
||||||
|
.map(|(a, b)| {
|
||||||
|
(
|
||||||
|
force_graph.node_weight(a).unwrap(),
|
||||||
|
force_graph.node_weight(b).unwrap(),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
draw_line(
|
||||||
|
translate_x(source.coords.column(0)[0], scale),
|
||||||
|
translate_y(source.coords.column(0)[1], scale),
|
||||||
|
translate_x(target.coords.column(0)[0], scale),
|
||||||
|
translate_y(target.coords.column(0)[1], scale),
|
||||||
|
4.0,
|
||||||
|
BLACK,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
for ((name, orient), pos) in force_graph.node_weights() {
|
||||||
|
let x = translate_x(pos.coords.column(0)[0], scale);
|
||||||
|
let y = translate_y(pos.coords.column(0)[1], scale);
|
||||||
|
|
||||||
|
draw_circle(x, y, 20.0 * scale, RED);
|
||||||
|
draw_text(
|
||||||
|
format!("{}{}", name, orient).as_str(),
|
||||||
|
x - 30.0 * scale,
|
||||||
|
y - 30.0 * scale,
|
||||||
|
40.0 * scale,
|
||||||
|
BLACK,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
next_frame().await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn translate_x(x: f32, scale: f32) -> f32 {
|
||||||
|
(screen_width() / 2.0) + (x * scale)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn translate_y(y: f32, scale: f32) -> f32 {
|
||||||
|
(screen_height() / 2.0) + (y * scale)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn calculate_scale(graph: &ForceGraph<f32, 2, (String, Orientation), ()>) -> f32 {
|
||||||
|
let (min_x, max_x, min_y, max_y) = graph.node_weights().fold(
|
||||||
|
(f32::MAX, f32::MIN, f32::MAX, f32::MIN),
|
||||||
|
|(min_x, max_x, min_y, max_y), (_, pos)| {
|
||||||
|
(
|
||||||
|
min_x.min(pos.coords.column(0)[0]),
|
||||||
|
max_x.max(pos.coords.column(0)[0]),
|
||||||
|
min_y.min(pos.coords.column(0)[1]),
|
||||||
|
max_y.max(pos.coords.column(0)[1]),
|
||||||
|
)
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
let graph_width = max_x - min_x;
|
||||||
|
let graph_height = max_y - min_y;
|
||||||
|
|
||||||
|
let scale_x = screen_width() / graph_width;
|
||||||
|
let scale_y = screen_height() / graph_height;
|
||||||
|
|
||||||
|
scale_x.min(scale_y) * 0.9 // add some padding
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue