SOARSを使ったエージェントベースシミュレーションでAI交通シミュレータを作ってみた

import java.io.FileWriter;
import java.io.IOException;
import java.util.Random;
 
public class OdListGenerator {
    private static final double LAT_MIN = 24.30, LAT_MAX = 24.55;
    private static final double LON_MIN = 124.10, LON_MAX = 124.35;
    private static final int NUM_AGENTS = 1000;
 
    public static void main(String[] args) throws IOException {
        Random rng = new Random();
        try (FileWriter fw = new FileWriter("od_list.csv")) {
            fw.write("agent_id,origin_lat,origin_lon,dest_lat,dest_lon,depart_time\n");
            for (int id = 0; id < NUM_AGENTS; id++) {
                double oLat = LAT_MIN + rng.nextDouble() * (LAT_MAX - LAT_MIN);
                double oLon = LON_MIN + rng.nextDouble() * (LON_MAX - LON_MIN);
                double dLat = LAT_MIN + rng.nextDouble() * (LAT_MAX - LAT_MIN);
                double dLon = LON_MIN + rng.nextDouble() * (LON_MAX - LON_MIN);
                int hour = 7 + rng.nextInt(14);
                int min  = rng.nextInt(60);
                fw.write(String.format(
                    "%d,%.6f,%.6f,%.6f,%.6f,%02d:%02d:00%n",
                    id, oLat, oLon, dLat, dLon, hour, min
                ));
            }
        }
    }
}

import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
 
public class OtpClient {
    private static final String OTP_BASE_URL = "http://localhost:8080";
    private final HttpClient http = HttpClient.newHttpClient();
 
    public String queryRoute(
            double originLat, double originLon,
            double destLat,   double destLon,
            String date,      String time) throws Exception {
 
        String url = String.format(
            "%s/otp/routers/default/plan"
                + "?fromPlace=%f,%f&toPlace=%f,%f"
                + "&date=%s&time=%s&mode=BUS,WALK&numItineraries=1",
            OTP_BASE_URL,
            originLat, originLon,
            destLat,   destLon,
            date, time
        );
 
        HttpRequest req = HttpRequest.newBuilder()
            .uri(URI.create(url))
            .GET()
            .build();
 
        // レスポンスはJSON文字列。各エージェントの legs（移動区間）を取り出して処理する
        HttpResponse<String> res = http.send(req, HttpResponse.BodyHandlers.ofString());
        return res.body();
    }
}

// エージェントクラス: OTPの経路情報を保持する
public class TransportAgent extends TAgent {
    private List<Leg> legs;   // OTPから取得した移動区間リスト
    private int currentLeg = 0;
 
    public TransportAgent(long id, List<Leg> legs) {
        super(id);
        this.legs = legs;
    }
 
    public Leg getNextLeg() {
        return legs.get(currentLeg);
    }
}
 
// ルール: 各ステップで次の区間に移動するかを判断する
public class MoveToNextStopRule extends TAgentRule {
    @Override
    public void doIt(TTime currentTime, TAgent agent, ...) {
        TransportAgent ta = (TransportAgent) agent;
        Leg next = ta.getNextLeg();
 
        // 次の区間の出発時刻になったらスポットを移動
        if (currentTime.isAfterOrEqual(next.departureTime)) {
            agent.moveTo(next.toSpot);
            ta.currentLeg++;
        }
    }
}

agent_id, timestamp,          lat,      lon,      mode
0,        2024-01-15T08:05:00, 24.3421, 124.1567, WALK
0,        2024-01-15T08:12:00, 24.3398, 124.1623, BUS
0,        2024-01-15T08:45:00, 24.4607, 124.2133, WALK
...