polars_plan/dsl/
selector.rs

1use std::ops::{Add, BitAnd, BitXor, Sub};
2
3#[cfg(feature = "serde")]
4use serde::{Deserialize, Serialize};
5
6use super::*;
7
8#[derive(Clone, PartialEq, Hash, Debug, Eq)]
9#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
10pub enum Selector {
11    Add(Box<Selector>, Box<Selector>),
12    Sub(Box<Selector>, Box<Selector>),
13    ExclusiveOr(Box<Selector>, Box<Selector>),
14    Intersect(Box<Selector>, Box<Selector>),
15    Root(Box<Expr>),
16}
17
18impl Selector {
19    pub fn new(e: Expr) -> Self {
20        Self::Root(Box::new(e))
21    }
22}
23
24impl Add for Selector {
25    type Output = Selector;
26
27    fn add(self, rhs: Self) -> Self::Output {
28        Selector::Add(Box::new(self), Box::new(rhs))
29    }
30}
31
32impl BitAnd for Selector {
33    type Output = Selector;
34
35    #[allow(clippy::suspicious_arithmetic_impl)]
36    fn bitand(self, rhs: Self) -> Self::Output {
37        Selector::Intersect(Box::new(self), Box::new(rhs))
38    }
39}
40
41impl BitXor for Selector {
42    type Output = Selector;
43
44    #[allow(clippy::suspicious_arithmetic_impl)]
45    fn bitxor(self, rhs: Self) -> Self::Output {
46        Selector::ExclusiveOr(Box::new(self), Box::new(rhs))
47    }
48}
49
50impl Sub for Selector {
51    type Output = Selector;
52
53    #[allow(clippy::suspicious_arithmetic_impl)]
54    fn sub(self, rhs: Self) -> Self::Output {
55        Selector::Sub(Box::new(self), Box::new(rhs))
56    }
57}
58
59impl From<&str> for Selector {
60    fn from(value: &str) -> Self {
61        Selector::new(col(PlSmallStr::from_str(value)))
62    }
63}
64
65impl From<String> for Selector {
66    fn from(value: String) -> Self {
67        Selector::new(col(PlSmallStr::from_string(value)))
68    }
69}
70
71impl From<PlSmallStr> for Selector {
72    fn from(value: PlSmallStr) -> Self {
73        Selector::new(Expr::Column(value))
74    }
75}
76
77impl From<Expr> for Selector {
78    fn from(value: Expr) -> Self {
79        Selector::new(value)
80    }
81}
polars_plan/dsl/selector.rs

polars_plan/dsl/
selector.rs
