org.optaplanner.examples.conferencescheduling.persistence.ConferenceSchedulingXlsxFileIO Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of optaplanner-examples Show documentation
Show all versions of optaplanner-examples Show documentation
OptaPlanner solves planning problems.
This lightweight, embeddable planning engine implements powerful and scalable algorithms
to optimize business resource scheduling and planning.
This module contains the examples which demonstrate how to use it in a normal Java application.
/*
* Copyright 2017 Red Hat, Inc. and/or its affiliates.
*
* 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.optaplanner.examples.conferencescheduling.persistence;
import static java.util.Collections.disjoint;
import static java.util.Collections.emptyList;
import static java.util.Collections.reverseOrder;
import static java.util.Comparator.comparing;
import static java.util.stream.Collectors.counting;
import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.joining;
import static java.util.stream.Collectors.mapping;
import static java.util.stream.Collectors.toCollection;
import static java.util.stream.Collectors.toList;
import static java.util.stream.Collectors.toMap;
import static java.util.stream.Collectors.toSet;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.AUDIENCE_LEVEL_DIVERSITY;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.AUDIENCE_TYPE_DIVERSITY;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.AUDIENCE_TYPE_THEME_TRACK_CONFLICT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.CONSECUTIVE_TALKS_PAUSE;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.CONTENT_CONFLICT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.CROWD_CONTROL;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.LANGUAGE_DIVERSITY;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.POPULAR_TALKS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.PUBLISHED_ROOM;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.PUBLISHED_TIMESLOT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.ROOM_CONFLICT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.ROOM_UNAVAILABLE_TIMESLOT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SAME_DAY_TALKS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SECTOR_CONFLICT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_CONFLICT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_PREFERRED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_PREFERRED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_PROHIBITED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_PROHIBITED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_REQUIRED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_REQUIRED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_UNAVAILABLE_TIMESLOT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_UNDESIRED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.SPEAKER_UNDESIRED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_MUTUALLY_EXCLUSIVE_TALKS_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_PREFERRED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_PREFERRED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_PREREQUISITE_TALKS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_PROHIBITED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_PROHIBITED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_REQUIRED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_REQUIRED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_UNDESIRED_ROOM_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.TALK_UNDESIRED_TIMESLOT_TAGS;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.THEME_TRACK_CONFLICT;
import static org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration.THEME_TRACK_ROOM_STABILITY;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.format.DateTimeParseException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.poi.ss.usermodel.ClientAnchor;
import org.apache.poi.ss.usermodel.Comment;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFColor;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.optaplanner.core.api.score.Score;
import org.optaplanner.core.api.score.buildin.hardmediumsoft.HardMediumSoftScore;
import org.optaplanner.core.api.score.constraint.ConstraintMatch;
import org.optaplanner.core.api.score.constraint.Indictment;
import org.optaplanner.examples.common.persistence.AbstractXlsxSolutionFileIO;
import org.optaplanner.examples.conferencescheduling.app.ConferenceSchedulingApp;
import org.optaplanner.examples.conferencescheduling.domain.ConferenceConstraintConfiguration;
import org.optaplanner.examples.conferencescheduling.domain.ConferenceSolution;
import org.optaplanner.examples.conferencescheduling.domain.Room;
import org.optaplanner.examples.conferencescheduling.domain.Speaker;
import org.optaplanner.examples.conferencescheduling.domain.Talk;
import org.optaplanner.examples.conferencescheduling.domain.TalkType;
import org.optaplanner.examples.conferencescheduling.domain.Timeslot;
import org.optaplanner.swing.impl.TangoColorFactory;
public class ConferenceSchedulingXlsxFileIO extends AbstractXlsxSolutionFileIO {
private static final String ROOM_UNAVAILABLE_TIMESLOT_DESCRIPTION =
"Penalty per talk with an unavailable room in its timeslot, per minute";
private static final String ROOM_CONFLICT_DESCRIPTION =
"Penalty per 2 talks in the same room and overlapping timeslots, per overlapping minute";
private static final String SPEAKER_UNAVAILABLE_TIMESLOT_DESCRIPTION =
"Penalty per talk with an unavailable speaker in its timeslot, per minute";
private static final String SPEAKER_CONFLICT_DESCRIPTION =
"Penalty per 2 talks with the same speaker and overlapping timeslots, per overlapping minute";
private static final String TALK_PREREQUISITE_TALKS_DESCRIPTION =
"Penalty per prerequisite talk of a talk that doesn't end before the second talk starts, per minute of either talk";
private static final String TALK_MUTUALLY_EXCLUSIVE_TALKS_TAGS_DESCRIPTION =
"Penalty per common mutually exclusive talks tag of 2 talks with overlapping timeslots, per overlapping minute";
private static final String CONSECUTIVE_TALKS_PAUSE_DESCRIPTION =
"Penalty per 2 consecutive talks for the same speaker with a pause less than the minimum pause, per minute of either talk";
private static final String CROWD_CONTROL_DESCRIPTION =
"Penalty per talk with a non-zero crowd control risk that are not in paired with exactly one other such talk, per minute of either talk";
private static final String SPEAKER_REQUIRED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per missing required tag in a talk's timeslot, per minute";
private static final String SPEAKER_PROHIBITED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per prohibited tag in a talk's timeslot, per minute";
private static final String TALK_REQUIRED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per missing required tag in a talk's timeslot, per minute";
private static final String TALK_PROHIBITED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per prohibited tag in a talk's timeslot, per minute";
private static final String SPEAKER_REQUIRED_ROOM_TAGS_DESCRIPTION =
"Penalty per missing required tag in a talk's room, per minute";
private static final String SPEAKER_PROHIBITED_ROOM_TAGS_DESCRIPTION =
"Penalty per prohibited tag in a talk's room, per minute";
private static final String TALK_REQUIRED_ROOM_TAGS_DESCRIPTION =
"Penalty per missing required tag in a talk's room, per minute";
private static final String TALK_PROHIBITED_ROOM_TAGS_DESCRIPTION =
"Penalty per prohibited tag in a talk's room, per minute";
private static final String PUBLISHED_TIMESLOT_DESCRIPTION =
"Penalty per published talk with a different timeslot than its published timeslot, per match";
private static final String PUBLISHED_ROOM_DESCRIPTION =
"Penalty per published talk with a different room than its published room, per match";
private static final String THEME_TRACK_CONFLICT_DESCRIPTION =
"Penalty per common theme track of 2 talks with overlapping timeslots, per overlapping minute";
private static final String THEME_TRACK_ROOM_STABILITY_DESCRIPTION =
"Penalty per common theme track of 2 talks in a different room on the same day, per minute of either talk";
private static final String SECTOR_CONFLICT_DESCRIPTION =
"Penalty per common sector of 2 talks with overlapping timeslots, per overlapping minute";
private static final String AUDIENCE_TYPE_DIVERSITY_DESCRIPTION =
"Reward per 2 talks with a different audience type and the same timeslot, per (overlapping) minute";
private static final String AUDIENCE_TYPE_THEME_TRACK_CONFLICT_DESCRIPTION =
"Penalty per 2 talks with a common audience type, a common theme track and overlapping timeslots, per overlapping minute";
private static final String AUDIENCE_LEVEL_DIVERSITY_DESCRIPTION =
"Reward per 2 talks with a different audience level and the same timeslot, per (overlapping) minute";
private static final String CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION_DESCRIPTION =
"Penalty per common content of 2 talks with a different audience level for which the easier talk isn't scheduled earlier than the other talk, per minute of either talk";
private static final String CONTENT_CONFLICT_DESCRIPTION =
"Penalty per common content of 2 talks with overlapping timeslots, per overlapping minute";
private static final String LANGUAGE_DIVERSITY_DESCRIPTION =
"Reward per 2 talks with a different language and the the same timeslot, per (overlapping) minute";
private static final String SAME_DAY_TALKS_DESCRIPTION =
"Penalty per common content or theme track of 2 talks with a different day, per minute of either talk";
private static final String POPULAR_TALKS_DESCRIPTION =
"Penalty per 2 talks where the less popular one (has lower favorite count) is assigned a larger room than the more popular talk";
private static final String SPEAKER_PREFERRED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per missing preferred tag in a talk's timeslot, per minute";
private static final String SPEAKER_UNDESIRED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per undesired tag in a talk's timeslot, per minute";
private static final String TALK_PREFERRED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per missing preferred tag in a talk's timeslot, per minute";
private static final String TALK_UNDESIRED_TIMESLOT_TAGS_DESCRIPTION =
"Penalty per undesired tag in a talk's timeslot, per minute";
private static final String SPEAKER_PREFERRED_ROOM_TAGS_DESCRIPTION =
"Penalty per missing preferred tag in a talk's room, per minute";
private static final String SPEAKER_UNDESIRED_ROOM_TAGS_DESCRIPTION =
"Penalty per undesired tag in a talk's room, per minute";
private static final String TALK_PREFERRED_ROOM_TAGS_DESCRIPTION =
"Penalty per missing preferred tag in a talk's room, per minute";
private static final String TALK_UNDESIRED_ROOM_TAGS_DESCRIPTION = "Penalty per undesired tag in a talk's room, per minute";
private static final Comparator COMPARATOR = comparing(Timeslot::getStartDateTime)
.thenComparing(reverseOrder(comparing(Timeslot::getEndDateTime)));
private boolean strict;
public ConferenceSchedulingXlsxFileIO() {
this(true);
}
public ConferenceSchedulingXlsxFileIO(boolean strict) {
super();
this.strict = strict;
}
@Override
public ConferenceSolution read(File inputSolutionFile) {
try (InputStream in = new BufferedInputStream(new FileInputStream(inputSolutionFile))) {
XSSFWorkbook workbook = new XSSFWorkbook(in);
return new ConferenceSchedulingXlsxReader(workbook).read();
} catch (IOException | RuntimeException e) {
throw new IllegalStateException("Failed reading inputSolutionFile ("
+ inputSolutionFile + ").", e);
}
}
private class ConferenceSchedulingXlsxReader extends AbstractXlsxReader {
private Map totalTalkTypeMap;
private Set totalTimeslotTagSet;
private Set totalRoomTagSet;
private Map totalTalkCodeMap;
public ConferenceSchedulingXlsxReader(XSSFWorkbook workbook) {
super(workbook, ConferenceSchedulingApp.SOLVER_CONFIG);
}
@Override
public ConferenceSolution read() {
solution = new ConferenceSolution();
totalTalkTypeMap = new HashMap<>();
totalTimeslotTagSet = new HashSet<>();
totalRoomTagSet = new HashSet<>();
totalTalkCodeMap = new HashMap<>();
readConfiguration();
readTimeslotList();
readRoomList();
readSpeakerList();
readTalkList();
// Needed for merging in the sheet Rooms views
solution.getTimeslotList().sort(COMPARATOR);
return solution;
}
private void readConfiguration() {
nextSheet("Configuration");
nextRow();
readHeaderCell("Conference name");
solution.setConferenceName(nextStringCell().getStringCellValue());
if (strict && !VALID_NAME_PATTERN.matcher(solution.getConferenceName()).matches()) {
throw new IllegalStateException(currentPosition() + ": The conference name (" + solution.getConferenceName()
+ ") must match to the regular expression (" + VALID_NAME_PATTERN + ").");
}
ConferenceConstraintConfiguration constraintConfiguration = new ConferenceConstraintConfiguration();
readIntConstraintParameterLine("Minimum consecutive talks pause in minutes",
constraintConfiguration::setMinimumConsecutiveTalksPauseInMinutes,
"The amount of time a speaker needs between 2 talks");
readScoreConstraintHeaders();
constraintConfiguration.setId(0L);
constraintConfiguration.setRoomUnavailableTimeslot(readScoreConstraintLine(ROOM_UNAVAILABLE_TIMESLOT,
ROOM_UNAVAILABLE_TIMESLOT_DESCRIPTION));
constraintConfiguration.setRoomConflict(readScoreConstraintLine(ROOM_CONFLICT,
ROOM_CONFLICT_DESCRIPTION));
constraintConfiguration.setSpeakerUnavailableTimeslot(readScoreConstraintLine(SPEAKER_UNAVAILABLE_TIMESLOT,
SPEAKER_UNAVAILABLE_TIMESLOT_DESCRIPTION));
constraintConfiguration.setSpeakerConflict(readScoreConstraintLine(SPEAKER_CONFLICT,
SPEAKER_CONFLICT_DESCRIPTION));
constraintConfiguration.setTalkPrerequisiteTalks(readScoreConstraintLine(TALK_PREREQUISITE_TALKS,
TALK_PREREQUISITE_TALKS_DESCRIPTION));
constraintConfiguration
.setTalkMutuallyExclusiveTalksTags(readScoreConstraintLine(TALK_MUTUALLY_EXCLUSIVE_TALKS_TAGS,
TALK_MUTUALLY_EXCLUSIVE_TALKS_TAGS_DESCRIPTION));
constraintConfiguration.setConsecutiveTalksPause(readScoreConstraintLine(CONSECUTIVE_TALKS_PAUSE,
CONSECUTIVE_TALKS_PAUSE_DESCRIPTION));
constraintConfiguration.setCrowdControl(readScoreConstraintLine(CROWD_CONTROL,
CROWD_CONTROL_DESCRIPTION));
constraintConfiguration.setSpeakerRequiredTimeslotTags(readScoreConstraintLine(SPEAKER_REQUIRED_TIMESLOT_TAGS,
SPEAKER_REQUIRED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setSpeakerProhibitedTimeslotTags(readScoreConstraintLine(SPEAKER_PROHIBITED_TIMESLOT_TAGS,
SPEAKER_PROHIBITED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setTalkRequiredTimeslotTags(readScoreConstraintLine(TALK_REQUIRED_TIMESLOT_TAGS,
TALK_REQUIRED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setTalkProhibitedTimeslotTags(readScoreConstraintLine(TALK_PROHIBITED_TIMESLOT_TAGS,
TALK_PROHIBITED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setSpeakerRequiredRoomTags(readScoreConstraintLine(SPEAKER_REQUIRED_ROOM_TAGS,
SPEAKER_REQUIRED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setSpeakerProhibitedRoomTags(readScoreConstraintLine(SPEAKER_PROHIBITED_ROOM_TAGS,
SPEAKER_PROHIBITED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setTalkRequiredRoomTags(readScoreConstraintLine(TALK_REQUIRED_ROOM_TAGS,
TALK_REQUIRED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setTalkProhibitedRoomTags(readScoreConstraintLine(TALK_PROHIBITED_ROOM_TAGS,
TALK_PROHIBITED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setPublishedTimeslot(readScoreConstraintLine(PUBLISHED_TIMESLOT,
PUBLISHED_TIMESLOT_DESCRIPTION));
constraintConfiguration.setPublishedRoom(readScoreConstraintLine(PUBLISHED_ROOM,
PUBLISHED_ROOM_DESCRIPTION));
constraintConfiguration.setThemeTrackConflict(readScoreConstraintLine(THEME_TRACK_CONFLICT,
THEME_TRACK_CONFLICT_DESCRIPTION));
constraintConfiguration.setThemeTrackRoomStability(readScoreConstraintLine(THEME_TRACK_ROOM_STABILITY,
THEME_TRACK_ROOM_STABILITY_DESCRIPTION));
constraintConfiguration.setSectorConflict(readScoreConstraintLine(SECTOR_CONFLICT,
SECTOR_CONFLICT_DESCRIPTION));
constraintConfiguration.setAudienceTypeDiversity(readScoreConstraintLine(AUDIENCE_TYPE_DIVERSITY,
AUDIENCE_TYPE_DIVERSITY_DESCRIPTION));
constraintConfiguration
.setAudienceTypeThemeTrackConflict(readScoreConstraintLine(AUDIENCE_TYPE_THEME_TRACK_CONFLICT,
AUDIENCE_TYPE_THEME_TRACK_CONFLICT_DESCRIPTION));
constraintConfiguration.setAudienceLevelDiversity(readScoreConstraintLine(AUDIENCE_LEVEL_DIVERSITY,
AUDIENCE_LEVEL_DIVERSITY_DESCRIPTION));
constraintConfiguration
.setContentAudienceLevelFlowViolation(readScoreConstraintLine(CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION,
CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION_DESCRIPTION));
constraintConfiguration.setContentConflict(readScoreConstraintLine(CONTENT_CONFLICT,
CONTENT_CONFLICT_DESCRIPTION));
constraintConfiguration.setLanguageDiversity(readScoreConstraintLine(LANGUAGE_DIVERSITY,
LANGUAGE_DIVERSITY_DESCRIPTION));
constraintConfiguration.setSameDayTalks(readScoreConstraintLine(SAME_DAY_TALKS,
SAME_DAY_TALKS_DESCRIPTION));
constraintConfiguration.setPopularTalks(readScoreConstraintLine(POPULAR_TALKS,
POPULAR_TALKS_DESCRIPTION));
constraintConfiguration.setSpeakerPreferredTimeslotTags(readScoreConstraintLine(SPEAKER_PREFERRED_TIMESLOT_TAGS,
SPEAKER_PREFERRED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setSpeakerUndesiredTimeslotTags(readScoreConstraintLine(SPEAKER_UNDESIRED_TIMESLOT_TAGS,
SPEAKER_UNDESIRED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setTalkPreferredTimeslotTags(readScoreConstraintLine(TALK_PREFERRED_TIMESLOT_TAGS,
TALK_PREFERRED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setTalkUndesiredTimeslotTags(readScoreConstraintLine(TALK_UNDESIRED_TIMESLOT_TAGS,
TALK_UNDESIRED_TIMESLOT_TAGS_DESCRIPTION));
constraintConfiguration.setSpeakerPreferredRoomTags(readScoreConstraintLine(SPEAKER_PREFERRED_ROOM_TAGS,
SPEAKER_PREFERRED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setSpeakerUndesiredRoomTags(readScoreConstraintLine(SPEAKER_UNDESIRED_ROOM_TAGS,
SPEAKER_UNDESIRED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setTalkPreferredRoomTags(readScoreConstraintLine(TALK_PREFERRED_ROOM_TAGS,
TALK_PREFERRED_ROOM_TAGS_DESCRIPTION));
constraintConfiguration.setTalkUndesiredRoomTags(readScoreConstraintLine(TALK_UNDESIRED_ROOM_TAGS,
TALK_UNDESIRED_ROOM_TAGS_DESCRIPTION));
solution.setConstraintConfiguration(constraintConfiguration);
}
private void readTimeslotList() {
nextSheet("Timeslots");
nextRow(false);
readHeaderCell("Day");
readHeaderCell("Start");
readHeaderCell("End");
readHeaderCell("Talk types");
readHeaderCell("Tags");
List talkTypeList = new ArrayList<>();
List timeslotList = new ArrayList<>(currentSheet.getLastRowNum() - 1);
long id = 0L;
long talkTypeId = 0L;
while (nextRow()) {
Timeslot timeslot = new Timeslot();
timeslot.setId(id++);
LocalDate day = LocalDate.parse(nextStringCell().getStringCellValue(), DAY_FORMATTER);
LocalTime startTime = LocalTime.parse(nextStringCell().getStringCellValue(), TIME_FORMATTER);
LocalTime endTime = LocalTime.parse(nextStringCell().getStringCellValue(), TIME_FORMATTER);
if (startTime.compareTo(endTime) >= 0) {
throw new IllegalStateException(currentPosition() + ": The startTime (" + startTime
+ ") must be less than the endTime (" + endTime + ").");
}
timeslot.setStartDateTime(LocalDateTime.of(day, startTime));
timeslot.setEndDateTime(LocalDateTime.of(day, endTime));
String[] talkTypeNames = nextStringCell().getStringCellValue().split(", ");
Set talkTypeSet = new LinkedHashSet<>(talkTypeNames.length);
for (String talkTypeName : talkTypeNames) {
TalkType talkType = totalTalkTypeMap.get(talkTypeName);
if (talkType == null) {
talkType = new TalkType(talkTypeId);
talkTypeId++;
if (strict && !VALID_TAG_PATTERN.matcher(talkTypeName).matches()) {
throw new IllegalStateException(currentPosition()
+ ": The timeslot (" + timeslot + ")'s talkType (" + talkTypeName
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
talkType.setName(talkTypeName);
talkType.setCompatibleTimeslotSet(new LinkedHashSet<>());
talkType.setCompatibleRoomSet(new LinkedHashSet<>());
totalTalkTypeMap.put(talkTypeName, talkType);
talkTypeList.add(talkType);
}
talkTypeSet.add(talkType);
talkType.getCompatibleTimeslotSet().add(timeslot);
}
if (talkTypeSet.isEmpty()) {
throw new IllegalStateException(currentPosition()
+ ": The timeslot (" + timeslot + ")'s talk types (" + timeslot.getTalkTypeSet()
+ ") must not be empty.");
}
timeslot.setTalkTypeSet(talkTypeSet);
timeslot.setTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
for (String tag : timeslot.getTagSet()) {
if (strict && !VALID_TAG_PATTERN.matcher(tag).matches()) {
throw new IllegalStateException(currentPosition()
+ ": The timeslot (" + timeslot + ")'s tag (" + tag
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
}
totalTimeslotTagSet.addAll(timeslot.getTagSet());
timeslotList.add(timeslot);
}
solution.setTimeslotList(timeslotList);
solution.setTalkTypeList(talkTypeList);
}
private void readRoomList() {
nextSheet("Rooms");
nextRow(false);
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readTimeslotDaysHeaders();
nextRow(false);
readHeaderCell("Name");
readHeaderCell("Capacity");
readHeaderCell("Talk types");
readHeaderCell("Tags");
readTimeslotHoursHeaders();
List roomList = new ArrayList<>(currentSheet.getLastRowNum() - 1);
long id = 0L;
while (nextRow()) {
Room room = new Room();
room.setId(id++);
room.setName(nextStringCell().getStringCellValue());
if (strict && !VALID_NAME_PATTERN.matcher(room.getName()).matches()) {
throw new IllegalStateException(currentPosition() + ": The room name (" + room.getName()
+ ") must match to the regular expression (" + VALID_NAME_PATTERN + ").");
}
room.setCapacity(getNextStrictlyPositiveIntegerCell("room name (" + room.getName(), "capacity"));
String[] talkTypeNames = nextStringCell().getStringCellValue().split(", ");
Set talkTypeSet;
if (talkTypeNames.length == 0 || (talkTypeNames.length == 1 && talkTypeNames[0].isEmpty())) {
talkTypeSet = new LinkedHashSet<>(totalTalkTypeMap.values());
for (TalkType talkType : talkTypeSet) {
talkType.getCompatibleRoomSet().add(room);
}
} else {
talkTypeSet = new LinkedHashSet<>(talkTypeNames.length);
for (String talkTypeName : talkTypeNames) {
TalkType talkType = totalTalkTypeMap.get(talkTypeName);
if (talkType == null) {
throw new IllegalStateException(currentPosition()
+ ": The room (" + room + ")'s talkType (" + talkTypeName
+ ") does not exist in the talk types (" + totalTalkTypeMap.keySet()
+ ") of the other sheet (Timeslots).");
}
talkTypeSet.add(talkType);
talkType.getCompatibleRoomSet().add(room);
}
}
room.setTalkTypeSet(talkTypeSet);
room.setTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
for (String tag : room.getTagSet()) {
if (strict && !VALID_TAG_PATTERN.matcher(tag).matches()) {
throw new IllegalStateException(currentPosition() + ": The room (" + room + ")'s tag (" + tag
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
}
totalRoomTagSet.addAll(room.getTagSet());
Set unavailableTimeslotSet = new LinkedHashSet<>();
for (Timeslot timeslot : solution.getTimeslotList()) {
XSSFCell cell = nextStringCell();
if (Objects.equals(extractColor(cell, UNAVAILABLE_COLOR), UNAVAILABLE_COLOR)) {
unavailableTimeslotSet.add(timeslot);
}
if (!cell.getStringCellValue().isEmpty()) {
throw new IllegalStateException(currentPosition() + ": The cell (" + cell.getStringCellValue()
+ ") should be empty. Use the talks sheet pre-assign rooms and timeslots.");
}
}
room.setUnavailableTimeslotSet(unavailableTimeslotSet);
roomList.add(room);
}
solution.setRoomList(roomList);
}
private void readSpeakerList() {
nextSheet("Speakers");
nextRow(false);
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readHeaderCell("");
readTimeslotDaysHeaders();
nextRow(false);
readHeaderCell("Name");
readHeaderCell("Required timeslot tags");
readHeaderCell("Preferred timeslot tags");
readHeaderCell("Prohibited timeslot tags");
readHeaderCell("Undesired timeslot tags");
readHeaderCell("Required room tags");
readHeaderCell("Preferred room tags");
readHeaderCell("Prohibited room tags");
readHeaderCell("Undesired room tags");
readTimeslotHoursHeaders();
List speakerList = new ArrayList<>(currentSheet.getLastRowNum() - 1);
long id = 0L;
while (nextRow()) {
Speaker speaker = new Speaker();
speaker.setId(id++);
speaker.setName(nextStringCell().getStringCellValue());
if (strict && !VALID_NAME_PATTERN.matcher(speaker.getName()).matches()) {
throw new IllegalStateException(currentPosition() + ": The speaker name (" + speaker.getName()
+ ") must match to the regular expression (" + VALID_NAME_PATTERN + ").");
}
speaker.setRequiredTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(speaker.getRequiredTimeslotTagSet());
speaker.setPreferredTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(speaker.getPreferredTimeslotTagSet());
speaker.setProhibitedTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(speaker.getProhibitedTimeslotTagSet());
speaker.setUndesiredTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(speaker.getUndesiredTimeslotTagSet());
speaker.setRequiredRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(speaker.getRequiredRoomTagSet());
speaker.setPreferredRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(speaker.getPreferredRoomTagSet());
speaker.setProhibitedRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(speaker.getProhibitedRoomTagSet());
speaker.setUndesiredRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(speaker.getUndesiredRoomTagSet());
Set unavailableTimeslotSet = new LinkedHashSet<>();
for (Timeslot timeslot : solution.getTimeslotList()) {
XSSFCell cell = nextStringCell();
if (Objects.equals(extractColor(cell, UNAVAILABLE_COLOR), UNAVAILABLE_COLOR)) {
unavailableTimeslotSet.add(timeslot);
}
if (!cell.getStringCellValue().isEmpty()) {
throw new IllegalStateException(currentPosition() + ": The cell (" + cell.getStringCellValue()
+ ") should be empty. Use the other sheet (Talks) to pre-assign rooms and timeslots.");
}
}
speaker.setUnavailableTimeslotSet(unavailableTimeslotSet);
speakerList.add(speaker);
}
solution.setSpeakerList(speakerList);
}
private void readTalkList() {
Map speakerMap = solution.getSpeakerList().stream().collect(
toMap(Speaker::getName, speaker -> speaker));
nextSheet("Talks");
nextRow(false);
readHeaderCell("Code");
readHeaderCell("Title");
readHeaderCell("Talk type");
readHeaderCell("Speakers");
readHeaderCell("Theme track tags");
readHeaderCell("Sector tags");
readHeaderCell("Audience types");
readHeaderCell("Audience level");
readHeaderCell("Content tags");
readHeaderCell("Language");
readHeaderCell("Required timeslot tags");
readHeaderCell("Preferred timeslot tags");
readHeaderCell("Prohibited timeslot tags");
readHeaderCell("Undesired timeslot tags");
readHeaderCell("Required room tags");
readHeaderCell("Preferred room tags");
readHeaderCell("Prohibited room tags");
readHeaderCell("Undesired room tags");
readHeaderCell("Mutually exclusive talks tags");
readHeaderCell("Prerequisite talks codes");
readHeaderCell("Favorite count");
readHeaderCell("Crowd control risk");
readHeaderCell("Pinned by user");
readHeaderCell("Timeslot day");
readHeaderCell("Start");
readHeaderCell("End");
readHeaderCell("Room");
readHeaderCell("Published Timeslot");
readHeaderCell("Published Start");
readHeaderCell("Published End");
readHeaderCell("Published Room");
List talkList = new ArrayList<>(currentSheet.getLastRowNum() - 1);
long id = 0L;
Map, Timeslot> timeslotMap = solution.getTimeslotList().stream().collect(
Collectors.toMap(timeslot -> Pair.of(timeslot.getStartDateTime(), timeslot.getEndDateTime()),
Function.identity()));
Map roomMap = solution.getRoomList().stream().collect(
Collectors.toMap(Room::getName, Function.identity()));
Map> talkToPrerequisiteTalkSetMap = new HashMap<>();
while (nextRow()) {
Talk talk = new Talk();
talk.setId(id++);
talk.setCode(nextStringCell().getStringCellValue());
totalTalkCodeMap.put(talk.getCode(), talk);
if (strict && !VALID_CODE_PATTERN.matcher(talk.getCode()).matches()) {
throw new IllegalStateException(currentPosition() + ": The talk code (" + talk.getCode()
+ ") must match to the regular expression (" + VALID_CODE_PATTERN + ").");
}
talk.setTitle(nextStringCell().getStringCellValue());
String talkTypeName = nextStringCell().getStringCellValue();
TalkType talkType = totalTalkTypeMap.get(talkTypeName);
if (talkType == null) {
throw new IllegalStateException(currentPosition()
+ ": The talk (" + talk + ")'s talkType (" + talkTypeName
+ ") does not exist in the talk types (" + totalTalkTypeMap.keySet()
+ ") of the other sheet (Timeslots).");
}
talk.setTalkType(talkType);
talk.setSpeakerList(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).map(speakerName -> {
Speaker speaker = speakerMap.get(speakerName);
if (speaker == null) {
throw new IllegalStateException(currentPosition() + ": The talk with code (" + talk.getCode()
+ ") has a speaker (" + speakerName + ") that doesn't exist in the speaker list.");
}
return speaker;
}).collect(toList()));
talk.setThemeTrackTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
for (String tag : talk.getThemeTrackTagSet()) {
if (strict && !VALID_TAG_PATTERN.matcher(tag).matches()) {
throw new IllegalStateException(currentPosition() + ": The talk (" + talk + ")'s theme tag (" + tag
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
}
talk.setSectorTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
for (String tag : talk.getSectorTagSet()) {
if (strict && !VALID_TAG_PATTERN.matcher(tag).matches()) {
throw new IllegalStateException(currentPosition() + ": The talk (" + talk + ")'s sector tag (" + tag
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
}
talk.setAudienceTypeSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
for (String audienceType : talk.getAudienceTypeSet()) {
if (strict && !VALID_TAG_PATTERN.matcher(audienceType).matches()) {
throw new IllegalStateException(
currentPosition() + ": The talk (" + talk + ")'s audience type (" + audienceType
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
}
talk.setAudienceLevel(
getNextStrictlyPositiveIntegerCell("talk with code (" + talk.getCode(), "an audience level"));
talk.setContentTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
for (String tag : talk.getContentTagSet()) {
if (strict && !VALID_TAG_PATTERN.matcher(tag).matches()) {
throw new IllegalStateException(currentPosition() + ": The talk (" + talk + ")'s content tag (" + tag
+ ") must match to the regular expression (" + VALID_TAG_PATTERN + ").");
}
}
talk.setLanguage(nextStringCell().getStringCellValue());
talk.setRequiredTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(talk.getRequiredTimeslotTagSet());
talk.setPreferredTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(talk.getPreferredTimeslotTagSet());
talk.setProhibitedTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(talk.getProhibitedTimeslotTagSet());
talk.setUndesiredTimeslotTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyTimeslotTags(talk.getUndesiredTimeslotTagSet());
talk.setRequiredRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(talk.getRequiredRoomTagSet());
talk.setPreferredRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(talk.getPreferredRoomTagSet());
talk.setProhibitedRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(talk.getProhibitedRoomTagSet());
talk.setUndesiredRoomTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(toCollection(LinkedHashSet::new)));
verifyRoomTags(talk.getUndesiredRoomTagSet());
talk.setMutuallyExclusiveTalksTagSet(Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(Collectors.toCollection(LinkedHashSet::new)));
talkToPrerequisiteTalkSetMap.put(talk, Arrays.stream(nextStringCell().getStringCellValue().split(", "))
.filter(tag -> !tag.isEmpty()).collect(Collectors.toCollection(LinkedHashSet::new)));
talk.setFavoriteCount(getNextPositiveIntegerCell("talk with code (" + talk.getCode(), "a Favorite count"));
talk.setCrowdControlRisk(
getNextPositiveIntegerCell("talk with code (" + talk.getCode(), "a crowd control risk"));
talk.setPinnedByUser(nextBooleanCell().getBooleanCellValue());
talk.setTimeslot(extractTimeslot(timeslotMap, talk));
talk.setRoom(extractRoom(roomMap, talk));
talk.setPublishedTimeslot(extractTimeslot(timeslotMap, talk));
talk.setPublishedRoom(extractRoom(roomMap, talk));
talkList.add(talk);
}
setPrerequisiteTalkSets(talkToPrerequisiteTalkSetMap);
solution.setTalkList(talkList);
}
private Timeslot extractTimeslot(Map, Timeslot> timeslotMap, Talk talk) {
Timeslot assignedTimeslot;
String dateString = nextStringCell().getStringCellValue();
String startTimeString = nextStringCell().getStringCellValue();
String endTimeString = nextStringCell().getStringCellValue();
if (!dateString.isEmpty() || !startTimeString.isEmpty() || !endTimeString.isEmpty()) {
LocalDateTime startDateTime;
LocalDateTime endDateTime;
try {
startDateTime = LocalDateTime.of(LocalDate.parse(dateString, DAY_FORMATTER),
LocalTime.parse(startTimeString, TIME_FORMATTER));
endDateTime = LocalDateTime.of(LocalDate.parse(dateString, DAY_FORMATTER),
LocalTime.parse(endTimeString, TIME_FORMATTER));
} catch (DateTimeParseException e) {
throw new IllegalStateException(currentPosition() + ": The talk with code (" + talk.getCode()
+ ") has a timeslot date (" + dateString
+ "), startTime (" + startTimeString + ") and endTime (" + endTimeString
+ ") that doesn't parse as a date or time.", e);
}
assignedTimeslot = timeslotMap.get(Pair.of(startDateTime, endDateTime));
if (assignedTimeslot == null) {
throw new IllegalStateException(currentPosition() + ": The talk with code (" + talk.getCode()
+ ") has a timeslot date (" + dateString
+ "), startTime (" + startTimeString + ") and endTime (" + endTimeString
+ ") that doesn't exist in the other sheet (Timeslots).");
}
return assignedTimeslot;
}
return null;
}
private Room extractRoom(Map roomMap, Talk talk) {
String roomName = nextStringCell().getStringCellValue();
if (!roomName.isEmpty()) {
Room room = roomMap.get(roomName);
if (room == null) {
throw new IllegalStateException(currentPosition() + ": The talk with code (" + talk.getCode()
+ ") has a roomName (" + roomName
+ ") that doesn't exist in the other sheet (Rooms).");
}
return room;
}
return null;
}
private int getNextStrictlyPositiveIntegerCell(String classSpecifier, String columnName) {
double cellValueDouble = nextNumericCell().getNumericCellValue();
if (strict && (cellValueDouble <= 0 || cellValueDouble != Math.floor(cellValueDouble))) {
throw new IllegalStateException(currentPosition() + ": The" + classSpecifier
+ ")'s has " + columnName + " (" + cellValueDouble
+ ") that isn't a strictly positive integer number.");
}
return (int) cellValueDouble;
}
private int getNextPositiveIntegerCell(String classSpecifier, String columnName) {
double cellValueDouble = nextNumericCell().getNumericCellValue();
if (strict && (cellValueDouble < 0 || cellValueDouble != Math.floor(cellValueDouble))) {
throw new IllegalStateException(currentPosition() + ": The " + classSpecifier
+ ")'s has " + columnName + " (" + cellValueDouble + ") that isn't a positive integer number.");
}
return (int) cellValueDouble;
}
private void verifyTimeslotTags(Set timeslotTagSet) {
for (String tag : timeslotTagSet) {
if (!totalTimeslotTagSet.contains(tag)) {
throw new IllegalStateException(currentPosition() + ": The timeslot tag (" + tag
+ ") does not exist in the tags (" + totalTimeslotTagSet
+ ") of the other sheet (Timeslots).");
}
}
}
private void verifyRoomTags(Set roomTagSet) {
for (String tag : roomTagSet) {
if (!totalRoomTagSet.contains(tag)) {
throw new IllegalStateException(currentPosition() + ": The room tag (" + tag
+ ") does not exist in the tags (" + totalRoomTagSet + ") of the other sheet (Rooms).");
}
}
}
private void setPrerequisiteTalkSets(Map> talkToPrerequisiteTalkSetMap) {
for (Map.Entry> entry : talkToPrerequisiteTalkSetMap.entrySet()) {
Talk currentTalk = entry.getKey();
Set prerequisiteTalkSet = new HashSet<>();
for (String prerequisiteTalkCode : entry.getValue()) {
Talk prerequisiteTalk = totalTalkCodeMap.get(prerequisiteTalkCode);
if (prerequisiteTalk == null) {
throw new IllegalStateException("The talk (" + currentTalk.getCode()
+ ") has a prerequisite talk (" + prerequisiteTalkCode
+ ") that doesn't exist in the talk list.");
}
prerequisiteTalkSet.add(prerequisiteTalk);
}
currentTalk.setPrerequisiteTalkSet(prerequisiteTalkSet);
}
}
private void readTimeslotDaysHeaders() {
LocalDate previousTimeslotDay = null;
for (Timeslot timeslot : solution.getTimeslotList()) {
LocalDate timeslotDay = timeslot.getDate();
if (timeslotDay.equals(previousTimeslotDay)) {
readHeaderCell("");
} else {
readHeaderCell(DAY_FORMATTER.format(timeslotDay));
previousTimeslotDay = timeslotDay;
}
}
}
private void readTimeslotHoursHeaders() {
for (Timeslot timeslot : solution.getTimeslotList()) {
readHeaderCell(TIME_FORMATTER.format(timeslot.getStartDateTime())
+ "-" + TIME_FORMATTER.format(timeslot.getEndDateTime()));
}
}
}
@Override
public void write(ConferenceSolution solution, File outputSolutionFile) {
try (FileOutputStream out = new FileOutputStream(outputSolutionFile)) {
Workbook workbook = new ConferenceSchedulingXlsxWriter(solution).write();
workbook.write(out);
} catch (IOException | RuntimeException e) {
throw new IllegalStateException("Failed writing outputSolutionFile ("
+ outputSolutionFile + ") for solution (" + solution + ").", e);
}
}
private class ConferenceSchedulingXlsxWriter extends AbstractXlsxWriter {
private Map themeTrackToStyleMap;
public ConferenceSchedulingXlsxWriter(ConferenceSolution solution) {
super(solution, ConferenceSchedulingApp.SOLVER_CONFIG);
}
@Override
public Workbook write() {
writeSetup();
initializeThemeTrackToStyleMap();
writeConfiguration();
writeTimeslotList();
writeRoomList();
writeSpeakerList();
writeTalkList();
writeInfeasibleView();
writeRoomsView();
writeSpeakersView();
writeThemeTracksView();
writeSectorsView();
writeAudienceTypesView();
writeAudienceLevelsView();
writeContentsView();
writeLanguagesView();
writeScoreView(justificationList -> justificationList.stream()
.filter(o -> o instanceof Talk).map(o -> ((Talk) o).getCode())
.collect(joining(", ")));
writeDaysSheets();
return workbook;
}
private void initializeThemeTrackToStyleMap() {
this.themeTrackToStyleMap = new HashMap<>();
TangoColorFactory tangoColorFactory = new TangoColorFactory();
List themeTrackList = solution.getTalkList().stream()
.flatMap(talk -> talk.getThemeTrackTagSet().stream())
.distinct().collect(toList());
for (String themeTrack : themeTrackList) {
XSSFCellStyle style = createStyle(new XSSFColor(tangoColorFactory.pickColor(themeTrack)));
themeTrackToStyleMap.put(themeTrack, style);
}
}
private void writeConfiguration() {
nextSheet("Configuration", 1, 4, false);
nextRow();
nextHeaderCell("Conference name");
nextCell().setCellValue(solution.getConferenceName());
ConferenceConstraintConfiguration constraintConfiguration = solution.getConstraintConfiguration();
writeIntConstraintParameterLine("Minimum consecutive talks pause in minutes",
constraintConfiguration::getMinimumConsecutiveTalksPauseInMinutes,
"The amount of time a speaker needs between 2 talks");
nextRow();
writeScoreConstraintHeaders();
writeScoreConstraintLine(ROOM_UNAVAILABLE_TIMESLOT, constraintConfiguration.getRoomUnavailableTimeslot(),
ROOM_UNAVAILABLE_TIMESLOT_DESCRIPTION);
writeScoreConstraintLine(ROOM_CONFLICT, constraintConfiguration.getRoomConflict(),
ROOM_CONFLICT_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_UNAVAILABLE_TIMESLOT, constraintConfiguration.getSpeakerUnavailableTimeslot(),
SPEAKER_UNAVAILABLE_TIMESLOT_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_CONFLICT, constraintConfiguration.getSpeakerConflict(),
SPEAKER_CONFLICT_DESCRIPTION);
writeScoreConstraintLine(TALK_PREREQUISITE_TALKS, constraintConfiguration.getTalkPrerequisiteTalks(),
TALK_PREREQUISITE_TALKS_DESCRIPTION);
writeScoreConstraintLine(TALK_MUTUALLY_EXCLUSIVE_TALKS_TAGS,
constraintConfiguration.getTalkMutuallyExclusiveTalksTags(),
TALK_MUTUALLY_EXCLUSIVE_TALKS_TAGS_DESCRIPTION);
writeScoreConstraintLine(CONSECUTIVE_TALKS_PAUSE, constraintConfiguration.getConsecutiveTalksPause(),
CONSECUTIVE_TALKS_PAUSE_DESCRIPTION);
writeScoreConstraintLine(CROWD_CONTROL, constraintConfiguration.getCrowdControl(),
CROWD_CONTROL_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_REQUIRED_TIMESLOT_TAGS, constraintConfiguration.getSpeakerRequiredTimeslotTags(),
SPEAKER_REQUIRED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_PROHIBITED_TIMESLOT_TAGS,
constraintConfiguration.getSpeakerProhibitedTimeslotTags(),
SPEAKER_PROHIBITED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_REQUIRED_TIMESLOT_TAGS, constraintConfiguration.getTalkRequiredTimeslotTags(),
TALK_REQUIRED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_PROHIBITED_TIMESLOT_TAGS, constraintConfiguration.getTalkProhibitedTimeslotTags(),
TALK_PROHIBITED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_REQUIRED_ROOM_TAGS, constraintConfiguration.getSpeakerRequiredRoomTags(),
SPEAKER_REQUIRED_ROOM_TAGS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_PROHIBITED_ROOM_TAGS, constraintConfiguration.getSpeakerProhibitedRoomTags(),
SPEAKER_PROHIBITED_ROOM_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_REQUIRED_ROOM_TAGS, constraintConfiguration.getTalkRequiredRoomTags(),
TALK_REQUIRED_ROOM_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_PROHIBITED_ROOM_TAGS, constraintConfiguration.getTalkProhibitedRoomTags(),
TALK_PROHIBITED_ROOM_TAGS_DESCRIPTION);
nextRow();
writeScoreConstraintLine(PUBLISHED_TIMESLOT, constraintConfiguration.getPublishedTimeslot(),
PUBLISHED_TIMESLOT_DESCRIPTION);
nextRow();
writeScoreConstraintLine(PUBLISHED_ROOM, constraintConfiguration.getPublishedRoom(),
PUBLISHED_ROOM_DESCRIPTION);
writeScoreConstraintLine(THEME_TRACK_CONFLICT, constraintConfiguration.getThemeTrackConflict(),
THEME_TRACK_CONFLICT_DESCRIPTION);
writeScoreConstraintLine(THEME_TRACK_ROOM_STABILITY, constraintConfiguration.getThemeTrackRoomStability(),
THEME_TRACK_ROOM_STABILITY_DESCRIPTION);
writeScoreConstraintLine(SECTOR_CONFLICT, constraintConfiguration.getSectorConflict(),
SECTOR_CONFLICT_DESCRIPTION);
writeScoreConstraintLine(AUDIENCE_TYPE_DIVERSITY, constraintConfiguration.getAudienceTypeDiversity(),
AUDIENCE_TYPE_DIVERSITY_DESCRIPTION);
writeScoreConstraintLine(AUDIENCE_TYPE_THEME_TRACK_CONFLICT,
constraintConfiguration.getAudienceTypeThemeTrackConflict(),
AUDIENCE_TYPE_THEME_TRACK_CONFLICT_DESCRIPTION);
writeScoreConstraintLine(AUDIENCE_LEVEL_DIVERSITY, constraintConfiguration.getAudienceLevelDiversity(),
AUDIENCE_LEVEL_DIVERSITY_DESCRIPTION);
writeScoreConstraintLine(CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION,
constraintConfiguration.getContentAudienceLevelFlowViolation(),
CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION_DESCRIPTION);
writeScoreConstraintLine(CONTENT_CONFLICT, constraintConfiguration.getContentConflict(),
CONTENT_CONFLICT_DESCRIPTION);
writeScoreConstraintLine(LANGUAGE_DIVERSITY, constraintConfiguration.getLanguageDiversity(),
LANGUAGE_DIVERSITY_DESCRIPTION);
writeScoreConstraintLine(SAME_DAY_TALKS, constraintConfiguration.getSameDayTalks(),
SAME_DAY_TALKS_DESCRIPTION);
writeScoreConstraintLine(POPULAR_TALKS, constraintConfiguration.getPopularTalks(),
POPULAR_TALKS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_PREFERRED_TIMESLOT_TAGS, constraintConfiguration.getSpeakerPreferredTimeslotTags(),
SPEAKER_PREFERRED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_UNDESIRED_TIMESLOT_TAGS, constraintConfiguration.getSpeakerUndesiredTimeslotTags(),
SPEAKER_UNDESIRED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_PREFERRED_TIMESLOT_TAGS, constraintConfiguration.getTalkPreferredTimeslotTags(),
TALK_PREFERRED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_UNDESIRED_TIMESLOT_TAGS, constraintConfiguration.getTalkUndesiredTimeslotTags(),
TALK_UNDESIRED_TIMESLOT_TAGS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_PREFERRED_ROOM_TAGS, constraintConfiguration.getSpeakerPreferredRoomTags(),
SPEAKER_PREFERRED_ROOM_TAGS_DESCRIPTION);
writeScoreConstraintLine(SPEAKER_UNDESIRED_ROOM_TAGS, constraintConfiguration.getSpeakerUndesiredRoomTags(),
SPEAKER_UNDESIRED_ROOM_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_PREFERRED_ROOM_TAGS, constraintConfiguration.getTalkPreferredRoomTags(),
TALK_PREFERRED_ROOM_TAGS_DESCRIPTION);
writeScoreConstraintLine(TALK_UNDESIRED_ROOM_TAGS, constraintConfiguration.getTalkUndesiredRoomTags(),
TALK_UNDESIRED_ROOM_TAGS_DESCRIPTION);
autoSizeColumnsWithHeader();
}
private void writeTimeslotList() {
nextSheet("Timeslots", 3, 1, false);
nextRow();
nextHeaderCell("Day");
nextHeaderCell("Start");
nextHeaderCell("End");
nextHeaderCell("Talk types");
nextHeaderCell("Tags");
for (Timeslot timeslot : solution.getTimeslotList()) {
nextRow();
nextCell().setCellValue(DAY_FORMATTER.format(timeslot.getDate()));
nextCell().setCellValue(TIME_FORMATTER.format(timeslot.getStartDateTime()));
nextCell().setCellValue(TIME_FORMATTER.format(timeslot.getEndDateTime()));
nextCell().setCellValue(
String.join(", ", timeslot.getTalkTypeSet().stream().map(TalkType::getName).collect(toList())));
nextCell().setCellValue(String.join(", ", timeslot.getTagSet()));
}
autoSizeColumnsWithHeader();
}
private void writeRoomList() {
nextSheet("Rooms", 1, 2, false);
nextRow();
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Name");
nextHeaderCell("Capacity");
nextHeaderCell("Talk types");
nextHeaderCell("Tags");
writeTimeslotHoursHeaders();
for (Room room : solution.getRoomList()) {
nextRow();
nextCell().setCellValue(room.getName());
nextCell().setCellValue(room.getCapacity());
nextCell().setCellValue(
String.join(", ", room.getTalkTypeSet().stream().map(TalkType::getName).collect(toList())));
nextCell().setCellValue(String.join(", ", room.getTagSet()));
for (Timeslot timeslot : solution.getTimeslotList()) {
nextCell(room.getUnavailableTimeslotSet().contains(timeslot) ? unavailableStyle : defaultStyle)
.setCellValue("");
}
}
autoSizeColumnsWithHeader();
}
private void writeSpeakerList() {
nextSheet("Speakers", 1, 2, false);
nextRow();
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Name");
nextHeaderCell("Required timeslot tags");
nextHeaderCell("Preferred timeslot tags");
nextHeaderCell("Prohibited timeslot tags");
nextHeaderCell("Undesired timeslot tags");
nextHeaderCell("Required room tags");
nextHeaderCell("Preferred room tags");
nextHeaderCell("Prohibited room tags");
nextHeaderCell("Undesired room tags");
writeTimeslotHoursHeaders();
for (Speaker speaker : solution.getSpeakerList()) {
nextRow();
nextCell().setCellValue(speaker.getName());
nextCell().setCellValue(String.join(", ", speaker.getRequiredTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getPreferredTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getProhibitedTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getUndesiredTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getRequiredRoomTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getPreferredRoomTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getProhibitedRoomTagSet()));
nextCell().setCellValue(String.join(", ", speaker.getUndesiredRoomTagSet()));
for (Timeslot timeslot : solution.getTimeslotList()) {
nextCell(speaker.getUnavailableTimeslotSet().contains(timeslot) ? unavailableStyle : defaultStyle)
.setCellValue("");
}
}
autoSizeColumnsWithHeader();
}
private void writeTalkList() {
nextSheet("Talks", 2, 1, false);
nextRow();
nextHeaderCell("Code");
nextHeaderCell("Title");
nextHeaderCell("Talk type");
nextHeaderCell("Speakers");
nextHeaderCell("Theme track tags");
nextHeaderCell("Sector tags");
nextHeaderCell("Audience types");
nextHeaderCell("Audience level");
nextHeaderCell("Content tags");
nextHeaderCell("Language");
nextHeaderCell("Required timeslot tags");
nextHeaderCell("Preferred timeslot tags");
nextHeaderCell("Prohibited timeslot tags");
nextHeaderCell("Undesired timeslot tags");
nextHeaderCell("Required room tags");
nextHeaderCell("Preferred room tags");
nextHeaderCell("Prohibited room tags");
nextHeaderCell("Undesired room tags");
nextHeaderCell("Mutually exclusive talks tags");
nextHeaderCell("Prerequisite talks codes");
nextHeaderCell("Favorite count");
nextHeaderCell("Crowd control risk");
nextHeaderCell("Pinned by user");
nextHeaderCell("Timeslot day");
nextHeaderCell("Start");
nextHeaderCell("End");
nextHeaderCell("Room");
nextHeaderCell("Published Timeslot");
nextHeaderCell("Published Start");
nextHeaderCell("Published End");
nextHeaderCell("Published Room");
for (Talk talk : solution.getTalkList()) {
nextRow();
nextCell().setCellValue(talk.getCode());
nextCell().setCellValue(talk.getTitle());
nextCell().setCellValue(talk.getTalkType().getName());
nextCell().setCellValue(talk.getSpeakerList()
.stream().map(Speaker::getName).collect(joining(", ")));
nextCell().setCellValue(String.join(", ", talk.getThemeTrackTagSet()));
nextCell().setCellValue(String.join(", ", talk.getSectorTagSet()));
nextCell().setCellValue(String.join(", ", talk.getAudienceTypeSet()));
nextCell().setCellValue(talk.getAudienceLevel());
nextCell().setCellValue(String.join(", ", talk.getContentTagSet()));
nextCell().setCellValue(talk.getLanguage());
nextCell().setCellValue(String.join(", ", talk.getRequiredTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", talk.getPreferredTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", talk.getProhibitedTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", talk.getUndesiredTimeslotTagSet()));
nextCell().setCellValue(String.join(", ", talk.getRequiredRoomTagSet()));
nextCell().setCellValue(String.join(", ", talk.getPreferredRoomTagSet()));
nextCell().setCellValue(String.join(", ", talk.getProhibitedRoomTagSet()));
nextCell().setCellValue(String.join(", ", talk.getUndesiredRoomTagSet()));
nextCell().setCellValue(String.join(", ", talk.getMutuallyExclusiveTalksTagSet()));
nextCell().setCellValue(
String.join(", ", talk.getPrerequisiteTalkSet().stream().map(Talk::getCode).collect(toList())));
nextCell().setCellValue(talk.getFavoriteCount());
nextCell().setCellValue(talk.getCrowdControlRisk());
nextCell(talk.isPinnedByUser() ? pinnedStyle : defaultStyle).setCellValue(talk.isPinnedByUser());
XSSFCellStyle timeslotStyle;
XSSFCellStyle roomStyle;
if (talk.isPinnedByUser()) {
timeslotStyle = (talk.getTimeslot() == null) ? hardPenaltyStyle : defaultStyle;
roomStyle = (talk.getRoom() == null) ? hardPenaltyStyle : defaultStyle;
} else {
timeslotStyle = (talk.getPublishedTimeslot() == null || talk.getTimeslot() == talk.getPublishedTimeslot())
? planningVariableStyle
: republishedStyle;
roomStyle = (talk.getPublishedRoom() == null || talk.getRoom() == talk.getPublishedRoom())
? planningVariableStyle
: republishedStyle;
}
nextCell(timeslotStyle)
.setCellValue(talk.getTimeslot() == null ? "" : DAY_FORMATTER.format(talk.getTimeslot().getDate()));
nextCell(timeslotStyle).setCellValue(
talk.getTimeslot() == null ? "" : TIME_FORMATTER.format(talk.getTimeslot().getStartDateTime()));
nextCell(timeslotStyle).setCellValue(
talk.getTimeslot() == null ? "" : TIME_FORMATTER.format(talk.getTimeslot().getEndDateTime()));
nextCell(roomStyle).setCellValue(talk.getRoom() == null ? "" : talk.getRoom().getName());
nextCell().setCellValue(
talk.getPublishedTimeslot() == null ? "" : DAY_FORMATTER.format(talk.getPublishedTimeslot().getDate()));
nextCell().setCellValue(talk.getPublishedTimeslot() == null ? ""
: TIME_FORMATTER.format(talk.getPublishedTimeslot().getStartDateTime()));
nextCell().setCellValue(talk.getPublishedTimeslot() == null ? ""
: TIME_FORMATTER.format(talk.getPublishedTimeslot().getEndDateTime()));
nextCell().setCellValue(talk.getPublishedRoom() == null ? "" : talk.getPublishedRoom().getName());
}
autoSizeColumnsWithHeader();
}
private void writeInfeasibleView() {
if (solution.getScore() == null || solution.getScore().isFeasible()) {
return;
}
nextSheet("Infeasible view", 1, 1, true);
nextRow();
nextHeaderCell("Score");
nextCell().setCellValue(solution.getScore() == null ? "Not yet solved" : solution.getScore().toShortString());
nextRow();
nextRow();
nextHeaderCell("Talk type");
nextHeaderCell("Count");
nextHeaderCell("Usable timeslots");
nextHeaderCell("Usable rooms");
nextHeaderCell("Usable sessions");
Map talkTypeToCountMap = solution.getTalkList().stream()
.collect(groupingBy(Talk::getTalkType, LinkedHashMap::new, counting()));
for (Map.Entry entry : talkTypeToCountMap.entrySet()) {
TalkType talkType = entry.getKey();
long count = entry.getValue();
nextRow();
nextHeaderCell(talkType.getName());
nextCell().setCellValue(count);
int timeslotListSize = talkType.getCompatibleTimeslotSet().size();
nextCell().setCellValue(timeslotListSize);
int roomListSize = talkType.getCompatibleRoomSet().size();
nextCell().setCellValue(roomListSize);
int sessionCount = timeslotListSize * roomListSize;
nextCell(sessionCount < count ? hardPenaltyStyle : defaultStyle).setCellValue(sessionCount);
}
nextRow();
nextRow();
nextHeaderCell("Total");
int talkListSize = solution.getTalkList().size();
nextCell().setCellValue(talkListSize);
int timeslotListSize = solution.getTimeslotList().size();
nextCell().setCellValue(timeslotListSize);
int roomListSize = solution.getRoomList().size();
nextCell().setCellValue(roomListSize);
int sessionCount = 0;
for (Timeslot timeslot : solution.getTimeslotList()) {
for (Room room : solution.getRoomList()) {
if (!disjoint(timeslot.getTalkTypeSet(), room.getTalkTypeSet())
&& !room.getUnavailableTimeslotSet().contains(timeslot)) {
sessionCount++;
}
}
}
nextCell(sessionCount < talkListSize ? hardPenaltyStyle : defaultStyle).setCellValue(sessionCount);
autoSizeColumnsWithHeader();
}
private void writeRoomsView() {
nextSheet("Rooms view", 1, 2, true);
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Room");
writeTimeslotHoursHeaders();
for (Room room : solution.getRoomList()) {
nextRow();
currentRow.setHeightInPoints(3 * currentSheet.getDefaultRowHeightInPoints());
nextCell().setCellValue(room.getName());
List roomTalkList = solution.getTalkList().stream()
.filter(talk -> talk.getRoom() == room)
.collect(toList());
Timeslot mergePreviousTimeslot = null;
int mergeStart = -1;
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = roomTalkList.stream()
.filter(talk -> talk.getTimeslot() == timeslot).collect(toList());
if (talkList.isEmpty() && mergePreviousTimeslot != null
&& timeslot.getStartDateTime().compareTo(mergePreviousTimeslot.getEndDateTime()) < 0) {
nextCell();
} else {
if (mergePreviousTimeslot != null && mergeStart < currentColumnNumber) {
currentSheet.addMergedRegion(
new CellRangeAddress(currentRowNumber, currentRowNumber, mergeStart, currentColumnNumber));
}
boolean unavailable = room.getUnavailableTimeslotSet().contains(timeslot)
|| disjoint(room.getTalkTypeSet(), timeslot.getTalkTypeSet());
nextTalkListCell(unavailable, talkList, talk -> talk.getCode() + ": " + talk.getTitle() + "\n "
+ talk.getSpeakerList().stream().map(Speaker::getName).collect(joining(", ")));
mergePreviousTimeslot = talkList.isEmpty() ? null : timeslot;
mergeStart = currentColumnNumber;
}
}
if (mergePreviousTimeslot != null && mergeStart < currentColumnNumber) {
currentSheet.addMergedRegion(
new CellRangeAddress(currentRowNumber, currentRowNumber, mergeStart, currentColumnNumber));
}
}
currentSheet.autoSizeColumn(0);
for (int i = 1; i < headerCellCount; i++) {
currentSheet.setColumnWidth(i, 20 * 256);
}
}
private void writeSpeakersView() {
nextSheet("Speakers view", 1, 2, true);
String[] filteredConstraintNames = {
SPEAKER_UNAVAILABLE_TIMESLOT, SPEAKER_CONFLICT,
SPEAKER_REQUIRED_TIMESLOT_TAGS, SPEAKER_PROHIBITED_TIMESLOT_TAGS,
SPEAKER_PREFERRED_TIMESLOT_TAGS, SPEAKER_UNDESIRED_TIMESLOT_TAGS,
SPEAKER_REQUIRED_ROOM_TAGS, SPEAKER_PROHIBITED_ROOM_TAGS,
SPEAKER_PREFERRED_ROOM_TAGS, SPEAKER_UNDESIRED_ROOM_TAGS };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Speaker");
writeTimeslotHoursHeaders();
for (Speaker speaker : solution.getSpeakerList()) {
nextRow();
nextCell().setCellValue(speaker.getName());
List timeslotTalkList = solution.getTalkList().stream()
.filter(talk -> talk.getSpeakerList().contains(speaker))
.collect(toList());
Timeslot mergePreviousTimeslot = null;
int mergeStart = -1;
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotTalkList.stream()
.filter(talk -> talk.getTimeslot() == timeslot).collect(toList());
if (talkList.isEmpty() && mergePreviousTimeslot != null
&& timeslot.getStartDateTime().compareTo(mergePreviousTimeslot.getEndDateTime()) < 0) {
nextCell();
} else {
if (mergePreviousTimeslot != null && mergeStart < currentColumnNumber) {
currentSheet.addMergedRegion(
new CellRangeAddress(currentRowNumber, currentRowNumber, mergeStart, currentColumnNumber));
}
boolean unavailable = speaker.getUnavailableTimeslotSet().contains(timeslot);
nextTalkListCell(unavailable, talkList, filteredConstraintNames);
mergePreviousTimeslot = talkList.isEmpty() ? null : timeslot;
mergeStart = currentColumnNumber;
}
}
if (mergePreviousTimeslot != null && mergeStart < currentColumnNumber) {
currentSheet.addMergedRegion(
new CellRangeAddress(currentRowNumber, currentRowNumber, mergeStart, currentColumnNumber));
}
}
currentSheet.autoSizeColumn(0);
for (int i = 1; i < headerCellCount; i++) {
currentSheet.setColumnWidth(i, 20 * 256);
}
}
private void writeThemeTracksView() {
nextSheet("Theme tracks view", 1, 2, true);
String[] filteredConstraintNames = { THEME_TRACK_CONFLICT, AUDIENCE_TYPE_THEME_TRACK_CONFLICT, SAME_DAY_TALKS };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Theme track tag");
writeTimeslotHoursHeaders();
Map>> tagToTimeslotToTalkListMap = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null)
.flatMap(talk -> talk.getThemeTrackTagSet().stream()
.map(tag -> Pair.of(tag, Pair.of(talk.getTimeslot(), talk))))
.collect(groupingBy(Pair::getLeft,
groupingBy(o -> o.getRight().getLeft(), mapping(o -> o.getRight().getRight(), toList()))));
for (Map.Entry>> entry : tagToTimeslotToTalkListMap.entrySet()) {
nextRow();
nextHeaderCell(entry.getKey());
Map> timeslotToTalkListMap = entry.getValue();
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotToTalkListMap.get(timeslot);
nextTalkListCell(talkList, filteredConstraintNames);
}
}
autoSizeColumnsWithHeader();
}
private void writeSectorsView() {
nextSheet("Sectors view", 1, 2, true);
String[] filteredConstraintNames = { SECTOR_CONFLICT };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Sector tag");
writeTimeslotHoursHeaders();
Map>> tagToTimeslotToTalkListMap = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null)
.flatMap(talk -> talk.getSectorTagSet().stream()
.map(tag -> Pair.of(tag, Pair.of(talk.getTimeslot(), talk))))
.collect(groupingBy(Pair::getLeft,
groupingBy(o -> o.getRight().getLeft(), mapping(o -> o.getRight().getRight(), toList()))));
for (Map.Entry>> entry : tagToTimeslotToTalkListMap.entrySet()) {
nextRow();
nextHeaderCell(entry.getKey());
Map> timeslotToTalkListMap = entry.getValue();
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotToTalkListMap.get(timeslot);
nextTalkListCell(talkList, filteredConstraintNames);
}
}
autoSizeColumnsWithHeader();
}
private void writeAudienceTypesView() {
nextSheet("Audience types view", 1, 2, true);
String[] filteredConstraintNames = { AUDIENCE_TYPE_DIVERSITY, AUDIENCE_TYPE_THEME_TRACK_CONFLICT };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Audience type");
writeTimeslotHoursHeaders();
Map>> audienceTypeToTimeslotToTalkListMap = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null)
.flatMap(talk -> talk.getAudienceTypeSet().stream()
.map(audienceType -> Pair.of(audienceType, Pair.of(talk.getTimeslot(), talk))))
.collect(groupingBy(Pair::getLeft,
groupingBy(o -> o.getRight().getLeft(), mapping(o -> o.getRight().getRight(), toList()))));
for (Map.Entry>> entry : audienceTypeToTimeslotToTalkListMap.entrySet()) {
nextRow();
nextHeaderCell(entry.getKey());
Map> timeslotToTalkListMap = entry.getValue();
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotToTalkListMap.get(timeslot);
nextTalkListCell(talkList, filteredConstraintNames);
}
}
autoSizeColumnsWithHeader();
}
private void writeAudienceLevelsView() {
nextSheet("Audience levels view", 1, 2, true);
String[] filteredConstraintNames = { AUDIENCE_LEVEL_DIVERSITY, CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Audience level");
writeTimeslotHoursHeaders();
Map>> levelToTimeslotToTalkListMap = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null)
.map(talk -> Pair.of(talk.getAudienceLevel(), Pair.of(talk.getTimeslot(), talk)))
.collect(groupingBy(Pair::getLeft,
groupingBy(o -> o.getRight().getLeft(), mapping(o -> o.getRight().getRight(), toList()))));
for (Map.Entry>> entry : levelToTimeslotToTalkListMap.entrySet()) {
nextRow();
nextHeaderCell(Integer.toString(entry.getKey()));
Map> timeslotToTalkListMap = entry.getValue();
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotToTalkListMap.get(timeslot);
nextTalkListCell(talkList, filteredConstraintNames);
}
}
autoSizeColumnsWithHeader();
}
private void writeContentsView() {
nextSheet("Contents view", 1, 2, true);
String[] filteredConstraintNames = { CONTENT_AUDIENCE_LEVEL_FLOW_VIOLATION, CONTENT_CONFLICT };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Content tag");
writeTimeslotHoursHeaders();
Map>> tagToTimeslotToTalkListMap = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null)
.flatMap(talk -> talk.getContentTagSet().stream()
.map(tag -> Pair.of(tag, Pair.of(talk.getTimeslot(), talk))))
.collect(groupingBy(Pair::getLeft,
groupingBy(o -> o.getRight().getLeft(), mapping(o -> o.getRight().getRight(), toList()))));
for (Map.Entry>> entry : tagToTimeslotToTalkListMap.entrySet()) {
nextRow();
nextHeaderCell(entry.getKey());
Map> timeslotToTalkListMap = entry.getValue();
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotToTalkListMap.get(timeslot);
nextTalkListCell(talkList,
talk -> talk.getCode() + " (level " + talk.getAudienceLevel() + ")",
filteredConstraintNames,
justificationList -> justificationList.stream()
.allMatch(justification -> !(justification instanceof Talk)
|| ((Talk) justification).getContentTagSet().contains(entry.getKey())));
}
}
autoSizeColumnsWithHeader();
}
private void writeLanguagesView() {
nextSheet("Languages view", 1, 2, true);
String[] filteredConstraintNames = { LANGUAGE_DIVERSITY };
nextRow();
nextHeaderCell("");
writeTimeslotDaysHeaders();
nextRow();
nextHeaderCell("Language");
writeTimeslotHoursHeaders();
Map>> languageToTimeslotToTalkListMap = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null)
.map(talk -> Pair.of(talk.getLanguage(), Pair.of(talk.getTimeslot(), talk)))
.collect(groupingBy(Pair::getLeft,
groupingBy(o -> o.getRight().getLeft(), mapping(o -> o.getRight().getRight(), toList()))));
for (Map.Entry>> entry : languageToTimeslotToTalkListMap.entrySet()) {
nextRow();
nextHeaderCell(entry.getKey());
Map> timeslotToTalkListMap = entry.getValue();
for (Timeslot timeslot : solution.getTimeslotList()) {
List talkList = timeslotToTalkListMap.get(timeslot);
nextTalkListCell(talkList, filteredConstraintNames);
}
}
autoSizeColumnsWithHeader();
}
private void writeDaysSheets() {
List dayList = solution.getTimeslotList().stream().map(Timeslot::getDate).distinct().collect(toList());
for (LocalDate day : dayList) {
List dayTimeslotList = solution.getTimeslotList().stream()
.filter(timeslot -> timeslot.getDate().equals(day)).collect(toList());
List dayTalkList = solution.getTalkList().stream()
.filter(talk -> talk.getTimeslot() != null && talk.getTimeslot().getDate().equals(day))
.collect(toList());
writeDaySheet(day, dayTimeslotList, dayTalkList);
}
}
private void writeDaySheet(LocalDate day, List timeslotList, List talkList) {
nextSheet(DAY_FORMATTER.format(day), 1, 1, true);
nextRow();
nextHeaderCell(DAY_FORMATTER.format(day));
writeTimeslotHoursVertically(timeslotList);
List dayRoomList = talkList.stream().map(Talk::getRoom)
.filter(Objects::nonNull)
.distinct()
.sorted(comparing(Room::getName))
.collect(toList());
for (Room room : dayRoomList) {
currentColumnNumber++;
currentRowNumber = -1;
nextCellVertically().setCellValue(room.getName());
List roomTalkList = talkList.stream()
.filter(talk -> talk.getRoom() == room)
.collect(toList());
writeRoomTalks(timeslotList, room, roomTalkList);
}
currentSheet.autoSizeColumn(0);
for (int i = 1; i < currentSheet.getRow(0).getPhysicalNumberOfCells(); i++) {
currentSheet.setColumnWidth(i, 15 * 256);
}
}
private void writeRoomTalks(List dayTimeslotList, Room room, List roomTalkList) {
Timeslot mergePreviousTimeslot = null;
int mergeStart = -1;
for (Timeslot timeslot : dayTimeslotList) {
List talkList = roomTalkList.stream()
.filter(talk -> talk.getTimeslot() == timeslot).collect(toList());
if (talkList.isEmpty() && mergePreviousTimeslot != null
&& timeslot.getStartDateTime().compareTo(mergePreviousTimeslot.getEndDateTime()) < 0) {
nextCellVertically();
} else {
if (mergePreviousTimeslot != null && mergeStart < currentRowNumber) {
currentSheet.addMergedRegion(
new CellRangeAddress(mergeStart, currentRowNumber, currentColumnNumber, currentColumnNumber));
}
boolean unavailable = room.getUnavailableTimeslotSet().contains(timeslot)
|| disjoint(room.getTalkTypeSet(), timeslot.getTalkTypeSet());
nextTalkListCell(unavailable, talkList, talk -> StringUtils.abbreviate(talk.getTitle(), 50) + "\n"
+ StringUtils.abbreviate(
talk.getSpeakerList().stream().map(Speaker::getName).collect(joining(", ")), 30),
true);
mergePreviousTimeslot = talkList.isEmpty() ? null : timeslot;
mergeStart = currentRowNumber;
}
}
if (mergePreviousTimeslot != null && mergeStart < currentRowNumber) {
currentSheet.addMergedRegion(
new CellRangeAddress(mergeStart, currentRowNumber, currentColumnNumber, currentColumnNumber));
}
}
private void writeTimeslotDaysHeaders() {
LocalDate previousTimeslotDay = null;
int mergeStart = -1;
for (Timeslot timeslot : solution.getTimeslotList()) {
LocalDate timeslotDay = timeslot.getDate();
if (timeslotDay.equals(previousTimeslotDay)) {
nextHeaderCell("");
} else {
if (previousTimeslotDay != null) {
currentSheet.addMergedRegion(
new CellRangeAddress(currentRowNumber, currentRowNumber, mergeStart, currentColumnNumber));
}
nextHeaderCell(DAY_FORMATTER.format(timeslotDay));
previousTimeslotDay = timeslotDay;
mergeStart = currentColumnNumber;
}
}
if (previousTimeslotDay != null) {
currentSheet.addMergedRegion(
new CellRangeAddress(currentRowNumber, currentRowNumber, mergeStart, currentColumnNumber));
}
}
private void writeTimeslotHoursHeaders() {
for (Timeslot timeslot : solution.getTimeslotList()) {
nextHeaderCell(TIME_FORMATTER.format(timeslot.getStartDateTime())
+ "-" + TIME_FORMATTER.format(timeslot.getEndDateTime()));
}
}
private void writeTimeslotHoursVertically(List dayTimeslotList) {
for (Timeslot timeslot : dayTimeslotList) {
nextRow();
nextCell().setCellValue(TIME_FORMATTER.format(timeslot.getStartDateTime())
+ "-" + TIME_FORMATTER.format(timeslot.getEndDateTime()));
currentRow.setHeightInPoints(3 * currentSheet.getDefaultRowHeightInPoints());
}
}
protected void nextTalkListCell(List talkList, String[] filteredConstraintNames) {
nextTalkListCell(false, talkList, filteredConstraintNames);
}
protected void nextTalkListCell(boolean unavailable, List talkList, String[] filteredConstraintNames) {
nextTalkListCell(unavailable, talkList,
talk -> talk.getCode() + " @ " + (talk.getRoom() == null ? "No room" : talk.getRoom().getName()),
filteredConstraintNames, false, null);
}
protected void nextTalkListCell(List talkList, Function stringFunction,
String[] filteredConstraintNames, Predicate> isValidJustificationList) {
nextTalkListCell(false, talkList, stringFunction, filteredConstraintNames, false, isValidJustificationList);
}
protected void nextTalkListCell(boolean unavailable, List talkList, Function stringFunction) {
nextTalkListCell(unavailable, talkList, stringFunction, null, false, null);
}
protected void nextTalkListCell(boolean unavailable, List talkList, Function stringFunction,
boolean isVerticalView) {
nextTalkListCell(unavailable, talkList, stringFunction, null, isVerticalView, null);
}
protected void nextTalkListCell(boolean unavailable, List talkList, Function stringFunction,
String[] filteredConstraintNames, boolean isPrintedView, Predicate> isValidJustificationList) {
List filteredConstraintNameList = (filteredConstraintNames == null) ? null
: Arrays.asList(filteredConstraintNames);
if (talkList == null) {
talkList = emptyList();
}
HardMediumSoftScore score = talkList.stream()
.map(indictmentMap::get)
.filter(Objects::nonNull)
.flatMap(indictment -> indictment.getConstraintMatchSet().stream())
// Filter out filtered constraints
.filter(constraintMatch -> filteredConstraintNameList == null
|| filteredConstraintNameList.contains(constraintMatch.getConstraintName()))
.filter(constraintMatch -> isValidJustificationList == null
|| isValidJustificationList.test(constraintMatch.getJustificationList()))
.map(constraintMatch -> (HardMediumSoftScore) constraintMatch.getScore())
// Filter out positive constraints
.filter(indictmentScore -> !(indictmentScore.getHardScore() >= 0 && indictmentScore.getMediumScore() >= 0
&& indictmentScore.getSoftScore() >= 0))
.reduce(Score::add).orElse(HardMediumSoftScore.ZERO);
XSSFCell cell;
if (isPrintedView) {
cell = nextCellVertically(talkList.isEmpty() || talkList.get(0).getThemeTrackTagSet().isEmpty() ? wrappedStyle
: themeTrackToStyleMap.get(talkList.get(0).getThemeTrackTagSet().iterator().next()));
} else if (talkList.stream().anyMatch(Talk::isPinnedByUser)) {
cell = nextCell(pinnedStyle);
} else if (!score.isFeasible()) {
cell = nextCell(hardPenaltyStyle);
} else if (unavailable) {
cell = nextCell(unavailableStyle);
} else if (score.getMediumScore() < 0) {
cell = nextCell(mediumPenaltyStyle);
} else if (score.getSoftScore() < 0) {
cell = nextCell(softPenaltyStyle);
} else {
cell = nextCell(wrappedStyle);
}
if (!talkList.isEmpty()) {
ClientAnchor anchor = creationHelper.createClientAnchor();
anchor.setCol1(cell.getColumnIndex());
anchor.setCol2(cell.getColumnIndex() + 4);
anchor.setRow1(currentRow.getRowNum());
anchor.setRow2(currentRow.getRowNum() + 4);
Comment comment = currentDrawing.createCellComment(anchor);
StringBuilder commentString = new StringBuilder(talkList.size() * 200);
for (Talk talk : talkList) {
commentString.append(talk.getCode()).append("-").append(String.join(", ", talk.getThemeTrackTagSet()))
.append(": ").append(talk.getTitle()).append("\n ")
.append(talk.getSpeakerList().stream().map(Speaker::getName).collect(joining(", ")))
.append(talk.isPinnedByUser() ? "\nPINNED BY USER" : "");
Indictment indictment = indictmentMap.get(talk);
if (indictment != null) {
commentString.append("\n").append(indictment.getScore().toShortString())
.append(" total");
Set constraintMatchSet = indictment.getConstraintMatchSet().stream()
.filter(constraintMatch -> filteredConstraintNameList == null
|| filteredConstraintNameList.contains(constraintMatch.getConstraintName()))
.collect(toSet());
List constraintNameList = constraintMatchSet.stream()
.map(ConstraintMatch::getConstraintName).distinct().collect(toList());
for (String constraintName : constraintNameList) {
List filteredConstraintMatchList = constraintMatchSet.stream()
.filter(constraintMatch -> constraintMatch.getConstraintName().equals(constraintName)
&& (isValidJustificationList == null
|| isValidJustificationList.test(constraintMatch.getJustificationList())))
.collect(toList());
HardMediumSoftScore sum = filteredConstraintMatchList.stream()
.map(constraintMatch -> (HardMediumSoftScore) constraintMatch.getScore())
.reduce(HardMediumSoftScore::add)
.orElse(HardMediumSoftScore.ZERO);
String justificationTalkCodes = filteredConstraintMatchList.stream()
.flatMap(constraintMatch -> constraintMatch.getJustificationList().stream())
.filter(justification -> justification instanceof Talk && justification != talk)
.distinct().map(o -> ((Talk) o).getCode()).collect(joining(", "));
commentString.append("\n ").append(sum.toShortString())
.append(" for ").append(filteredConstraintMatchList.size())
.append(" ").append(constraintName).append("s")
.append("\n ").append(justificationTalkCodes);
}
}
commentString.append("\n\n");
}
comment.setString(creationHelper.createRichTextString(commentString.toString()));
cell.setCellComment(comment);
}
cell.setCellValue(talkList.stream().map(stringFunction).collect(joining("\n")));
currentRow.setHeightInPoints(
Math.max(currentRow.getHeightInPoints(), talkList.size() * currentSheet.getDefaultRowHeightInPoints()));
}
}
}