codegen entity dimensions

4 files changed, 201 insertions, 5 deletions

diff --git a/azalea-entity/src/dimensions.rs b/azalea-entity/src/dimensions.rs
index 3db7e304..12a815d1 100755
--- a/azalea-entity/src/dimensions.rs
+++ b/azalea-entity/src/dimensions.rs
@@ -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),
+        }
+    }
+}
diff --git a/azalea-entity/src/lib.rs b/azalea-entity/src/lib.rs
index 1512bbd3..4c0201ce 100644
--- a/azalea-entity/src/lib.rs
+++ b/azalea-entity/src/lib.rs
@@ -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 {
diff --git a/codegen/genentities.py b/codegen/genentities.py
index bb1c9b24..944a1a4e 100644
--- a/codegen/genentities.py
+++ b/codegen/genentities.py
@@ -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()
 
diff --git a/codegen/lib/code/entity.py b/codegen/lib/code/entity.py
index 1d15bd0b..a1f5525d 100644
--- a/codegen/lib/code/entity.py
+++ b/codegen/lib/code/entity.py
@@ -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)
+