buildFullGraph.py

#!/usr/bin/env python
# Eclipse SUMO, Simulation of Urban MObility; see https://eclipse.dev/sumo
# Copyright (C) 2011-2025 German Aerospace Center (DLR) and others.
# This program and the accompanying materials are made available under the
# terms of the Eclipse Public License 2.0 which is available at
# https://www.eclipse.org/legal/epl-2.0/
# This Source Code may also be made available under the following Secondary
# Licenses when the conditions for such availability set forth in the Eclipse
# Public License 2.0 are satisfied: GNU General Public License, version 2
# or later which is available at
# https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html
# SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later

# @file    buildFullGraph.py
# @author  Jakob Erdmann
# @date    2024-05-02

"""
Extend network with direct connection between all edges that permit the given vClass
"""
from __future__ import absolute_import
from __future__ import print_function

import sys
import os
import subprocess
sys.path.append(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
import sumolib  # noqa
from sumolib.options import ArgumentParser  # noqa


def parse_args():
    optParser = ArgumentParser()
    optParser.add_option("net", category="input", type=optParser.net_file,
                         help="the network to patch")
    optParser.add_option("-o", "--output-file", dest="outfile", category="output", type=optParser.net_file,
                         help="the output network file")
    optParser.add_option("--vclass",
                         help="class for which the fully connected graph should be built")
    optParser.add_option("--speed", type=float, default=100/3.6,
                         help="speed of added edges")
    optParser.add_option("--numlanes", type=int, default=1,
                         help="number of lanes for added edges")
    optParser.add_option("--width", type=float,
                         help="width of added edges")
    options = optParser.parse_args()
    if not options.outfile:
        options.outfile = "patched." + options.net

    return options


def main(options):
    net = sumolib.net.readNet(options.net)
    baseEdges = [e for e in net.getEdges() if e.allows(options.vclass)]
    prefix = options.net
    if prefix.endswith(".net.xml.gz"):
        prefix = prefix[:-11]
    elif prefix.endswith(".net.xml"):
        prefix = prefix[:-8]

    edgePatch = prefix + ".patch.edg.xml"
    conPatch = prefix + ".patch.con.xml"
    with open(edgePatch, 'w') as outfe, open(conPatch, 'w') as outfc:
        sumolib.writeXMLHeader(outfe, "$Id$", "edges", options=options)
        sumolib.writeXMLHeader(outfc, "$Id$", "connections", options=options)
        for e1 in baseEdges:
            for e2 in baseEdges:
                if e1 != e2 and e2 not in e1.getOutgoing():
                    newEID = "%s_%s" % (e1.getToNode().getID(), e2.getFromNode().getID())
                    width = options.width if options.width is not None else e1.getLanes()[-1].getWidth()
                    outfe.write('    <edge id="%s" from="%s" to="%s" speed="%s" numLanes="%s" width="%s" allow="%s"/>\n' % (  # noqa
                        newEID, e1.getToNode().getID(), e2.getFromNode().getID(),
                        options.speed, options.numlanes, width, options.vclass))
                    outfc.write('    <connection from="%s" to="%s" fromLane="0" toLane="0"/>\n' % (e1.getID(), newEID))
                    outfc.write('    <connection from="%s" to="%s" fromLane="0" toLane="0"/>\n' % (newEID, e2.getID()))

        outfe.write("</edges>\n")
        outfc.write("</connections>\n")

    NETCONVERT = sumolib.checkBinary('netconvert')
    subprocess.call([NETCONVERT,
                     '-s', options.net,
                     '-e', edgePatch,
                     '-x', conPatch,
                     '-o', options.outfile])


if __name__ == "__main__":
    main(parse_args())