Skip to content

Commit

Permalink
boilerplate drt test, wip, for JB to complete
Browse files Browse the repository at this point in the history
  • Loading branch information
@jakobrehmann
jakobrehmann committed May 10, 2024
1 parent c77a9f6 commit 1f12853
Showing 1 changed file with 271 additions and 0 deletions.
271 changes: 271 additions & 0 deletions ...sibility/src/test/java/org/matsim/contrib/accessibility/run/TinyDRTAccessibilityTest.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,271 @@
/* *********************************************************************** *
* project: org.matsim.* *
* *
* *********************************************************************** *
* *
* copyright : (C) 2013 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.contrib.accessibility.run;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
import org.matsim.api.core.v01.Coord;
import org.matsim.api.core.v01.Id;
import org.matsim.api.core.v01.Scenario;
import org.matsim.api.core.v01.TransportMode;
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.contrib.accessibility.*;
import org.matsim.contrib.accessibility.AccessibilityConfigGroup.AreaOfAccesssibilityComputation;
import org.matsim.core.config.Config;
import org.matsim.core.config.ConfigUtils;
import org.matsim.core.config.groups.FacilitiesConfigGroup;
import org.matsim.core.config.groups.ScoringConfigGroup;
import org.matsim.core.controler.Controler;
import org.matsim.core.controler.OutputDirectoryHierarchy.OverwriteFileSetting;
import org.matsim.core.network.NetworkUtils;
import org.matsim.core.scenario.MutableScenario;
import org.matsim.core.scenario.ScenarioUtils;
import org.matsim.core.utils.collections.Tuple;
import org.matsim.facilities.ActivityFacilities;
import org.matsim.facilities.ActivityFacility;
import org.matsim.testcases.MatsimTestUtils;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
* A small test that enables to easily compare results with hand-computed results.
*
* @author dziemke
*/
public class TinyDRTAccessibilityTest {

private static final Logger LOG = LogManager.getLogger(TinyDRTAccessibilityTest.class);

@RegisterExtension private MatsimTestUtils utils = new MatsimTestUtils();

@Test
void testWithBoundingBox() {
final Config config = createTestConfig();


ScoringConfigGroup.ModeParams drtParams = new ScoringConfigGroup.ModeParams("drt");
drtParams.setMarginalUtilityOfDistance(0.0);
drtParams.setMarginalUtilityOfTraveling(0.0);
drtParams.setConstant(0.0);
config.scoring().addModeParams(drtParams);

double min = 0.; // Values for bounding box usually come from a config file
double max = 200.;

AccessibilityConfigGroup acg = ConfigUtils.addOrGetModule(config, AccessibilityConfigGroup.class) ;
acg.setAreaOfAccessibilityComputation(AreaOfAccesssibilityComputation.fromBoundingBox);
acg.setBoundingBoxBottom(min);
acg.setBoundingBoxTop(max);
acg.setBoundingBoxLeft(min);
acg.setBoundingBoxRight(max);
acg.setUseParallelization(false);

final Scenario sc = createTestScenario(config);

Controler controler = new Controler(sc);

final AccessibilityModule module = new AccessibilityModule();
final ResultsComparator resultsComparator = new ResultsComparator();
module.addFacilityDataExchangeListener(resultsComparator);
controler.addOverridingModule(module);

controler.run();
}


private Config createTestConfig() {
final Config config = ConfigUtils.createConfig();

final AccessibilityConfigGroup acg = ConfigUtils.addOrGetModule(config, AccessibilityConfigGroup.class);
acg.setTileSize_m(100);
acg.setComputingAccessibilityForMode(Modes4Accessibility.freespeed, true);
acg.setComputingAccessibilityForMode(Modes4Accessibility.car, true);
acg.setComputingAccessibilityForMode(Modes4Accessibility.estimatedDrt, true);

config.controller().setLastIteration(0);
config.controller().setOutputDirectory(utils.getOutputDirectory());
config.controller().setOverwriteFileSetting(OverwriteFileSetting.deleteDirectoryIfExists);

config.routing().setRoutingRandomness(0.);

return config;
}


private static Scenario createTestScenario(final Config config) {
// final Scenario scenario = ScenarioUtils.loadScenario(config);
MutableScenario scenario = (MutableScenario) ScenarioUtils.loadScenario(config);
Network network = createLessSymmetricTestNetwork();
scenario.setNetwork(network);

// Creating test opportunities (facilities); one on each link with same ID as link and coord on center of link
final ActivityFacilities opportunities = scenario.getActivityFacilities();
ActivityFacility facility1 = opportunities.getFactory().createActivityFacility(Id.create("1", ActivityFacility.class), new Coord(200, 0));
opportunities.addActivityFacility(facility1);
ActivityFacility facility2 = opportunities.getFactory().createActivityFacility(Id.create("2", ActivityFacility.class), new Coord(200, 200));
opportunities.addActivityFacility(facility2);
scenario.getConfig().facilities().setFacilitiesSource(FacilitiesConfigGroup.FacilitiesSource.setInScenario);
return scenario;
}


/**
* This method creates a test network. It is used for example in PtMatrixTest.java to test the pt simulation in MATSim.
* The network has 9 nodes and 8 links (see the sketch below).
*
* @return the created test network
*
* @author thomas
* @author tthunig
*/
public static Network createLessSymmetricTestNetwork() {
/*
* (2) (5)------(8)
* | |
* | |
* (1)------(4)------(7)
* | |
* | |
* (3) (6)------(9)
*/
double freespeed = 2.7;
double capacity = 500.;
double numLanes = 1.;

MutableScenario scenario = (MutableScenario) ScenarioUtils.createScenario(ConfigUtils.createConfig());

Network network = (Network) scenario.getNetwork();

// Nodes
Node node1 = NetworkUtils.createAndAddNode(network, Id.create(1, Node.class), new Coord((double) 10, (double) 100));
Node node2 = NetworkUtils.createAndAddNode(network, Id.create(2, Node.class), new Coord((double) 10, (double) 190));
Node node3 = NetworkUtils.createAndAddNode(network, Id.create(3, Node.class), new Coord((double) 10, (double) 10));
Node node4 = NetworkUtils.createAndAddNode(network, Id.create(4, Node.class), new Coord((double) 120, (double) 100));
Node node5 = NetworkUtils.createAndAddNode(network, Id.create(5, Node.class), new Coord((double) 120, (double) 190));
Node node6 = NetworkUtils.createAndAddNode(network, Id.create(6, Node.class), new Coord((double) 120, (double) 10));
Node node7 = NetworkUtils.createAndAddNode(network, Id.create(7, Node.class), new Coord((double) 190, (double) 100));
Node node8 = NetworkUtils.createAndAddNode(network, Id.create(8, Node.class), new Coord((double) 190, (double) 190));
Node node9 = NetworkUtils.createAndAddNode(network, Id.create(9, Node.class), new Coord((double) 190, (double) 10));

Set<String> modes = new HashSet<>();
modes.add("car");

// Links (bi-directional)
NetworkUtils.createAndAddLink(network,Id.create(1, Link.class), node1, node2, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(1, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(2, Link.class), node2, node1, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(2, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(3, Link.class), node1, node3, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(3, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(4, Link.class), node3, node1, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(4, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(5, Link.class), node1, node4, (double) 110, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(5, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(6, Link.class), node4, node1, (double) 110, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(6, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(7, Link.class), node4, node5, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(7, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(8, Link.class), node5, node4, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(8, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(9, Link.class), node4, node6, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(9, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(10, Link.class), node6, node4, (double) 90, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(10, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(11, Link.class), node4, node7, (double) 70, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(11, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(12, Link.class), node7, node4, (double) 70, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(12, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(13, Link.class), node5, node8, (double) 70, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(13, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(14, Link.class), node8, node5, (double) 70, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(14, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(15, Link.class), node6, node9, (double) 70, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(15, Link.class)).setAllowedModes(modes);

NetworkUtils.createAndAddLink(network,Id.create(16, Link.class), node9, node6, (double) 70, freespeed, capacity, numLanes);
network.getLinks().get(Id.create(16, Link.class)).setAllowedModes(modes);

return network;
}


static class ResultsComparator implements FacilityDataExchangeInterface{
private Map<Tuple<ActivityFacility, Double>, Map<String,Double>> accessibilitiesMap = new HashMap<>() ;

@Override
public void setFacilityAccessibilities(ActivityFacility measurePoint, Double timeOfDay, String mode, double accessibility) {
Tuple<ActivityFacility, Double> key = new Tuple<>(measurePoint, timeOfDay);
if (!accessibilitiesMap.containsKey(key)) {
Map<String,Double> accessibilitiesByMode = new HashMap<>();
accessibilitiesMap.put(key, accessibilitiesByMode);
}
accessibilitiesMap.get(key).put(mode, accessibility);
}

@Override
public void finish() {
for (Tuple<ActivityFacility, Double> tuple : accessibilitiesMap.keySet()) {
LOG.warn("CHECK X = " + tuple.getFirst().getCoord().getX() + " -- Y = " + tuple.getFirst().getCoord().getY() + " -- freespeed value = " + accessibilitiesMap.get(tuple).get("freespeed"));
LOG.warn("CHECK X = " + tuple.getFirst().getCoord().getX() + " -- Y = " + tuple.getFirst().getCoord().getY() + " -- car value = " + accessibilitiesMap.get(tuple).get(TransportMode.car));
if (tuple.getFirst().getCoord().getX() == 50.) {
if (tuple.getFirst().getCoord().getY() == 50.) {
// Assertions.assertEquals(-0.017248522428805767, accessibilitiesMap.get(tuple).get("freespeed"), MatsimTestUtils.EPSILON);
// Assertions.assertEquals(-0.017240250823867296, accessibilitiesMap.get(tuple).get(TransportMode.car), MatsimTestUtils.EPSILON);

} else if (tuple.getFirst().getCoord().getY() == 150.) {
// Assertions.assertEquals(-0.017248522428805767, accessibilitiesMap.get(tuple).get("freespeed"), MatsimTestUtils.EPSILON);
// Assertions.assertEquals(-0.017240250823867296, accessibilitiesMap.get(tuple).get(TransportMode.car), MatsimTestUtils.EPSILON);
}
}
if (tuple.getFirst().getCoord().getX() == 150.) {
if (tuple.getFirst().getCoord().getY() == 50.) {
// Assertions.assertEquals(0.2758252376673665, accessibilitiesMap.get(tuple).get("freespeed"), MatsimTestUtils.EPSILON);
// Assertions.assertEquals(0.27582980607476704, accessibilitiesMap.get(tuple).get(TransportMode.car), MatsimTestUtils.EPSILON);
} else if (tuple.getFirst().getCoord().getY() == 150.) {
// Assertions.assertEquals(0.2758252376673665, accessibilitiesMap.get(tuple).get("freespeed"), MatsimTestUtils.EPSILON);
// Assertions.assertEquals(0.27582980607476704, accessibilitiesMap.get(tuple).get(TransportMode.car), MatsimTestUtils.EPSILON);
}
}
}
}
}
}

0 comments on commit 1f12853

Please sign in to comment.