95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
import pydantic
|
|
import pygame
|
|
from noise import snoise2
|
|
from PIL import Image, ImageFilter
|
|
from rich.console import Console
|
|
|
|
from cave_survival.config import config
|
|
|
|
console = Console()
|
|
|
|
|
|
class Point(pydantic.BaseModel):
|
|
x: int
|
|
y: int
|
|
|
|
|
|
class Map:
|
|
def __init__(self, game):
|
|
self.game = game
|
|
self.grass = pygame.image.load("grass.webp").convert_alpha()
|
|
self.rock = pygame.image.load("rock.jpg").convert_alpha()
|
|
self.dirt = pygame.image.load("dirt.jpg").convert_alpha()
|
|
self.resolution = 16
|
|
self.scale = 0.14 # Determines the "smoothness" of the terrain
|
|
self.offset = Point(x=0, y=0)
|
|
self.last_offset = self.offset
|
|
self.screen_width = config.screen_width
|
|
self.screen_height = config.screen_height
|
|
self.octaves = 2 # Number of layers of noise to combine
|
|
self.persistence = 0.05 # Amplitude of each octave
|
|
self.lacunarity = 1.0 # Frequency of each octave
|
|
self.thresh = 128
|
|
self.pre_draw()
|
|
|
|
def refresh_surf(self):
|
|
self.surf = pygame.Surface((self.screen_width, self.screen_height))
|
|
|
|
def get_noise(self, x, y):
|
|
value = snoise2(
|
|
(x + self.offset.x) * self.scale,
|
|
(y + self.offset.y) * self.scale,
|
|
self.octaves,
|
|
self.persistence,
|
|
self.lacunarity,
|
|
)
|
|
value = (value + 1) / 2 * 255
|
|
return value
|
|
|
|
def draw(self):
|
|
self.game.screen.blit(
|
|
pygame.transform.scale(self.surf, (self.screen_width, self.screen_height)),
|
|
(0, 0),
|
|
)
|
|
|
|
def point_check_collision(self, x, y, thresh=None):
|
|
return self.get_noise(x / self.resolution, y / self.resolution) < (
|
|
thresh or self.thresh
|
|
)
|
|
|
|
def pre_draw(self):
|
|
console.log("drawing")
|
|
self.refresh_surf()
|
|
for x in range(int(self.screen_width)):
|
|
for y in range(int(self.screen_height)):
|
|
if not self.point_check_collision(x, y):
|
|
pygame.draw.rect(
|
|
self.surf,
|
|
config.WHITE,
|
|
(
|
|
x,
|
|
y,
|
|
1,
|
|
1,
|
|
),
|
|
)
|
|
pygame.image.save(self.surf, "map.png")
|
|
av1 = (
|
|
Image.open("rock.jpg")
|
|
.convert("RGB")
|
|
.resize((self.screen_width, self.screen_height))
|
|
)
|
|
av2 = (
|
|
Image.open("dirt.jpg")
|
|
.convert("RGB")
|
|
.resize((self.screen_width, self.screen_height))
|
|
)
|
|
mask = (
|
|
Image.open("map.png")
|
|
.convert("L")
|
|
.resize((self.screen_width, self.screen_height))
|
|
.filter(ImageFilter.GaussianBlur(3))
|
|
)
|
|
Image.composite(av2, av1, mask).save("result.png")
|
|
result = pygame.image.load("result.png")
|
|
self.surf.blit(result, (0, 0))
|