33 lines
998 B
Python
33 lines
998 B
Python
import os
|
|
import shutil
|
|
|
|
src_base = "assets/DEMO_FAZA1/Vegetation"
|
|
dst_base = "assets/vegetation"
|
|
|
|
mapping = {
|
|
"trava_sop.png": "trava_zelena.png",
|
|
"visoka_trava.png": "trava_suha.png",
|
|
"grass_cluster_dense.png": "trava_divja.png"
|
|
}
|
|
|
|
for src_name, dst_name in mapping.items():
|
|
src = os.path.join(src_base, src_name)
|
|
dst = os.path.join(dst_base, dst_name)
|
|
|
|
if os.path.exists(src):
|
|
shutil.copy(src, dst)
|
|
print(f"Copied {src_name} -> {dst_name}")
|
|
else:
|
|
# Fallback: Generate if missing
|
|
print(f"Missing {src_name}, generating placeholder for {dst_name}")
|
|
from PIL import Image, ImageDraw
|
|
img = Image.new('RGBA', (64, 64), (0,0,0,0))
|
|
draw = ImageDraw.Draw(img)
|
|
color = (0, 255, 0)
|
|
if "suha" in dst_name: color = (200, 200, 100)
|
|
if "divja" in dst_name: color = (0, 100, 0)
|
|
draw.ellipse([10, 10, 54, 54], fill=color + (200,))
|
|
img.save(dst)
|
|
|
|
print("Vegetation variation ready.")
|