Whitelist - gather coordinates for one to many routing

src/lib/compositionTypes.ts

@@ -2,6 +2,27 @@ import type { Capacities } from './capacities';
 import type { Interval } from './interval';
 import type { Coordinates } from './location';
 
+export type Company = {
+	id: number;
+	coordinates: Coordinates;
+	vehicles: Vehicle[];
+	zoneId: number;
+};
+
+export type Vehicle = {
+	id: number;
+	capacities: Capacities;
+	tours: Tour[];
+	availabilities: Interval[];
+};
+
+export type Tour = {
+	departure: Date;
+	arrival: Date;
+	id: number;
+	events: Event[];
+};
+
 export type Event = {
 	capacities: Capacities;
 	is_pickup: boolean;

src/routes/api/whitelist/gatherCoordinates.test.ts

@@ -0,0 +1,113 @@
+import { describe, it, expect } from 'vitest';
+import { type Range } from './capacitySimulation';
+import { Coordinates } from '$lib/location';
+import type { Vehicle, Company, Tour, Event } from '$lib/compositionTypes';
+import { Interval } from '$lib/interval';
+import { gatherRoutingCoordinates } from './routing';
+
+const createCompany = (vehicles: Vehicle[], coordinates: Coordinates): Company => {
+	return {
+		id: 1,
+		coordinates: coordinates,
+		vehicles,
+		zoneId: 1
+	};
+};
+
+const createVehicle = (id: number, tours: Tour[]): Vehicle => {
+	return {
+		id,
+		capacities: { passengers: 0, bikes: 0, wheelchairs: 0, luggage: 0 },
+		tours,
+		availabilities: []
+	};
+};
+
+const createTour = (events: Event[]): Tour => {
+	return {
+		departure: new Date(),
+		arrival: new Date(),
+		id: 1,
+		events
+	};
+};
+
+const createEvent = (coordinates: Coordinates): Event => {
+	return {
+		capacities: { passengers: 0, bikes: 0, wheelchairs: 0, luggage: 0 },
+		is_pickup: true,
+		time: new Interval(new Date(), new Date()),
+		id: 1,
+		coordinates,
+		tourId: 1
+	};
+};
+
+const createBusStop = () => {
+	return { coordinates: new Coordinates(1, 1), times: [] };
+};
+
+describe('gather coordinates test', () => {
+	it('TODO', () => {
+		let eventLatLng = 100;
+		let companyLatLng = 5;
+		const companies = [
+			createCompany(
+				[
+					createVehicle(1, [
+						createTour([
+							createEvent(new Coordinates(eventLatLng, eventLatLng++)),
+							createEvent(new Coordinates(eventLatLng, eventLatLng++))
+						])
+					])
+				],
+				new Coordinates(companyLatLng, companyLatLng++)
+			),
+			createCompany(
+				[
+					createVehicle(2, [
+						createTour([
+							createEvent(new Coordinates(eventLatLng, eventLatLng++)),
+							createEvent(new Coordinates(eventLatLng, eventLatLng++))
+						])
+					])
+				],
+				new Coordinates(companyLatLng, companyLatLng++)
+			)
+		];
+		const busStops = [createBusStop(), createBusStop()];
+		const insertions = new Map<number, Range[]>();
+		insertions.set(1, [{ earliestPickup: 0, latestDropoff: 2 }]);
+		insertions.set(2, [{ earliestPickup: 0, latestDropoff: 2 }]);
+		const coordinates = gatherRoutingCoordinates(companies, busStops, insertions);
+		expect(coordinates.busStopMany).toHaveLength(2);
+		expect(coordinates.userChosenMany).toHaveLength(6);
+		busStops.forEach((_, idx) => {
+			expect(coordinates.busStopMany[idx]).toHaveLength(6);
+		});
+		// Company coordinates
+		expect(coordinates.userChosenMany[0].lat).toBe(5);
+		expect(coordinates.busStopMany[0][0].lat).toBe(5);
+		expect(coordinates.busStopMany[1][0].lat).toBe(5);
+		expect(coordinates.userChosenMany[1].lat).toBe(6);
+		expect(coordinates.busStopMany[0][1].lat).toBe(6);
+		expect(coordinates.busStopMany[1][1].lat).toBe(6);
+
+		// Event coordinates
+		expect(coordinates.userChosenMany[2].lat).toBe(100);
+		expect(coordinates.busStopMany[0][2].lat).toBe(100);
+		expect(coordinates.busStopMany[1][2].lat).toBe(100);
+
+		expect(coordinates.userChosenMany[3].lat).toBe(101);
+		expect(coordinates.busStopMany[0][3].lat).toBe(101);
+		expect(coordinates.busStopMany[1][3].lat).toBe(101);
+
+		expect(coordinates.userChosenMany[4].lat).toBe(102);
+		expect(coordinates.busStopMany[0][4].lat).toBe(102);
+		expect(coordinates.busStopMany[1][4].lat).toBe(102);
+
+		expect(coordinates.userChosenMany[5].lat).toBe(103);
+		expect(coordinates.busStopMany[0][5].lat).toBe(103);
+		expect(coordinates.busStopMany[1][5].lat).toBe(103);
+	});
+});

src/routes/api/whitelist/routing.ts

@@ -0,0 +1,68 @@
+import type { BusStop } from '$lib/busStop';
+import type { Company, Event } from '$lib/compositionTypes';
+import { Coordinates } from '$lib/location';
+import type { Range } from './capacitySimulation';
+
+type RoutingCoordinates = {
+	busStopMany: Coordinates[][];
+	userChosenMany: Coordinates[];
+};
+
+function iterateAllInsertions(
+	companies: Company[],
+	insertions: Map<number, Range[]>,
+	insertionFn: (events: Event[], insertionIdx: number, companyPos: number, eventPos: number) => void
+) {
+	let companyPos = 0;
+	let eventPos = companies.length;
+	companies.forEach((company) => {
+		company.vehicles.forEach((vehicle) => {
+			const events = vehicle.tours.flatMap((t) => t.events);
+			insertions.get(vehicle.id)!.forEach((insertion) => {
+				for (
+					let insertionIdx = insertion.earliestPickup;
+					insertionIdx != insertion.latestDropoff;
+					++insertionIdx
+				) {
+					insertionFn(events, insertionIdx, companyPos, eventPos++);
+				}
+			});
+		});
+		companyPos++;
+	});
+}
+
+export function gatherRoutingCoordinates(
+	companies: Company[],
+	busStops: BusStop[],
+	insertionsByVehicle: Map<number, Range[]>
+): RoutingCoordinates {
+	const eventInsertionCount = companies
+		.flatMap((c) => c.vehicles)
+		.flatMap((v) => insertionsByVehicle.get(v.id)!)
+		.reduce((sum, current) => sum + current.latestDropoff - current.earliestPickup, 0);
+	const insertionCount = companies.length + eventInsertionCount;
+	const busStopMany = new Array<Coordinates[]>(busStops.length);
+	for (let i = 0; i != busStopMany.length; ++i) {
+		busStopMany[i] = new Array<Coordinates>(insertionCount);
+	}
+	const userChosenMany = new Array<Coordinates>(insertionCount);
+	companies.forEach((company, companyPos) => {
+		for (let busStopIdx = 0; busStopIdx != busStops.length; ++busStopIdx) {
+			busStopMany[busStopIdx][companyPos] = company.coordinates;
+		}
+		userChosenMany[companyPos] = company.coordinates;
+	});
+	iterateAllInsertions(companies, insertionsByVehicle, (events, insertionIdx, _, eventPos) => {
+		const eventCoordinates = events[insertionIdx].coordinates;
+		for (let busStopIdx = 0; busStopIdx != busStops.length; ++busStopIdx) {
+			busStopMany[busStopIdx][eventPos] = eventCoordinates;
+		}
+		userChosenMany[eventPos] = eventCoordinates;
+	});
+	console.log(userChosenMany);
+	return {
+		busStopMany,
+		userChosenMany
+	};
+}