mirror of
https://github.com/mat-1/azalea.git
synced 2025-08-02 14:26:04 +00:00
codegen entity dimensions
This commit is contained in:
parent
e03101a7b7
commit
a5cb21f039
4 changed files with 201 additions and 5 deletions
|
|
@ -1,4 +1,5 @@
|
|||
use azalea_core::{aabb::AABB, position::Vec3};
|
||||
use azalea_registry::EntityKind;
|
||||
|
||||
#[derive(Debug, Default, Clone)]
|
||||
pub struct EntityDimensions {
|
||||
|
|
@ -7,6 +8,10 @@ pub struct EntityDimensions {
|
|||
}
|
||||
|
||||
impl EntityDimensions {
|
||||
pub fn new(width: f32, height: f32) -> Self {
|
||||
Self { width, height }
|
||||
}
|
||||
|
||||
pub fn make_bounding_box(&self, pos: &Vec3) -> AABB {
|
||||
let radius = (self.width / 2.0) as f64;
|
||||
let height = self.height as f64;
|
||||
|
|
@ -16,3 +21,161 @@ impl EntityDimensions {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<EntityKind> for EntityDimensions {
|
||||
fn from(entity: EntityKind) -> Self {
|
||||
// this match statement is automatically generated by codegen/genentities.py,
|
||||
// don't edit it manually!
|
||||
match entity {
|
||||
EntityKind::AcaciaBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::AcaciaChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Allay => EntityDimensions::new(0.35, 0.6),
|
||||
EntityKind::AreaEffectCloud => EntityDimensions::new(6.0, 0.5),
|
||||
EntityKind::Armadillo => EntityDimensions::new(0.7, 0.65),
|
||||
EntityKind::ArmorStand => EntityDimensions::new(0.5, 1.975),
|
||||
EntityKind::Arrow => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::Axolotl => EntityDimensions::new(0.75, 0.42),
|
||||
EntityKind::BambooChestRaft => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::BambooRaft => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Bat => EntityDimensions::new(0.5, 0.9),
|
||||
EntityKind::Bee => EntityDimensions::new(0.7, 0.6),
|
||||
EntityKind::BirchBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::BirchChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Blaze => EntityDimensions::new(0.6, 1.8),
|
||||
EntityKind::BlockDisplay => EntityDimensions::new(0.0, 0.0),
|
||||
EntityKind::Bogged => EntityDimensions::new(0.6, 1.99),
|
||||
EntityKind::Breeze => EntityDimensions::new(0.6, 1.77),
|
||||
EntityKind::BreezeWindCharge => EntityDimensions::new(0.3125, 0.3125),
|
||||
EntityKind::Camel => EntityDimensions::new(1.7, 2.375),
|
||||
EntityKind::Cat => EntityDimensions::new(0.6, 0.7),
|
||||
EntityKind::CaveSpider => EntityDimensions::new(0.7, 0.5),
|
||||
EntityKind::CherryBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::CherryChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::ChestMinecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::Chicken => EntityDimensions::new(0.4, 0.7),
|
||||
EntityKind::Cod => EntityDimensions::new(0.5, 0.3),
|
||||
EntityKind::CommandBlockMinecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::Cow => EntityDimensions::new(0.9, 1.4),
|
||||
EntityKind::Creaking => EntityDimensions::new(0.9, 2.7),
|
||||
EntityKind::Creeper => EntityDimensions::new(0.6, 1.7),
|
||||
EntityKind::DarkOakBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::DarkOakChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Dolphin => EntityDimensions::new(0.9, 0.6),
|
||||
EntityKind::Donkey => EntityDimensions::new(1.39648, 1.5),
|
||||
EntityKind::DragonFireball => EntityDimensions::new(1.0, 1.0),
|
||||
EntityKind::Drowned => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Egg => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::ElderGuardian => EntityDimensions::new(1.9975, 1.9975),
|
||||
EntityKind::EndCrystal => EntityDimensions::new(2.0, 2.0),
|
||||
EntityKind::EnderDragon => EntityDimensions::new(16.0, 8.0),
|
||||
EntityKind::EnderPearl => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::Enderman => EntityDimensions::new(0.6, 2.9),
|
||||
EntityKind::Endermite => EntityDimensions::new(0.4, 0.3),
|
||||
EntityKind::Evoker => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::EvokerFangs => EntityDimensions::new(0.5, 0.8),
|
||||
EntityKind::ExperienceBottle => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::ExperienceOrb => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::EyeOfEnder => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::FallingBlock => EntityDimensions::new(0.98, 0.98),
|
||||
EntityKind::Fireball => EntityDimensions::new(1.0, 1.0),
|
||||
EntityKind::FireworkRocket => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::FishingBobber => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::Fox => EntityDimensions::new(0.6, 0.7),
|
||||
EntityKind::Frog => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::FurnaceMinecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::Ghast => EntityDimensions::new(4.0, 4.0),
|
||||
EntityKind::Giant => EntityDimensions::new(3.6, 12.0),
|
||||
EntityKind::GlowItemFrame => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::GlowSquid => EntityDimensions::new(0.8, 0.8),
|
||||
EntityKind::Goat => EntityDimensions::new(0.9, 1.3),
|
||||
EntityKind::Guardian => EntityDimensions::new(0.85, 0.85),
|
||||
EntityKind::Hoglin => EntityDimensions::new(1.39648, 1.4),
|
||||
EntityKind::HopperMinecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::Horse => EntityDimensions::new(1.39648, 1.6),
|
||||
EntityKind::Husk => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Illusioner => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Interaction => EntityDimensions::new(0.0, 0.0),
|
||||
EntityKind::IronGolem => EntityDimensions::new(1.4, 2.7),
|
||||
EntityKind::Item => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::ItemDisplay => EntityDimensions::new(0.0, 0.0),
|
||||
EntityKind::ItemFrame => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::JungleBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::JungleChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::LeashKnot => EntityDimensions::new(0.375, 0.5),
|
||||
EntityKind::LightningBolt => EntityDimensions::new(0.0, 0.0),
|
||||
EntityKind::Llama => EntityDimensions::new(0.9, 1.87),
|
||||
EntityKind::LlamaSpit => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::MagmaCube => EntityDimensions::new(0.52, 0.52),
|
||||
EntityKind::MangroveBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::MangroveChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Marker => EntityDimensions::new(0.0, 0.0),
|
||||
EntityKind::Minecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::Mooshroom => EntityDimensions::new(0.9, 1.4),
|
||||
EntityKind::Mule => EntityDimensions::new(1.39648, 1.6),
|
||||
EntityKind::OakBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::OakChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Ocelot => EntityDimensions::new(0.6, 0.7),
|
||||
EntityKind::OminousItemSpawner => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::Painting => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::PaleOakBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::PaleOakChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Panda => EntityDimensions::new(1.3, 1.25),
|
||||
EntityKind::Parrot => EntityDimensions::new(0.5, 0.9),
|
||||
EntityKind::Phantom => EntityDimensions::new(0.9, 0.5),
|
||||
EntityKind::Pig => EntityDimensions::new(0.9, 0.9),
|
||||
EntityKind::Piglin => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::PiglinBrute => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Pillager => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Player => EntityDimensions::new(0.6, 1.8),
|
||||
EntityKind::PolarBear => EntityDimensions::new(1.4, 1.4),
|
||||
EntityKind::Potion => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::Pufferfish => EntityDimensions::new(0.7, 0.7),
|
||||
EntityKind::Rabbit => EntityDimensions::new(0.4, 0.5),
|
||||
EntityKind::Ravager => EntityDimensions::new(1.95, 2.2),
|
||||
EntityKind::Salmon => EntityDimensions::new(0.7, 0.4),
|
||||
EntityKind::Sheep => EntityDimensions::new(0.9, 1.3),
|
||||
EntityKind::Shulker => EntityDimensions::new(1.0, 1.0),
|
||||
EntityKind::ShulkerBullet => EntityDimensions::new(0.3125, 0.3125),
|
||||
EntityKind::Silverfish => EntityDimensions::new(0.4, 0.3),
|
||||
EntityKind::Skeleton => EntityDimensions::new(0.6, 1.99),
|
||||
EntityKind::SkeletonHorse => EntityDimensions::new(1.39648, 1.6),
|
||||
EntityKind::Slime => EntityDimensions::new(0.52, 0.52),
|
||||
EntityKind::SmallFireball => EntityDimensions::new(0.3125, 0.3125),
|
||||
EntityKind::Sniffer => EntityDimensions::new(1.9, 1.75),
|
||||
EntityKind::SnowGolem => EntityDimensions::new(0.7, 1.9),
|
||||
EntityKind::Snowball => EntityDimensions::new(0.25, 0.25),
|
||||
EntityKind::SpawnerMinecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::SpectralArrow => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::Spider => EntityDimensions::new(1.4, 0.9),
|
||||
EntityKind::SpruceBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::SpruceChestBoat => EntityDimensions::new(1.375, 0.5625),
|
||||
EntityKind::Squid => EntityDimensions::new(0.8, 0.8),
|
||||
EntityKind::Stray => EntityDimensions::new(0.6, 1.99),
|
||||
EntityKind::Strider => EntityDimensions::new(0.9, 1.7),
|
||||
EntityKind::Tadpole => EntityDimensions::new(0.4, 0.3),
|
||||
EntityKind::TextDisplay => EntityDimensions::new(0.0, 0.0),
|
||||
EntityKind::Tnt => EntityDimensions::new(0.98, 0.98),
|
||||
EntityKind::TntMinecart => EntityDimensions::new(0.98, 0.7),
|
||||
EntityKind::TraderLlama => EntityDimensions::new(0.9, 1.87),
|
||||
EntityKind::Trident => EntityDimensions::new(0.5, 0.5),
|
||||
EntityKind::TropicalFish => EntityDimensions::new(0.5, 0.4),
|
||||
EntityKind::Turtle => EntityDimensions::new(1.2, 0.4),
|
||||
EntityKind::Vex => EntityDimensions::new(0.4, 0.8),
|
||||
EntityKind::Villager => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Vindicator => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::WanderingTrader => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Warden => EntityDimensions::new(0.9, 2.9),
|
||||
EntityKind::WindCharge => EntityDimensions::new(0.3125, 0.3125),
|
||||
EntityKind::Witch => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::Wither => EntityDimensions::new(0.9, 3.5),
|
||||
EntityKind::WitherSkeleton => EntityDimensions::new(0.7, 2.4),
|
||||
EntityKind::WitherSkull => EntityDimensions::new(0.3125, 0.3125),
|
||||
EntityKind::Wolf => EntityDimensions::new(0.6, 0.85),
|
||||
EntityKind::Zoglin => EntityDimensions::new(1.39648, 1.4),
|
||||
EntityKind::Zombie => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::ZombieHorse => EntityDimensions::new(1.39648, 1.6),
|
||||
EntityKind::ZombieVillager => EntityDimensions::new(0.6, 1.95),
|
||||
EntityKind::ZombifiedPiglin => EntityDimensions::new(0.6, 1.95),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -449,11 +449,7 @@ impl EntityBundle {
|
|||
kind: azalea_registry::EntityKind,
|
||||
world_name: ResourceLocation,
|
||||
) -> Self {
|
||||
// TODO: get correct entity dimensions by having them codegen'd somewhere
|
||||
let dimensions = EntityDimensions {
|
||||
width: 0.6,
|
||||
height: 1.8,
|
||||
};
|
||||
let dimensions = EntityDimensions::from(kind);
|
||||
let eye_height = dimensions.height * 0.85;
|
||||
|
||||
Self {
|
||||
|
|
|
|||
|
|
@ -12,6 +12,7 @@ burger_data = lib.extract.get_burger_data_for_version(version_id)
|
|||
burger_entities_data = burger_data[0]['entities']
|
||||
|
||||
lib.code.entity.generate_entity_metadata(burger_entities_data, mappings)
|
||||
lib.code.entity.generate_entity_dimensions(burger_entities_data)
|
||||
|
||||
lib.code.utils.fmt()
|
||||
|
||||
|
|
|
|||
|
|
@ -11,6 +11,9 @@ METADATA_RS_DIR = get_dir_location(
|
|||
DATA_RS_DIR = get_dir_location(
|
||||
'../azalea-entity/src/data.rs')
|
||||
|
||||
DIMENSIONS_RS_DIR = get_dir_location(
|
||||
'../azalea-entity/src/dimensions.rs')
|
||||
|
||||
def generate_metadata_names(burger_dataserializers: dict, mappings: Mappings):
|
||||
serializer_names: list[Optional[str]] = [None] * len(burger_dataserializers)
|
||||
for burger_serializer in burger_dataserializers.values():
|
||||
|
|
@ -542,6 +545,38 @@ impl From<EntityDataValue> for UpdateMetadataError {
|
|||
with open(METADATA_RS_DIR, 'w') as f:
|
||||
f.write('\n'.join(code))
|
||||
|
||||
def generate_entity_dimensions(burger_entities_data: dict):
|
||||
# lines look like
|
||||
# EntityKind::Player => EntityDimensions::new(0.6, 1.8),
|
||||
new_match_lines = []
|
||||
for entity_id, entity_data in burger_entities_data['entity'].items():
|
||||
if entity_id.startswith('~'):
|
||||
# not actually an entity
|
||||
continue
|
||||
variant_name: str = upper_first_letter(to_camel_case(entity_id))
|
||||
width = entity_data['width']
|
||||
height = entity_data['height']
|
||||
new_match_lines.append(
|
||||
f' EntityKind::{variant_name} => EntityDimensions::new({width}, {height}),')
|
||||
|
||||
with open(DIMENSIONS_RS_DIR, 'r') as f:
|
||||
lines = f.read().split('\n')
|
||||
new_lines = []
|
||||
|
||||
in_match = False
|
||||
for i, line in enumerate(lines):
|
||||
if not in_match:
|
||||
new_lines.append(line)
|
||||
if line.strip() == 'match entity {':
|
||||
in_match = True
|
||||
else:
|
||||
if line.strip() == '}':
|
||||
new_lines.extend(new_match_lines)
|
||||
new_lines.extend(lines[i:])
|
||||
break
|
||||
|
||||
with open(DIMENSIONS_RS_DIR, 'w') as f:
|
||||
f.write('\n'.join(new_lines))
|
||||
|
||||
def get_entity_parents(entity_id: str, burger_entity_metadata: dict):
|
||||
parents = []
|
||||
|
|
@ -633,3 +668,4 @@ def prettify_mojang_method(mojang_method: str):
|
|||
if re.match(r'is[A-Z]', better_name):
|
||||
better_name = better_name[2:]
|
||||
return to_snake_case(better_name)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue