Mirror von
https://github.com/Chaoscaot/schemsearch
synchronisiert 2024-11-17 01:10:19 +01:00
Commits vergleichen
Keine gemeinsamen Commits. "a357da2ce8f3c4956ec004ac88d61d805450906c" und "25c4e97f714e083dc591778de4c350bca942763b" haben vollständig unterschiedliche Historien.
a357da2ce8
...
25c4e97f71
@ -39,7 +39,7 @@ use schemsearch_sql::load_all_schematics;
|
|||||||
#[cfg(feature = "sql")]
|
#[cfg(feature = "sql")]
|
||||||
use crate::types::SqlSchematicSupplier;
|
use crate::types::SqlSchematicSupplier;
|
||||||
use indicatif::*;
|
use indicatif::*;
|
||||||
use schemsearch_files::SpongeSchematic;
|
use schemsearch_files::{SchematicVersioned};
|
||||||
use crate::sinks::{OutputFormat, OutputSink};
|
use crate::sinks::{OutputFormat, OutputSink};
|
||||||
use crate::stderr::MaschineStdErr;
|
use crate::stderr::MaschineStdErr;
|
||||||
|
|
||||||
@ -206,7 +206,7 @@ fn main() {
|
|||||||
threshold: *matches.get_one::<f32>("threshold").expect("Couldn't get threshold"),
|
threshold: *matches.get_one::<f32>("threshold").expect("Couldn't get threshold"),
|
||||||
};
|
};
|
||||||
|
|
||||||
let pattern = match SpongeSchematic::load(&PathBuf::from(matches.get_one::<String>("pattern").unwrap())) {
|
let pattern = match SchematicVersioned::load(&PathBuf::from(matches.get_one::<String>("pattern").unwrap())) {
|
||||||
Ok(x) => x,
|
Ok(x) => x,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
cmd.error(ErrorKind::Io, format!("Error while loading Pattern: {}", e.to_string())).exit();
|
cmd.error(ErrorKind::Io, format!("Error while loading Pattern: {}", e.to_string())).exit();
|
||||||
@ -334,8 +334,8 @@ fn main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn load_schem(schem_path: &PathBuf) -> Option<SpongeSchematic> {
|
fn load_schem(schem_path: &PathBuf) -> Option<SchematicVersioned> {
|
||||||
match SpongeSchematic::load(schem_path) {
|
match SchematicVersioned::load(schem_path) {
|
||||||
Ok(x) => Some(x),
|
Ok(x) => Some(x),
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
println!("Error while loading schematic ({}): {}", schem_path.to_str().unwrap(), e.to_string());
|
println!("Error while loading schematic ({}): {}", schem_path.to_str().unwrap(), e.to_string());
|
||||||
|
@ -15,13 +15,11 @@
|
|||||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#[cfg(feature = "sql")]
|
|
||||||
use std::io::Cursor;
|
use std::io::Cursor;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
#[cfg(feature = "sql")]
|
#[cfg(feature = "sql")]
|
||||||
use futures::executor::block_on;
|
use futures::executor::block_on;
|
||||||
#[cfg(feature = "sql")]
|
use schemsearch_files::SchematicVersioned;
|
||||||
use schemsearch_files::SpongeSchematic;
|
|
||||||
#[cfg(feature = "sql")]
|
#[cfg(feature = "sql")]
|
||||||
use schemsearch_sql::{load_schemdata, SchematicNode};
|
use schemsearch_sql::{load_schemdata, SchematicNode};
|
||||||
|
|
||||||
@ -48,9 +46,9 @@ pub struct SqlSchematicSupplier {
|
|||||||
|
|
||||||
#[cfg(feature = "sql")]
|
#[cfg(feature = "sql")]
|
||||||
impl SqlSchematicSupplier {
|
impl SqlSchematicSupplier {
|
||||||
pub fn get_schematic(&self) -> Result<SpongeSchematic, String> {
|
pub fn get_schematic(&self) -> Result<SchematicVersioned, String> {
|
||||||
let mut schemdata = block_on(load_schemdata(self.node.id));
|
let mut schemdata = block_on(load_schemdata(self.node.id));
|
||||||
SpongeSchematic::load_data(&mut Cursor::new(schemdata.as_mut_slice()))
|
SchematicVersioned::load_data(&mut Cursor::new(schemdata.as_mut_slice()))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_name(&self) -> String {
|
pub fn get_name(&self) -> String {
|
||||||
|
@ -21,7 +21,92 @@ use std::path::PathBuf;
|
|||||||
use nbt::{CompoundTag, Tag};
|
use nbt::{CompoundTag, Tag};
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct SpongeSchematic {
|
pub enum SchematicVersioned {
|
||||||
|
V1(SpongeV1Schematic),
|
||||||
|
V2(SpongeV2Schematic),
|
||||||
|
V3(SpongeV3Schematic),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SchematicVersioned {
|
||||||
|
#[inline]
|
||||||
|
pub fn get_width(&self) -> u16 {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => schematic.width,
|
||||||
|
SchematicVersioned::V2(schematic) => schematic.width,
|
||||||
|
SchematicVersioned::V3(schematic) => schematic.width,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
pub fn get_height(&self) -> u16 {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => schematic.height,
|
||||||
|
SchematicVersioned::V2(schematic) => schematic.height,
|
||||||
|
SchematicVersioned::V3(schematic) => schematic.height,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
pub fn get_length(&self) -> u16 {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => schematic.length,
|
||||||
|
SchematicVersioned::V2(schematic) => schematic.length,
|
||||||
|
SchematicVersioned::V3(schematic) => schematic.length,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
pub fn get_palette_max(&self) -> i32 {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => schematic.palette_max,
|
||||||
|
SchematicVersioned::V2(schematic) => schematic.palette_max,
|
||||||
|
SchematicVersioned::V3(schematic) => schematic.blocks.palette.len() as i32,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
pub fn get_palette(&self) -> &HashMap<String, i32> {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => &schematic.palette,
|
||||||
|
SchematicVersioned::V2(schematic) => &schematic.palette,
|
||||||
|
SchematicVersioned::V3(schematic) => &schematic.blocks.palette,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
pub fn get_block_data(&self) -> &Vec<i32> {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => &schematic.block_data,
|
||||||
|
SchematicVersioned::V2(schematic) => &schematic.block_data,
|
||||||
|
SchematicVersioned::V3(schematic) => &schematic.blocks.block_data,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
pub fn get_block_entities(&self) -> &Vec<BlockEntity> {
|
||||||
|
return match self {
|
||||||
|
SchematicVersioned::V1(schematic) => &schematic.tile_entities,
|
||||||
|
SchematicVersioned::V2(schematic) => &schematic.block_entities,
|
||||||
|
SchematicVersioned::V3(schematic) => &schematic.blocks.block_entities,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct SpongeV1Schematic {
|
||||||
|
pub metadata: CompoundTag,
|
||||||
|
pub width: u16,
|
||||||
|
pub height: u16,
|
||||||
|
pub length: u16,
|
||||||
|
pub offset: [i32; 3],
|
||||||
|
pub palette_max: i32,
|
||||||
|
pub palette: HashMap<String, i32>,
|
||||||
|
pub block_data: Vec<i32>,
|
||||||
|
pub tile_entities: Vec<BlockEntity>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct SpongeV2Schematic {
|
||||||
pub data_version: i32,
|
pub data_version: i32,
|
||||||
pub metadata: CompoundTag,
|
pub metadata: CompoundTag,
|
||||||
pub width: u16,
|
pub width: u16,
|
||||||
@ -35,6 +120,18 @@ pub struct SpongeSchematic {
|
|||||||
pub entities: Option<Vec<Entity>>,
|
pub entities: Option<Vec<Entity>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct SpongeV3Schematic {
|
||||||
|
pub data_version: i32,
|
||||||
|
pub metadata: CompoundTag,
|
||||||
|
pub width: u16,
|
||||||
|
pub height: u16,
|
||||||
|
pub length: u16,
|
||||||
|
pub offset: [i32; 3],
|
||||||
|
pub blocks: BlockContainer,
|
||||||
|
pub entities: Option<Vec<Entity>>,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct BlockContainer {
|
pub struct BlockContainer {
|
||||||
pub palette: HashMap<String, i32>,
|
pub palette: HashMap<String, i32>,
|
||||||
@ -61,27 +158,28 @@ pub struct Entity {
|
|||||||
pub pos: [i32; 3],
|
pub pos: [i32; 3],
|
||||||
}
|
}
|
||||||
|
|
||||||
impl SpongeSchematic {
|
impl SchematicVersioned {
|
||||||
pub fn load_data<R>(data: &mut R) -> Result<SpongeSchematic, String> where R: Read {
|
pub fn load_data<R>(data: &mut R) -> Result<SchematicVersioned, String> where R: Read {
|
||||||
let nbt: CompoundTag = nbt::decode::read_gzip_compound_tag(data).map_err(|e| e.to_string())?;
|
let nbt: CompoundTag = nbt::decode::read_gzip_compound_tag(data).map_err(|e| e.to_string())?;
|
||||||
let version = nbt.get_i32("Version").map_err(|e| e.to_string())?;
|
let version = nbt.get_i32("Version").map_err(|e| e.to_string())?;
|
||||||
|
|
||||||
match version {
|
match version {
|
||||||
1 => SpongeSchematic::from_nbt_1(nbt),
|
1 => Ok(SchematicVersioned::V1(SpongeV1Schematic::from_nbt(nbt)?)),
|
||||||
2 => SpongeSchematic::from_nbt_2(nbt),
|
2 => Ok(SchematicVersioned::V2(SpongeV2Schematic::from_nbt(nbt)?)),
|
||||||
3 => SpongeSchematic::from_nbt_3(nbt),
|
3 => Ok(SchematicVersioned::V3(SpongeV3Schematic::from_nbt(nbt)?)),
|
||||||
_ => Err("Invalid schematic: Unknown Version".to_string()),
|
_ => Err("Invalid schematic: Unknown Version".to_string()),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn load(path: &PathBuf) -> Result<SpongeSchematic, String> {
|
pub fn load(path: &PathBuf) -> Result<SchematicVersioned, String> {
|
||||||
let mut file = std::fs::File::open(path).map_err(|e| e.to_string())?;
|
let mut file = std::fs::File::open(path).map_err(|e| e.to_string())?;
|
||||||
Self::load_data(&mut file)
|
Self::load_data(&mut file)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn from_nbt_1(nbt: CompoundTag) -> Result<Self, String> {
|
impl SpongeV1Schematic {
|
||||||
|
pub fn from_nbt(nbt: CompoundTag) -> Result<Self, String> {
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
data_version: 0,
|
|
||||||
metadata: nbt.get_compound_tag("Metadata").map_err(|e| e.to_string())?.clone(),
|
metadata: nbt.get_compound_tag("Metadata").map_err(|e| e.to_string())?.clone(),
|
||||||
width: nbt.get_i16("Width").map_err(|e| e.to_string())? as u16,
|
width: nbt.get_i16("Width").map_err(|e| e.to_string())? as u16,
|
||||||
height: nbt.get_i16("Height").map_err(|e| e.to_string())? as u16,
|
height: nbt.get_i16("Height").map_err(|e| e.to_string())? as u16,
|
||||||
@ -90,12 +188,13 @@ impl SpongeSchematic {
|
|||||||
palette_max: nbt.get_i32("PaletteMax").map_err(|e| e.to_string())?,
|
palette_max: nbt.get_i32("PaletteMax").map_err(|e| e.to_string())?,
|
||||||
palette: read_palette(nbt.get_compound_tag("Palette").map_err(|e| e.to_string())?),
|
palette: read_palette(nbt.get_compound_tag("Palette").map_err(|e| e.to_string())?),
|
||||||
block_data: read_blocks(nbt.get_i8_vec("BlockData").map_err(|e| e.to_string())?),
|
block_data: read_blocks(nbt.get_i8_vec("BlockData").map_err(|e| e.to_string())?),
|
||||||
block_entities: read_tile_entities(nbt.get_compound_tag_vec("TileEntities").map_err(|e| e.to_string())?)?,
|
tile_entities: read_tile_entities(nbt.get_compound_tag_vec("TileEntities").map_err(|e| e.to_string())?)?,
|
||||||
entities: None,
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn from_nbt_2(nbt: CompoundTag) -> Result<Self, String> {
|
impl SpongeV2Schematic {
|
||||||
|
pub fn from_nbt(nbt: CompoundTag) -> Result<Self, String> {
|
||||||
Ok(Self{
|
Ok(Self{
|
||||||
data_version: nbt.get_i32("DataVersion").map_err(|e| e.to_string())?,
|
data_version: nbt.get_i32("DataVersion").map_err(|e| e.to_string())?,
|
||||||
metadata: nbt.get_compound_tag("Metadata").map_err(|e| e.to_string())?.clone(),
|
metadata: nbt.get_compound_tag("Metadata").map_err(|e| e.to_string())?.clone(),
|
||||||
@ -110,8 +209,10 @@ impl SpongeSchematic {
|
|||||||
entities: None,
|
entities: None,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn from_nbt_3(nbt: CompoundTag) -> Result<Self, String> {
|
impl SpongeV3Schematic {
|
||||||
|
pub fn from_nbt(nbt: CompoundTag) -> Result<Self, String> {
|
||||||
let blocks = nbt.get_compound_tag("Blocks").map_err(|e| e.to_string())?;
|
let blocks = nbt.get_compound_tag("Blocks").map_err(|e| e.to_string())?;
|
||||||
Ok(Self{
|
Ok(Self{
|
||||||
data_version: nbt.get_i32("DataVersion").map_err(|e| e.to_string())?,
|
data_version: nbt.get_i32("DataVersion").map_err(|e| e.to_string())?,
|
||||||
@ -120,14 +221,14 @@ impl SpongeSchematic {
|
|||||||
height: nbt.get_i16("Height").map_err(|e| e.to_string())? as u16,
|
height: nbt.get_i16("Height").map_err(|e| e.to_string())? as u16,
|
||||||
length: nbt.get_i16("Length").map_err(|e| e.to_string())? as u16,
|
length: nbt.get_i16("Length").map_err(|e| e.to_string())? as u16,
|
||||||
offset: read_offset(nbt.get_i32_vec("Offset").map_err(|e| e.to_string())?)?,
|
offset: read_offset(nbt.get_i32_vec("Offset").map_err(|e| e.to_string())?)?,
|
||||||
palette_max: compute_palette_max(blocks.get_compound_tag("Palette").map_err(|e| e.to_string())?),
|
blocks: BlockContainer {
|
||||||
palette: read_palette(blocks.get_compound_tag("Palette").map_err(|e| e.to_string())?),
|
palette: read_palette(blocks.get_compound_tag("Palette").map_err(|e| e.to_string())?),
|
||||||
block_data: read_blocks(blocks.get_i8_vec("BlockData").map_err(|e| e.to_string())?),
|
block_data: read_blocks(blocks.get_i8_vec("BlockData").map_err(|e| e.to_string())?),
|
||||||
block_entities: read_tile_entities(blocks.get_compound_tag_vec("BlockEntities").map_err(|e| e.to_string())?)?,
|
block_entities: read_tile_entities(blocks.get_compound_tag_vec("BlockEntities").map_err(|e| e.to_string())?)?,
|
||||||
|
},
|
||||||
entities: None,
|
entities: None,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_tile_entities(tag: Vec<&CompoundTag>) -> Result<Vec<BlockEntity>, String> {
|
fn read_tile_entities(tag: Vec<&CompoundTag>) -> Result<Vec<BlockEntity>, String> {
|
||||||
@ -161,14 +262,6 @@ fn read_palette(p: &CompoundTag) -> HashMap<String, i32> {
|
|||||||
palette
|
palette
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
|
||||||
fn compute_palette_max(palette: &CompoundTag) -> i32 {
|
|
||||||
palette.iter().map(|(_, v)| v).filter_map(|v| match v {
|
|
||||||
Tag::Int(n) => Some(*n),
|
|
||||||
_ => None,
|
|
||||||
}).max().unwrap_or(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn read_blocks(blockdata: &Vec<i8>) -> Vec<i32> {
|
fn read_blocks(blockdata: &Vec<i8>) -> Vec<i32> {
|
||||||
read_varint_array(blockdata)
|
read_varint_array(blockdata)
|
||||||
|
@ -19,7 +19,7 @@ pub mod pattern_mapper;
|
|||||||
|
|
||||||
use serde::{Serialize, Deserialize};
|
use serde::{Serialize, Deserialize};
|
||||||
use pattern_mapper::match_palette;
|
use pattern_mapper::match_palette;
|
||||||
use schemsearch_files::SpongeSchematic;
|
use schemsearch_files::SchematicVersioned;
|
||||||
use crate::pattern_mapper::match_palette_adapt;
|
use crate::pattern_mapper::match_palette_adapt;
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy, Deserialize, Serialize)]
|
#[derive(Debug, Clone, Copy, Deserialize, Serialize)]
|
||||||
@ -33,15 +33,15 @@ pub struct SearchBehavior {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn search(
|
pub fn search(
|
||||||
schem: SpongeSchematic,
|
schem: SchematicVersioned,
|
||||||
pattern_schem: &SpongeSchematic,
|
pattern_schem: &SchematicVersioned,
|
||||||
search_behavior: SearchBehavior,
|
search_behavior: SearchBehavior,
|
||||||
) -> Vec<Match> {
|
) -> Vec<Match> {
|
||||||
if schem.width < pattern_schem.width || schem.height < pattern_schem.height || schem.length < pattern_schem.length {
|
if schem.get_width() < pattern_schem.get_width() || schem.get_height() < pattern_schem.get_height() || schem.get_length() < pattern_schem.get_length() {
|
||||||
return vec![];
|
return vec![];
|
||||||
}
|
}
|
||||||
|
|
||||||
if pattern_schem.palette.len() > schem.palette.len() {
|
if pattern_schem.get_palette().len() > schem.get_palette().len() {
|
||||||
return vec![];
|
return vec![];
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -49,27 +49,27 @@ pub fn search(
|
|||||||
|
|
||||||
let mut matches: Vec<Match> = Vec::new();
|
let mut matches: Vec<Match> = Vec::new();
|
||||||
|
|
||||||
let pattern_data = pattern_schem.block_data.as_slice();
|
let pattern_data = pattern_schem.get_block_data().as_slice();
|
||||||
|
|
||||||
let schem_data = if search_behavior.ignore_block_data {
|
let schem_data = if search_behavior.ignore_block_data {
|
||||||
match_palette_adapt(&schem, &pattern_schem.palette, search_behavior.ignore_block_data)
|
match_palette_adapt(&schem, &pattern_schem.get_palette(), search_behavior.ignore_block_data)
|
||||||
} else {
|
} else {
|
||||||
schem.block_data.clone()
|
schem.get_block_data().clone()
|
||||||
};
|
};
|
||||||
|
|
||||||
let schem_data = schem_data.as_slice();
|
let schem_data = schem_data.as_slice();
|
||||||
|
|
||||||
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 air_id = if search_behavior.ignore_air || search_behavior.air_as_any { pattern_schem.get_palette().get("minecraft:air").unwrap_or(&-1) } else { &-1};
|
||||||
|
|
||||||
let pattern_blocks = pattern_data.len() as f32;
|
let pattern_blocks = pattern_data.len() as f32;
|
||||||
|
|
||||||
let pattern_width = pattern_schem.width as usize;
|
let pattern_width = pattern_schem.get_width() as usize;
|
||||||
let pattern_height = pattern_schem.height as usize;
|
let pattern_height = pattern_schem.get_height() as usize;
|
||||||
let pattern_length = pattern_schem.length as usize;
|
let pattern_length = pattern_schem.get_length() as usize;
|
||||||
|
|
||||||
let schem_width = schem.width as usize;
|
let schem_width = schem.get_width() as usize;
|
||||||
let schem_height = schem.height as usize;
|
let schem_height = schem.get_height() as usize;
|
||||||
let schem_length = schem.length as usize;
|
let schem_length = schem.get_length() as usize;
|
||||||
|
|
||||||
for y in 0..=schem_height - pattern_height {
|
for y in 0..=schem_height - pattern_height {
|
||||||
for z in 0..=schem_length - pattern_length {
|
for z in 0..=schem_length - pattern_length {
|
||||||
@ -136,12 +136,19 @@ pub fn normalize_data(data: &str, ignore_data: bool) -> &str {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use std::path::{Path, PathBuf};
|
use std::path::{Path, PathBuf};
|
||||||
|
use schemsearch_files::SchematicVersioned::V2;
|
||||||
|
use schemsearch_files::SpongeV2Schematic;
|
||||||
use crate::pattern_mapper::strip_data;
|
use crate::pattern_mapper::strip_data;
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn read_schematic() {
|
fn read_schematic() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
||||||
|
|
||||||
|
let schematic = match schematic {
|
||||||
|
V2 (schematic) => schematic,
|
||||||
|
_ => panic!("Invalid schematic version"),
|
||||||
|
};
|
||||||
|
|
||||||
assert_eq!(schematic.width as usize * schematic.height as usize * schematic.length as usize, schematic.block_data.len());
|
assert_eq!(schematic.width as usize * schematic.height as usize * schematic.length as usize, schematic.block_data.len());
|
||||||
assert_eq!(schematic.palette_max, schematic.palette.len() as i32);
|
assert_eq!(schematic.palette_max, schematic.palette.len() as i32);
|
||||||
@ -149,7 +156,12 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_parse_function() {
|
fn test_parse_function() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
let schematic: SchematicVersioned = SchematicVersioned::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
||||||
|
|
||||||
|
let schematic = match schematic {
|
||||||
|
V2 (schematic) => schematic,
|
||||||
|
_ => panic!("Invalid schematic version"),
|
||||||
|
};
|
||||||
|
|
||||||
assert_eq!(schematic.width as usize * schematic.height as usize * schematic.length as usize, schematic.block_data.len());
|
assert_eq!(schematic.width as usize * schematic.height as usize * schematic.length as usize, schematic.block_data.len());
|
||||||
assert_eq!(schematic.palette_max, schematic.palette.len() as i32);
|
assert_eq!(schematic.palette_max, schematic.palette.len() as i32);
|
||||||
@ -157,32 +169,32 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_strip_schem() {
|
fn test_strip_schem() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
||||||
let stripped = strip_data(&schematic);
|
let stripped = strip_data(&schematic);
|
||||||
|
|
||||||
assert_eq!(stripped.palette.keys().any(|k| k.contains('[')), false);
|
assert_eq!(stripped.get_palette().keys().any(|k| k.contains('[')), false);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_match_palette() {
|
fn test_match_palette() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
||||||
let endstone = SpongeSchematic::load(&PathBuf::from("../tests/endstone.schem")).unwrap();
|
let endstone = SchematicVersioned::load(&PathBuf::from("../tests/endstone.schem")).unwrap();
|
||||||
|
|
||||||
let _ = match_palette(&schematic, &endstone, true);
|
let _ = match_palette(&schematic, &endstone, true);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_match_palette_ignore_data() {
|
fn test_match_palette_ignore_data() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
||||||
let endstone = SpongeSchematic::load(&PathBuf::from("../tests/endstone.schem")).unwrap();
|
let endstone = SchematicVersioned::load(&PathBuf::from("../tests/endstone.schem")).unwrap();
|
||||||
|
|
||||||
let _ = match_palette(&schematic, &endstone, false);
|
let _ = match_palette(&schematic, &endstone, false);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
pub fn test_big_search() {
|
pub fn test_big_search() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/simple.schem")).unwrap();
|
||||||
let endstone = SpongeSchematic::load(&PathBuf::from("../tests/endstone.schem")).unwrap();
|
let endstone = SchematicVersioned::load(&PathBuf::from("../tests/endstone.schem")).unwrap();
|
||||||
|
|
||||||
let _ = search(schematic, &endstone, SearchBehavior {
|
let _ = search(schematic, &endstone, SearchBehavior {
|
||||||
ignore_block_data: true,
|
ignore_block_data: true,
|
||||||
@ -196,8 +208,8 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
pub fn test_search() {
|
pub fn test_search() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/Random.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/Random.schem")).unwrap();
|
||||||
let pattern = SpongeSchematic::load(&PathBuf::from("../tests/Pattern.schem")).unwrap();
|
let pattern = SchematicVersioned::load(&PathBuf::from("../tests/Pattern.schem")).unwrap();
|
||||||
|
|
||||||
let matches = search(schematic, &pattern, SearchBehavior {
|
let matches = search(schematic, &pattern, SearchBehavior {
|
||||||
ignore_block_data: true,
|
ignore_block_data: true,
|
||||||
@ -218,8 +230,8 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
pub fn test_search_ws() {
|
pub fn test_search_ws() {
|
||||||
let schematic = SpongeSchematic::load(&PathBuf::from("../tests/warships/GreyFly-by-Bosslar.schem")).unwrap();
|
let schematic = SchematicVersioned::load(&PathBuf::from("../tests/warships/GreyFly-by-Bosslar.schem")).unwrap();
|
||||||
let pattern = SpongeSchematic::load(&PathBuf::from("../tests/gray_castle_complex.schem")).unwrap();
|
let pattern = SchematicVersioned::load(&PathBuf::from("../tests/gray_castle_complex.schem")).unwrap();
|
||||||
|
|
||||||
let matches = search(schematic, &pattern, SearchBehavior {
|
let matches = search(schematic, &pattern, SearchBehavior {
|
||||||
ignore_block_data: false,
|
ignore_block_data: false,
|
||||||
|
@ -17,26 +17,26 @@
|
|||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use nbt::CompoundTag;
|
use nbt::CompoundTag;
|
||||||
use schemsearch_files::SpongeSchematic;
|
use schemsearch_files::{SchematicVersioned, SpongeV2Schematic};
|
||||||
use crate::normalize_data;
|
use crate::normalize_data;
|
||||||
|
|
||||||
fn create_reverse_palette(schem: &SpongeSchematic) -> Vec<&str> {
|
fn create_reverse_palette(schem: &SchematicVersioned) -> Vec<&str> {
|
||||||
let mut reverse_palette = Vec::with_capacity(schem.palette_max as usize);
|
let mut reverse_palette = Vec::with_capacity(schem.get_palette_max() as usize);
|
||||||
(0..schem.palette_max).for_each(|_| reverse_palette.push(""));
|
(0..schem.get_palette_max()).for_each(|_| reverse_palette.push(""));
|
||||||
for (key, value) in schem.palette.iter() {
|
for (key, value) in schem.get_palette().iter() {
|
||||||
reverse_palette[*value as usize] = key;
|
reverse_palette[*value as usize] = key;
|
||||||
}
|
}
|
||||||
reverse_palette
|
reverse_palette
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn strip_data(schem: &SpongeSchematic) -> SpongeSchematic {
|
pub fn strip_data(schem: &SchematicVersioned) -> SchematicVersioned {
|
||||||
let mut data: Vec<i32> = Vec::new();
|
let mut data: Vec<i32> = Vec::new();
|
||||||
|
|
||||||
let mut palette: HashMap<String, i32> = HashMap::new();
|
let mut palette: HashMap<String, i32> = HashMap::new();
|
||||||
let mut palette_max: i32 = 0;
|
let mut palette_max: i32 = 0;
|
||||||
let reverse_palette = create_reverse_palette(schem);
|
let reverse_palette = create_reverse_palette(schem);
|
||||||
|
|
||||||
for block in schem.block_data.iter() {
|
for block in schem.get_block_data().iter() {
|
||||||
let block_name = reverse_palette[*block as usize].clone();
|
let block_name = reverse_palette[*block as usize].clone();
|
||||||
let block_name = block_name.split('[').next().unwrap().to_string();
|
let block_name = block_name.split('[').next().unwrap().to_string();
|
||||||
|
|
||||||
@ -48,28 +48,28 @@ pub fn strip_data(schem: &SpongeSchematic) -> SpongeSchematic {
|
|||||||
data.push(*entry);
|
data.push(*entry);
|
||||||
}
|
}
|
||||||
|
|
||||||
SpongeSchematic {
|
SchematicVersioned::V2(SpongeV2Schematic {
|
||||||
data_version: 1,
|
data_version: 1,
|
||||||
palette,
|
palette,
|
||||||
palette_max,
|
palette_max,
|
||||||
block_data: data,
|
block_data: data,
|
||||||
block_entities: schem.block_entities.clone(),
|
block_entities: schem.get_block_entities().clone(),
|
||||||
height: schem.height,
|
height: schem.get_height(),
|
||||||
length: schem.length,
|
length: schem.get_length(),
|
||||||
width: schem.width,
|
width: schem.get_width(),
|
||||||
metadata: CompoundTag::new(),
|
metadata: CompoundTag::new(),
|
||||||
offset: [0; 3],
|
offset: [0; 3],
|
||||||
entities: None,
|
entities: None,
|
||||||
}
|
},)
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn match_palette_adapt(schem: &SpongeSchematic, matching_palette: &HashMap<String, i32>, ignore_data: bool) -> Vec<i32> {
|
pub fn match_palette_adapt(schem: &SchematicVersioned, matching_palette: &HashMap<String, i32>, ignore_data: bool) -> Vec<i32> {
|
||||||
let mut data: Vec<i32> = Vec::new();
|
let mut data: Vec<i32> = Vec::new();
|
||||||
let reverse_palette = create_reverse_palette(schem);
|
let reverse_palette = create_reverse_palette(schem);
|
||||||
|
|
||||||
for x in schem.block_data.iter() {
|
for x in schem.get_block_data() {
|
||||||
let blockname = reverse_palette[*x as usize];
|
let blockname = reverse_palette[*x as usize];
|
||||||
let blockname = if ignore_data { normalize_data(blockname, ignore_data) } else { blockname };
|
let blockname = if ignore_data { normalize_data(blockname, ignore_data) } else { blockname };
|
||||||
let block_id = match matching_palette.get(&*blockname) {
|
let block_id = match matching_palette.get(&*blockname) {
|
||||||
@ -83,10 +83,10 @@ pub fn match_palette_adapt(schem: &SpongeSchematic, matching_palette: &HashMap<S
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn match_palette(
|
pub fn match_palette(
|
||||||
schem: &SpongeSchematic,
|
schem: &SchematicVersioned,
|
||||||
pattern: &SpongeSchematic,
|
pattern: &SchematicVersioned,
|
||||||
ignore_data: bool,
|
ignore_data: bool,
|
||||||
) -> SpongeSchematic {
|
) -> SchematicVersioned {
|
||||||
if ignore_data {
|
if ignore_data {
|
||||||
match_palette_internal(&strip_data(schem), &strip_data(pattern), ignore_data)
|
match_palette_internal(&strip_data(schem), &strip_data(pattern), ignore_data)
|
||||||
} else {
|
} else {
|
||||||
@ -95,23 +95,23 @@ pub fn match_palette(
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn match_palette_internal(
|
fn match_palette_internal(
|
||||||
schem: &SpongeSchematic,
|
schem: &SchematicVersioned,
|
||||||
pattern: &SpongeSchematic,
|
pattern: &SchematicVersioned,
|
||||||
ignore_data: bool,
|
ignore_data: bool,
|
||||||
) -> SpongeSchematic {
|
) -> SchematicVersioned {
|
||||||
let data_pattern: Vec<i32> = match_palette_adapt(&pattern, &schem.palette, ignore_data);
|
let data_pattern: Vec<i32> = match_palette_adapt(&pattern, schem.get_palette(), ignore_data);
|
||||||
|
|
||||||
SpongeSchematic {
|
SchematicVersioned::V2(SpongeV2Schematic {
|
||||||
data_version: 0,
|
data_version: 0,
|
||||||
palette: schem.palette.clone(),
|
palette: schem.get_palette().clone(),
|
||||||
palette_max: schem.palette_max,
|
palette_max: schem.get_palette_max(),
|
||||||
block_data: data_pattern,
|
block_data: data_pattern,
|
||||||
block_entities: pattern.block_entities.clone(),
|
block_entities: pattern.get_block_entities().clone(),
|
||||||
height: pattern.height,
|
height: pattern.get_height(),
|
||||||
length: pattern.length,
|
length: pattern.get_length(),
|
||||||
width: pattern.width,
|
width: pattern.get_width(),
|
||||||
metadata: CompoundTag::new(),
|
metadata: CompoundTag::new(),
|
||||||
offset: [0; 3],
|
offset: [0; 3],
|
||||||
entities: None,
|
entities: None,
|
||||||
}
|
})
|
||||||
}
|
}
|
Laden…
In neuem Issue referenzieren
Einen Benutzer sperren