PopulationWriterHandlerImplV4.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * PlansWriterHandlerImplV4.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2007 by the members listed in the COPYING,        *
 *                   LICENSE and WARRANTY file.                            *
 * email           : info at matsim dot org                                *
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 *   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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *   See also COPYING, LICENSE and WARRANTY file                           *
 *                                                                         *
 * *********************************************************************** */

package org.matsim.core.population.io;

import static org.matsim.core.utils.io.XmlUtils.encodeAttributeValue;

import java.io.BufferedWriter;
import java.io.IOException;

import org.matsim.api.core.v01.Coord;
import org.matsim.api.core.v01.network.Network;
import org.matsim.api.core.v01.network.Node;
import org.matsim.api.core.v01.population.Activity;
import org.matsim.api.core.v01.population.Leg;
import org.matsim.api.core.v01.population.Person;
import org.matsim.api.core.v01.population.Plan;
import org.matsim.api.core.v01.population.Population;
import org.matsim.api.core.v01.population.Route;
import org.matsim.core.population.PersonUtils;
import org.matsim.core.population.routes.NetworkRoute;
import org.matsim.core.population.routes.RouteUtils;
import org.matsim.core.utils.geometry.CoordinateTransformation;
import org.matsim.core.utils.geometry.transformations.IdentityTransformation;
import org.matsim.core.utils.io.MatsimXmlWriter;
import org.matsim.core.utils.io.XmlUtils;
import org.matsim.core.utils.misc.Time;

