org.optaplanner.examples.common.persistence.AbstractTxtSolutionImporter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of optaplanner-examples Show documentation
Show all versions of optaplanner-examples Show documentation
OptaPlanner solves planning problems.
This lightweight, embeddable planning engine implements powerful and scalable algorithms
to optimize business resource scheduling and planning.
This module contains the examples which demonstrate how to use it in a normal Java application.
/*
* Copyright 2010 JBoss Inc
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.optaplanner.examples.common.persistence;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.math.BigInteger;
import java.net.URL;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.IOUtils;
import org.optaplanner.core.impl.solution.Solution;
public abstract class AbstractTxtSolutionImporter extends AbstractSolutionImporter {
private static final String DEFAULT_INPUT_FILE_SUFFIX = ".txt";
protected AbstractTxtSolutionImporter(SolutionDao solutionDao) {
super(solutionDao);
}
public String getInputFileSuffix() {
return DEFAULT_INPUT_FILE_SUFFIX;
}
public abstract TxtInputBuilder createTxtInputBuilder();
public Solution readSolution(File inputFile) {
Solution solution;
BufferedReader bufferedReader = null;
try {
bufferedReader = new BufferedReader(new InputStreamReader(new FileInputStream(inputFile), "UTF-8"));
TxtInputBuilder txtInputBuilder = createTxtInputBuilder();
txtInputBuilder.setInputFile(inputFile);
txtInputBuilder.setBufferedReader(bufferedReader);
try {
solution = txtInputBuilder.readSolution();
} catch (IllegalArgumentException e) {
throw new IllegalArgumentException("Exception in inputFile (" + inputFile + ")", e);
} catch (IllegalStateException e) {
throw new IllegalStateException("Exception in inputFile (" + inputFile + ")", e);
}
} catch (IOException e) {
throw new IllegalArgumentException("Could not read the file (" + inputFile.getName() + ").", e);
} finally {
IOUtils.closeQuietly(bufferedReader);
}
logger.info("Imported: {}", inputFile);
return solution;
}
public Solution readSolution(URL inputURL) {
BufferedReader bufferedReader = null;
try {
bufferedReader = new BufferedReader(new InputStreamReader(inputURL.openStream(), "UTF-8"));
TxtInputBuilder txtInputBuilder = createTxtInputBuilder();
txtInputBuilder.setInputFile(new File(inputURL.getFile()));
txtInputBuilder.setBufferedReader(bufferedReader);
try {
return txtInputBuilder.readSolution();
} catch (IllegalArgumentException e) {
throw new IllegalArgumentException("Exception in inputURL (" + inputURL + ")", e);
} catch (IllegalStateException e) {
throw new IllegalStateException("Exception in inputURL (" + inputURL + ")", e);
}
} catch (IOException e) {
throw new IllegalArgumentException("Could not read the inputURL (" + inputURL + ").", e);
} finally {
IOUtils.closeQuietly(bufferedReader);
}
}
public abstract class TxtInputBuilder {
protected File inputFile;
protected BufferedReader bufferedReader;
public void setInputFile(File inputFile) {
this.inputFile = inputFile;
}
public void setBufferedReader(BufferedReader bufferedReader) {
this.bufferedReader = bufferedReader;
}
public abstract Solution readSolution() throws IOException;
// ************************************************************************
// Helper methods
// ************************************************************************
public String getInputId() {
return FilenameUtils.getBaseName(inputFile.getPath());
}
public void readEmptyLine() throws IOException {
readConstantLine("");
}
public void readConstantLine(String constantValue) throws IOException {
String line = bufferedReader.readLine();
if (line == null) {
throw new IllegalArgumentException("File ends before a line is expected to be a constant value ("
+ constantValue + ").");
}
String value = line.trim();
if (!value.equals(constantValue)) {
throw new IllegalArgumentException("Read line (" + line + ") is expected to be a constant value ("
+ constantValue + ").");
}
}
public void readUntilConstantLine(String constantValue) throws IOException {
String line;
String value;
do {
line = bufferedReader.readLine();
if (line == null) {
throw new IllegalArgumentException("File ends before a line is expected to be a constant value ("
+ constantValue + ").");
}
value = line.trim();
} while (!value.equals(constantValue));
}
public void readRegexConstantLine(String regex) throws IOException {
String line = bufferedReader.readLine();
if (line == null) {
throw new IllegalArgumentException("File ends before a line is expected to be a regex ("
+ regex + ").");
}
String value = line.trim();
if (!value.matches(regex)) {
throw new IllegalArgumentException("Read line (" + line + ") is expected to be a regex ("
+ regex + ").");
}
}
public int readIntegerValue() throws IOException {
return readIntegerValue("");
}
public int readIntegerValue(String prefix) throws IOException {
return readIntegerValue(prefix, "");
}
public int readIntegerValue(String prefix, String suffix) throws IOException {
String line = bufferedReader.readLine();
if (line == null) {
throw new IllegalArgumentException("File ends before a line is expected to contain an integer value ("
+ prefix + "" + suffix + ").");
}
String value = removePrefixSuffixFromLine(line, prefix, suffix);
try {
return Integer.parseInt(value);
} catch (NumberFormatException e) {
throw new IllegalArgumentException("Read line (" + line + ") is expected to contain an integer value ("
+ value + ").", e);
}
}
public long readLongValue() throws IOException {
return readLongValue("");
}
public long readLongValue(String prefix) throws IOException {
return readLongValue(prefix, "");
}
public long readLongValue(String prefix, String suffix) throws IOException {
String line = bufferedReader.readLine();
if (line == null) {
throw new IllegalArgumentException("File ends before a line is expected to contain an integer value ("
+ prefix + "" + suffix + ").");
}
String value = removePrefixSuffixFromLine(line, prefix, suffix);
try {
return Long.parseLong(value);
} catch (NumberFormatException e) {
throw new IllegalArgumentException("Read line (" + line + ") is expected to contain an integer value ("
+ value + ").", e);
}
}
public String readStringValue() throws IOException {
return readStringValue("");
}
public String readStringValue(String prefix) throws IOException {
return readStringValue(prefix, "");
}
public String readStringValue(String prefix, String suffix) throws IOException {
String line = bufferedReader.readLine();
if (line == null) {
throw new IllegalArgumentException("File ends before a line is expected to contain an string value ("
+ prefix + "" + suffix + ").");
}
return removePrefixSuffixFromLine(line, prefix, suffix);
}
public String removePrefixSuffixFromLine(String line, String prefix, String suffix) {
String value = line.trim();
if (!value.startsWith(prefix)) {
throw new IllegalArgumentException("Read line (" + line + ") is expected to start with prefix ("
+ prefix + ").");
}
value = value.substring(prefix.length());
if (!value.endsWith(suffix)) {
throw new IllegalArgumentException("Read line (" + line + ") is expected to end with suffix ("
+ suffix + ").");
}
value = value.substring(0, value.length() - suffix.length());
value = value.trim();
return value;
}
public String[] splitBySpace(String line) {
return splitBySpace(line, null);
}
public String[] splitBySpace(String line, Integer numberOfTokens) {
return splitBy(line, "\\ ", "a space ( )", numberOfTokens, false, false);
}
public String[] splitBySpace(String line, Integer minimumNumberOfTokens, Integer maximumNumberOfTokens) {
return splitBy(line, "\\ ", "a space ( )", minimumNumberOfTokens, maximumNumberOfTokens, false, false);
}
public String[] splitBySpacesOrTabs(String line) {
return splitBySpacesOrTabs(line, null);
}
public String[] splitBySpacesOrTabs(String line, Integer numberOfTokens) {
return splitBy(line, "[\\ \\t]+", "spaces or tabs", numberOfTokens, false, false);
}
public String[] splitByPipelineAndTrim(String line, int numberOfTokens) {
return splitBy(line, "\\|", "a pipeline (|)", numberOfTokens, true, false);
}
public String[] splitBySemicolonSeparatedValue(String line, int numberOfTokens) {
return splitBy(line, ";", "a semicolon (;)", numberOfTokens, false, true);
}
public String[] splitBy(String line, String tokenRegex, String tokenName,
Integer numberOfTokens, boolean trim, boolean removeQuotes) {
return splitBy(line, tokenRegex, tokenName, numberOfTokens, numberOfTokens, trim, removeQuotes);
}
public String[] splitBy(String line, String tokenRegex, String tokenName,
Integer minimumNumberOfTokens, Integer maximumNumberOfTokens, boolean trim, boolean removeQuotes) {
String[] lineTokens = line.split(tokenRegex);
if (minimumNumberOfTokens != null && lineTokens.length < minimumNumberOfTokens) {
throw new IllegalArgumentException("Read line (" + line + ") has " + lineTokens.length
+ " tokens but is expected to contain at least " + minimumNumberOfTokens
+ " tokens separated by " + tokenName + ".");
}
if (maximumNumberOfTokens != null && lineTokens.length > maximumNumberOfTokens) {
throw new IllegalArgumentException("Read line (" + line + ") has " + lineTokens.length
+ " tokens but is expected to contain at most " + maximumNumberOfTokens
+ " tokens separated by " + tokenName + ".");
}
if (trim) {
for (int i = 0; i < lineTokens.length; i++) {
lineTokens[i] = lineTokens[i].trim();
}
}
if (removeQuotes) {
for (int i = 0; i < lineTokens.length; i++) {
if (lineTokens[i].startsWith("\"") && lineTokens[i].endsWith("\"")) {
lineTokens[i] = lineTokens[i].substring(1, lineTokens[i].length() - 1);
}
}
}
return lineTokens;
}
public boolean parseBooleanFromNumber(String token) {
if (token.equals("0")) {
return false;
} else if (token.equals("1")) {
return true;
} else {
throw new IllegalArgumentException("The token (" + token
+ ") is expected to be 0 or 1 representing a boolean.");
}
}
public BigInteger factorial(int base) {
BigInteger value = BigInteger.ONE;
for (int i = 1; i <= base; i++) {
value = value.multiply(BigInteger.valueOf(base));
}
return value;
}
}
}