Simmer/Waywrite
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Partially completed input conversion to matrix

Browse Source
This commit is contained in:
Ethan Simmons
2024-02-18 01:44:14 -06:00
parent 6825f69118
commit b9e8521b19
4 changed files with 34 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/process_point.rs
View File

@@ -8,7 +8,9 @@ fn process(points: Vec<Point>) -> NdTensor<f32, 3> {
NdTensor::zeros([1, 1, 1])
}
pub fn to_matrix(points: &Vec<Point>) -> Vec<Box<[f64; 1000]>>{
pub fn to_matrix(points: &Vec<Point>) -> Vec<Box<[f64; 200]>>{
const MATRIX_SIZE: f64 = 200.0;
let min_x = points.iter().min_by_key(|p| p.x as i32).unwrap().x;
let min_y = points.iter().min_by_key(|p| p.y as i32).unwrap().y;
@@ -18,27 +20,28 @@ pub fn to_matrix(points: &Vec<Point>) -> Vec<Box<[f64; 1000]>>{
let x_len = max_x - min_x;
let y_len = max_y - min_y;
let x_scale = 800.0 / x_len;
let y_scale = (y_len / x_len) * x_scale;
let y_ratio = y_len / x_len;
let x_scale = (0.8 * MATRIX_SIZE) / x_len;
let y_scale = ((0.8 * MATRIX_SIZE) * y_ratio) / y_len;
let scaled_points = points.iter().map(|point| {
let x_scaled = ((point.x - min_x) * x_scale) + 100.0;
let y_scaled = ((point.y - min_y) * x_scale) + (100.0 * y_scale);
let x_scaled = ((point.x - min_x) * x_scale) + (0.1 * MATRIX_SIZE);
let y_scaled = ((point.y - min_y) * y_scale) + ((0.1 * MATRIX_SIZE) * y_ratio);
((x_scaled, y_scaled), point.new_line)
});
let mut matrix: Vec<Box<[f64; 1000]>> = vec![Box::new([-0.0; 1000]); (y_len * y_scale) as u32 as usize];
let mut matrix: Vec<Box<[f64; MATRIX_SIZE as usize]>> = vec![Box::new([0.0; MATRIX_SIZE as u32 as usize]); (MATRIX_SIZE * y_ratio) as u64 as usize];
for ((x, y), newline) in scaled_points {
let start_x = x - 1.5;
let end_x = x + 1.5;
let start_y = y - 1.5;
let end_y = y + 1.5;
let start_x = x - (MATRIX_SIZE / 100.0);
let end_x = x + (MATRIX_SIZE / 100.0);
let start_y = y - ((MATRIX_SIZE / 100.0) * y_ratio);
let end_y = y + ((MATRIX_SIZE / 100.0) * y_ratio);
matrix.iter_mut().enumerate().for_each(|(y, line)| line.iter_mut().for_each(|x| {
if (start_x < *x && *x < end_x) && (start_y < (y as f64) && (y as f64) < end_y) {
*x = 1.0;
matrix.iter_mut().enumerate().for_each(|(mat_y, line)| line.iter_mut().enumerate().for_each(|(mat_x, val)| {
if (start_x < (mat_x as f64) && (mat_x as f64) < end_x) && (start_y < (mat_y as f64) && (mat_y as f64) < end_y) {
*val = 1.0;
}
}));
}