1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use vortex_error::{vortex_err, VortexResult};

use crate::{Array, ArrayDType};

pub trait FillForwardFn {
    fn fill_forward(&self) -> VortexResult<Array>;
}

pub fn fill_forward(array: &Array) -> VortexResult<Array> {
    if !array.dtype().is_nullable() {
        return Ok(array.clone());
    }

    array.with_dyn(|a| {
        a.fill_forward()
            .map(|t| t.fill_forward())
            .unwrap_or_else(|| {
                Err(vortex_err!(
                    NotImplemented: "fill_forward",
                    array.encoding().id()
                ))
            })
    })
}