org.chocosolver.parser.handlers.LimitHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of choco-parsers Show documentation

Provide parsers from FlatZinc, XCSP3 and MSP to Choco-solver.

The newest version!

/*
 * This file is part of choco-parsers, http://choco-solver.org/
 *
 * Copyright (c) 2024, IMT Atlantique. All rights reserved.
 *
 * Licensed under the BSD 4-clause license.
 *
 * See LICENSE file in the project root for full license information.
 */
package org.chocosolver.parser.handlers;

import org.chocosolver.solver.search.strategy.SearchParams;
import org.chocosolver.util.tools.TimeUtils;
import org.kohsuke.args4j.CmdLineException;
import org.kohsuke.args4j.CmdLineParser;
import org.kohsuke.args4j.OptionDef;
import org.kohsuke.args4j.spi.OneArgumentOptionHandler;
import org.kohsuke.args4j.spi.Setter;

/**
 * 

 *
 * @author Charles Prud'homme
 * @since 15/06/2020
 */
public class LimitHandler extends OneArgumentOptionHandler {

    public LimitHandler(CmdLineParser parser, OptionDef option, Setter setter) {
        super(parser, option, setter);
    }

    /**
     * Returns {@code "[String+]"}.
     *
     * @return return "[String+]";
     */
    @Override
    public String getDefaultMetaVariable() {
        return "[String+]";
    }


    @Override
    protected SearchParams.LimConf parse(String argument) throws NumberFormatException, CmdLineException {
        if (argument.startsWith("[")) argument = argument.substring(1);
        if (argument.endsWith("]")) argument = argument.substring(0, argument.length() - 1);
        String[] pars = argument.split(",");
        long time = -1;
        int sols = -1;
        int runs = -1;
        for (String params : pars) {
            if (params.endsWith("runs")) {
                params = params.substring(0, params.length() - 4);
                runs = Integer.parseInt(params);
                continue;
            }
            if (params.endsWith("sols")) {
                params = params.substring(0, params.length() - 4);
                sols = Integer.parseInt(params);
                continue;
            }
            time = TimeUtils.convertInMilliseconds(params);
        }
        return new SearchParams.LimConf(time, sols, runs);
    }
}