Compare commits
23 commits
master
...
launcher0.
Author | SHA1 | Date | |
---|---|---|---|
|
e46b09ba06 | ||
|
e45f8f6f85 | ||
|
de08850f24 | ||
|
a7ffda822e | ||
|
118bcd02a7 | ||
|
c21c0b9535 | ||
|
d831e996e7 | ||
|
70e12f180d | ||
|
a9aabc7465 | ||
|
376628d9a6 | ||
|
54a134c73e | ||
|
46b46df60f | ||
|
1bb8893635 | ||
|
45d7142feb | ||
|
c224ba4074 | ||
|
a034975928 | ||
|
8236efcffc | ||
|
f21499062d | ||
|
2cc82a2724 | ||
|
d26c7dd536 | ||
|
b2e3a1d486 | ||
|
3c6eed8f39 | ||
|
355f156295 |
17
.gitignore
vendored
|
@ -34,3 +34,20 @@ nosetests.xml
|
||||||
.mr.developer.cfg
|
.mr.developer.cfg
|
||||||
.project
|
.project
|
||||||
.pydevproject
|
.pydevproject
|
||||||
|
|
||||||
|
.idea
|
||||||
|
punyverse/*.obj
|
||||||
|
*.html
|
||||||
|
*.exp
|
||||||
|
*.lib
|
||||||
|
punyverse/assets/textures/*_medium.*
|
||||||
|
punyverse/assets/textures/*_small.*
|
||||||
|
punyverse/assets/textures/*/*_medium.*
|
||||||
|
punyverse/assets/textures/*/*_small.*
|
||||||
|
temp
|
||||||
|
|
||||||
|
# There is a py2exe package that can execute punyverse directly
|
||||||
|
library.zip
|
||||||
|
*.exe
|
||||||
|
*.log
|
||||||
|
*.dll
|
||||||
|
|
35
bootloader.py
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
import pyglet
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import uuid
|
||||||
|
import imp
|
||||||
|
|
||||||
|
def load_dll(dir, module):
|
||||||
|
name = 'punyverse.' + module
|
||||||
|
path = os.path.join(dir, 'punyverse', module + '.pyd')
|
||||||
|
if not os.path.exists(path):
|
||||||
|
path = os.path.join(dir, 'punyverse.%s.pyd' % module)
|
||||||
|
if not os.path.exists(path):
|
||||||
|
raise ImportError('No module named %s' % module)
|
||||||
|
return imp.load_dynamic(name, path)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
try:
|
||||||
|
dir = os.path.dirname(sys.executable)
|
||||||
|
if sys.frozen == 'windows_exe':
|
||||||
|
sys.stderr = open(os.path.join(dir, 'punyverse.log'), 'a')
|
||||||
|
except AttributeError:
|
||||||
|
sys.exit('This is only meant to be ran frozen.')
|
||||||
|
|
||||||
|
sys.path.insert(0, dir)
|
||||||
|
|
||||||
|
# Model indirectly depends on _glgeom to handle textures
|
||||||
|
load_dll(dir, '_glgeom')
|
||||||
|
|
||||||
|
# Model path needs special handling
|
||||||
|
_model = load_dll(dir, '_model')
|
||||||
|
_model.model_base = os.path.join(dir, 'punyverse', 'assets', 'models')
|
||||||
|
|
||||||
|
with open('punyverse\__main__.py', 'r') as code:
|
||||||
|
exec(code)
|
29
launcher.py
Normal file
|
@ -0,0 +1,29 @@
|
||||||
|
from distutils.core import setup
|
||||||
|
import py2exe
|
||||||
|
from glob import glob
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
|
||||||
|
if len(sys.argv) < 2:
|
||||||
|
sys.argv.append('py2exe')
|
||||||
|
|
||||||
|
parent = os.path.dirname(__file__)
|
||||||
|
join = os.path.join
|
||||||
|
|
||||||
|
setup(
|
||||||
|
console=[{'dest_base': 'punyverse_debug', 'script': 'bootloader.py'}, 'small_images.py'],
|
||||||
|
windows=[{'dest_base': 'punyverse', 'script': 'bootloader.py'}],
|
||||||
|
options={'py2exe': {
|
||||||
|
'unbuffered': True, 'optimize': 2,
|
||||||
|
'excludes': [
|
||||||
|
'_ssl', 'unittest', 'doctest', 'PIL', 'email', 'distutils',
|
||||||
|
'pyglet.window.carbon', 'pyglet.window.xlib',
|
||||||
|
'pyglet.media.drivers.alsa',
|
||||||
|
'win32wnet', 'netbios', 'pgmagick'
|
||||||
|
],
|
||||||
|
'includes': ['punyverse._model', 'punyverse._glgeom'],
|
||||||
|
'dll_excludes': ['MPR.dll', 'w9xpopen.exe'],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
1
punyverse/__init__.py
Normal file
|
@ -0,0 +1 @@
|
||||||
|
|
17
punyverse/__main__.py
Normal file
|
@ -0,0 +1,17 @@
|
||||||
|
#!/usr/bin/python
|
||||||
|
INITIAL_WIN_HEIGHT = 540
|
||||||
|
INITIAL_WIN_WIDTH = 700
|
||||||
|
WIN_TITLE = "Punyverse"
|
||||||
|
|
||||||
|
def main():
|
||||||
|
import pyglet
|
||||||
|
from punyverse.game import Applet
|
||||||
|
|
||||||
|
pyglet.options['shadow_window'] = False
|
||||||
|
|
||||||
|
Applet(width=INITIAL_WIN_WIDTH, height=INITIAL_WIN_HEIGHT, caption=WIN_TITLE, resizable=True, vsync=0)
|
||||||
|
pyglet.app.run()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
4
punyverse/__main__.pyw
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
from punyverse.__main__ import main
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
1209
punyverse/_cyopengl.pxi
Normal file
3348
punyverse/_glgeom.c
Normal file
83
punyverse/_glgeom.pyx
Normal file
|
@ -0,0 +1,83 @@
|
||||||
|
from libc.math cimport sin, cos, sqrt
|
||||||
|
from libc.stdlib cimport malloc, free
|
||||||
|
from libc.string cimport memcpy
|
||||||
|
cimport cython
|
||||||
|
|
||||||
|
include "_cyopengl.pxi"
|
||||||
|
cdef float PI = 3.1415926535897932324626
|
||||||
|
cdef float TWOPI = PI * 2
|
||||||
|
|
||||||
|
cdef extern from "Python.h":
|
||||||
|
object PyBytes_FromStringAndSize(const char *s, Py_ssize_t len)
|
||||||
|
const char* PyBytes_AsString(bytes o)
|
||||||
|
|
||||||
|
@cython.cdivision(True)
|
||||||
|
cpdef torus(float major_radius, float minor_radius, int n_major, int n_minor, tuple material, int shininess=125):
|
||||||
|
"""
|
||||||
|
Torus function from the OpenGL red book.
|
||||||
|
"""
|
||||||
|
glPushAttrib(GL_CURRENT_BIT)
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, [material[0], material[1], material[2], material[3]])
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, [1, 1, 1, 1])
|
||||||
|
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, shininess)
|
||||||
|
|
||||||
|
assert n_major > 0 and n_minor > 0
|
||||||
|
assert minor_radius > 0 and major_radius > 0
|
||||||
|
|
||||||
|
cdef float major_s, minor_s
|
||||||
|
major_s = TWOPI / n_major
|
||||||
|
minor_s = TWOPI / n_minor
|
||||||
|
|
||||||
|
cdef float a0, a1, x0, y0, x1, y1, b, c, r, z, m, x, y, z2
|
||||||
|
cdef int i, j
|
||||||
|
|
||||||
|
for i in xrange(n_major):
|
||||||
|
a0 = i * major_s
|
||||||
|
a1 = a0 + major_s
|
||||||
|
x0 = cos(a0)
|
||||||
|
y0 = sin(a0)
|
||||||
|
x1 = cos(a1)
|
||||||
|
y1 = sin(a1)
|
||||||
|
|
||||||
|
glBegin(GL_TRIANGLE_STRIP)
|
||||||
|
|
||||||
|
for j in xrange(n_minor + 1):
|
||||||
|
b = j * minor_s
|
||||||
|
c = cos(b)
|
||||||
|
r = minor_radius * c + major_radius
|
||||||
|
z = minor_radius * sin(b)
|
||||||
|
|
||||||
|
x = x0 * c
|
||||||
|
y = y0 * c
|
||||||
|
z2 = z / minor_radius
|
||||||
|
m = 1.0 / sqrt(x * x + y * y + z2 * z2)
|
||||||
|
glNormal3f(x * m, y * z, z2 * m)
|
||||||
|
glVertex3f(x0 * r, y0 * r, z)
|
||||||
|
|
||||||
|
x = x1 * c
|
||||||
|
y = y1 * c
|
||||||
|
m = 1.0 / sqrt(x * x + y * y + z2 * z2)
|
||||||
|
glNormal3f(x * m, y * z, z2 * m)
|
||||||
|
glVertex3f(x1 * r, y1 * r, z)
|
||||||
|
|
||||||
|
glEnd()
|
||||||
|
glPopAttrib()
|
||||||
|
|
||||||
|
cpdef bytes bgr_to_rgb(bytes buffer, int width, int height, bint alpha=0, bint bottom_up=1):
|
||||||
|
cdef int length = len(buffer)
|
||||||
|
cdef int depth = length / (width * height)
|
||||||
|
cdef int depth2 = depth - alpha
|
||||||
|
cdef char *result = <char*>malloc(length)
|
||||||
|
cdef const char *source = PyBytes_AsString(buffer)
|
||||||
|
cdef int x, y, ioffset, ooffset, i, row = width * depth
|
||||||
|
for y in xrange(height):
|
||||||
|
for x in xrange(width):
|
||||||
|
ioffset = y * width * depth + x * depth
|
||||||
|
ooffset = (height - y - 1 if bottom_up else y) * row + x * depth
|
||||||
|
for i in xrange(depth2):
|
||||||
|
result[ooffset+i] = source[ioffset+depth2-i-1]
|
||||||
|
if alpha:
|
||||||
|
result[ooffset+depth2] = source[ioffset+depth2]
|
||||||
|
cdef object final = PyBytes_FromStringAndSize(result, length)
|
||||||
|
free(result)
|
||||||
|
return final
|
12929
punyverse/_model.c
Normal file
357
punyverse/_model.pyx
Normal file
|
@ -0,0 +1,357 @@
|
||||||
|
from libc.string cimport strcmp, strlen
|
||||||
|
from libc.stdlib cimport malloc, free, atof
|
||||||
|
from libc.stdio cimport fopen, fclose, fgets, FILE
|
||||||
|
cimport cython
|
||||||
|
|
||||||
|
from punyverse.texture import load_texture
|
||||||
|
include "_cyopengl.pxi"
|
||||||
|
from uuid import uuid4
|
||||||
|
import os
|
||||||
|
|
||||||
|
cdef enum:
|
||||||
|
FACE_TRIANGLES
|
||||||
|
FACE_QUADS
|
||||||
|
|
||||||
|
cdef class Face(object):
|
||||||
|
cdef public int type
|
||||||
|
cdef public list verts, norms, texs, vertices, normals, textures
|
||||||
|
|
||||||
|
def __init__(self, int type, list verts, list norms, list texs,
|
||||||
|
list vertices, list normals, list textures):
|
||||||
|
self.type = type
|
||||||
|
self.verts = verts
|
||||||
|
self.norms = norms
|
||||||
|
self.texs = texs
|
||||||
|
self.vertices = vertices
|
||||||
|
self.normals = normals
|
||||||
|
self.textures = textures
|
||||||
|
|
||||||
|
cdef class Material(object):
|
||||||
|
cdef public str name, texture
|
||||||
|
cdef public tuple Ka, Kd, Ks
|
||||||
|
cdef public double shininess
|
||||||
|
|
||||||
|
def __init__(self, str name, str texture=None, tuple Ka=(0, 0, 0),
|
||||||
|
tuple Kd=(0, 0, 0), tuple Ks=(0, 0, 0), double shininess=0.0):
|
||||||
|
self.name = name
|
||||||
|
self.texture = texture
|
||||||
|
self.Ka = Ka
|
||||||
|
self.Kd = Kd
|
||||||
|
self.Ks = Ks
|
||||||
|
self.shininess = shininess
|
||||||
|
|
||||||
|
cdef class Group(object):
|
||||||
|
cdef public str name
|
||||||
|
cdef public tuple min
|
||||||
|
cdef public Material material
|
||||||
|
cdef public list faces, indices, vertices, normals, textures
|
||||||
|
cdef public int idx_count
|
||||||
|
|
||||||
|
def __init__(self, str name=None):
|
||||||
|
if name is None:
|
||||||
|
self.name = str(uuid4())
|
||||||
|
else:
|
||||||
|
self.name = name
|
||||||
|
self.min = ()
|
||||||
|
self.material = None
|
||||||
|
self.faces = []
|
||||||
|
self.indices = []
|
||||||
|
self.vertices = []
|
||||||
|
self.normals = []
|
||||||
|
self.textures = []
|
||||||
|
self.idx_count = 0
|
||||||
|
|
||||||
|
def pack(self):
|
||||||
|
min_x, min_y, min_z = 0, 0, 0
|
||||||
|
for face in self.faces:
|
||||||
|
for x, y, z in face.vertices:
|
||||||
|
min_x = max(min_x, abs(x))
|
||||||
|
min_y = max(min_y, abs(y))
|
||||||
|
min_z = max(min_x, abs(z))
|
||||||
|
self.min = (min_x, min_y, min_z)
|
||||||
|
|
||||||
|
cdef class WavefrontObject(object):
|
||||||
|
cdef unicode root
|
||||||
|
cdef public list vertices, normals, textures, groups
|
||||||
|
cdef public dict materials
|
||||||
|
cdef unicode path
|
||||||
|
cdef Material current_material
|
||||||
|
cdef Group current_group
|
||||||
|
def __init__(self, unicode path):
|
||||||
|
self.path = path
|
||||||
|
self.root = os.path.abspath(os.path.dirname(path))
|
||||||
|
self.vertices = []
|
||||||
|
self.normals = []
|
||||||
|
self.textures = []
|
||||||
|
self.groups = []
|
||||||
|
self.materials = {}
|
||||||
|
|
||||||
|
self.perform_io(self.path)
|
||||||
|
|
||||||
|
cdef void new_material(self, list words):
|
||||||
|
material = Material(words[1])
|
||||||
|
self.materials[words[1]] = material
|
||||||
|
self.current_material = material
|
||||||
|
|
||||||
|
cdef void Ka(self, list words):
|
||||||
|
self.current_material.Ka = (float(words[1]), float(words[2]), float(words[3]))
|
||||||
|
|
||||||
|
cdef void Kd(self, list words):
|
||||||
|
self.current_material.Kd = (float(words[1]), float(words[2]), float(words[3]))
|
||||||
|
|
||||||
|
cdef void Ks(self, list words):
|
||||||
|
self.current_material.Ks = (float(words[1]), float(words[2]), float(words[3]))
|
||||||
|
|
||||||
|
cdef void material_shininess(self, list words):
|
||||||
|
self.current_material.shininess = min(float(words[1]), 125)
|
||||||
|
|
||||||
|
cdef void material_texture(self, list words):
|
||||||
|
self.current_material.texture = words[-1]
|
||||||
|
|
||||||
|
@cython.nonecheck(False)
|
||||||
|
cdef void vertex(self, list words):
|
||||||
|
self.vertices.append((float(words[1]), float(words[2]), float(words[3])))
|
||||||
|
|
||||||
|
@cython.nonecheck(False)
|
||||||
|
cdef void normal(self, list words):
|
||||||
|
self.normals.append((float(words[1]), float(words[2]), float(words[3])))
|
||||||
|
|
||||||
|
cdef void texture(self, list words):
|
||||||
|
cdef int l = len(words)
|
||||||
|
cdef object x = 0, y = 0, z = 0
|
||||||
|
if l >= 2:
|
||||||
|
x = float(words[1])
|
||||||
|
if l >= 3:
|
||||||
|
# OBJ origin is at upper left, OpenGL origin is at lower left
|
||||||
|
y = 1 - float(words[2])
|
||||||
|
if l >= 4:
|
||||||
|
z = float(words[3])
|
||||||
|
self.textures.append((x, y, z))
|
||||||
|
|
||||||
|
cdef void face(self, list words):
|
||||||
|
cdef int l = len(words)
|
||||||
|
cdef int type = -1
|
||||||
|
cdef int vertex_count = l - 1
|
||||||
|
cdef list face_vertices = [], face_normals = [], face_textures = []
|
||||||
|
|
||||||
|
if vertex_count == 3:
|
||||||
|
type = FACE_TRIANGLES
|
||||||
|
else:
|
||||||
|
type = FACE_QUADS
|
||||||
|
|
||||||
|
cdef int current_value = -1, texture_len = len(self.textures)
|
||||||
|
cdef list raw_faces, vindices = [], nindices = [], tindices = []
|
||||||
|
|
||||||
|
for i in xrange(1, vertex_count + 1):
|
||||||
|
raw_faces = words[i].split('/')
|
||||||
|
l = len(raw_faces)
|
||||||
|
|
||||||
|
current_value = int(raw_faces[0])
|
||||||
|
vindices.append(current_value - 1)
|
||||||
|
face_vertices.append(self.vertices[current_value - 1])
|
||||||
|
|
||||||
|
if l == 1:
|
||||||
|
continue
|
||||||
|
if l >= 2 and raw_faces[1]:
|
||||||
|
current_value = int(raw_faces[1])
|
||||||
|
if current_value <= texture_len:
|
||||||
|
tindices.append(current_value - 1)
|
||||||
|
face_textures.append(self.textures[current_value - 1])
|
||||||
|
if l >= 3 and raw_faces[2]:
|
||||||
|
current_value = int(raw_faces[2])
|
||||||
|
nindices.append(current_value - 1)
|
||||||
|
face_normals.append(self.normals[current_value - 1])
|
||||||
|
|
||||||
|
cdef Group group
|
||||||
|
if self.current_group is None:
|
||||||
|
self.current_group = group = Group()
|
||||||
|
self.groups.append(group)
|
||||||
|
else:
|
||||||
|
group = self.current_group
|
||||||
|
|
||||||
|
group.vertices += face_vertices
|
||||||
|
group.normals += face_normals
|
||||||
|
group.textures += face_textures
|
||||||
|
idx_count = group.idx_count
|
||||||
|
group.indices += (idx_count + 1, idx_count + 2, idx_count + 3)
|
||||||
|
group.idx_count += 3
|
||||||
|
|
||||||
|
group.faces.append(Face(type, vindices, nindices, tindices, face_vertices, face_normals, face_textures))
|
||||||
|
|
||||||
|
cdef void material(self, list words):
|
||||||
|
self.perform_io(os.path.join(self.root, words[1]))
|
||||||
|
|
||||||
|
cdef void use_material(self, list words):
|
||||||
|
mat = words[1]
|
||||||
|
try:
|
||||||
|
self.current_group.material = self.materials[mat]
|
||||||
|
except KeyError:
|
||||||
|
print "Warning: material %s undefined, only %s defined." % (mat, self.materials)
|
||||||
|
except AttributeError:
|
||||||
|
print "Warning: no group"
|
||||||
|
|
||||||
|
cdef void group(self, list words):
|
||||||
|
name = words[1]
|
||||||
|
group = Group(name)
|
||||||
|
|
||||||
|
if self.groups:
|
||||||
|
self.current_group.pack()
|
||||||
|
self.groups.append(group)
|
||||||
|
self.current_group = group
|
||||||
|
|
||||||
|
cdef inline void perform_io(self, unicode file):
|
||||||
|
mbcsfile = file.encode('mbcs')
|
||||||
|
cdef char* fname = mbcsfile
|
||||||
|
|
||||||
|
cdef FILE* cfile
|
||||||
|
cfile = fopen(fname, 'rb')
|
||||||
|
if cfile == NULL:
|
||||||
|
raise IOError(2, "No such file or directory: '%s'" % file)
|
||||||
|
|
||||||
|
cdef size_t bufsize = 2048
|
||||||
|
cdef char *buf = <char*>malloc(bufsize)
|
||||||
|
cdef ssize_t read
|
||||||
|
cdef char *type
|
||||||
|
cdef list words
|
||||||
|
cdef int hash, length
|
||||||
|
|
||||||
|
while fgets(buf, bufsize, cfile):
|
||||||
|
if buf[0] in (0, 10, 13, 35):
|
||||||
|
continue # Empty or comment
|
||||||
|
words = buf.split()
|
||||||
|
type = words[0]
|
||||||
|
|
||||||
|
length = strlen(type)
|
||||||
|
if not length:
|
||||||
|
continue
|
||||||
|
elif length < 3:
|
||||||
|
hash = type[0] << 8 | type[1]
|
||||||
|
if hash == 0x7600: # v\0
|
||||||
|
self.vertex(words)
|
||||||
|
elif hash == 0x766e: # vn
|
||||||
|
self.normal(words)
|
||||||
|
elif hash == 0x7674: # vt
|
||||||
|
self.texture(words)
|
||||||
|
elif hash == 0x6600: # f
|
||||||
|
self.face(words)
|
||||||
|
elif hash == 0x6700: # g
|
||||||
|
self.group(words)
|
||||||
|
elif hash == 0x6f00: # o
|
||||||
|
self.group(words)
|
||||||
|
elif hash == 0x4b61: # Ka
|
||||||
|
self.Ka(words)
|
||||||
|
elif hash == 0x4b64: # Kd
|
||||||
|
self.Kd(words)
|
||||||
|
elif hash == 0x4b73: # Ks
|
||||||
|
self.Ks(words)
|
||||||
|
elif hash == 0x4e73: # Ns
|
||||||
|
self.material_shininess(words)
|
||||||
|
elif strcmp(type, b'mtllib') == 0:
|
||||||
|
self.material(words)
|
||||||
|
elif strcmp(type, b'usemtl') == 0:
|
||||||
|
self.use_material(words)
|
||||||
|
elif strcmp(type, b'newmtl') == 0:
|
||||||
|
self.new_material(words)
|
||||||
|
elif strcmp(type, b'map_Kd') == 0:
|
||||||
|
self.material_texture(words)
|
||||||
|
free(buf)
|
||||||
|
fclose(cfile)
|
||||||
|
|
||||||
|
import sys
|
||||||
|
if hasattr(sys, 'frozen'):
|
||||||
|
model_base = os.path.dirname(sys.executable)
|
||||||
|
else:
|
||||||
|
model_base = os.path.join(os.path.dirname(__file__), 'assets', 'models')
|
||||||
|
del sys
|
||||||
|
|
||||||
|
def load_model(path):
|
||||||
|
if not os.path.isabs(path):
|
||||||
|
path = os.path.join(model_base, path)
|
||||||
|
if not isinstance(path, unicode):
|
||||||
|
path = path.decode('mbcs')
|
||||||
|
return WavefrontObject(path)
|
||||||
|
|
||||||
|
@cython.nonecheck(False)
|
||||||
|
cdef inline void point(Face f, WavefrontObject m, int tex_id, float sx, float sy, float sz, int n):
|
||||||
|
cdef float x, y, z
|
||||||
|
cdef tuple normal, texture
|
||||||
|
if f.norms:
|
||||||
|
normal = m.normals[f.norms[n]]
|
||||||
|
glNormal3f(normal[0], normal[1], normal[2])
|
||||||
|
if tex_id:
|
||||||
|
texture = m.textures[f.texs[n]]
|
||||||
|
glTexCoord2f(texture[0], texture[1])
|
||||||
|
|
||||||
|
x, y, z = m.vertices[f.verts[n]]
|
||||||
|
glVertex3f(x * sx, y * sy, z * sz)
|
||||||
|
|
||||||
|
cpdef int model_list(WavefrontObject model, float sx=1, float sy=1, float sz=1, object rotation=(0, 0, 0)):
|
||||||
|
for m, text in model.materials.iteritems():
|
||||||
|
if text.texture:
|
||||||
|
load_texture(os.path.join(model.root, text.texture))
|
||||||
|
|
||||||
|
cdef int display = glGenLists(1)
|
||||||
|
|
||||||
|
glNewList(display, GL_COMPILE)
|
||||||
|
glPushMatrix()
|
||||||
|
glPushAttrib(GL_CURRENT_BIT)
|
||||||
|
|
||||||
|
cdef float pitch, yaw, roll
|
||||||
|
cdef float kx, ky, kz
|
||||||
|
|
||||||
|
pitch, yaw, roll = rotation
|
||||||
|
glPushAttrib(GL_TRANSFORM_BIT)
|
||||||
|
glRotatef(pitch, 1, 0, 0)
|
||||||
|
glRotatef(yaw, 0, 1, 0)
|
||||||
|
glRotatef(roll, 0, 0, 1)
|
||||||
|
glPopAttrib()
|
||||||
|
|
||||||
|
cdef Face f
|
||||||
|
cdef Group g
|
||||||
|
cdef int tex_id
|
||||||
|
|
||||||
|
for g in model.groups:
|
||||||
|
tex_id = load_texture(os.path.join(model.root, g.material.texture)) if (g.material and g.material.texture) else 0
|
||||||
|
|
||||||
|
if tex_id:
|
||||||
|
glEnable(GL_TEXTURE_2D)
|
||||||
|
glBindTexture(GL_TEXTURE_2D, tex_id)
|
||||||
|
else:
|
||||||
|
glBindTexture(GL_TEXTURE_2D, 0)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
|
||||||
|
if g.material is not None:
|
||||||
|
if g.material.texture is not None:
|
||||||
|
tex_id = load_texture(os.path.join(model.root, g.material.texture))
|
||||||
|
if g.material.Ka:
|
||||||
|
kx, ky, kz = g.material.Ka
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT, [kx, ky, kz, 1])
|
||||||
|
if g.material.Kd:
|
||||||
|
kx, ky, kz = g.material.Kd
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_DIFFUSE, [kx, ky, kz, 1])
|
||||||
|
if g.material.Ks:
|
||||||
|
kx, ky, kz = g.material.Ks
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, [kx, ky, kz, 1])
|
||||||
|
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, g.material.shininess)
|
||||||
|
|
||||||
|
glBegin(GL_TRIANGLES)
|
||||||
|
for f in g.faces:
|
||||||
|
point(f, model, tex_id, sx, sy, sz, 0)
|
||||||
|
point(f, model, tex_id, sx, sy, sz, 1)
|
||||||
|
point(f, model, tex_id, sx, sy, sz, 2)
|
||||||
|
|
||||||
|
if f.type == FACE_QUADS:
|
||||||
|
point(f, model, tex_id, sx, sy, sz, 2)
|
||||||
|
point(f, model, tex_id, sx, sy, sz, 3)
|
||||||
|
point(f, model, tex_id, sx, sy, sz, 0)
|
||||||
|
glEnd()
|
||||||
|
|
||||||
|
if tex_id:
|
||||||
|
glBindTexture(GL_TEXTURE_2D, 0)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
|
||||||
|
glPopAttrib()
|
||||||
|
glPopMatrix()
|
||||||
|
|
||||||
|
glEndList()
|
||||||
|
return display
|
4619
punyverse/assets/models/asteroids/01.obj
Normal file
18443
punyverse/assets/models/asteroids/02.obj
Normal file
18443
punyverse/assets/models/asteroids/03.obj
Normal file
18443
punyverse/assets/models/asteroids/04.obj
Normal file
4619
punyverse/assets/models/asteroids/05.obj
Normal file
11
punyverse/assets/models/asteroids/asteroid.mtl
Normal file
|
@ -0,0 +1,11 @@
|
||||||
|
# Blender MTL File: 'None'
|
||||||
|
# Material Count: 1
|
||||||
|
|
||||||
|
newmtl Material
|
||||||
|
Ns 96.078431
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.520996 0.473531 0.337164
|
||||||
|
Ks 0.079916 0.069096 0.039636
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
110
punyverse/assets/models/satellites/iss.mtl
Normal file
|
@ -0,0 +1,110 @@
|
||||||
|
# Blender MTL File: 'None'
|
||||||
|
# Material Count: 12
|
||||||
|
|
||||||
|
newmtl Brown
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.238431 0.189074 0.102853
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl ISSDarkMetal
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.070138 0.070138 0.070138
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl ISSGray2
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.241098 0.241098 0.241098
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl ISSMetal1
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.381373 0.381373 0.381373
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl ISSSolarPanels
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.043620 0.023622 0.301177
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl ISSWhiteMetal
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.561093 0.561093 0.561093
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl _Brown
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.238431 0.189074 0.102853
|
||||||
|
Ks 0.000000 0.000000 0.000000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl _ISSDarkMetal
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.070138 0.070138 0.070138
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl _ISSGray2
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.241098 0.241098 0.241098
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl _ISSMetal1
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.381373 0.381373 0.381373
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl _ISSSolarPanels
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.043620 0.023622 0.301177
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
||||||
|
|
||||||
|
newmtl _ISSWhiteMetal
|
||||||
|
Ns 25
|
||||||
|
Ka 0.000000 0.000000 0.000000
|
||||||
|
Kd 0.561093 0.561093 0.561093
|
||||||
|
Ks 0.500000 0.500000 0.500000
|
||||||
|
Ni 1.000000
|
||||||
|
d 1.000000
|
||||||
|
illum 2
|
21459
punyverse/assets/models/satellites/iss.obj
Normal file
BIN
punyverse/assets/models/satellites/phobos.jpg
Normal file
After Width: | Height: | Size: 100 KiB |
20
punyverse/assets/models/satellites/phobos.mtl
Normal file
|
@ -0,0 +1,20 @@
|
||||||
|
# Ns = Phong specular component. Ranges from 0 to 1000. (I've seen various statements about this range (see below))
|
||||||
|
# Kd = Diffuse color weighted by the diffuse coefficient.
|
||||||
|
# Ka = Ambient color weighted by the ambient coefficient.
|
||||||
|
# Ks = Specular color weighted by the specular coefficient.
|
||||||
|
# d = Dissolve factor (pseudo-transparency). Values are from 0-1. 0 is completely transparent, 1 is opaque.
|
||||||
|
# Ni = Refraction index. Values range from 1 upwards. A value of 1 will cause no refraction. A higher value implies refraction.
|
||||||
|
# illum = (0, 1, or 2) 0 to disable lighting, 1 for ambient & diffuse only (specular color set to black), 2 for full lighting (see below)
|
||||||
|
# sharpness = ? (see below)
|
||||||
|
# c
|
||||||
|
# map_Bump = Bump texture map.
|
||||||
|
# map_d = Opacity texture map.
|
||||||
|
# refl = reflection type and filename (?)
|
||||||
|
|
||||||
|
newmtl white
|
||||||
|
Ka 1 1 1
|
||||||
|
Kd 1 1 1
|
||||||
|
Ks 0 0 0
|
||||||
|
map_Kd phobos.jpg
|
||||||
|
illum 2
|
||||||
|
Ns 8
|
5131
punyverse/assets/models/satellites/phobos.obj
Normal file
BIN
punyverse/assets/textures/atmosphere_earth.png
Normal file
After Width: | Height: | Size: 114 B |
BIN
punyverse/assets/textures/cloudmap.png
Normal file
After Width: | Height: | Size: 1.1 MiB |
BIN
punyverse/assets/textures/earth.jpg
Normal file
After Width: | Height: | Size: 1.7 MiB |
BIN
punyverse/assets/textures/jupiter.jpg
Normal file
After Width: | Height: | Size: 482 KiB |
BIN
punyverse/assets/textures/mars.jpg
Normal file
After Width: | Height: | Size: 2 MiB |
BIN
punyverse/assets/textures/moon.jpg
Normal file
After Width: | Height: | Size: 1.3 MiB |
BIN
punyverse/assets/textures/moons/callisto.jpg
Normal file
After Width: | Height: | Size: 364 KiB |
BIN
punyverse/assets/textures/moons/dione.jpg
Normal file
After Width: | Height: | Size: 236 KiB |
BIN
punyverse/assets/textures/moons/enceladus.jpg
Normal file
After Width: | Height: | Size: 206 KiB |
BIN
punyverse/assets/textures/moons/europa.jpg
Normal file
After Width: | Height: | Size: 244 KiB |
BIN
punyverse/assets/textures/moons/ganymede.jpg
Normal file
After Width: | Height: | Size: 284 KiB |
BIN
punyverse/assets/textures/moons/iapetus.jpg
Normal file
After Width: | Height: | Size: 175 KiB |
BIN
punyverse/assets/textures/moons/io.jpg
Normal file
After Width: | Height: | Size: 268 KiB |
BIN
punyverse/assets/textures/moons/mimas.jpg
Normal file
After Width: | Height: | Size: 278 KiB |
BIN
punyverse/assets/textures/moons/rhea.jpg
Normal file
After Width: | Height: | Size: 303 KiB |
BIN
punyverse/assets/textures/moons/tethys.jpg
Normal file
After Width: | Height: | Size: 281 KiB |
BIN
punyverse/assets/textures/moons/titan.jpg
Normal file
After Width: | Height: | Size: 210 KiB |
BIN
punyverse/assets/textures/neptune.jpg
Normal file
After Width: | Height: | Size: 9.6 KiB |
BIN
punyverse/assets/textures/ring_saturn.png
Normal file
After Width: | Height: | Size: 3 KiB |
BIN
punyverse/assets/textures/ring_uranus.png
Normal file
After Width: | Height: | Size: 1.6 KiB |
BIN
punyverse/assets/textures/saturn.jpg
Normal file
After Width: | Height: | Size: 133 KiB |
BIN
punyverse/assets/textures/sky.jpg
Normal file
After Width: | Height: | Size: 7.2 MiB |
BIN
punyverse/assets/textures/uranus.jpg
Normal file
After Width: | Height: | Size: 5.9 KiB |
BIN
punyverse/assets/textures/venus.jpg
Normal file
After Width: | Height: | Size: 176 KiB |
41
punyverse/camera.py
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
from math import sin, cos, radians
|
||||||
|
|
||||||
|
|
||||||
|
class Camera(object):
|
||||||
|
def __init__(self, x=0, y=0, z=0, pitch=0, yaw=0, roll=0):
|
||||||
|
self.x = x
|
||||||
|
self.y = y
|
||||||
|
self.z = z
|
||||||
|
self.pitch = pitch
|
||||||
|
self.yaw = yaw
|
||||||
|
self.roll = roll
|
||||||
|
|
||||||
|
def move(self, speed):
|
||||||
|
dx, dy, dz = self.direction()
|
||||||
|
self.x += dx * speed
|
||||||
|
self.y += dy * speed
|
||||||
|
self.z += dz * speed
|
||||||
|
|
||||||
|
def mouse_move(self, dx, dy):
|
||||||
|
if self.pitch > 90 or self.pitch < -90:
|
||||||
|
dx = -dx
|
||||||
|
if self.yaw + dx >= 360:
|
||||||
|
self.yaw = self.yaw + dx - 360
|
||||||
|
elif self.yaw + dx < 0:
|
||||||
|
self.yaw = 360 - self.yaw + dx
|
||||||
|
else:
|
||||||
|
self.yaw += dx
|
||||||
|
|
||||||
|
self.pitch -= dy
|
||||||
|
if self.pitch < -180:
|
||||||
|
self.pitch += 360
|
||||||
|
elif self.pitch > 180:
|
||||||
|
self.pitch -= 360
|
||||||
|
|
||||||
|
def direction(self):
|
||||||
|
m = cos(radians(self.pitch))
|
||||||
|
|
||||||
|
dy = -sin(radians(self.pitch))
|
||||||
|
dx = cos(radians(self.yaw - 90)) * m
|
||||||
|
dz = sin(radians(self.yaw - 90)) * m
|
||||||
|
return dx, dy, dz
|
115
punyverse/entity.py
Normal file
|
@ -0,0 +1,115 @@
|
||||||
|
from punyverse.orbit import KeplerOrbit
|
||||||
|
|
||||||
|
from pyglet.gl import *
|
||||||
|
|
||||||
|
|
||||||
|
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 Body(Entity):
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
self.rotation_angle = kwargs.pop('rotation_angle', 5)
|
||||||
|
self.atmosphere = kwargs.pop('atmosphere', 0)
|
||||||
|
self.cloudmap = kwargs.pop('cloudmap', 0)
|
||||||
|
self.last_tick = 0
|
||||||
|
self.mass = kwargs.pop('mass', None)
|
||||||
|
self.world = kwargs.pop('world')
|
||||||
|
orbit_distance = kwargs.pop('orbit_distance', 40000) + .0
|
||||||
|
self.orbit_show = orbit_distance * 1.25
|
||||||
|
self.orbit_blend = orbit_distance / 4
|
||||||
|
self.orbit_opaque = orbit_distance
|
||||||
|
super(Body, self).__init__(*args, **kwargs)
|
||||||
|
self.initial_roll = self.rotation[2]
|
||||||
|
|
||||||
|
def update(self):
|
||||||
|
super(Body, self).update()
|
||||||
|
|
||||||
|
if self.last_tick != self.world.tick:
|
||||||
|
self.last_tick = self.world.tick
|
||||||
|
pitch, yaw, roll = self.rotation
|
||||||
|
roll = (self.initial_roll + self.world.tick * self.rotation_angle) % 360
|
||||||
|
self.rotation = pitch, yaw, roll
|
||||||
|
|
||||||
|
|
||||||
|
class Satellite(Body):
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
self.parent = kwargs.pop('parent')
|
||||||
|
self.orbit_speed = kwargs.pop('orbit_speed', 1)
|
||||||
|
|
||||||
|
# Semi-major axis and eccentricity defines orbit
|
||||||
|
distance = kwargs.pop('distance', 100)
|
||||||
|
eccentricity = kwargs.pop('eccentricity', 0)
|
||||||
|
|
||||||
|
# Inclination, longitude of ascending node, and argument of periapsis defines orbital plane
|
||||||
|
inclination = kwargs.pop('inclination', 0)
|
||||||
|
longitude = kwargs.pop('longitude', 0)
|
||||||
|
argument = kwargs.pop('argument', 0)
|
||||||
|
|
||||||
|
# Orbit calculation
|
||||||
|
self.orbit_id = None
|
||||||
|
self.orbit_cache = None
|
||||||
|
|
||||||
|
self.theta = 0
|
||||||
|
# OpenGL's z-axis is reversed
|
||||||
|
self.orbit = KeplerOrbit(distance, eccentricity, inclination, longitude, argument)
|
||||||
|
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
|
||||||
|
|
||||||
|
if self.orbit_id is not None:
|
||||||
|
glDeleteLists(self.orbit_id, 1)
|
||||||
|
|
||||||
|
id = glGenLists(1)
|
||||||
|
glNewList(id, GL_COMPILE)
|
||||||
|
glBegin(GL_LINE_LOOP)
|
||||||
|
for theta in xrange(360):
|
||||||
|
x, z, y = self.orbit.orbit(theta)
|
||||||
|
glVertex3f(x, y, z)
|
||||||
|
glEnd()
|
||||||
|
glEndList()
|
||||||
|
|
||||||
|
self.orbit_id = id
|
||||||
|
self.orbit_cache = cache
|
||||||
|
return id
|
||||||
|
|
||||||
|
def update(self):
|
||||||
|
super(Body, self).update() # Notice how the parent class is skipped
|
||||||
|
|
||||||
|
if self.last_tick != self.world.tick:
|
||||||
|
self.last_tick = self.world.tick
|
||||||
|
pitch, yaw, roll = self.rotation
|
||||||
|
roll = (self.initial_roll + self.world.tick * self.rotation_angle) % 360
|
||||||
|
self.rotation = pitch, yaw, roll
|
||||||
|
|
||||||
|
self.parent.update()
|
||||||
|
px, py, pz = self.parent.location
|
||||||
|
self.theta = self.world.tick * self.orbit_speed % 360
|
||||||
|
x, z, y = self.orbit.orbit(self.theta)
|
||||||
|
self.location = (x + px, y + py, z + pz)
|
||||||
|
|
354
punyverse/game.py
Normal file
|
@ -0,0 +1,354 @@
|
||||||
|
#!/usr/bin/python
|
||||||
|
from operator import attrgetter
|
||||||
|
from math import hypot, sqrt, atan2, degrees
|
||||||
|
from time import clock
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
|
||||||
|
from punyverse.camera import Camera
|
||||||
|
from punyverse.world import load_world
|
||||||
|
from punyverse.glgeom import *
|
||||||
|
from punyverse.entity import Asteroid
|
||||||
|
from punyverse import texture
|
||||||
|
|
||||||
|
try:
|
||||||
|
from punyverse._model import model_list, load_model
|
||||||
|
except ImportError:
|
||||||
|
from punyverse.model import model_list, load_model
|
||||||
|
|
||||||
|
from pyglet.gl import *
|
||||||
|
from pyglet.window import key
|
||||||
|
import pyglet
|
||||||
|
|
||||||
|
|
||||||
|
INITIAL_SPEED = 0 # The initial speed of the player
|
||||||
|
TICKS_PER_SECOND = 20 # How many times to update game per second
|
||||||
|
MOUSE_SENSITIVITY = 0.3 # Mouse sensitivity, 0..1, none...hyperspeed
|
||||||
|
|
||||||
|
MAX_DELTA = 5
|
||||||
|
SEED = int(time.time())
|
||||||
|
|
||||||
|
|
||||||
|
def entity_distance(x0, y0, z0):
|
||||||
|
def distance(entity):
|
||||||
|
x1, y1, z1 = entity.location
|
||||||
|
return hypot(hypot(x1 - x0, y1 - y0), z1 - z0)
|
||||||
|
|
||||||
|
return distance
|
||||||
|
|
||||||
|
|
||||||
|
class Applet(pyglet.window.Window):
|
||||||
|
def update(self, dt):
|
||||||
|
cam = self.cam
|
||||||
|
if self.exclusive:
|
||||||
|
if key.A in self.keys:
|
||||||
|
cam.roll += 4
|
||||||
|
if key.S in self.keys:
|
||||||
|
cam.roll -= 4
|
||||||
|
cam.move(self.speed)
|
||||||
|
|
||||||
|
if self.running:
|
||||||
|
self.world.tick += self.tick
|
||||||
|
for entity in self.world.tracker:
|
||||||
|
entity.update()
|
||||||
|
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super(Applet, self).__init__(*args, **kwargs)
|
||||||
|
l = clock()
|
||||||
|
self.fps = 0
|
||||||
|
self.world = load_world("world.json")
|
||||||
|
self.speed = INITIAL_SPEED
|
||||||
|
self.keys = set()
|
||||||
|
self.info = True
|
||||||
|
self.debug = False
|
||||||
|
self.orbit = True
|
||||||
|
self.running = True
|
||||||
|
|
||||||
|
self.tick = self.world.tick_length
|
||||||
|
# On standard world: 10x is one day per second, 100x is 10 days, 300x is a month
|
||||||
|
# 900x is a quarter, 1825 is a half year, 3650 is a year, 36500 is a decade, 365000 is a century
|
||||||
|
# and yes the individual hours and seconds look ugly
|
||||||
|
self.ticks = [20, 40, 60, # Second range
|
||||||
|
120, 300, 600, 1200, 1800, 2700, 3600, # Minute range
|
||||||
|
7200, 14400, 21600, 43200, 86400, # Hour range
|
||||||
|
172800, 432000, 604800, # 2, 5, 7 days
|
||||||
|
1209600, 2592000, # 2 week, 1 month
|
||||||
|
5270400, 7884000, 15768000, 31536000, # 2, 3, 6, 12 months
|
||||||
|
63072000, 157680000, 315360000, # 2, 5, 10 years
|
||||||
|
630720000, 1576800000, 3153600000, # 20, 50, 100 years
|
||||||
|
]
|
||||||
|
self.ticks = [i / 20 for i in self.ticks]
|
||||||
|
self.__time_per_second_cache = None
|
||||||
|
self.__time_per_second_value = None
|
||||||
|
|
||||||
|
self.label = pyglet.text.Label('', font_name='Consolas', font_size=12, x=10, y=self.height - 20,
|
||||||
|
color=(255,) * 4, multiline=True, width=600)
|
||||||
|
self.cam = Camera()
|
||||||
|
|
||||||
|
self.exclusive = False
|
||||||
|
pyglet.clock.schedule_interval(self.update, 1.0 / TICKS_PER_SECOND)
|
||||||
|
|
||||||
|
def update_fps(dt):
|
||||||
|
self.fps = pyglet.clock.get_fps()
|
||||||
|
|
||||||
|
pyglet.clock.schedule_interval(update_fps, 1)
|
||||||
|
|
||||||
|
glClearColor(0, 0, 0, 1)
|
||||||
|
glClearDepth(1.0)
|
||||||
|
|
||||||
|
texture.init()
|
||||||
|
if not texture.badcard:
|
||||||
|
glEnable(GL_BLEND)
|
||||||
|
glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
|
||||||
|
glEnable(GL_LINE_SMOOTH)
|
||||||
|
glEnable(GL_POLYGON_SMOOTH)
|
||||||
|
glHint(GL_LINE_SMOOTH_HINT, GL_NICEST)
|
||||||
|
glHint(GL_POLYGON_SMOOTH_HINT, GL_NICEST)
|
||||||
|
|
||||||
|
glAlphaFunc(GL_GEQUAL, 0.9)
|
||||||
|
glDepthFunc(GL_LEQUAL)
|
||||||
|
glEnable(GL_DEPTH_TEST)
|
||||||
|
glShadeModel(GL_SMOOTH)
|
||||||
|
|
||||||
|
glMatrixMode(GL_MODELVIEW)
|
||||||
|
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glEnable(GL_LIGHT0)
|
||||||
|
glEnable(GL_LIGHT1)
|
||||||
|
|
||||||
|
glEnable(GL_POLYGON_OFFSET_FILL)
|
||||||
|
|
||||||
|
fv4 = GLfloat * 4
|
||||||
|
|
||||||
|
glLightfv(GL_LIGHT0, GL_POSITION, fv4(.5, .5, 1, 0))
|
||||||
|
glLightfv(GL_LIGHT0, GL_SPECULAR, fv4(.5, .5, 1, 1))
|
||||||
|
glLightfv(GL_LIGHT0, GL_DIFFUSE, fv4(1, 1, 1, 1))
|
||||||
|
glLightfv(GL_LIGHT1, GL_POSITION, fv4(1, 0, .5, 0))
|
||||||
|
glLightfv(GL_LIGHT1, GL_DIFFUSE, fv4(.5, .5, .5, 1))
|
||||||
|
glLightfv(GL_LIGHT1, GL_SPECULAR, fv4(1, 1, 1, 1))
|
||||||
|
|
||||||
|
self.asteroid_ids = [model_list(load_model(r"asteroids\01.obj"), 5, 5, 5, (0, 0, 0)),
|
||||||
|
model_list(load_model(r"asteroids\02.obj"), 5, 5, 5, (0, 0, 0)),
|
||||||
|
model_list(load_model(r"asteroids\03.obj"), 5, 5, 5, (0, 0, 0)),
|
||||||
|
model_list(load_model(r"asteroids\04.obj"), 5, 5, 5, (0, 0, 0)),
|
||||||
|
model_list(load_model(r"asteroids\05.obj"), 5, 5, 5, (0, 0, 0))]
|
||||||
|
|
||||||
|
c = self.cam
|
||||||
|
c.x, c.y, c.z = self.world.start
|
||||||
|
c.pitch, c.yaw, c.roll = self.world.direction
|
||||||
|
|
||||||
|
for entity in self.world.tracker:
|
||||||
|
entity.update()
|
||||||
|
|
||||||
|
print "Loaded in %s seconds." % (clock() - l)
|
||||||
|
|
||||||
|
def set_exclusive_mouse(self, exclusive):
|
||||||
|
super(Applet, self).set_exclusive_mouse(exclusive) # Pass to parent
|
||||||
|
self.exclusive = exclusive # Toggle flag
|
||||||
|
|
||||||
|
def on_mouse_press(self, x, y, button, modifiers):
|
||||||
|
if not self.exclusive:
|
||||||
|
self.set_exclusive_mouse(True)
|
||||||
|
|
||||||
|
def on_mouse_motion(self, x, y, dx, dy):
|
||||||
|
if self.exclusive: # Only handle camera movement if mouse is grabbed
|
||||||
|
self.cam.mouse_move(dx * MOUSE_SENSITIVITY, dy * MOUSE_SENSITIVITY)
|
||||||
|
|
||||||
|
def on_key_press(self, symbol, modifiers):
|
||||||
|
if self.exclusive: # Only handle keyboard input if mouse is grabbed
|
||||||
|
if symbol == key.ESCAPE:
|
||||||
|
pyglet.app.exit()
|
||||||
|
elif symbol == key.E:
|
||||||
|
self.set_exclusive_mouse(False) # Escape mouse
|
||||||
|
elif symbol == key.F:
|
||||||
|
self.set_fullscreen(not self.fullscreen)
|
||||||
|
elif symbol == key.NUM_ADD:
|
||||||
|
self.speed += 1
|
||||||
|
elif symbol == key.NUM_SUBTRACT:
|
||||||
|
self.speed -= 1
|
||||||
|
elif symbol == key.NUM_MULTIPLY:
|
||||||
|
self.speed += 10
|
||||||
|
elif symbol == key.NUM_DIVIDE:
|
||||||
|
self.speed -= 10
|
||||||
|
elif symbol == key.PAGEUP:
|
||||||
|
self.speed += 100
|
||||||
|
elif symbol == key.PAGEDOWN:
|
||||||
|
self.speed -= 100
|
||||||
|
elif symbol == key.HOME:
|
||||||
|
self.speed += 1000
|
||||||
|
elif symbol == key.END:
|
||||||
|
self.speed -= 1000
|
||||||
|
elif symbol == key.I:
|
||||||
|
self.info = not self.info
|
||||||
|
elif symbol == key.D:
|
||||||
|
self.debug = not self.debug
|
||||||
|
elif symbol == key.O:
|
||||||
|
self.orbit = not self.orbit
|
||||||
|
elif symbol == key.ENTER:
|
||||||
|
self.running = not self.running
|
||||||
|
elif symbol == key.INSERT:
|
||||||
|
index = self.ticks.index(self.tick) + 1
|
||||||
|
if index < len(self.ticks):
|
||||||
|
self.tick = self.ticks[index]
|
||||||
|
elif symbol == key.DELETE:
|
||||||
|
index = self.ticks.index(self.tick) - 1
|
||||||
|
if index >= 0:
|
||||||
|
self.tick = self.ticks[index]
|
||||||
|
elif symbol == key.SPACE:
|
||||||
|
c = self.cam
|
||||||
|
dx, dy, dz = c.direction()
|
||||||
|
speed = self.speed * 1.1 + 5
|
||||||
|
dx *= speed
|
||||||
|
dy *= speed
|
||||||
|
dz *= speed
|
||||||
|
self.world.tracker.append(
|
||||||
|
Asteroid(random.choice(self.asteroid_ids), (c.x, c.y - 3, c.z + 5), direction=(dx, dy, dz)))
|
||||||
|
else:
|
||||||
|
self.keys.add(symbol)
|
||||||
|
|
||||||
|
def on_key_release(self, symbol, modifiers):
|
||||||
|
if symbol in self.keys:
|
||||||
|
self.keys.remove(symbol)
|
||||||
|
|
||||||
|
def on_resize(self, width, height):
|
||||||
|
height = max(height, 1) # Prevent / by 0
|
||||||
|
self.label.y = height - 20
|
||||||
|
glViewport(0, 0, width, height)
|
||||||
|
glMatrixMode(GL_PROJECTION)
|
||||||
|
glLoadIdentity()
|
||||||
|
# A field of view of 45
|
||||||
|
gluPerspective(45.0, width / float(height), 0.1, 50000000.0)
|
||||||
|
glMatrixMode(GL_MODELVIEW)
|
||||||
|
|
||||||
|
def get_time_per_second(self):
|
||||||
|
if self.__time_per_second_cache == self.tick:
|
||||||
|
return self.__time_per_second_value
|
||||||
|
time = self.tick * TICKS_PER_SECOND + .0
|
||||||
|
unit = 'seconds'
|
||||||
|
for size, name in ((60, 'minutes'), (60, 'hours'), (24, 'days'), (365, 'years')):
|
||||||
|
if time < size:
|
||||||
|
break
|
||||||
|
time /= size
|
||||||
|
unit = name
|
||||||
|
result = '%s %s' % (round(time, 1), unit)
|
||||||
|
self.__time_per_second_cache = self.tick
|
||||||
|
self.__time_per_second_value = result
|
||||||
|
return result
|
||||||
|
|
||||||
|
def on_draw(self):
|
||||||
|
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
|
||||||
|
glLoadIdentity()
|
||||||
|
|
||||||
|
c = self.cam
|
||||||
|
x, y, z = c.x, c.y, c.z
|
||||||
|
glRotatef(c.pitch, 1, 0, 0)
|
||||||
|
glRotatef(c.yaw, 0, 1, 0)
|
||||||
|
glRotatef(c.roll, 0, 0, 1)
|
||||||
|
glTranslatef(-x, -y, -z)
|
||||||
|
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glEnable(GL_BLEND)
|
||||||
|
world = self.world
|
||||||
|
get_distance = entity_distance(x, y, z)
|
||||||
|
if x != world.x or y != world.y or z != world.z:
|
||||||
|
world.tracker.sort(key=get_distance, reverse=True)
|
||||||
|
world.tracker.sort(key=attrgetter('background'), reverse=True)
|
||||||
|
world.x, world.y, world.z = x, y, z
|
||||||
|
|
||||||
|
for entity in world.tracker:
|
||||||
|
x, y, z = entity.location
|
||||||
|
pitch, yaw, roll = entity.rotation
|
||||||
|
|
||||||
|
glPushMatrix()
|
||||||
|
glTranslatef(x, y, z)
|
||||||
|
glRotatef(pitch, 1, 0, 0)
|
||||||
|
glRotatef(yaw, 0, 1, 0)
|
||||||
|
glRotatef(roll, 0, 0, 1)
|
||||||
|
glPushAttrib(GL_CURRENT_BIT)
|
||||||
|
glCallList(entity.id)
|
||||||
|
if self.debug:
|
||||||
|
glPushMatrix()
|
||||||
|
glLineWidth(0.25)
|
||||||
|
glPolygonOffset(1, 1)
|
||||||
|
glDisable(GL_LIGHTING)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
glColor3f(0, 1, 0)
|
||||||
|
glPolygonMode(GL_FRONT_AND_BACK, GL_LINE)
|
||||||
|
glCallList(entity.id)
|
||||||
|
glPolygonMode(GL_FRONT_AND_BACK, GL_FILL)
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glEnable(GL_TEXTURE_2D)
|
||||||
|
glPopMatrix()
|
||||||
|
glPopAttrib()
|
||||||
|
glPopMatrix()
|
||||||
|
|
||||||
|
if hasattr(entity, 'atmosphere') and entity.atmosphere:
|
||||||
|
glPushMatrix()
|
||||||
|
x0, y0, z0 = entity.location
|
||||||
|
dx, dy, dz = x - x0, y - y0, z - z0
|
||||||
|
|
||||||
|
distance = sqrt(dz * dz + dx * dx)
|
||||||
|
pitch = (360 - degrees(atan2(dy, distance)))
|
||||||
|
yaw = degrees(atan2(dx, dz))
|
||||||
|
|
||||||
|
glTranslatef(x0, y0, z0)
|
||||||
|
glRotatef(pitch, 1, 0, 0)
|
||||||
|
glRotatef(yaw, 0, 1, 0)
|
||||||
|
glCallList(entity.atmosphere)
|
||||||
|
glPopMatrix()
|
||||||
|
|
||||||
|
if hasattr(entity, "cloudmap") and entity.cloudmap:
|
||||||
|
glPushMatrix()
|
||||||
|
glEnable(GL_ALPHA_TEST)
|
||||||
|
glTranslatef(*entity.location)
|
||||||
|
pitch, yaw, roll = entity.rotation
|
||||||
|
glRotatef(pitch, 1, 0, 0)
|
||||||
|
glRotatef(yaw, 0, 1, 0)
|
||||||
|
glRotatef(roll, 0, 0, 1)
|
||||||
|
glCallList(entity.cloudmap)
|
||||||
|
glDisable(GL_ALPHA_TEST)
|
||||||
|
glPopMatrix()
|
||||||
|
|
||||||
|
if self.orbit and hasattr(entity, 'get_orbit') and hasattr(entity, 'parent'):
|
||||||
|
parent = entity.parent
|
||||||
|
distance = get_distance(parent)
|
||||||
|
if distance < parent.orbit_show:
|
||||||
|
glPushMatrix()
|
||||||
|
glTranslatef(*entity.parent.location)
|
||||||
|
glDisable(GL_LIGHTING)
|
||||||
|
glPushAttrib(GL_LINE_BIT | GL_CURRENT_BIT)
|
||||||
|
glColor4f(1, 1, 1, 1 if distance < parent.orbit_opaque else
|
||||||
|
(1 - (distance - parent.orbit_opaque) / parent.orbit_blend))
|
||||||
|
glLineWidth(1)
|
||||||
|
glCallList(entity.get_orbit())
|
||||||
|
glPopAttrib()
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glPopMatrix()
|
||||||
|
|
||||||
|
glColor4f(1, 1, 1, 1)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
|
||||||
|
width, height = self.get_size()
|
||||||
|
|
||||||
|
if self.info:
|
||||||
|
ortho(width, height)
|
||||||
|
|
||||||
|
self.label.text = ('%d FPS @ (x=%.2f, y=%.2f, z=%.2f) @ %s, %s/s\n'
|
||||||
|
'Direction(pitch=%.2f, yaw=%.2f, roll=%.2f)\nTick: %d' %
|
||||||
|
(self.fps, c.x, c.y, c.z, self.speed, self.get_time_per_second(),
|
||||||
|
c.pitch, c.yaw, c.roll, self.world.tick))
|
||||||
|
self.label.draw()
|
||||||
|
|
||||||
|
glPushAttrib(GL_CURRENT_BIT | GL_LINE_BIT)
|
||||||
|
|
||||||
|
glLineWidth(2)
|
||||||
|
|
||||||
|
cx, cy = width / 2, height / 2
|
||||||
|
|
||||||
|
glColor3f(0, 0, 1)
|
||||||
|
crosshair(15, (cx, cy))
|
||||||
|
glColor4f(0, 1, 0, 1)
|
||||||
|
circle(20, 30, (cx, cy))
|
||||||
|
glPopAttrib()
|
||||||
|
|
||||||
|
frustrum()
|
28
punyverse/gencheap.bat
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
@echo off
|
||||||
|
cd %~dp0assets\textures
|
||||||
|
call :convert earth.jpg earth_medium.jpg 2048x1024
|
||||||
|
call :convert earth.jpg earth_small.jpg 1024x512
|
||||||
|
call :convert moon.jpg moon_medium.jpg 2048x1024
|
||||||
|
call :convert moon.jpg moon_small.jpg 1024x512
|
||||||
|
call :convert mars.jpg mars_medium.jpg 2048x1024
|
||||||
|
call :convert mars.jpg mars_small.jpg 1024x512
|
||||||
|
call :convert jupiter.jpg jupiter_medium.jpg 2048x1024
|
||||||
|
call :convert jupiter.jpg jupiter_small.jpg 1024x512
|
||||||
|
call :convert saturn.jpg saturn_medium.jpg 2048x1024
|
||||||
|
call :convert saturn.jpg saturn_small.jpg 1024x512
|
||||||
|
call :convert moons\io.jpg moons\io_small.jpg 1024x512
|
||||||
|
call :convert moons\europa.jpg moons\europa_small.jpg 1024x512
|
||||||
|
call :convert moons\ganymede.jpg moons\ganymede_small.jpg 1024x512
|
||||||
|
call :convert moons\callisto.jpg moons\callisto_small.jpg 1024x512
|
||||||
|
call :convert moons\titan.jpg moons\titan_small.jpg 1024x512
|
||||||
|
call :convert moons\rhea.jpg moons\rhea_small.jpg 1024x512
|
||||||
|
call :convert moons\iapetus.jpg moons\iapetus_small.jpg 1024x512
|
||||||
|
call :convert moons\dione.jpg moons\dione_small.jpg 1024x512
|
||||||
|
call :convert moons\tethys.jpg moons\tethys_small.jpg 1024x512
|
||||||
|
call :convert moons\enceladus.jpg moons\enceladus_small.jpg 1024x512
|
||||||
|
call :convert moons\mimas.jpg moons\mimas_small.jpg 1024x512
|
||||||
|
goto :eof
|
||||||
|
|
||||||
|
:convert
|
||||||
|
echo Converting %1 to %2, size %3...
|
||||||
|
if not exist %2 gm convert %1 -resize %3 %2
|
167
punyverse/glgeom.py
Normal file
|
@ -0,0 +1,167 @@
|
||||||
|
from math import *
|
||||||
|
from pyglet.gl import *
|
||||||
|
|
||||||
|
TWOPI = pi * 2
|
||||||
|
|
||||||
|
__all__ = ['compile', 'ortho', 'frustrum', 'crosshair', 'circle', 'disk', 'sphere', 'colourball', 'torus']
|
||||||
|
|
||||||
|
|
||||||
|
def compile(pointer, *args, **kwargs):
|
||||||
|
display = glGenLists(1)
|
||||||
|
glNewList(display, GL_COMPILE)
|
||||||
|
pointer(*args, **kwargs)
|
||||||
|
glEndList()
|
||||||
|
return display
|
||||||
|
|
||||||
|
|
||||||
|
def ortho(width, height):
|
||||||
|
glDisable(GL_LIGHTING)
|
||||||
|
glDisable(GL_DEPTH_TEST)
|
||||||
|
glMatrixMode(GL_PROJECTION)
|
||||||
|
glPushMatrix()
|
||||||
|
glLoadIdentity()
|
||||||
|
glOrtho(0, width, 0, height, -1, 1)
|
||||||
|
glMatrixMode(GL_MODELVIEW)
|
||||||
|
glPushMatrix()
|
||||||
|
glLoadIdentity()
|
||||||
|
|
||||||
|
|
||||||
|
def frustrum():
|
||||||
|
glMatrixMode(GL_PROJECTION)
|
||||||
|
glPopMatrix()
|
||||||
|
glMatrixMode(GL_MODELVIEW)
|
||||||
|
glPopMatrix()
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glEnable(GL_DEPTH_TEST)
|
||||||
|
|
||||||
|
|
||||||
|
def crosshair(size, (cx, cy)):
|
||||||
|
glBegin(GL_LINES)
|
||||||
|
glVertex2f(cx - size, cy)
|
||||||
|
glVertex2f(cx + size, cy)
|
||||||
|
glVertex2f(cx, cy - size)
|
||||||
|
glVertex2f(cx, cy + size)
|
||||||
|
glEnd()
|
||||||
|
|
||||||
|
|
||||||
|
def circle(r, seg, (cx, cy)):
|
||||||
|
glBegin(GL_LINE_LOOP)
|
||||||
|
for i in xrange(seg):
|
||||||
|
theta = TWOPI * i / seg
|
||||||
|
glVertex2f(cx + cos(theta) * r, cy + sin(theta) * r)
|
||||||
|
glEnd()
|
||||||
|
|
||||||
|
|
||||||
|
def disk(rinner, router, segs, tex):
|
||||||
|
glEnable(GL_TEXTURE_2D)
|
||||||
|
glDisable(GL_LIGHTING)
|
||||||
|
glBindTexture(GL_TEXTURE_2D, tex)
|
||||||
|
res = segs * 5
|
||||||
|
|
||||||
|
glBegin(GL_TRIANGLE_STRIP)
|
||||||
|
texture = 0
|
||||||
|
factor = TWOPI / res
|
||||||
|
theta = 0
|
||||||
|
for n in xrange(res + 1):
|
||||||
|
theta += factor
|
||||||
|
x = cos(theta)
|
||||||
|
y = sin(theta)
|
||||||
|
glTexCoord2f(0, texture)
|
||||||
|
glVertex2f(rinner * x, rinner * y)
|
||||||
|
glTexCoord2f(1, texture)
|
||||||
|
glVertex2f(router * x, router * y)
|
||||||
|
texture ^= 1
|
||||||
|
glEnd()
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
|
||||||
|
|
||||||
|
def sphere(r, lats, longs, tex, lighting=True, fv4=GLfloat * 4):
|
||||||
|
"""
|
||||||
|
Sphere function from the OpenGL red book.
|
||||||
|
"""
|
||||||
|
sphere = gluNewQuadric()
|
||||||
|
gluQuadricDrawStyle(sphere, GLU_FILL)
|
||||||
|
gluQuadricTexture(sphere, True)
|
||||||
|
if lighting:
|
||||||
|
gluQuadricNormals(sphere, GLU_SMOOTH)
|
||||||
|
|
||||||
|
glEnable(GL_TEXTURE_2D)
|
||||||
|
if lighting:
|
||||||
|
glDisable(GL_BLEND)
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, fv4(1, 1, 1, 0))
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, fv4(1, 1, 1, 0))
|
||||||
|
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, 125)
|
||||||
|
else:
|
||||||
|
glDisable(GL_LIGHTING)
|
||||||
|
glBindTexture(GL_TEXTURE_2D, tex)
|
||||||
|
|
||||||
|
gluSphere(sphere, r, lats, longs)
|
||||||
|
|
||||||
|
glBindTexture(GL_TEXTURE_2D, 0)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
glEnable(GL_LIGHTING)
|
||||||
|
glEnable(GL_BLEND)
|
||||||
|
gluDeleteQuadric(sphere)
|
||||||
|
|
||||||
|
|
||||||
|
def colourball(r, lats, longs, colour, fv4=GLfloat * 4):
|
||||||
|
"""
|
||||||
|
Sphere function from the OpenGL red book.
|
||||||
|
"""
|
||||||
|
sphere = gluNewQuadric()
|
||||||
|
|
||||||
|
glDisable(GL_BLEND)
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, fv4(*colour))
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, fv4(1, 1, 1, 1))
|
||||||
|
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, 125)
|
||||||
|
|
||||||
|
gluSphere(sphere, r, lats, longs)
|
||||||
|
|
||||||
|
glEnable(GL_BLEND)
|
||||||
|
gluDeleteQuadric(sphere)
|
||||||
|
|
||||||
|
|
||||||
|
try:
|
||||||
|
from _glgeom import torus
|
||||||
|
except ImportError:
|
||||||
|
def torus(major_radius, minor_radius, n_major, n_minor, material, shininess=125, fv4=GLfloat * 4):
|
||||||
|
"""
|
||||||
|
Torus function from the OpenGL red book.
|
||||||
|
"""
|
||||||
|
glPushAttrib(GL_CURRENT_BIT)
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, fv4(*material))
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, fv4(1, 1, 1, 1))
|
||||||
|
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, shininess)
|
||||||
|
|
||||||
|
major_s = TWOPI / n_major
|
||||||
|
minor_s = TWOPI / n_minor
|
||||||
|
|
||||||
|
def n(x, y, z):
|
||||||
|
m = 1.0 / sqrt(x * x + y * y + z * z)
|
||||||
|
return x * m, y * m, z * m
|
||||||
|
|
||||||
|
for i in xrange(n_major):
|
||||||
|
a0 = i * major_s
|
||||||
|
a1 = a0 + major_s
|
||||||
|
x0 = cos(a0)
|
||||||
|
y0 = sin(a0)
|
||||||
|
x1 = cos(a1)
|
||||||
|
y1 = sin(a1)
|
||||||
|
|
||||||
|
glBegin(GL_TRIANGLE_STRIP)
|
||||||
|
|
||||||
|
for j in xrange(n_minor + 1):
|
||||||
|
b = j * minor_s
|
||||||
|
c = cos(b)
|
||||||
|
r = minor_radius * c + major_radius
|
||||||
|
z = minor_radius * sin(b)
|
||||||
|
|
||||||
|
glNormal3f(*n(x0 * c, y0 * c, z / minor_radius))
|
||||||
|
glVertex3f(x0 * r, y0 * r, z)
|
||||||
|
|
||||||
|
glNormal3f(*n(x1 * c, y1 * c, z / minor_radius))
|
||||||
|
glVertex3f(x1 * r, y1 * r, z)
|
||||||
|
|
||||||
|
glEnd()
|
||||||
|
glPopAttrib()
|
288
punyverse/model.py
Normal file
|
@ -0,0 +1,288 @@
|
||||||
|
from time import clock
|
||||||
|
import os.path
|
||||||
|
|
||||||
|
from pyglet.gl import *
|
||||||
|
|
||||||
|
from punyverse.texture import load_texture
|
||||||
|
|
||||||
|
FACE_TRIANGLES = 0
|
||||||
|
FACE_QUADS = 1
|
||||||
|
|
||||||
|
|
||||||
|
def model_list(model, sx=1, sy=1, sz=1, rotation=(0, 0, 0)):
|
||||||
|
for m, text in model.materials.iteritems():
|
||||||
|
if text.texture:
|
||||||
|
load_texture(os.path.join(model.root, text.texture))
|
||||||
|
|
||||||
|
display = glGenLists(1)
|
||||||
|
|
||||||
|
glNewList(display, GL_COMPILE)
|
||||||
|
glPushMatrix()
|
||||||
|
glPushAttrib(GL_CURRENT_BIT)
|
||||||
|
|
||||||
|
pitch, yaw, roll = rotation
|
||||||
|
glPushAttrib(GL_TRANSFORM_BIT)
|
||||||
|
glRotatef(pitch, 1, 0, 0)
|
||||||
|
glRotatef(yaw, 0, 1, 0)
|
||||||
|
glRotatef(roll, 0, 0, 1)
|
||||||
|
glPopAttrib()
|
||||||
|
|
||||||
|
vertices = model.vertices
|
||||||
|
textures = model.textures
|
||||||
|
normals = model.normals
|
||||||
|
|
||||||
|
for g in model.groups:
|
||||||
|
material = g.material
|
||||||
|
|
||||||
|
tex_id = load_texture(os.path.join(model.root, material.texture)) if (material and material.texture) else 0
|
||||||
|
|
||||||
|
if tex_id:
|
||||||
|
glEnable(GL_TEXTURE_2D)
|
||||||
|
glBindTexture(GL_TEXTURE_2D, tex_id)
|
||||||
|
else:
|
||||||
|
glBindTexture(GL_TEXTURE_2D, 0)
|
||||||
|
glDisable(GL_TEXTURE_2D)
|
||||||
|
|
||||||
|
if material:
|
||||||
|
fv4 = GLfloat * 4
|
||||||
|
|
||||||
|
if material.Ka:
|
||||||
|
kx, ky, kz = material.Ka
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT, fv4(kx, ky, kz, 1))
|
||||||
|
if material.Kd:
|
||||||
|
kx, ky, kz = material.Kd
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_DIFFUSE, fv4(kx, ky, kz, 1))
|
||||||
|
if material.Ks:
|
||||||
|
kx, ky, kz = material.Ks
|
||||||
|
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, fv4(kx, ky, kz, 1))
|
||||||
|
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, material.shininess)
|
||||||
|
|
||||||
|
type = -1
|
||||||
|
|
||||||
|
def point(f, vertices, normals, textures, n):
|
||||||
|
if f.norms:
|
||||||
|
glNormal3f(*normals[f.norms[n]])
|
||||||
|
if tex_id:
|
||||||
|
glTexCoord2f(*textures[f.texs[n]][:2])
|
||||||
|
|
||||||
|
x, y, z = vertices[f.verts[n]]
|
||||||
|
glVertex3f(x * sx, y * sy, z * sz)
|
||||||
|
|
||||||
|
for f in g.faces:
|
||||||
|
if type != f.type:
|
||||||
|
if type != -1:
|
||||||
|
glEnd()
|
||||||
|
glBegin(GL_TRIANGLES)
|
||||||
|
type = f.type
|
||||||
|
|
||||||
|
point(f, vertices, normals, textures, 0)
|
||||||
|
point(f, vertices, normals, textures, 1)
|
||||||
|
point(f, vertices, normals, textures, 2)
|
||||||
|
|
||||||
|
if type == FACE_QUADS:
|
||||||
|
point(f, vertices, normals, textures, 2)
|
||||||
|
point(f, vertices, normals, textures, 3)
|
||||||
|
point(f, vertices, normals, textures, 0)
|
||||||
|
glEnd()
|
||||||
|
|
||||||
|
glPopAttrib()
|
||||||
|
glPopMatrix()
|
||||||
|
|
||||||
|
glEndList()
|
||||||
|
return display
|
||||||
|
|
||||||
|
def load_model(path):
|
||||||
|
path = os.path.join(os.path.dirname(__file__), "assets", "models", path)
|
||||||
|
root = os.path.dirname(path)
|
||||||
|
|
||||||
|
vertices = []
|
||||||
|
normals = []
|
||||||
|
textures = []
|
||||||
|
groups = []
|
||||||
|
materials = {}
|
||||||
|
|
||||||
|
# Allez au diable, Python
|
||||||
|
current_group = [None]
|
||||||
|
current_material = [None]
|
||||||
|
|
||||||
|
def dispatch(p, commands):
|
||||||
|
with open(p, 'r') as file:
|
||||||
|
for line in file:
|
||||||
|
line = line.strip()
|
||||||
|
if not line or line[0] == '#':
|
||||||
|
continue # Empty or comment
|
||||||
|
words = line.split()
|
||||||
|
type = words[0]
|
||||||
|
if type in commands:
|
||||||
|
commands[type](words)
|
||||||
|
|
||||||
|
def newmtl(words):
|
||||||
|
material = Material(words[1])
|
||||||
|
materials[words[1]] = material
|
||||||
|
current_material[0] = material
|
||||||
|
|
||||||
|
def Ka(words):
|
||||||
|
current_material[0].Ka = (float(words[1]), float(words[2]), float(words[3]))
|
||||||
|
|
||||||
|
def Kd(words):
|
||||||
|
current_material[0].Kd = (float(words[1]), float(words[2]), float(words[3]))
|
||||||
|
|
||||||
|
def Ks(words):
|
||||||
|
current_material[0].Ks = (float(words[1]), float(words[2]), float(words[3]))
|
||||||
|
|
||||||
|
def Ns(words):
|
||||||
|
current_material[0].shininess = min(float(words[1]), 125) # Seems to sometimes be > 125. TODO: find out why
|
||||||
|
|
||||||
|
def map_Kd(words):
|
||||||
|
current_material[0].texture = words[-1]
|
||||||
|
|
||||||
|
def v(words):
|
||||||
|
vertices.append((float(words[1]), float(words[2]), float(words[3])))
|
||||||
|
|
||||||
|
def vn(words):
|
||||||
|
normals.append((float(words[1]), float(words[2]), float(words[3])))
|
||||||
|
|
||||||
|
def vt(words):
|
||||||
|
l = len(words)
|
||||||
|
x, y, z = 0, 0, 0
|
||||||
|
if l >= 2:
|
||||||
|
x = float(words[1])
|
||||||
|
if l >= 3:
|
||||||
|
# OBJ origin is at upper left, OpenGL origin is at lower left
|
||||||
|
y = 1 - float(words[2])
|
||||||
|
if l >= 4:
|
||||||
|
z = float(words[3])
|
||||||
|
textures.append((x, y, z))
|
||||||
|
|
||||||
|
def f(words):
|
||||||
|
l = len(words)
|
||||||
|
type = -1
|
||||||
|
face_vertices = []
|
||||||
|
face_normals = []
|
||||||
|
face_textures = []
|
||||||
|
|
||||||
|
vertex_count = l - 1
|
||||||
|
if vertex_count == 3:
|
||||||
|
type = FACE_TRIANGLES
|
||||||
|
else:
|
||||||
|
type = FACE_QUADS
|
||||||
|
|
||||||
|
raw_faces = []
|
||||||
|
current_value = -1
|
||||||
|
vindices = []
|
||||||
|
nindices = []
|
||||||
|
tindices = []
|
||||||
|
|
||||||
|
for i in xrange(1, vertex_count + 1):
|
||||||
|
raw_faces = words[i].split('/')
|
||||||
|
l = len(raw_faces)
|
||||||
|
|
||||||
|
current_value = int(raw_faces[0])
|
||||||
|
|
||||||
|
vindices.append(current_value - 1)
|
||||||
|
face_vertices.append(vertices[current_value - 1])
|
||||||
|
|
||||||
|
if l == 1:
|
||||||
|
continue
|
||||||
|
|
||||||
|
if l >= 2 and raw_faces[1]:
|
||||||
|
current_value = int(raw_faces[1])
|
||||||
|
if current_value <= len(textures):
|
||||||
|
tindices.append(current_value - 1)
|
||||||
|
face_textures.append(textures[current_value - 1])
|
||||||
|
if l >= 3 and raw_faces[2]:
|
||||||
|
current_value = int(raw_faces[2])
|
||||||
|
nindices.append(current_value - 1)
|
||||||
|
face_normals.append(normals[current_value - 1])
|
||||||
|
|
||||||
|
if not groups:
|
||||||
|
group = Group()
|
||||||
|
groups.append(group)
|
||||||
|
else:
|
||||||
|
group = groups[-1]
|
||||||
|
group.vertices += face_vertices
|
||||||
|
group.normals += face_normals
|
||||||
|
group.textures += face_textures
|
||||||
|
|
||||||
|
idx_count = group.idx_count
|
||||||
|
group.indices += (idx_count + 1, idx_count + 2, idx_count + 3)
|
||||||
|
group.idx_count += 3
|
||||||
|
|
||||||
|
group.faces.append(Face(type, vindices, nindices, tindices, face_vertices, face_normals, face_textures))
|
||||||
|
|
||||||
|
mtllib = lambda words: dispatch(os.path.join(root, words[1]), obj)
|
||||||
|
|
||||||
|
def usemtl(words):
|
||||||
|
mat = words[1]
|
||||||
|
if mat in materials:
|
||||||
|
groups[-1].material = materials[mat]
|
||||||
|
else:
|
||||||
|
print "Warning: material %s undefined." % mat
|
||||||
|
|
||||||
|
g = lambda words: groups.append(Group(words[1]))
|
||||||
|
|
||||||
|
obj = {
|
||||||
|
'v': v,
|
||||||
|
'vn': vn,
|
||||||
|
'vt': vt,
|
||||||
|
'f': f,
|
||||||
|
'mtllib': mtllib,
|
||||||
|
'usemtl': usemtl,
|
||||||
|
'g': g,
|
||||||
|
'o': g, # TODO: o is not really g
|
||||||
|
'newmtl': newmtl,
|
||||||
|
'Ka': Ka,
|
||||||
|
'Kd': Kd,
|
||||||
|
'Ks': Ks,
|
||||||
|
'Ns': Ns,
|
||||||
|
'map_Kd': map_Kd
|
||||||
|
}
|
||||||
|
|
||||||
|
dispatch(path, obj)
|
||||||
|
|
||||||
|
return WavefrontObject(root, vertices, normals, textures, groups, materials)
|
||||||
|
|
||||||
|
|
||||||
|
class WavefrontObject(object):
|
||||||
|
def __init__(self, root, vertices=None, normals=None, textures=None, groups=None, materials=None):
|
||||||
|
self.root = root
|
||||||
|
self.vertices = vertices or []
|
||||||
|
self.normals = normals or []
|
||||||
|
self.textures = textures or []
|
||||||
|
self.groups = groups or []
|
||||||
|
self.materials = materials or []
|
||||||
|
|
||||||
|
|
||||||
|
class Group(object):
|
||||||
|
def __init__(self, name=None):
|
||||||
|
if not name:
|
||||||
|
name = clock()
|
||||||
|
self.name = name
|
||||||
|
self.material = None
|
||||||
|
self.faces = []
|
||||||
|
self.indices = []
|
||||||
|
self.vertices = []
|
||||||
|
self.normals = []
|
||||||
|
self.textures = []
|
||||||
|
self.idx_count = 0
|
||||||
|
|
||||||
|
|
||||||
|
class Face(object):
|
||||||
|
def __init__(self, type, verts, norms, texs, vertices, normals, textures):
|
||||||
|
self.type = type
|
||||||
|
self.verts = verts
|
||||||
|
self.norms = norms
|
||||||
|
self.texs = texs
|
||||||
|
self.vertices = vertices
|
||||||
|
self.normals = normals
|
||||||
|
self.textures = textures
|
||||||
|
|
||||||
|
|
||||||
|
class Material(object):
|
||||||
|
def __init__(self, name, texture=None, Ka=0, Kd=0, Ks=0, shininess=0.0):
|
||||||
|
self.name = name
|
||||||
|
self.texture = texture
|
||||||
|
self.Ka = Ka
|
||||||
|
self.Kd = Kd
|
||||||
|
self.Ks = Ks
|
||||||
|
self.shininess = shininess
|
95
punyverse/orbit.py
Normal file
|
@ -0,0 +1,95 @@
|
||||||
|
from math import sin, cos, tan, atan, sqrt, radians, degrees
|
||||||
|
|
||||||
|
|
||||||
|
class KeplerOrbit(object):
|
||||||
|
def __init__(self, sma, eccentricity, inclination=0, longitude=0, argument=0):
|
||||||
|
self.sma = sma
|
||||||
|
self.eccentricity = eccentricity
|
||||||
|
self.inclination = inclination
|
||||||
|
self.longitude = longitude
|
||||||
|
self.argument = argument
|
||||||
|
self.__true_anomaly_factor = sqrt((1 + eccentricity)/(1 - eccentricity))
|
||||||
|
self.__distance_factor = sma * (1 - eccentricity ** 2)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def inclination(self):
|
||||||
|
return degrees(self._inclination)
|
||||||
|
|
||||||
|
@inclination.setter
|
||||||
|
def inclination(self, value):
|
||||||
|
self.inclination_radian = radians(value)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def inclination_radian(self):
|
||||||
|
return self._inclination
|
||||||
|
|
||||||
|
@inclination_radian.setter
|
||||||
|
def inclination_radian(self, value):
|
||||||
|
self._inclination = value
|
||||||
|
self.__sin_inclination = sin(self._inclination)
|
||||||
|
self.__cos_inclination = cos(self._inclination)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def longitude(self):
|
||||||
|
return degrees(self._longitude)
|
||||||
|
|
||||||
|
@longitude.setter
|
||||||
|
def longitude(self, value):
|
||||||
|
self.longitude_radian = radians(value)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def longitude_radian(self):
|
||||||
|
return self._longitude
|
||||||
|
|
||||||
|
@longitude_radian.setter
|
||||||
|
def longitude_radian(self, value):
|
||||||
|
self._longitude = value
|
||||||
|
self.__sin_longitude = sin(self._longitude)
|
||||||
|
self.__cos_longitude = cos(self._longitude)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def argument(self):
|
||||||
|
return degrees(self._argument)
|
||||||
|
|
||||||
|
@argument.setter
|
||||||
|
def argument(self, value):
|
||||||
|
self.argument_radian = radians(value)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def argument_radian(self):
|
||||||
|
return self._argument
|
||||||
|
|
||||||
|
@argument_radian.setter
|
||||||
|
def argument_radian(self, value):
|
||||||
|
self._argument = value
|
||||||
|
self.__sin_argument = sin(self._argument)
|
||||||
|
self.__cos_argument = cos(self._argument)
|
||||||
|
|
||||||
|
def eccentric_anomaly(self, mean_anomaly):
|
||||||
|
e1 = mean_anomaly
|
||||||
|
e2 = mean_anomaly + self.eccentricity * sin(e1)
|
||||||
|
while abs(e1 - e2) > 0.000001:
|
||||||
|
e1, e2 = e2, mean_anomaly + self.eccentricity * sin(e2)
|
||||||
|
return e2
|
||||||
|
|
||||||
|
def true_anomaly(self, mean_anomaly):
|
||||||
|
eccentric_anomaly = self.eccentric_anomaly(mean_anomaly)
|
||||||
|
return 2 * atan(self.__true_anomaly_factor * tan(eccentric_anomaly / 2))
|
||||||
|
|
||||||
|
def orbit(self, mean_anomaly):
|
||||||
|
mean_anomaly = radians(mean_anomaly)
|
||||||
|
phi = self.true_anomaly(mean_anomaly)
|
||||||
|
r = self.__distance_factor / (1 + self.eccentricity * cos(phi))
|
||||||
|
x = r * cos(phi)
|
||||||
|
y = r * sin(phi)
|
||||||
|
z = 0
|
||||||
|
|
||||||
|
# phi = longitude, theta = inclination, psi = argument
|
||||||
|
x, y = (x * self.__cos_longitude + y * self.__sin_longitude,
|
||||||
|
-x * self.__sin_longitude + y * self.__cos_longitude)
|
||||||
|
x, z = (x * self.__cos_inclination + z * self.__sin_inclination,
|
||||||
|
-x * self.__sin_inclination + z * self.__cos_inclination)
|
||||||
|
x, y = (x * self.__cos_argument + y * self.__sin_argument,
|
||||||
|
-x * self.__sin_argument + y * self.__cos_argument)
|
||||||
|
|
||||||
|
return x, y, z
|
222
punyverse/texture.py
Normal file
|
@ -0,0 +1,222 @@
|
||||||
|
from pyglet import image
|
||||||
|
from pyglet.gl import *
|
||||||
|
from ctypes import c_int, byref, c_ulong
|
||||||
|
import os.path
|
||||||
|
import struct
|
||||||
|
|
||||||
|
try:
|
||||||
|
from _glgeom import bgr_to_rgb
|
||||||
|
except ImportError:
|
||||||
|
import warnings
|
||||||
|
warnings.warn('Compile _glgeom.c, or double the start up time.')
|
||||||
|
|
||||||
|
# Use magick when _glgeom is not compiled (is actually slower)
|
||||||
|
try:
|
||||||
|
from pgmagick import Blob, Image
|
||||||
|
except ImportError:
|
||||||
|
magick = False
|
||||||
|
else:
|
||||||
|
magick = True
|
||||||
|
|
||||||
|
def bgr_to_rgb(source, width, height, alpha=False, bottom_up=True):
|
||||||
|
length = len(source)
|
||||||
|
depth = length / (width * height)
|
||||||
|
depth2 = depth - alpha
|
||||||
|
result = bytearray(length)
|
||||||
|
row = width * depth
|
||||||
|
for y in xrange(height):
|
||||||
|
for x in xrange(width):
|
||||||
|
ioffset = y * width * depth + x * depth
|
||||||
|
ooffset = (height - y - 1 if bottom_up else y) * row + x * depth
|
||||||
|
for i in xrange(depth2):
|
||||||
|
result[ooffset+i] = source[ioffset+depth2-i-1]
|
||||||
|
if alpha:
|
||||||
|
result[ooffset+depth2] = source[ioffset+depth2]
|
||||||
|
return str(result)
|
||||||
|
else:
|
||||||
|
magick = False
|
||||||
|
|
||||||
|
try:
|
||||||
|
from cStringIO import StringIO
|
||||||
|
except ImportError:
|
||||||
|
from StringIO import StringIO
|
||||||
|
|
||||||
|
__all__ = ['load_texture']
|
||||||
|
|
||||||
|
id = 0
|
||||||
|
cache = {}
|
||||||
|
|
||||||
|
max_texture = None
|
||||||
|
power_of_two = None
|
||||||
|
badcard = False
|
||||||
|
bgra = False
|
||||||
|
|
||||||
|
|
||||||
|
def init():
|
||||||
|
global max_texture, power_of_two, badcard, bgra, magick
|
||||||
|
|
||||||
|
if max_texture is None:
|
||||||
|
buf = c_int()
|
||||||
|
glGetIntegerv(GL_MAX_TEXTURE_SIZE, byref(buf))
|
||||||
|
max_texture = buf.value
|
||||||
|
badcard = gl_info.get_renderer() in ('GDI Generic',)
|
||||||
|
if badcard:
|
||||||
|
import warnings
|
||||||
|
warnings.warn('Please update your graphics drivers if possible')
|
||||||
|
#extensions = gl_info.get_extensions()
|
||||||
|
#bgra = 'GL_EXT_bgra' in extensions
|
||||||
|
#if bgra and magick:
|
||||||
|
# magick = False # Disable magick because BGRA needs it not
|
||||||
|
|
||||||
|
if power_of_two is None:
|
||||||
|
power_of_two = gl_info.have_version(2) or gl_info.have_extension('GL_ARB_texture_non_power_of_two')
|
||||||
|
|
||||||
|
is_power2 = lambda num: num != 0 and ((num & (num - 1)) == 0)
|
||||||
|
|
||||||
|
|
||||||
|
def image_info(data):
|
||||||
|
data = str(data)
|
||||||
|
size = len(data)
|
||||||
|
height = -1
|
||||||
|
width = -1
|
||||||
|
content_type = ''
|
||||||
|
|
||||||
|
# handle GIFs
|
||||||
|
if (size >= 10) and data[:6] in ('GIF87a', 'GIF89a'):
|
||||||
|
# Check to see if content_type is correct
|
||||||
|
content_type = 'image/gif'
|
||||||
|
w, h = struct.unpack("<HH", data[6:10])
|
||||||
|
width = int(w)
|
||||||
|
height = int(h)
|
||||||
|
|
||||||
|
# See PNG 2. Edition spec (http://www.w3.org/TR/PNG/)
|
||||||
|
# Bytes 0-7 are below, 4-byte chunk length, then 'IHDR'
|
||||||
|
# and finally the 4-byte width, height
|
||||||
|
elif ((size >= 24) and data.startswith('\211PNG\r\n\032\n')
|
||||||
|
and (data[12:16] == 'IHDR')):
|
||||||
|
content_type = 'image/png'
|
||||||
|
w, h = struct.unpack(">LL", data[16:24])
|
||||||
|
width = int(w)
|
||||||
|
height = int(h)
|
||||||
|
|
||||||
|
# Maybe this is for an older PNG version.
|
||||||
|
elif (size >= 16) and data.startswith('\211PNG\r\n\032\n'):
|
||||||
|
# Check to see if we have the right content type
|
||||||
|
content_type = 'image/png'
|
||||||
|
w, h = struct.unpack(">LL", data[8:16])
|
||||||
|
width = int(w)
|
||||||
|
height = int(h)
|
||||||
|
|
||||||
|
# handle JPEGs
|
||||||
|
elif (size >= 2) and data.startswith('\377\330'):
|
||||||
|
content_type = 'image/jpeg'
|
||||||
|
jpeg = StringIO(data)
|
||||||
|
jpeg.read(2)
|
||||||
|
b = jpeg.read(1)
|
||||||
|
try:
|
||||||
|
while b and ord(b) != 0xDA:
|
||||||
|
while ord(b) != 0xFF:
|
||||||
|
b = jpeg.read(1)
|
||||||
|
while ord(b) == 0xFF:
|
||||||
|
b = jpeg.read(1)
|
||||||
|
if 0xC0 <= ord(b) <= 0xC3:
|
||||||
|
jpeg.read(3)
|
||||||
|
h, w = struct.unpack(">HH", jpeg.read(4))
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
jpeg.read(int(struct.unpack(">H", jpeg.read(2))[0])-2)
|
||||||
|
b = jpeg.read(1)
|
||||||
|
width = int(w)
|
||||||
|
height = int(h)
|
||||||
|
except struct.error:
|
||||||
|
pass
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return content_type, width, height
|
||||||
|
|
||||||
|
|
||||||
|
def check_size(width, height):
|
||||||
|
init()
|
||||||
|
if width > max_texture or height > max_texture:
|
||||||
|
print 'too large'
|
||||||
|
raise ValueError('Texture too large')
|
||||||
|
elif not power_of_two:
|
||||||
|
if not is_power2(width) or not is_power2(height):
|
||||||
|
print 'not power of two'
|
||||||
|
raise ValueError('Texture not power of two')
|
||||||
|
|
||||||
|
|
||||||
|
def load_texture(file):
|
||||||
|
if os.path.isabs(file):
|
||||||
|
path = file
|
||||||
|
file = os.path.basename(path)
|
||||||
|
else:
|
||||||
|
path = os.path.join(os.path.dirname(__file__), "assets", "textures", file)
|
||||||
|
|
||||||
|
if path in cache:
|
||||||
|
return cache[path]
|
||||||
|
print "Loading image %s..." % file,
|
||||||
|
|
||||||
|
try:
|
||||||
|
file = open(path, 'rb')
|
||||||
|
except IOError:
|
||||||
|
print 'exists not'
|
||||||
|
raise ValueError('Texture exists not')
|
||||||
|
type, width, height = image_info(file.read(8192))
|
||||||
|
file.seek(0, 0)
|
||||||
|
if type:
|
||||||
|
check_size(width, height)
|
||||||
|
|
||||||
|
if magick:
|
||||||
|
file.close()
|
||||||
|
file = Image(path.encode('mbcs' if os.name == 'nt' else 'utf8'))
|
||||||
|
geo = file.size()
|
||||||
|
check_size(geo.width(), geo.height())
|
||||||
|
print
|
||||||
|
blob = Blob()
|
||||||
|
file.flip()
|
||||||
|
file.write(blob, 'RGBA')
|
||||||
|
texture = blob.data
|
||||||
|
mode = GL_RGBA
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
raw = image.load(path, file=file)
|
||||||
|
except IOError:
|
||||||
|
print 'exists not'
|
||||||
|
raise ValueError('Texture exists not')
|
||||||
|
|
||||||
|
width, height = raw.width, raw.height
|
||||||
|
check_size(width, height)
|
||||||
|
print
|
||||||
|
|
||||||
|
mode = GL_RGBA if 'A' in raw.format else GL_RGB
|
||||||
|
# Flip from BGR to RGB
|
||||||
|
# I hate you too, Pyglet...
|
||||||
|
# REGULAR EXPRESSIONS ARE NOT MEANT TO PARSE BINARY DATA!!!
|
||||||
|
#texture = raw.get_data('RGBA', width * 4) if safe else raw.data[::-1] if 'BGR' in raw.format else raw.data
|
||||||
|
if raw.format in ('BGR', 'BGRA'):
|
||||||
|
if bgra:
|
||||||
|
mode = {GL_RGBA: GL_BGRA, GL_RGB: GL_BGR}[mode]
|
||||||
|
texture = raw.data
|
||||||
|
else:
|
||||||
|
texture = bgr_to_rgb(raw.data, width, height, 'A' in raw.format)
|
||||||
|
elif raw.format in ('RGB', 'RGBA'):
|
||||||
|
texture = raw.data
|
||||||
|
else:
|
||||||
|
texture = raw.get_data('RGBA', width * 4)
|
||||||
|
|
||||||
|
buffer = c_ulong()
|
||||||
|
glGenTextures(1, byref(buffer))
|
||||||
|
id = buffer.value
|
||||||
|
|
||||||
|
glBindTexture(GL_TEXTURE_2D, id)
|
||||||
|
|
||||||
|
filter = GL_NEAREST if badcard else GL_LINEAR
|
||||||
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, filter)
|
||||||
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, filter)
|
||||||
|
gluBuild2DMipmaps(GL_TEXTURE_2D, len(raw.format), width, height, mode, GL_UNSIGNED_BYTE, texture)
|
||||||
|
|
||||||
|
cache[path] = id
|
||||||
|
|
||||||
|
return id
|
255
punyverse/world.json
Normal file
|
@ -0,0 +1,255 @@
|
||||||
|
{
|
||||||
|
"comments": {
|
||||||
|
"au": "greatly shrunk so planets can actually be seen",
|
||||||
|
"tick": "real world second per game tick",
|
||||||
|
"length": "kilometre per world units for satellite distance and planetary radius",
|
||||||
|
"radius": "note that radius is equatorial, flattening will be implemented later",
|
||||||
|
"distance": "virtual distance to look better, in km",
|
||||||
|
"sma": "semi-major axis used with mass of parent to calculate orbit, in km",
|
||||||
|
"mass": "mass in kg",
|
||||||
|
"texture": "a group of texture to use, tried in that order. a list means a colour",
|
||||||
|
"model": "used to load a wavefront object instead of a textured sphere"
|
||||||
|
},
|
||||||
|
"au": 10000,
|
||||||
|
"tick": 180,
|
||||||
|
"length": 63.7,
|
||||||
|
"bodies": {
|
||||||
|
"earth": {
|
||||||
|
"texture": ["earth.jpg", "earth_medium.jpg", "earth_small.jpg", [0, 0.28, 1, 1]],
|
||||||
|
"radius": 6378.1,
|
||||||
|
"z": "AU",
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 23.4,
|
||||||
|
"roll": -90,
|
||||||
|
"mass": 5.97219e+24,
|
||||||
|
"rotation": 86400,
|
||||||
|
"atmosphere": {
|
||||||
|
"cloud_texture": "cloudmap.png",
|
||||||
|
"diffuse_texture": "atmosphere_earth.png",
|
||||||
|
"diffuse_size": 30
|
||||||
|
},
|
||||||
|
"orbit_distance": "AU",
|
||||||
|
"satellites": {
|
||||||
|
"moon": {
|
||||||
|
"texture": ["moon.jpg", "moon_medium.jpg", "moon_small.jpg", [0.53, 0.53, 0.53, 1]],
|
||||||
|
"radius": 1738.14,
|
||||||
|
"distance": 38439,
|
||||||
|
"sma": 384399,
|
||||||
|
"eccentricity": 0.0549,
|
||||||
|
"inclination": 5.145,
|
||||||
|
"rotation": 0,
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 6.68,
|
||||||
|
"roll": -90
|
||||||
|
},
|
||||||
|
"iss": {
|
||||||
|
"model": "satellites/iss.obj",
|
||||||
|
"inclination": 51.65,
|
||||||
|
"distance": 6800,
|
||||||
|
"scale": 5
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"mars": {
|
||||||
|
"texture": ["mars.jpg", "mars_small.jpg", "mars_medium.jpg", [0.85, 0.47, 0.2, 1]],
|
||||||
|
"radius": 3396.2,
|
||||||
|
"z": "1.524 * AU",
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 25.19,
|
||||||
|
"mass": 6.4185e+23,
|
||||||
|
"rotation": 88643,
|
||||||
|
"orbit_distance": "AU",
|
||||||
|
"satellites": {
|
||||||
|
"phobos": {
|
||||||
|
"distance": 9377,
|
||||||
|
"inclination": 26.04,
|
||||||
|
"eccentricity": 0.0151,
|
||||||
|
"model": "satellites/phobos.obj"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"jupiter": {
|
||||||
|
"texture": ["jupiter.jpg", "jupiter_medium.jpg", "jupiter_small.jpg", [0.65, 0.36, 0.19, 1]],
|
||||||
|
"radius": 71492,
|
||||||
|
"mass": 1.8986e+27,
|
||||||
|
"z": "5.2 * AU",
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 3.13,
|
||||||
|
"comment": "satellites here are 3/10 the virtual distance than physical, and five times the size",
|
||||||
|
"rotation": 35730,
|
||||||
|
"orbit_distance": "3 * AU",
|
||||||
|
"satellites": {
|
||||||
|
"io": {
|
||||||
|
"texture": ["moons/io.jpg", "moons/io_small.jpg", [0.62, 0.56, 0.35, 1]],
|
||||||
|
"radius": "1821.3 * 5",
|
||||||
|
"distance": 126510,
|
||||||
|
"sma": 421700,
|
||||||
|
"rotation": 0,
|
||||||
|
"pitch": -90,
|
||||||
|
"inclination": 2.21,
|
||||||
|
"eccentricity": 0.0041
|
||||||
|
},
|
||||||
|
"europa": {
|
||||||
|
"texture": ["moons/europa.jpg", "moons/europa_small.jpg", [0.77, 0.74, 0.65, 1]],
|
||||||
|
"radius": "1560.8 * 5",
|
||||||
|
"distance": 201270,
|
||||||
|
"sma": 670900,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 2.71,
|
||||||
|
"eccentricity": 0.009
|
||||||
|
},
|
||||||
|
"ganymede": {
|
||||||
|
"texture": ["moons/ganymede.jpg", "moons/ganymede_small.jpg", [0.52, 0.47, 0.46, 1]],
|
||||||
|
"radius": "2634.1 * 5",
|
||||||
|
"distance": 321120,
|
||||||
|
"sma": 1070400,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 2.51,
|
||||||
|
"eccentricity": 0.0013
|
||||||
|
},
|
||||||
|
"callisto": {
|
||||||
|
"texture": ["moons/callisto.jpg", "moons/callisto_small.jpg", [0.49, 0.43, 0.34, 1]],
|
||||||
|
"radius": "2410.3 * 5",
|
||||||
|
"distance": 564810,
|
||||||
|
"sma": 1882700,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 0.192,
|
||||||
|
"eccentricity": 0.0074
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"saturn": {
|
||||||
|
"texture": ["saturn.jpg", "saturn_medium.jpg", "saturn_small.jpg", [0.9, 0.8, 0.64, 1]],
|
||||||
|
"radius": 60268,
|
||||||
|
"mass": 5.6846e+26,
|
||||||
|
"z": "9.58 * AU",
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 26.73,
|
||||||
|
"rotation": 38052,
|
||||||
|
"ring": {
|
||||||
|
"texture": "ring_saturn.png",
|
||||||
|
"distance": 1169,
|
||||||
|
"size": 2247
|
||||||
|
},
|
||||||
|
"orbit_distance": "4 * AU",
|
||||||
|
"satellites": {
|
||||||
|
"titan": {
|
||||||
|
"texture": ["moons/titan.jpg", "moons/titan_small.jpg", [0.52, 0.39, 0.23, 1]],
|
||||||
|
"radius": "2576 * 10",
|
||||||
|
"distance": "1221870 / 3 + 200000",
|
||||||
|
"sma": 1221870,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 27.07854,
|
||||||
|
"eccentricity": 0.0288
|
||||||
|
},
|
||||||
|
"rhea": {
|
||||||
|
"texture": ["moons/rhea.jpg", "moons/rhea_small.jpg", [0.62, 0.60, 0.59, 1]],
|
||||||
|
"radius": "763.8 * 10",
|
||||||
|
"distance": "527108 / 3 + 200000",
|
||||||
|
"sma": 527108,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 27.075,
|
||||||
|
"eccentricity": 0.0012583
|
||||||
|
},
|
||||||
|
"iapetus": {
|
||||||
|
"texture": ["moons/iapetus.jpg", "moons/iapetus_small.jpg", [0.62, 0.60, 0.59, 1]],
|
||||||
|
"radius": "734.5 * 10",
|
||||||
|
"distance": "3560820 / 3 + 200000",
|
||||||
|
"sma": 3560820,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 17.28,
|
||||||
|
"eccentricity": 0.0286125
|
||||||
|
},
|
||||||
|
"dione": {
|
||||||
|
"texture": ["moons/dione.jpg", "moons/dione_small.jpg", [0.46, 0.46, 0.46, 1]],
|
||||||
|
"radius": "561.4 * 10",
|
||||||
|
"distance": "377396 / 3 + 200000",
|
||||||
|
"sma": 377396,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 26.711,
|
||||||
|
"eccentricity": 0.0022
|
||||||
|
},
|
||||||
|
"tethys": {
|
||||||
|
"texture": ["moons/tethys.jpg", "moons/tethys_small.jpg", [0.68, 0.68, 0.66, 1]],
|
||||||
|
"radius": "531.1 * 10",
|
||||||
|
"distance": "294619 / 3 + 200000",
|
||||||
|
"sma": 294619,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 25.61,
|
||||||
|
"eccentricity": 0.0001
|
||||||
|
},
|
||||||
|
"enceladus": {
|
||||||
|
"texture": ["moons/enceladus.jpg", "moons/enceladus_small.jpg", [0.74, 0.74, 0.74, 1]],
|
||||||
|
"radius": "252.1 * 10",
|
||||||
|
"distance": "237948 / 3 + 200000",
|
||||||
|
"sma": 237948,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 26.711,
|
||||||
|
"eccentricity": 0.0047
|
||||||
|
},
|
||||||
|
"mimas": {
|
||||||
|
"texture": ["moons/mimas.jpg", "moons/mimas_small.jpg", [0.47, 0.47, 0.47, 1]],
|
||||||
|
"radius": "198.2 * 10",
|
||||||
|
"distance": "181902 / 3 + 200000",
|
||||||
|
"sma": 181902,
|
||||||
|
"pitch": -90,
|
||||||
|
"rotation": 0,
|
||||||
|
"inclination": 28.304,
|
||||||
|
"eccentricity": 0.0196
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"uranus": {
|
||||||
|
"texture": ["uranus.jpg", [0, 0.53, 0.84, 1]],
|
||||||
|
"radius": 25559,
|
||||||
|
"mass": 8.6810e+25,
|
||||||
|
"z": "19.23 * AU",
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 97.77,
|
||||||
|
"rotation": -62064,
|
||||||
|
"orbit_distance": "6 * AU",
|
||||||
|
"ring": {
|
||||||
|
"texture": "ring_uranus.png",
|
||||||
|
"pitch": 0,
|
||||||
|
"yaw": 0,
|
||||||
|
"roll": 90,
|
||||||
|
"distance": 421,
|
||||||
|
"size": 781
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"neptune": {
|
||||||
|
"texture": ["neptune.jpg", [0.31, 0.49, 0.59, 1]],
|
||||||
|
"radius": 24764,
|
||||||
|
"mass": 1.0243e+26,
|
||||||
|
"z": "30.5 * AU",
|
||||||
|
"orbit_distance": "6 * AU",
|
||||||
|
"rotation": 57996,
|
||||||
|
"pitch": -90,
|
||||||
|
"yaw": 28.32
|
||||||
|
},
|
||||||
|
"sky": {
|
||||||
|
"texture": "sky.jpg",
|
||||||
|
"optional": true,
|
||||||
|
"lighting": false,
|
||||||
|
"radius": 190000,
|
||||||
|
"pitch": 90,
|
||||||
|
"yaw": 30,
|
||||||
|
"roll": 180,
|
||||||
|
"delta": 0,
|
||||||
|
"background": true
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"start": {
|
||||||
|
"z": "AU - 400",
|
||||||
|
"yaw": 180
|
||||||
|
}
|
||||||
|
}
|
186
punyverse/world.py
Normal file
|
@ -0,0 +1,186 @@
|
||||||
|
from collections import OrderedDict
|
||||||
|
import os.path
|
||||||
|
|
||||||
|
try:
|
||||||
|
import json
|
||||||
|
except ImportError:
|
||||||
|
try:
|
||||||
|
import simplejson as json
|
||||||
|
except ImportError:
|
||||||
|
raise SystemExit('No JSON module found')
|
||||||
|
|
||||||
|
try:
|
||||||
|
from punyverse._model import model_list, load_model
|
||||||
|
except ImportError:
|
||||||
|
from punyverse.model import model_list, load_model
|
||||||
|
|
||||||
|
from punyverse.glgeom import *
|
||||||
|
from punyverse.entity import *
|
||||||
|
from punyverse.texture import *
|
||||||
|
|
||||||
|
from math import pi, sqrt
|
||||||
|
|
||||||
|
G = 6.67384e-11 # Gravitation Constant
|
||||||
|
|
||||||
|
|
||||||
|
def get_best_texture(info, optional=False):
|
||||||
|
cheap = False
|
||||||
|
skip = False
|
||||||
|
texture = None
|
||||||
|
if isinstance(info, list):
|
||||||
|
for item in info:
|
||||||
|
if isinstance(item, list):
|
||||||
|
if len(item) == 4:
|
||||||
|
cheap = True
|
||||||
|
texture = item
|
||||||
|
break
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
texture = load_texture(item)
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
texture = load_texture(info)
|
||||||
|
except ValueError:
|
||||||
|
if optional:
|
||||||
|
skip = True
|
||||||
|
else:
|
||||||
|
cheap = True
|
||||||
|
texture = [1, 1, 1, 1]
|
||||||
|
return cheap, skip, texture
|
||||||
|
|
||||||
|
|
||||||
|
def load_world(file):
|
||||||
|
with open(os.path.join(os.path.dirname(__file__), file)) as f:
|
||||||
|
root = json.load(f, object_pairs_hook=OrderedDict)
|
||||||
|
|
||||||
|
world = World()
|
||||||
|
au = root.get('au', 2000)
|
||||||
|
e = lambda x: eval(str(x), {'__builtins__': None}, {'AU': au})
|
||||||
|
tick = root.get('tick', 4320) # How many second is a tick?
|
||||||
|
length = root.get('length', 4320) # Satellite distance is in km, divide by this gets in world units
|
||||||
|
world.tick_length = tick
|
||||||
|
|
||||||
|
if 'start' in root:
|
||||||
|
info = root['start']
|
||||||
|
x = e(info.get('x', 0))
|
||||||
|
y = e(info.get('y', 0))
|
||||||
|
z = e(info.get('z', 0))
|
||||||
|
pitch = e(info.get('pitch', 0))
|
||||||
|
yaw = e(info.get('yaw', 0))
|
||||||
|
roll = e(info.get('roll', 0))
|
||||||
|
world.start = (x, y, z)
|
||||||
|
world.direction = (pitch, yaw, roll)
|
||||||
|
|
||||||
|
def body(name, info, parent=None):
|
||||||
|
lighting = info.get('lighting', True)
|
||||||
|
x = e(info.get('x', 0))
|
||||||
|
y = e(info.get('y', 0))
|
||||||
|
z = e(info.get('z', 0))
|
||||||
|
pitch = e(info.get('pitch', 0))
|
||||||
|
yaw = e(info.get('yaw', 0))
|
||||||
|
roll = e(info.get('roll', 0))
|
||||||
|
rotation = e(info.get('rotation', 86400))
|
||||||
|
radius = e(info.get('radius', length)) / length
|
||||||
|
background = info.get('background', False)
|
||||||
|
orbit_distance = e(info.get('orbit_distance', au))
|
||||||
|
|
||||||
|
if 'texture' in info:
|
||||||
|
cheap, skip, texture = get_best_texture(info['texture'], optional=info.get('optional', False))
|
||||||
|
if skip:
|
||||||
|
return
|
||||||
|
division = min(int(radius / 2), 100)
|
||||||
|
if cheap:
|
||||||
|
object_id = compile(colourball, radius, division, division, texture)
|
||||||
|
else:
|
||||||
|
object_id = compile(sphere, radius, division, division, texture, lighting=lighting)
|
||||||
|
elif 'model' in info:
|
||||||
|
scale = info.get('scale', 1)
|
||||||
|
object_id = model_list(load_model(info['model']), info.get('sx', scale), info.get('sy', scale),
|
||||||
|
info.get('sz', scale), (0, 0, 0))
|
||||||
|
else:
|
||||||
|
print 'Nothing to load for %s.' % name
|
||||||
|
return
|
||||||
|
|
||||||
|
params = {'world': world, 'orbit_distance': orbit_distance}
|
||||||
|
if parent is None:
|
||||||
|
type = Body
|
||||||
|
else:
|
||||||
|
x, y, z = parent.location
|
||||||
|
distance = e(info.get('distance', 100)) # Semi-major axis when actually displayed in virtual space
|
||||||
|
sma = e(info.get('sma', distance)) # Semi-major axis used to calculate orbital speed
|
||||||
|
if hasattr(parent, 'mass') and parent.mass is not None:
|
||||||
|
period = 2 * pi * sqrt((sma * 1000) ** 3 / (G * parent.mass))
|
||||||
|
speed = 360 / (period + .0)
|
||||||
|
if not rotation: # Rotation = 0 assumes tidal lock
|
||||||
|
rotation = period
|
||||||
|
else:
|
||||||
|
speed = info.get('orbit_speed', 1)
|
||||||
|
type = Satellite
|
||||||
|
params.update(parent=parent, orbit_speed=speed,
|
||||||
|
distance=distance / length, eccentricity=info.get('eccentricity', 0),
|
||||||
|
inclination=info.get('inclination', 0), longitude=info.get('longitude', 0),
|
||||||
|
argument=info.get('argument', 0))
|
||||||
|
|
||||||
|
if 'mass' in info:
|
||||||
|
params['mass'] = info['mass']
|
||||||
|
|
||||||
|
atmosphere_id = 0
|
||||||
|
cloudmap_id = 0
|
||||||
|
if 'atmosphere' in info:
|
||||||
|
atmosphere_data = info['atmosphere']
|
||||||
|
size = e(atmosphere_data.get('diffuse_size', None))
|
||||||
|
atm_texture = atmosphere_data.get('diffuse_texture', None)
|
||||||
|
cloud_texture = atmosphere_data.get('cloud_texture', None)
|
||||||
|
cheap, _, cloud_texture = get_best_texture(cloud_texture)
|
||||||
|
if not cheap:
|
||||||
|
division = min(int(radius / 2), 100)
|
||||||
|
cloudmap_id = compile(sphere, radius + 2, division, division, cloud_texture,
|
||||||
|
lighting=False)
|
||||||
|
cheap, _, atm_texture = get_best_texture(atm_texture)
|
||||||
|
if not cheap:
|
||||||
|
atmosphere_id = compile(disk, radius, radius + size, 30, atm_texture)
|
||||||
|
|
||||||
|
object = type(object_id, (x, y, z), (pitch, yaw, roll), rotation_angle=360 / (rotation + .0),
|
||||||
|
atmosphere=atmosphere_id, cloudmap=cloudmap_id, background=background, **params)
|
||||||
|
world.tracker.append(object)
|
||||||
|
|
||||||
|
if 'ring' in info:
|
||||||
|
ring_data = info['ring']
|
||||||
|
texture = ring_data.get('texture', None)
|
||||||
|
distance = e(ring_data.get('distance', radius * 1.2))
|
||||||
|
size = e(ring_data.get('size', radius / 2))
|
||||||
|
pitch = e(ring_data.get('pitch', pitch))
|
||||||
|
yaw = e(ring_data.get('yaw', yaw))
|
||||||
|
roll = e(ring_data.get('roll', roll))
|
||||||
|
|
||||||
|
cheap, _, texture = get_best_texture(texture)
|
||||||
|
if not cheap:
|
||||||
|
world.tracker.append(
|
||||||
|
type(compile(disk, distance, distance + size, 30, texture), (x, y, z),
|
||||||
|
(pitch, yaw, roll), **params))
|
||||||
|
|
||||||
|
for satellite, info in info.get('satellites', {}).iteritems():
|
||||||
|
print "Loading %s, satellite of %s." % (satellite, name)
|
||||||
|
body(satellite, info, object)
|
||||||
|
|
||||||
|
for planet, info in root['bodies'].iteritems():
|
||||||
|
print "Loading %s." % planet
|
||||||
|
body(planet, info)
|
||||||
|
|
||||||
|
return world
|
||||||
|
|
||||||
|
|
||||||
|
class World(object):
|
||||||
|
def __init__(self):
|
||||||
|
self.tracker = []
|
||||||
|
self.start = (0, 0, 0)
|
||||||
|
self.direction = (0, 0, 0)
|
||||||
|
self.x = None
|
||||||
|
self.y = None
|
||||||
|
self.z = None
|
||||||
|
self.tick_length = 1
|
||||||
|
self.tick = 0
|
118
small_images.py
Normal file
|
@ -0,0 +1,118 @@
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
from pyglet.gl import GLint, glGetIntegerv, GL_MAX_TEXTURE_SIZE
|
||||||
|
from ctypes import byref
|
||||||
|
|
||||||
|
buf = GLint()
|
||||||
|
glGetIntegerv(GL_MAX_TEXTURE_SIZE, byref(buf))
|
||||||
|
max_texture = buf.value
|
||||||
|
|
||||||
|
del byref, GLint, glGetIntegerv, GL_MAX_TEXTURE_SIZE, buf
|
||||||
|
|
||||||
|
def resize(width, height, target):
|
||||||
|
factor = (target + .0) / max(width, height)
|
||||||
|
return int(width * factor), int(height * factor)
|
||||||
|
|
||||||
|
def fits(width, height):
|
||||||
|
return width < max_texture and height < max_texture
|
||||||
|
|
||||||
|
def make_name(image, suffix):
|
||||||
|
name, ext = os.path.splitext(image)
|
||||||
|
return '%s_%s%s' % (name, suffix, ext)
|
||||||
|
|
||||||
|
try:
|
||||||
|
import pgmagick
|
||||||
|
|
||||||
|
def get_image(image):
|
||||||
|
return pgmagick.Image(image)
|
||||||
|
|
||||||
|
def get_size(image):
|
||||||
|
size = image.size()
|
||||||
|
return size.width(), size.height()
|
||||||
|
|
||||||
|
def scale(image, width, height):
|
||||||
|
image.filterType(pgmagick.FilterTypes.LanczosFilter)
|
||||||
|
image.scale(pgmagick.Geometry(width, height))
|
||||||
|
return image
|
||||||
|
|
||||||
|
def save(image, file):
|
||||||
|
image.write(file)
|
||||||
|
except ImportError:
|
||||||
|
import Image
|
||||||
|
|
||||||
|
def get_image(image):
|
||||||
|
return Image.open(image)
|
||||||
|
|
||||||
|
def get_size(image):
|
||||||
|
return image.size
|
||||||
|
|
||||||
|
def scale(image, width, height):
|
||||||
|
original_width, original_height = image.size
|
||||||
|
if width * 3 < original_width and height * 3 < original_height:
|
||||||
|
image = image.resize((width * 2, height * 2))
|
||||||
|
return image.resize((width, height), Image.ANTIALIAS)
|
||||||
|
|
||||||
|
def save(image, file):
|
||||||
|
image.save(file)
|
||||||
|
|
||||||
|
def shrink(file):
|
||||||
|
image = get_image(file)
|
||||||
|
width, height = get_size(image)
|
||||||
|
if fits(width, height):
|
||||||
|
print 'no need'
|
||||||
|
return
|
||||||
|
width, height = resize(width, height, 2048)
|
||||||
|
if fits(width, height):
|
||||||
|
size = 'medium'
|
||||||
|
else:
|
||||||
|
width, height = resize(width, height, 1024) # 1024 is minimum
|
||||||
|
size = 'small'
|
||||||
|
print 'size %s, %dx%d...' % (size, width, height),
|
||||||
|
name = make_name(file, size)
|
||||||
|
if not os.path.exists(name):
|
||||||
|
image = scale(image, width, height)
|
||||||
|
print 'saved to:', os.path.basename(name)
|
||||||
|
save(image, name)
|
||||||
|
else:
|
||||||
|
print
|
||||||
|
|
||||||
|
textures = [
|
||||||
|
'earth.jpg',
|
||||||
|
'moon.jpg',
|
||||||
|
'mars.jpg',
|
||||||
|
'jupiter.jpg',
|
||||||
|
'saturn.jpg',
|
||||||
|
'moons/io.jpg',
|
||||||
|
'moons/europa.jpg',
|
||||||
|
'moons/ganymede.jpg',
|
||||||
|
'moons/callisto.jpg',
|
||||||
|
'moons/titan.jpg',
|
||||||
|
'moons/rhea.jpg',
|
||||||
|
'moons/iapetus.jpg',
|
||||||
|
'moons/dione.jpg',
|
||||||
|
'moons/tethys.jpg',
|
||||||
|
'moons/enceladus.jpg',
|
||||||
|
'moons/mimas.jpg',
|
||||||
|
]
|
||||||
|
|
||||||
|
def frozen():
|
||||||
|
import imp
|
||||||
|
return (hasattr(sys, "frozen") or # new py2exe
|
||||||
|
hasattr(sys, "importers") # old py2exe
|
||||||
|
or imp.is_frozen("__main__")) # tools/freeze
|
||||||
|
|
||||||
|
def get_main_dir():
|
||||||
|
if frozen():
|
||||||
|
return os.path.dirname(sys.executable)
|
||||||
|
return os.path.dirname(__file__)
|
||||||
|
|
||||||
|
def main():
|
||||||
|
texture = os.path.join(get_main_dir(), 'punyverse', 'assets', 'textures')
|
||||||
|
for file in textures:
|
||||||
|
print 'Resizing %s:' % file,
|
||||||
|
file = os.path.join(texture, file.replace('/', os.sep))
|
||||||
|
shrink(file)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|