Pumpkin/pumpkin-codegen/src/remap/sound_id.rs
Alexander Medvedev 23e7992b2d Port to 26.2
2026-06-20 20:07:31 +02:00

122 lines
4.2 KiB
Rust

use proc_macro2::{Literal, TokenStream};
use quote::{format_ident, quote};
use crate::remap::{MappingNode, ParsedMappings, Remapper};
use crate::version::JavaMinecraftVersion;
/// Generates the `TokenStream` for per-version sound ID remap tables and the
/// `remap_sound_id_for_version` function.
pub fn build() -> TokenStream {
let node_1_20_5 = MappingNode {
version: JavaMinecraftVersion::V_1_20_5,
value: "../assets/viabackwards/data/mappings-1.21to1.20.5.nbt",
child: None,
};
let node_1_21 = MappingNode {
version: JavaMinecraftVersion::V_1_21,
value: "../assets/viabackwards/data/mappings-1.21.2to1.21.nbt",
child: Some(&node_1_20_5),
};
let node_1_21_2 = MappingNode {
version: JavaMinecraftVersion::V_1_21_2,
value: "../assets/viabackwards/data/mappings-1.21.4to1.21.2.nbt",
child: Some(&node_1_21),
};
let node_1_21_4 = MappingNode {
version: JavaMinecraftVersion::V_1_21_4,
value: "../assets/viabackwards/data/mappings-1.21.5to1.21.4.nbt",
child: Some(&node_1_21_2),
};
let node_1_21_5 = MappingNode {
version: JavaMinecraftVersion::V_1_21_5,
value: "../assets/viabackwards/data/mappings-1.21.6to1.21.5.nbt",
child: Some(&node_1_21_4),
};
let node_1_21_6 = MappingNode {
version: JavaMinecraftVersion::V_1_21_6,
value: "../assets/viabackwards/data/mappings-1.21.7to1.21.6.nbt",
child: Some(&node_1_21_5),
};
let node_1_21_7 = MappingNode {
version: JavaMinecraftVersion::V_1_21_7,
value: "../assets/viabackwards/data/mappings-1.21.9to1.21.7.nbt",
child: Some(&node_1_21_6),
};
let node_1_21_9 = MappingNode {
version: JavaMinecraftVersion::V_1_21_9,
value: "../assets/viabackwards/data/mappings-1.21.11to1.21.9.nbt",
child: Some(&node_1_21_7),
};
let node_1_21_11 = MappingNode {
version: JavaMinecraftVersion::V_1_21_11,
value: "../assets/viabackwards/data/mappings-26.1to1.21.11.nbt",
child: Some(&node_1_21_9),
};
let node_26_1 = MappingNode {
version: JavaMinecraftVersion::V_26_1,
value: "../assets/viabackwards/data/mappings-26.2to26.1.nbt",
child: Some(&node_1_21_11),
};
let remapper: Remapper<_, Option<Vec<u16>>> = Remapper {
version: JavaMinecraftVersion::V_26_2,
remapper: |first, second| match (first, second) {
(Some(first), Some(second)) => Some(
first
.iter()
.map(|id| second.get(usize::from(*id)).copied().unwrap_or(0))
.collect(),
),
(None, Some(second)) => Some(
(0..second.len())
.map(|id| second.get(id).copied().unwrap_or(0))
.collect(),
),
(Some(first), None) => Some(first.clone()),
_ => None,
},
serializer: |&file| {
ParsedMappings::parse_mapping_file(file, "sounds").map(|mappings| mappings.to_u16(file))
},
};
let all_mappings = remapper.process(&node_26_1);
let mut static_values = TokenStream::new();
let mut match_arms = TokenStream::new();
for (ver, mapping) in &all_mappings {
let ident = format_ident!(
"{}",
format!("SOUND_ID_REMAP_{:?}_TO_{:?}", remapper.version, ver).to_uppercase()
);
let mapping_tokens: Vec<_> = mapping
.as_ref()
.unwrap()
.iter()
.copied()
.map(Literal::u16_unsuffixed)
.collect();
static_values.extend(quote! {
const #ident: &[u16] = &[#(#mapping_tokens),*];
});
match_arms.extend(quote! {
#ver => #ident
.get(usize::from(sound_id))
.copied()
.unwrap_or(sound_id),
});
}
quote! {
use pumpkin_util::version::JavaMinecraftVersion;
#static_values
#[must_use]
pub fn remap_sound_id_for_version(sound_id: u16, version: JavaMinecraftVersion) -> u16 {
match version {
#match_arms
_ => sound_id,
}
}
}
}