@@ -8,15 +8,24 @@
88
99import Foundation
1010
11+typealias Lines = [String]
12+typealias Element = [String]
13+let commentChars = CharacterSet.init(charactersIn: "#/")
14+let styleCharecters = CharacterSet.init(charactersIn: "_*")
15+
16+
1117enum BracketStyle {
1218 case square
1319 case round
20+ case line
1421 var opening: Character {
1522 switch self {
1623 case .round:
1724 return "("
1825 case .square:
1926 return "["
27+ case .line:
28+ return "|"
2029 }
2130 }
2231 var closing: Character {
@@ -25,6 +34,8 @@ enum BracketStyle {
2534 return ")"
2635 case .square:
2736 return "]"
37+ case .line:
38+ return "|"
2839 }
2940 }
3041}
@@ -56,10 +67,12 @@ default:
5667 if let surl = args.popFirst() {
5768 outputUrl = URL(fileURLWithPath: surl)
5869 }
59− case "-s":
70+ case "-b:s":
6071 style = .square
61− case "-r":
72+ case "-b:r":
6273 style = .round
74+ case "-b:l":
75+ style = .line
6376 default:
6477 return
6578 }
@@ -74,7 +87,6 @@ enum ElementStyle: String {
7487 case italic = "italic"
7588 case boldItalic = "bold-italic"
7689}
77−let styleCharecters = CharacterSet.init(charactersIn: "_*")
7890extension String.SubSequence {
7991 var style: ElementStyle {
8092 let num = self.prefix(while: {styleCharecters.contains($0.unicodeScalars.first!)}).count
@@ -93,38 +105,209 @@ extension String.SubSequence {
93105 return self.trimmingCharacters(in: styleCharecters)
94106 }
95107}
96−func process(mth: String) -> String {
97− let whitespace = CharacterSet.whitespaces
108+
109+func process(element: Element) -> String {
110+ var out = ""
98111
99− var out = "<math>"
100− let lines = mth.split(separator: "\n", omittingEmptySubsequences: false)
101− .map {$0.trimmingCharacters(in: .whitespaces)}
102− var insideMatrix = false
103− for line in lines {
104− switch line {
105− case "":
106− if insideMatrix {
107− out.append("</mtable><mo>\(style.closing)</mo></mrow>")
108− insideMatrix = false
109− }
112+ var inMatrix = false
113+ var first = false
114+ var seperated = false
115+ var afterSeparator: [Substring] = []
116+
117+ var small = false
118+
119+ let arrows = CharacterSet.init(charactersIn: "⟺⟷⟼⟻⟶⟹⟵⟸")
120+
121+ let nEl = element.map { (el) -> String in
122+ switch el {
123+ case let l where l.last == ">" && l.first == "<" && l.dropLast().dropFirst().filter({$0 == "="}).count > 0:
124+ small = true
125+ return "⟺"
126+ case let l where l.last == ">" && l.first == "<" && l.dropLast().dropFirst().filter({$0 == "-"}).count > 0:
127+ small = true
128+ return "⟷"
129+ case let l where l.last == ">" && l.first == "|" && l.dropLast().dropFirst().filter({$0 == "-"}).count > 0:
130+ small = true
131+ return "⟼"
132+ case let l where l.last == "|" && l.first == "<" && l.dropLast().dropFirst().filter({$0 == "-"}).count > 0:
133+ small = true
134+ return "⟻"
135+ case let l where l.last == ">" && l.dropLast().filter({$0 == "-"}).count > 0:
136+ small = true
137+ return "⟶"
138+ case let l where l.last == ">" && l.dropLast().filter({$0 == "="}).count > 0:
139+ small = true
140+ return "⟹"
141+ case let l where l.first == "<" && l.dropFirst().filter({$0 == "-"}).count > 0:
142+ small = true
143+ return "⟵"
144+ case let l where l.first == "<" && l.dropFirst().filter({$0 == "="}).count > 0:
145+ small = true
146+ return "⟸"
110147 default:
111− if !insideMatrix {
112− out.append("<mrow><mo>\(style.opening)</mo><mtable>")
113− insideMatrix = true
148+ return el
149+ }
150+ }
151+
152+ func process(lineElement: Substring, withAttributes attributes: String = "", style: ElementStyle = .none) -> Substring {
153+ func extractNextElement(from line: Substring) -> (element: Substring, left: Substring) {
154+ if line.first != "(" {
155+ var i = line.startIndex
156+ var el: Substring = ""
157+ while i < line.endIndex && CharacterSet.alphanumerics.contains(line[i].unicodeScalars.first!) {
158+ el.append(line[i])
159+ i = line.index(after: i)
160+ }
161+ return (el, line[i...])
162+ } else {
163+ var brackets = 0
164+ var i = line.startIndex
165+ var el: Substring = ""
166+ while i < line.endIndex {
167+ switch el[i] {
168+ case "(":
169+ brackets += 1
170+ case ")":
171+ brackets -= 1
172+ default:
173+ break
174+ }
175+ i = line.index(after: i)
176+ if brackets == 0 {
177+ break
178+ }
179+ el.append(el[i])
180+ }
181+ if brackets != 0 {
182+ print("Mismatched brackets")
183+ exit(0)
184+ }
185+ return (el.dropFirst(), line[i...])
114186 }
115− out.append("<mtr>")
116− for element in line.split(
117− whereSeparator: {c in whitespace.contains(c.unicodeScalars.first!)}) {
187+ }
188+ var i = lineElement.startIndex
189+ while i < lineElement.endIndex {
190+ switch lineElement[i] {
191+ case "^":
192+ let parts = extractNextElement(from: lineElement[lineElement.index(after: i)...])
193+ return "<mo mathvariant=\"\(style.rawValue)\" \(attributes)>\(lineElement[..<i].filter({$0 != "\\"}))</mo>" + "<msup><mo></mo><mn>\(parts.element)</mn></msup>" + process(lineElement: parts.left, withAttributes: attributes, style: style)
194+ case "!":
195+ let parts = extractNextElement(from: lineElement[lineElement.index(after: i)...])
196+ return "<mo mathvariant=\"\(style.rawValue)\" \(attributes)>\(lineElement[..<i].filter({$0 != "\\"}))</mo>" + "<mover accent=\"true\"><mrow><mo>\(parts.element)</mo></mrow><mo>_</mo></mover>" + process(lineElement: parts.left, withAttributes: attributes, style: style)
197+ case "\\":
198+ i = lineElement.index(after: i)
199+ case "/":
200+ let parts = extractNextElement(from: lineElement[lineElement.index(after: i)...])
201+ return "<mo mathvariant=\"\(style.rawValue)\" \(attributes)>\(lineElement[..<i].filter({$0 != "\\"}))</mo>" + "<msub><mo></mo><mn>\(parts.element)</mn></msub>" + process(lineElement: parts.left, withAttributes: attributes, style: style)
202+ default:
203+ break
204+ }
205+ i = lineElement.index(after: i)
206+ }
207+ return Substring("<mo mathvariant=\"\(style.rawValue)\" \(attributes)>\(lineElement.filter({$0 != "\\"}))</mo>")
208+ }
209+ func process(line: Substring) {
210+ out.append("<mtr>")
211+ for element in line.split(
212+ whereSeparator: {c in CharacterSet.whitespaces.contains(c.unicodeScalars.first!)}) {
213+ switch element {
214+ case "_":
215+ out.append("<mtd><mspace height=\"0.8em\"/></mtd>")
216+ default:
118217 let eStyle = element.style
119− out.append("<mtd><mn mathvariant=\"\(eStyle.rawValue)\">\(element.trimmingStyleCharecters())</mn></mtd>")
218+ out.append("<mtd>\(process(lineElement: Substring(element.trimmingStyleCharecters()), withAttributes: "height=\"0.8em\"", style: eStyle))</mtd>")
219+ }
220+ }
221+ out.append("</mtr>")
222+ }
223+ func process(comment: String) {
224+ if comment.hasPrefix("#") {
225+ out.append("<!-- \(comment.trimmingCharacters(in: commentChars).trimmingCharacters(in: .whitespaces)) -->")
226+ }
227+ }
228+
229+ for line in nEl {
230+ if inMatrix {
231+ if first && line.contains("|") {
232+ seperated = true
233+ }
234+ switch line {
235+ case "]":
236+ if seperated {
237+ out.append("</mtable><mo stretchy=\"true\">|</mo><mtable>")
238+ for line in afterSeparator {
239+ process(line: line)
240+ }
241+ }
242+ out.append("</mtable><mo stretchy=\"true\">\(style.closing)</mo></mrow></mtd></mtr>")
243+ inMatrix = false
244+ seperated = false
245+ afterSeparator = []
246+ default:
247+ if seperated {
248+ let split = line.split(separator: "|", maxSplits: 1, omittingEmptySubsequences: false)
249+ if split.count != 2 {
250+ print("Inconsistent syntax. The `|` seperator isn't consistent within a matrix.")
251+ exit(0)
252+ }
253+ process(line: split[0])
254+ afterSeparator.append(split[1])
255+ } else {
256+ process(line: Substring(line))
257+ }
258+ }
259+ } else {
260+ switch line {
261+ case let l where l.hasPrefix("//"):
262+ break
263+ case let l where l.hasPrefix("#"):
264+ out.append("</mtable></math>\n\n")
265+ process(comment: l)
266+ out.append("\n\n<math><mtable>")
267+ break // TODO: Figure out what to do
268+ case "[":
269+ out.append("<mtr><mtd><mrow><mo stretchy=\"true\">\(style.opening)</mo><mtable>")
270+ inMatrix = true
271+ first = true
272+ continue
273+ case "":
274+ if !first {
275+ out.append("</mtable><mtable>")
276+ }
277+ first = true
278+ continue
279+ case let l where arrows.contains(l.first!.unicodeScalars.first!):
280+ out.append("<mtr><mtd><mo>\(line)</mo></mtd></mtr>")
281+ default:
282+ out.append("<mtr><mtd>\(small ? "<mstyle scriptlevel=\"1\">" : "")\(process(lineElement: Substring(line)))\(small ? "</mstyle>" : "")</mtd></mtr>")
120283 }
121− out.append("</mtr>")
122284 }
285+ first = false
123286 }
124− if insideMatrix {
125− out.append("</mtable><mo>\(style.closing)</mo></mrow>")
287+ return out
288+}
289+
290+func process(mth: String) -> String {
291+ let elements = mth
292+ .split(separator: "\n", omittingEmptySubsequences: false)
293+ .map({$0.trimmingCharacters(in: .whitespaces)})
294+ .map({ l in
295+ var i = l.startIndex
296+ for u in 0..<max(0, l.count - 1) {
297+ if l[i] == "/" && l[l.index(after: i)] == "/" {
298+ return String(l.dropLast(l.count - u))
299+ }
300+ i = l.index(after: i)
301+ }
302+ return l
303+ })
304+ .split(separator: "")
305+ var out = "<math><mtable>"
306+ for el in elements {
307+ out.append(process(element: Array(el)))
308+ out.append("</mtable><mtable>")
126309 }
127− out.append("</math>")
310+ out.append("</mtable></math>")
128311 return out
129312}
130313