org.drools.planner.examples.pas.solver.move.factory.BedDesignationPillarPartSwapMoveFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of drools-planner-examples Show documentation
Show all versions of drools-planner-examples Show documentation
Drools Planner optimizes automated planning by combining metaheuristic search algorithms with rule
engine powered score calculation. This is the drools-planner-examples module which contains examples on how to use
Drools Planner.
/*
* 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.drools.planner.examples.pas.solver.move.factory;
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.HashMap;
import java.util.Collections;
import java.util.Comparator;
import java.util.Iterator;
import java.util.NoSuchElementException;
import org.drools.planner.core.heuristic.selector.move.factory.MoveListFactory;
import org.drools.planner.core.move.Move;
import org.drools.planner.core.move.CompositeMove;
import org.drools.planner.core.solution.Solution;
import org.drools.planner.examples.pas.domain.BedDesignation;
import org.drools.planner.examples.pas.domain.PatientAdmissionSchedule;
import org.drools.planner.examples.pas.domain.Bed;
import org.drools.planner.examples.pas.solver.move.BedChangeMove;
import org.apache.commons.lang.builder.CompareToBuilder;
public class BedDesignationPillarPartSwapMoveFactory implements MoveListFactory {
public List createMoveList(Solution solution) {
PatientAdmissionSchedule patientAdmissionSchedule = (PatientAdmissionSchedule) solution;
Map> bedToBedDesignationList = new HashMap>(
patientAdmissionSchedule.getBedList().size());
for (BedDesignation bedDesignation : patientAdmissionSchedule.getBedDesignationList()) {
List bedDesignationListPerBed = bedToBedDesignationList.get(bedDesignation.getBed());
if (bedDesignationListPerBed == null) {
// Note: the initialCapacity is probably to high,
// which is bad for memory, but the opposite is bad for performance (which is worse)
bedDesignationListPerBed = new ArrayList(
patientAdmissionSchedule.getNightList().size());
bedToBedDesignationList.put(bedDesignation.getBed(), bedDesignationListPerBed);
}
bedDesignationListPerBed.add(bedDesignation);
}
for (List bedDesignationListPerBed : bedToBedDesignationList.values()) {
Collections.sort(bedDesignationListPerBed, new Comparator() {
public int compare(BedDesignation a, BedDesignation b) {
// This comparison is sameBedInSameNight safe.
return new CompareToBuilder()
.append(a.getAdmissionPart().getFirstNight(), b.getAdmissionPart().getFirstNight())
.append(a.getAdmissionPart().getLastNight(), b.getAdmissionPart().getLastNight())
.append(a.getAdmissionPart(), b.getAdmissionPart())
.toComparison();
}
});
}
List bedList = patientAdmissionSchedule.getBedList();
List moveList = new ArrayList();
// For every 2 distinct beds
for (ListIterator leftBedIt = bedList.listIterator(); leftBedIt.hasNext();) {
Bed leftBed = leftBedIt.next();
for (ListIterator rightBedIt = bedList.listIterator(leftBedIt.nextIndex());
rightBedIt.hasNext();) {
Bed rightBed = rightBedIt.next();
List leftBedDesignationList = bedToBedDesignationList.get(leftBed);
if (leftBedDesignationList == null) {
leftBedDesignationList = Collections.emptyList();
}
List rightBedDesignationList = bedToBedDesignationList.get(rightBed);
if (rightBedDesignationList == null) {
rightBedDesignationList = Collections.emptyList();
}
LowestFirstNightBedDesignationIterator lowestIt = new LowestFirstNightBedDesignationIterator(
leftBedDesignationList, rightBedDesignationList);
// For every pillar part duo
while (lowestIt.hasNext()) {
BedDesignation pillarPartBedDesignation = lowestIt.next();
// Note: the initialCapacity is probably to high,
// which is bad for memory, but the opposite is bad for performance (which is worse)
List moveListByPillarPartDuo = new ArrayList(
leftBedDesignationList.size() + rightBedDesignationList.size());
int lastNightIndex = pillarPartBedDesignation.getAdmissionPart().getLastNight().getIndex();
Bed otherBed;
int leftMinimumFirstNightIndex = Integer.MIN_VALUE;
int rightMinimumFirstNightIndex = Integer.MIN_VALUE;
if (lowestIt.isLastNextWasLeft()) {
otherBed = rightBed;
leftMinimumFirstNightIndex = lastNightIndex;
} else {
otherBed = leftBed;
rightMinimumFirstNightIndex = lastNightIndex;
}
moveListByPillarPartDuo.add(new BedChangeMove(pillarPartBedDesignation, otherBed));
// For every BedDesignation in that pillar part duo
while (lowestIt.hasNextWithMaximumFirstNightIndexes(
leftMinimumFirstNightIndex, rightMinimumFirstNightIndex)) {
pillarPartBedDesignation = lowestIt.next();
lastNightIndex = pillarPartBedDesignation.getAdmissionPart().getLastNight().getIndex();
if (lowestIt.isLastNextWasLeft()) {
otherBed = rightBed;
leftMinimumFirstNightIndex = Math.max(leftMinimumFirstNightIndex, lastNightIndex);
} else {
otherBed = leftBed;
rightMinimumFirstNightIndex = Math.max(rightMinimumFirstNightIndex, lastNightIndex);
}
moveListByPillarPartDuo.add(new BedChangeMove(pillarPartBedDesignation, otherBed));
}
moveList.add(new CompositeMove(moveListByPillarPartDuo));
}
}
}
return moveList;
}
private static class LowestFirstNightBedDesignationIterator implements Iterator {
private Iterator leftIterator;
private Iterator rightIterator;
private boolean leftHasNext = true;
private boolean rightHasNext = true;
private BedDesignation nextLeft;
private BedDesignation nextRight;
private boolean lastNextWasLeft;
public LowestFirstNightBedDesignationIterator(
List leftBedDesignationList, List rightBedDesignationList) {
// Buffer the nextLeft and nextRight
leftIterator = leftBedDesignationList.iterator();
if (leftIterator.hasNext()) {
nextLeft = leftIterator.next();
} else {
leftHasNext = false;
nextLeft = null;
}
rightIterator = rightBedDesignationList.iterator();
if (rightIterator.hasNext()) {
nextRight = rightIterator.next();
} else {
rightHasNext = false;
nextRight = null;
}
}
public boolean hasNext() {
return leftHasNext || rightHasNext;
}
public boolean hasNextWithMaximumFirstNightIndexes(
int leftMinimumFirstNightIndex, int rightMinimumFirstNightIndex) {
if (!hasNext()) {
return false;
}
boolean nextIsLeft = nextIsLeft();
if (nextIsLeft) {
int firstNightIndex = nextLeft.getAdmissionPart().getFirstNight().getIndex();
// It should not be conflict in the same pillar and it should be in conflict with the other pillar
return firstNightIndex > leftMinimumFirstNightIndex && firstNightIndex <= rightMinimumFirstNightIndex;
} else {
int firstNightIndex = nextRight.getAdmissionPart().getFirstNight().getIndex();
// It should not be conflict in the same pillar and it should be in conflict with the other pillar
return firstNightIndex > rightMinimumFirstNightIndex && firstNightIndex <= leftMinimumFirstNightIndex;
}
}
public BedDesignation next() {
lastNextWasLeft = nextIsLeft();
// Buffer the nextLeft or nextRight
BedDesignation lowest;
if (lastNextWasLeft) {
lowest = nextLeft;
if (leftIterator.hasNext()) {
nextLeft = leftIterator.next();
} else {
leftHasNext = false;
nextLeft = null;
}
} else {
lowest = nextRight;
if (rightIterator.hasNext()) {
nextRight = rightIterator.next();
} else {
rightHasNext = false;
nextRight = null;
}
}
return lowest;
}
private boolean nextIsLeft() {
boolean returnLeft;
if (leftHasNext) {
if (rightHasNext) {
int leftFirstNightIndex = nextLeft.getAdmissionPart().getFirstNight().getIndex();
int rightFirstNightIndex = nextRight.getAdmissionPart().getFirstNight().getIndex();
returnLeft = leftFirstNightIndex <= rightFirstNightIndex;
} else {
returnLeft = true;
}
} else {
if (rightHasNext) {
returnLeft = false;
} else {
throw new NoSuchElementException();
}
}
return returnLeft;
}
public void remove() {
throw new UnsupportedOperationException("Remove is not supported.");
}
public boolean isLastNextWasLeft() {
return lastNextWasLeft;
}
}
}