forked from fogleman/Craft
-
Notifications
You must be signed in to change notification settings - Fork 1
/
world.py
48 lines (40 loc) · 1.55 KB
/
world.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# gcc -std=c99 -O3 -shared -o world \
# -I src -I deps/noise deps/noise/noise.c src/world.c
from ctypes import CDLL, CFUNCTYPE, c_float, c_int, c_void_p
from collections import OrderedDict
dll = CDLL('./world')
WORLD_FUNC = CFUNCTYPE(None, c_int, c_int, c_int, c_int, c_void_p)
def dll_seed(x):
dll.seed(x)
def dll_create_world(p, q):
result = {}
def world_func(x, y, z, w, arg):
result[(x, y, z)] = w
dll.create_world(p, q, WORLD_FUNC(world_func), None)
return result
dll.simplex2.restype = c_float
dll.simplex2.argtypes = [c_float, c_float, c_int, c_float, c_float]
def dll_simplex2(x, y, octaves=1, persistence=0.5, lacunarity=2.0):
return dll.simplex2(x, y, octaves, persistence, lacunarity)
dll.simplex3.restype = c_float
dll.simplex3.argtypes = [c_float, c_float, c_float, c_int, c_float, c_float]
def dll_simplex3(x, y, z, octaves=1, persistence=0.5, lacunarity=2.0):
return dll.simplex3(x, y, z, octaves, persistence, lacunarity)
class World(object):
def __init__(self, seed=None, cache_size=64):
self.seed = seed
self.cache = OrderedDict()
self.cache_size = cache_size
def create_chunk(self, p, q):
if self.seed is not None:
dll_seed(self.seed)
return dll_create_world(p, q)
def get_chunk(self, p, q):
try:
chunk = self.cache.pop((p, q))
except KeyError:
chunk = self.create_chunk(p, q)
self.cache[(p, q)] = chunk
if len(self.cache) > self.cache_size:
self.cache.popitem(False)
return chunk