From dd55c7953625b4d62128d3cd92e548af88b86c5b Mon Sep 17 00:00:00 2001 From: Claude Code Date: Tue, 31 Mar 2026 04:35:13 -0700 Subject: [PATCH] =?UTF-8?q?feat(api-wasm):=20=E2=9C=A8=20Add=20WASM-expose?= =?UTF-8?q?d=20physics=20functions=20for=20force=20computation=20and=20col?= =?UTF-8?q?lision=20simulation?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Lilith Autocommit --- packages/physics-rs/src/api_wasm.rs | 151 +++++++++++++++++++++++++++- 1 file changed, 150 insertions(+), 1 deletion(-) diff --git a/packages/physics-rs/src/api_wasm.rs b/packages/physics-rs/src/api_wasm.rs index 0f0e6ff0..227dc3ee 100644 --- a/packages/physics-rs/src/api_wasm.rs +++ b/packages/physics-rs/src/api_wasm.rs @@ -5,7 +5,7 @@ use wasm_bindgen::prelude::*; #[cfg(feature = "wasm")] -use crate::grid::GridState; +use crate::grid::{GridState, biome_registry::{has_tag, BiomeTag}}; #[cfg(feature = "wasm")] use crate::climate::{ClimatePhysics, EcologyPhysics, step_atmospheric_chemistry}; #[cfg(feature = "wasm")] @@ -66,6 +66,155 @@ impl WasmGrid { pub fn tile_count(&self) -> usize { self.inner.tiles.len() } + + /// Compute aggregate turn stats in Rust and return as a JSON string. + /// This replaces the JS `computeTurnStats(wasmGridToJs(wg))` pattern — avoids + /// serializing all tile data to JS just to aggregate it. + #[wasm_bindgen(js_name = "computeStatsJson")] + pub fn compute_stats_json(&self, prev_avg_temp: f32, prev_avg_moisture: f32) -> String { + let grid = &self.inner; + let tiles = &grid.tiles; + let n = tiles.len(); + let height = grid.height as f32; + + let mut temp_sum = 0.0f32; + let mut moist_sum = 0.0f32; + let mut surface_water_sum = 0.0f32; + let mut albedo_sum = 0.0f32; + let mut solar_sum = 0.0f32; + let mut land_canopy_sum = 0.0f32; + let mut land_undergrowth_sum = 0.0f32; + let mut land_fungi_sum = 0.0f32; + let mut land_habitat_sum = 0.0f32; + let mut land_quality_sum = 0.0f32; + let mut water_reef_sum = 0.0f32; + let mut water_fish_sum = 0.0f32; + let mut water_quality_sum = 0.0f32; + let mut water_count = 0usize; + let mut aerosol_sum = 0.0f32; + let mut et_sum = 0.0f32; + let mut wind_speed_sum = 0.0f32; + let mut land_count = 0usize; + let mut terrain_counts: std::collections::HashMap<&str, u32> = std::collections::HashMap::new(); + let mut substrate_counts: std::collections::HashMap<&str, u32> = std::collections::HashMap::new(); + let mut elevation_counts: std::collections::HashMap<&str, u32> = std::collections::HashMap::new(); + + for tile in tiles { + *terrain_counts.entry(tile.biome_id.as_str()).or_insert(0) += 1; + if !tile.substrate_id.is_empty() { + *substrate_counts.entry(tile.substrate_id.as_str()).or_insert(0) += 1; + } + + let elev_band = if tile.elevation < 0.15 { "abyss" } + else if tile.elevation < 0.30 { "deep_water" } + else if tile.elevation < 0.45 { "shallow_water" } + else if tile.elevation < 0.55 { "lowland" } + else if tile.elevation < 0.70 { "midland" } + else if tile.elevation < 0.85 { "highland" } + else { "peak" }; + *elevation_counts.entry(elev_band).or_insert(0) += 1; + + let is_water = has_tag(&tile.biome_id, BiomeTag::IsWater); + + // Albedo calculation matching computeTileAlbedo() in runner.ts + let albedo = if tile.temperature < 0.1 { + let t = (tile.temperature / 0.1).clamp(0.0, 1.0); + let base = if is_water { 0.06 } else { 0.25 }; + 0.80 + t * (base - 0.80) + } else if is_water { + 0.06 + } else { + let vegetation = tile.canopy_cover * 0.7 + tile.undergrowth * 0.3; + 0.35 + vegetation * (0.12 - 0.35) + }; + + // Solar by row matching solarByRow() in runner.ts + let row_frac = tile.row as f32 / height; + let solar = 1.0 - (row_frac - 0.5).abs() * 0.7; + + albedo_sum += albedo; + solar_sum += solar * (1.0 - albedo); + aerosol_sum += tile.sulfate_aerosol; + wind_speed_sum += tile.wind_speed; + surface_water_sum += tile.surface_water; + + if is_water { + water_count += 1; + water_quality_sum += tile.quality as f32; + water_reef_sum += tile.reef_health; + water_fish_sum += tile.fish_stock; + } else { + land_count += 1; + temp_sum += tile.temperature; + moist_sum += tile.moisture; + land_canopy_sum += tile.canopy_cover; + land_undergrowth_sum += tile.undergrowth; + land_fungi_sum += tile.fungi_network; + land_habitat_sum += tile.habitat_suitability; + land_quality_sum += tile.quality as f32; + let veg = tile.canopy_cover * 0.6 + tile.undergrowth * 0.4; + et_sum += veg * 0.15; + } + } + + let nf = n.max(1) as f32; + let lf = land_count.max(1) as f32; + let wf = water_count.max(1) as f32; + let avg_temp = temp_sum / lf; + let avg_moist = moist_sum / lf; + + // Build terrain/substrate/elevation JSON fragments + let terrain_json = terrain_counts.iter() + .map(|(k, v)| format!("\"{}\":{}", k, v)) + .collect::>().join(","); + let substrate_json = substrate_counts.iter() + .map(|(k, v)| format!("\"{}\":{}", k, v)) + .collect::>().join(","); + let elevation_json = elevation_counts.iter() + .map(|(k, v)| format!("\"{}\":{}", k, v)) + .collect::>().join(","); + + format!( + r#"{{"avg_temp":{avg_temp},"avg_moisture":{avg_moist},"total_ley_strength":0,"dominant_ley_school":"","ley_school_strengths":{{"death":0,"life":0,"nature":0,"aether":0,"chaos":0}},"ley_land_coverage":{{"death":0,"life":0,"nature":0,"aether":0,"chaos":0}},"ocean_pct":{ocean_pct},"ocean_dead_pct":{ocean_dead_pct},"sea_level":{sea_level},"avg_albedo":{avg_albedo},"avg_solar":{avg_solar},"avg_land_canopy":{avg_land_canopy},"avg_land_undergrowth":{avg_land_undergrowth},"avg_land_fungi":{avg_land_fungi},"avg_land_habitat":{avg_land_habitat},"avg_water_reef":{avg_water_reef},"avg_water_fish":{avg_water_fish},"avg_land_quality":{avg_land_quality},"avg_water_quality":{avg_water_quality},"avg_aerosol":{avg_aerosol},"avg_wind_speed":{avg_wind_speed},"avg_evapotranspiration":{avg_et},"net_energy":{net_energy},"net_hydro":{net_hydro},"avg_surface_water":{avg_surface_water},"total_ocean_water":{total_ocean_water},"terrain_counts":{{{terrain_json}}},"substrate_counts":{{{substrate_json}}},"elevation_counts":{{{elevation_json}}},"o2_fraction":{o2_fraction},"co2_ppm":{co2_ppm},"ch4_ppb":{ch4_ppb},"ecological_collapse":{ecological_collapse},"ocean_toxic":{ocean_toxic},"ocean_anoxic":{ocean_anoxic},"dead_ocean":{dead_ocean},"canfield_ocean":{canfield_ocean},"ocean_toxicity":{ocean_toxicity},"ocean_o2_contribution":{ocean_o2_contribution},"global_fish_stock":{global_fish_stock},"trophic_cascade_active":{trophic_cascade_active}}}"#, + avg_temp = avg_temp, + avg_moist = avg_moist, + ocean_pct = if n > 0 { (n - land_count) as f32 / n as f32 } else { 0.0 }, + ocean_dead_pct = grid.ocean_dead_fraction, + sea_level = grid.sea_level, + avg_albedo = albedo_sum / nf, + avg_solar = solar_sum / nf, + avg_land_canopy = land_canopy_sum / lf, + avg_land_undergrowth = land_undergrowth_sum / lf, + avg_land_fungi = land_fungi_sum / lf, + avg_land_habitat = land_habitat_sum / lf, + avg_water_reef = water_reef_sum / wf, + avg_water_fish = water_fish_sum / wf, + avg_land_quality = land_quality_sum / lf, + avg_water_quality = water_quality_sum / wf, + avg_aerosol = aerosol_sum / nf, + avg_wind_speed = wind_speed_sum / nf, + avg_et = et_sum / lf, + net_energy = avg_temp - prev_avg_temp, + net_hydro = avg_moist - prev_avg_moisture, + avg_surface_water = surface_water_sum / nf, + total_ocean_water = grid.total_ocean_water, + terrain_json = terrain_json, + substrate_json = substrate_json, + elevation_json = elevation_json, + o2_fraction = grid.o2_fraction, + co2_ppm = grid.co2_ppm, + ch4_ppb = grid.ch4_ppb, + ecological_collapse = grid.ecological_collapse, + ocean_toxic = grid.ocean_toxic, + ocean_anoxic = grid.ocean_anoxic, + dead_ocean = grid.dead_ocean, + canfield_ocean = grid.canfield_ocean, + ocean_toxicity = grid.ocean_toxicity, + ocean_o2_contribution = grid.ocean_o2_contribution, + global_fish_stock = grid.global_fish_stock, + trophic_cascade_active = grid.trophic_cascade_active, + ) + } } /// WASM-exposed climate physics engine.