def run(probability, samples=1, turning=False, fname=None):
# Run simulation for each radius
average_collisions = []
for i, sample in enumerate(range(samples)):
print '\tSim {} of {}'.format(i + 1, samples)
sim = Simulation(probability, turning=turning)
sim.run(animate=False, fname=fname)
average_collisions.append(sim.average_collisions())
return float(sum(average_collisions)) / len(average_collisions)
def main():
# Read arguments
args = parser().parse_args()
seed(args.seed)
# Run once
if not args.run_all:
sim = Simulation(args.probability, turning=args.turning)
sim.run(animate=args.animate, fname=args.output_file)
print 'collisions = {}'.format(sim.average_collisions())
# Run all
else:
probabilities = np.linspace(0.04, 0.2, 17)
run_all(probabilities, turning=args.turning, fname=args.output_file)
