]> git.r.bdr.sh - rbdr/map/blob - Map/Logic/MapParser/Strategies/EdgeParserStrategy.swift
f8b78df1888d5e558c4220844b9a036c45abac0f
[rbdr/map] / Map / Logic / MapParser / Strategies / EdgeParserStrategy.swift
1 /*
2 Copyright (C) 2024 Rubén Beltrán del Río
3
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see https://map.tranquil.systems.
16 */
17 import Foundation
18
19 struct EdgeParserStrategy: MapParserStrategy {
20 private let regex = MapParsingPatterns.edge
21
22 func canHandle(line: String) -> Bool {
23 let range = NSRange(location: 0, length: line.utf16.count)
24 let matches = regex.matches(in: String(line), options: [], range: range)
25 return matches.count > 0 && matches[0].numberOfRanges == 4
26 }
27
28 func handle(index: Int, line: String, vertices: [String: Vertex]) -> (Any.Type, Any) {
29 let range = NSRange(location: 0, length: line.utf16.count)
30 let matches = regex.matches(in: String(line), options: [], range: range)
31
32 let match = matches[0]
33 let arrowhead = String(line[Range(match.range(at: 2), in: line)!]) == ">"
34 let vertexA = String(line[Range(match.range(at: 1), in: line)!])
35 let vertexB = String(line[Range(match.range(at: 3), in: line)!])
36
37 if let origin = vertices[vertexA] {
38 if let destination = vertices[vertexB] {
39 let edge = MapEdge(
40 id: index, origin: origin.position, destination: destination.position,
41 arrowhead: arrowhead)
42 return (MapEdge.self, edge)
43 }
44 }
45
46 return (NSObject.self, NSObject()) // No matching object
47 }
48 }