snake_multiturn_inScreen.py

#####################################################################
#   Snakes in this file can take multiple turns at the same time    #
#           Snakes will not move outside of the screen              #
#####################################################################
# vis 142 fall 2022 project 2
# This can take 20 minutes to hours to run.

# imports, don't change these but you can add imports you need
import pygame
from pygame.locals import *
from sys import exit
import random
import time
# record the start time
start_time = time.time()

#####################################################################
# We will be producing 4K video from an image sequence
# Important: you might want to work at lower resolution that fits, 
# your screens! Such as 1920 x 1080.
# And then change these back to 4096 × 2160 for production.
# On the other hand, you will have to deal with scaling issues if you do.
#####################################################################
width = 1000
height = 500

# width = 4096 
# height = 2160 

#####################################################################
# Name and title, update to your name and title
#####################################################################
name = "Meihui Liu"
title = "SSSSnakeS" 

#####################################################################
# IMPORTANT - you will get your start sequence number from your TA
# If you use the default number 1000000 below, your work will not 
# be part of the class reel, as in every student must have different
# start sequence numbers.
#####################################################################
start_sequence_num = 2042000 # CHANGE HERE

# Do not change these variables
# normal pygame stuff
clock = pygame.time.Clock()
pygame.init()
screen = pygame.display.set_mode((width, height))
pygame.display.set_caption('generate 4K animation pngs')
frame_num = start_sequence_num
titles_font = pygame.font.SysFont(None, int(width/12)) # if name or title run off screen, try setting the literal to 16 instead of 12
name_f = titles_font.render(name, True, (255,255,255))
title_f = titles_font.render(title, True, (255,255,255))

# print resolution warning
if (width != 4096 and height != 2160):
    print("Warning: dimensions not 4K, be sure width and height are set to 4096 and 2160.")

# this function makes one second of black frames
def make_black():
    global frame_num
    screen.fill((0,0,0))
    pygame.display.update()
    for i in range(0, 60):
       pygame.image.save(screen, "./frames/" + str(frame_num) + ".png")
       frame_num = frame_num + 1
       clock.tick(60)
        
