133 lines
4.6 KiB
Python
133 lines
4.6 KiB
Python
"""HeightMap Voronoi Demo
|
|
|
|
Demonstrates: add_voronoi with different coefficients
|
|
Shows cell-based patterns useful for biomes, regions, and organic structures.
|
|
"""
|
|
import mcrfpy
|
|
from mcrfpy import automation
|
|
|
|
GRID_WIDTH, GRID_HEIGHT = 64, 48
|
|
CELL_SIZE = 16
|
|
|
|
def biome_color(h):
|
|
"""Color cells as distinct biomes."""
|
|
# Use value ranges to create distinct regions
|
|
h = max(0.0, min(1.0, h))
|
|
|
|
if h < 0.15:
|
|
return mcrfpy.Color(30, 60, 120) # Deep water
|
|
elif h < 0.25:
|
|
return mcrfpy.Color(50, 100, 180) # Shallow water
|
|
elif h < 0.35:
|
|
return mcrfpy.Color(194, 178, 128) # Beach/desert
|
|
elif h < 0.5:
|
|
return mcrfpy.Color(80, 160, 60) # Grassland
|
|
elif h < 0.65:
|
|
return mcrfpy.Color(40, 100, 40) # Forest
|
|
elif h < 0.8:
|
|
return mcrfpy.Color(100, 80, 60) # Hills
|
|
elif h < 0.9:
|
|
return mcrfpy.Color(130, 130, 130) # Mountains
|
|
else:
|
|
return mcrfpy.Color(240, 240, 250) # Snow
|
|
|
|
def cell_edges_color(h):
|
|
"""Highlight cell boundaries."""
|
|
h = max(0.0, min(1.0, h))
|
|
if h < 0.3:
|
|
return mcrfpy.Color(40, 40, 60)
|
|
elif h < 0.6:
|
|
return mcrfpy.Color(80, 80, 100)
|
|
else:
|
|
return mcrfpy.Color(200, 200, 220)
|
|
|
|
def run_demo(runtime):
|
|
panel_w = GRID_WIDTH // 3
|
|
panel_h = GRID_HEIGHT // 2
|
|
|
|
# Panel 1: Standard Voronoi (cell centers high)
|
|
# coefficients (1, 0) = distance to nearest point
|
|
v_standard = mcrfpy.HeightMap((panel_w, panel_h), fill=0.0)
|
|
v_standard.add_voronoi(num_points=15, coefficients=(1.0, 0.0), seed=42)
|
|
v_standard.normalize(0.0, 1.0)
|
|
|
|
# Panel 2: Inverted (cell centers low, edges high)
|
|
# coefficients (-1, 0) = inverted distance
|
|
v_inverted = mcrfpy.HeightMap((panel_w, panel_h), fill=0.0)
|
|
v_inverted.add_voronoi(num_points=15, coefficients=(-1.0, 0.0), seed=42)
|
|
v_inverted.normalize(0.0, 1.0)
|
|
|
|
# Panel 3: Cell difference (creates ridges)
|
|
# coefficients (1, -1) = distance to nearest - distance to second nearest
|
|
v_ridges = mcrfpy.HeightMap((panel_w, panel_h), fill=0.0)
|
|
v_ridges.add_voronoi(num_points=15, coefficients=(1.0, -1.0), seed=42)
|
|
v_ridges.normalize(0.0, 1.0)
|
|
|
|
# Panel 4: Few large cells (biome-scale)
|
|
v_biomes = mcrfpy.HeightMap((panel_w, panel_h), fill=0.0)
|
|
v_biomes.add_voronoi(num_points=6, coefficients=(1.0, -0.3), seed=99)
|
|
v_biomes.normalize(0.0, 1.0)
|
|
|
|
# Panel 5: Many small cells (texture-scale)
|
|
v_texture = mcrfpy.HeightMap((panel_w, panel_h), fill=0.0)
|
|
v_texture.add_voronoi(num_points=50, coefficients=(1.0, -0.5), seed=77)
|
|
v_texture.normalize(0.0, 1.0)
|
|
|
|
# Panel 6: Voronoi + noise blend (natural regions)
|
|
noise = mcrfpy.NoiseSource(dimensions=2, algorithm='simplex', seed=42)
|
|
v_natural = mcrfpy.HeightMap((panel_w, panel_h), fill=0.0)
|
|
v_natural.add_voronoi(num_points=12, coefficients=(0.8, -0.4), seed=42)
|
|
v_natural.add_noise(noise, world_size=(15, 15), mode='fbm', octaves=3, scale=0.3)
|
|
v_natural.normalize(0.0, 1.0)
|
|
|
|
# Apply to grid
|
|
panels = [
|
|
(v_standard, 0, 0, "Standard (1,0)", biome_color),
|
|
(v_inverted, panel_w, 0, "Inverted (-1,0)", biome_color),
|
|
(v_ridges, panel_w * 2, 0, "Ridges (1,-1)", cell_edges_color),
|
|
(v_biomes, 0, panel_h, "Biomes (6 pts)", biome_color),
|
|
(v_texture, panel_w, panel_h, "Texture (50 pts)", cell_edges_color),
|
|
(v_natural, panel_w * 2, panel_h, "Voronoi + Noise", biome_color),
|
|
]
|
|
|
|
for hmap, ox, oy, name, color_func in panels:
|
|
for y in range(panel_h):
|
|
for x in range(panel_w):
|
|
h = hmap.get((x, y))
|
|
color_layer.set(((ox + x, oy + y)), color_func(h))
|
|
|
|
label = mcrfpy.Caption(text=name, pos=(ox * CELL_SIZE + 5, oy * CELL_SIZE + 5))
|
|
label.fill_color = mcrfpy.Color(255, 255, 255)
|
|
label.outline = 1
|
|
label.outline_color = mcrfpy.Color(0, 0, 0)
|
|
scene.children.append(label)
|
|
|
|
# Grid lines
|
|
for y in range(GRID_HEIGHT):
|
|
color_layer.set(((panel_w - 1, y)), mcrfpy.Color(60, 60, 60))
|
|
color_layer.set(((panel_w * 2 - 1, y)), mcrfpy.Color(60, 60, 60))
|
|
for x in range(GRID_WIDTH):
|
|
color_layer.set(((x, panel_h - 1)), mcrfpy.Color(60, 60, 60))
|
|
|
|
|
|
# Setup
|
|
scene = mcrfpy.Scene("voronoi_demo")
|
|
|
|
grid = mcrfpy.Grid(
|
|
grid_size=(GRID_WIDTH, GRID_HEIGHT),
|
|
pos=(0, 0),
|
|
size=(GRID_WIDTH * CELL_SIZE, GRID_HEIGHT * CELL_SIZE),
|
|
layers={}
|
|
)
|
|
grid.fill_color = mcrfpy.Color(0, 0, 0)
|
|
color_layer = grid.add_layer("color", z_index=-1)
|
|
scene.children.append(grid)
|
|
|
|
scene.activate()
|
|
|
|
# Run the demo
|
|
run_demo(0)
|
|
|
|
# Take screenshot
|
|
automation.screenshot("procgen_06_heightmap_voronoi.png")
|
|
print("Screenshot saved: procgen_06_heightmap_voronoi.png")
|