Compare commits

...

4 commits

Author SHA1 Message Date
Jarrett Ye
2f62fb8dd1
Merge 1afc7e35ca into 3890e12c9e 2025-09-17 14:51:58 +08:00
Jarrett Ye
1afc7e35ca
Merge branch 'main' into Refactor-reviews_for_fsrs-function-for-improved-performance 2025-09-17 13:43:13 +08:00
Jarrett Ye
38990fe68d
collapse if statement 2025-09-17 13:38:05 +08:00
Jarrett Ye
ecd1cf45e9
Refactor reviews_for_fsrs function for improved performance
Replaced the previous implementation with a more efficient approach using a single loop and pre-allocated vectors. This change reduces the complexity of creating FSRSItems and enhances overall performance, especially for larger datasets.
2025-09-17 12:28:24 +08:00

View file

@ -481,24 +481,20 @@ pub(crate) fn reviews_for_fsrs(
let skip = if training { 1 } else { 0 };
// Convert the remaining entries into separate FSRSItems, where each item
// contains all reviews done until then.
let items: Vec<(RevlogId, FSRSItem)> = entries
.iter()
.enumerate()
.skip(skip)
.map(|(outer_idx, entry)| {
let reviews = entries
.iter()
.take(outer_idx + 1)
.enumerate()
.map(|(inner_idx, r)| FSRSReview {
rating: r.button_chosen as u32,
delta_t: delta_ts[inner_idx],
})
.collect();
(entry.id, FSRSItem { reviews })
})
.filter(|(_, item)| !training || item.reviews.last().unwrap().delta_t > 0)
.collect_vec();
let mut items = Vec::with_capacity(entries.len());
let mut current_reviews = Vec::with_capacity(entries.len());
for (idx, (entry, &delta_t)) in entries.iter().zip(delta_ts.iter()).enumerate() {
current_reviews.push(FSRSReview {
rating: entry.button_chosen as u32,
delta_t,
});
if idx >= skip && (!training || current_reviews.last().unwrap().delta_t > 0) {
let item = FSRSItem {
reviews: current_reviews.clone(),
};
items.push((entry.id, item));
}
}
if items.is_empty() {
None
} else {