import numpy as np
from landskap import Landskap
import matplotlib.pyplot as plt
from mpl_toolkits import mplot3d
#import pgzrun
import sys
np.random.seed(1)

# Dette er prekode som gjoer at pygame-zero fungerer. Ikke endre dette:
WIDTH = 800
HEIGHT = WIDTH
gridstorrelse = 100

landskap = Landskap(gridstorrelse, gridstorrelse, WIDTH)
landskap.lag_landskap()


def plot_3d(landskap):
    print("Plott")
    fig = plt.figure()
    ax = plt.axes(projection='3d')
    x= np.arange(0, landskap._bredde)
    y = np.arange(0, landskap._hoyde)
    x, y = np.meshgrid(x, y)
    ax.plot_surface(x, y, landskap._landskap, rstride=1, cstride=1,
                    cmap='viridis', edgecolor='none')
    plt.show()

plot_3d(landskap)

def draw():
    return
    screen.fill((128, 128, 128))
    landskap.tegn(screen)

def update():
    return
    fortsett = landskap.oppdater()

#pgzrun.go()