#!/usr/bin/env python

"""
Redrock for BOSS - MPI entry point
"""
import os
import sys

from redrock.utils import nersc_login_node

# MPI environment availability
have_mpi = None
if nersc_login_node():
    have_mpi = False
else:
    have_mpi = True
    try:
        import mpi4py.MPI as MPI
    except ImportError:
        have_mpi = False

from redrock.external import boss

if have_mpi:
    boss.rrboss(comm=MPI.COMM_WORLD)
else:
    print("MPI not available, falling back to serial / multiprocessing case")
    sys.stdout.flush()
    boss.rrboss(comm=None)
