Wizzard/radarflow2-kvm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimize bomb search

Browse Source 
More optimized bomb search, now searching in chunks of 128 indexes.
This commit is contained in:
Janek 2024-01-08 03:33:11 +01:00
parent 16f7791628
commit 510a74a802
1 changed files with 12 additions and 11 deletions

23
src/dma/threaddata/mod.rs

@ -34,41 +34,42 @@ impl CsData {
// If the bomb is dropped, do a reverse entity list loop with early exit when we found the bomb. ( Now with BATCHING!!! :O ) // If the bomb is dropped, do a reverse entity list loop with early exit when we found the bomb. ( Now with BATCHING!!! :O )
if self.bomb_dropped { if self.bomb_dropped {
// We search 16 entities at a time. // We search in chunks of 128 indexes
for chunk in &(0..=self.highest_index).rev().chunks(16) { for chunk in &(0..=self.highest_index).rev().into_iter().chunks(128) {
let mut data_vec: Vec<(u64, i32)> = chunk
.map(|idx| (0u64, idx))
.collect();
let indexes: Vec<i32> = chunk.collect();
let mut data_array = [(0u64, 0i32); 16];
{ {
let mut batcher = ctx.process.batcher(); let mut batcher = ctx.process.batcher();
let ent_list: Address = self.entity_list.into(); let ent_list: Address = self.entity_list.into();
data_array.iter_mut().zip(indexes).for_each(|((data_ptr, data_idx), index)| { data_vec.iter_mut().for_each(|(data, idx)| {
batcher.read_into(ent_list + 8 * (index >> 9) + 16, data_ptr); batcher.read_into(ent_list + 8 * (*idx >> 9) + 16, data);
*data_idx = index;
}); });
} }
{ {
let mut batcher = ctx.process.batcher(); let mut batcher = ctx.process.batcher();
data_array.iter_mut().for_each(|(ptr, index)| { data_vec.iter_mut().for_each(|(ptr, index)| {
let handle: Address = (*ptr).into(); let handle: Address = (*ptr).into();
batcher.read_into(handle + 120 * (*index & 0x1FF), ptr); batcher.read_into(handle + 120 * (*index & 0x1FF), ptr);
}); });
} }
// You can actually optimize this EVEN more // You can actually optimize this EVEN more
let bomb = data_array.into_iter().find(|(ptr, _)| { let bomb = data_vec.into_iter().find(|(ptr, _)| {
// By doing this with a batcher too... // By doing this with a batcher too...
ctx.is_dropped_c4((*ptr).into()).unwrap_or(false) ctx.is_dropped_c4((*ptr).into()).unwrap_or(false)
}); });
if let Some(bomb) = bomb { if let Some(bomb) = bomb {
self.bomb = bomb.0.into(); self.bomb = bomb.0.into();
break;
} }
} }
} else if self.bomb_planted { } else if self.bomb_planted {
let bomb = ctx.get_plantedc4() let bomb = ctx.get_plantedc4()
.expect("Failed to get planted bomb"); .expect("Failed to get planted bomb");