opendp 0.14.2-dev.20260401.2

A library of differential privacy algorithms for the statistical analysis of sensitive private data.
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

use chrono::{NaiveDate, NaiveTime};
use opendp_derive::bootstrap;

use crate::{
    core::FfiResult,
    domains::{AtomDomain, CategoricalDomain, DatetimeDomain, DynSeriesElementDomain, EnumDomain},
    error::Fallible,
    ffi::any::{AnyDomain, Downcast},
};

use super::ArrayDomain;

#[bootstrap(
    name = "array_domain",
    arguments(categories(rust_type = "Series")),
    returns(c_type = "FfiResult<AnyDomain *>")
)]
/// Construct an instance of `ArrayDomain`.
/// Can be used as an argument to a Polars series domain.
///
/// # Arguments
/// * `element_domain` - The domain of each element in the array.
/// * `width` - The width of the array.
#[unsafe(no_mangle)]
pub extern "C" fn opendp_domains__array_domain(
    element_domain: *const AnyDomain,
    width: u32,
) -> FfiResult<*mut AnyDomain> {
    let element_domain = try_as_ref!(element_domain);

    fn monomorphize<D: DynSeriesElementDomain + Clone>(
        element_domain: &AnyDomain,
        width: u32,
    ) -> Fallible<AnyDomain> {
        let element_domain = try_!(element_domain.downcast_ref::<D>()).clone();
        Ok(AnyDomain::new(ArrayDomain::new(
            element_domain,
            width as usize,
        )))
    }

    let D = element_domain.type_.clone();

    dispatch!(monomorphize, [
        (D, [
            AtomDomain<bool>, AtomDomain<String>,
            AtomDomain<u32>, AtomDomain<u64>,
            AtomDomain<i8>, AtomDomain<i16>, AtomDomain<i32>, AtomDomain<i64>,
            AtomDomain<f32>, AtomDomain<f64>,
            AtomDomain<NaiveTime>, AtomDomain<NaiveDate>,
            CategoricalDomain, DatetimeDomain, EnumDomain
        ])
    ], (element_domain, width))
    .into()
}