mirror of
https://github.com/quantum5/punyverse.git
synced 2025-08-05 01:28:57 -04:00
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
|
from punyverse import framedata
|
||
|
|
from math import radians, sin, cos
|
||
|
|
|
||
|
|
class Entity(object):
|
||
|
|
def __init__(self, id, location, rotation=(0, 0, 0), direction=(0, 0, 0), background=False):
|
||
|
|
self.id = id
|
||
|
|
self.location = location
|
||
|
|
self.rotation = rotation
|
||
|
|
self.direction = direction
|
||
|
|
self.background = background
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
x, y, z = self.location
|
||
|
|
dx, dy, dz = self.direction
|
||
|
|
self.location = x + dx, y + dy, z + dz
|
||
|
|
|
||
|
|
|
||
|
|
class Asteroid(Entity):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super(Asteroid, self).__init__(*args, **kwargs)
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
super(Asteroid, self).update()
|
||
|
|
rx, ry, rz = self.rotation
|
||
|
|
# Increment all axis to 'spin'
|
||
|
|
self.rotation = rx + 1, ry + 1, rz + 1
|
||
|
|
|
||
|
|
|
||
|
|
class Planet(Entity):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
self.delta = kwargs.pop('delta', 5)
|
||
|
|
self.atmosphere = kwargs.pop('atmosphere', 0)
|
||
|
|
self.cloudmap = kwargs.pop('cloudmap', 0)
|
||
|
|
self.last_tick = 0
|
||
|
|
super(Planet, self).__init__(*args, **kwargs)
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
super(Planet, self).update()
|
||
|
|
|
||
|
|
if self.last_tick != framedata.tick:
|
||
|
|
self.last_tick = framedata.tick
|
||
|
|
pitch, yaw, roll = self.rotation
|
||
|
|
roll += self.delta / 100.0
|
||
|
|
self.rotation = pitch, yaw, roll
|
||
|
|
|
||
|
|
|
||
|
|
class Satellite(Planet):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
self.parent = kwargs.pop('parent')
|
||
|
|
self.orbit_speed = kwargs.pop('orbit_speed', 1)
|
||
|
|
self.inclination = kwargs.pop('inclination', 0)
|
||
|
|
self.distance = kwargs.pop('distance', 100)
|
||
|
|
self.theta = 0
|
||
|
|
super(Satellite, self).__init__(*args, **kwargs)
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
super(Planet, self).update()
|
||
|
|
|
||
|
|
if self.last_tick != framedata.tick:
|
||
|
|
self.last_tick = framedata.tick
|
||
|
|
pitch, yaw, roll = self.rotation
|
||
|
|
roll += self.delta / 100.0
|
||
|
|
self.rotation = pitch, yaw, roll
|
||
|
|
|
||
|
|
self.parent.update()
|
||
|
|
x, y, z = self.location
|
||
|
|
px, py, pz = self.parent.location
|
||
|
|
self.theta += self.orbit_speed
|
||
|
|
x = cos(radians(self.theta)) * self.distance + px
|
||
|
|
z = sin(radians(self.theta)) * self.distance + pz
|
||
|
|
self.location = (x, y, z)
|
||
|
|
|