def init_animation_loader(self, use_atlases):
# this method should not be called from init to catch any bugs caused by the loader changing after it.
self.use_atlases = use_atlases
if self.use_atlases:
self.animationloader = SQLiteAtlasLoader()
else:
self.animationloader = SQLiteAnimationLoader()
def _loadObjects(self): # get fifedit objects self.model = self._engine.getModel() self.all_action_sets = ActionSetLoader.get_sets() TileSetLoader.load() ActionSetLoader.load() self.animationloader = SQLiteAnimationLoader() self._loadGroundTiles() self._loadBuildings()
class UHObjectLoader(scripts.plugin.Plugin):
""" The B{UHObjectLoader} allows to load the UH objects into the FIFEdit object selector """
def __init__(self):
# Editor instance
self._editor = None
# Plugin variables
self._enabled = False
# Current mapview
self._mapview = None
#--- Plugin functions ---#
def enable(self):
""" Enable plugin """
if self._enabled is True:
return
# Fifedit plugin data
self._editor = scripts.editor.getEditor()
self._engine = self._editor.getEngine()
# load UH objects
self._fixupFife()
self._fixupHorizons()
self._loadObjects()
def disable(self):
""" Disable plugin """
if self._enabled is False:
return
def isEnabled(self):
""" Returns True if plugin is enabled """
return self._enabled;
def getName(self):
""" Return plugin name """
return u"UHObjectLoader"
#--- End plugin functions ---#
def _fixupHorizons(self):
"""Fixes some UH quirks that have to do with globals"""
class PatchedFife:
imagemanager = self._engine.getImageManager()
use_atlases = False
pass
horizons.globals.fife = PatchedFife()
uh_path = util.getUHPath()
horizons.main.PATHS.TILE_SETS_DIRECTORY = os.path.join(uh_path, horizons.main.PATHS.TILE_SETS_DIRECTORY)
horizons.main.PATHS.ACTION_SETS_DIRECTORY = os.path.join(uh_path, horizons.main.PATHS.ACTION_SETS_DIRECTORY)
horizons.main.PATHS.DB_FILES = map(lambda file: os.path.join(uh_path, file), horizons.main.PATHS.DB_FILES)
def _fixupFife(self):
"""Fixes some FIFE quirks that have to do with VFS"""
vfs = self._engine.getVFS()
vfs.addNewSource(util.getUHPath())
vfs.addNewSource("/")
def _loadObjects(self):
# get fifedit objects
self.model = self._engine.getModel()
self.all_action_sets = ActionSetLoader.get_sets()
TileSetLoader.load()
ActionSetLoader.load()
self.animationloader = SQLiteAnimationLoader()
self._loadGroundTiles()
self._loadBuildings()
def _loadGroundTiles(self):
print("loading UH ground tiles...")
tile_sets = TileSetLoader.get_sets()
for tile_set_id in tile_sets:
tile_set = tile_sets[tile_set_id]
object = self.model.createObject(str(tile_set_id), util.GROUND_NAMESPACE)
fife.ObjectVisual.create(object)
# load animations
for action_id in tile_sets[tile_set_id].iterkeys():
action = object.createAction(action_id+"_"+str(tile_set_id))
fife.ActionVisual.create(action)
for rotation in tile_sets[tile_set_id][action_id].iterkeys():
anim = self.animationloader.loadResource( \
str(tile_set_id)+"+"+str(action_id)+"+"+ \
str(rotation) + ':shift:center+0,bottom+8')
action.get2dGfxVisual().addAnimation(int(rotation), anim)
action.setDuration(anim.getDuration())
def _loadBuildings(self):
print("loading UH buildings...")
for root, dirnames, filenames in os.walk(util.getUHPath() + '/content/objects/buildings'):
for filename in fnmatch.filter(filenames, '*.yaml'):
# This is needed for dict lookups! Do not convert to os.join!
full_file = root + "/" + filename
result = YamlCache.get_file(full_file, game_data=True)
result['yaml_file'] = full_file
self._loadBuilding(result)
print("finished loading UH objects")
def _loadBuilding(self, yaml):
id = int(yaml['id'])
name_data = yaml['name']
if isinstance(name_data, dict): # level-sensitive names as dict
name_data = sorted(name_data.iteritems())[0][1]
name = name_data[2:] # drop translation mark
size_x = yaml['size_x']
size_y = yaml['size_y']
action_sets = yaml['actionsets']
object = self.model.createObject(str(name), util.BUILDING_NAMESPACE)
fife.ObjectVisual.create(object)
main_action = 'idle'
for action_set_list in action_sets.itervalues():
for action_set_id in action_set_list.iterkeys():
for action_id in self.all_action_sets[action_set_id].iterkeys():
main_action = action_id+"_"+str(action_set_id)
action = object.createAction(main_action)
fife.ActionVisual.create(action)
for rotation in self.all_action_sets[action_set_id][action_id].iterkeys():
if rotation == 45:
command = 'left-32,bottom+' + str(size_x * 16)
elif rotation == 135:
command = 'left-' + str(size_y * 32) + ',bottom+16'
elif rotation == 225:
command = 'left-' + str((size_x + size_y - 1) * 32) + ',bottom+' + str(size_y * 16)
elif rotation == 315:
command = 'left-' + str(size_x * 32) + ',bottom+' + str((size_x + size_y - 1) * 16)
else:
assert False, "Bad rotation for action_set %(id)s: %(rotation)s for action: %(action_id)s" % \
{ 'id': action_set_id, 'rotation': rotation, 'action_id': action_id }
anim = self.animationloader.loadResource(str(action_set_id)+"+"+str(action_id)+"+"+str(rotation) + ':shift:' + command)
action.get2dGfxVisual().addAnimation(int(rotation), anim)
action.setDuration(anim.getDuration())
util.addBuilding(id, name, main_action)