ArgumentParser.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * ArgumentParser.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2007 by the members listed in the COPYING,        *
 *                   LICENSE and WARRANTY file.                            *
 * email           : info at matsim dot org                                *
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *   See also COPYING, LICENSE and WARRANTY file                           *
 *                                                                         *
 * *********************************************************************** */

package org.matsim.core.utils.misc;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class ArgumentParser implements Iterable<String> {

    private List<String> args = null;
    private boolean enableShortOptions = true;

    public ArgumentParser(final String[] args) {
        this(args, false);
    }

    public ArgumentParser(final String[] args, final boolean enableShortOptions) {
        this.args = new ArrayList<String>();
        this.enableShortOptions = enableShortOptions;
        parse(args);
    }


    @Override
    public Iterator<String> iterator() {
        return this.args.iterator();
    }


    private void parse(final String[] args) {
        if (this.enableShortOptions) {
            for (String arg : args) {
                if (arg.startsWith("--")) {
                    parseLongOption(arg.substring(2));
                } else if (arg.startsWith("-")) {
                    parseShortOption(arg.substring(1));
                } else {
                    this.args.add(arg);
                }
            }
        } else {
            for (String arg : args) {
                if (arg.startsWith("-")) {
                    parseOption(arg);
                } else {
                    this.args.add(arg);
                }
            }
        }
    }

    private void parseShortOption(final String arg) {
        if (arg.length() == 0) {
            this.args.add("-");
            return;
        }
        for (int i = 0; i < arg.length(); i++) {
            char ch = arg.charAt(i);
            if (ch == '=') {
                if ((i == 0) && (arg.length() == 1)) {
                    // '=' is the only char
                    this.args.add("-=");
                    return;
                } else if (i == 0) {
                    // arg is a string in the form of '-=*', for me this makes only sense as a string-param, so just add it
                    this.args.add('-' + arg);
                    return;
                } else {
                    this.args.add(arg.substring(i+1));
                    return;
                }
            }
            // else...
            this.args.add("-" + ch);
        }
    }

    private void parseLongOption(final String arg) {
        StringBuilder argname = new StringBuilder("--");
        for (int i = 0; i < arg.length(); i++) {
            char ch = arg.charAt(i);
            if (ch == '=') {
                if ((i == 0) && (arg.length() == 1)) {
                    // '=' is the only char
                    this.args.add("--=");
                    return;
                } else if (i == 0) {
                    // arg is a string in the form of '--=*', interpret it as a string argument
                    this.args.add("--" + arg);
                    return;
                } else {
                    this.args.add(argname.toString());
                    this.args.add(arg.substring(i + 1));
                    return;
                }
            }
            // else...
            argname.append(ch);
        }
        this.args.add(argname.toString());
    }

    private void parseOption(final String arg) {
        StringBuilder argname = new StringBuilder();
        argname.append('-');  // every option starts with '-'
        for (int i = 1; i < arg.length(); i++) { // options always start with '-', so we can start at char 1
            char ch = arg.charAt(i);
            if (ch == '=') {
                if (argname.toString().equals("-")) {
                    // arg is a string in the form of '-=*', interpret it as a strig argument
                    this.args.add(arg);
                    return;
                }
                if (argname.toString().equals("--")) {
                    // arg is a string in the form of '--=*', interpret it as a strig argument
                    this.args.add(arg);
                    return;
                }
                // it seems there are already some real characters before '=' that make a correct argument
                this.args.add(argname.toString());
                this.args.add(arg.substring(i + 1));
                return;
            }
            // else...
            argname.append(ch);
        }
        this.args.add(argname.toString());
    }

}