polars_python/series/
arithmetic.rs

1use polars::prelude::*;
2use pyo3::prelude::*;
3
4use super::PySeries;
5use crate::utils::EnterPolarsExt;
6
7#[pymethods]
8impl PySeries {
9    fn add(&self, py: Python<'_>, other: &PySeries) -> PyResult<Self> {
10        py.enter_polars_series(|| &self.series + &other.series)
11    }
12    fn sub(&self, py: Python<'_>, other: &PySeries) -> PyResult<Self> {
13        py.enter_polars_series(|| &self.series - &other.series)
14    }
15    fn mul(&self, py: Python<'_>, other: &PySeries) -> PyResult<Self> {
16        py.enter_polars_series(|| &self.series * &other.series)
17    }
18    fn div(&self, py: Python<'_>, other: &PySeries) -> PyResult<Self> {
19        py.enter_polars_series(|| &self.series / &other.series)
20    }
21    fn rem(&self, py: Python<'_>, other: &PySeries) -> PyResult<Self> {
22        py.enter_polars_series(|| &self.series % &other.series)
23    }
24}
25
26macro_rules! impl_arithmetic {
27    ($name:ident, $type:ty, $operand:tt) => {
28        #[pymethods]
29        impl PySeries {
30            fn $name(&self, py: Python<'_>, other: $type) -> PyResult<Self> {
31                py.enter_polars_series(|| Ok({&self.series $operand other}))
32            }
33        }
34    };
35}
36
37impl_arithmetic!(add_u8, u8, +);
38impl_arithmetic!(add_u16, u16, +);
39impl_arithmetic!(add_u32, u32, +);
40impl_arithmetic!(add_u64, u64, +);
41impl_arithmetic!(add_i8, i8, +);
42impl_arithmetic!(add_i16, i16, +);
43impl_arithmetic!(add_i32, i32, +);
44impl_arithmetic!(add_i64, i64, +);
45impl_arithmetic!(add_datetime, i64, +);
46impl_arithmetic!(add_duration, i64, +);
47impl_arithmetic!(add_f32, f32, +);
48impl_arithmetic!(add_f64, f64, +);
49impl_arithmetic!(sub_u8, u8, -);
50impl_arithmetic!(sub_u16, u16, -);
51impl_arithmetic!(sub_u32, u32, -);
52impl_arithmetic!(sub_u64, u64, -);
53impl_arithmetic!(sub_i8, i8, -);
54impl_arithmetic!(sub_i16, i16, -);
55impl_arithmetic!(sub_i32, i32, -);
56impl_arithmetic!(sub_i64, i64, -);
57impl_arithmetic!(sub_datetime, i64, -);
58impl_arithmetic!(sub_duration, i64, -);
59impl_arithmetic!(sub_f32, f32, -);
60impl_arithmetic!(sub_f64, f64, -);
61impl_arithmetic!(div_u8, u8, /);
62impl_arithmetic!(div_u16, u16, /);
63impl_arithmetic!(div_u32, u32, /);
64impl_arithmetic!(div_u64, u64, /);
65impl_arithmetic!(div_i8, i8, /);
66impl_arithmetic!(div_i16, i16, /);
67impl_arithmetic!(div_i32, i32, /);
68impl_arithmetic!(div_i64, i64, /);
69impl_arithmetic!(div_f32, f32, /);
70impl_arithmetic!(div_f64, f64, /);
71impl_arithmetic!(mul_u8, u8, *);
72impl_arithmetic!(mul_u16, u16, *);
73impl_arithmetic!(mul_u32, u32, *);
74impl_arithmetic!(mul_u64, u64, *);
75impl_arithmetic!(mul_i8, i8, *);
76impl_arithmetic!(mul_i16, i16, *);
77impl_arithmetic!(mul_i32, i32, *);
78impl_arithmetic!(mul_i64, i64, *);
79impl_arithmetic!(mul_f32, f32, *);
80impl_arithmetic!(mul_f64, f64, *);
81impl_arithmetic!(rem_u8, u8, %);
82impl_arithmetic!(rem_u16, u16, %);
83impl_arithmetic!(rem_u32, u32, %);
84impl_arithmetic!(rem_u64, u64, %);
85impl_arithmetic!(rem_i8, i8, %);
86impl_arithmetic!(rem_i16, i16, %);
87impl_arithmetic!(rem_i32, i32, %);
88impl_arithmetic!(rem_i64, i64, %);
89impl_arithmetic!(rem_f32, f32, %);
90impl_arithmetic!(rem_f64, f64, %);
91
92macro_rules! impl_rhs_arithmetic {
93    ($name:ident, $type:ty, $operand:ident) => {
94        #[pymethods]
95        impl PySeries {
96            fn $name(&self, py: Python<'_>, other: $type) -> PyResult<Self> {
97                py.enter_polars_series(|| Ok(other.$operand(&self.series)))
98            }
99        }
100    };
101}
102
103impl_rhs_arithmetic!(add_u8_rhs, u8, add);
104impl_rhs_arithmetic!(add_u16_rhs, u16, add);
105impl_rhs_arithmetic!(add_u32_rhs, u32, add);
106impl_rhs_arithmetic!(add_u64_rhs, u64, add);
107impl_rhs_arithmetic!(add_i8_rhs, i8, add);
108impl_rhs_arithmetic!(add_i16_rhs, i16, add);
109impl_rhs_arithmetic!(add_i32_rhs, i32, add);
110impl_rhs_arithmetic!(add_i64_rhs, i64, add);
111impl_rhs_arithmetic!(add_f32_rhs, f32, add);
112impl_rhs_arithmetic!(add_f64_rhs, f64, add);
113impl_rhs_arithmetic!(sub_u8_rhs, u8, sub);
114impl_rhs_arithmetic!(sub_u16_rhs, u16, sub);
115impl_rhs_arithmetic!(sub_u32_rhs, u32, sub);
116impl_rhs_arithmetic!(sub_u64_rhs, u64, sub);
117impl_rhs_arithmetic!(sub_i8_rhs, i8, sub);
118impl_rhs_arithmetic!(sub_i16_rhs, i16, sub);
119impl_rhs_arithmetic!(sub_i32_rhs, i32, sub);
120impl_rhs_arithmetic!(sub_i64_rhs, i64, sub);
121impl_rhs_arithmetic!(sub_f32_rhs, f32, sub);
122impl_rhs_arithmetic!(sub_f64_rhs, f64, sub);
123impl_rhs_arithmetic!(div_u8_rhs, u8, div);
124impl_rhs_arithmetic!(div_u16_rhs, u16, div);
125impl_rhs_arithmetic!(div_u32_rhs, u32, div);
126impl_rhs_arithmetic!(div_u64_rhs, u64, div);
127impl_rhs_arithmetic!(div_i8_rhs, i8, div);
128impl_rhs_arithmetic!(div_i16_rhs, i16, div);
129impl_rhs_arithmetic!(div_i32_rhs, i32, div);
130impl_rhs_arithmetic!(div_i64_rhs, i64, div);
131impl_rhs_arithmetic!(div_f32_rhs, f32, div);
132impl_rhs_arithmetic!(div_f64_rhs, f64, div);
133impl_rhs_arithmetic!(mul_u8_rhs, u8, mul);
134impl_rhs_arithmetic!(mul_u16_rhs, u16, mul);
135impl_rhs_arithmetic!(mul_u32_rhs, u32, mul);
136impl_rhs_arithmetic!(mul_u64_rhs, u64, mul);
137impl_rhs_arithmetic!(mul_i8_rhs, i8, mul);
138impl_rhs_arithmetic!(mul_i16_rhs, i16, mul);
139impl_rhs_arithmetic!(mul_i32_rhs, i32, mul);
140impl_rhs_arithmetic!(mul_i64_rhs, i64, mul);
141impl_rhs_arithmetic!(mul_f32_rhs, f32, mul);
142impl_rhs_arithmetic!(mul_f64_rhs, f64, mul);
143impl_rhs_arithmetic!(rem_u8_rhs, u8, rem);
144impl_rhs_arithmetic!(rem_u16_rhs, u16, rem);
145impl_rhs_arithmetic!(rem_u32_rhs, u32, rem);
146impl_rhs_arithmetic!(rem_u64_rhs, u64, rem);
147impl_rhs_arithmetic!(rem_i8_rhs, i8, rem);
148impl_rhs_arithmetic!(rem_i16_rhs, i16, rem);
149impl_rhs_arithmetic!(rem_i32_rhs, i32, rem);
150impl_rhs_arithmetic!(rem_i64_rhs, i64, rem);
151impl_rhs_arithmetic!(rem_f32_rhs, f32, rem);
152impl_rhs_arithmetic!(rem_f64_rhs, f64, rem);
polars_python/series/arithmetic.rs

polars_python/series/
arithmetic.rs
