def mkdir_p(path):
try:
os.makedirs(path)
except OSError as e:
if e.errno == errno.EEXIST and os.path.isdir(path):
pass
else:
raise
logging.getLogger().setLevel(logging.INFO)
import main
main.init()
for behavior_type in ['skills', 'tactics', 'plays']:
entries = class_import.recursive_import_classes('.', [behavior_type], fsm.StateMachine)
for entry in entries:
try:
klass = entry[1]
module_path = entry[0]
dirpath = 'diagrams/' + '/'.join(module_path[:-1])
mkdir_p(dirpath)
filepath = dirpath + "/" + klass.__name__ + ".png"
klass().write_diagram_png(filepath)
print("generated " + filepath)
except Exception as e:
logging.error("Error generating fsm diagram for behavior '" + klass.__name__ + "':" + str(e))
traceback.print_exc()
def init():
# by default, the logger only shows messages at the WARNING level or greater
logging.getLogger().setLevel(logging.INFO)
global _has_initialized
if _has_initialized:
logging.warn(
"main robocoup python init() method called twice - ignoring")
return
# init root play
global _root_play
import root_play as root_play_module
_root_play = root_play_module.RootPlay()
# init play registry
global _play_registry
_play_registry = play_registry_module.PlayRegistry()
# load all plays
play_classes = class_import.recursive_import_classes(GAMEPLAY_DIR,
['plays'], play.Play)
for entry in play_classes:
# keep in mind that @entry is a tuple
mod_path = entry[0][1:]
_play_registry.insert(mod_path, entry[1])
# this callback lets us do cool stuff when our python files change on disk
def fswatch_callback(event_type, module_path):
# the top-level folders we care about watching
autoloadables = ['plays', 'skills', 'tactics', 'evaluation',
'visualization']
# Don't load if we aren't a special module or if the filename is hidden
if module_path[0] in autoloadables and module_path[-1][0] != '.':
logging.info('.'.join(module_path) + " " + event_type)
is_play = module_path[0] == 'plays'
if event_type == 'created':
if is_play:
# we load the module and register the play class it contains with the play registry
# this makes it automatically show up in the play config tab in the gui
try:
module = importlib.import_module('.'.join(module_path))
except:
logging.error("Error creating module '" + '.'.join(
module_path) + "':")
traceback.print_exc()
return
try:
play_class = class_import.find_subclasses(module,
play.Play)[0]
_play_registry.insert(module_path[1:], play_class) # note: skipping index zero of module_path cuts off the 'plays' part
except IndexError as e:
# we'll get an IndexError exception if the module didn't contain any Plays
# FIXME: instead, we should unload the module and just log a warning
raise Exception(
"Error: python files within the plays directory must contain a subclass of play.Play")
elif event_type == 'modified':
try:
# reload the module
containing_dict = sys.modules
for modname in module_path[:-1]:
containing_dict = containing_dict[modname].__dict__
if module_path[-1] not in containing_dict:
logging.error("failed reloading module '" + '.'.join(
module_path) + "'")
return
module = containing_dict[module_path[-1]]
try:
module = imp.reload(module)
except:
logging.error("Error reloading module '" + '.'.join(
module_path) + "':")
traceback.print_exc()
return
logging.info("reloaded module '" + '.'.join(module_path) +
"'")
# TODO: https://github.com/PyCQA/pylint/issues/1328
#pylint: disable=no-member
if is_play:
# re-register the new play class
# FIXME: this logic should go inside the play_registry
play_reg_node = _play_registry.node_for_module_path(
module_path[1:])
play_reg_node.play_class = class_import.find_subclasses(
module, play.Play)[0]
# _play_registry.modelReset.emit()
# kill currently-running stuff if needed
if not is_play:
_root_play.drop_current_play()
_root_play.drop_goalie_behavior()
elif is_play and root_play != None and root_play.__class__.__name__ == play_reg_node.play_class.__name__:
_root_play.drop_current_play()
except Exception as e:
logging.error("EXCEPTION in file modified event: " + repr(
e))
traceback.print_exc()
raise e
elif event_type == 'deleted':
if is_play:
node = _play_registry.node_for_module_path(module_path[1:])
if node is None:
logging.error("Error removing module '" + '.'.join(
module_path) + "'")
return
if _root_play.play != None and _root_play.play.__class__.__name__ == node.play_class.__name__:
_root_play.drop_current_play()
_play_registry.delete(module_path[1:])
else:
_root_play.drop_current_play()
_root_play.drop_goalie_behavior()
else:
raise AssertionError("Unknown FsWatcher event type: '" +
event_type + "'")
# start up filesystem-watching
watcher = fs_watcher.FsWatcher(GAMEPLAY_DIR)
watcher.subscribe(fswatch_callback)
watcher.start()
_has_initialized = True
def init():
# by default, the logger only shows messages at the WARNING level or greater
logging.getLogger().setLevel(logging.INFO)
global _has_initialized
if _has_initialized:
logging.warn(
"main robocoup python init() method called twice - ignoring")
return
# init root play
global _root_play
import root_play as root_play_module
_root_play = root_play_module.RootPlay()
# init play registry
global _play_registry
_play_registry = play_registry_module.PlayRegistry()
# load all plays
play_classes = class_import.recursive_import_classes(
GAMEPLAY_DIR, ['plays'], play.Play)
for entry in play_classes:
# keep in mind that @entry is a tuple
mod_path = entry[0][1:]
_play_registry.insert(mod_path, entry[1])
# this callback lets us do cool stuff when our python files change on disk
def fswatch_callback(event_type, module_path):
# the top-level folders we care about watching
autoloadables = ['plays', 'skills', 'tactics', 'evaluation']
if module_path[0] in autoloadables:
logging.info('.'.join(module_path) + " " + event_type)
is_play = module_path[0] == 'plays'
if event_type == 'created':
if is_play:
# we load the module and register the play class it contains with the play registry
# this makes it automatically show up in the play config tab in the gui
try:
module = importlib.import_module('.'.join(module_path))
except:
logging.error("Error reloading module '" +
'.'.join(module_path) + "': e")
traceback.print_exc()
return
try:
play_class = class_import.find_subclasses(
module, play.Play)[0]
_play_registry.insert(
module_path[1:], play_class
) # note: skipping index zero of module_path cuts off the 'plays' part
except IndexError as e:
# we'll get an IndexError exception if the module didn't contain any Plays
# FIXME: instead, we should unload the module and just log a warning
raise Exception(
"Error: python files within the plays directory must contain a subclass of play.Play"
)
elif event_type == 'modified':
try:
# reload the module
containing_dict = sys.modules
for modname in module_path[:-1]:
containing_dict = containing_dict[modname].__dict__
module = containing_dict[module_path[-1]]
try:
module = imp.reload(module)
except:
logging.error("Error reloading module '" +
'.'.join(module_path) + "': e")
traceback.print_exc()
return
logging.info("reloaded module '" + '.'.join(module_path) +
"'")
if is_play:
# re-register the new play class
# FIXME: this logic should go inside the play_registry
play_reg_node = _play_registry.node_for_module_path(
module_path[1:])
play_reg_node.play_class = class_import.find_subclasses(
module, play.Play)[0]
# _play_registry.modelReset.emit()
# kill currently-running stuff if needed
if not is_play:
_root_play.drop_current_play()
_root_play.drop_goalie_behavior()
elif is_play and root_play != None and root_play.__class__.__name__ == play_reg_node.play_class.__name__:
_root_play.drop_current_play()
except Exception as e:
logging.error("EXCEPTION in file modified event: " +
repr(e))
traceback.print_exc()
raise e
elif event_type == 'deleted':
if is_play:
node = _play_registry.node_for_module_path(module_path[1:])
if _root_play.play != None and _root_play.play.__class__.__name__ == node.play_class.__name__:
_root_play.drop_current_play()
_play_registry.delete(module_path[1:])
else:
_root_play.drop_current_play()
_root_play.drop_goalie_behavior()
else:
raise AssertionError("Unknown FsWatcher event type: '" +
event_type + "'")
# start up filesystem-watching
watcher = fs_watcher.FsWatcher(GAMEPLAY_DIR)
watcher.subscribe(fswatch_callback)
watcher.start()
_has_initialized = True
else:
raise
logging.getLogger().setLevel(logging.INFO)
import main
import ui.main
main.init()
# Creates a mock ball so that no error is thrown when a ball isn't found
context = robocup.Context()
main.set_context(context)
for behavior_type in ['skills', 'tactics', 'plays']:
entries = class_import.recursive_import_classes('.', [behavior_type],
fsm.StateMachine)
for entry in entries:
try:
klass = entry[1]
module_path = entry[0]
dirpath = 'diagrams/' + '/'.join(module_path[:-1])
mkdir_p(dirpath)
filepath = dirpath + "/" + klass.__name__
klass().write_diagram_png(filepath)
print("generated " + filepath)
except Exception as e:
logging.error("Error generating fsm diagram for behavior '" +
klass.__name__ + "':" + str(e))
traceback.print_exc()
