LanesWriter.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * MatsimLaneWriter
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2009 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.lanes;

import java.io.BufferedWriter;
import java.io.IOException;
import java.util.Map;

import jakarta.xml.bind.JAXBContext;
import jakarta.xml.bind.JAXBException;
import jakarta.xml.bind.Marshaller;
import javax.xml.crypto.MarshalException;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.matsim.api.core.v01.Id;
import org.matsim.api.core.v01.network.Link;
import org.matsim.core.api.internal.MatsimSomeWriter;
import org.matsim.core.gbl.Gbl;
import org.matsim.core.utils.io.IOUtils;
import org.matsim.core.utils.io.MatsimJaxbXmlWriter;
import org.matsim.jaxb.lanedefinitions20.*;
import org.matsim.utils.objectattributes.ObjectAttributesConverter;

public final class LanesWriter extends MatsimJaxbXmlWriter implements MatsimSomeWriter {

    private static final Logger log = LogManager.getLogger(LanesWriter.class);

    private Lanes laneDefinitions;

    private final ObjectAttributesConverter attributesConverter = new ObjectAttributesConverter();

    public LanesWriter(Lanes lanedefs) {
//      log.info("Using LaneDefinitionWriter20...");
        this.laneDefinitions = lanedefs;
    }

    @Override
    public void write(String filename) {
        log.info( Gbl.aboutToWrite( "lanes", filename ) ) ;
    JAXBContext jc;
        try {
            XMLLaneDefinitions xmlLaneDefinitions = convertDataToXml();
            jc = JAXBContext.newInstance(org.matsim.jaxb.lanedefinitions20.ObjectFactory.class);
            Marshaller m = jc.createMarshaller();
            super.setMarshallerProperties(LanesReader.SCHEMALOCATIONV20, m);
            BufferedWriter bufout = IOUtils.getBufferedWriter(filename);
            m.marshal(xmlLaneDefinitions, bufout);
            bufout.close();
        } catch (JAXBException | IOException | MarshalException e) {
            throw new RuntimeException(e);
        }
    }

    private XMLLaneDefinitions convertDataToXml() throws MarshalException {
        ObjectFactory fac = new ObjectFactory();
        XMLLaneDefinitions xmllaneDefs = fac.createXMLLaneDefinitions();

        for (LanesToLinkAssignment ltla : this.laneDefinitions.getLanesToLinkAssignments().values()) {
            XMLLanesToLinkAssignmentType xmlltla = fac.createXMLLanesToLinkAssignmentType();
            xmlltla.setLinkIdRef(ltla.getLinkId().toString());

            for (Lane bl : ltla.getLanes().values()) {
                XMLLaneType xmllane = fac.createXMLLaneType();
                xmllane.setId(bl.getId().toString());

                if ((bl.getToLinkIds() == null && bl.getToLaneIds() != null) || 
                        (bl.getToLinkIds() != null && bl.getToLaneIds() == null)){
                    xmllane.setLeadsTo(fac.createXMLLaneTypeXMLLeadsTo());
                }
                else {
                    throw new MarshalException("Either at least one toLinkId or (exclusive) one toLaneId must" +
                            "be set for Lane Id " + bl.getId() + " on link Id " + ltla.getLinkId() + "! Cannot write according to XML grammar.");
                }
                
                if (bl.getToLinkIds() != null){
                    for (Id<Link> id : bl.getToLinkIds()) {
                        XMLIdRefType xmlToLink = fac.createXMLIdRefType();
                        xmlToLink.setRefId(id.toString());
                        xmllane.getLeadsTo().getToLink().add(xmlToLink);
                    }
                }
                else if (bl.getToLaneIds() != null){
                    for (Id<Lane> id : bl.getToLaneIds()) {
                        XMLIdRefType xmlToLink = fac.createXMLIdRefType();
                        xmlToLink.setRefId(id.toString());
                        xmllane.getLeadsTo().getToLane().add(xmlToLink);
                    }
                }
                
                XMLLaneType.XMLCapacity capacity = new XMLLaneType.XMLCapacity();
                capacity.setVehiclesPerHour(bl.getCapacityVehiclesPerHour());
                xmllane.setCapacity(capacity);

                XMLLaneType.XMLRepresentedLanes lanes = new XMLLaneType.XMLRepresentedLanes();
                lanes.setNumber(bl.getNumberOfRepresentedLanes());
                xmllane.setRepresentedLanes(lanes);

                XMLLaneType.XMLStartsAt startsAt = new XMLLaneType.XMLStartsAt();
                startsAt.setMeterFromLinkEnd(bl.getStartsAtMeterFromLinkEnd());
                xmllane.setStartsAt(startsAt);

                xmllane.setAlignment(bl.getAlignment());

                if (bl.getAttributes() != null) {
                    xmllane.setAttributes(fac.createXMLLaneTypeXMLAttributes());

                    // write attributes
                    for (Map.Entry<String, Object> objAttribute : bl.getAttributes().getAsMap().entrySet()) {
                        Class<?> clazz = objAttribute.getValue().getClass();
                        String converted = attributesConverter.convertToString(objAttribute.getValue());
                        if (converted != null) {
                            XMLAttributeType att = fac.createXMLAttributeType();
                            att.setName(objAttribute.getKey());
                            att.setValue(converted);
                            att.setClazz(clazz.getCanonicalName());
                            xmllane.getAttributes().getAttributeList().add(att);
                        }
                    }
                }

                xmlltla.getLane().add(xmllane);
            }
            xmllaneDefs.getLanesToLinkAssignment().add(xmlltla);
        } //end writing lanesToLinkAssignments

        return xmllaneDefs;
    }

}