picopong.py

#Import libraries
#import picounicorn

import picographics
from picographics import PicoGraphics, DISPLAY_PICO_DISPLAY_2, PEN_RGB332
from pimoroni import Button
import jpegdec
import sys
import utime
#picounicorn.init()

display = picographics.PicoGraphics(display=picographics.DISPLAY_PICO_DISPLAY_2, pen_type=PEN_RGB332)
display.clear()
display.update()

# From CPython Lib/colorsys.py
def hsv_to_rgb(h, s, v):
    if s == 0.0:
        return v, v, v
    i = int(h * 6.0)
    f = (h * 6.0) - i
    p = v * (1.0 - s)
    q = v * (1.0 - s * f)
    t = v * (1.0 - s * (1.0 - f))
    i = i % 6
    if i == 0:
        return v, t, p
    if i == 1:
        return q, v, p
    if i == 2:
        return p, v, t
    if i == 3:
        return p, q, v
    if i == 4:
        return t, p, v
    if i == 5:
        return v, p, q
        
#Define setup variables
#w = picounicorn.get_width()#16
#h = picounicorn.get_height()#7
w, h = display.get_bounds()

startballx=7
startbally=3
startdirH=1
startdirV=0
startscoreXY=0
startscoreAB=0
onlistAB=[2,3,4]
onlistXY=[2,3,4]

#Create lists of pixels for the height/width of the whole display/ball area
listW=[]#makes a list 0-15
for i in range(w):
    listW.append(i)

listH=[]#makes a list 0-6
for i in range(h):
    listH.append(i)
    
listWball=[]#makes a list 1-14
for i in range(1,15):
    listWball.append(i)

listHball=[]#makes a list 0-6
for i in range(0,7):
    listHball.append(i)
    

# Define colours
playerXYcolours=(0,114,178)
playerABcolours=(255,70,160)
ballcolours=playerABcolours

#Set constant variables for pong title, scoring and win message

pongtitle=[["                "],
           ["RRR RRR RRR RRRX"],
           ["R R R R R R R RX"],
           ["RRR R R R R RRRX"],
           ["R   R R R R   R "],
           ["R   RRR R R RRRX"],
           ["                "]] 


BLANKSECTION= ["   " for i in range(h)] 

NUMZERO=["   ",
         "XXX",
         "X X",
         "X X",
         "X X",
         "XXX",
         "   "]
         
NUMONE=["   ",
        " X ",
        "XX ",
        " X ",
        " X ",
        "XXX",
        "   "]
        
NUMTWO=["   ",
        "XXX",
        "  X",
        "XXX",
        "X  ",
        "XXX",
        "   "]
        
NUMTHREE=["   ",
          "XXX",
          "  X",
          "XXX",
          "  X",
          "XXX",
          "   "]

NUMFOUR=["   ",
         "X X",
         "X X",
         "XXX",
         "  X",
         "  X",
         "   "]

NUMFIVE=["   ",
          "XXX",
          "X  ",
          "XXX",
          "  X",
          "XXX",
          "   "]

DASH=["    ",
      "    ",
      "    ",
      " XX ",
      "    ",
      "    ",
      "    "]
         
LETTERW=["     ",
         "X   X",
         "X   X",
         "X   X",
         "X X X",
         "XXXXX",
         "     "]
 
LETTERI=["   ",
         "XXX",
         " X ",
         " X ",
         " X ",
         "XXX",
         "   "]
 
LETTERN=["    ",
         "X  X",
         "XX X",
         "X XX",
         "X  X",
         "X  X",
         "    "]

PUNCEXCALM=["  ",
            "X ",
            "X ",
            "X ",
            "  ",
            "X ",
            "  "]

#Create dictionaries to store scoring numbers/text
scoredict={0:NUMZERO,1:NUMONE,2:NUMTWO,3:NUMTHREE,4:NUMFOUR,5:NUMFIVE,"dash":DASH}
textdict={"W":LETTERW,"I":LETTERI,"N":LETTERN,"!":PUNCEXCALM}


#Function to clear display by setting pixels to black
def cleardisplay():
    for x in range(w):
        for y in range(h):
            picounicorn.set_pixel(x, y, 0, 0, 0)            

