def process_boids(self):
self.control = ControlBoids(self.settings)
self.boid_anim = self.control.run_animation()
if self.arguments.save is not None:
self.boid_anim.save(self.arguments.save,
metadata={'title': 'Boids'})
else:
plt.show()
class ParseBoids(object):
def __init__(self):
with open(os.path.join(os.path.dirname(__file__), 'parser_text.yml')) \
as text_file, open(os.path.join(os.path.dirname(__file__),
'boids_default_config.yml')) \
as default_config:
parse_text = yaml.load(text_file)
self.settings = yaml.load(default_config)
self.text_main = parse_text['main']
self.text_epi = parse_text['epi']
self.text_save = parse_text['save']
self.text_config = parse_text['config']
self.text_generate = parse_text['generate']
def entry_point(self):
self.parser = ArgumentParser(description=self.text_main,
epilog=self.text_epi)
self.parser.add_argument('--save', '-s', type=str, help=self.text_save)
self.parser.add_argument('--config',
'-c',
type=str,
help=self.text_config)
self.parser.add_argument('--generate',
'-g',
type=str,
help=self.text_generate)
self.arguments = self.parser.parse_args()
self._arg_check()
self.process_config()
def _arg_check(self):
args = vars(self.arguments)
for arg in args:
if args[arg] is not None:
extensions = {
'save': '.mp4',
'config': '.yml',
'generate': '.yml'
}
actual_arg = args[arg].lower()
expect_ext = extensions[arg]
arg_len = len(actual_arg) > 4
arg_end = actual_arg.endswith(expect_ext)
if not (arg_len & arg_end):
self.parser.error(arg.upper() + ' argument ' +
'should have extension ' + expect_ext)
def process_config(self):
if self.arguments.generate is not None:
self.generate_config()
elif self.arguments.config is not None:
self.open_config()
self.process_boids()
else:
self.process_boids()
def generate_config(self):
try:
config_generate = open(self.arguments.generate, 'w')
config_generate.write(yaml.dump(self.settings))
config_generate.close()
except IOError:
raise IOError('Unable to write configuration file')
def open_config(self):
try:
expected_keys = set(self.settings.keys())
config_read = open(self.arguments.config, 'r')
self.settings = yaml.load(config_read)
actual_keys = set(self.settings.keys())
assert expected_keys.issubset(actual_keys)
config_read.close()
except AssertionError:
raise KeyError('Key(s) missing form given configuration file')
except AttributeError:
raise IOError('Incorrectly formatted YAML file')
except IOError:
raise IOError('Unable to open given configuration file')
def process_boids(self):
self.control = ControlBoids(self.settings)
self.boid_anim = self.control.run_animation()
if self.arguments.save is not None:
self.boid_anim.save(self.arguments.save,
metadata={'title': 'Boids'})
else:
plt.show()
