import argparse as argparse; parser = argparse.ArgumentParser(prog="python main.py", description = "Main.py handles everything. Use python main.py -h to see your options."); parser.add_argument('-m', '--mode', help='Set the simulation mode.', default = 3, action='store', type = int); starlings = Flock() args = parser.parse_args(); flocksim = args.mode if flocksim == 0: starlings.mode = 1 starlings.number = 40 starlings.tau = 0.1 starlings.eta = 0.25 starlings.speed = 25.0 starlings.boxSize = 100.0 starlings.habitatSize = 200.00 starlings.habitatStrength = 20.0 starlings.initBoids() starlings.velocities = np.ones((starlings.number,3), dtype=np.float) starlings.positions[ int(starlings.number/2):,:] += 0.8 * starlings.habitatSize / (2.0**0.5) starlings.sensitivities *= 30.00 starlings.display = 1 starlings.length = 10.0 starlings.show() elif flocksim == 1:
'--filename', help='Set the simulation mode.', default="default", action='store') starlings = Flock() args = parser.parse_args() flocksim = args.mode filename = args.filename + ".mp4" snapshots = args.snapshots #number of frames if flocksim == 0: starlings.mode = 1 starlings.number = 40 starlings.tau = 0.1 starlings.eta = 0.25 starlings.speed = 25.0 starlings.boxSize = 100.0 starlings.habitatSize = 200.00 starlings.habitatStrength = 20.0 starlings.initBoids() starlings.velocities = np.ones((starlings.number, 3), dtype=np.float) starlings.positions[int(starlings.number / 2):, :] += 0.8 * starlings.habitatSize / (2.0**0.5) starlings.sensitivities *= 30.00 starlings.display = 1 starlings.length = 10.0 starlings.record(filename, snapshots)