import numpy as np from pygame import Rect import matplotlib.pyplot as plt import math class Landskap: def __init__(self, bredde, hoyde, skjermstorrelse=None, antall_peaks = 12, random_seed=1, maks_score_sjekk=1000): np.random.seed(random_seed) self._skjermstorrelse = skjermstorrelse # brukes hvis man skal tegne med pygame zero self._bredde = bredde self._hoyde = hoyde self._landskap = np.zeros((self._hoyde, self._bredde)) self._individer = [] self._max_hoyde = None self._antall_peaks = antall_peaks self._antall_score_sjekk = 0 self._maks_score_sjekk = maks_score_sjekk def hent_score(self, x, y): if self._antall_score_sjekk >= self._maks_score_sjekk: print("Ikke lov ? sjekke scoren mer") return 0 else: self._antall_score_sjekk += 1 return self._landskap[x, y] def _lag_peak(self, x, y, hoyde): # sett peaken til ? v?re h?yden, fyll inn "skr?ning" rundt self._landskap[y, x] = hoyde peak_radius = self._bredde // 5 # fyll inn ruter med avstand radius fra peaken for i in range(x-peak_radius, x+peak_radius): for j in range(y-peak_radius, y+peak_radius): avstand_fra_peak = math.sqrt((x-i)**2 + (y-j)**2) if avstand_fra_peak <= peak_radius: if i >= 0 and i < self._bredde and j >= 0 and j < self._hoyde: self._landskap[j, i] = hoyde - (avstand_fra_peak * hoyde/peak_radius) def _legg_til_stoy(self): self._landskap += np.random.randint(0, 10, size=(self._hoyde, self._bredde)) / 10 def plott_3d(self): fig = plt.figure() ax = plt.axes(projection='3d') x = np.arange(0, self._bredde) y = np.arange(0, self._hoyde) x, y = np.meshgrid(x, y) ax.plot_surface(x, y, self._landskap, rstride=1, cstride=1, cmap='viridis', edgecolor='none') plt.show() def lag_landskap(self): for peak in range(self._antall_peaks): pos_x = np.random.randint(0, self._bredde) pos_y = np.random.randint(0, self._hoyde) peak_hoyde = np.random.normal(10, 10) + np.random.randint(0, 10) self._lag_peak(pos_x, pos_y, peak_hoyde) self._legg_til_stoy() self._max_hoyde = np.max(self._landskap) # fungerer med pygame zero def tegn(self, skjerm): rutestorrelse_x = self._skjermstorrelse / self._bredde rutestorrelse_y = self._skjermstorrelse / self._hoyde for x in range(self._bredde): for y in range(self._hoyde): farge = 128 * self._landskap[y, x] / self._max_hoyde skjerm.draw.filled_rect(Rect((x*rutestorrelse_x, y*rutestorrelse_y), ((x+1)*rutestorrelse_x, (y+1)*rutestorrelse_y)), (farge, farge, farge)) def oppdater(self): pass if __name__ == "__main__": landskap = Landskap(70, 70, maks_score_sjekk=200) landskap.lag_landskap() landskap.plott_3d() landskap.sjekk_score(5, 20)