fixed unsorted triangle bug, back to area z-score

This commit is contained in:
randogoth 2024-03-18 10:26:05 +02:00
parent a57e095e1a
commit 2bf2e1293f

View file

@ -141,23 +141,22 @@ fn mean_std(dataset: Vec<f32>) -> (f32, f32) {
fn delfin( fn delfin(
geometry_data: &GeometryData, geometry_data: &GeometryData,
min_voidness: f32, min_area: f32,
min_distance: f32, min_distance: f32,
) -> Vec<HashSet<usize>> { ) -> Vec<HashSet<usize>> {
// Sort all triangles by the longest terminal edge // Sort all triangles by the longest terminal edge
let triangles_sorted: Vec<usize> = geometry_data.triangles.iter() let triangles_sorted: Vec<(usize, f32)> = geometry_data.triangles.iter()
.map(|triangle_data: &TriangleData| { .map(|triangle_data: &TriangleData| {
let terminal_edge_length: f32 = geometry_data.edge_lengths[&triangle_data.terminal_edge]; let terminal_edge_length: f32 = geometry_data.edge_lengths[&triangle_data.terminal_edge];
(triangle_data.index, terminal_edge_length) (triangle_data.index, terminal_edge_length)
}) })
.sorted_by(|a, b| b.1.partial_cmp(&a.1).unwrap()) // Sort in descending order by edge length .sorted_by(|a, b| b.1.partial_cmp(&a.1).unwrap()) // Sort in descending order by edge length
.map(|(index, _)| index) // Extract triangle indices
.collect(); .collect();
// Calculate densities based on reverse area // Calculate densities based on reverse area
let densities: Vec<f32> = geometry_data.triangles.par_iter() let areas: Vec<f32> = geometry_data.triangles.par_iter()
.map(|triangle_data: &TriangleData| 1.0 / triangle_data.area) .map(|triangle_data: &TriangleData| triangle_data.area)
.collect(); .collect();
// Calculate mean and standard deviation of terminal edges lengths // Calculate mean and standard deviation of terminal edges lengths
@ -166,32 +165,27 @@ fn delfin(
.collect(); .collect();
let (mean_terminal_edge, std_terminal_edge) = mean_std(terminal_edge_lengths); let (mean_terminal_edge, std_terminal_edge) = mean_std(terminal_edge_lengths);
let (mean_density, std_density) = mean_std(densities); let (mean_area, std_area) = mean_std(areas);
let mut void_polygons: Vec<HashSet<usize>> = Vec::new(); let mut void_polygons: Vec<HashSet<usize>> = Vec::new();
let mut processed_triangles: HashSet<usize> = HashSet::new(); let mut processed_triangles: HashSet<usize> = HashSet::new();
for &triangle_index in &triangles_sorted { for &(triangle_index, terminal_edge_length) in &triangles_sorted {
// Skip if this triangle has already been processed // Skip if this triangle has already been processed
if processed_triangles.contains(&triangle_index) { if processed_triangles.contains(&triangle_index) {
continue; continue;
} }
// Retrieve the terminal edge for the current triangle
let triangle_data: &TriangleData = &geometry_data.triangles[triangle_index];
let terminal_edge: Edge = triangle_data.terminal_edge;
// Calculate the Z-score for the terminal edge length // Calculate the Z-score for the terminal edge length
let terminal_edge_length: f32 = geometry_data.edge_lengths[&terminal_edge]; let distance_z_score: f32 = (terminal_edge_length - mean_terminal_edge) / std_terminal_edge;
let z_score: f32 = (terminal_edge_length - mean_terminal_edge) / std_terminal_edge;
// println!("Terminal Length: {}", terminal_edge_length);
// println!("Terminal Length Z-Score: {}", z_score);
// Continue if the Z-score is below the minimum distance threshold // Continue if the Z-score is below the minimum distance threshold
if z_score < min_distance { if distance_z_score < min_distance {
continue; continue;
} }
// Retrieve triangles that share the terminal edge, continue if less than 2 triangles share it // Retrieve triangles that share the terminal edge, continue if less than 2 triangles share it
let triangle_data: &TriangleData = &geometry_data.triangles[triangle_index];
let terminal_edge: Edge = triangle_data.terminal_edge;
if let Some(connected_triangles) = geometry_data.edge_to_triangles.get(&terminal_edge) { if let Some(connected_triangles) = geometry_data.edge_to_triangles.get(&terminal_edge) {
if connected_triangles.len() < 2 { if connected_triangles.len() < 2 {
continue; continue;
@ -239,15 +233,11 @@ fn delfin(
.map(|triangle_data: &TriangleData| triangle_data.area) .map(|triangle_data: &TriangleData| triangle_data.area)
.sum(); .sum();
// Calculate the polygon density // Calculate the area Z-score
let polygon_density: f32 = if total_area > 0.0 { 1.0 / total_area } else { 0.0 }; let area_z_score: f32 = (total_area - mean_area) / std_area;
// Calculate the density Z-score // Filter based on the area Z-score and the minimum number of triangles
let density_z_score: f32 = (polygon_density - mean_density) / std_density; area_z_score >= min_area && poly_set.len() >= 3
// println!("Density Z-Score: {}", density_z_score.abs());
// Filter based on the density Z-score and the minimum number of triangles
density_z_score.abs() >= min_voidness && poly_set.len() >= 3
}); });
return void_polygons; return void_polygons;
@ -256,18 +246,17 @@ fn delfin(
fn main() { fn main() {
let points: Vec<Point<f32>> = random_points((0.0, 0.0), 1000.0, 10000); let points: Vec<Point<f32>> = random_points((0.0, 0.0), 1000.0, 10000);
let triangles_indices: Vec<usize> = delaunay(&points); let triangles_indices: Vec<usize> = delaunay(&points);
// println!("{:?}", triangles_indices);
// Preprocess to create GeometryData // Preprocess to create GeometryData
let geometry_data: GeometryData = preprocess(&points, &triangles_indices); let geometry_data: GeometryData = preprocess(&points, &triangles_indices);
// Define minimum voidness and minimum distance for delfin function // Define minimum voidness and minimum distance for delfin function
let min_voidness: f32 = 0.2; // Example threshold for voidness let min_area: f32 = 4.0; // Example threshold for voidness
let min_distance: f32 = 0.0; // Example threshold for minimum distance (Z-score) let min_distance: f32 = 1.0; // Example threshold for minimum distance (Z-score)
// Execute delfin function with the generated GeometryData // Execute delfin function with the generated GeometryData
let void_polygons: Vec<HashSet<usize>> = delfin(&geometry_data, min_voidness, min_distance); let void_polygons: Vec<HashSet<usize>> = delfin(&geometry_data, min_area, min_distance);
// To display the result, let's just print the count of void polygons found // To display the result, let's just print the count of void polygons found
println!("Void Polygons Found: {}", void_polygons.len()); println!("Void Polygons Found: {:?}", void_polygons.len());
} }