dicom-core 0.9.1

//! Helper module for handling pixel encapsulation into fragments
use crate::value::{InMemFragment, PixelFragmentSequence, C};

/// Represents the fragments of a single frame.
///
/// A [`PixelFragmentSequence`] can be generated from a list of [`Fragments`].
/// In case of multi-frame, a list of frames composed by 1 fragment is expected.
///
/// The frames can be independently processed, so parallel execution is possible.
///
/// # Example
/// ```
/// use dicom_core::{DataElement, Tag};
/// use dicom_core::header::EmptyObject;
/// use dicom_core::value::Value::PixelSequence;
/// use dicom_core::value::fragments::Fragments;
/// use dicom_core::value::InMemFragment;
/// use dicom_core::VR::OB;
///
/// // Frames are represented as Vec<Vec<u8>>
/// // Single 512x512 frame
/// let frames = vec![vec![0; 262144]];
/// let fragments = frames
///     .into_iter()
///     .map(|frame| Fragments::new(frame, 0))
///     .collect::<Vec<Fragments>>();
///
/// let element = DataElement::new(
///     Tag(0x7FE0, 0x0008),
///     OB,
///     PixelSequence::<EmptyObject, InMemFragment>(fragments.into())
/// );
/// ```
///
/// From this last example, it is possible to extend it to implement a pipeline, and even use rayon
/// for parallel processing of the frames.
#[derive(Debug)]
pub struct Fragments {
    fragments: Vec<InMemFragment>,
}

impl Fragments {
    pub fn new(data: Vec<u8>, fragment_size: u32) -> Self {
        let fragment_size: u32 = if fragment_size == 0 {
            data.len() as u32
        } else {
            fragment_size
        };

        let fragment_size = if fragment_size % 2 == 0 {
            fragment_size
        } else {
            fragment_size + 1
        };

        let number_of_fragments = (data.len() as f32 / fragment_size as f32).ceil() as u32;

        // Calculate the encapsulated size. If necessary pad the vector with zeroes so all the
        // chunks have the same fragment_size
        let mut data = data;
        let encapsulated_size = (fragment_size * number_of_fragments) as usize;
        if encapsulated_size > data.len() {
            data.resize(encapsulated_size, 0);
        }

        let fragments = data
            .chunks_exact(fragment_size as usize)
            .map(|fragment| fragment.to_vec())
            .collect::<Vec<InMemFragment>>();

        Fragments { fragments }
    }

    pub fn is_empty(&self) -> bool {
        self.fragments.len() == 0
    }

    pub fn is_multiframe(&self) -> bool {
        self.fragments.len() > 1
    }

    pub fn len(&self) -> u32 {
        self.fragments
            .iter()
            .fold(0u32, |acc, fragment| acc + fragment.len() as u32 + 8u32)
    }
}

impl From<Vec<Fragments>> for PixelFragmentSequence<InMemFragment> {
    fn from(value: Vec<Fragments>) -> Self {
        if value.is_empty() {
            // return an empty fragment list
            return PixelFragmentSequence {
                offset_table: C::new(),
                fragments: C::new(),
            };
        }

        let mut offset_table = C::with_capacity(value.len() + 1);
        offset_table.push(0u32);
        let mut current_offset = 0u32;

        let mut fragments = Vec::new();
        let is_multiframe = value.len() > 1;
        let last_frame = value.len() - 1;

        for (index, mut frame) in value.into_iter().enumerate() {
            if frame.is_multiframe() && is_multiframe {
                panic!("More than 1 fragment per frame is invalid for multi frame pixel data");
            }

            if index < last_frame {
                let offset = frame.len();
                offset_table.push(current_offset + offset);
                current_offset += offset;
            }

            fragments.append(&mut frame.fragments);
        }

        PixelFragmentSequence {
            offset_table,
            fragments: C::from_vec(fragments),
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::value::fragments::Fragments;
    use crate::value::{InMemFragment, PixelFragmentSequence};

    #[test]
    fn test_fragment_frame() {
        let fragment = Fragments::new(vec![150, 164, 200], 0);
        assert_eq!(fragment.fragments.len(), 1, "1 fragment should be present");
        assert_eq!(
            fragment.fragments[0].len(),
            4,
            "The fragment size should be 4"
        );
        assert_eq!(
            fragment.fragments[0],
            vec![150, 164, 200, 0],
            "The data should be 0 padded"
        );

        let fragment = Fragments::new(vec![150, 164, 200, 222], 4);
        assert_eq!(fragment.fragments.len(), 1, "1 fragment should be present");
        assert_eq!(
            fragment.fragments[0].len(),
            4,
            "The fragment size should be 4"
        );
        assert_eq!(
            fragment.fragments[0],
            vec![150, 164, 200, 222],
            "The data should be what was sent"
        );

        let fragment = Fragments::new(vec![150, 164, 200, 222], 2);
        assert_eq!(fragment.fragments.len(), 2, "2 fragments should be present");
        assert_eq!(fragment.fragments[0].len(), 2);
        assert_eq!(fragment.fragments[1].len(), 2);
        assert_eq!(fragment.fragments[0], vec![150, 164]);
        assert_eq!(fragment.fragments[1], vec![200, 222]);

        let fragment = Fragments::new(vec![150, 164, 200], 1);
        assert_eq!(
            fragment.fragments.len(),
            2,
            "2 fragments should be present as fragment_size < 2"
        );
        assert_eq!(fragment.fragments[0].len(), 2);
        assert_eq!(fragment.fragments[0], vec![150, 164]);
        assert_eq!(fragment.fragments[1].len(), 2);
        assert_eq!(fragment.fragments[1], vec![200, 0]);

        let fragment = Fragments::new(vec![150, 164, 200, 222], 1);
        assert_eq!(
            fragment.fragments.len(),
            2,
            "2 fragments should be present as fragment_size < 2"
        );
        assert_eq!(fragment.fragments[0].len(), 2);
        assert_eq!(fragment.fragments[0], vec![150, 164]);
        assert_eq!(fragment.fragments[1].len(), 2);
        assert_eq!(fragment.fragments[1], vec![200, 222]);
    }

    #[test]
    fn test_bot_single_fragment_generation() {
        let data = vec![Fragments::new(vec![0u8; 2], 2)];
        let fragment_sequence: PixelFragmentSequence<InMemFragment> = data.into();
        assert_eq!(fragment_sequence.offset_table.len(), 1);
        assert_eq!(fragment_sequence.offset_table[0], 0);
    }

    #[test]
    fn test_bot_multi_fragments_generation() {
        let data = vec![Fragments::new(vec![0u8; 4], 2)];
        let fragment_sequence: PixelFragmentSequence<InMemFragment> = data.into();
        assert_eq!(fragment_sequence.offset_table.len(), 1);
        assert_eq!(fragment_sequence.offset_table[0], 0);
    }

    #[test]
    fn test_bot_multi_frame_generation() {
        let data = vec![
            Fragments::new(vec![0u8; 4], 0),
            Fragments::new(vec![1u8; 6], 0),
        ];
        let fragment_sequence: PixelFragmentSequence<InMemFragment> = data.into();
        assert_eq!(fragment_sequence.offset_table.len(), 2);
        assert_eq!(fragment_sequence.offset_table[0], 0);
        assert_eq!(fragment_sequence.offset_table[1], 12); // 8 separator bytes + 4 data bytes
    }
}