explorer

// MiMalloc won´t compile on Windows with the GCC compiler.
// On Linux with Musl it won´t load correctly.
#[cfg(not(any(
    all(windows, target_env = "gnu"),
    all(target_os = "linux", target_env = "musl")
)))]
use mimalloc::MiMalloc;
use rustler::{Env, Term};

#[cfg(not(any(
    all(windows, target_env = "gnu"),
    all(target_os = "linux", target_env = "musl")
)))]
#[global_allocator]
static GLOBAL: MiMalloc = MiMalloc;

mod dataframe;
#[allow(clippy::extra_unused_lifetimes)]
mod datatypes;
mod error;
mod expressions;
mod lazyframe;
mod series;

use dataframe::*;
pub use datatypes::{
    ExDataFrame, ExDataFrameRef, ExExpr, ExExprRef, ExLazyFrame, ExLazyFrameRef, ExSeries,
    ExSeriesRef,
};
pub use error::ExplorerError;
use expressions::*;
use lazyframe::*;
use series::*;

fn on_load(env: Env, _info: Term) -> bool {
    rustler::resource!(ExDataFrameRef, env);
    rustler::resource!(ExExprRef, env);
    rustler::resource!(ExLazyFrameRef, env);
    rustler::resource!(ExSeriesRef, env);
    true
}

mod atoms {
    rustler::atoms! {
        calendar_atom = "Elixir.Calendar.ISO",
        hour,
        minute,
        second,
        day,
        month,
        year,
        microsecond,
        calendar,
        nan,
        infinity,
        neg_infinity
    }
}

rustler::init!(
    "Elixir.Explorer.PolarsBackend.Native",
    [
        df_arrange_with,
        df_column,
        df_names,
        df_drop,
        df_drop_duplicates,
        df_drop_nulls,
        df_dtypes,
        df_filter_with,
        df_get_columns,
        df_groups,
        df_groupby_agg,
        df_groupby_agg_with,
        df_group_indices,
        df_head,
        df_height,
        df_hstack_many,
        df_join,
        df_mask,
        df_melt,
        df_new,
        df_pivot_wider,
        df_read_csv,
        df_read_ipc,
        df_read_parquet,
        df_read_ndjson,
        df_write_ndjson,
        df_select,
        df_select_at_idx,
        df_rename_columns,
        df_shape,
        df_slice,
        df_slice_by_indices,
        df_sort,
        df_tail,
        df_to_csv,
        df_to_csv_file,
        df_to_dummies,
        df_to_lazy,
        df_vstack_many,
        df_width,
        df_with_columns,
        df_with_column_exprs,
        df_write_ipc,
        df_write_parquet,
        // expressions
        expr_boolean,
        expr_cast,
        expr_column,
        expr_date,
        expr_datetime,
        expr_float,
        expr_integer,
        expr_string,
        expr_slice,
        expr_head,
        expr_tail,
        expr_peaks,
        expr_fill_missing,
        expr_fill_missing_with_value,
        // sort
        expr_argsort,
        expr_distinct,
        expr_unordered_distinct,
        expr_reverse,
        expr_sort,
        // comparison expressions
        expr_binary_and,
        expr_binary_or,
        expr_eq,
        expr_gt,
        expr_gt_eq,
        expr_all_equal,
        expr_is_nil,
        expr_is_not_nil,
        expr_lt,
        expr_lt_eq,
        expr_neq,
        // arithmetic expressions
        expr_add,
        expr_subtract,
        expr_divide,
        expr_multiply,
        expr_pow,
        // slice and dice expressions
        expr_coalesce,
        // agg expressions
        expr_sum,
        expr_min,
        expr_max,
        expr_mean,
        expr_median,
        expr_n_distinct,
        expr_std,
        expr_var,
        expr_quantile,
        expr_alias,
        expr_count,
        expr_first,
        expr_last,
        // window expressions
        expr_cumulative_max,
        expr_cumulative_min,
        expr_cumulative_sum,
        expr_window_max,
        expr_window_mean,
        expr_window_min,
        expr_window_sum,
        // inspect expressions
        expr_describe_filter_plan,
        // lazyframe
        lf_collect,
        lf_describe_plan,
        lf_drop,
        lf_dtypes,
        lf_fetch,
        lf_head,
        lf_names,
        lf_select,
        lf_tail,
        // series
        s_add,
        s_and,
        s_append,
        s_argsort,
        s_as_str,
        s_cast,
        s_coalesce,
        s_cum_max,
        s_cum_min,
        s_cum_sum,
        s_distinct,
        s_div,
        s_dtype,
        s_eq,
        s_fill_none,
        s_fill_none_with_int,
        s_fill_none_with_float,
        s_fill_none_with_bin,
        s_filter,
        s_get,
        s_gt,
        s_gt_eq,
        s_head,
        s_is_not_null,
        s_is_null,
        s_len,
        s_lt,
        s_lt_eq,
        s_max,
        s_mean,
        s_median,
        s_min,
        s_mul,
        s_n_unique,
        s_name,
        s_neq,
        s_new_bool,
        s_new_date32,
        s_new_date64,
        s_new_f64,
        s_new_i64,
        s_new_str,
        s_or,
        s_peak_max,
        s_peak_min,
        s_pow,
        s_int_pow,
        s_quantile,
        s_rename,
        s_reverse,
        s_rolling_max,
        s_rolling_mean,
        s_rolling_min,
        s_rolling_sum,
        s_seedable_random_indices,
        s_series_equal,
        s_slice,
        s_slice_by_indices,
        s_sort,
        s_std,
        s_sub,
        s_sum,
        s_tail,
        s_take_every,
        s_to_list,
        s_unordered_distinct,
        s_var,
        s_value_counts,
    ],
    load = on_load
);