runner.py

#!/usr/bin/env python
# Eclipse SUMO, Simulation of Urban MObility; see https://eclipse.dev/sumo
# Copyright (C) 2019-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    runner.py
# @author  Michael Behrisch
# @date    2019-05-01

from __future__ import absolute_import
from __future__ import print_function


import os
import subprocess
import sys
import time
import glob
if "SUMO_HOME" in os.environ:
    sys.path.append(os.path.join(os.environ["SUMO_HOME"], "tools"))
from sumolib import checkBinary  # noqa

javac = "javac"
java = "java"
if 'JAVA_HOME' in os.environ:
    javac = os.path.join(os.environ['JAVA_HOME'], "bin", javac)
    java = os.path.join(os.environ['JAVA_HOME'], "bin", java)

# use latest version
useLibsumo = 'LIBSUMO_AS_TRACI' in os.environ
prefix = "libsumo" if useLibsumo else "libtraci"
prefix = os.path.join(os.environ['SUMO_HOME'], "bin", prefix)
files = [f for f in glob.glob(prefix + "-*.jar") if not f.endswith("sources.jar")]
traciJar = max(files, key=os.path.getmtime)
# print("traciJar", traciJar)

assert os.path.exists(traciJar)

for f in sys.argv[1:]:
    fname = "data/%s.java" % f
    if useLibsumo:
        with open(fname, encoding="utf8") as fin:
            filedata = fin.read()
        with open(fname, 'w', encoding="utf8") as fob:
            fob.write(filedata.replace('libtraci', 'libsumo'))
    subprocess.check_call([javac, "-cp", traciJar, fname])

os.environ["PATH"] += os.pathsep + os.path.join(os.environ['SUMO_HOME'], "bin")
procs = [subprocess.Popen([java, "-Djava.library.path=" + os.path.join(os.environ['SUMO_HOME'], "bin"),
                           "-cp", os.pathsep.join([traciJar, "data"]), sys.argv[1],
                           checkBinary('sumo'), "data/config.sumocfg"])]
if len(sys.argv) > 2:
    time.sleep(5)  # give sumo some time to start
    procs += [subprocess.Popen([java, "-Djava.library.path=" + os.path.join(os.environ['SUMO_HOME'], "bin"),
                                "-cp", os.pathsep.join([traciJar, "data"]), f]) for f in sys.argv[2:]]
for p in procs:
    p.wait()