EZPPFarm/akatsuki-pp-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: handle taiko passed objects properly & improve testing

Browse Source
This commit is contained in:
MaxOhn
2023-11-13 13:38:14 +01:00
parent 037654c771
commit 1ff200dbda
4 changed files with 142 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/taiko/difficulty_object.rs
+10 -2
View File
@@ -111,8 +111,7 @@ fn closest_rhythm(
.unwrap()
}
// TODO: Remove Default impl and replace with `with_capacity` method for efficiency
#[derive(Clone, Debug, Default)]
#[derive(Clone, Debug)]
pub(crate) struct ObjectLists {
pub(crate) all: Vec<Rc<RefCell<TaikoDifficultyObject>>>,
pub(crate) centres: Vec<usize>,
@@ -121,6 +120,15 @@ pub(crate) struct ObjectLists {
}
impl ObjectLists {
pub(crate) fn with_capacity(capacity: usize) -> Self {
Self {
all: Vec::with_capacity(capacity),
centres: Vec::new(),
rims: Vec::new(),
notes: Vec::new(),
}
}
pub(crate) fn prev_mono(
&self,
curr: usize,
src/taiko/gradual_difficulty.rs
+72 -19
View File
@@ -51,9 +51,18 @@ pub struct TaikoGradualDifficulty {
lists: ObjectLists,
peaks: Peaks,
total_hits: usize,
first_combos: FirstTwoCombos,
is_convert: bool,
}
#[derive(Copy, Clone, Debug)]
enum FirstTwoCombos {
None,
OnlyFirst,
OnlySecond,
Both,
}
impl TaikoGradualDifficulty {
/// Create a new difficulty attributes iterator for osu!taiko maps.
pub fn new(map: &Beatmap, mods: u32) -> Self {
@@ -68,41 +77,44 @@ impl TaikoGradualDifficulty {
.clock_rate(clock_rate)
.hit_windows();
let mut attrs = TaikoDifficultyAttributes {
stamina: 0.0,
rhythm: 0.0,
colour: 0.0,
peak: 0.0,
let attrs = TaikoDifficultyAttributes {
hit_window,
stars: 0.0,
max_combo: 0,
..Default::default()
};
if map.hit_objects.len() < 2 {
return Self {
idx: 0,
diff_objects: Vec::new().into_iter(),
lists: ObjectLists::default(),
lists: ObjectLists::with_capacity(0),
peaks,
attrs,
total_hits: 0,
first_combos: FirstTwoCombos::None,
is_convert,
};
}
attrs.max_combo += map.hit_objects[0].is_circle() as usize;
attrs.max_combo += map.hit_objects[1].is_circle() as usize;
let mut total_hits = attrs.max_combo;
let mut diff_objects = ObjectLists::default();
let first_combos = match (
map.hit_objects[0].is_circle(),
map.hit_objects[1].is_circle(),
) {
(false, false) => FirstTwoCombos::None,
(true, false) => FirstTwoCombos::OnlyFirst,
(false, true) => FirstTwoCombos::OnlySecond,
(true, true) => FirstTwoCombos::Both,
};
let mut total_hits = 0;
let mut diff_objects = ObjectLists::with_capacity(map.hit_objects.len().saturating_sub(2));
map.taiko_objects()
.inspect(|(h, _)| total_hits += h.is_hit as usize)
.skip(2)
.zip(map.hit_objects.iter().skip(1))
.zip(map.hit_objects.iter())
.enumerate()
.for_each(|(idx, (((base, base_start_time), last), last_last))| {
total_hits += base.is_hit as usize;
let diff_obj = TaikoDifficultyObject::new(
base,
base_start_time,
@@ -135,6 +147,7 @@ impl TaikoGradualDifficulty {
peaks,
attrs,
total_hits,
first_combos,
is_convert,
}
}
@@ -162,6 +175,14 @@ impl Iterator for TaikoGradualDifficulty {
}
} else if self.lists.all.is_empty() {
return None;
} else {
match self.first_combos {
FirstTwoCombos::OnlyFirst => self.attrs.max_combo = 1,
FirstTwoCombos::OnlySecond if self.idx == 1 => self.attrs.max_combo = 1,
FirstTwoCombos::Both if self.idx == 0 => self.attrs.max_combo = 1,
FirstTwoCombos::Both if self.idx == 1 => self.attrs.max_combo = 2,
_ => {}
}
}
self.idx += 1;
@@ -211,11 +232,43 @@ impl Iterator for TaikoGradualDifficulty {
fn nth(&mut self, n: usize) -> Option<Self::Item> {
let mut take = n.min(self.len().saturating_sub(1));
// The first two notes have no difficulty object
if self.idx < 2 && take > 0 {
let skipped = take.min(2);
take -= skipped;
self.idx += skipped;
// The first two notes have no difficulty object but might add to combo
match (take, self.idx) {
(_, 2..) | (0, _) => {}
(1, 0) => {
take -= 1;
self.idx += 1;
match self.first_combos {
FirstTwoCombos::None => {}
FirstTwoCombos::OnlyFirst => self.attrs.max_combo = 1,
FirstTwoCombos::OnlySecond => {}
FirstTwoCombos::Both => self.attrs.max_combo = 1,
}
}
(_, 0) => {
take -= 2;
self.idx += 2;
match self.first_combos {
FirstTwoCombos::None => {}
FirstTwoCombos::OnlyFirst => self.attrs.max_combo = 1,
FirstTwoCombos::OnlySecond => self.attrs.max_combo = 1,
FirstTwoCombos::Both => self.attrs.max_combo = 2,
}
}
(_, 1) => {
take -= 1;
self.idx += 1;
match self.first_combos {
FirstTwoCombos::None => {}
FirstTwoCombos::OnlyFirst => self.attrs.max_combo = 1,
FirstTwoCombos::OnlySecond => self.attrs.max_combo = 1,
FirstTwoCombos::Both => self.attrs.max_combo = 2,
}
}
_ => unreachable!(),
}
for _ in 0..take {
src/taiko/mod.rs
+12 -14
View File
@@ -227,26 +227,19 @@ fn calculate_skills(params: TaikoStars<'_>) -> (Peaks, usize) {
is_convert: _,
} = params;
let mut take = passed_objects.unwrap_or(map.hit_objects.len());
let take = passed_objects.unwrap_or(map.hit_objects.len());
let clock_rate = clock_rate.unwrap_or_else(|| mods.clock_rate());
let mut peaks = Peaks::new();
let mut max_combo = 0;
let mut n_diff_objects = 0;
let mut diff_objects = ObjectLists::default();
let mut diff_objects = ObjectLists::with_capacity(map.hit_objects.len().saturating_sub(2));
map.taiko_objects()
.take_while(|(h, _)| {
if h.is_hit {
if take == 0 {
return false;
}
max_combo += 1;
take -= 1;
}
true
.inspect(|(h, _)| {
n_diff_objects += (max_combo < take) as usize;
max_combo += (max_combo < take && h.is_hit) as usize;
})
.skip(2)
.zip(map.hit_objects.iter().skip(1))
@@ -276,9 +269,14 @@ fn calculate_skills(params: TaikoStars<'_>) -> (Peaks, usize) {
diff_objects.all.push(Rc::new(RefCell::new(diff_obj)));
});
map.hit_objects
.iter()
.take(2)
.for_each(|h| n_diff_objects = n_diff_objects.saturating_sub(h.is_circle() as usize));
ColourDifficultyPreprocessor::process_and_assign(&mut diff_objects);
for hit_object in diff_objects.all.iter() {
for hit_object in diff_objects.all.iter().take(n_diff_objects) {
peaks.process(&hit_object.borrow(), &diff_objects);
}