blob: f2863d0c2de134ab7797dc5a2b64c8f38e9f17ac (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
package com.rockymadden.stringmetric
object Algorithm {
import scala.collection.immutable.Map
import Transform._
trait Algorithm[A] {
def compute(a: A): Option[A]
}
object Algorithm {
implicit def toStringAlgorithmDecorator(sa: StringAlgorithm): StringAlgorithmDecorator =
new StringAlgorithmDecorator(sa)
}
trait StringAlgorithm extends Algorithm[Array[Char]] {
def compute(a: String): Option[String]
}
object StringAlgorithm {
val Metaphone = phonetic.MetaphoneAlgorithm
val Nysiis = phonetic.NysiisAlgorithm
val RefinedNysiis = phonetic.RefinedNysiisAlgorithm
val RefinedSoundex = phonetic.RefinedSoundexAlgorithm
val Soundex = phonetic.SoundexAlgorithm
def computeWithMetaphone(a: Array[Char]) = Metaphone.compute(a)
def computeWithNysiis(a: Array[Char]) = Nysiis.compute(a)
def computeWithRefinedNysiis(a: Array[Char]) = RefinedNysiis.compute(a)
def computeWithRefinedSoundex(a: Array[Char]) = RefinedSoundex.compute(a)
def computeWithSoundex(a: Array[Char]) = Soundex.compute(a)
}
sealed trait AlgorithmDecorator[A] {
val withMemoization: Algorithm[A]
val withTransform: (Transform[A] => Algorithm[A])
}
final case class StringAlgorithmDecorator(sa: StringAlgorithm) extends AlgorithmDecorator[Array[Char]] {
override val withMemoization: StringAlgorithm = new StringAlgorithm {
private val base: StringAlgorithm = sa
private var memo: Map[String, Option[String]] = Map()
override def compute(a: Array[Char]): Option[Array[Char]] =
compute(a.toString).map(_.toCharArray)
override def compute(a: String): Option[String] =
if (memo.contains(a)) memo(a)
else {
memo = memo + (a -> base.compute(a))
memo(a)
}
}
override val withTransform: (StringTransform => StringAlgorithm) = (st) => new StringAlgorithm {
private val base: StringAlgorithm = sa
private val transform: StringTransform = st
override def compute(a: Array[Char]): Option[Array[Char]] = base.compute(transform(a))
override def compute(a: String): Option[String] = compute(a.toCharArray).map(_.mkString)
}
}
}
|
