from celle import Celle
from random import randint
import os

class Spillebrett:
    def __init__(self, rader, kolonner):
        self._rader = rader
        self._kolonner = kolonner
        self._spillebrett = []
        for rad in range(rader):
            self._spillebrett.append([])
            for col in range(kolonner):
                self._spillebrett[rad].append(Celle(rad,col))
        self.generer()

    def printSpillebrett(self):
        os.system("clear")
        for rad in self._spillebrett:
            #celler = ""
            for celle in rad:
                #celler += celle.toString()
                print(celle.toString(), end="")
            print()
            #print(celler)

    def generer(self):
        for rad in self._spillebrett:
            for celle in rad:
                if randint(0, 2) == 1:
                    celle.settLevende()

    def finnNabo(self, rad, col):
        naboer = []
        if (rad > 0):
            if col > 0:
                naboer.append(self._spillebrett[rad-1][col-1])
            naboer.append(self._spillebrett[rad-1][col])
            if col < self._kolonner -1:
                naboer.append(self._spillebrett[rad-1][col+1])
        if col > 0:
            naboer.append(self._spillebrett[rad][col-1])
        if col < self._kolonner -1:
           naboer.append(self._spillebrett[rad][col+1])
        if rad < self._rader -1:
            if col > 0:
                naboer.append(self._spillebrett[rad+1][col-1])
            naboer.append(self._spillebrett[rad+1][col])
            if col < self._kolonner -1:
                naboer.append(self._spillebrett[rad+1][col+1])

        return naboer



for i in range(1000):
    sb = Spillebrett(5, 6)
    #sb.generer()
    sb.printSpillebrett()
    print(sb.finnNabo(1,1))