def run_gen(self, seed):
tgen.initialize(seed, self.data_path)
self.tgen_init = True
while True:
data = self.gen_queue.get()
if data is None:
break
func, args, f = data
self.print_debug(*data)
try:
res = func(*args)
def set_res(f, res):
if f.cancelled():
return
f.set_result(res)
self.loop.call_soon_threadsafe(set_res, f, res)
except Exception as e:
def set_exc(f, e):
if f.cancelled():
return
f.set_exception(e)
self.loop.call_soon_threadsafe(set_exc, f, e)
def run_gen(self, seed):
tgen.initialize(seed, self.data_path)
while True:
data = self.gen_queue.get()
if data is None:
break
pos, f = data
res = tgen.generate(*pos)
self.loop.call_soon_threadsafe(f.set_result, res)
def run_gen(self, seed):
tgen.initialize(seed, self.data_path)
while True:
data = self.gen_queue.get()
if data is None:
break
pos, f = data
res = tgen.generate(*pos)
self.loop.call_soon_threadsafe(f.set_result, res)
def run_gen(self, seed):
tgen.initialize(seed)
while True:
data = self.gen_queue.get()
if data is None:
break
key = (data.x, data.y)
res = self.cache.get(key, None)
if res is None:
res = tgen.generate(data.x, data.y)
self.cache[key] = res
reactor.callFromThread(data.d.callback, res)
def run(self):
print('Initializing tgen...')
tgen.initialize(self.parent.seed, './data/')
while self.running:
key = self.gen_queue.get()
x, y = key
# print 'Generating chunk', x, y
off_x = (x - self.parent.chunk_x) * 256.0
off_y = (y - self.parent.chunk_y) * 256.0
data = tgen.generate(x, y).get_render(off_x, off_y)
res = ChunkData(data)
self.cache[key] = res
def run(self):
print('Initializing tgen...')
tgen.initialize(self.parent.seed, './data/')
while self.running:
key = self.gen_queue.get()
x, y = key
# print 'Generating chunk', x, y
off_x = (x - self.parent.chunk_x) * 256.0
off_y = (y - self.parent.chunk_y) * 256.0
data = tgen.generate(x, y).get_render(off_x, off_y)
res = ChunkData(data)
self.cache[key] = res
def run_gen(self, seed):
tgen.initialize(seed, self.data_path)
self.tgen_init = True
while True:
data = self.gen_queue.get()
if data is None:
break
pos, f = data
res = tgen.generate(*pos)
def set_result():
try:
f.set_result(res)
except asyncio.InvalidStateError:
return
self.loop.call_soon_threadsafe(set_result)
def main():
parser = argparse.ArgumentParser(description='Generates chunks using tgen')
parser.add_argument('seed', help='input seed', type=int)
parser.add_argument('coords', help='input coordinates',
type=coords, nargs='+')
args = parser.parse_args()
print('Initializing...')
val = './data/'
tgen.initialize(args.seed, val)
for coord in args.coords:
x, y = coord
f = '%s_%s.qmo' % (x, y)
convert_pos(x, y, f)
print('Done')
def run_gen(self, seed):
tgen.initialize(seed, self.data_path)
self.tgen_init = True
while True:
data = self.gen_queue.get()
if data is None:
break
pos, f = data
res = tgen.generate(*pos)
def set_result():
try:
f.set_result(res)
except asyncio.InvalidStateError:
return
self.loop.call_soon_threadsafe(set_result)
def main():
parser = argparse.ArgumentParser(description='Generates chunks using tgen')
parser.add_argument('seed', help='input seed', type=int)
parser.add_argument('coords',
help='input coordinates',
type=coords,
nargs='+')
args = parser.parse_args()
print('Initializing...')
val = './data/'
tgen.initialize(args.seed, val)
for coord in args.coords:
x, y = coord
f = '%s_%s.qmo' % (x, y)
convert_pos(x, y, f)
print('Done')
def run_gen(self, seed):
tgen.initialize(seed, self.data_path)
self.tgen_init = True
while True:
data = self.gen_queue.get()
if data is None:
break
func, args, f = data
self.print_debug(*data)
try:
res = func(*args)
def set_res(f, res):
if f.cancelled():
return
f.set_result(res)
self.loop.call_soon_threadsafe(set_res, f, res)
except Exception as e:
def set_exc(f, e):
if f.cancelled():
return
f.set_exception(e)
self.loop.call_soon_threadsafe(set_exc, f, e)
def setUp(self):
print('Initializing tgen')
tgen.initialize(26879, './data/')
print('Done')
def main():
structs = get_structs()
out = FormattedOutput('Entity data read/write')
out.putln('cimport cython')
out.putln('from libc.stdint cimport int64_t, uint64_t')
out.putln('from cuwo.bytes cimport ByteReader, ByteWriter')
out.putln('from cuwo.vector import Vector3')
out.putln('from cuwo import strings')
out.putln()
out.putln()
for struct in structs.values():
if struct.defs:
for d in struct.defs:
name, val = d
out.putln('%s = %s' % (name, val))
if not name.endswith('_BIT'):
continue
val = 1 << eval(val)
name = '%s_FLAG' % name.replace('_BIT', '')
out.putln('%s = %s' % (name, hex(val)))
out.putln()
out.putln()
# out.putln('@cython.final')
out.putln('cdef class %s:' % struct.name)
out.indent()
# reset func
# out.putln('def reset(self):')
# out.indent()
# for attr in struct.attrs:
# if attr.default is None:
# continue
# if attr.dim and attr.typ != 'int8':
# out.putln('self.%s = [%s] * %s' % (attr.name, attr.default,
# attr.dim))
# else:
# out.putln('self.%s = %s' % (attr.name, attr.default))
# out.putln('pass')
# out.dedent()
# out.putln()
out.putln('cdef public:')
out.indent()
has_cinit = struct.name == 'EntityData'
lists = []
objects = []
defaults = []
for attr in struct.attrs:
name, typ, dim, spec = attr.get()
if name == 'pad':
continue
if dim:
if typ == 'int8':
typ = 'str'
defaults.append((name, "''"))
else:
has_cinit = True
lists.append((typ, name, dim))
typ = 'list'
else:
if attr.default:
defaults.append((name, attr.default))
has_cinit = True
elif typ in PYTHON_OBJECTS or spec:
has_cinit = True
objects.append((typ, name))
typ = CYTHON_TYPES.get(typ, typ)
out.putln('%s %s' % (typ, name))
if struct.name == 'EntityData':
out.putln('uint64_t mask')
out.dedent()
out.putln()
if has_cinit:
out.putln('def __cinit__(self):')
out.indent()
if struct.name == 'EntityData':
out.putln('self.mask = 0')
for (typ, name, dim) in lists:
out.putln('self.%s = []' % name)
out.putln('for _ in range(%s):' % dim)
out.indent()
if typ in DEFAULTS:
value = DEFAULTS[typ]
else:
value = '%s.__new__(%s)' % (typ, typ)
out.putln('self.%s.append(%s)' % (name, value))
out.dedent()
for (typ, name) in objects:
if typ in DEFAULTS:
value = DEFAULTS[typ]
else:
value = '%s.__new__(%s)' % (typ, typ)
out.putln('self.%s = %s' % (name, value))
for (name, default) in defaults:
out.putln('self.%s = %s' % (name, default))
out.dedent()
out.putln()
# read func
out.putln('cpdef read(self, ByteReader reader):')
out.indent()
for attr in struct.attrs:
name, typ, dim, spec = attr.get()
if name == 'pad':
out.putln('reader.skip(%s)' % dim)
continue
if dim is not None and typ == 'int8':
out.putln('self.%s = reader.read_ascii(%s)' % (name, dim))
continue
if dim is not None:
out.putln('for i in range(%s):' % dim)
out.indent()
if spec:
out.putln('(<%s>self.%s[i]).read(reader)' % (typ, name))
else:
out.putln('self.%s[i] = reader.read_%s()' % (name, typ))
out.dedent()
continue
if spec:
out.putln('self.%s.read(reader)' % name)
else:
out.putln('self.%s = reader.read_%s()' % (name, typ))
out.dedent()
out.putln()
# write func
out.putln('cpdef write(self, ByteWriter writer):')
out.indent()
for attr in struct.attrs:
name, typ, dim, spec = attr.get()
if name == 'pad':
out.putln('writer.pad(%s)' % dim)
continue
if dim is not None and typ == 'int8':
out.putln('writer.write_ascii(self.%s, %s)' % (name, dim))
continue
if dim is not None:
out.putln('for item in self.%s:' % name)
out.indent()
if spec:
out.putln('item.write(writer)')
else:
out.putln('writer.write_%s(item)' % typ)
out.dedent()
continue
if spec:
out.putln('self.%s.write(writer)' % name)
else:
out.putln('writer.write_%s(self.%s)' % (typ, name))
out.dedent()
if struct.name == 'EntityData':
# setters
bits = {}
for d in struct.defs:
orig_name, value = d
if not orig_name.endswith('_BIT'):
continue
value = int(value)
name = orig_name[:-4].lower()
bits[value] = name
for index in sorted(bits):
name = bits[index]
elif struct.name == 'AppearanceData':
for attr in struct.attrs:
if attr.name.endswith('_model'):
name = attr.name.split('_')[0]
out.putln()
out.putln('def get_%s(self):' % name)
out.indent()
out.putln('return strings.MODEL_NAMES[self.%s]' %
attr.name)
out.dedent()
out.putln()
out.putln('def set_%s(self, name):' % name)
out.indent()
out.putln('self.%s = strings.MODEL_IDS[name]' % attr.name)
out.dedent()
out.dedent()
out.putln()
out.putln()
# mask data
mask_data = []
mask_input = open('masked_read.h', 'rb').read().decode('utf-8')
mask_input = mask_input.splitlines()
entity_struct = structs['EntityData']
for bit_index, line in enumerate(mask_input):
start = line.find('read_') + 5
end = line.find('_masked', start)
size = int(line[start:end])
attr_start = line.find('->')
if attr_start == -1:
attr = 'x'
else:
attr_end = line.find(');')
attr = line[attr_start + 2:attr_end]
offset = 0
for struct_attr in entity_struct.attrs:
name, typ, dim, spec = struct_attr.get()
if name == attr:
break
offset += 1
else:
print(attr)
raise NotImplementedError()
mask_data.append((offset, size))
for index in sorted(bits):
name = bits[index]
func_name = 'is_%s_set' % name
out.putln('cpdef inline bint %s(uint64_t mask):' % func_name)
out.indent()
out.putln('return %s' % get_mask_condition(index))
out.dedent()
out.putln()
out.putln()
# read masked data
out.putln('cpdef uint64_t read_masked_data(EntityData entity, '
'ByteReader reader):')
out.indent()
out.putln('cdef uint64_t mask = reader.read_uint64()')
for bit_index, v in enumerate(mask_data):
offset, size = v
out.putln('if %s:' % get_mask_condition(bit_index))
out.indent()
name, typ, dim, spec = entity_struct.attrs[offset].get()
if spec:
if dim:
out.putln('for item in entity.%s:' % name)
out.indent()
out.putln('(<%s>item).read(reader)' % typ)
out.dedent()
else:
out.putln('entity.%s.read(reader)' % name)
else:
while size > 0:
name, typ, dim, spec = entity_struct.attrs[offset].get()
if typ == 'int8' and dim is not None:
out.putln('entity.%s = reader.read_ascii(%s)' %
(name, dim))
elif dim is not None:
out.putln('entity.%s = []' % name)
out.putln('for _ in xrange(%s):' % dim)
out.indent()
out.putln('entity.%s.append(reader.read_%s())' %
(name, typ))
out.dedent()
else:
out.putln('entity.%s = reader.read_%s()' % (name, typ))
size -= TYPE_SIZE[typ] * (dim or 1)
offset += 1
if size != 0:
raise NotImplementedError()
out.dedent()
out.putln()
out.putln('return mask')
out.dedent()
out.putln()
out.putln()
# get masked size
out.putln('cpdef unsigned int get_masked_size(uint64_t mask):')
out.indent()
out.putln('cdef unsigned int size = 0')
for bit_index, v in enumerate(mask_data):
offset, size = v
out.putln('if %s:' % get_mask_condition(bit_index))
out.indent()
out.putln('size += %s' % size)
out.dedent()
out.putln('return size')
out.dedent()
out.putln()
out.putln()
# write masked data (send everything)
out.putln('cpdef write_masked_data(EntityData entity, ByteWriter writer, '
'uint64_t mask):')
out.indent()
out.putln('writer.write_uint64(mask)')
for bit_index, v in enumerate(mask_data):
out.putln('if %s:' % get_mask_condition(bit_index))
out.indent()
offset, size = v
name, typ, dim, spec = entity_struct.attrs[offset].get()
if spec:
if dim:
out.putln('for item in entity.%s:' % name)
out.indent()
out.putln('(<%s>item).write(writer)' % typ)
out.dedent()
else:
out.putln('entity.%s.write(writer)' % name)
else:
while size > 0:
name, typ, dim, spec = entity_struct.attrs[offset].get()
if typ == 'int8' and dim is not None:
out.putln('writer.write_ascii(entity.%s, %s)' %
(name, dim))
elif dim is not None:
out.putln('for item in entity.%s:' % name)
out.indent()
out.putln('writer.write_%s(item)' % (typ))
out.dedent()
else:
out.putln('writer.write_%s(entity.%s)' % (typ, name))
size -= TYPE_SIZE[typ] * (dim or 1)
offset += 1
out.dedent()
out.dedent()
open('../cuwo/entity.pyx', 'wb').write(out.get())
# sounds
sound_input = open('sounds.h', 'rb').read().decode('utf-8')
sound_input = sound_input.splitlines()
sounds = {}
current_index = None
for line in sound_input:
if 'case ' in line:
start = line.find('case ') + 5
end = line.find(':', start)
current_index = int(line[start:end])
elif '.wav"' in line:
end = line.find('.wav"')
start = line.rfind('"', 0, end) + 1
sound_name = line[start:end]
sounds[current_index] = sound_name
# models
model_input = open('models.h', 'rb').read().decode('utf-8')
model_input = model_input.splitlines()
models = {-1: None}
current_name = None
for line in model_input:
if 'std_string_append' in line:
if current_name is not None:
print(line)
raise NotImplementedError()
start = line.rfind(', ') + 2
line = line[start:-2]
if line.startswith('(const char *)'):
line = line.replace('(const char *)', '')
if line in ('"cubequest4.cub"', '&byte_135B530', '"data1.db"'):
continue
name, ext = eval(line).split('.')
current_name = name
elif current_name is None:
continue
model_id = None
if line.endswith('];'):
start = line.rfind('[') + 1
model_id = int(line[start:-2])
elif 'this_4pointer_off' in line:
start = line.rfind(', ') + 2
model_id = int(line[start:-2])
else:
continue
models[model_id] = current_name
current_name = None
out = FormattedOutput('Constant string definitions')
out.write_dict('SOUND_NAMES', sounds)
out.write_inverse_dict('SOUND_IDS', 'SOUND_NAMES')
out.write_dict('MODEL_NAMES', models)
out.write_inverse_dict('MODEL_IDS', 'MODEL_NAMES')
print('Writing defs from tgen')
tgen.initialize(1234, '../data/')
out.write_dict('ITEM_NAMES', tgen.get_item_names())
out.write_inverse_dict('ITEM_IDS', 'ITEM_NAMES')
out.write_dict('STATIC_NAMES', tgen.get_static_names())
out.write_inverse_dict('STATIC_IDS', 'STATIC_NAMES')
out.write_dict('ENTITY_NAMES', tgen.get_entity_names())
out.write_inverse_dict('ENTITY_IDS', 'ENTITY_NAMES')
out.write_dict('LOCATION_NAMES', tgen.get_location_names())
out.write_inverse_dict('LOCATION_IDS', 'LOCATION_NAMES')
out.write_dict('QUARTER_NAMES', tgen.get_quarter_names())
out.write_inverse_dict('QUARTER_IDS', 'QUARTER_NAMES')
def remove(value, rem):
new_dict = {}
for k, v in value.items():
new_dict[k] = v.replace(rem, '')
return new_dict
out.write_dict('SKILL_NAMES', remove(tgen.get_skill_names(), 'Skill'))
out.write_inverse_dict('SKILL_IDS', 'SKILL_NAMES')
out.write_dict('ABILITY_NAMES', remove(tgen.get_ability_names(),
'Ability'))
out.write_inverse_dict('ABILITY_IDS', 'ABILITY_NAMES')
print('Done')
open('../cuwo/strings.py', 'wb').write(out.get())
def main():
structs = get_structs()
out = FormattedOutput('Entity data read/write')
out.putln('cimport cython')
out.putln('from libc.stdint cimport int64_t, uint64_t')
out.putln('from cuwo.bytes cimport ByteReader, ByteWriter')
out.putln('from cuwo.vector import Vector3')
out.putln('from cuwo import strings')
out.putln()
out.putln()
for struct in structs.values():
if struct.defs:
for d in struct.defs:
name, val = d
out.putln('%s = %s' % (name, val))
if not name.endswith('_BIT'):
continue
val = 1 << eval(val)
name = '%s_FLAG' % name.replace('_BIT', '')
out.putln('%s = %s' % (name, hex(val)))
out.putln()
out.putln()
# out.putln('@cython.final')
out.putln('cdef class %s:' % struct.name)
out.indent()
# reset func
# out.putln('def reset(self):')
# out.indent()
# for attr in struct.attrs:
# if attr.default is None:
# continue
# if attr.dim and attr.typ != 'int8':
# out.putln('self.%s = [%s] * %s' % (attr.name, attr.default,
# attr.dim))
# else:
# out.putln('self.%s = %s' % (attr.name, attr.default))
# out.putln('pass')
# out.dedent()
# out.putln()
out.putln('cdef public:')
out.indent()
has_cinit = struct.name == 'EntityData'
lists = []
objects = []
defaults = []
for attr in struct.attrs:
name, typ, dim, spec = attr.get()
if name == 'pad':
continue
if dim:
if typ == 'int8':
typ = 'str'
defaults.append((name, "''"))
else:
has_cinit = True
lists.append((typ, name, dim))
typ = 'list'
else:
if attr.default:
defaults.append((name, attr.default))
has_cinit = True
elif typ in PYTHON_OBJECTS or spec:
has_cinit = True
objects.append((typ, name))
typ = CYTHON_TYPES.get(typ, typ)
out.putln('%s %s' % (typ, name))
if struct.name == 'EntityData':
out.putln('uint64_t mask')
out.dedent()
out.putln()
if has_cinit:
out.putln('def __cinit__(self):')
out.indent()
if struct.name == 'EntityData':
out.putln('self.mask = 0')
for (typ, name, dim) in lists:
out.putln('self.%s = []' % name)
out.putln('for _ in range(%s):' % dim)
out.indent()
if typ in DEFAULTS:
value = DEFAULTS[typ]
else:
value = '%s.__new__(%s)' % (typ, typ)
out.putln('self.%s.append(%s)' % (name, value))
out.dedent()
for (typ, name) in objects:
if typ in DEFAULTS:
value = DEFAULTS[typ]
else:
value = '%s.__new__(%s)' % (typ, typ)
out.putln('self.%s = %s' % (name, value))
for (name, default) in defaults:
out.putln('self.%s = %s' % (name, default))
out.dedent()
out.putln()
# read func
out.putln('cpdef read(self, ByteReader reader):')
out.indent()
for attr in struct.attrs:
name, typ, dim, spec = attr.get()
if name == 'pad':
out.putln('reader.skip(%s)' % dim)
continue
if dim is not None and typ == 'int8':
out.putln('self.%s = reader.read_ascii(%s)' % (name, dim))
continue
if dim is not None:
out.putln('for i in range(%s):' % dim)
out.indent()
if spec:
out.putln('(<%s>self.%s[i]).read(reader)' % (typ, name))
else:
out.putln('self.%s[i] = reader.read_%s()' % (name, typ))
out.dedent()
continue
if spec:
out.putln('self.%s.read(reader)' % name)
else:
out.putln('self.%s = reader.read_%s()' % (name, typ))
out.dedent()
out.putln()
# write func
out.putln('cpdef write(self, ByteWriter writer):')
out.indent()
for attr in struct.attrs:
name, typ, dim, spec = attr.get()
if name == 'pad':
out.putln('writer.pad(%s)' % dim)
continue
if dim is not None and typ == 'int8':
out.putln('writer.write_ascii(self.%s, %s)' % (name, dim))
continue
if dim is not None:
out.putln('for item in self.%s:' % name)
out.indent()
if spec:
out.putln('item.write(writer)')
else:
out.putln('writer.write_%s(item)' % typ)
out.dedent()
continue
if spec:
out.putln('self.%s.write(writer)' % name)
else:
out.putln('writer.write_%s(self.%s)' % (typ, name))
out.dedent()
if struct.name == 'EntityData':
# setters
bits = {}
for d in struct.defs:
orig_name, value = d
if not orig_name.endswith('_BIT'):
continue
value = int(value)
name = orig_name[:-4].lower()
bits[value] = name
for index in sorted(bits):
name = bits[index]
elif struct.name == 'AppearanceData':
for attr in struct.attrs:
if attr.name.endswith('_model'):
name = attr.name.split('_')[0]
out.putln()
out.putln('def get_%s(self):' % name)
out.indent()
out.putln('return strings.MODEL_NAMES[self.%s]'
% attr.name)
out.dedent()
out.putln()
out.putln('def set_%s(self, name):' % name)
out.indent()
out.putln('self.%s = strings.MODEL_IDS[name]' % attr.name)
out.dedent()
out.dedent()
out.putln()
out.putln()
# mask data
mask_data = []
mask_input = open('masked_read.h', 'rb').read().decode('utf-8')
mask_input = mask_input.splitlines()
entity_struct = structs['EntityData']
for bit_index, line in enumerate(mask_input):
start = line.find('read_') + 5
end = line.find('_masked', start)
size = int(line[start:end])
attr_start = line.find('->')
if attr_start == -1:
attr = 'x'
else:
attr_end = line.find(');')
attr = line[attr_start+2:attr_end]
offset = 0
for struct_attr in entity_struct.attrs:
name, typ, dim, spec = struct_attr.get()
if name == attr:
break
offset += 1
else:
print(attr)
raise NotImplementedError()
mask_data.append((offset, size))
for index in sorted(bits):
name = bits[index]
func_name = 'is_%s_set' % name
out.putln('cpdef inline bint %s(uint64_t mask):' % func_name)
out.indent()
out.putln('return %s' % get_mask_condition(index))
out.dedent()
out.putln()
out.putln()
# read masked data
out.putln('cpdef uint64_t read_masked_data(EntityData entity, '
'ByteReader reader):')
out.indent()
out.putln('cdef uint64_t mask = reader.read_uint64()')
for bit_index, v in enumerate(mask_data):
offset, size = v
out.putln('if %s:' % get_mask_condition(bit_index))
out.indent()
name, typ, dim, spec = entity_struct.attrs[offset].get()
if spec:
if dim:
out.putln('for item in entity.%s:' % name)
out.indent()
out.putln('(<%s>item).read(reader)' % typ)
out.dedent()
else:
out.putln('entity.%s.read(reader)' % name)
else:
while size > 0:
name, typ, dim, spec = entity_struct.attrs[offset].get()
if typ == 'int8' and dim is not None:
out.putln('entity.%s = reader.read_ascii(%s)' % (
name, dim))
elif dim is not None:
out.putln('entity.%s = []' % name)
out.putln('for _ in xrange(%s):' % dim)
out.indent()
out.putln('entity.%s.append(reader.read_%s())' % (name,
typ))
out.dedent()
else:
out.putln('entity.%s = reader.read_%s()' % (name, typ))
size -= TYPE_SIZE[typ] * (dim or 1)
offset += 1
if size != 0:
raise NotImplementedError()
out.dedent()
out.putln()
out.putln('return mask')
out.dedent()
out.putln()
out.putln()
# get masked size
out.putln('cpdef unsigned int get_masked_size(uint64_t mask):')
out.indent()
out.putln('cdef unsigned int size = 0')
for bit_index, v in enumerate(mask_data):
offset, size = v
out.putln('if %s:' % get_mask_condition(bit_index))
out.indent()
out.putln('size += %s' % size)
out.dedent()
out.putln('return size')
out.dedent()
out.putln()
out.putln()
# write masked data (send everything)
out.putln('cpdef write_masked_data(EntityData entity, ByteWriter writer, '
'uint64_t mask):')
out.indent()
out.putln('writer.write_uint64(mask)')
for bit_index, v in enumerate(mask_data):
out.putln('if %s:' % get_mask_condition(bit_index))
out.indent()
offset, size = v
name, typ, dim, spec = entity_struct.attrs[offset].get()
if spec:
if dim:
out.putln('for item in entity.%s:' % name)
out.indent()
out.putln('(<%s>item).write(writer)' % typ)
out.dedent()
else:
out.putln('entity.%s.write(writer)' % name)
else:
while size > 0:
name, typ, dim, spec = entity_struct.attrs[offset].get()
if typ == 'int8' and dim is not None:
out.putln('writer.write_ascii(entity.%s, %s)' % (
name, dim))
elif dim is not None:
out.putln('for item in entity.%s:' % name)
out.indent()
out.putln('writer.write_%s(item)' % (typ))
out.dedent()
else:
out.putln('writer.write_%s(entity.%s)' % (typ, name))
size -= TYPE_SIZE[typ] * (dim or 1)
offset += 1
out.dedent()
out.dedent()
open('../cuwo/entity.pyx', 'wb').write(out.get())
# sounds
sound_input = open('sounds.h', 'rb').read().decode('utf-8')
sound_input = sound_input.splitlines()
sounds = {}
current_index = None
for line in sound_input:
if 'case ' in line:
start = line.find('case ') + 5
end = line.find(':', start)
current_index = int(line[start:end])
elif '.wav"' in line:
end = line.find('.wav"')
start = line.rfind('"', 0, end) + 1
sound_name = line[start:end]
sounds[current_index] = sound_name
# models
model_input = open('models.h', 'rb').read().decode('utf-8')
model_input = model_input.splitlines()
models = {-1: None}
current_name = None
for line in model_input:
if 'std_string_append' in line:
if current_name is not None:
print(line)
raise NotImplementedError()
start = line.rfind(', ') + 2
line = line[start:-2]
if line.startswith('(const char *)'):
line = line.replace('(const char *)', '')
if line in ('"cubequest4.cub"', '&byte_135B530', '"data1.db"'):
continue
name, ext = eval(line).split('.')
current_name = name
elif current_name is None:
continue
model_id = None
if line.endswith('];'):
start = line.rfind('[') + 1
model_id = int(line[start:-2])
elif 'this_4pointer_off' in line:
start = line.rfind(', ') + 2
model_id = int(line[start:-2])
else:
continue
models[model_id] = current_name
current_name = None
out = FormattedOutput('Constant string definitions')
out.write_dict('SOUND_NAMES', sounds)
out.write_inverse_dict('SOUND_IDS', 'SOUND_NAMES')
out.write_dict('MODEL_NAMES', models)
out.write_inverse_dict('MODEL_IDS', 'MODEL_NAMES')
print('Writing defs from tgen')
tgen.initialize(1234, '../data/')
out.write_dict('ITEM_NAMES', tgen.get_item_names())
out.write_inverse_dict('ITEM_IDS', 'ITEM_NAMES')
out.write_dict('STATIC_NAMES', tgen.get_static_names())
out.write_inverse_dict('STATIC_IDS', 'STATIC_NAMES')
out.write_dict('ENTITY_NAMES', tgen.get_entity_names())
out.write_inverse_dict('ENTITY_IDS', 'ENTITY_NAMES')
out.write_dict('LOCATION_NAMES', tgen.get_location_names())
out.write_inverse_dict('LOCATION_IDS', 'LOCATION_NAMES')
out.write_dict('QUARTER_NAMES', tgen.get_quarter_names())
out.write_inverse_dict('QUARTER_IDS', 'QUARTER_NAMES')
def remove(value, rem):
new_dict = {}
for k, v in value.items():
new_dict[k] = v.replace(rem, '')
return new_dict
out.write_dict('SKILL_NAMES', remove(tgen.get_skill_names(), 'Skill'))
out.write_inverse_dict('SKILL_IDS', 'SKILL_NAMES')
out.write_dict('ABILITY_NAMES', remove(tgen.get_ability_names(),
'Ability'))
out.write_inverse_dict('ABILITY_IDS', 'ABILITY_NAMES')
print('Done')
open('../cuwo/strings.py', 'wb').write(out.get())
# Copyright (c) Mathias Kaerlev 2013-2017.
#
# This file is part of cuwo.
#
# cuwo is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# cuwo is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with cuwo. If not, see <http://www.gnu.org/licenses/>.
import sys
sys.path.append('.')
from cuwo import tgen
tgen.initialize(123456, './data/')
tgen.generate(5000, 5000)
def setUp(self):
print('Initializing tgen')
tgen.initialize(26879, './data/')
print('Done')
