ArrayRoutingNetworkFactory.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * ArrayRoutingNetworkFactory.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2012 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.router.util;

import java.util.HashMap;
import java.util.Map;

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.api.core.v01.network.Network;
import org.matsim.api.core.v01.network.Node;
import org.matsim.core.network.LinkFactory;

public class ArrayRoutingNetworkFactory extends AbstractRoutingNetworkFactory {
    
    private final static Logger log = LogManager.getLogger(ArrayRoutingNetworkFactory.class);
    
    private int nodeArrayIndexCounter;
    private int linkArrayIndexCounter;

    @Override
    public synchronized ArrayRoutingNetwork createRoutingNetwork(final Network network) {
        this.nodeArrayIndexCounter = 0;
        this.linkArrayIndexCounter = 0;
        
        ArrayRoutingNetwork routingNetwork = new ArrayRoutingNetwork(network);
        
        for (Node node : network.getNodes().values()) {
            RoutingNetworkNode routingNode = createRoutingNetworkNode(node, node.getOutLinks().size());
            routingNetwork.addNode(routingNode);
        }
        Map<Id<Link>, RoutingNetworkLink> routingLinks = new HashMap<Id<Link>, RoutingNetworkLink>();
        for (Link link : network.getLinks().values()) {
            RoutingNetworkNode fromNode = routingNetwork.getNodes().get(link.getFromNode().getId());
            RoutingNetworkNode toNode = routingNetwork.getNodes().get(link.getToNode().getId());
            RoutingNetworkLink dijkstraLink = createRoutingNetworkLink(link, fromNode, toNode);
            routingLinks.put(dijkstraLink.getId(), dijkstraLink);
        }
        
        for (Node node : network.getNodes().values()) {
            RoutingNetworkLink[] outLinks = new RoutingNetworkLink[node.getOutLinks().size()];
            
            int i = 0;
            for (Link outLink : node.getOutLinks().values()) {
                outLinks[i] = routingLinks.remove(outLink.getId());
                i++;
            }
            
            RoutingNetworkNode dijkstraNode = routingNetwork.getNodes().get(node.getId());
            dijkstraNode.setOutLinksArray(outLinks);
        }
        
        if (routingLinks.size() > 0) log.warn("Not all links have been use in the ArrayRoutingNetwork - check connectivity of input network!");
        
        return routingNetwork;
    }

    @Override
    public ArrayRoutingNetworkNode createRoutingNetworkNode(final Node node, final int numOutLinks) {
        return new ArrayRoutingNetworkNode(node, numOutLinks, this.nodeArrayIndexCounter++);
    }

    @Override
    public ArrayRoutingNetworkLink createRoutingNetworkLink(final Link link, final RoutingNetworkNode fromNode, final RoutingNetworkNode toNode) {
        return new ArrayRoutingNetworkLink(link, fromNode, toNode, this.linkArrayIndexCounter++);
    }
}