org.unitime.timetable.dataexchange.TravelTimesImport.java Source code

Introduction

Here is the source code for org.unitime.timetable.dataexchange.TravelTimesImport.java

Source

/*
 * UniTime 3.4 - 3.5 (University Timetabling Application)
 * Copyright (C) 2012 - 2013, UniTime LLC, and individual contributors
 * as indicated by the @authors tag.
 * 
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 * 
*/
package org.unitime.timetable.dataexchange;

import java.util.Iterator;

import org.dom4j.Element;
import org.unitime.timetable.model.Location;
import org.unitime.timetable.model.NonUniversityLocation;
import org.unitime.timetable.model.Room;
import org.unitime.timetable.model.Session;
import org.unitime.timetable.model.TravelTime;

/**
 * @author Tomas Muller
 */
public class TravelTimesImport extends BaseImport {

    public void loadXml(Element root) throws Exception {
        if (!root.getName().equalsIgnoreCase("traveltimes")) {
            throw new Exception("Given XML file is not a Travel Times load file.");
        }
        try {
            beginTransaction();

            String campus = root.attributeValue("campus");
            String year = root.attributeValue("year");
            String term = root.attributeValue("term");

            Session session = Session.getSessionUsingInitiativeYearTerm(campus, year, term);
            if (session == null) {
                throw new Exception("No session found for the given campus, year, and term.");
            }

            info("Deleting existing travel times...");
            getHibSession().createQuery("delete from TravelTime where session.uniqueId = :sessionId")
                    .setLong("sessionId", session.getUniqueId()).executeUpdate();

            info("Importing travel times...");
            for (Iterator i = root.elementIterator("from"); i.hasNext();) {
                Element fromEl = (Element) i.next();
                Location from = findLocation(session.getUniqueId(), fromEl);
                if (from == null)
                    continue;

                for (Iterator j = fromEl.elementIterator("to"); j.hasNext();) {
                    Element toEl = (Element) j.next();
                    Location to = findLocation(session.getUniqueId(), toEl);
                    if (to == null)
                        continue;

                    TravelTime time = new TravelTime();
                    time.setSession(session);
                    time.setLocation1Id(Math.min(from.getUniqueId(), to.getUniqueId()));
                    time.setLocation2Id(Math.max(from.getUniqueId(), to.getUniqueId()));
                    time.setDistance(Integer.parseInt(toEl.getTextTrim()));

                    getHibSession().saveOrUpdate(time);
                }

            }

            info("All done.");

            commitTransaction();
        } catch (Exception e) {
            fatal("Exception: " + e.getMessage(), e);
            rollbackTransaction();
            throw e;
        }
    }

    private Location findLocation(Long sessionId, Element element) {
        if (element.attributeValue("id") != null) {
            Room room = (Room) getHibSession().createQuery(
                    "select r from Room r where r.externalUniqueId=:externalId and r.building.session.uniqueId=:sessionId")
                    .setLong("sessionId", sessionId).setString("externalId", element.attributeValue("id"))
                    .setCacheable(true).setMaxResults(1).uniqueResult();
            if (room != null)
                return room;
        }
        if (element.attributeValue("building") != null && element.attributeValue("roomNbr") != null) {
            Room room = (Room) getHibSession().createQuery(
                    "select  r from Room r where r.roomNumber=:roomNbr and r.building.abbreviation = :building and r.session.uniqueId=:sessionId")
                    .setLong("sessionId", sessionId).setString("building", element.attributeValue("building"))
                    .setString("roomNbr", element.attributeValue("roomNbr")).setCacheable(true).setMaxResults(1)
                    .uniqueResult();
            if (room != null)
                return room;
        }
        if (element.attributeValue("name") != null) {
            Room room = (Room) getHibSession().createQuery(
                    "select  r from Room r where (r.building.abbreviation || ' ' || r.roomNumber) = :name and r.session.uniqueId=:sessionId")
                    .setLong("sessionId", sessionId).setString("name", element.attributeValue("name"))
                    .setCacheable(true).setMaxResults(1).uniqueResult();
            if (room != null)
                return room;

            NonUniversityLocation location = (NonUniversityLocation) getHibSession().createQuery(
                    "select  r from NonUniversityLocation r where r.name = :name and r.session.uniqueId=:sessionId")
                    .setLong("sessionId", sessionId).setString("name", element.attributeValue("name"))
                    .setCacheable(true).setMaxResults(1).uniqueResult();
            if (location != null)
                return location;
        }
        warn("Location " + element.asXML() + " not found.");
        return null;
    }

}