org.drools.planner.examples.common.business.ScoreDetail 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.common.business;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.apache.commons.lang.builder.CompareToBuilder;
import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;
import org.drools.planner.core.score.constraint.ConstraintOccurrence;
import org.drools.planner.core.score.constraint.ConstraintType;
import org.drools.planner.core.score.constraint.DoubleConstraintOccurrence;
import org.drools.planner.core.score.constraint.IntConstraintOccurrence;
import org.drools.planner.core.score.constraint.LongConstraintOccurrence;
import org.drools.planner.core.score.constraint.UnweightedConstraintOccurrence;
/**
* TODO Replace this class with the ConstraintOccurrenceTotal class: https://jira.jboss.org/jira/browse/JBRULES-2510
*/
public class ScoreDetail implements Comparable {
private String ruleId;
private ConstraintType constraintType;
private Set constraintOccurrenceSet = new HashSet();
private double scoreTotal = 0.0;
public ScoreDetail(String ruleId, ConstraintType constraintType) {
this.ruleId = ruleId;
this.constraintType = constraintType;
}
public String getRuleId() {
return ruleId;
}
public ConstraintType getConstraintType() {
return constraintType;
}
public Set getConstraintOccurrenceSet() {
return constraintOccurrenceSet;
}
public int getOccurrenceSize() {
return constraintOccurrenceSet.size();
}
public double getScoreTotal() {
return scoreTotal;
}
public void addConstraintOccurrence(ConstraintOccurrence constraintOccurrence) {
boolean added = constraintOccurrenceSet.add(constraintOccurrence);
if (!added) {
throw new IllegalArgumentException("Add the same constraintOccurrence (" + constraintOccurrence
+ ") twice.");
}
double occurrenceScore;
if (constraintOccurrence instanceof IntConstraintOccurrence) {
occurrenceScore = ((IntConstraintOccurrence) constraintOccurrence).getWeight();
} else if (constraintOccurrence instanceof DoubleConstraintOccurrence) {
occurrenceScore = ((DoubleConstraintOccurrence) constraintOccurrence).getWeight();
} else if (constraintOccurrence instanceof LongConstraintOccurrence) {
occurrenceScore = ((LongConstraintOccurrence) constraintOccurrence).getWeight();
} else if (constraintOccurrence instanceof UnweightedConstraintOccurrence) {
occurrenceScore = 1.0;
} else {
throw new IllegalStateException("Cannot determine occurrenceScore of ConstraintOccurrence class: "
+ constraintOccurrence.getClass());
}
scoreTotal += occurrenceScore;
}
public boolean equals(Object o) {
if (this == o) {
return true;
} else if (o instanceof ScoreDetail) {
ScoreDetail other = (ScoreDetail) o;
return new EqualsBuilder()
.append(ruleId, other.ruleId)
.append(constraintType, other.constraintType)
.isEquals();
} else {
return false;
}
}
public int hashCode() {
return new HashCodeBuilder()
.append(ruleId)
.append(constraintType)
.toHashCode();
}
public int compareTo(ScoreDetail other) {
return new CompareToBuilder()
.append(constraintType, other.constraintType)
.append(ruleId, other.ruleId)
.toComparison();
}
public String toString() {
return ruleId + "/" + constraintType + " (" + getOccurrenceSize() + ") = " + scoreTotal;
}
public String buildConstraintOccurrenceListText() {
List constraintOccurrenceList = new ArrayList(constraintOccurrenceSet);
Collections.sort(constraintOccurrenceList);
StringBuilder text = new StringBuilder(constraintOccurrenceList.size() * 80);
for (ConstraintOccurrence constraintOccurrence : constraintOccurrenceList) {
text.append(constraintOccurrence.toString()).append("\n");
}
return text.toString();
}
}