main.py

#Main class for project, this is where the camera feed loop will exist
import cv2
import numpy as np

#main method is where camera loop exists
import algorithm
import detectCard
import identifyCard


def main():

    # set up text prompts
    userPrompt = "To scan your cards, focus your cards"
    dealerPrompt = "To scan dealer's card, focus the dealers card"
    playAgainPrompt = "To play again, press 'w'. To quit, press 'e'"
    continuePrompt = "Press 'w' when ready to scan your next card."

    #setup video feed, index depends on where the camera is plugged in
    cap = cv2.VideoCapture(1)
    cap.set(3, 640) #set width
    cap.set(4, 480) #set height

    while True:
        # call method to prompt the user to scan the players cards
        playerCardNames, playerCardValues = scanCards(cap, userPrompt)
        print(playerCardNames)
        print(playerCardValues)


        # call method to prompt the user to scan the dealers cards
        dealerCardNames, dealerCardValues = scanCards(cap, dealerPrompt)
        print(dealerCardNames)
        print(dealerCardValues)


        # call to algorithm here using "userCardValues" and "dealerCardValues"
        suggestedMove = algorithm.determineMove(playerCardValues, dealerCardValues, False)
        #testing print(suggestedMove)

        # in the event the algorithm tells you to hit, continue to prompt user to scan cards
        # until result is a stand or bust
        while suggestedMove == "Hit":
            displayMove(cap, continuePrompt, suggestedMove)
            playerCardNames, playerCardValues = scanCards(cap, userPrompt)
            print(playerCardNames)
            print(playerCardValues)
            suggestedMove = algorithm.determineMove(playerCardValues, dealerCardValues, False)

        # display move to the user and prompt to play again or not
        playAgain = displayMove(cap, playAgainPrompt, suggestedMove)


        # if user selects play again, run game again. If quit, exit loop and program
        if playAgain:
            pass
        else:
            break


# this method is how we prompt the user to scan the correct cards
def scanCards(cap, prompt1):
    while True:
        #preset color values for detecting white
        h_min = 0
        h_max = 179
        l_min = 200 #233
        l_max = 255
        s_min = 0
        s_max = 255

        #read video feed and create mask
        _, img = cap.read()
        imgHLS = cv2.cvtColor(img, cv2.COLOR_BGR2HLS)

        #prompt user to scan cards
        prompt2 = "and press \'q\' to scan."
        cv2.putText(img, prompt1, (10, 420), cv2.FONT_HERSHEY_COMPLEX, .75, (255, 0, 0), 2)
        cv2.putText(img, prompt2, (10, 450), cv2.FONT_HERSHEY_COMPLEX, .75, (255, 0, 0), 2)

        #display original image, create mask and display mask
        cv2.imshow("Original", img)
        lower = np.array([h_min, l_min, s_min])
        upper = np.array([h_max, l_max, s_max])
        mask = cv2.inRange(imgHLS, lower, upper)
        imgResults = cv2.bitwise_and(img, img, mask=mask)
        cv2.imshow("Results", imgResults)

        # wait for user to press button to scan cards
        if cv2.waitKey(1) & 0xFF == ord('q'):
            cardArray2 = detectCard.getContours(imgResults, img)
            cardNames, cardValues = identifyCard.matchCards(cardArray2)

            return cardNames, cardValues

# this method is where we display the suggested move to the user and prompt the next action,
# either play again continue to scan the next cards in event of a hit
def displayMove(cap, prompt, move):
    move = move + "!"
    while True:
        _, img = cap.read()

        cv2.putText(img, move, (10, 50), cv2.FONT_HERSHEY_COMPLEX, .9, (0, 0, 255), 2)
        cv2.putText(img, prompt, (10, 450), cv2.FONT_HERSHEY_COMPLEX, .75, (0, 0, 0), 2)

        cv2.imshow("Original", img)

        # wait for user input on what to do next
        if cv2.waitKey(1) & 0xFF == ord('w'):
            return True
        elif cv2.waitKey(1) & 0xFF == ord('e'):
            return False


main()