| | | 1 | | using System; |
| | | 2 | | using System.Collections.Generic; |
| | | 3 | | using System.Linq; |
| | | 4 | | using Itinero.IO.Osm.Streams; |
| | | 5 | | using OsmSharp; |
| | | 6 | | using OsmSharp.Complete; |
| | | 7 | | using OsmSharp.Streams; |
| | | 8 | | using OsmSharp.Tags; |
| | | 9 | | |
| | | 10 | | namespace Itinero.IO.Osm; |
| | | 11 | | |
| | | 12 | | /// <summary> |
| | | 13 | | /// Contains extensions method for the router db. |
| | | 14 | | /// </summary> |
| | | 15 | | public static class RouterDbExtensions |
| | | 16 | | { |
| | | 17 | | /// <summary> |
| | | 18 | | /// Loads the given OSM data into the router db. |
| | | 19 | | /// </summary> |
| | | 20 | | /// <param name="routerDb">The router db.</param> |
| | | 21 | | /// <param name="data">The data.</param> |
| | | 22 | | /// <param name="configure">The configure function.</param> |
| | | 23 | | public static void UseOsmData(this RouterDb routerDb, OsmStreamSource data, |
| | | 24 | | Action<DataProviderSettings>? configure = null) |
| | 3 | 25 | | { |
| | | 26 | | // get writer. |
| | 3 | 27 | | if (routerDb.HasMutableNetwork) |
| | 0 | 28 | | { |
| | 0 | 29 | | throw new InvalidOperationException( |
| | 0 | 30 | | $"Cannot add data to a {nameof(RouterDb)} that is only being written to."); |
| | | 31 | | } |
| | | 32 | | |
| | 3 | 33 | | using var routerDbWriter = routerDb.GetMutableNetwork(); |
| | | 34 | | |
| | | 35 | | // create settings. |
| | 3 | 36 | | var settings = new DataProviderSettings(); |
| | 3 | 37 | | configure?.Invoke(settings); |
| | | 38 | | |
| | | 39 | | // do filtering. |
| | | 40 | | |
| | | 41 | | // 1: complete objects. |
| | 3 | 42 | | if (settings.TagsFilter.CompleteFilter != null) |
| | 0 | 43 | | { |
| | 0 | 44 | | data = data.ApplyCompleteFilter(settings.TagsFilter.CompleteFilter); |
| | 0 | 45 | | } |
| | | 46 | | |
| | | 47 | | // 2: filter relations. |
| | 3 | 48 | | if (settings.TagsFilter.MemberFilter != null) |
| | 0 | 49 | | { |
| | 0 | 50 | | data = data.ApplyRelationMemberFilter(settings.TagsFilter.MemberFilter); |
| | 0 | 51 | | } |
| | | 52 | | |
| | | 53 | | // 3: filter tags on ways and relations. |
| | 3 | 54 | | if (settings.TagsFilter.Filter != null) |
| | 2 | 55 | | { |
| | 2 | 56 | | data = data.ApplyFilter(settings.TagsFilter.Filter); |
| | 2 | 57 | | } |
| | | 58 | | |
| | | 59 | | // use writer to fill router db. |
| | 3 | 60 | | var routerDbStreamTarget = new RouterDbStreamTarget(routerDbWriter, settings.ElevationHandler); |
| | 3 | 61 | | routerDbStreamTarget.RegisterSource(data); |
| | 3 | 62 | | routerDbStreamTarget.Initialize(); |
| | 3 | 63 | | routerDbStreamTarget.Pull(); |
| | 6 | 64 | | } |
| | | 65 | | } |