org.drools.planner.examples.travelingtournament.solver.smart.move.MultipleMatchListRotateMove 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.travelingtournament.solver.smart.move;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;
import org.drools.WorkingMemory;
import org.drools.FactHandle;
import org.drools.planner.core.localsearch.decider.acceptor.tabu.TabuPropertyEnabled;
import org.drools.planner.core.move.Move;
import org.drools.planner.examples.travelingtournament.domain.Day;
import org.drools.planner.examples.travelingtournament.domain.Match;
public class MultipleMatchListRotateMove implements Move, TabuPropertyEnabled {
private List firstMatchList;
private List secondMatchList;
public MultipleMatchListRotateMove(List firstMatchList, List secondMatchList) {
this.firstMatchList = firstMatchList;
this.secondMatchList = secondMatchList;
}
public boolean isMoveDoable(WorkingMemory workingMemory) {
return true;
}
public Move createUndoMove(WorkingMemory workingMemory) {
List inverseFirstMatchList = new ArrayList(firstMatchList);
Collections.reverse(inverseFirstMatchList);
List inverseSecondMatchList = new ArrayList(secondMatchList);
Collections.reverse(inverseSecondMatchList);
return new MultipleMatchListRotateMove(inverseFirstMatchList, inverseSecondMatchList);
}
public void doMove(WorkingMemory workingMemory) {
rotateList(firstMatchList, workingMemory);
if (!secondMatchList.isEmpty()) { // TODO create SingleMatchListRotateMove
rotateList(secondMatchList, workingMemory);
}
}
private void rotateList(List matchList, WorkingMemory workingMemory) {
Iterator it = matchList.iterator();
Match firstMatch = it.next();
Match secondMatch = null;
Day startDay = firstMatch.getDay();
while (it.hasNext()) {
secondMatch = it.next();
FactHandle firstMatchHandle = workingMemory.getFactHandle(firstMatch);
firstMatch.setDay(secondMatch.getDay());
workingMemory.update(firstMatchHandle, firstMatch);
firstMatch = secondMatch;
}
FactHandle secondMatchHandle = workingMemory.getFactHandle(firstMatch);
secondMatch.setDay(startDay);
workingMemory.update(secondMatchHandle, secondMatch);
}
public Collection getTabuProperties() {
List tabuPropertyList = new ArrayList(firstMatchList.size() + secondMatchList.size());
tabuPropertyList.addAll(firstMatchList);
tabuPropertyList.addAll(secondMatchList);
return tabuPropertyList;
}
public boolean equals(Object o) {
if (this == o) {
return true;
} else if (o instanceof MultipleMatchListRotateMove) {
MultipleMatchListRotateMove other = (MultipleMatchListRotateMove) o;
return new EqualsBuilder()
.append(firstMatchList, other.firstMatchList)
.append(secondMatchList, other.secondMatchList)
.isEquals();
} else {
return false;
}
}
public int hashCode() {
return new HashCodeBuilder()
.append(firstMatchList)
.append(secondMatchList)
.toHashCode();
}
public String toString() {
return firstMatchList + " & " + secondMatchList;
}
}