added basic test image generator
Todo: make video from images
This commit is contained in:
parent
39a5fb6066
commit
7470513019
|
|
@ -0,0 +1,49 @@
|
|||
|
||||
import math
|
||||
from PIL import Image, ImageDraw
|
||||
import random
|
||||
|
||||
|
||||
def getRandomColorString():
|
||||
return '#{:06x}'.format(random.randint(0, 256**3))
|
||||
|
||||
fps = 30
|
||||
xmax = 1920
|
||||
ymax = 1080
|
||||
# in minutes
|
||||
length = 1
|
||||
numberOfEvents = 3
|
||||
|
||||
|
||||
|
||||
# creating new Image object
|
||||
img = Image.new("RGB", (xmax, ymax))
|
||||
|
||||
for i in range(numberOfEvents):
|
||||
|
||||
objectWidth = (5 + random.randint(0, 5)) * xmax / 100
|
||||
objectHeight = (10 + random.randint(-5, 5)) * ymax / 100
|
||||
|
||||
objectX = random.randint(0, xmax)
|
||||
objectY = random.randint(0, ymax)
|
||||
|
||||
objectSpeedX = random.randint( -int(objectWidth) - 1, int(objectWidth) + 1)
|
||||
objectSpeedY = random.randint( -int(objectWidth) - 1, int(objectWidth) + 1)
|
||||
color = getRandomColorString()
|
||||
|
||||
for j in range(int(fps*length*60 / numberOfEvents)):
|
||||
|
||||
objectX -= objectSpeedX
|
||||
objectY -= objectSpeedY
|
||||
|
||||
objectShape = [(objectX, objectY), (objectX + objectWidth, objectY + objectHeight)]
|
||||
|
||||
|
||||
|
||||
# create rectangle image
|
||||
img1 = ImageDraw.Draw(img)
|
||||
|
||||
img1.rectangle(objectShape, fill = color)
|
||||
|
||||
img.show()
|
||||
|
||||
Loading…
Reference in New Issue