pycollimate.py 文件源码

def process_command_line_old(argv):
    """
    Process command line arguments
    """

    help_text = 'Usage: python pycollimate.py 1 ti2_track_out_1.tfsone initial_distribution\n'

    if argv is None:

        argv = sys.argv[1:]

    # initialize the parser object:
    parser = optparse.OptionParser(
        formatter=optparse.TitledHelpFormatter(width=78),
        add_help_option=None)

    # define options here:
    parser.add_option(      # customized description; put --help last
                            '-h', '--help', action='help',
                            help=help_text)

    parser.set_defaults(coll_index=1, file_input='twiss_input.tfs',
                        file_out='initial_distribution', black=False)

    parser.add_option('-o', '--output', dest='file_out')

    parser.add_option('-i', '--input', dest='file_input')

    parser.add_option('-b', '--black', action='store_true', dest='black', help='Black absorber switch')

    parser.add_option('-n', '--number', type="int", dest="coll_index", help='Collimator index and seed')

    options, args = parser.parse_args(argv)

    return options, args