Skip to content

bad_apple

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import cv2
from gridmas import *

name = "Bad Apple"
author = "Ciaran"


def draw():
    cap = cv2.VideoCapture("patterns/badapple.mp4")
    coords = []

    for pixel in pixels():
        videox = max(min(int(((-pixel.x + 1) / 2) * 480), 479), 0)
        videoy = max(min(int((1 - pixel.z / 2) * 360), 359), 0)
        coords.append((videox, videoy))

    set_fps(30)

    while True:

        ret, frame = cap.read()
        if not ret:
            break

        for i, pixel in enumerate(pixels()):
            if 0.5 < pixel.z < 2.5:
                value = int(frame[coords[i][1]][coords[i][0]][0])
                pixel.set_rgb(value, value, value)

            else:
                pixel.set_rgb(0, 0, 0)
        yield

    cap.release()