#!/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='MPI demo.')
    parser.add_argument("n_ranks", type=int, default=1, help="number of MPI ranks (default: 1)")
    args = parser.parse_args()

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

    # run parallel program
    cfp.run_shell_command('mpirun -np '+str(args.n_ranks)+' ./mpi_test')

    # let the timer report
    timer.report()