org.optaplanner.examples.common.business.SolutionBusiness 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.business;
import java.io.File;
import java.io.FileFilter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import javax.swing.SwingUtilities;
import org.apache.commons.io.filefilter.DirectoryFileFilter;
import org.optaplanner.core.api.score.Score;
import org.optaplanner.core.api.score.constraint.ConstraintMatchTotal;
import org.optaplanner.core.api.solver.Solver;
import org.optaplanner.core.impl.event.BestSolutionChangedEvent;
import org.optaplanner.core.impl.event.SolverEventListener;
import org.optaplanner.core.impl.move.Move;
import org.optaplanner.core.impl.score.director.ScoreDirector;
import org.optaplanner.core.impl.score.director.ScoreDirectorFactory;
import org.optaplanner.core.impl.solution.Solution;
import org.optaplanner.core.impl.solver.ProblemFactChange;
import org.optaplanner.examples.common.persistence.AbstractSolutionExporter;
import org.optaplanner.examples.common.persistence.AbstractSolutionImporter;
import org.optaplanner.examples.common.persistence.SolutionDao;
import org.optaplanner.examples.common.swingui.SolverAndPersistenceFrame;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class SolutionBusiness {
private static final ProblemFileComparator FILE_COMPARATOR = new ProblemFileComparator();
protected final transient Logger logger = LoggerFactory.getLogger(getClass());
private SolutionDao solutionDao;
private AbstractSolutionImporter importer;
private AbstractSolutionExporter exporter;
private File importDataDir;
private File unsolvedDataDir;
private File solvedDataDir;
private File exportDataDir;
// volatile because the solve method doesn't come from the event thread (like every other method call)
private volatile Solver solver;
private ScoreDirector guiScoreDirector;
public void setSolutionDao(SolutionDao solutionDao) {
this.solutionDao = solutionDao;
}
public void setImporter(AbstractSolutionImporter importer) {
this.importer = importer;
}
public void setExporter(AbstractSolutionExporter exporter) {
this.exporter = exporter;
}
public File getDataDir() {
return solutionDao.getDataDir();
}
public String getDirName() {
return solutionDao.getDirName();
}
public boolean hasImporter() {
return importer != null;
}
public boolean hasExporter() {
return exporter != null;
}
public void updateDataDirs() {
File dataDir = getDataDir();
if (!dataDir.exists()) {
throw new IllegalStateException("The directory dataDir (" + dataDir.getAbsolutePath()
+ ") does not exist." +
" The working directory should be set to the directory that contains the data directory." +
" This is different in a git clone (optaplanner/optaplanner-examples)" +
" and the release zip (examples).");
}
if (hasImporter()) {
importDataDir = new File(dataDir, "input");
if (!importDataDir.exists()) {
throw new IllegalStateException("The directory importDataDir (" + importDataDir.getAbsolutePath()
+ ") does not exist.");
}
}
unsolvedDataDir = new File(dataDir, "unsolved");
if (!unsolvedDataDir.exists()) {
throw new IllegalStateException("The directory unsolvedDataDir (" + unsolvedDataDir.getAbsolutePath()
+ ") does not exist.");
}
solvedDataDir = new File(dataDir, "solved");
if (!solvedDataDir.exists() && !solvedDataDir.mkdir()) {
throw new IllegalStateException("The directory solvedDataDir (" + solvedDataDir.getAbsolutePath()
+ ") does not exist and could not be created.");
}
if (hasExporter()) {
exportDataDir = new File(dataDir, "output");
if (!exportDataDir.exists() && !exportDataDir.mkdir()) {
throw new IllegalStateException("The directory exportDataDir (" + exportDataDir.getAbsolutePath()
+ ") does not exist and could not be created.");
}
}
}
public File getImportDataDir() {
return importDataDir;
}
public boolean acceptImportFile(File file) {
return importer.acceptInputFile(file);
}
public String getImportFileSuffix() {
return importer.getInputFileSuffix();
}
public File getUnsolvedDataDir() {
return unsolvedDataDir;
}
public File getSolvedDataDir() {
return solvedDataDir;
}
public File getExportDataDir() {
return exportDataDir;
}
public void setSolver(Solver solver) {
this.solver = solver;
ScoreDirectorFactory scoreDirectorFactory = solver.getScoreDirectorFactory();
guiScoreDirector = scoreDirectorFactory.buildScoreDirector();
}
public List getUnsolvedFileList() {
List fileList = extractFileList(unsolvedDataDir);
Collections.sort(fileList, FILE_COMPARATOR);
return fileList;
}
public List getSolvedFileList() {
List fileList = extractFileList(solvedDataDir);
Collections.sort(fileList, FILE_COMPARATOR);
return fileList;
}
public List extractFileList(File directory) {
List fileList = Arrays.asList(directory.listFiles(new SolutionFileFilter(solutionDao)));
for (File subDirectory : directory.listFiles((FileFilter) DirectoryFileFilter.INSTANCE)) {
fileList.addAll(extractFileList(subDirectory));
}
return fileList;
}
public Solution getSolution() {
return guiScoreDirector.getWorkingSolution();
}
public void setSolution(Solution solution) {
guiScoreDirector.setWorkingSolution(solution);
}
public Score getScore() {
return guiScoreDirector.calculateScore();
}
public boolean isSolving() {
return solver.isSolving();
}
public void registerForBestSolutionChanges(final SolverAndPersistenceFrame solverAndPersistenceFrame) {
solver.addEventListener(new SolverEventListener() {
// Not called on the event thread
public void bestSolutionChanged(BestSolutionChangedEvent event) {
// Avoid ConcurrentModificationException when there is an unprocessed ProblemFactChange
// because the paint method uses the problem facts instances as the solver
// unlike the planning entities which are cloned
if (solver.isEveryProblemFactChangeProcessed()) {
// final is also needed for thread visibility
final Solution latestBestSolution = event.getNewBestSolution();
// Migrate it to the event thread
SwingUtilities.invokeLater(new Runnable() {
public void run() {
guiScoreDirector.setWorkingSolution(latestBestSolution);
solverAndPersistenceFrame.bestSolutionChanged();
}
});
}
}
});
}
public boolean isConstraintMatchEnabled() {
return guiScoreDirector.isConstraintMatchEnabled();
}
public List getConstraintMatchTotalList() {
List constraintMatchTotalList = new ArrayList(
guiScoreDirector.getConstraintMatchTotals());
Collections.sort(constraintMatchTotalList);
return constraintMatchTotalList;
}
public void importSolution(File file) {
Solution solution = importer.readSolution(file);
guiScoreDirector.setWorkingSolution(solution);
}
public void openSolution(File file) {
Solution solution = solutionDao.readSolution(file);
guiScoreDirector.setWorkingSolution(solution);
}
public void saveSolution(File file) {
Solution solution = guiScoreDirector.getWorkingSolution();
solutionDao.writeSolution(solution, file);
}
public void exportSolution(File file) {
Solution solution = guiScoreDirector.getWorkingSolution();
exporter.writeSolution(solution, file);
}
public void doMove(Move move) {
if (solver.isSolving()) {
logger.error("Not doing user move ({}) because the solver is solving.", move);
return;
}
if (!move.isMoveDoable(guiScoreDirector)) {
logger.warn("Not doing user move ({}) because it is not doable.", move);
return;
}
logger.info("Doing user move ({}).", move);
move.doMove(guiScoreDirector);
}
public void doProblemFactChange(ProblemFactChange problemFactChange) {
if (solver.isSolving()) {
solver.addProblemFactChange(problemFactChange);
} else {
problemFactChange.doChange(guiScoreDirector);
}
}
/**
* Can be called on any thread.
*
* Note: This method does not change the guiScoreDirector because that can only be changed on the event thread.
* @param planningProblem never null
* @return never null
*/
public Solution solve(Solution planningProblem) {
solver.setPlanningProblem(planningProblem);
solver.solve();
return solver.getBestSolution();
}
public void terminateSolvingEarly() {
solver.terminateEarly();
}
}