88 lines
3.0 KiB
Python
Executable File
88 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Sat Jan 25 20:54:34 2020
|
|
|
|
@author: suwako
|
|
"""
|
|
|
|
import sys, pygame, time, random
|
|
import numpy as np
|
|
|
|
|
|
pygame.init()
|
|
black=(0,0,0)
|
|
size = width, height = 1366, 720
|
|
TICK_RATE=60
|
|
camera=np.array([0.,0.])
|
|
origin=camera=np.array([0.,0.])
|
|
follow=0
|
|
screen = pygame.display.set_mode(size)
|
|
|
|
class satellite(pygame.sprite.Sprite):
|
|
def __init__(self):
|
|
pygame.sprite.Sprite.__init__(self)
|
|
self.image = pygame.image.load("intro_ball.gif").convert_alpha()
|
|
self.position = self.image.get_rect()
|
|
self.speed=np.array([0.,0.])
|
|
self.pos = np.array([0.,0.])
|
|
self.mass=1
|
|
self.G=0.0000001
|
|
def tick(self, camera=np.array([0.,0.]), origin=np.array([0.,0.])):
|
|
self.pos += self.speed
|
|
def apply(self, camera=np.array([0.,0.]), origin=np.array([0.,0.])):
|
|
self.position.x, self.position.y = map(int, self.pos-camera-origin)
|
|
def apply_gravity(self, elements):
|
|
for element in elements:
|
|
if element != self:
|
|
d=(element.pos - self.pos )
|
|
u=d/np.linalg.norm(d)
|
|
norm = self.G * element.mass*(d[0]**2+d[1]**2)
|
|
self.speed+=norm*u
|
|
premier = satellite()
|
|
#premier.position.center = (width/2, height/2)
|
|
tick=(time.time(), time.time(), 0)
|
|
def clip(val, minval, maxval):
|
|
return min(max(val, minval), maxval)
|
|
elements=[satellite() for _ in range(3)]
|
|
for element in elements:
|
|
element.pos=np.array([random.randint(0,width), random.randint(0,height)], dtype='float64')
|
|
element.speed=np.array([random.randint(0,50), random.randint(0,50)], dtype='float64')
|
|
elements[1].pos= np.array([float(width/2), float(height/2)])
|
|
elements[0].pos= np.array([float(width/4), float(height/4)])
|
|
elements[0].speed[1]=10
|
|
elements[1].mass=100
|
|
elements[2].mass=30
|
|
keys=set()
|
|
while 1:
|
|
dtick=tick
|
|
tick=(tick[0], time.time(), int((time.time() - tick[0])*TICK_RATE))
|
|
dtick=tick[2] - dtick[2]
|
|
if dtick>2:
|
|
print(dtick)
|
|
for _ in range(dtick):
|
|
for event in pygame.event.get():
|
|
#print(event.type)
|
|
if event.type == pygame.QUIT: sys.exit()
|
|
if event.type == pygame.KEYDOWN:
|
|
keys.update({event.key})
|
|
print(event.key)
|
|
if event.key == 102:
|
|
follow = (follow + 1)%len(elements)
|
|
if event.type == pygame.KEYUP:
|
|
if event.key in keys:
|
|
keys.remove(event.key)
|
|
screen.fill(black)
|
|
for key in keys:
|
|
if key in [275,276,273,274]:
|
|
exec("camera" + ("[0]+","[0]-","[1]-","[1]+")[[275,276,273,274].index(key)] + "=30")
|
|
for element in elements:
|
|
element.apply_gravity(elements)
|
|
element.tick(camera=camera)
|
|
origin=elements[follow].pos
|
|
for element in elements:
|
|
element.apply(camera=camera, origin=origin)
|
|
screen.blit(element.image, element.position)
|
|
pygame.display.flip()
|
|
time.sleep(0.001)
|