psyche_utils/bucket_strainer/rules/
sum_rule.rs

1use crate::bucket_strainer::{Bucket, Rule};
2use crate::Scalar;
3
4/// Bucket rule that sum scores of all of its subrules.
5#[derive(Clone)]
6pub struct SumRule<T> {
7    /// List of subrules to score.
8    pub rules: Vec<Box<dyn Rule<T>>>,
9}
10
11impl<T> SumRule<T> {
12    pub fn new(rules: Vec<Box<dyn Rule<T>>>) -> Self {
13        Self { rules }
14    }
15}
16
17impl<T> Rule<T> for SumRule<T>
18where
19    T: Clone + 'static,
20{
21    fn score(&self, item: &T, bucket: &Bucket<T>) -> Scalar {
22        self.rules
23            .iter()
24            .fold(0.0, |a, r| a + r.score(item, bucket))
25    }
26
27    fn box_clone(&self) -> Box<dyn Rule<T>> {
28        Box::new((*self).clone())
29    }
30}