# Anzahl Quadrate in x-Richtung
breite = 10

start = {
	'left' : [0, -1],
	'right' : [breite-1, 1],
}

def verschiebe(bewegung):
	global feld
	# Verschiebt alle 2028-Steine in die angegebene Bewegungsrichtung ('left' oder 'right'),
	# so dass keine Lücken verbleiben.
	startx, richtung = start[bewegung]
	aenderung = False
	a = startx
	while feld[a] != -1:
		if feld[a] >= 2:
			p = a
			while feld[p+richtung] == 0:
				p += richtung
			if p != a:
				feld[p] = feld[a]
				feld[a] = 0
				aenderung = True
		a -= richtung
	# Beim späteren Verwenden dieser Funktion in der pygame-Version sollte der Rückgabewert 
	# True sein, wenn Fusionen stattgefunden haben, sonst False.
	return aenderung

def fusioniere(bewegung):
	global feld
	# Fusioniert jeweils zwei direkt nebeneinanderliegende Steine derselben Beschriftung.
	# Die Fusion geschieht in der Bewegungsrichtung.
	startx, richtung = start[bewegung]
	aenderung = False
	a = startx
	while feld[a] != -1:
		if feld[a] >= 2:
			if feld[a] == feld[a+richtung]:
				feld[a+richtung] *= 2
				feld[a] = 0
				aenderung = True
		a -= richtung
	# Beim späteren Verwenden dieser Funktion in der pygame-Version sollte der Rückgabewert 
	# True sein, wenn sich Steine bewegt haben, sonst False.
	return aenderung

# Test 1:
print('\n Test 1\n')
feld = [2, 0, 2, 0, 2, 0, 2, 0, 2, 0, -1, -1]
# Bemerkung: Diese Variable speichert eine Zeile aus 10 Feldern im Spiel 2028.
# Die eigentlichen Einträge stehen an den Positionen 0 bis 9.
# Die beiden "Dummy-Einträge" -1 an den Positionen -1 und 10 markieren den linken bzw. rechten Rand.
# Mit ihrer Hilfe lässt sich bisweilen der Zugriff auf nicht existente Positionen der Liste elegant vermeiden.

print(feld)

verschiebe('right')
assert feld == [0, 0, 0, 0, 0, 2, 2, 2, 2, 2, -1, -1]
print(feld)
fusioniere('right')
assert feld == [0, 0, 0, 0, 0, 2, 0, 4, 0, 4, -1, -1]
print(feld)
verschiebe('right')
assert feld == [0, 0, 0, 0, 0, 0, 0, 2, 4, 4, -1, -1]
print(feld)

# Test 2:
print('\n Test 2\n')

feld = [2, 2, 0, 4, 4, 4, 2, 2, 2, 2, -1, -1]
print(feld)

verschiebe('right')
assert feld == [0, 2, 2, 4, 4, 4, 2, 2, 2, 2, -1, -1]
print(feld)
fusioniere('right')
assert feld == [0, 0, 4, 4, 0, 8, 0, 4, 0, 4, -1, -1]
print(feld)
verschiebe('right')
assert feld == [0, 0, 0, 0, 0, 4, 4, 8, 4, 4, -1, -1]
print(feld)

# Test 3:
print('\n Test 3\n')

feld = [4, 0, 0, 4, 8, 8, 0, 8, 0, 8, -1, -1]
print(feld)

verschiebe('right')
assert feld == [0, 0, 0, 0, 4, 4, 8, 8, 8, 8, -1, -1]
print(feld)
fusioniere('right')
assert feld == [0, 0, 0, 0, 0, 8, 0, 16, 0, 16, -1, -1]
print(feld)
verschiebe('right')
assert feld == [0, 0, 0, 0, 0, 0, 0, 8, 16, 16, -1, -1]
print(feld)

# Test 4: (nun mal nach links)
print('\n Test 4\n')

feld = [4, 0, 0, 4, 8, 8, 0, 8, 0, 8, -1, -1]
print(feld)

verschiebe('left')
assert feld == [4, 4, 8, 8, 8, 8, 0, 0, 0, 0, -1, -1]
print(feld)
fusioniere('left')
assert feld == [8, 0, 16, 0, 16, 0, 0, 0, 0, 0, -1, -1]
print(feld)
verschiebe('left')
assert feld == [8, 16, 16, 0, 0, 0, 0, 0, 0, 0, -1, -1]
print(feld)