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

use flatbuffers::{FlatBufferBuilder, WIPOffset};
use serde::{Deserialize, Serialize};
use vortex_dtype::DType;
use vortex_error::{VortexError, VortexExpect as _};
use vortex_flatbuffers::{scalar as fb, WriteFlatBuffer};

use crate::{Scalar, ScalarValue};

impl TryFrom<fb::Scalar<'_>> for Scalar {
    type Error = VortexError;

    fn try_from(value: fb::Scalar<'_>) -> Result<Self, Self::Error> {
        let dtype = value.dtype();
        let dtype = DType::try_from(dtype)?;

        let reader = flexbuffers::Reader::get_root(value.value().flex().bytes())?;
        let value = ScalarValue::deserialize(reader)?;

        Ok(Self { dtype, value })
    }
}

impl TryFrom<fb::ScalarValue<'_>> for ScalarValue {
    type Error = VortexError;

    fn try_from(value: fb::ScalarValue<'_>) -> Result<Self, Self::Error> {
        let reader = flexbuffers::Reader::get_root(value.flex().bytes())?;
        Ok(Self::deserialize(reader)?)
    }
}

impl WriteFlatBuffer for Scalar {
    type Target<'a> = fb::Scalar<'a>;

    fn write_flatbuffer<'fb>(
        &self,
        fbb: &mut FlatBufferBuilder<'fb>,
    ) -> WIPOffset<Self::Target<'fb>> {
        let dtype = Some(self.dtype.write_flatbuffer(fbb));
        let value = Some(self.value.write_flatbuffer(fbb));
        fb::Scalar::create(fbb, &fb::ScalarArgs { dtype, value })
    }
}

impl WriteFlatBuffer for ScalarValue {
    type Target<'a> = fb::ScalarValue<'a>;

    fn write_flatbuffer<'fb>(
        &self,
        fbb: &mut FlatBufferBuilder<'fb>,
    ) -> WIPOffset<Self::Target<'fb>> {
        let mut value_se = flexbuffers::FlexbufferSerializer::new();
        self.serialize(&mut value_se)
            .map_err(VortexError::FlexBuffersSerError)
            .vortex_expect("Failed to serialize ScalarValue");
        let flex = Some(fbb.create_vector(value_se.view()));
        fb::ScalarValue::create(fbb, &fb::ScalarValueArgs { flex })
    }
}