optflags.rs

use parking_lot::RwLock;
use polars::prelude::OptFlags;
use pyo3::pymethods;

use super::PyOptFlags;

macro_rules! flag_getter_setters {
    ($(($flag:ident, $getter:ident, $setter:ident, clear=$clear:literal))+) => {
        #[pymethods]
        impl PyOptFlags {
            #[staticmethod]
            pub fn empty() -> Self {
                Self {
                    inner: RwLock::new(OptFlags::empty())
                }
            }

            #[staticmethod]
            #[allow(clippy::should_implement_trait)]
            pub fn default() -> Self {
                Self { inner: RwLock::default() }
            }

            pub fn no_optimizations(&self) {
                $(if $clear {
                    self.inner.write().remove(OptFlags::$flag);
                })+
            }

            pub fn copy(&self) -> Self {
                Self { inner: RwLock::new(self.inner.read().clone()) }
            }

            $(
            #[getter]
            fn $getter(&self) -> bool {
                self.inner.read().contains(OptFlags::$flag)
            }
            #[setter]
            fn $setter(&self, value: bool) {
                self.inner.write().set(OptFlags::$flag, value)
            }
            )+
        }
    };
}

flag_getter_setters! {
    (TYPE_COERCION, get_type_coercion, set_type_coercion, clear=false)
    (TYPE_CHECK, get_type_check, set_type_check, clear=false)

    (PROJECTION_PUSHDOWN, get_projection_pushdown, set_projection_pushdown, clear=true)
    (PREDICATE_PUSHDOWN, get_predicate_pushdown, set_predicate_pushdown, clear=true)
    (CLUSTER_WITH_COLUMNS, get_cluster_with_columns, set_cluster_with_columns, clear=true)
    (SIMPLIFY_EXPR, get_simplify_expression, set_simplify_expression, clear=true)
    (SLICE_PUSHDOWN, get_slice_pushdown, set_slice_pushdown, clear=true)
    (COMM_SUBPLAN_ELIM, get_comm_subplan_elim, set_comm_subplan_elim, clear=true)
    (COMM_SUBEXPR_ELIM, get_comm_subexpr_elim, set_comm_subexpr_elim, clear=true)
    (CHECK_ORDER_OBSERVE, get_check_order_observe, set_check_order_observe, clear=true)
    (FAST_PROJECTION, get_fast_projection, set_fast_projection, clear=true)

    (EAGER, get_eager, set_eager, clear=true)
    (NEW_STREAMING, get_streaming, set_streaming, clear=true)
}