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

use snafu::OptionExt;

use crate::arrow_reader::column::present::new_present_iter;
use crate::arrow_reader::column::{Column, NullableIterator};
use crate::error::{InvalidColumnSnafu, Result};
use crate::proto::stream::Kind;
use crate::reader::decode::float::FloatIter;

macro_rules! impl_float_iter {
    ($tp:ident) => {
        paste::item! {
            pub fn [<new_ $tp _iter>] (column: &Column) -> Result<NullableIterator<$tp>> {
                let present = new_present_iter(column)?.try_collect::<Vec<_>>()?;
                let rows: usize = present.iter().filter(|&p| *p).count();
                let iter = column
                    .stream(Kind::Data)
                    .transpose()?
                    .map(|reader| Box::new(FloatIter::<$tp, _>::new(reader, rows)))
                    .context(InvalidColumnSnafu { name: &column.name })?;

                Ok(NullableIterator {
                    present: Box::new(present.into_iter()),
                    iter,
                })
            }
        }
    };
}

impl_float_iter!(f32);
impl_float_iter!(f64);