cardiod_new.py

# Import a library of functions called 'pygame'
import pygame
import math

# Initialize the game engine
pygame.init()

# Define the colors we will use in RGB format
black = [0, 0, 0]
white = [255, 255, 255]
blue = [0, 0, 255]
green = [0, 255, 0]
red = [255, 0, 0]

PI = 3.141592653
r = 100
R = 100

# Set the height and width of the screen
size = [800, 480]
screen = pygame.display.set_mode(size)

pygame.display.set_caption("Sansagraphics World")

# Loop until the user clicks the close button.
done = False
clock = pygame.time.Clock()

colour = red
thickness = 2
i = 0
ntotal = 60
ddelta = 2 * PI / ntotal / 2
delta = 0.0
a = 2.0
b = 1.0

while not done:

    # This limits the while loop to a max of 10 times per second.
    # Leave this out and we will use all CPU we can.
    clock.tick(10)

    for event in pygame.event.get():  # User did something
        if event.type == pygame.QUIT:  # If user clicked close
            done = True  # Flag that we are done so we exit this loop

    # All drawing code happens after the for loop and but
    # inside the main while done==False loop.

    # Clear the screen and set the screen background
    screen.fill(white)
    prevx = 400.0
    prevy = 240.0

    # Draw on the screen several green lines from (0,10) to (100,110)
    # 5 pixels wide using a loop
    colour = red
    for n in range(ntotal * 2):
        theta = n * 2 * PI / ntotal
        y_offset = R * math.cos((theta - delta) * a) + r * math.cos((theta - delta + PI) * b) + 240.0
        x_offset = R * math.sin((theta - delta) * a) + r * math.sin((theta - delta + PI) * b) + 400.0
        cx = int(120 * (math.sin((theta - delta) / 10) + 1) + 10)
        cy = int(120 * (math.sin((theta - delta) * 6 / 10) + 1) + 10)
        cz = int(120 * (math.sin((theta - delta) / 10 - PI) + 1) + 10)

        if n > 0:
            pygame.draw.line(screen, [cx, cy, cz], [prevx, prevy], [x_offset, y_offset], thickness)

        prevx = x_offset
        prevy = y_offset

    # Go ahead and update the screen with what we've drawn.
    # This MUST happen after all the other drawing commands.
    delta += ddelta
    pygame.display.flip()

# Be IDLE friendly
pygame.quit()