def run(self):
# Initialise variables
self.physics = False
self.physics_engine = Physics(self)
self.raw_blocks = None # Read this from any thread but please update through TASK_BLOCKSET so queued_blocks is updated
self.running = True
self.unflooding = False
self.finite_water = False
self.queued_blocks = {} # Blocks which need to be flushed into the file.
self.create_raw_blocks()
# Start physics engine
self.physics_engine.start()
# Main eval loop
while self.running:
try:
# Pop something off the queue
task = self.in_queue.get()
# If we've been asked to flush, do so, and say we did.
if task[0] is TASK_FLUSH:
self.flush()
self.out_queue.put([TASK_FLUSH])
# New block?
elif task[0] is TASK_BLOCKSET:
try:
self[task[1]] = task[2]
if len(task) == 4 and task[3] == True:
# Tells the server to update the given block for clients.
self.out_queue.put([TASK_BLOCKSET, (task[1][0], task[1][1], task[1][2], task[2])])
except AssertionError:
self.logger.warning("Tried to set a block at %s in %s!" % (task[1], self.world_name))
# Asking for a block?
elif task[0] is TASK_BLOCKGET:
self.out_queue.put([TASK_BLOCKGET, task[1], self[task[1]]])
# Perhaps physics was enabled?
elif task[0] is TASK_PHYSICSOFF:
self.logger.debug("Disabling physics on '%s'..." % self.world_name)
self.disable_physics()
# Or disabled?
elif task[0] is TASK_PHYSICSON:
self.logger.debug("Enabling physics on '%s'..." % self.world_name)
self.enable_physics()
# I can haz finite water tiem?
elif task[0] is TASK_FWATERON:
self.logger.debug("Enabling finite water on '%s'..." % self.world_name)
self.finite_water = True
# Noes, no more finite water.
elif task[0] is TASK_FWATEROFF:
self.logger.debug("Disabling finite water on '%s'..." % self.world_name)
self.finite_water = False
# Do they need to do a Moses?
elif task[0] is TASK_UNFLOOD:
self.logger.debug("Unflood started on '%s'..." % self.world_name)
self.unflooding = True
# Perhaps that's it, and we need to stop?
elif task[0] is TASK_STOP:
self.logger.debug("Stopping block store '%s'..." % self.world_name)
self.physics_engine.stop()
self.flush()
self.logger.debug("Stopped block store '%s'." % self.world_name)
return
# ???
else:
raise ValueError("Unknown BlockStore task: %s" % task)
except (KeyboardInterrupt, IOError):
pass
class BlockStore(Thread):
"""
A class which deals with storing the block worlds, flushing them, etc.
"""
def __init__(self, blocks_path, sx, sy, sz):
Thread.__init__(self)
self.x, self.y, self.z = sx, sy, sz
self.blocks_path = blocks_path
self.world_name = os.path.basename(os.path.dirname(blocks_path))
self.in_queue = Queue()
self.out_queue = Queue()
self.logger = ColouredLogger()
def run(self):
# Initialise variables
self.physics = False
self.physics_engine = Physics(self)
self.raw_blocks = None # Read this from any thread but please update through TASK_BLOCKSET so queued_blocks is updated
self.running = True
self.unflooding = False
self.finite_water = False
self.queued_blocks = {} # Blocks which need to be flushed into the file.
self.create_raw_blocks()
# Start physics engine
self.physics_engine.start()
# Main eval loop
while self.running:
try:
# Pop something off the queue
task = self.in_queue.get()
# If we've been asked to flush, do so, and say we did.
if task[0] is TASK_FLUSH:
self.flush()
self.out_queue.put([TASK_FLUSH])
# New block?
elif task[0] is TASK_BLOCKSET:
try:
self[task[1]] = task[2]
if len(task) == 4 and task[3] == True:
# Tells the server to update the given block for clients.
self.out_queue.put([TASK_BLOCKSET, (task[1][0], task[1][1], task[1][2], task[2])])
except AssertionError:
self.logger.warning("Tried to set a block at %s in %s!" % (task[1], self.world_name))
# Asking for a block?
elif task[0] is TASK_BLOCKGET:
self.out_queue.put([TASK_BLOCKGET, task[1], self[task[1]]])
# Perhaps physics was enabled?
elif task[0] is TASK_PHYSICSOFF:
self.logger.debug("Disabling physics on '%s'..." % self.world_name)
self.disable_physics()
# Or disabled?
elif task[0] is TASK_PHYSICSON:
self.logger.debug("Enabling physics on '%s'..." % self.world_name)
self.enable_physics()
# I can haz finite water tiem?
elif task[0] is TASK_FWATERON:
self.logger.debug("Enabling finite water on '%s'..." % self.world_name)
self.finite_water = True
# Noes, no more finite water.
elif task[0] is TASK_FWATEROFF:
self.logger.debug("Disabling finite water on '%s'..." % self.world_name)
self.finite_water = False
# Do they need to do a Moses?
elif task[0] is TASK_UNFLOOD:
self.logger.debug("Unflood started on '%s'..." % self.world_name)
self.unflooding = True
# Perhaps that's it, and we need to stop?
elif task[0] is TASK_STOP:
self.logger.debug("Stopping block store '%s'..." % self.world_name)
self.physics_engine.stop()
self.flush()
self.logger.debug("Stopped block store '%s'." % self.world_name)
return
# ???
else:
raise ValueError("Unknown BlockStore task: %s" % task)
except (KeyboardInterrupt, IOError):
pass
def enable_physics(self):
"Turns on physics"
self.physics = True
def disable_physics(self):
"Disables physics, and clears the in-memory store."
self.physics = False
def create_raw_blocks(self):
"Reads in the gzipped data into a raw array"
# Open the blocks file
fh = gzip.GzipFile(self.blocks_path)
self.raw_blocks = array('c')
# Read off the size header
fh.read(4)
# Copy into the array in chunks
chunk = fh.read(2048)
while chunk:
self.raw_blocks.extend(chunk)
chunk = fh.read(2048)
fh.close()
def get_offset(self, x, y, z):
"Turns block coordinates into a data offset"
assert 0 <= x < self.x
assert 0 <= y < self.y
assert 0 <= z < self.z
return y * (self.x * self.z) + z * (self.x) + x
def get_coords(self, offset):
"Turns a data offset into coordinates"
x = offset % self.x
z = (offset // self.x) % self.z
y = offset // (self.x * self.z)
return x, y, z
def message(self, message):
"Sends a message out to users about this World."
self.out_queue.put([TASK_MESSAGE, message])
def __setitem__(self, (x, y, z), block):
"Set a block in this level to the given value."
assert isinstance(block, str) and len(block) == 1
# Save to queued blocks
offset = self.get_offset(x, y, z)
self.queued_blocks[offset] = block
# And directly to raw blocks, if we must
if self.raw_blocks:
self.raw_blocks[offset] = block
# Ask the physics engine if they'd like a look at that
self.physics_engine.handle_change(offset, block)