/*package*/ class PopulationWriterHandlerImplV4 extends AbstractPopulationWriterHandler {

    private final CoordinateTransformation coordinateTransformation;
    private final Network network;

    public PopulationWriterHandlerImplV4(
            final Network network) {
        this( new IdentityTransformation() , network );
    }

    public PopulationWriterHandlerImplV4(
            final CoordinateTransformation coordinateTransformation,
            final Network network) {
        this.coordinateTransformation = coordinateTransformation;
        this.network = network;
    }

    @Override
    public void writeHeaderAndStartElement(final BufferedWriter out) throws IOException {
        out.write("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n");
        out.write("<!DOCTYPE plans SYSTEM \"" + MatsimXmlWriter.DEFAULT_DTD_LOCATION + "plans_v4.dtd\">\n\n");
    }

    // <plans ... > ... </plans>

    @Override
    public void startPlans(final Population plans, final BufferedWriter out) throws IOException {
        out.write("<plans");
        if (plans.getName() != null) {
            out.write(" name=\"" + encodeAttributeValue(plans.getName()) + "\"");
        }
        out.write(">\n\n");
    }

    @Override
    public void endPlans(final BufferedWriter out) throws IOException {
        out.write("</plans>\n");
        out.flush();
    }

    // <person ... > ... </person>

    @Override
    public void startPerson(final Person person, final BufferedWriter out) throws IOException {
        out.write("\t<person id=\"");
        out.write(encodeAttributeValue(person.getId().toString()));
        out.write("\"");
        if (PersonUtils.getSex(person) != null) {
            out.write(" sex=\"");
            out.write(PersonUtils.getSex(person));
            out.write("\"");
        }
        if (PersonUtils.getAge(person) != null) {
            out.write(" age=\"");
            out.write(Integer.toString(PersonUtils.getAge(person)));
            out.write("\"");
        }
        if (PersonUtils.getLicense(person) != null) {
            out.write(" license=\"");
            out.write(PersonUtils.getLicense(person));
            out.write("\"");
        }
        if (PersonUtils.getCarAvail(person) != null) {
            out.write(" car_avail=\"");
            out.write(PersonUtils.getCarAvail(person));
            out.write("\"");
        }
        if (PersonUtils.isEmployed(person) != null) {
            out.write(" employed=\"");
            out.write((PersonUtils.isEmployed(person) ? "yes" : "no"));
            out.write("\"");
        }
        out.write(">\n");
    }

    @Override
    public void endPerson(final BufferedWriter out) throws IOException {
        out.write("\t</person>\n\n");
    }

    // <travelcard ... />

    @Override
    public void startTravelCard(final String travelcard, final BufferedWriter out) throws IOException {
        out.write("\t\t<travelcard type=\"");
        out.write(travelcard);
        out.write("\" />\n\n");
    }

    @Override
    public void endTravelCard(final BufferedWriter out) throws IOException {
    }

    // <plan ... > ... </plan>

    @Override
    public void startPlan(final Plan plan, final BufferedWriter out) throws IOException {
        out.write("\t\t<plan");
        if (plan.getScore() != null) {
            out.write(" score=\"");
            out.write(plan.getScore().toString());
            out.write("\"");
        }
        if (PersonUtils.isSelected(plan))
            out.write(" selected=\"yes\"");
        else
            out.write(" selected=\"no\"");
        if ((plan.getType() != null)) {
            out.write(" type=\"");
            out.write(encodeAttributeValue(plan.getType()));
            out.write("\"");
        }
        out.write(">\n");
    }

    @Override
    public void endPlan(final BufferedWriter out) throws IOException {
        out.write("\t\t</plan>\n\n");
    }

    // <act ... > ... </act>

    @Override
    public void startAct(final Activity act, final BufferedWriter out) throws IOException {
        out.write("\t\t\t<act type=\"");
        out.write(encodeAttributeValue(act.getType()));
        out.write("\"");
        if (act.getLinkId() != null) {
            out.write(" link=\"");
            out.write(encodeAttributeValue(act.getLinkId().toString()));
            out.write("\"");
        }
        if (act.getFacilityId() != null) {
            out.write(" facility=\"");
            out.write(encodeAttributeValue(act.getFacilityId().toString()));
            out.write("\"");
        }
        if (act.getCoord() != null) {
            final Coord coord = coordinateTransformation.transform( act.getCoord() );
            out.write(" x=\"");
            out.write(Double.toString( coord.getX() ));
            out.write("\" y=\"");
            out.write(Double.toString( coord.getY() ));
            out.write("\"");
        }
        if (act.getStartTime().isDefined()) {
            out.write(" start_time=\"");
            out.write(Time.writeTime(act.getStartTime().seconds()));
            out.write("\"");
        }
        if (act.getMaximumDuration().isDefined()) {
            out.write(" dur=\"");
            out.write(Time.writeTime(act.getMaximumDuration().seconds()));
            out.write("\"");
        }
        if (act.getEndTime().isDefined()) {
            out.write(" end_time=\"");
            out.write(Time.writeTime(act.getEndTime().seconds()));
            out.write("\"");
        }
        out.write(" />\n");
    }

    @Override
    public void endAct(final BufferedWriter out) throws IOException {
    }

    // <leg ... > ... </leg>

    @Override
    public void startLeg(final Leg leg, final BufferedWriter out) throws IOException {
        out.write("\t\t\t<leg mode=\"");
        out.write(leg.getMode());
        out.write("\"");
        if (leg.getDepartureTime().isDefined()) {
            out.write(" dep_time=\"");
            out.write(Time.writeTime(leg.getDepartureTime().seconds()));
            out.write("\"");
        }
        if (leg.getTravelTime().isDefined()) {
            out.write(" trav_time=\"");
            out.write(Time.writeTime(leg.getTravelTime().seconds()));
            out.write("\"");
        }
//      if (leg instanceof LegImpl) {
//          LegImpl l = (LegImpl)leg;
//          if (l.getDepartureTime() + l.getTravelTime() != Time.getUndefinedTime()) {
//              out.write(" arr_time=\"");
//              out.write(Time.writeTime(l.getDepartureTime() + l.getTravelTime()));
//              out.write("\"");
//          }
//      }
        // arrival time is in dtd, but no longer evaluated in code (according to not being in API).  kai, jun'16
        out.write(">\n");
    }

    @Override
    public void endLeg(final BufferedWriter out) throws IOException {
        out.write("\t\t\t</leg>\n");
    }

    // <route ... > ... </route>

    @Override
    public void startRoute(final Route route, final BufferedWriter out) throws IOException {
        out.write("\t\t\t\t<route");
        if (!Double.isNaN(route.getDistance())) {
            out.write(" dist=\"");
            out.write(Double.toString(route.getDistance()));
            out.write("\"");
        }
        if (route.getTravelTime().isDefined()) {
            out.write(" trav_time=\"");
            out.write(Time.writeTime(route.getTravelTime().seconds()));
            out.write("\"");
        }
        out.write(">\n");

        out.write("\t\t\t\t\t");
        if (route instanceof NetworkRoute) {
            for (Node n : RouteUtils.getNodes((NetworkRoute) route, this.network)) {
                out.write(n.getId().toString());
                out.write(" ");
            }
        } else {
            String rd = route.getRouteDescription();
            if (rd != null) {
                out.write(XmlUtils.encodeContent(rd));
                out.write(" "); // this is at the moment only to maintain binary compatibility
            }
        }

        out.write("\n");
    }

    @Override
    public void endRoute(final BufferedWriter out) throws IOException {
        out.write("\t\t\t\t</route>\n");
    }

    // <!-- ============ ... ========== -->

    @Override
    public void writeSeparator(final BufferedWriter out) throws IOException {
        out.write("<!-- ====================================================================== -->\n\n");
    }

}