#!python

'''
join mock targets and truth from individual pixels into summary tables
'''

from __future__ import absolute_import, division, print_function
import argparse
parser = argparse.ArgumentParser(usage = "{prog} [options]")
parser.add_argument("--mockdir", type=str, required=True, help="input mock directory")
parser.add_argument("--outdir", type=str,  help="output directory [default same as mockdir]")
parser.add_argument("--force", action="store_true", help="[deprecated; see --overwrite]")
parser.add_argument("--overwrite", action="store_true", help="overwrite pre-existing output files")
parser.add_argument("--mpi", action="store_true", help="Use MPI parallelism")
args = parser.parse_args()

#- Establish MPI communication before other imports so SLURM knows we're alive
if args.mpi:
    from mpi4py import MPI
    comm = MPI.COMM_WORLD
else:
    comm = None

from desitarget.mock.build import join_targets_truth

if args.force:
    args.overwrite = args.force
    if comm is None or comm.rank == 0:
        from desiutil.log import get_logger
        log = get_logger()
        log.warning('--force is deprecated; please use --overwrite instead')

join_targets_truth(args.mockdir, outdir=args.outdir, overwrite=args.overwrite, comm=comm)





