batch.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Run rasim batchly

Created on Wed Nov 12 02:50:02 2014

@author: Mehmet Emre
"""

import argparse

parser = argparse.ArgumentParser(description="rasim - A radio network simulator")
parser.add_argument('-j', '--jobs', action='store', help='number of parallel jobs to run', type=int)
argv = parser.parse_args()

import subprocess
import multiprocessing
import sys

output_dir = 'data'

N_runs = 30
t_total = 12000
N_agents = [1, 5, 10]
N_channel = 5
N_good_channels = [1, 2, 4, 5]
buffer_levels = [2, 4, 10]
buffer_size = 512
packet_size = 1024
min_packet_rate = 0
max_packet_rate = 6
beta_idles = [1, 2, 4]

total_runs = len(N_agents) * len(N_good_channels) * len(buffer_levels) * len(beta_idles)
current_run = 1
processes = []
for N_agent in N_agents:
    for N_good_channel in N_good_channels:
        for buffer_level in buffer_levels:
            for beta_idle in beta_idles:
                processes.append(('python sim.py --N-runs %d --t-total %d --individual-q --random-channel\
                --highest-snr --output-dir %s --n-agent %d --n-channel %d --n-good-channel %d --buffer-levels %d --buffer-size %d\
                --batch-run --packet-size %d --min-packet-rate %d --max-packet-rate %d --beta-idle %d' % (
                    N_runs, t_total, output_dir, N_agent, N_channel, N_good_channel, buffer_level, buffer_size, packet_size,
                    min_packet_rate, max_packet_rate, beta_idle
                )).split())

def work(arg):
    command, current_run, total_runs = arg # to comply with multiprocessing
    subprocess.call(command)
    print "[RUN #%d of %d completed]" % (current_run, total_runs)
    sys.stdout.flush()

if argv.jobs:
    count = argv.jobs
else:
    count = multiprocessing.cpu_count()
print "Running %d processes parallelly" % count
pool = multiprocessing.Pool(processes=count)
pool.map_async(work, zip(processes, xrange(1, total_runs + 1), [total_runs] * total_runs))
pool.close()
pool.join()