#Function to update display with letters/numbers
def updatedisplay(displaymap):
    x=0
    y=0
    for row in displaymap: 
        for line in row:
            for char in line:
                if x < w and y <h:
                    if char == "X":
                        r, g, b = 255,255,255
                    elif char == "R":  
                        r, g, b = [int(c * 255) for c in hsv_to_rgb(x / w, y / h, 1.0)]
                    elif char == "D":
                        r, g, b = 200,200,40
                    elif char == "A":
                        r, g, b = playerABcolours
                    elif char == "Y":
                        r, g, b = playerXYcolours
                    else:
                        r,g,b=0,0,0
                    picounicorn.set_pixel(x, y, r, g, b)    
                x+=1
            x=0
        y+=1
    return displaymap

#Function to generate an 2D array to represent the current score in the format: PlayerABscore - PlayerXYscore
def generatescore(scoreAB,scoreXY):
    scoreABpix = [item.replace("X","A") for item in scoredict[scoreAB]]
    scoreXYpix = [item.replace("X","Y") for item in scoredict[scoreXY]]
    dashpix = [item.replace("X","D") for item in scoredict["dash"]]
    fulldisplay = [["{}{}{}{}{}".format(BLANKSECTION[i],scoreABpix[i],dashpix[i],scoreXYpix[i],BLANKSECTION[i])] for i in range(h)]
    return fulldisplay
    
#Function to generate a 2D array of specified text e.g. "WIN!"
def generatemessage(winningcolour):
    text1=[item.replace("X",winningcolour) for item in textdict["W"]]
    text2=[item.replace("X",winningcolour) for item in textdict["I"]]
    text3=[item.replace("X",winningcolour) for item in textdict["N"]]
    text4=[item.replace("X",winningcolour) for item in textdict["!"]]
    fulldisplay = [["{} {} {} {}   ".format(text1[i],text2[i],text3[i],text4[i])] for i in range(h)]
    return fulldisplay

#Function to make the "display map" seem as though it is "scrolling" (loop)
def scrolldisplay(displaymap):
    scrollmap=[]
    rowcount=0
    for row in displaymap:
        scrollmap.append([])
        for line in row:
            scrollmap[rowcount]= [line[1:]+line[0]]
        rowcount+=1
    return scrollmap

#Display the title screen for 2 seconds
updatedisplay(pongtitle)
utime.sleep(2)

#Function to create the ball and ball trail(using previous ball co-ords)
def ball(currentballx,currentbally,prevballx,prevbally,prevball2x,prevball2y,ballcolours):
    for x in listWball:
        for y in listHball:
            if x == currentballx:
                if y == currentbally:
                    r,g,b=ballcolours
                    picounicorn.set_pixel(x, y, r, g, b)
            elif x == prevballx:
                if y == prevbally:
                    r,g,b=[round(element * 0.3) for element in ballcolours]
                    picounicorn.set_pixel(x, y, r, g, b)
            elif x == prevball2x:
                if y == prevball2y:
                    r,g,b=[round(element * 0.2) for element in ballcolours]
                    picounicorn.set_pixel(x, y, r, g, b)          
            else:
                r,g,b=0,0,0
                picounicorn.set_pixel(x, y, r, g, b)
    return currentballx,currentbally,prevballx,prevbally,prevball2x,prevball2y

#Initial invocation of "ball" function
ball(startballx,startbally,startballx,startbally,startballx,startbally,ballcolours)

ballx=startballx
bally=startbally
dirH=startdirH
dirV=startdirV
scoreAB=startscoreAB
scoreXY=startscoreXY

#Function to create playerAB paddle (in the first column)
def lightcontrolAB(onlistAB):
    for x in range(1):
        for y in listH:
            if y in onlistAB:
                r,g,b=playerABcolours
            else:
                r,g,b=0,0,0
            picounicorn.set_pixel(x, y, r, g, b)
                
#Function to create playerXY paddle (in the last column)
def lightcontrolXY(onlistXY):
    for x in range(15,16):
        for y in listH:
            if y in onlistXY:
                r,g,b=playerXYcolours
            else:
                r,g,b=0,0,0
            picounicorn.set_pixel(x, y, r, g, b)
  
