from subprocess import Popen 
import project
from os.path import join
import os
import glob

vmem={}
wall={}

vmem[1] = "6GB"
wall[1] = "48:00:00"
vmem[2] = "8700MB"
wall[2] = "40:00:00"
vmem[3] = "11GB"
wall[3] = "32:00:00"
vmem[4] = "13GB"
wall[4] = "24:00:00"
vmem[5] = "16GB"
wall[5] = "22:00:00"
vmem[6] = "18GB"
wall[6] = "20:00:00"
vmem[7] = "21GB"
wall[7] = "19:00:00"
vmem[8] = "24GB"
wall[8] = "17:00:00"
vmem[16] = "48GB"
wall[16] = "10:00:00"
vmem[24] = "72GB"
wall[24] = "09:00:00"
vmem[32] = "96GB"
wall[32] = "07:00:00"

for np in project.np:
    JobScriptName = project.event +"_"+ project.setup +"_"+ str(np) 
    f = open(JobScriptName, 'w')
    f.writelines(["#!/usr/bin/env tcsh\n",\
                 "#PBS -q normal\n",\
                 "#PBS -P w85\n",\
                 "#PBS -l walltime="+wall[np]+",vmem="+vmem[np]+",ncpus="+str(np)+"\n",\
                 "#PBS -wd\n",\
                 "mpirun -np "+str(np)+" python2.6 "+project.ParScript])
    f.close
    
    Popen(["qsub", JobScriptName])