2018-07-07 14:43:29 +02:00
|
|
|
use std::cmp::Ordering;
|
2019-01-30 16:30:27 +01:00
|
|
|
use slice_group_by::GroupBy;
|
2019-02-24 19:44:24 +01:00
|
|
|
use crate::criterion::Criterion;
|
|
|
|
use crate::RawDocument;
|
2018-07-07 14:43:29 +02:00
|
|
|
|
2018-08-25 13:15:04 +02:00
|
|
|
#[inline]
|
2019-02-02 14:22:31 +01:00
|
|
|
fn sum_matches_attributes(query_index: &[u32], attribute: &[u16]) -> usize {
|
|
|
|
let mut sum_attributes = 0;
|
|
|
|
let mut index = 0;
|
|
|
|
|
2019-02-12 15:43:43 +01:00
|
|
|
for group in query_index.linear_group() {
|
2019-02-02 14:22:31 +01:00
|
|
|
sum_attributes += attribute[index] as usize;
|
|
|
|
index += group.len();
|
|
|
|
}
|
|
|
|
|
|
|
|
sum_attributes
|
2018-08-25 13:15:04 +02:00
|
|
|
}
|
|
|
|
|
2018-10-11 14:04:41 +02:00
|
|
|
#[derive(Debug, Clone, Copy)]
|
|
|
|
pub struct SumOfWordsAttribute;
|
|
|
|
|
2019-02-02 14:22:31 +01:00
|
|
|
impl Criterion for SumOfWordsAttribute {
|
|
|
|
fn evaluate(&self, lhs: &RawDocument, rhs: &RawDocument) -> Ordering {
|
|
|
|
let lhs = {
|
|
|
|
let query_index = lhs.query_index();
|
|
|
|
let attribute = lhs.attribute();
|
|
|
|
sum_matches_attributes(query_index, attribute)
|
|
|
|
};
|
|
|
|
|
|
|
|
let rhs = {
|
|
|
|
let query_index = rhs.query_index();
|
|
|
|
let attribute = rhs.attribute();
|
|
|
|
sum_matches_attributes(query_index, attribute)
|
|
|
|
};
|
2018-07-07 14:43:29 +02:00
|
|
|
|
2018-10-11 14:04:41 +02:00
|
|
|
lhs.cmp(&rhs)
|
|
|
|
}
|
2018-07-07 14:43:29 +02:00
|
|
|
}
|