run-mc

#!/usr/bin/env python

from argparse import ArgumentParser

parser = ArgumentParser()
parser.add_argument('--year', type=int, choices=(11, 12), default=11)
parser.add_argument('--student', default='hhskim.py')
parser.add_argument('--systematics', default=None)
parser.add_argument('--nproc', type=int, default=1)
parser.add_argument('--queue', default='short')
parser.add_argument('--output-path',
                    default='/cluster/data01/endw/ntuples/running')
parser.add_argument('--db', default='datasets_hh')
parser.add_argument('--nice', type=int, default=10)
parser.add_argument('--nominal-only', action='store_true', default=False)
parser.add_argument('--systematics-only', action='store_true', default=False)
parser.add_argument('--dry', action='store_true', default=False)
parser.add_argument('--use-ssh', dest='use_qsub', action='store_false', default=True)
parser.add_argument('--warnings-as-errors', action='store_true', default=False)
parser.add_argument('--include-embedded', action='store_true', default=False)
parser.add_argument('samples', nargs='*', default=None)

args, student_args = parser.parse_known_args()

import sys
import cluster
from higgstautau import samples

hosts = cluster.get_hosts('hosts.sfu.txt')
setup = cluster.get_setup('setup.noel.sfu.txt')

if not args.systematics_only:
    # nominal values
    datasets = samples.samples('hadhad', args.year, args.samples,
                               include_embedded=args.include_embedded)
    cluster.run(args.student,
                db=args.db,
                datasets=datasets,
                hosts=hosts,
                nproc=args.nproc,
                nice=args.nice,
                setup=setup,
                output_path=args.output_path,
                use_qsub=args.use_qsub,
                qsub_queue=args.queue,
                dry_run=args.dry,
                separate_student_output=True,
                warnings_as_errors=args.warnings_as_errors,
                student_args=student_args)

if not args.nominal_only:
    if args.systematics is not None:
        args.systematics = [
                tuple(s.split('+')) for s in
                args.systematics.split(',')]
    # systematics
    for datasets, systematics in samples.iter_samples('hadhad', args.year,
            args.samples, systematics=True,
            include_embedded=args.include_embedded):
        cluster.run_systematics_new(
            'HADHAD',
            args.student,
            db=args.db,
            systematics=systematics,
            filter_systematics=args.systematics,
            datasets=datasets,
            hosts=hosts,
            nproc=args.nproc,
            nice=args.nice,
            setup=setup,
            output_path=args.output_path,
            use_qsub=args.use_qsub,
            qsub_queue=args.queue,
            dry_run=args.dry,
            separate_student_output=True,
            warnings_as_errors=args.warnings_as_errors,
            student_args=student_args)