scirs2-ndimage 0.4.2

//! Functions for creating and manipulating structuring elements

use scirs2_core::ndarray::{Array, Dimension, IxDyn};

use super::Connectivity;
use crate::error::{NdimageError, NdimageResult};

/// Generate a binary structure for morphological operations
///
/// # Arguments
///
/// * `rank` - The number of dimensions of the array
/// * `connectivity` - Connectivity type (Face, FaceEdge, or Full)
///
/// # Returns
///
/// * `Result<Array<bool, IxDyn>>` - Binary structuring element
///
/// # Examples
///
/// ```
/// use scirs2_ndimage::morphology::{generate_binary_structure, Connectivity};
///
/// // Create a 2D structure with face connectivity (4-connectivity in 2D)
/// let structure = generate_binary_structure(2, Connectivity::Face).expect("Operation failed");
/// assert_eq!(structure.shape(), &[3, 3]);
/// // Center element and face neighbors
/// assert!(structure[[1, 1]]);  // Center
/// assert!(structure[[0, 1]]);  // Top
/// assert!(structure[[1, 0]]);  // Left
/// assert!(structure[[1, 2]]);  // Right
/// assert!(structure[[2, 1]]);  // Bottom
/// // Corner elements should be false for face connectivity
/// assert!(!structure[[0, 0]]);  // Top-left
/// assert!(!structure[[0, 2]]);  // Top-right
/// assert!(!structure[[2, 0]]);  // Bottom-left
/// assert!(!structure[[2, 2]]);  // Bottom-right
/// ```
#[allow(dead_code)]
pub fn generate_binary_structure(
    rank: usize,
    connectivity: Connectivity,
) -> NdimageResult<Array<bool, IxDyn>> {
    // Validate inputs
    if rank == 0 {
        return Err(NdimageError::InvalidInput(
            "Rank must be greater than 0".into(),
        ));
    }

    // Create a structure of shape (3, 3, ..., 3) with rank dimensions
    let shape = vec![3; rank];
    let mut structure = Array::<bool, IxDyn>::from_elem(IxDyn(&shape), false);

    // Center indices (1, 1, ..., 1)
    let center = vec![1; rank];

    // Set the center element to true
    structure[IxDyn(&center)] = true;

    // For each dimension, create indices that are adjacent in that dimension
    for dim in 0..rank {
        let mut lower_idx = center.clone();
        let mut upper_idx = center.clone();

        lower_idx[dim] = 0;
        upper_idx[dim] = 2;

        // Set adjacent elements to true for face connectivity
        structure[IxDyn(&lower_idx)] = true;
        structure[IxDyn(&upper_idx)] = true;
    }

    // For FaceEdge and Full connectivity, add edges and vertices
    if connectivity == Connectivity::FaceEdge || connectivity == Connectivity::Full {
        // Recursively add all combinations of indices that differ by at most 1
        // from the center in each dimension
        let mut indices = vec![1; rank];
        add_connected_indices(&mut structure, &mut indices, 0, connectivity);
    }

    Ok(structure)
}

/// Recursively add connected indices to the structure
#[allow(dead_code)]
fn add_connected_indices(
    structure: &mut Array<bool, IxDyn>,
    indices: &mut Vec<usize>,
    dim: usize,
    connectivity: Connectivity,
) {
    if dim == indices.len() {
        // Set the current indices to true
        structure[IxDyn(indices)] = true;
        return;
    }

    // Save the original value for this dimension
    let orig_val = indices[dim];

    // Try all possible values for this dimension (0, 1, 2)
    for val in 0..3 {
        indices[dim] = val;

        // Check if this combination is valid based on connectivity
        let center_dist = indices
            .iter()
            .enumerate()
            .map(|(i, &idx)| {
                if i == dim {
                    0 // Don't count the current dimension
                } else {
                    match idx {
                        0 | 2 => 1, // Distance from center (1)
                        _ => 0,     // No distance (at center)
                    }
                }
            })
            .sum::<usize>();

        let is_valid = match connectivity {
            Connectivity::Face => {
                // Only direct neighbors (at most one dimension can differ from center)
                center_dist == 0
            }
            Connectivity::FaceEdge => {
                // Neighbors and diagonal connections (at most two dimensions can differ)
                center_dist <= 1
            }
            Connectivity::Full => {
                // All elements within the 3x3x... cube
                true
            }
        };

        if is_valid {
            add_connected_indices(structure, indices, dim + 1, connectivity);
        }
    }

    // Restore the original value
    indices[dim] = orig_val;
}

/// Iterate binary erosion or dilation until convergence
///
/// # Arguments
///
/// * `structure` - Input structuring element
/// * `iterations` - Number of iterations
///
/// # Returns
///
/// * `Result<Array<bool, D>>` - Iterated structuring element
#[allow(dead_code)]
pub fn iterate_structure<D>(
    structure: &Array<bool, D>,
    iterations: usize,
) -> NdimageResult<Array<bool, D>>
where
    D: Dimension,
{
    // Validate inputs
    if structure.ndim() == 0 {
        return Err(NdimageError::InvalidInput(
            "Structure cannot be 0-dimensional".into(),
        ));
    }

    if iterations == 0 {
        return Ok(structure.to_owned());
    }

    // Placeholder implementation returning a copy of the input
    // This will be implemented with proper iteration of the structure
    Ok(structure.to_owned())
}

