matrix falling code in python using pygame

matrix.py

import pygame, pygame.font
import random


def IsWritten():
    defTemp = True
    for x in xrange((lettersOnScreen[0] / 2) - (len(str) / 2), (lettersOnScreen[0] / 2) + (len(str) / 2) + 1):
        if xHeads[x] == -1:
            defTemp = False
    return defTemp

def getColor(fx,fy):
    defTemp=xHeads[fx]-fy

    if (maxCol>defTemp>0):
        return defTemp
    else:
        return maxCol-1


try:
    fo = open("indata.txt", "r+")
    str = fo.readline()
    # Close opend file
    fo.close()
except:
    str = ''
str = str.upper()  # for better placement


# Pygame init
pygame.init()
temp = pygame.display.Info()
displLength = (temp.current_w, temp.current_h)
surface = pygame.display.set_mode(displLength, pygame.FULLSCREEN)
# Font init
pygame.font.init()
fontObj = pygame.font.Font(pygame.font.get_default_font(), 14)
sampleLetter = fontObj.render('_', False, (0, 111, 0))
letterSize = (sampleLetter.get_width(), sampleLetter.get_height())
lettersOnScreen = (int(float(displLength[0]) / letterSize[0]),
                   int(float(displLength[1]) / letterSize[1]))

# color init
colorList = [(255, 255, 255)]
primeColors = len(colorList)+1
colorList += [(10, 210, 10)] * ((lettersOnScreen[1] - 10))
endColors = len(colorList)
colorList += [(0, 150, 0),(0, 100, 0),(0, 0, 0)]
endColors = len(colorList) - endColors+1

maxCol = len(colorList)


# char generator
letters = [[0 for _ in xrange(lettersOnScreen[1] + 1)] for _ in xrange(lettersOnScreen[0])]
char = chr(random.randint(32, 126))

for y in xrange(lettersOnScreen[1] + 1):
    for x in xrange(lettersOnScreen[0]):
        letters[x][y] = [fontObj.render(char, False, colorList[col]) for col in xrange(maxCol)]
        char = chr(random.randint(32, 126))


# word write
wordMode = False
if len(str) > 0:
    wordMode = True
    for x in xrange((lettersOnScreen[0] / 2) - (len(str) / 2),
                    (lettersOnScreen[0] / 2) + (len(str) / 2)):
        letters[x][lettersOnScreen[1] / 2] = [fontObj.render(str[x - ((lettersOnScreen[0] / 2) - (len(str) / 2))],
                                                             False, (255, 255, 255))
                                              for col in xrange(maxCol)]

    for y in xrange(lettersOnScreen[1] / 2 + 1,
                    lettersOnScreen[1] + 1):
        for x in xrange((lettersOnScreen[0] / 2) - (len(str) / 2),
                        (lettersOnScreen[0] / 2) + (len(str) / 2)):
            letters[x][y] = [fontObj.render(char, False, (0, 0, 0)) for col in xrange(maxCol)]
            char = chr(random.randint(32, 126))

    if len(str) % 2 == 1:

        letters[(lettersOnScreen[0] / 2) + (len(str) / 2)][lettersOnScreen[1] / 2] = \
            [fontObj.render(str[len(str) - 1], False, (255, 255, 255)) for col in xrange(maxCol)]

        for y in xrange(lettersOnScreen[1] / 2 + 1,
                        lettersOnScreen[1] + 1):
            letters[(lettersOnScreen[0] / 2) + (len(str) / 2)][y] = \
                [fontObj.render(char, False, (0, 0, 0)) for col in xrange(maxCol)]
            char = chr(random.randint(32, 126))

if wordMode:
    xHeads = [-1 for _ in xrange(lettersOnScreen[0] + 1)]
else:
    xHeads = [0 for _ in xrange(lettersOnScreen[0] + 1)]


