/// <summary>
/// Sorts the stops and updates the connections.
/// </summary>
public static void SortStops(this TransitDb db)
{
if (db.HasTransfersDb())
{
throw new ArgumentException("Cannot sort stops for a transit db with tranfers db's. Add tranfer db's after sorting.");
}
// add default stop ids.
var stopIds = new MemoryArray <uint>(db.StopsCount);
for (uint s = 0; s < db.StopsCount; s++)
{
stopIds[s] = s;
}
// sort stops.
db.SortStops((s1, s2) =>
{
var temp = stopIds[s1];
stopIds[s1] = stopIds[s2];
stopIds[s2] = temp;
});
var reverseStopIds = new MemoryArray <uint>(db.StopsCount);
for (uint s = 0; s < db.StopsCount; s++)
{
reverseStopIds[stopIds[s]] = s;
}
stopIds = null;
// update stops on connections.
var connectionEnumerator = db.GetConnectionsEnumerator(DefaultSorting.DepartureTime);
while (connectionEnumerator.MoveNext())
{
db.ConnectionsDb.UpdateStops(connectionEnumerator.Id,
reverseStopIds[connectionEnumerator.DepartureStop],
reverseStopIds[connectionEnumerator.ArrivalStop]);
}
// update stops on shapes.
var shapesEnumerator = db.ShapesDb.GetEnumerator();
while (shapesEnumerator.MoveNext())
{
db.ShapesDb.UpdateStops(shapesEnumerator,
reverseStopIds[shapesEnumerator.Stop1],
reverseStopIds[shapesEnumerator.Stop2]);
}
db.ShapesDb.Sort(); // sort again, stops have changed!
}
/// <summary>
/// Copies all core data stops, schedules, trips, and connections from the given transit db.
/// </summary>
public static void CopyFrom(this TransitDb db, TransitDb other)
{
if (db == null)
{
throw new ArgumentNullException("db");
}
if (other == null)
{
throw new ArgumentNullException("other");
}
if (other.ConnectionSorting == null)
{
throw new ArgumentException("A database can only be copied if connections are sorted.");
}
var stopIdsBlockSize = 1024 * 32;
var stopIds = new MemoryArray <uint>(stopIdsBlockSize);
var tripIdsBlockSize = 1024 * 32;
var tripIds = new MemoryArray <uint>(tripIdsBlockSize);
var scheduleIds = new Dictionary <uint, uint>();
var agencyIds = new Dictionary <uint, uint>();
// copy stops and keep id transformations.
var stopsEnumerator = other.GetStopsEnumerator();
while (stopsEnumerator.MoveNext())
{
var stopsMeta = other.StopAttributes.Get(stopsEnumerator.MetaId);
var newMetaId = db.StopAttributes.Add(stopsMeta);
var newStopId = db.AddStop(stopsEnumerator.Latitude, stopsEnumerator.Longitude, newMetaId);
if (stopsEnumerator.Id >= stopIds.Length)
{
stopIds.Resize(stopIds.Length + stopIdsBlockSize);
}
stopIds[stopsEnumerator.Id] = newStopId;
}
// copy trips, copy schedules that have not been copied yet, and keep trip id transformations.
var tripsEnumerator = other.GetTripsEnumerator();
var scheduleEnumerator = other.GetSchedulesEnumerator();
while (tripsEnumerator.MoveNext())
{
var tripsMeta = other.TripAttributes.Get(tripsEnumerator.MetaId);
var newMetaId = db.TripAttributes.Add(tripsMeta);
uint newAgencyMetaId = uint.MaxValue;
if (!agencyIds.TryGetValue(tripsEnumerator.AgencyId, out newAgencyMetaId))
{
var agencyMeta = other.AgencyAttributes.Get(tripsEnumerator.AgencyId);
newAgencyMetaId = db.AgencyAttributes.Add(agencyMeta);
agencyIds.Add(tripsEnumerator.AgencyId, newAgencyMetaId);
}
uint newScheduleId = uint.MaxValue;
if (!scheduleIds.TryGetValue(tripsEnumerator.ScheduleId, out newScheduleId))
{
if (scheduleEnumerator.MoveTo(tripsEnumerator.ScheduleId))
{
newScheduleId = scheduleEnumerator.CopyTo(db.SchedulesDb);
scheduleIds[tripsEnumerator.ScheduleId] = newScheduleId;
}
}
var newTripId = db.AddTrip(newScheduleId, newAgencyMetaId, newMetaId);
if (tripsEnumerator.Id >= tripIds.Length)
{
tripIds.Resize(tripIds.Length + tripIdsBlockSize);
}
tripIds[tripsEnumerator.Id] = newTripId;
}
// copy connections.
var connectionEnumerator = other.GetConnectionsEnumerator(other.ConnectionSorting.Value);
while (connectionEnumerator.MoveNext())
{
var newArrivalStop = stopIds[connectionEnumerator.ArrivalStop];
var newDepartureStop = stopIds[connectionEnumerator.DepartureStop];
var newTripId = tripIds[connectionEnumerator.TripId];
db.AddConnection(newDepartureStop, newArrivalStop, newTripId, connectionEnumerator.DepartureTime,
connectionEnumerator.ArrivalTime);
}
// copy shapes.
if (other.ShapesDb != null)
{
var shapesEnumerator = other.ShapesDb.GetEnumerator();
while (shapesEnumerator.MoveNext())
{
var stop1 = stopIds[shapesEnumerator.Stop1];
var stop2 = stopIds[shapesEnumerator.Stop2];
db.ShapesDb.Add(stop1, stop2, shapesEnumerator.Shape);
}
}
}
