use of org.cpsolver.studentsct.model.Enrollment in project cpsolver by UniTime.
the class StudentRequestXml method exportModel.
public static Document exportModel(Assignment<Request, Enrollment> assignment, StudentSectioningModel model) {
Document document = DocumentHelper.createDocument();
Element requestElement = document.addElement("request");
requestElement.addAttribute("campus", model.getProperties().getProperty("Data.Initiative"));
requestElement.addAttribute("year", model.getProperties().getProperty("Data.Year"));
requestElement.addAttribute("term", model.getProperties().getProperty("Data.Term"));
for (Student student : model.getStudents()) {
Element studentElement = requestElement.addElement("student");
studentElement.addAttribute("key", String.valueOf(student.getId()));
Element courseRequestsElement = studentElement.addElement("updateCourseRequests");
courseRequestsElement.addAttribute("commit", "true");
Collections.sort(student.getRequests(), new Comparator<Request>() {
@Override
public int compare(Request r1, Request r2) {
if (r1.isAlternative() != r2.isAlternative()) {
return (r1.isAlternative() ? 1 : -1);
}
return Double.compare(r1.getPriority(), r2.getPriority());
}
});
boolean hasSchedule = false;
for (Request request : student.getRequests()) {
if (assignment.getValue(request) != null)
hasSchedule = true;
if (request instanceof FreeTimeRequest) {
FreeTimeRequest ftReq = (FreeTimeRequest) request;
Element ftReqElement = courseRequestsElement.addElement("freeTime");
requestElement.addAttribute("days", ftReq.getTime().getDayHeader());
int startSlot = ftReq.getTime().getStartSlot();
int startTime = startSlot * Constants.SLOT_LENGTH_MIN + Constants.FIRST_SLOT_TIME_MIN;
ftReqElement.addAttribute("startTime", s2zDF.format(startTime / 60) + s2zDF.format(startTime % 60));
int endTime = startTime + ftReq.getTime().getLength() * Constants.SLOT_LENGTH_MIN - ftReq.getTime().getBreakTime();
ftReqElement.addAttribute("endTime", s2zDF.format(endTime / 60) + s2zDF.format(endTime % 60));
ftReqElement.addAttribute("length", String.valueOf(ftReq.getTime().getLength() * Constants.SLOT_LENGTH_MIN));
} else {
CourseRequest crReq = (CourseRequest) request;
Element crReqElement = courseRequestsElement.addElement("courseOffering");
Course course = crReq.getCourses().get(0);
crReqElement.addAttribute("subjectArea", course.getSubjectArea());
crReqElement.addAttribute("courseNumber", course.getCourseNumber());
crReqElement.addAttribute("waitlist", crReq.isWaitlist() ? "true" : "false");
crReqElement.addAttribute("alternative", crReq.isAlternative() ? "true" : "false");
for (int i = 1; i < crReq.getCourses().size(); i++) {
Course altCourse = crReq.getCourses().get(i);
Element altCourseElement = crReqElement.addElement("alternative");
altCourseElement.addAttribute("subjectArea", altCourse.getSubjectArea());
altCourseElement.addAttribute("courseNumber", altCourse.getCourseNumber());
}
}
}
if (hasSchedule) {
Element requestScheduleElement = studentElement.addElement("requestSchedule");
requestScheduleElement.addAttribute("type", "commit");
for (Request request : student.getRequests()) {
if (request instanceof CourseRequest) {
CourseRequest crReq = (CourseRequest) request;
Enrollment enrollment = assignment.getValue(crReq);
if (enrollment == null)
continue;
Element crReqElement = requestScheduleElement.addElement("courseOffering");
Course course = enrollment.getCourse();
crReqElement.addAttribute("subjectArea", course.getSubjectArea());
crReqElement.addAttribute("courseNumber", course.getCourseNumber());
for (Section section : enrollment.getSections()) {
Element classEl = crReqElement.addElement("class");
classEl.addAttribute("id", section.getSubpart().getInstructionalType());
classEl.addAttribute("assignmentId", String.valueOf(section.getId()));
}
}
}
}
}
return document;
}
use of org.cpsolver.studentsct.model.Enrollment in project cpsolver by UniTime.
the class StudentRequestXml method main.
public static void main(String[] args) {
try {
ToolBox.configureLogging();
StudentSectioningModel model = new StudentSectioningModel(new DataProperties());
Assignment<Request, Enrollment> assignment = new DefaultSingleAssignment<Request, Enrollment>();
StudentSectioningXMLLoader xmlLoad = new StudentSectioningXMLLoader(model, assignment);
xmlLoad.setInputFile(new File(args[0]));
xmlLoad.load();
Document document = exportModel(assignment, model);
FileOutputStream fos = new FileOutputStream(new File(args[1]));
(new XMLWriter(fos, OutputFormat.createPrettyPrint())).write(document);
fos.flush();
fos.close();
} catch (Exception e) {
e.printStackTrace();
}
}
use of org.cpsolver.studentsct.model.Enrollment in project cpsolver by UniTime.
the class StudentSctBBTest method getSolution.
/**
* Compute and return the sectioning solution. It contains only the given
* student with his/her schedule
* @return current solution
*/
public Solution<Request, Enrollment> getSolution() {
if (iSolution == null) {
iSolution = new Solution<Request, Enrollment>(this, new DefaultSingleAssignment<Request, Enrollment>());
BranchBoundSelection.Selection selection = new BranchBoundSelection(new DataProperties()).getSelection(iSolution.getAssignment(), getStudent());
Neighbour<Request, Enrollment> neighbour = selection.select();
if (neighbour != null)
neighbour.assign(iSolution.getAssignment(), 0);
iTime = selection.getTime();
iTimeoutReached = selection.isTimeoutReached();
}
return iSolution;
}
use of org.cpsolver.studentsct.model.Enrollment in project cpsolver by UniTime.
the class StudentSectioningXMLLoader method loadStudents.
/**
* Load students
* @param studentsEl students element
* @param offeringTable offering table
* @param courseTable course table
*/
protected void loadStudents(Element studentsEl, Map<Long, Offering> offeringTable, Map<Long, Course> courseTable) {
List<Enrollment> bestEnrollments = new ArrayList<Enrollment>();
List<Enrollment> currentEnrollments = new ArrayList<Enrollment>();
for (Iterator<?> i = studentsEl.elementIterator("student"); i.hasNext(); ) {
Element studentEl = (Element) i.next();
Student student = loadStudent(studentEl, offeringTable);
if (iStudentFilter != null && !iStudentFilter.accept(student))
continue;
for (Iterator<?> j = studentEl.elementIterator(); j.hasNext(); ) {
Element requestEl = (Element) j.next();
Request request = loadRequest(requestEl, student, offeringTable, courseTable);
if (request == null)
continue;
Element initialEl = requestEl.element("initial");
if (iLoadInitial && initialEl != null) {
Enrollment enrollment = loadEnrollment(initialEl, request);
if (enrollment != null)
request.setInitialAssignment(enrollment);
}
Element currentEl = requestEl.element("current");
if (iLoadCurrent && currentEl != null) {
Enrollment enrollment = loadEnrollment(currentEl, request);
if (enrollment != null)
currentEnrollments.add(enrollment);
}
Element bestEl = requestEl.element("best");
if (iLoadBest && bestEl != null) {
Enrollment enrollment = loadEnrollment(bestEl, request);
if (enrollment != null)
bestEnrollments.add(enrollment);
}
}
getModel().addStudent(student);
}
if (!bestEnrollments.isEmpty()) {
// Enrollments with a reservation must go first
for (Enrollment enrollment : bestEnrollments) {
if (enrollment.getReservation() == null)
continue;
if (!enrollment.getStudent().isAvailable(enrollment)) {
sLogger.warn("Enrollment " + enrollment + " is conflicting: student not available.");
continue;
}
Map<Constraint<Request, Enrollment>, Set<Enrollment>> conflicts = getModel().conflictConstraints(getAssignment(), enrollment);
if (conflicts.isEmpty())
getAssignment().assign(0, enrollment);
else
sLogger.warn("Enrollment " + enrollment + " conflicts with " + conflicts);
}
for (Enrollment enrollment : bestEnrollments) {
if (enrollment.getReservation() != null)
continue;
if (!enrollment.getStudent().isAvailable(enrollment)) {
sLogger.warn("Enrollment " + enrollment + " is conflicting: student not available.");
continue;
}
Map<Constraint<Request, Enrollment>, Set<Enrollment>> conflicts = getModel().conflictConstraints(getAssignment(), enrollment);
if (conflicts.isEmpty())
getAssignment().assign(0, enrollment);
else
sLogger.warn("Enrollment " + enrollment + " conflicts with " + conflicts);
}
getModel().saveBest(getAssignment());
}
if (!currentEnrollments.isEmpty()) {
for (Request request : getModel().variables()) getAssignment().unassign(0, request);
// Enrollments with a reservation must go first
for (Enrollment enrollment : currentEnrollments) {
if (enrollment.getReservation() == null)
continue;
if (!enrollment.getStudent().isAvailable(enrollment)) {
sLogger.warn("Enrollment " + enrollment + " is conflicting: student not available.");
continue;
}
Map<Constraint<Request, Enrollment>, Set<Enrollment>> conflicts = getModel().conflictConstraints(getAssignment(), enrollment);
if (conflicts.isEmpty())
getAssignment().assign(0, enrollment);
else
sLogger.warn("Enrollment " + enrollment + " conflicts with " + conflicts);
}
for (Enrollment enrollment : currentEnrollments) {
if (enrollment.getReservation() != null)
continue;
if (!enrollment.getStudent().isAvailable(enrollment)) {
sLogger.warn("Enrollment " + enrollment + " is conflicting: student not available.");
continue;
}
Map<Constraint<Request, Enrollment>, Set<Enrollment>> conflicts = getModel().conflictConstraints(getAssignment(), enrollment);
if (conflicts.isEmpty())
getAssignment().assign(0, enrollment);
else
sLogger.warn("Enrollment " + enrollment + " conflicts with " + conflicts);
}
}
}
use of org.cpsolver.studentsct.model.Enrollment in project cpsolver by UniTime.
the class StudentSectioningModel method computeOnlineSectioningInfos.
/**
* Compute online student sectioning infos for all sections (see
* {@link Section#getSpaceExpected()} and {@link Section#getSpaceHeld()}).
* @param assignment current assignment
*/
public void computeOnlineSectioningInfos(Assignment<Request, Enrollment> assignment) {
clearOnlineSectioningInfos();
for (Student student : getStudents()) {
if (!student.isDummy())
continue;
for (Request request : student.getRequests()) {
if (!(request instanceof CourseRequest))
continue;
CourseRequest courseRequest = (CourseRequest) request;
Enrollment enrollment = assignment.getValue(courseRequest);
if (enrollment != null) {
for (Section section : enrollment.getSections()) {
section.setSpaceHeld(courseRequest.getWeight() + section.getSpaceHeld());
}
}
List<Enrollment> feasibleEnrollments = new ArrayList<Enrollment>();
int totalLimit = 0;
for (Enrollment enrl : courseRequest.values(assignment)) {
boolean overlaps = false;
for (Request otherRequest : student.getRequests()) {
if (otherRequest.equals(courseRequest) || !(otherRequest instanceof CourseRequest))
continue;
Enrollment otherErollment = assignment.getValue(otherRequest);
if (otherErollment == null)
continue;
if (enrl.isOverlapping(otherErollment)) {
overlaps = true;
break;
}
}
if (!overlaps) {
feasibleEnrollments.add(enrl);
if (totalLimit >= 0) {
int limit = enrl.getLimit();
if (limit < 0)
totalLimit = -1;
else
totalLimit += limit;
}
}
}
double increment = courseRequest.getWeight() / (totalLimit > 0 ? totalLimit : feasibleEnrollments.size());
for (Enrollment feasibleEnrollment : feasibleEnrollments) {
for (Section section : feasibleEnrollment.getSections()) {
if (totalLimit > 0) {
section.setSpaceExpected(section.getSpaceExpected() + increment * feasibleEnrollment.getLimit());
} else {
section.setSpaceExpected(section.getSpaceExpected() + increment);
}
}
}
}
}
}
Aggregations