org.drools.planner.examples.travelingtournament.solver.smart.move.MultipleMatchListRotateMove Maven / Gradle / Ivy
/*
* 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.planner.core.move.Move;
import org.drools.planner.core.score.director.ScoreDirector;
import org.drools.planner.examples.travelingtournament.domain.Day;
import org.drools.planner.examples.travelingtournament.domain.Match;
import org.drools.planner.examples.travelingtournament.solver.move.TravelingTournamentMoveHelper;
public class MultipleMatchListRotateMove implements Move {
private List firstMatchList;
private List secondMatchList;
public MultipleMatchListRotateMove(List firstMatchList, List secondMatchList) {
this.firstMatchList = firstMatchList;
this.secondMatchList = secondMatchList;
}
public boolean isMoveDoable(ScoreDirector scoreDirector) {
return true;
}
public Move createUndoMove(ScoreDirector scoreDirector) {
List inverseFirstMatchList = new ArrayList(firstMatchList);
Collections.reverse(inverseFirstMatchList);
List inverseSecondMatchList = new ArrayList(secondMatchList);
Collections.reverse(inverseSecondMatchList);
return new MultipleMatchListRotateMove(inverseFirstMatchList, inverseSecondMatchList);
}
public void doMove(ScoreDirector scoreDirector) {
rotateList(scoreDirector, firstMatchList);
if (!secondMatchList.isEmpty()) { // TODO create SingleMatchListRotateMove
rotateList(scoreDirector, secondMatchList);
}
}
private void rotateList(ScoreDirector scoreDirector, List matchList) {
Iterator it = matchList.iterator();
Match previousMatch = it.next();
Match match = null;
Day firstDay = previousMatch.getDay();
while (it.hasNext()) {
match = it.next();
TravelingTournamentMoveHelper.moveDay(scoreDirector, previousMatch, match.getDay());
previousMatch = match;
}
TravelingTournamentMoveHelper.moveDay(scoreDirector, match, firstDay);
}
public Collection getPlanningEntities() {
List entities = new ArrayList(firstMatchList.size() + secondMatchList.size());
entities.addAll(firstMatchList);
entities.addAll(secondMatchList);
return entities;
}
public Collection getPlanningValues() {
List values = new ArrayList(firstMatchList.size() + secondMatchList.size());
for (Match match : firstMatchList) {
values.add(match.getDay());
}
for (Match match : secondMatchList) {
values.add(match.getDay());
}
return values;
}
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;
}
}