#Function to determine the ball colour & direction (based on it's position)and update scoring
def ballposition(currentballx,currentbally,currentdirH,currentdirV,scoreAB,scoreXY,ballcolours,startdirH):
    if currentballx == listWball[0]:#if ball is in the furthest left column
        ballcolours=playerABcolours
        if currentbally in onlistAB:
            currentdirH=1
            if currentbally == onlistAB[0]:
                currentdirV=-1
            if currentbally == onlistAB[2]:
                currentdirV=1
        else:
            scoreXY+=1
            startdirH=1
            currentballx=startballx
            currentbally=startbally
            currentdirH=startdirH
            currentdirV=startdirV
            cleardisplay()
            updatedisplay(generatescore(scoreAB,scoreXY))
            utime.sleep(1)
    elif currentballx == listWball[-1]:#if ball is in the furthest right column
        ballcolours=playerXYcolours
        if currentbally in onlistXY:
            currentdirH= -1
            if currentbally == onlistXY[0]:
                currentdirV=-1
            if currentbally == onlistXY[2]:
                currentdirV=1 
        else:
            scoreAB+=1
            startdirH=- 1
            currentballx=startballx
            currentbally=startbally
            currentdirH=startdirH
            currentdirV=startdirV
            cleardisplay()
            updatedisplay(generatescore(scoreAB,scoreXY))
            utime.sleep(1)
    elif currentballx in listWball:
        if currentbally == listHball[0]:
            currentdirV=1
        elif currentbally == listHball[-1]:
            currentdirV=-1
    prevball2x=currentballx-currentdirH
    prevball2y=currentbally-currentdirV
    prevballx=currentballx
    prevbally=currentbally
    currentballx=currentballx+currentdirH
    currentbally=currentbally+currentdirV
    ball(currentballx,currentbally,prevballx,prevbally,prevball2x,prevball2y,ballcolours)
    return currentballx,currentbally,currentdirH,currentdirV,scoreAB,scoreXY,ballcolours,startdirH

#Function which runs the game       
while True:
    ballx,bally,dirH,dirV,scoreAB,scoreXY,ballcolours,startdirH = ballposition(ballx,bally,dirH,dirV,scoreAB,scoreXY,ballcolours,startdirH)
    if scoreAB >=5 or scoreXY >=5:
        if scoreAB>=5:
            winningcolour="A"
        else:
            winningcolour="Y"
        cleardisplay()
        currentdisplaymap=updatedisplay(generatemessage(winningcolour))
        # for i in range(w*3):
        while True:
            currentdisplaymap=updatedisplay(scrolldisplay(currentdisplaymap))
            utime.sleep(0.1)
            if picounicorn.is_pressed(picounicorn.BUTTON_A) or picounicorn.is_pressed(picounicorn.BUTTON_B) or picounicorn.is_pressed(picounicorn.BUTTON_X) or picounicorn.is_pressed(picounicorn.BUTTON_Y):
                ballx=startballx
                bally=startbally
                dirH=startdirH
                dirV=startdirV
                scoreAB=startscoreAB
                scoreXY=startscoreXY
                updatedisplay(pongtitle)
                utime.sleep(2)
                ball(startballx,startbally,startballx,startbally,startballx,startbally,ballcolours)
                break
    else:
        if picounicorn.is_pressed(picounicorn.BUTTON_A):
            if onlistAB[0]!=0:
                onlistAB.insert(0,onlistAB[0]-1)
                onlistAB=onlistAB[:-1]
        elif picounicorn.is_pressed(picounicorn.BUTTON_B):
            if onlistAB[-1]!=6:
                onlistAB.insert(len(onlistAB),onlistAB[-1]+1)
                onlistAB=onlistAB[1:]
        if picounicorn.is_pressed(picounicorn.BUTTON_X):
            if onlistXY[0]!=0:
                onlistXY.insert(0,onlistXY[0]-1)
                onlistXY=onlistXY[:-1]
        elif picounicorn.is_pressed(picounicorn.BUTTON_Y):
            if onlistXY[-1]!=6:
                onlistXY.insert(len(onlistXY),onlistXY[-1]+1)
                onlistXY=onlistXY[1:]
    lightcontrolAB(onlistAB)
    lightcontrolXY(onlistXY)
    utime.sleep(0.1)