1 files changed, 52 insertions, 16 deletions
diff --git a/src/main/kotlin/moe/nea/ledger/NumberUtil.kt b/src/main/kotlin/moe/nea/ledger/NumberUtil.kt
index d26b047..b0e47db 100644
--- a/src/main/kotlin/moe/nea/ledger/NumberUtil.kt
+++ b/src/main/kotlin/moe/nea/ledger/NumberUtil.kt
@@ -6,28 +6,64 @@ import java.util.regex.Pattern
 // language=regexp
 val SHORT_NUMBER_PATTERN = "[0-9]+(?:,[0-9]+)*(?:\\.[0-9]+)?[kKmMbB]?"
 
+// language=regexp
+val ROMAN_NUMBER_PATTERN = "[IVXLCDM]+"
+
+val romanNumbers = mapOf(
+	'I' to 1,
+	'V' to 5,
+	'X' to 10,
+	'L' to 50,
+	'C' to 100,
+	'D' to 500,
+	'M' to 1000
+)
+
+fun parseRomanNumber(string: String): Int {
+	var smallestSeenSoFar = Int.MAX_VALUE
+	var lastSeenOfSmallest = 0
+	var amount = 0
+	for (c in string) {
+		val cV = romanNumbers[c]!!
+		if (cV == smallestSeenSoFar) {
+			lastSeenOfSmallest++
+			amount += cV
+		} else if (cV < smallestSeenSoFar) {
+			smallestSeenSoFar = cV
+			amount += cV
+			lastSeenOfSmallest = 1
+		} else {
+			amount -= lastSeenOfSmallest * smallestSeenSoFar * 2
+			smallestSeenSoFar = cV
+			amount += cV
+			lastSeenOfSmallest = 1
+		}
+	}
+	return amount
+}
+
 val siScalars = mapOf(
-    'k' to 1_000.0,
-    'K' to 1_000.0,
-    'm' to 1_000_000.0,
-    'M' to 1_000_000.0,
-    'b' to 1_000_000_000.0,
-    'B' to 1_000_000_000.0,
+	'k' to 1_000.0,
+	'K' to 1_000.0,
+	'm' to 1_000_000.0,
+	'M' to 1_000_000.0,
+	'b' to 1_000_000_000.0,
+	'B' to 1_000_000_000.0,
 )
 
 fun parseShortNumber(string: String): Double {
-    var k = string.replace(",", "")
-    val scalar = k.last()
-    var scalarMultiplier = siScalars[scalar]
-    if (scalarMultiplier == null) {
-        scalarMultiplier = 1.0
-    } else {
-        k = k.dropLast(1)
-    }
-    return k.toDouble() * scalarMultiplier
+	var k = string.replace(",", "")
+	val scalar = k.last()
+	var scalarMultiplier = siScalars[scalar]
+	if (scalarMultiplier == null) {
+		scalarMultiplier = 1.0
+	} else {
+		k = k.dropLast(1)
+	}
+	return k.toDouble() * scalarMultiplier
 }
 
 inline fun <T> Pattern.useMatcher(string: String, block: Matcher.() -> T): T? =
-    matcher(string).takeIf { it.matches() }?.let(block)
+	matcher(string).takeIf { it.matches() }?.let(block)
 
 fun String.unformattedString(): String = replace("§.".toRegex(), "")
 \ No newline at end of file