FilesExpand file tree

 master

/

matplotlib_demo.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

76 lines (55 loc) · 1.45 KB

 master

/

matplotlib_demo.py

Copy path

Top

File metadata and controls

• Code
• Blame

76 lines (55 loc) · 1.45 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

import pygame, sys

from pygame.locals import *

import numpy as np

import matplotlib as mpl

mpl.use("Agg")

import matplotlib.pyplot as plt

import matplotlib.backends.backend_agg as agg

fig = plt.figure(figsize=[3, 3])

ax = fig.add_subplot(111)

canvas = agg.FigureCanvasAgg(fig)

def plot(data):

ax.plot(data)

canvas.draw()

renderer = canvas.get_renderer()

raw_data = renderer.tostring_rgb()

size = canvas.get_width_height()

return pygame.image.fromstring(raw_data, size, "RGB")

pygame.init()

clock = pygame.time.Clock()

screen = pygame.display.set_mode((400, 400))

pygame.display.set_caption('Animating Objects')

img = pygame.image.load('head.jpg')

steps = np.linspace(20, 360, 40).astype(int)

right = np.zeros((2, len(steps)))

down = np.zeros((2, len(steps)))

left = np.zeros((2, len(steps)))

up = np.zeros((2, len(steps)))

right[0] = steps

right[1] = 20

down[0] = 360

down[1] = steps

left[0] = steps[::-1]

left[1] = 360

up[0] = 20

up[1] = steps[::-1]

pos = np.concatenate((right.T, down.T, left.T, up.T))

i = 0

history = np.array([])

surf = plot(history)

while True:

# Erase screen

screen.fill((255, 255, 255))

if i >= len(pos):

i = 0

surf = plot(history)

screen.blit(img, pos[i])

history = np.append(history, pos[i])

screen.blit(surf, (100, 100))

i += 1

for event in pygame.event.get():

if event.type == QUIT:

pygame.quit()

sys.exit()

pygame.display.update()

clock.tick(30)