# 1st loop - word write, no char switch
notDone = True
ticksLeft = lettersOnScreen[1] + maxCol
while ticksLeft > 0 and (notDone) and (wordMode):
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            notDone = False
        if event.type == pygame.KEYDOWN:
            notDone = False
    if IsWritten():
        ticksLeft -= 1
    if random.randint(1, 2) == 1:
        randomInt = random.randint(0, lettersOnScreen[0])
        if wordMode:
            if xHeads[randomInt] == -1:
                xHeads[randomInt] = 1
            if random.randint(1, 6):
                randomInt = random.randint((lettersOnScreen[0] / 2) - len(str),
                                           (lettersOnScreen[0] / 2) + len(str) + 1)
                if xHeads[randomInt] == -1:
                    xHeads[randomInt] = 1
        else:
            if xHeads[randomInt] == 0:
                xHeads[randomInt] = 1
    for x in xrange(lettersOnScreen[0]):
        col = 0
        counter = xHeads[x]
        while (counter > 0) and (col < maxCol):
            if (counter < lettersOnScreen[1] + 2) and (col < primeColors or
                                    col > (maxCol - endColors)):
                surface.blit(letters[x][counter - 1][col], (x * letterSize[0],
                                                            (counter - 1) * letterSize[1]))
            col += 1
            counter -= 1
        if xHeads[x] > 0:
            xHeads[x] += 1
        if xHeads[x] - maxCol > lettersOnScreen[1]:
            xHeads[x] = 0

    pygame.display.update()
    clock = pygame.time.Clock()
    clock.tick(20)

# word delete
if len(str) % 2 == 1:
    strLen = (lettersOnScreen[0] / 2) + (len(str) / 2) + 1
else:
    strLen = (lettersOnScreen[0] / 2) + (len(str) / 2)

for x in xrange((lettersOnScreen[0] / 2) - (len(str) / 2),
                strLen):
    letters[x][lettersOnScreen[1] / 2] = \
        [fontObj.render(str[x - ((lettersOnScreen[0] / 2) - (len(str) / 2))], False, colorList[col])
         for col in xrange(maxCol)]

char = chr(random.randint(32, 126))
for y in xrange(lettersOnScreen[1] / 2 + 1,
                lettersOnScreen[1] + 1):
    for x in xrange((lettersOnScreen[0] / 2) - (len(str) / 2),
                    (lettersOnScreen[0] / 2) + (len(str) / 2) + 1):
        letters[x][y] = [fontObj.render(char, False, colorList[col]) for col in xrange(maxCol)]
        char = chr(random.randint(32, 126))


# main matrix, has char switch
while notDone:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            notDone = False
        if event.type == pygame.KEYDOWN:
            notDone = False
    if random.randint(1, 2) == 1:
        randomInt = random.randint(0, lettersOnScreen[0])
        if xHeads[randomInt] <= 0:
            xHeads[randomInt] = 1
    for x in xrange(lettersOnScreen[0]):
        col = 0
        counter = xHeads[x]
        # main loop for redraw
        while (counter > 0) and (col < maxCol):
            if (counter < lettersOnScreen[1] + 2) and (col < primeColors or
                                    col > (maxCol - endColors)):
                surface.blit(letters[x][counter - 1][col], (x * letterSize[0],
                                                            (counter - 1) * letterSize[1]))
            col += 1
            counter -= 1

        # charswirch
        randomInt = random.randint(1, maxCol - 1)
        charPosY = xHeads[x] - randomInt
        if (lettersOnScreen[1] - 1 > charPosY > 0):
            temp = letters[x][charPosY]
            randomX = random.randint(1, lettersOnScreen[0] - 1)
            randomY = random.randint(1,lettersOnScreen[1] - 1)

            surface.blit(letters[x][charPosY][maxCol - 1], (x * letterSize[0],
                                                            charPosY * letterSize[1]))
            surface.blit(letters[randomX][randomY][maxCol - 1], (randomX * letterSize[0],
                                                            randomY * letterSize[1]))
            # char swap
            letters[x][charPosY] = letters[randomX][randomY]
            letters[randomX][randomY] = temp

            surface.blit(letters[x][charPosY][randomInt], (x * letterSize[0], charPosY * letterSize[1]))
            surface.blit(letters[randomX][randomY][getColor(randomX,randomY)],
                         (randomX * letterSize[0], randomY * letterSize[1]))
        # check if is out of screen
        if xHeads[x] > 0:
            xHeads[x] += 1
        if xHeads[x] - maxCol > lettersOnScreen[1]:
            xHeads[x] = 0

    pygame.display.update()
    clock = pygame.time.Clock()
    clock.tick(20)

Awesome code.. Thanks for providing this.

here is a better version https://gist.github.com/MrKioZ/c07b9377d20bab53af6ebcdfbdeabb64