156 lines
8.1 KiB
Python
156 lines
8.1 KiB
Python
import math
|
|
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
class drawConnectFour():
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
# Draws the whole thing
|
|
def drawImage(self, channel):
|
|
self.bot.log("Drawing connect four board", level = 10)
|
|
game = self.bot.database["connect 4 games"].find_one({"_id":channel})
|
|
|
|
board = game["board"]
|
|
|
|
border = 40
|
|
gridBorder = 40
|
|
cornerSize = 300
|
|
boardOutlineSize = 10
|
|
pieceOutlineSize = 10
|
|
emptyOutlineSize = 0
|
|
bottomBorder = 110
|
|
exampleCircles = 100
|
|
w, h = 2800,2400
|
|
backgroundColor = (230,230,234,255)
|
|
boardOutlineColor = (0,0,0)
|
|
pieceOutlineColor = (244,244,248)
|
|
emptyOutlineColor = (0,0,0)
|
|
player1Color = (254,74,73)
|
|
player2Color = (254,215,102)
|
|
boardColor = (42,183,202)
|
|
placeSize = 300
|
|
white = (255,255,255,160)
|
|
winBarColor = (250,250,250,255)
|
|
|
|
fnt = ImageFont.truetype('resources/fonts/futura-bold.ttf', exampleCircles)
|
|
|
|
boardSize = [w-(2*(border+gridBorder)),h-(2*(border+gridBorder))]
|
|
placeGridSize = [math.floor(boardSize[0]/7),math.floor(boardSize[1]/6)]
|
|
pieceStartx = (border+gridBorder)+math.floor(placeGridSize[0]/2)-math.floor(placeSize/2)
|
|
pieceStarty = (border+gridBorder)+math.floor(placeGridSize[1]/2)-math.floor(placeSize/2)
|
|
|
|
if game["players"][0] == "Gwendolyn":
|
|
player1 = "Gwendolyn"
|
|
else:
|
|
player1 = self.bot.databaseFuncs.getName(game["players"][0])
|
|
|
|
if game["players"][1] == "Gwendolyn":
|
|
player2 = "Gwendolyn"
|
|
else:
|
|
player2 = self.bot.databaseFuncs.getName(game["players"][1])
|
|
|
|
|
|
background = Image.new("RGB", (w,h+bottomBorder),backgroundColor)
|
|
d = ImageDraw.Draw(background,"RGBA")
|
|
|
|
# This whole part was the easiest way to make a rectangle with rounded corners and an outline
|
|
# - Corners:
|
|
d.ellipse([(border,border),(border+cornerSize,border+cornerSize)],fill=boardColor,outline=boardOutlineColor,width=boardOutlineSize)
|
|
d.ellipse([(w-(border+cornerSize),h-(border+cornerSize)),(w-border,h-border)],fill=boardColor,outline=boardOutlineColor,width=boardOutlineSize)
|
|
d.ellipse([(border,h-(border+cornerSize)),(border+cornerSize,h-border)],fill=boardColor,outline=boardOutlineColor,width=boardOutlineSize)
|
|
d.ellipse([(w-(border+cornerSize),border),(w-border,border+cornerSize)],fill=boardColor,outline=boardOutlineColor,width=boardOutlineSize)
|
|
# - Rectangle:
|
|
d.rectangle([(border+math.floor(cornerSize/2),border),(w-(border+math.floor(cornerSize/2)),h-border)],fill=boardColor,outline=boardOutlineColor,width=boardOutlineSize)
|
|
d.rectangle([(border,border+math.floor(cornerSize/2)),(w-border,h-(border+math.floor(cornerSize/2)))],fill=boardColor,outline=boardOutlineColor,width=boardOutlineSize)
|
|
# - Removing outline on the inside:
|
|
d.rectangle([(border+math.floor(cornerSize/2),border+math.floor(cornerSize/2)),(w-(border+math.floor(cornerSize/2)),h-(border+math.floor(cornerSize/2)))],fill=boardColor)
|
|
d.rectangle([(border+math.floor(cornerSize/2),border+boardOutlineSize),(w-(border+math.floor(cornerSize/2)),h-(border+boardOutlineSize))],fill=boardColor)
|
|
d.rectangle([(border+boardOutlineSize,border+math.floor(cornerSize/2)),(w-(border+boardOutlineSize),h-(border+math.floor(cornerSize/2)))],fill=boardColor)
|
|
|
|
for x, line in enumerate(board):
|
|
for y, piece in enumerate(line):
|
|
|
|
if piece == 1:
|
|
pieceColor = player1Color
|
|
outlineWidth = pieceOutlineSize
|
|
outlineColor = pieceOutlineColor
|
|
elif piece == 2:
|
|
pieceColor = player2Color
|
|
outlineWidth = pieceOutlineSize
|
|
outlineColor = pieceOutlineColor
|
|
else:
|
|
pieceColor = backgroundColor
|
|
outlineWidth = emptyOutlineSize
|
|
outlineColor = emptyOutlineColor
|
|
|
|
startx = pieceStartx + placeGridSize[0]*y
|
|
starty = pieceStarty + placeGridSize[1]*x
|
|
|
|
d.ellipse([(startx,starty),(startx+placeSize,starty+placeSize)],fill=pieceColor,outline=outlineColor,width=outlineWidth)
|
|
|
|
if game["winner"] != 0:
|
|
coordinates = game["win coordinates"]
|
|
startx = border + placeGridSize[0]*coordinates[1] + gridBorder
|
|
starty = border + placeGridSize[1]*coordinates[0] + gridBorder
|
|
a = (placeGridSize[0]*4-gridBorder-border)**2
|
|
b = (placeGridSize[1]*4-gridBorder-border)**2
|
|
diagonalLength = (math.sqrt(a+b))/placeGridSize[0]
|
|
diagonalAngle = math.degrees(math.atan(placeGridSize[1]/placeGridSize[0]))
|
|
|
|
if game["win direction"] == "h":
|
|
winBar = Image.new("RGBA",(placeGridSize[0]*4,placeGridSize[1]),(0,0,0,0))
|
|
winD = ImageDraw.Draw(winBar)
|
|
winD.ellipse([(0,0),(placeGridSize[0],placeGridSize[1])],fill=white)
|
|
winD.ellipse([((placeGridSize[0]*3),0),(placeGridSize[0]*4,placeGridSize[1])],fill=white)
|
|
winD.rectangle([(int(placeGridSize[0]*0.5),0),(int(placeGridSize[0]*3.5),placeGridSize[1])],fill=white)
|
|
|
|
elif game["win direction"] == "v":
|
|
winBar = Image.new("RGBA",(placeGridSize[0],placeGridSize[1]*4),(0,0,0,0))
|
|
winD = ImageDraw.Draw(winBar)
|
|
winD.ellipse([(0,0),(placeGridSize[0],placeGridSize[1])],fill=white)
|
|
winD.ellipse([(0,(placeGridSize[1]*3)),(placeGridSize[0],placeGridSize[1]*4)],fill=white)
|
|
winD.rectangle([0,(int(placeGridSize[1]*0.5)),(placeGridSize[0],int(placeGridSize[1]*3.5))],fill=white)
|
|
|
|
elif game["win direction"] == "r":
|
|
winBar = Image.new("RGBA",(int(placeGridSize[0]*diagonalLength),placeGridSize[1]),(0,0,0,0))
|
|
winD = ImageDraw.Draw(winBar)
|
|
winD.ellipse([(0,0),(placeGridSize[0],placeGridSize[1])],fill=white)
|
|
winD.ellipse([((placeGridSize[0]*(diagonalLength-1)),0),(placeGridSize[0]*diagonalLength,placeGridSize[1])],fill=white)
|
|
winD.rectangle([(int(placeGridSize[0]*0.5),0),(int(placeGridSize[0]*(diagonalLength-0.5)),placeGridSize[1])],fill=white)
|
|
winBar = winBar.rotate(-diagonalAngle,expand=1)
|
|
startx -= 90
|
|
starty -= 100
|
|
|
|
elif game["win direction"] == "l":
|
|
winBar = Image.new("RGBA",(int(placeGridSize[0]*diagonalLength),placeGridSize[1]),(0,0,0,0))
|
|
winD = ImageDraw.Draw(winBar)
|
|
winD.ellipse([(0,0),(placeGridSize[0],placeGridSize[1])],fill=white)
|
|
winD.ellipse([((placeGridSize[0]*(diagonalLength-1)),0),(placeGridSize[0]*diagonalLength,placeGridSize[1])],fill=white)
|
|
winD.rectangle([(int(placeGridSize[0]*0.5),0),(int(placeGridSize[0]*(diagonalLength-0.5)),placeGridSize[1])],fill=white)
|
|
winBar = winBar.rotate(diagonalAngle,expand=1)
|
|
startx -= placeGridSize[0]*3 + 90
|
|
starty -= gridBorder + 60
|
|
|
|
|
|
mask = winBar.copy()#.convert("L")
|
|
#mask.putalpha(128)
|
|
#mask.save("test.png")
|
|
|
|
winBarImage = Image.new("RGBA",mask.size,color=winBarColor)
|
|
background.paste(winBarImage,(startx,starty),mask)
|
|
|
|
# Bottom
|
|
textPadding = 20
|
|
|
|
exampleHeight = h - border + int((bottomBorder+border)/2) - int(exampleCircles/2)
|
|
d.ellipse([(border,exampleHeight),(border+exampleCircles),(exampleHeight+exampleCircles)],fill=player1Color,outline=boardOutlineColor,width=3)
|
|
d.text((border+exampleCircles+textPadding,exampleHeight),player1,font=fnt,fill=(0,0,0))
|
|
|
|
textWidth = fnt.getsize(player2)[0]
|
|
d.ellipse([(w-border-exampleCircles-textWidth-textPadding,exampleHeight),(w-border-textWidth-textPadding),(exampleHeight+exampleCircles)],fill=player2Color,outline=boardOutlineColor,width=3)
|
|
d.text((w-border-textWidth,exampleHeight),player2,font=fnt,fill=(0,0,0))
|
|
|
|
|
|
background.save("resources/games/connect4Boards/board"+channel+".png")
|