/** * 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.nurserostering.domain.solver; import java.io.Serializable; 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.examples.nurserostering.domain.DayOfWeek; import org.drools.planner.examples.nurserostering.domain.Employee; import org.drools.planner.examples.nurserostering.domain.ShiftDate; import org.drools.planner.examples.nurserostering.domain.WeekendDefinition; import org.drools.planner.examples.nurserostering.domain.contract.Contract; /** * @author Geoffrey De Smet */ public class EmployeeConsecutiveAssignmentStart implements Comparable<EmployeeConsecutiveAssignmentStart>, Serializable { private Employee employee; private ShiftDate shiftDate; public EmployeeConsecutiveAssignmentStart(Employee employee, ShiftDate shiftDate) { this.employee = employee; this.shiftDate = shiftDate; } public Employee getEmployee() { return employee; } public void setEmployee(Employee employee) { this.employee = employee; } public ShiftDate getShiftDate() { return shiftDate; } public void setShiftDate(ShiftDate shiftDate) { this.shiftDate = shiftDate; } public boolean equals(Object o) { if (this == o) { return true; } else if (o instanceof EmployeeConsecutiveAssignmentStart) { EmployeeConsecutiveAssignmentStart other = (EmployeeConsecutiveAssignmentStart) o; return new EqualsBuilder() .append(employee, other.employee) .append(shiftDate, other.shiftDate) .isEquals(); } else { return false; } } public int hashCode() { return new HashCodeBuilder() .append(employee) .append(shiftDate) .toHashCode(); } public int compareTo(EmployeeConsecutiveAssignmentStart other) { return new CompareToBuilder() .append(employee, other.employee) .append(shiftDate, other.shiftDate) .toComparison(); } @Override public String toString() { return employee + " " + shiftDate + " - ..."; } public Contract getContract() { return employee.getContract(); } public int getShiftDateDayIndex() { return shiftDate.getDayIndex(); } public boolean isWeekendAndNotFirstDayOfWeekend() { WeekendDefinition weekendDefinition = employee.getContract().getWeekendDefinition(); DayOfWeek dayOfWeek = shiftDate.getDayOfWeek(); return weekendDefinition.isWeekend(dayOfWeek) && weekendDefinition.getFirstDayOfWeekend() != dayOfWeek; } public int getDistanceToFirstDayOfWeekend() { WeekendDefinition weekendDefinition = employee.getContract().getWeekendDefinition(); DayOfWeek dayOfWeek = shiftDate.getDayOfWeek(); return weekendDefinition.getFirstDayOfWeekend().getDistanceToNext(dayOfWeek); } }