diff --git a/Cargo.toml b/Cargo.toml index 8b6b139..3eaf2e8 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -8,13 +8,6 @@ members = [ "schemsearch-java" ] -[profile.test] -inherits = "release" -lto = true - -[profile.release] -lto = true - [profile.small] inherits = "release" lto = true diff --git a/schemsearch-cli/src/main.rs b/schemsearch-cli/src/main.rs index 3708a54..8dc424f 100644 --- a/schemsearch-cli/src/main.rs +++ b/schemsearch-cli/src/main.rs @@ -29,14 +29,14 @@ fn main() { ) .arg( Arg::new("ignore-block-entities") - .help("Ignores block entities when searching") + .help("Ignores block entities when searching [Not Implemented]") .short('b') .long("ignore-block-entities") .action(ArgAction::SetTrue), ) .arg( Arg::new("ignore-entities") - .help("Ignores entities when searching") + .help("Ignores entities when searching [Not Implemented]") .short('e') .long("ignore-entities") .action(ArgAction::SetTrue), @@ -184,7 +184,7 @@ fn search_schempath(cmd: &mut Command, search_behavior: SearchBehavior, pattern: writeln!(file_out.as_mut().unwrap(), "Searching in schematic: {}", schem_path.file_name().unwrap().to_str().unwrap()).unwrap(); } - let matches = search(&schematic, &pattern, search_behavior); + let matches = search(schematic, pattern, search_behavior); for x in matches { if *output_std { diff --git a/schemsearch-files/src/lib.rs b/schemsearch-files/src/lib.rs index 0b90cdb..804e328 100644 --- a/schemsearch-files/src/lib.rs +++ b/schemsearch-files/src/lib.rs @@ -34,7 +34,7 @@ fn read_blockdata<'de, D>(deserializer: D) -> Result, D::Error> where D: Deserializer<'de>, { - let s: Vec = Deserialize::deserialize(deserializer)?; + let s: Vec = Deserialize::deserialize(deserializer)?; Ok(read_varint_array(&s)) } @@ -68,7 +68,7 @@ impl Schematic { } } -pub fn read_varint_array(read: &Vec) -> Vec { +pub fn read_varint_array(read: &Vec) -> Vec { let mut data = Vec::new(); let mut value: i32 = 0; let mut position = 0; @@ -76,7 +76,7 @@ pub fn read_varint_array(read: &Vec) -> Vec { let mut cursor = 0; loop { match read.get(cursor) { - Some(byte) => { current_byte = *byte; cursor += 1; }, + Some(byte) => { current_byte = *byte as u8; cursor += 1; }, None => break, }; diff --git a/schemsearch-java/src/lib.rs b/schemsearch-java/src/lib.rs index 3c13f0b..dd20b3e 100644 --- a/schemsearch-java/src/lib.rs +++ b/schemsearch-java/src/lib.rs @@ -18,7 +18,7 @@ pub extern "system" fn Java_SchemSearch_search<'local>(mut env: JNIEnv<'local>, let schematic = Schematic::load(Path::new(&schematic_path)).unwrap(); let pattern = Schematic::load(Path::new(&pattern_path)).unwrap(); - let matches = search(&schematic, &pattern, SearchBehavior { + let matches = search(schematic, &pattern, SearchBehavior { ignore_block_data: true, ignore_block_entities: true, ignore_entities: true, diff --git a/schemsearch-lib/src/lib.rs b/schemsearch-lib/src/lib.rs index a1da5a4..00c3819 100644 --- a/schemsearch-lib/src/lib.rs +++ b/schemsearch-lib/src/lib.rs @@ -2,6 +2,7 @@ pub mod pattern_mapper; use pattern_mapper::match_palette; use schemsearch_files::Schematic; +use crate::pattern_mapper::match_palette_adapt; #[derive(Debug, Clone, Copy)] pub struct SearchBehavior { @@ -14,7 +15,7 @@ pub struct SearchBehavior { } pub fn search( - schem: &Schematic, + schem: Schematic, pattern_schem: &Schematic, search_behavior: SearchBehavior, ) -> Vec<(u16, u16, u16, f64)> { @@ -26,15 +27,18 @@ pub fn search( return vec![]; } - let pattern_schem = match match_palette(&schem, &pattern_schem, search_behavior.ignore_block_data) { - Some(x) => x, - None => return vec![], - }; + let pattern_schem = match_palette(&schem, &pattern_schem, search_behavior.ignore_block_data); let mut matches: Vec<(u16, u16, u16, f64)> = Vec::new(); let pattern_data = pattern_schem.block_data; - let schem_data = &schem.block_data; + + let schem_data = if search_behavior.ignore_block_data { + match_palette_adapt(&schem, &pattern_schem.palette, search_behavior.ignore_block_data) + } else { + schem.block_data + }; + let air_id = if search_behavior.ignore_air || search_behavior.air_as_any { pattern_schem.palette.get("minecraft:air").unwrap_or(&-1) } else { &-1}; let pattern_blocks = (pattern_schem.width * pattern_schem.height * pattern_schem.length) as f64; @@ -46,11 +50,11 @@ pub fn search( for i in 0..pattern_schem.width as usize { for j in 0..pattern_schem.height as usize { for k in 0..pattern_schem.length as usize { - let index = (x + i) + (y + j) * (schem.width as usize) + (z + k) * (schem.width as usize) * (schem.height as usize); - let pattern_index = i + j * pattern_schem.width as usize + k * pattern_schem.width as usize * pattern_schem.height as usize; + let index = (x + i) + (z + k) * (schem.width as usize) + (y + j) * (schem.width as usize) * (schem.length as usize); + let pattern_index = i + k * pattern_schem.width as usize + j * pattern_schem.width as usize * pattern_schem.length as usize; let data = schem_data.get(index as usize).expect("Index out of bounds"); let pattern_data = pattern_data.get(pattern_index as usize).expect("Index out of bounds"); - if data == pattern_data || (search_behavior.ignore_air && *data == *air_id) || (search_behavior.air_as_any && *pattern_data == *air_id) { + if *data == *pattern_data || (search_behavior.ignore_air && *data == *air_id) || (search_behavior.air_as_any && *pattern_data == *air_id) { matching += 1; } } @@ -136,7 +140,7 @@ mod tests { let schematic = Schematic::load(Path::new("../tests/simple.schem")).unwrap(); let endstone = Schematic::load(Path::new("../tests/endstone.schem")).unwrap(); - let _ = search(&schematic, &endstone, SearchBehavior { + let _ = search(schematic, &endstone, SearchBehavior { ignore_block_data: true, ignore_block_entities: true, ignore_entities: true, @@ -151,7 +155,7 @@ mod tests { let schematic = Schematic::load(Path::new("../tests/Random.schem")).unwrap(); let pattern = Schematic::load(Path::new("../tests/Pattern.schem")).unwrap(); - let matches = search(&schematic, &pattern, SearchBehavior { + let matches = search(schematic, &pattern, SearchBehavior { ignore_block_data: true, ignore_block_entities: true, ignore_entities: true, @@ -164,4 +168,22 @@ mod tests { assert_eq!(matches.len(), 1); assert_eq!(matches[0], (1, 0, 3, 1.0)); } + + #[test] + pub fn test_search_ws() { + let schematic = Schematic::load(Path::new("../tests/warships/GreyFly-by-Bosslar.schem")).unwrap(); + let pattern = Schematic::load(Path::new("../tests/gray_castle_complex.schem")).unwrap(); + + let matches = search(schematic, &pattern, SearchBehavior { + ignore_block_data: false, + ignore_block_entities: false, + ignore_entities: false, + ignore_air: false, + air_as_any: false, + threshold: 0.9 + }); + + println!("{:?}", matches); + assert_eq!(matches.len(), 1); + } } diff --git a/schemsearch-lib/src/pattern_mapper.rs b/schemsearch-lib/src/pattern_mapper.rs index 38a7b97..2f6e483 100644 --- a/schemsearch-lib/src/pattern_mapper.rs +++ b/schemsearch-lib/src/pattern_mapper.rs @@ -17,9 +17,8 @@ pub fn strip_data(schem: &Schematic) -> Schematic { let mut palette: Map = Map::new(); let mut palette_max: i32 = 0; let reverse_palette = create_reverse_palette(schem); - let dat = &schem.block_data; - for block in dat.iter() { + for block in schem.block_data.iter() { let block_name = reverse_palette[*block as usize].clone(); let block_name = block_name.split('[').next().unwrap().to_string(); @@ -47,27 +46,28 @@ pub fn strip_data(schem: &Schematic) -> Schematic { } } -fn match_palette_adapt(schem: &Schematic, matching_palette: &Map, ignore_data: bool) -> Option> { +pub fn match_palette_adapt(schem: &Schematic, matching_palette: &Map, ignore_data: bool) -> Vec { let mut data: Vec = Vec::new(); + let reverse_palette = create_reverse_palette(schem); - for x in schem.block_data.iter() { - let blockname = schem.palette.iter().find(|(_, &v)| v == *x).expect("Invalid Schematic").0; + for x in &schem.block_data { + let blockname = &reverse_palette[*x as usize]; let blockname = if ignore_data { normalize_data(&blockname, ignore_data) } else { blockname.clone() }; let block_id = match matching_palette.get(&blockname) { - None => return None, - Some(x) => x + None => -1, + Some(x) => *x }; - data.push(*block_id); + data.push(block_id); } - Some(data) + data } pub fn match_palette( schem: &Schematic, pattern: &Schematic, ignore_data: bool, -) -> Option { +) -> Schematic { if ignore_data { match_palette_internal(&strip_data(schem), &strip_data(pattern), ignore_data) } else { @@ -79,16 +79,10 @@ fn match_palette_internal( schem: &Schematic, pattern: &Schematic, ignore_data: bool, -) -> Option { - if pattern.palette.iter().any(|(k, _)| schem.palette.get(k).is_none()) { - return None; - } +) -> Schematic { + let data_pattern: Vec = match_palette_adapt(&pattern, &schem.palette, ignore_data); - let data_pattern: Vec = match match_palette_adapt(&pattern, &schem.palette, ignore_data) { - None => return None, - Some(x) => x - }; - Some(Schematic { + Schematic { version: pattern.version.clone(), data_version: pattern.data_version.clone(), palette: schem.palette.clone(), @@ -101,5 +95,5 @@ fn match_palette_internal( metadata: pattern.metadata.clone(), offset: pattern.offset.clone(), entities: None, - }) + } } \ No newline at end of file diff --git a/tests/endstone_large.schem b/tests/endstone_large.schem new file mode 100644 index 0000000..74b26ce Binary files /dev/null and b/tests/endstone_large.schem differ diff --git a/tests/gray_castle_complex.schem b/tests/gray_castle_complex.schem new file mode 100644 index 0000000..b889dcf Binary files /dev/null and b/tests/gray_castle_complex.schem differ diff --git a/tests/iron_block.schem b/tests/iron_block.schem new file mode 100644 index 0000000..d6d2210 Binary files /dev/null and b/tests/iron_block.schem differ diff --git a/tests/wargears/Blue_Fortress.schem b/tests/wargears/Blue_Fortress.schem new file mode 100644 index 0000000..b570c4b Binary files /dev/null and b/tests/wargears/Blue_Fortress.schem differ diff --git a/tests/wargears/Dear-God.schem b/tests/wargears/Dear-God.schem new file mode 100644 index 0000000..7e6b153 Binary files /dev/null and b/tests/wargears/Dear-God.schem differ diff --git a/tests/wargears/GrayCastle.schem b/tests/wargears/GrayCastle.schem new file mode 100644 index 0000000..95166c7 Binary files /dev/null and b/tests/wargears/GrayCastle.schem differ diff --git a/tests/wargears/PXSuperStarVI.schem b/tests/wargears/PXSuperStarVI.schem new file mode 100644 index 0000000..4e05949 Binary files /dev/null and b/tests/wargears/PXSuperStarVI.schem differ diff --git a/tests/wargears/Predator_X.schem b/tests/wargears/Predator_X.schem new file mode 100644 index 0000000..d5d1a0b Binary files /dev/null and b/tests/wargears/Predator_X.schem differ diff --git a/tests/wargears/WarpedOwl.schem b/tests/wargears/WarpedOwl.schem new file mode 100644 index 0000000..f09a27d Binary files /dev/null and b/tests/wargears/WarpedOwl.schem differ diff --git a/tests/warships/AaZ_Buttercreme.schem b/tests/warships/AaZ_Buttercreme.schem new file mode 100644 index 0000000..7d6bfc6 Binary files /dev/null and b/tests/warships/AaZ_Buttercreme.schem differ diff --git a/tests/warships/Donauwelle.schem b/tests/warships/Donauwelle.schem new file mode 100644 index 0000000..f3a2c5b Binary files /dev/null and b/tests/warships/Donauwelle.schem differ diff --git a/tests/warships/GreyFly-by-Bosslar.schem b/tests/warships/GreyFly-by-Bosslar.schem new file mode 100644 index 0000000..f9eb617 Binary files /dev/null and b/tests/warships/GreyFly-by-Bosslar.schem differ diff --git a/tests/warships/HMS-Invincible_by_aaZ.schem b/tests/warships/HMS-Invincible_by_aaZ.schem new file mode 100644 index 0000000..9190e9f Binary files /dev/null and b/tests/warships/HMS-Invincible_by_aaZ.schem differ diff --git a/tests/warships/HRMS_Rotterdam.schem b/tests/warships/HRMS_Rotterdam.schem new file mode 100644 index 0000000..bb08762 Binary files /dev/null and b/tests/warships/HRMS_Rotterdam.schem differ diff --git a/tests/warships/SMS_Helgoland.schem b/tests/warships/SMS_Helgoland.schem new file mode 100644 index 0000000..ee2f233 Binary files /dev/null and b/tests/warships/SMS_Helgoland.schem differ diff --git a/tests/warships/USS_Oklahoma.schem b/tests/warships/USS_Oklahoma.schem new file mode 100644 index 0000000..65f1136 Binary files /dev/null and b/tests/warships/USS_Oklahoma.schem differ