rosu_memory_lib/reader/beatmap/stable/
memory.rs

1use crate::common::GameMode;
2use crate::generate_offset_getter;
3use crate::reader::beatmap::common::{
4    BeatmapInfo, BeatmapLocation, BeatmapMetadata, BeatmapStats, BeatmapStatus,
5    BeatmapTechnicalInfo,
6};
7use crate::reader::beatmap::stable::{beatmap_addr, offset::BEATMAP_OFFSET};
8use crate::reader::helpers::{read_f32, read_i32, read_string, read_u32};
9use crate::reader::structs::State;
10use crate::Error;
11use rosu_mem::process::{Process, ProcessTraits};
12
13generate_offset_getter! {
14    id: i32 = read_i32(BEATMAP_OFFSET.technical.id, beatmap_addr);
15    set_id: i32 = read_i32(BEATMAP_OFFSET.technical.set_id, beatmap_addr);
16    tags: String = read_string(BEATMAP_OFFSET.metadata.tags, beatmap_addr);
17    length: i32 = read_i32(BEATMAP_OFFSET.stats.total_length, beatmap_addr);
18    drain_time: i32 = read_i32(BEATMAP_OFFSET.stats.drain_time, beatmap_addr);
19    author: String = read_string(BEATMAP_OFFSET.metadata.author, beatmap_addr);
20    creator: String = read_string(BEATMAP_OFFSET.metadata.creator, beatmap_addr);
21    md5: String = read_string(BEATMAP_OFFSET.technical.md5, beatmap_addr);
22    title_romanized: String = read_string(BEATMAP_OFFSET.metadata.title_romanized, beatmap_addr);
23    title: String = read_string(BEATMAP_OFFSET.metadata.title_original, beatmap_addr);
24    difficulty: String = read_string(BEATMAP_OFFSET.metadata.difficulty, beatmap_addr);
25    od: f32 = read_f32(BEATMAP_OFFSET.stats.od, beatmap_addr);
26    ar: f32 = read_f32(BEATMAP_OFFSET.stats.ar, beatmap_addr);
27    cs: f32 = read_f32(BEATMAP_OFFSET.stats.cs, beatmap_addr);
28    hp: f32 = read_f32(BEATMAP_OFFSET.stats.hp, beatmap_addr);
29    object_count: u32 = read_u32(BEATMAP_OFFSET.stats.object_count, beatmap_addr);
30    slider_count: i32 = read_i32(BEATMAP_OFFSET.stats.slider_count, beatmap_addr);
31    folder: String = read_string(BEATMAP_OFFSET.location.folder, beatmap_addr);
32    filename: String = read_string(BEATMAP_OFFSET.location.filename, beatmap_addr);
33    audio: String = read_string(BEATMAP_OFFSET.location.audio, beatmap_addr);
34    cover: String = read_string(BEATMAP_OFFSET.location.cover, beatmap_addr);
35    mode: GameMode = read_i32(BEATMAP_OFFSET.technical.mode, beatmap_addr);
36    status: BeatmapStatus = read_i32(BEATMAP_OFFSET.technical.ranked_status, beatmap_addr);
37}
38
39pub fn stats(p: &Process, state: &mut State) -> Result<BeatmapStats, Error> {
40    let beatmap_addr = beatmap_addr(p, state)?;
41    // faster than using read_fn because we dont need to reload addr everytime
42    Ok(BeatmapStats {
43        ar: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.ar)?,
44        od: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.od)?,
45        cs: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.cs)?,
46        hp: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.hp)?,
47        length: p.read_i32(beatmap_addr + BEATMAP_OFFSET.stats.total_length)?,
48        star_rating: crate::reader::beatmap::stable::file::star_rating(p, state)?,
49        object_count: p.read_i32(beatmap_addr + BEATMAP_OFFSET.stats.object_count)?,
50        slider_count: p.read_i32(beatmap_addr + BEATMAP_OFFSET.stats.slider_count)?,
51    })
52}
53
54pub fn info(p: &Process, state: &mut State) -> Result<BeatmapInfo, Error> {
55    let beatmap_addr = beatmap_addr(p, state)?;
56    // done like that to be more efficient reading the string one by one would need to reload addr everytime which cost more
57    Ok(BeatmapInfo {
58        technical: BeatmapTechnicalInfo {
59            md5: p.read_string(beatmap_addr + BEATMAP_OFFSET.technical.md5)?,
60            id: p.read_i32(beatmap_addr + BEATMAP_OFFSET.technical.id)?,
61            set_id: p.read_i32(beatmap_addr + BEATMAP_OFFSET.technical.set_id)?,
62            mode: GameMode::Osu,
63            ranked_status: BeatmapStatus::from(
64                p.read_i32(beatmap_addr + BEATMAP_OFFSET.technical.ranked_status)?,
65            ),
66        },
67        metadata: BeatmapMetadata {
68            author: p.read_string(beatmap_addr + BEATMAP_OFFSET.metadata.author)?,
69            creator: p.read_string(beatmap_addr + BEATMAP_OFFSET.metadata.creator)?,
70            title_romanized: p
71                .read_string(beatmap_addr + BEATMAP_OFFSET.metadata.title_romanized)?,
72            title_original: p.read_string(beatmap_addr + BEATMAP_OFFSET.metadata.title_original)?,
73            difficulty: p.read_string(beatmap_addr + BEATMAP_OFFSET.metadata.difficulty)?,
74            tags: p.read_string(beatmap_addr + BEATMAP_OFFSET.metadata.tags)?,
75        },
76        stats: BeatmapStats {
77            ar: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.ar)?,
78            od: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.od)?,
79            cs: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.cs)?,
80            hp: p.read_f32(beatmap_addr + BEATMAP_OFFSET.stats.hp)?,
81            length: p.read_i32(beatmap_addr + BEATMAP_OFFSET.stats.total_length)?,
82            star_rating: crate::reader::beatmap::stable::file::star_rating(p, state)?,
83            object_count: p.read_i32(beatmap_addr + BEATMAP_OFFSET.stats.object_count)?,
84            slider_count: p.read_i32(beatmap_addr + BEATMAP_OFFSET.stats.slider_count)?,
85        },
86        location: BeatmapLocation {
87            folder: p.read_string(beatmap_addr + BEATMAP_OFFSET.location.folder)?,
88            filename: p.read_string(beatmap_addr + BEATMAP_OFFSET.location.filename)?,
89            audio: p.read_string(beatmap_addr + BEATMAP_OFFSET.location.audio)?,
90            cover: p.read_string(beatmap_addr + BEATMAP_OFFSET.location.cover)?,
91        },
92    })
93}
rosu_memory_lib/reader/beatmap/stable/memory.rs

rosu_memory_lib/reader/beatmap/stable/
memory.rs
