直接使用unsafe,效率++

This commit is contained in:
shenjack 2024-01-12 23:43:26 +08:00
parent 61801ba555
commit d5f80198c6
Signed by: shenjack
GPG Key ID: 7B1134A979775551

View File

@ -1,4 +1,4 @@
use std::{borrow::Cow, collections::HashMap, str::Bytes}; use std::{borrow::Cow, collections::HashMap};
/// NBT 里除了字符串的长度量都是 i32 /// NBT 里除了字符串的长度量都是 i32
#[allow(unused)] #[allow(unused)]
@ -16,9 +16,7 @@ pub struct NbtData {
#[allow(unused)] #[allow(unused)]
impl NbtData { impl NbtData {
pub fn new(data: Vec<u8>) -> Self { pub fn new(data: Vec<u8>) -> Self { Self { head: 0, data } }
Self { head: 0, data }
}
pub fn get_mut(&mut self) -> &mut [u8] { pub fn get_mut(&mut self) -> &mut [u8] {
let (_, data) = self.data.split_at_mut(self.head); let (_, data) = self.data.split_at_mut(self.head);
data data
@ -92,16 +90,56 @@ impl NbtData {
} }
} }
#[allow(unused)] pub mod raw_reading {
pub struct RawData { /// 多少有点脱裤子放屁
pub raw_data: Vec<u8>, pub fn slice_as_byte_array(slice: &[u8]) -> Vec<i8> {
pub length: usize, slice.iter().map(|&x| x as i8).collect::<Vec<i8>>()
} }
/// unsafe 从这里开始
#[allow(unused)] pub fn slice_as_short_array(slice: &[u8]) -> Option<&[i16]> {
impl RawData { let length = if slice.len() % 2 != 0 {
pub fn new(raw_data: Vec<u8>, length: usize) -> Self { return None;
Self { raw_data, length } } else {
(slice.len() / 2) as usize
};
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const i16, length) })
}
/// 开始 unsafe 了
/// unsafe rust, 小子!
pub fn slice_as_int_array(slice: &[u8]) -> Option<&[i32]> {
let length = if slice.len() % 4 != 0 {
return None;
} else {
(slice.len() / 4) as usize
};
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const i32, length) })
}
/// 这边也是 unsafe 捏
pub fn slice_as_long_array(slice: &[u8]) -> Option<&[i64]> {
let length = if slice.len() % 8 != 0 {
return None;
} else {
(slice.len() / 8) as usize
};
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const i64, length) })
}
/// 这边也是 unsafe 捏
pub fn slice_as_float_array(slice: &[u8]) -> Option<&[f32]> {
let length = if slice.len() % 4 != 0 {
return None;
} else {
(slice.len() / 4) as usize
};
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const f32, length) })
}
/// 这边也是 unsafe 捏
pub fn slice_as_double_array(slice: &[u8]) -> Option<&[f64]> {
let length = if slice.len() % 8 != 0 {
return None;
} else {
(slice.len() / 8) as usize
};
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const f64, length) })
} }
} }
@ -123,11 +161,11 @@ pub enum Value<'value> {
/// 8 /// 8
String(Cow<'value, str>), String(Cow<'value, str>),
/// 7 /// 7
ByteArray(RawData), ByteArray(Vec<i8>),
/// 11 /// 11
IntArray(RawData), IntArray(Vec<i32>),
/// 12 /// 12
LongArray(RawData), LongArray(Vec<i64>),
/// 9 /// 9
List(ListContent<'value>), List(ListContent<'value>),
/// 10 /// 10
@ -144,39 +182,27 @@ pub enum ListContent<'value> {
FloatList(Vec<f32>), FloatList(Vec<f32>),
DoubleList(Vec<f64>), DoubleList(Vec<f64>),
StringList(Vec<String>), StringList(Vec<String>),
ByteArrayList(Vec<RawData>), ByteArrayList(Vec<Vec<i8>>),
IntArrayList(Vec<RawData>), IntArrayList(Vec<Vec<i32>>),
LongArrayList(Vec<RawData>), LongArrayList(Vec<Vec<i64>>),
CompoundList(Vec<HashMap<String, Value<'value>>>), CompoundList(Vec<HashMap<String, Value<'value>>>),
ListList(Vec<ListContent<'value>>), ListList(Vec<ListContent<'value>>),
} }
// #[allow(unused)] #[allow(unused)]
impl<'value> Value<'value> { impl<'value> Value<'value> {
#[inline(always)] #[inline(always)]
pub fn read_byte(data: &mut NbtData) -> Self { pub fn read_byte(data: &mut NbtData) -> Self { Self::Byte(data.read_byte()) }
Self::Byte(data.read_byte())
}
#[inline(always)] #[inline(always)]
pub fn read_short(data: &mut NbtData) -> Self { pub fn read_short(data: &mut NbtData) -> Self { Self::Short(data.read_short()) }
Self::Short(data.read_short())
}
#[inline(always)] #[inline(always)]
pub fn read_int(data: &mut NbtData) -> Self { pub fn read_int(data: &mut NbtData) -> Self { Self::Int(data.read_int()) }
Self::Int(data.read_int())
}
#[inline(always)] #[inline(always)]
pub fn read_long(data: &mut NbtData) -> Self { pub fn read_long(data: &mut NbtData) -> Self { Self::Long(data.read_long()) }
Self::Long(data.read_long())
}
#[inline(always)] #[inline(always)]
pub fn read_float(data: &mut NbtData) -> Self { pub fn read_float(data: &mut NbtData) -> Self { Self::Float(data.read_float()) }
Self::Float(data.read_float())
}
#[inline(always)] #[inline(always)]
pub fn read_double(data: &mut NbtData) -> Self { pub fn read_double(data: &mut NbtData) -> Self { Self::Double(data.read_double()) }
Self::Double(data.read_double())
}
#[inline(always)] #[inline(always)]
pub fn read_string(data: &mut NbtData) -> Self { pub fn read_string(data: &mut NbtData) -> Self {
let length = data.read_short(); let length = data.read_short();
@ -191,56 +217,43 @@ impl<'value> Value<'value> {
match type_id { match type_id {
0 => panic!("WTF, type_id = 0"), 0 => panic!("WTF, type_id = 0"),
1 => { 1 => {
let mut list = Vec::with_capacity(length as usize); let raw_data = data.read_bytes(length as usize);
for _ in 0..length { let list = raw_reading::slice_as_byte_array(raw_data.as_slice());
list.push(data.read_byte());
}
Self::List(ListContent::ByteList(list)) Self::List(ListContent::ByteList(list))
} }
2 => { 2 => {
let mut list = Vec::with_capacity(length as usize); let raw_data = data.read_bytes(length as usize * 2);
for _ in 0..length { let list = raw_reading::slice_as_short_array(raw_data.as_slice()).unwrap().to_vec();
list.push(data.read_short());
}
Self::List(ListContent::ShortList(list)) Self::List(ListContent::ShortList(list))
} }
3 => { 3 => {
let mut list = Vec::with_capacity(length as usize); let raw_data = data.read_bytes(length as usize * 4);
for _ in 0..length { let list = raw_reading::slice_as_int_array(raw_data.as_slice()).unwrap().to_vec();
list.push(data.read_int());
}
Self::List(ListContent::IntList(list)) Self::List(ListContent::IntList(list))
} }
4 => { 4 => {
let mut list = Vec::with_capacity(length as usize); let raw_data = data.read_bytes(length as usize * 8);
for _ in 0..length { let list = raw_reading::slice_as_long_array(raw_data.as_slice()).unwrap().to_vec();
list.push(data.read_long());
}
Self::List(ListContent::LongList(list)) Self::List(ListContent::LongList(list))
} }
5 => { 5 => {
let mut list = Vec::with_capacity(length as usize); let raw_data = data.read_bytes(length as usize * 4);
for _ in 0..length { let list = raw_reading::slice_as_float_array(raw_data.as_slice()).unwrap().to_vec();
list.push(data.read_float());
}
Self::List(ListContent::FloatList(list)) Self::List(ListContent::FloatList(list))
} }
6 => { 6 => {
let mut list = Vec::with_capacity(length as usize); let raw_data = data.read_bytes(length as usize * 8);
for _ in 0..length { let list =
list.push(data.read_double()); raw_reading::slice_as_double_array(raw_data.as_slice()).unwrap().to_vec();
}
Self::List(ListContent::DoubleList(list)) Self::List(ListContent::DoubleList(list))
} }
7 => { 7 => {
let mut list = Vec::with_capacity(length as usize); let mut list = Vec::with_capacity(length as usize);
for _ in 0..length { for _ in 0..length {
let length = data.read_int(); let length = data.read_int();
let value = data.read_bytes(length as usize); let raw_data = data.read_bytes(length as usize);
list.push(RawData { let value = raw_reading::slice_as_byte_array(raw_data.as_slice());
raw_data: value, list.push(value);
length: length as usize,
});
} }
Self::List(ListContent::ByteArrayList(list)) Self::List(ListContent::ByteArrayList(list))
} }
@ -259,11 +272,109 @@ impl<'value> Value<'value> {
// 好好好, list 嵌套 list 是吧 // 好好好, list 嵌套 list 是吧
let mut list = Vec::with_capacity(length as usize); let mut list = Vec::with_capacity(length as usize);
for _ in 0..length { for _ in 0..length {
// let inner_list = Self::read_list(data); let inner_list = Self::read_list(data);
// list.push(inner_list);
} }
Self::List(ListContent::ListList(list)) Self::List(ListContent::ListList(list))
} }
10 => {
todo!("CompoundList, wait for Compound impl")
}
11 => {
let mut list = Vec::with_capacity(length as usize);
for _ in 0..length {
let length = data.read_int();
let raw_data = data.read_bytes(length as usize * 4);
let value =
raw_reading::slice_as_int_array(raw_data.as_slice()).unwrap().to_vec();
list.push(value);
}
Self::List(ListContent::IntArrayList(list))
}
12 => {
let mut list = Vec::with_capacity(length as usize);
for _ in 0..length {
let length = data.read_int();
let raw_data = data.read_bytes(length as usize * 8);
let value =
raw_reading::slice_as_long_array(raw_data.as_slice()).unwrap().to_vec();
list.push(value);
}
Self::List(ListContent::LongArrayList(list))
}
_ => panic!("WTF, type_id = {}", type_id), _ => panic!("WTF, type_id = {}", type_id),
} }
} }
pub fn as_byte(&self) -> Option<i8> {
match self {
Self::Byte(value) => Some(*value),
_ => None,
}
}
pub fn as_short(&self) -> Option<i16> {
match self {
Self::Short(value) => Some(*value),
_ => None,
}
}
pub fn as_int(&self) -> Option<i32> {
match self {
Self::Int(value) => Some(*value),
_ => None,
}
}
pub fn as_long(&self) -> Option<i64> {
match self {
Self::Long(value) => Some(*value),
_ => None,
}
}
pub fn as_float(&self) -> Option<f32> {
match self {
Self::Float(value) => Some(*value),
_ => None,
}
}
pub fn as_double(&self) -> Option<f64> {
match self {
Self::Double(value) => Some(*value),
_ => None,
}
}
pub fn as_string(&self) -> Option<&str> {
match self {
Self::String(value) => Some(value.as_ref()),
_ => None,
}
}
pub fn as_list(&self) -> Option<&ListContent> {
match self {
Self::List(value) => Some(value),
_ => None,
}
}
pub fn as_compound(&self) -> Option<&HashMap<Cow<'value, str>, Value<'value>>> {
match self {
Self::Compound(value) => Some(value),
_ => None,
}
}
pub fn as_byte_array(&self) -> Option<&[i8]> {
match self {
Self::ByteArray(value) => Some(value),
_ => None,
}
}
pub fn as_int_array(&self) -> Option<&[i32]> {
match self {
Self::IntArray(value) => Some(value),
_ => None,
}
}
pub fn as_long_array(&self) -> Option<&[i64]> {
match self {
Self::LongArray(value) => Some(value),
_ => None,
}
}
} }