Notes

from m5stack import *
from m5ui import *
from uiflow import *


setScreenColor(0x222222)


def make_point_from_indices(r, c):
return Point(80 + c * 80, 40 + r * 80)

class Point:

def __init__(self, x_, y_):
self.x = x_
self.y = y_

class Cross:

def __init__(self, r, c):
center = make_point_from_indices(r, c)
self.line0 = M5Line(M5Line.PLINE, center.x - 30,
center.y + 30, center.x + 30,
center.y - 30, 0xFFFFFF)
self.line1 = M5Line(M5Line.PLINE, center.x + 30,
center.y + 30, center.x - 30,
center.y - 30, 0xFFFFFF)

def hide(self):
self.line0.hide()
self.line1.hide()


class Circle:

def __init__(self, r, c):
center = make_point_from_indices(r, c)
self.circle = M5Circle(center.x, center.y,
30, 0x222222, 0xffffff)

def hide(self):
self.circle.hide()


CROSS = 1
CIRCLE = -1
EMPTY = 0

class Field:


def __init__(self):
self.fields = [[EMPTY, EMPTY, EMPTY],
[EMPTY, EMPTY, EMPTY],
[EMPTY, EMPTY, EMPTY]]
self.figures = []

def check_free_cell(self, i, j):
return self.fields[i][j] == EMPTY

def add_figure(self, r, c, player):
self.fields[r][c] = player.figure
if player.figure == CROSS:
self.figures.append(Cross(r, c))
else:
self.figures.append(Circle(r, c))

def check_win(self, player):
return self.check_rows(player.figure) or
self.check_diagonal(player.figure)

def check_draw(self):
return len(self.figures) == 9

def check_rows(self, figure):
for i in range(3):
rows = 0
cols = 0
for j in range(3):
if self.fields[i][j] == figure:
rows += 1
if self.fields[j][i] == figure:
cols += 1

if cols == 3 or rows == 3:
return True

return False

def check_diagonal(self, figure):
left = 0
right = 0
for i in range(3):
if self.fields[i][i] == figure:
left += 1
if self.fields[i][2 - i] == figure:
right += 1

if left == 3 or right == 3:
return True

return False

def draw(self):
self.lines = []
self.lines.append(M5Line(M5Line.PLINE, 40, 160, 280, 160, 0xFFFFFF))
self.lines.append(M5Line(M5Line.PLINE, 40, 80, 280, 80, 0xFFFFFF))
self.lines.append(M5Line(M5Line.PLINE, 120, 0, 120, 240, 0xFFFFFF))
self.lines.append(M5Line(M5Line.PLINE, 200, 0, 200, 240, 0xFFFFFF))

def clean(self):
for i in range(3):
for j in range(3):
self.fields[i][j] = EMPTY

for figure in self.figures:
figure.hide()
figures = []

for line in self.lines:
line.hide()
lines = []



class Player:

def __init__(self):
self.figure = CROSS

def next_figure(self):
if self.figure == CROSS:
self.figure = CIRCLE
else:
self.figure = CROSS

field = Field()
field.draw()



label_row = M5TextBox(0, 25, "Text", lcd.FONT_Default, 0xFFFFFF, rotate=0)
label_column = M5TextBox(0, 45, "Text", lcd.FONT_Default, 0xFFFFFF, rotate=0)

row = 0
column = 0
def increase_column():
global column
column = (column + 1) % 3
wait_ms(50)

def increase_row():
global row
row = (row + 1) % 3
wait_ms(50)

btnA.wasPressed(increase_row)
btnB.wasPressed(increase_column)

player = Player()

while True:

while btnC.isReleased():
label_row.setText("row: {}".format(row))
label_column.setText("column: {}".format(column))
while btnC.isPressed():
pass

if field.check_free_cell(row, column):
field.add_figure(row, column, player)

if field.check_win(player):
label_row.setText(str(player.figure) + " wins!")
break

if field.check_draw():
label_row.setText("Draw")
break

player.next_figure()