batch.py

import subprocess
import os

from os import listdir
from os.path import isfile, join

import argparse

parser = argparse.ArgumentParser(description='Batch proccess IPC sims.')
parser.add_argument(
    "--offline", help="run IPC without the viewer", action='store_true')
args = parser.parse_args()

inputFolderPath = os.path.realpath('.') + '/input/'
# on Mac:
# progPath = os.path.realpath('.') + '/build/Release/IPC_bin'
# on Minchen's Mac:
# progPath = '/Users/mincli/Library/Developer/Xcode/DerivedData/IPC-cegibpdumtrmuqbjruacrqwltitb/Build/Products/Release/IPC'
# on Ubuntu:
progPath = os.path.realpath('.') + '/build/IPC_bin'
# progPath = os.path.realpath('.') + '/src/Projects/DistortionMin/DistortionMin'

# envSetStr = 'export LD_LIBRARY_PATH=/usr/local/lib\n'
# for Ubuntu or Mac when CHOLMOD is compiled with MKL LAPACK and BLAS
NTSetStr0 = 'export MKL_NUM_THREADS='
# for Ubuntu when CHOLMOD is compiled with libopenblas
NTSetStr1 = 'export OMP_NUM_THREADS='
# for Mac when CHOLMOD is compiled with default LAPACK and BLAS
NTSetStr2 = 'export VECLIB_MAXIMUM_THREADS='

for numOfThreads in ['1', '8', '12']:
    inputFolderPath = os.path.realpath('.') + '/input/' + numOfThreads + '/'
    if(not os.path.isdir(inputFolderPath)):
        continue
    onlyfiles = [
        f for f in listdir(inputFolderPath) if isfile(join(inputFolderPath, f))
    ]
    for inputModelNameI in onlyfiles:
        runCommand = NTSetStr0 + numOfThreads + '\n'
        runCommand += NTSetStr1 + numOfThreads + '\n'
        runCommand += NTSetStr2 + numOfThreads + '\n'
        runCommand += "{} {} {} t{}".format(
            progPath, '100' if args.offline else '10',
            inputFolderPath + inputModelNameI, numOfThreads)
        if subprocess.call([runCommand], shell=True):
            continue