#############################################################
# Object Sanke
# Defines all the proporties of a snake object and makes it movable.
class Snake:
    def __init__(self):
        self.color = (random.randint(1, 255), random.randint(1, 255), random.randint(1, 255))
        self.length = random.randint(2, width - int(width/3))
        self.size = random.randint(3, 6)
        self.heading = random.randint(0, 3)
        self.prev_heading = self.heading
        self.turning_points = []
        startPosX = random.randint(50, width-50) 
        startPosY = random.randint(50, height-50)
        if (self.heading == 0): # heading right
            endPosX = startPosX - self.length
            endPosY = startPosY
        if (self.heading == 1): # heading left
            endPosX = startPosX + self.length
            endPosY = startPosY
        if (self.heading == 2): # heading up
            endPosX = startPosX
            endPosY = startPosY + self.length
        if (self.heading == 3): # heading down
            endPosX = startPosX
            endPosY = startPosY - self.length
        self.turning_points.append((startPosX, startPosY))
        self.turning_points.append((endPosX, endPosY))
    
    def choose_turn_heading(self):
        self.prev_heading = self.heading
        up_down = [2, 3]
        left_right = [1, 0]
        startPosX = self.turning_points[0][0]
        startPosY = self.turning_points[0][1]
        if(self.prev_heading == 0 or self.prev_heading == 1):
            if(startPosY == 0): # at the top, can't go up
                self.heading = 3
            elif (startPosY == (height)): # at the bottom, can't got down
                self.heading = 2
            else:
                self.heading = random.choice(up_down)
        else:
            if(startPosX == 0): # at the left end, can't go left
                self.heading = 0
            elif (startPosX == (width)): # at the right end, can't go right
                self.heading = 1
            else: 
                self.heading = random.choice(left_right)
    
    def set_turning_point(self):
        turningPosX = self.turning_points[0][0]
        turningPosY = self.turning_points[0][1]
        self.turning_points.insert(1, (turningPosX, turningPosY))

    def turn(self):
        startPosX = self.turning_points[0][0]
        startPosY = self.turning_points[0][1]
        endPosX = self.turning_points[len(self.turning_points) - 1][0]
        endPosY = self.turning_points[len(self.turning_points) - 1][1]
        beforeEndX = self.turning_points[len(self.turning_points) - 2][0]
        beforeEndY = self.turning_points[len(self.turning_points) - 2][1]
            
        # checking if the first turn has ended
        if(endPosX == beforeEndX and endPosY == beforeEndY):
            self.turning_points.pop(len(self.turning_points) - 2) # remove the turning point cz the turn has ended
            beforeEndX = self.turning_points[len(self.turning_points) - 2][0] #get the new turning location
            beforeEndY = self.turning_points[len(self.turning_points) - 2][1]

        #move the snake
        if (self.heading == 0): # start is heading right
            startPosX += 1
        if (self.heading == 1): # start is heading left
            startPosX -= 1
        if (self.heading == 2): # start is heading up
            startPosY -= 1
        if (self.heading == 3): # start is heading down
            startPosY += 1
        if(endPosX == beforeEndX):
            if(endPosY > beforeEndY): #heading upward
                endPosY -= 1
            else: #heading downward
                endPosY += 1
        if(endPosY == beforeEndY):
            if(endPosX > beforeEndX): #heading left
                endPosX -= 1
            else: #heading right
                endPosX += 1
        #update start and end locations as snake has moved
        self.turning_points[0] = (startPosX, startPosY)
        self.turning_points[len(self.turning_points) - 1] = (endPosX, endPosY)
        #checking if we are done turning
        # print("endPosX " + str(self.endPosX) + " endPosY " + str(self.endPosY))
        # print("turningPosX " + str(self.turningPosX) + " turningPosY " + str(self.turningPosY))
        # print("startPosX " + str(self.startPosX) + " startPosY " + str(self.startPosY))
        
    def move(self):
        # print(self.turning_points)
        startPosX = self.turning_points[0][0]
        startPosY = self.turning_points[0][1]
        afterStartX = self.turning_points[1][0]
        afterStartY = self.turning_points[1][1]
        endPosX = self.turning_points[len(self.turning_points) - 1][0]
        endPosY = self.turning_points[len(self.turning_points) - 1][1]
        beforeEndX = self.turning_points[len(self.turning_points) - 2][0]
        beforeEndY = self.turning_points[len(self.turning_points) - 2][1]
            
        # checking if the first turn has ended
        if(endPosX == beforeEndX and endPosY == beforeEndY):
            self.turning_points.pop(len(self.turning_points) - 2) # remove the turning point cz the turn has ended
            beforeEndX = self.turning_points[len(self.turning_points) - 2][0] #get the new turning location
            beforeEndY = self.turning_points[len(self.turning_points) - 2][1]

        #move the snake
        if(startPosX == afterStartX):
            if(startPosY > afterStartY): #start is heading downward
                startPosY += 1
            else: #start is heading upward
                startPosY -= 1
        if(startPosY == afterStartY):
            if(startPosX > afterStartX): #start is heading right
                startPosX += 1
            else: #start is heading left
                startPosX -= 1
        if(endPosX == beforeEndX):
            if(endPosY > beforeEndY): #end is heading upward
                endPosY -= 1
            else: #end is heading downward
                endPosY += 1
        if(endPosY == beforeEndY):
            if(endPosX > beforeEndX): #end is heading left
                endPosX -= 1
            else: #end is heading right
                endPosX += 1
        #hit the edge
        if(startPosX == (width + 1) or startPosX == -1 \
            or startPosY == (height + 1) or startPosY == -1):
            self.choose_turn_heading()
            self.set_turning_point()
            self.turn()
            return
        #update start and end locations as snake has moved
        self.turning_points[0] = (startPosX, startPosY)
        self.turning_points[len(self.turning_points) - 1] = (endPosX, endPosY)

    def draw(self):
        p_turn = random.randint(0, 100)
        if(p_turn > 99):
            self.choose_turn_heading()
            self.set_turning_point()
            # print("CURRENT " + str(self.heading))
            # print("PREV " + str(self.prev_heading))
            self.turn()
        else:
            self.move()
        pygame.draw.lines(screen, self.color, False, self.turning_points, self.size)
################################ end object

# this is the credits loop, which puts the title of your work
# and your name on the screen
make_black() # one second black
# produce title sequence
screen.fill((0,0,0))
screen.blit(name_f, (int(width/8), int(width/8)))
screen.blit(title_f, (int(width/8), int(width/4))) 
for i in range(0, 3*60): # three seconds of title and name
    pygame.display.update()
    pygame.image.save(screen, "./frames/" + str(frame_num) + ".png")
    frame_num = frame_num + 1
    clock.tick(60)
make_black() # one second black

# make a list of snakes
snakes = []
for i in range (0, 35):
    snakes.append(Snake())

# here is the main animation loop
for i in range(0, 30*60): # 20*60 frames is 20 seconds
    #########################################################
    # in the skeleton, your animation goes from here ########
    #########################################################
    screen.fill((0,0,0))
    for thing in snakes:
        thing.draw()
    #########################################################
    # to here ###############################################
    #########################################################

    # The next line can be commented out to speed up testing frame rate
    # by not writing the file. But for output to final frames,
    # you will need to ucomment it.
    pygame.image.save(screen, "./frames/" + str(frame_num) + ".png")
    frame_num = frame_num + 1
    pygame.display.update()
    clock.tick(60)

# print out stats
print("seconds:", int(time.time() - start_time))
print("~minutes: ", int((time.time() - start_time)/60))
# we just quit here
pygame.display.quit()
pygame.quit()
exit()

# you can make your files into a movie with ffmpeg:
# ffmpeg -r 60 -start_number 1000000 -s 4096x2160 -i %d.png -vcodec libx264 -crf 31 -pix_fmt yuv420p final.mp4
# with a few changes such as to start number, but this is just extra info here