89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
"""HeightMap Hills and Craters Demo
|
|
|
|
Demonstrates: add_hill, dig_hill
|
|
Creates volcanic terrain with mountains and craters using ColorLayer visualization.
|
|
"""
|
|
import mcrfpy
|
|
from mcrfpy import automation
|
|
|
|
# Full screen grid: 60x48 tiles at 16x16 = 960x768
|
|
GRID_WIDTH, GRID_HEIGHT = 64, 48
|
|
CELL_SIZE = 16
|
|
|
|
def height_to_color(h):
|
|
"""Convert height value to terrain color."""
|
|
if h < 0.1:
|
|
return mcrfpy.Color(20, 40, int(80 + h * 400))
|
|
elif h < 0.3:
|
|
t = (h - 0.1) / 0.2
|
|
return mcrfpy.Color(int(40 + t * 30), int(60 + t * 40), 30)
|
|
elif h < 0.5:
|
|
t = (h - 0.3) / 0.2
|
|
return mcrfpy.Color(int(70 - t * 20), int(100 + t * 50), int(30 + t * 20))
|
|
elif h < 0.7:
|
|
t = (h - 0.5) / 0.2
|
|
return mcrfpy.Color(int(120 + t * 40), int(100 + t * 30), int(60 + t * 20))
|
|
elif h < 0.85:
|
|
t = (h - 0.7) / 0.15
|
|
return mcrfpy.Color(int(140 + t * 40), int(130 + t * 40), int(120 + t * 40))
|
|
else:
|
|
t = (h - 0.85) / 0.15
|
|
return mcrfpy.Color(int(180 + t * 75), int(180 + t * 75), int(180 + t * 75))
|
|
|
|
# Setup scene
|
|
scene = mcrfpy.Scene("hills_demo")
|
|
|
|
# Create grid with color layer
|
|
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)
|
|
|
|
# Create heightmap
|
|
hmap = mcrfpy.HeightMap((GRID_WIDTH, GRID_HEIGHT), fill=0.3)
|
|
|
|
# Add volcanic mountains - large hills
|
|
hmap.add_hill((15, 24), 18.0, 0.6) # Central volcano base
|
|
hmap.add_hill((15, 24), 10.0, 0.3) # Volcano peak
|
|
hmap.add_hill((45, 15), 12.0, 0.5) # Eastern mountain
|
|
hmap.add_hill((35, 38), 14.0, 0.45) # Southern mountain
|
|
hmap.add_hill((8, 10), 8.0, 0.35) # Small northern hill
|
|
|
|
# Create craters using dig_hill
|
|
hmap.dig_hill((15, 24), 5.0, 0.1) # Volcanic crater
|
|
hmap.dig_hill((45, 15), 4.0, 0.25) # Eastern crater
|
|
hmap.dig_hill((25, 30), 6.0, 0.05) # Impact crater (deep)
|
|
hmap.dig_hill((50, 40), 3.0, 0.2) # Small crater
|
|
|
|
# Add some smaller features for variety
|
|
for i in range(8):
|
|
x = 5 + (i * 7) % 55
|
|
y = 5 + (i * 11) % 40
|
|
hmap.add_hill((x, y), float(3 + (i % 4)), 0.15)
|
|
|
|
# Normalize to use full color range
|
|
hmap.normalize(0.0, 1.0)
|
|
|
|
# Apply heightmap to color layer
|
|
for y in range(GRID_HEIGHT):
|
|
for x in range(GRID_WIDTH):
|
|
h = hmap.get((x, y))
|
|
color_layer.set((x, y), height_to_color(h))
|
|
|
|
# Title
|
|
title = mcrfpy.Caption(text="HeightMap: add_hill + dig_hill (volcanic terrain)", pos=(10, 10))
|
|
title.fill_color = mcrfpy.Color(255, 255, 255)
|
|
title.outline = 1
|
|
title.outline_color = mcrfpy.Color(0, 0, 0)
|
|
scene.children.append(title)
|
|
|
|
scene.activate()
|
|
|
|
# Take screenshot directly (works in headless mode)
|
|
automation.screenshot("procgen_01_heightmap_hills.png")
|
|
print("Screenshot saved: procgen_01_heightmap_hills.png")
|