ndarray/
argument_traits.rs

1use std::cell::Cell;
2use std::mem::MaybeUninit;
3
4use crate::math_cell::MathCell;
5
6/// A producer element that can be assigned to once
7pub trait AssignElem<T> {
8    /// Assign the value `input` to the element that self represents.
9    fn assign_elem(self, input: T);
10}
11
12/// Assignable element, simply `*self = input`.
13impl<'a, T> AssignElem<T> for &'a mut T {
14    fn assign_elem(self, input: T) {
15        *self = input;
16    }
17}
18
19/// Assignable element, simply `self.set(input)`.
20impl<'a, T> AssignElem<T> for &'a Cell<T> {
21    fn assign_elem(self, input: T) {
22        self.set(input);
23    }
24}
25
26/// Assignable element, simply `self.set(input)`.
27impl<'a, T> AssignElem<T> for &'a MathCell<T> {
28    fn assign_elem(self, input: T) {
29        self.set(input);
30    }
31}
32
33/// Assignable element, the item in the MaybeUninit is overwritten (prior value, if any, is not
34/// read or dropped).
35impl<'a, T> AssignElem<T> for &'a mut MaybeUninit<T> {
36    fn assign_elem(self, input: T) {
37        *self = MaybeUninit::new(input);
38    }
39}