/// Create a box structuring element of given size
///
/// # Arguments
///
/// * `shape` - The shape of the element
///
/// # Returns
///
/// * `Result<Array<bool, IxDyn>>` - Box structuring element
#[allow(dead_code)]
pub fn box_structure(shape: &[usize]) -> NdimageResult<Array<bool, IxDyn>> {
    // Validate inputs
    if shape.is_empty() {
        return Err(NdimageError::InvalidInput("Shape cannot be empty".into()));
    }

    for &s in shape {
        if s == 0 {
            return Err(NdimageError::InvalidInput(
                "Shape dimensions must be greater than 0".into(),
            ));
        }
    }

    // Create a box of ones
    let structure = Array::<bool, IxDyn>::from_elem(IxDyn(shape), true);
    Ok(structure)
}

/// Create a disk structuring element
///
/// # Arguments
///
/// * `radius` - The radius of the disk
/// * `dimension` - The number of dimensions (default: 2)
///
/// # Returns
///
/// * `Result<Array<bool, IxDyn>>` - Disk structuring element
#[allow(dead_code)]
pub fn disk_structure(radius: f64, dimension: Option<usize>) -> NdimageResult<Array<bool, IxDyn>> {
    // Validate inputs
    if radius <= 0.0 {
        return Err(NdimageError::InvalidInput(
            "Radius must be greater than 0".into(),
        ));
    }

    let dim = dimension.unwrap_or(2);
    if dim < 2 {
        return Err(NdimageError::InvalidInput(
            "Dimension must be at least 2".into(),
        ));
    }

    // Create a disk structure by checking if each position is within the _radius
    let size = (2.0 * radius).round() as usize;
    if size % 2 == 0 {
        // Make it odd for symmetry
        let size = size + 1;
        let shape = vec![size; dim];
        let center = (size - 1) as f64 / 2.0;

        let mut structure = Array::<bool, IxDyn>::from_elem(IxDyn(&shape), false);

        // For 2D case
        if dim == 2 {
            for i in 0..size {
                for j in 0..size {
                    let dx = i as f64 - center;
                    let dy = j as f64 - center;
                    let distance = (dx * dx + dy * dy).sqrt();
                    if distance <= radius {
                        structure[[i, j]] = true;
                    }
                }
            }
        } else {
            // For higher dimensions, still create a simple box for now
            let size = (2.0 * radius.ceil() + 1.0) as usize;
            let shape = vec![size; dim];
            return box_structure(&shape);
        }

        Ok(structure)
    } else {
        let shape = vec![size; dim];
        let center = (size - 1) as f64 / 2.0;

        let mut structure = Array::<bool, IxDyn>::from_elem(IxDyn(&shape), false);

        // For 2D case
        if dim == 2 {
            for i in 0..size {
                for j in 0..size {
                    let dx = i as f64 - center;
                    let dy = j as f64 - center;
                    let distance = (dx * dx + dy * dy).sqrt();
                    if distance <= radius {
                        structure[[i, j]] = true;
                    }
                }
            }
        } else {
            // For higher dimensions, still create a simple box for now
            let size = (2.0 * radius.ceil() + 1.0) as usize;
            let shape = vec![size; dim];
            return box_structure(&shape);
        }

        Ok(structure)
    }
}

/// Generate a binary structure for morphological operations (dynamic dimension version)
///
/// Creates a default cross-shaped structure for each rank
pub(crate) fn generate_binary_structure_dyn(rank: usize) -> NdimageResult<Array<bool, IxDyn>> {
    generate_binary_structure(rank, Connectivity::Face)
}

#[cfg(test)]
mod tests {
    use super::*;
    use scirs2_core::ndarray::Array2;

    #[test]
    fn test_iterate_structure() {
        let input = Array2::from_elem((3, 3), true);
        let result = iterate_structure(&input, 1).expect("Operation failed");
        assert_eq!(result.shape(), input.shape());
    }

    #[test]
    fn test_box_structure() {
        let result = box_structure(&[3, 3]).expect("Operation failed");
        assert_eq!(result.shape(), &[3, 3]);
        assert!(result.iter().all(|&x| x));
    }

    #[test]
    fn test_disk_structure() {
        let result = disk_structure(1.5, None).expect("Operation failed");
        // For radius 1.5, we expect a 3x3 structure (2 * 1.5 = 3)
        assert_eq!(result.shape(), &[3, 3]);

        // First, let's see what we actually get
        println!("Actual disk structure:");
        let rows = result.shape()[0];
        let cols = result.shape()[1];
        for i in 0..rows {
            for j in 0..cols {
                print!("{}", if result[[i, j]] { "1" } else { "0" });
            }
            println!();
        }

        // Check that it's a reasonable disk shape
        // The center should be set
        assert!(result[[1, 1]]);

        // Check the total number of true values
        let count: usize = result.iter().filter(|&&x| x).count();
        assert!(
            count >= 5 && count <= 9,
            "Expected between 5 and 9 pixels, got {}",
            count
        );

        // Verify it forms a disk-like pattern
        // For radius 1.5 in a 3x3 grid, we expect a cross or full grid
        // Corner pixels (0,0), (0,2), (2,0), (2,2) might or might not be set
        // depending on exact distance calculation
    }
}