#!/usr/bin/env python
# Christoph Federrath

import cfpack as cfp
import argparse

# ===== the following applies in case we are running this in script mode =====
if __name__ == "__main__":

    parser = argparse.ArgumentParser(description='OpenMP demo.')
    parser.add_argument("n_threads", type=int, default=1, help="number of OpenMP threads (default: 1)")
    args = parser.parse_args()

    # start a new timer
    timer = cfp.timer('OMP test')

    # run parallel program
    cfp.run_shell_command('export OMP_NUM_THREADS='+str(args.n_threads)+'; ./omp_test')

    # let the timer report
    timer.report()