반응형
이제 집에 가자. 많이 했다.
실행 명령은
$ python tetrisdigit.py
실행하면 아름다운 테트리스 숫자가 보인다. 테트리스 실행 화면이 너무 커서 중간에 메모리 문제로 정지하는 현상...
코드 참고는 Creating TetrisClock using OpenCV #1
import numpy as np
import cv2
import time
from PIL import Image, ImageDraw
gif_frames = []
def save_gif(frames, gifname, speed): #speed 100
frames[0].save(gifname, format='GIF', append_images=frames[1:], save_all=True, duration=speed, loop=0)
num_0 = (
(2, 5, 4, 16, 0),
(4, 7, 2, 16, 1),
(3, 4, 0, 16, 1),
(6, 6, 1, 16, 1),
(5, 1, 4, 14, 0),
(6, 6, 0, 13, 3),
(5, 1, 4, 12, 0),
(5, 1, 0, 11, 0),
(6, 6, 4, 10, 1),
(6, 6, 0, 9, 1),
(5, 1, 1, 8, 1),
(2, 5, 3, 8, 3))
num_1 = (
(2, 5, 4, 16, 0),
(3, 4, 4, 15, 1),
(3, 4, 5, 13, 3),
(2, 5, 4, 11, 2),
(0, 0, 4, 8, 0))
num_2 = (
(0, 0, 4, 16, 0),
(3, 4, 0, 16, 1),
(1, 2, 1, 16, 3),
(1, 2, 1, 15, 0),
(3, 4, 1, 12, 2),
(1, 2, 0, 12, 1),
(2, 5, 3, 12, 3),
(0, 0, 4, 10, 0),
(3, 4, 1, 8, 0),
(2, 5, 3, 8, 3),
(1, 2, 0, 8, 1))
num_3 = (
(1, 2, 3, 16, 3),
(2, 5, 0, 16, 1),
(3, 4, 1, 15, 2),
(0, 0, 4, 14, 0),
(3, 4, 1, 12, 2),
(1, 2, 0, 12, 1),
(3, 4, 5, 12, 3),
(2, 5, 3, 11, 0),
(3, 4, 1, 8, 0),
(1, 2, 0, 8, 1),
(2, 5, 3, 8, 3))
num_4 = (
(0, 0, 4, 16, 0),
(0, 0, 4, 14, 0),
(3, 4, 1, 12, 0),
(1, 2, 0, 12, 1),
(2, 5, 0, 10, 0),
(2, 5, 3, 12, 3),
(3, 4, 4, 10, 3),
(2, 5, 0, 9, 2),
(3, 4, 5, 10, 1))
num_5 = (
(0, 0, 0, 16, 0),
(2, 5, 2, 16, 1),
(2, 5, 3, 15, 0),
(3, 4, 5, 16, 1),
(3, 4, 1, 12, 0),
(1, 2, 0, 12, 1),
(2, 5, 3, 12, 3),
(0, 0, 0, 10, 0),
(3, 4, 1, 8, 2),
(1, 2, 0, 8, 1),
(2, 5, 3, 8, 3))
num_6 = (
(2, 5, 0, 16, 1),
(5, 1, 2, 16, 1),
(6, 6, 0, 15, 3),
(6, 6, 4, 16, 3),
(5, 1, 4, 14, 0),
(3, 4, 1, 12, 2),
(2, 5, 0, 13, 2),
(3, 4, 2, 11, 0),
(0, 0, 0, 10, 0),
(3, 4, 1, 8, 0),
(1, 2, 0, 8, 1),
(2, 5, 3, 8, 3))
num_7 = (
(0, 0, 4, 16, 0),
(1, 2, 4, 14, 0),
(3, 4, 5, 13, 1),
(2, 5, 4, 11, 2),
(3, 4, 1, 8, 2),
(2, 5, 3, 8, 3),
(1, 2, 0, 8, 1))
num_8 = (
(3, 4, 1, 16, 0),
(6, 6, 0, 16, 1),
(3, 4, 5, 16, 1),
(1, 2, 2, 15, 3),
(4, 7, 0, 14, 0),
(1, 2, 1, 12, 3),
(6, 6, 4, 13, 1),
(2, 5, 0, 11, 1),
(4, 7, 0, 10, 0),
(4, 7, 4, 11, 0),
(5, 1, 0, 8, 1),
(5, 1, 2, 8, 1),
(1, 2, 4, 9, 2))
num_9 = (
(0, 0, 0, 16, 0),
(3, 4, 2, 16, 0),
(1, 2, 2, 15, 3),
(1, 2, 4, 15, 2),
(3, 4, 1, 12, 2),
(3, 4, 5, 12, 3),
(5, 1, 0, 12, 0),
(1, 2, 2, 11, 3),
(5, 1, 4, 9, 0),
(6, 6, 0, 10, 1),
(5, 1, 0, 8, 1),
(6, 6, 2, 8, 2))
'''BGR'''
myRED = (0, 0, 255)
myGREEN = (0, 255, 0)
myBLUE = (255, 73, 48)
myWHITE = (255, 255, 255)
myYELLOW = (0, 255, 255)
myCYAN = (255, 255, 0)
myMAGENTA = (255, 0, 255)
myORANGE = (0, 96, 255)
myBLACK = (45, 45, 45)
myCOLORS = (myRED, myGREEN, myBLUE, myWHITE, myYELLOW, myCYAN, myMAGENTA, myBLACK)
mynums = (num_0, num_1, num_2, num_3, num_4, num_5, num_6, num_7, num_8, num_9)
scale = 20
x_shift = 2
y_shift = 1
'''
Don't make shape, just paint the pixel.
rotate: %4 => 0 ~ 3
'''
def draw_shape(canvas, x, y, color, shape, rotate, y_pos):
tcanvas = canvas.copy()
rot = rotate % 4
ret = False
if shape == 0: #rantangle
if rot == 0 or rot == 1 or rot == 2 or rot == 3:
for i in range(x, x + 2*scale):
for j in range(y + 0*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif shape == 1:
if rot == 3:
for i in range(x + 2*scale, x + 3*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x , x + 3*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 0:
for i in range(x , x + 1*scale):
for j in range(y, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 1:
for i in range(x , x + 3*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x, x + 1*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 2:
for i in range(x , x + 2*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y + 2*scale, y + 4*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif shape == 2:
if rot == 1:
for i in range(x, x + 1*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x , x + 3*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 2:
for i in range(x, x + 2*scale):
for j in range(y + 0*scale, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x , x + 1*scale):
for j in range(y + 0*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 3:
for i in range(x, x + 3*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 2*scale , x + 3*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 0:
for i in range(x, x + 1*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale , x + 2*scale):
for j in range(y, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif shape == 3:
if rot == 0 or rot == 2:
for i in range(x, x + 4*scale):
for j in range(y + 2*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 1 or rot == 3:
for i in range(x, x + 1*scale):
for j in range(y, y + 4*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif shape == 4:
if rot == 0:
for i in range(x, x + 1*scale):
for j in range(y, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y + 1*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 1:
for i in range(x + 1*scale, x + 3*scale):
for j in range(y, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x , x + 2*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 2:
for i in range(x, x + 1*scale):
for j in range(y, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale , x + 2*scale):
for j in range(y + 1*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 3:
for i in range(x + 1*scale, x + 3*scale):
for j in range(y, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x, x + 2*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif shape == 5:
if rot == 0:
for i in range(x, x + 1*scale):
for j in range(y + 1*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 1:
for i in range(x, x + 2*scale):
for j in range(y, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 3*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 2:
for i in range(x, x + 1*scale):
for j in range(y + 1*scale, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 3:
for i in range(x, x + 2*scale):
for j in range(y, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 3*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif shape == 6:
if rot == 0:
for i in range(x + 1*scale, x + 2*scale):
for j in range(y, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x, x + 3*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 1:
for i in range(x, x + 1*scale):
for j in range(y, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 2:
for i in range(x, x + 3*scale):
for j in range(y, y + 1*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
elif rot == 3:
for i in range(x, x + 1*scale):
for j in range(y + 1*scale, y + 2*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
for i in range(x + 1*scale, x + 2*scale):
for j in range(y, y + 3*scale):
tcanvas[j, i] = color
if j == (y_pos - 1):
ret = True
return tcanvas, ret
def animate_number(canvas, n, x_shift, y_shift):
tcanvas = canvas.copy()
num = mynums[n]
for i in num:
print(i)
shape = i[0]
color = myCOLORS[i[1]]
x_pos = i[2] + x_shift
y_pos = i[3] - y_shift
rotation = i[4]
y = 0
rot = 0
while True:
mycanvas, ret = draw_shape(tcanvas, x_pos * scale, y * scale, color, shape, rot, y_pos * scale)
# cv2.imshow("%d"%(n), mycanvas)
cv2.imshow("Tetris", mycanvas)
rgb = cv2.cvtColor(mycanvas, cv2.COLOR_BGR2RGB)
im_pil = Image.fromarray(rgb)
gif_frames.append(im_pil)
cv2.waitKey(50)
y += 1
if rot != rotation:
rot += 1
if ret == True:
break
tcanvas = mycanvas.copy()
cv2.waitKey(100)
return mycanvas
cv2.waitKey(0)
def make_canvas(h, w, color):
# canvas = np.zeros([(h + 1) * scale,(w + 1) * scale,3], dtype=np.uint8)
canvas = np.zeros([(h ) * scale,(w ) * scale,3], dtype=np.uint8)
canvas.fill(color)
return canvas
def test_shape():
for shape in range(0, 7):
for rotate in range(0, 4):
print(shape)
tcanvas, _ = draw_shape(canvas, 0, 0, myBLUE, shape, rotate, y_shift)
cv2.imshow("tetris", tcanvas)
cv2.waitKey(0)
def set_scale(s):
global scale
scale = s
if __name__ == '__main__':
set_scale(20)
canvas = make_canvas(16 * int(scale / 20 + 0.5), 32 * int(scale / 4 + 0.5), 0)
# test_shape()
tcanv = canvas
shift = x_shift
for i in range(0, 10):
tcanv = animate_number(tcanv, i, shift , y_shift)
shift += 11
# save_gif(gif_frames, "f:\\tmp\\tetris.gif", 50)
cv2.waitKey(0)
cv2.destroyAllWindows()
반응형
취업, 창업의 막막함, 외주 관리, 제품 부재!
당신의 고민은 무엇입니까? 현실과 동떨어진 교육, 실패만 반복하는 외주 계약,
아이디어는 있지만 구현할 기술이 없는 막막함.
우리는 알고 있습니다. 문제의 원인은 '명확한 학습, 실전 경험과 신뢰할 수 있는 기술력의 부재'에서 시작됩니다.
이제 고민을 멈추고, 캐어랩을 만나세요!
코딩(펌웨어), 전자부품과 디지털 회로설계, PCB 설계 제작, 고객(시장/수출) 발굴과 마케팅 전략으로 당신을 지원합니다.
제품 설계의 고수는 성공이 만든 게 아니라 실패가 만듭니다. 아이디어를 양산 가능한 제품으로!
귀사의 제품을 만드세요. 교육과 개발 실적으로 신뢰할 수 있는 파트너를 확보하세요.
캐어랩