def test_path(): assert_equals(expandfilename('config.ini'), 'config.ini') assert_equals( expandfilename(['~', 'home']), os.path.expanduser('~/home'))
import os import subprocess from simcity.util import expandfilename, copyglob if __name__ == '__main__': print("Input:") print(sys.argv) tmp_dir = sys.argv[1] in_dir = sys.argv[2] out_dir = sys.argv[3] with open(os.path.join(in_dir, 'input.json'), 'r') as f: params = json.load(f) src = expandfilename(('~', 'transims-us101')) dst = expandfilename((tmp_dir, 'transims-us101')) dirnames = ['control', 'input', 'output', 'ArcGis'] dirs = dict((name, expandfilename([dst, name])) for name in dirnames) os.mkdir(dst) for dstdir in dirs.values(): os.mkdir(dstdir) # Set up runtime directories copyglob((src, 'control', '*.ctl'), dirs['control']) copyglob((src, 'control', '*.sh'), dirs['control']) copyglob((src, 'input', '*'), dirs['input']) # Run simulation
tmp_dir = sys.argv[1] in_dir = sys.argv[2] out_dir = sys.argv[3] with open(os.path.join(in_dir, 'input.json'), 'r') as f: params = json.load(f) print("Parameters:") print(params) try: model = params['model'] except: model = 'tutorial' src = expandfilename(['~', 'bangalore-matsim']) indir = expandfilename([tmp_dir, 'input']) moddir = expandfilename([indir, model]) outdir = expandfilename([tmp_dir, 'output', model]) os.mkdir(indir) os.mkdir(moddir) # don't make output directory # Set up runtime directories copyglob([src, 'input', model, '*'], moddir) # Run simulation os.chdir(tmp_dir) subprocess.call([expandfilename([src, 'scripts', 'matsim-runner.sh']),