quantum/punyverse
1
0
Fork 0
mirror of https://github.com/quantum5/punyverse.git synced 2025-04-24 13:11:57 -04:00
Code Issues Packages Projects Releases Wiki Activity

Added orbital display.

Browse source
This commit is contained in:
Quantum 2013-10-26 14:25:37 -04:00
parent f21499062d
commit 8236efcffc
2 changed files with 43 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
punyverse/entity.py
View file

@ -1,7 +1,8 @@
from punyverse import framedata from punyverse import framedata
from math import radians, sin, cos
from punyverse.orbit import KeplerOrbit from punyverse.orbit import KeplerOrbit
from pyglet.gl import *
class Entity(object): class Entity(object):
def __init__(self, id, location, rotation=(0, 0, 0), direction=(0, 0, 0), background=False): def __init__(self, id, location, rotation=(0, 0, 0), direction=(0, 0, 0), background=False):
@ -60,11 +61,46 @@ class Satellite(Planet):
longitude = kwargs.pop('longitude', 0) longitude = kwargs.pop('longitude', 0)
argument = kwargs.pop('argument', 0) argument = kwargs.pop('argument', 0)
# Orbit calculation
self.orbit_id = None
self.orbit_cache = None
self.theta = 0 self.theta = 0
# OpenGL's z-axis is reversed # OpenGL's z-axis is reversed
self.orbit = KeplerOrbit(distance, eccentricity, inclination, longitude, argument) self.orbit = KeplerOrbit(distance, eccentricity, inclination, longitude, argument)
super(Satellite, self).__init__(*args, **kwargs) super(Satellite, self).__init__(*args, **kwargs)
def get_orbit(self):
# Cache key is the three orbital plane parameters and eccentricity
cache = (self.orbit.eccentricity, self.orbit.longitude, self.orbit.inclination, self.orbit.argument)
if self.orbit_cache == cache:
return self.orbit_id
print 'Orbit cache miss'
if self.orbit_id is not None:
glDeleteLists(self.orbit_id, 1)
id = glGenLists(1)
glNewList(id, GL_COMPILE)
glDisable(GL_LIGHTING)
glDisable(GL_TEXTURE_2D)
glPushAttrib(GL_LINE_BIT | GL_CURRENT_BIT)
glColor3f(0, 1, 0)
glLineWidth(3)
glBegin(GL_LINE_LOOP)
for theta in xrange(360):
x, z, y = self.orbit.orbit(theta)
glVertex3f(x, y, z)
glEnd()
glPopAttrib()
glEnable(GL_TEXTURE_2D)
glEnable(GL_LIGHTING)
glEndList()
self.orbit_id = id
self.orbit_cache = cache
return id
def update(self): def update(self):
super(Planet, self).update() super(Planet, self).update()

6
punyverse/game.py
View file

@ -265,6 +265,12 @@ class Applet(pyglet.window.Window):
glDisable(GL_ALPHA_TEST) glDisable(GL_ALPHA_TEST)
glPopMatrix() glPopMatrix()
if self.debug and hasattr(entity, 'get_orbit') and hasattr(entity, 'parent'):
glPushMatrix()
glTranslatef(*entity.parent.location)
glCallList(entity.get_orbit())
glPopMatrix()
glColor4f(1, 1, 1, 1) glColor4f(1, 1, 1, 1)
glDisable(GL_TEXTURE_2D) glDisable(GL_TEXTURE_2D)