fits-io 0.1.0

A pure-Rust FITS file handling library inspired by CFITSIO, focused on safety, clarity, and performance.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use crate::hdu::hdu::HDU;
use crate::header::{BayerPattern, ImageType};
use crate::image::Image;
use alloc::boxed::Box;
#[cfg(feature = "image")]
use image::{ImageBuffer, Luma, Primitive};
use std::error::Error;
use std::fmt;
use std::prelude::rust_2015::Vec;

pub trait ImageHDU: HDU + fmt::Debug + Send + Sync {
    fn image_count(&self) -> usize;
    fn images_width(&self) -> u32;
    fn images_height(&self) -> u32;
    fn images_bayer_pattern(&self) -> Option<BayerPattern>;
    fn images_type(&self) -> Option<&ImageType>;
    fn images_exposure_time(&self) -> Option<core::time::Duration>;
    fn read_image(&self, index: usize) -> Result<Option<Image>, Box<dyn Error + Send + Sync>>;

    #[cfg(feature = "image")]
    fn set_images_u8(
        &mut self,
        images: &[&ImageBuffer<Luma<u8>, Vec<u8>>],
    ) -> Result<(), Box<dyn Error + Send + Sync>> {
        get_raw_data_from_image(self, images, Self::set_raw_images_u8)
    }
    #[cfg(feature = "image")]
    fn set_images_i16(
        &mut self,
        images: &[&ImageBuffer<Luma<i16>, Vec<i16>>],
    ) -> Result<(), Box<dyn Error + Send + Sync>> {
        get_raw_data_from_image(self, images, Self::set_raw_images_i16)
    }
    #[cfg(feature = "image")]
    fn set_images_i32(
        &mut self,
        images: &[&ImageBuffer<Luma<i32>, Vec<i32>>],
    ) -> Result<(), Box<dyn Error + Send + Sync>> {
        get_raw_data_from_image(self, images, Self::set_raw_images_i32)
    }
    #[cfg(feature = "image")]
    fn set_images_f32(
        &mut self,
        images: &[&ImageBuffer<Luma<f32>, Vec<f32>>],
    ) -> Result<(), Box<dyn Error + Send + Sync>> {
        get_raw_data_from_image(self, images, Self::set_raw_images_f32)
    }
    #[cfg(feature = "image")]
    fn set_images_f64(
        &mut self,
        images: &[&ImageBuffer<Luma<f64>, Vec<f64>>],
    ) -> Result<(), Box<dyn Error + Send + Sync>> {
        get_raw_data_from_image(self, images, Self::set_raw_images_f64)
    }

    fn clear_images(&mut self) -> Result<(), Box<dyn Error + Send + Sync>>;

    fn set_raw_images_u8(
        &mut self,
        width: u32,
        height: u32,
        images: &[&[u8]],
    ) -> Result<(), Box<dyn Error + Send + Sync>>;
    fn set_raw_images_i16(
        &mut self,
        width: u32,
        height: u32,
        images: &[&[i16]],
    ) -> Result<(), Box<dyn Error + Send + Sync>>;
    fn set_raw_images_i32(
        &mut self,
        width: u32,
        height: u32,
        images: &[&[i32]],
    ) -> Result<(), Box<dyn Error + Send + Sync>>;
    fn set_raw_images_f32(
        &mut self,
        width: u32,
        height: u32,
        images: &[&[f32]],
    ) -> Result<(), Box<dyn Error + Send + Sync>>;
    fn set_raw_images_f64(
        &mut self,
        width: u32,
        height: u32,
        images: &[&[f64]],
    ) -> Result<(), Box<dyn Error + Send + Sync>>;

    #[cfg(feature = "tokio")]
    fn stream_normalised_image(
        &self,
        index: usize,
    ) -> Result<Option<futures::stream::BoxStream<(u32, u32, f64)>>, Box<dyn Error + Send + Sync>>;
    fn image_data_size(&self) -> u64;
}

fn get_raw_data_from_image<
    'a,
    T: Primitive,
    S: ImageHDU + ?Sized,
    CB: FnOnce(&mut S, u32, u32, &[&[T]]) -> Result<(), Box<dyn Error + Send + Sync>>,
>(
    hdu: &mut S,
    images: &'a [&'a ImageBuffer<Luma<T>, Vec<T>>],
    callback: CB,
) -> Result<(), Box<dyn Error + Send + Sync>> {
    if images.is_empty() {
        hdu.clear_images()?;
        Ok(())
    } else {
        let width = images[0].width();
        let height = images[0].width();

        let data = images
            .into_iter()
            .map(|image| image.iter().as_slice())
            .collect::<Vec<_>>();

        callback(hdu, width, height, &data)
    }
}