mirror of
https://github.com/gsi-upm/soil
synced 2024-09-27 18:21:41 +00:00
7d1c800490
* Graphs are not saved by default (not backwards compatible) * Modified newsspread examples * More granular options to save results (exporting to CSV and GEXF are now optional) * Updated tutorial to include exporting options * Removed references from environment to simulation * Added parallelism to simulations (can be turned off with a flag or argument).
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import importlib
|
|
import sys
|
|
import os
|
|
import pdb
|
|
import logging
|
|
|
|
__version__ = "0.10.1"
|
|
|
|
try:
|
|
basestring
|
|
except NameError:
|
|
basestring = str
|
|
|
|
logging.basicConfig()
|
|
|
|
from . import agents
|
|
from . import simulation
|
|
from . import environment
|
|
from . import utils
|
|
from . import analysis
|
|
|
|
|
|
def main():
|
|
import argparse
|
|
from . import simulation
|
|
|
|
parser = argparse.ArgumentParser(description='Run a SOIL simulation')
|
|
parser.add_argument('file', type=str,
|
|
nargs="?",
|
|
default='simulation.yml',
|
|
help='python module containing the simulation configuration.')
|
|
parser.add_argument('--module', '-m', type=str,
|
|
help='file containing the code of any custom agents.')
|
|
parser.add_argument('--dry-run', '--dry', action='store_true',
|
|
help='Do not store the results of the simulation.')
|
|
parser.add_argument('--pdb', action='store_true',
|
|
help='Use a pdb console in case of exception.')
|
|
parser.add_argument('--graph', '-g', action='store_true',
|
|
help='Dump GEXF graph. Defaults to false.')
|
|
parser.add_argument('--csv', action='store_true',
|
|
help='Dump history in CSV format. Defaults to false.')
|
|
parser.add_argument('--output', '-o', type=str, default="soil_output",
|
|
help='folder to write results to. It defaults to the current directory.')
|
|
parser.add_argument('--synchronous', action='store_true',
|
|
help='Run trials serially and synchronously instead of in parallel. Defaults to false.')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.module:
|
|
sys.path.append(os.getcwd())
|
|
importlib.import_module(args.module)
|
|
|
|
logging.info('Loading config file: {}'.format(args.file, args.output))
|
|
|
|
try:
|
|
dump = []
|
|
if not args.dry_run:
|
|
if args.csv:
|
|
dump.append('csv')
|
|
if args.graph:
|
|
dump.append('gexf')
|
|
simulation.run_from_config(args.file,
|
|
dry_run=args.dry_run,
|
|
dump=dump,
|
|
parallel=(not args.synchronous),
|
|
results_dir=args.output)
|
|
except Exception as ex:
|
|
if args.pdb:
|
|
pdb.post_mortem()
|
|
else:
|
|
raise
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|