floraachy
/
vChewing-macOS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

UOM // Use class in lieu of struct in certain types.

This commit is contained in:
ShikiSuen 2022-06-14 18:38:51 +08:00
parent 4f205f35f0
commit d768c12d5a
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Source/Modules/LangModelRelated/SubLMs/lmUserOverride.swift
View File

@ -28,28 +28,35 @@ import Foundation
extension vChewing { extension vChewing {
public class LMUserOverride { public class LMUserOverride {
// MARK: - Private Structures // MARK: - Private Structures
// 這些型別必須得用 class，不然會導致拿不到有效建議。
struct Override { class Override {
var count: Int = 0 var count: Int = 0
var timestamp: Double = 0.0 var timestamp: Double = 0.0
} }
struct Observation { class Observation {
var count: Int = 0 var count: Int = 0
var overrides: [String: Override] = [:] var overrides: [String: Override] = [:]
mutating func update(candidate: String, timestamp: Double) { func update(candidate: String, timestamp: Double) {
count += 1 count += 1
if var neta = overrides[candidate] { if let neta = overrides[candidate] {
neta.timestamp = timestamp neta.timestamp = timestamp
neta.count += 1 neta.count += 1
overrides[candidate] = neta
} }
} }
} }
struct KeyObservationPair { class KeyObservationPair {
var key: String var key: String
var observation: Observation var observation: Observation
init(key: String, observation: Observation) {
self.key = key
self.observation = observation
}
} }
// MARK: - Main // MARK: - Main
@ -74,7 +81,7 @@ extension vChewing {
let key = convertKeyFrom(walkedNodes: walkedNodes, cursorIndex: cursorIndex) let key = convertKeyFrom(walkedNodes: walkedNodes, cursorIndex: cursorIndex)
guard mutLRUMap[key] != nil else { guard mutLRUMap[key] != nil else {
var observation: Observation = .init() let observation: Observation = .init()
observation.update(candidate: candidate, timestamp: timestamp) observation.update(candidate: candidate, timestamp: timestamp)
let koPair = KeyObservationPair(key: key, observation: observation) let koPair = KeyObservationPair(key: key, observation: observation)
mutLRUMap[key] = koPair mutLRUMap[key] = koPair
@ -87,7 +94,7 @@ extension vChewing {
IME.prtDebugIntel("UOM: Observation finished with new observation: \(key)") IME.prtDebugIntel("UOM: Observation finished with new observation: \(key)")
return return
} }
if var theNeta = mutLRUMap[key] { if let theNeta = mutLRUMap[key] {
theNeta.observation.update(candidate: candidate, timestamp: timestamp) theNeta.observation.update(candidate: candidate, timestamp: timestamp)
mutLRUList.insert(theNeta, at: 0) mutLRUList.insert(theNeta, at: 0)
mutLRUMap[key] = theNeta mutLRUMap[key] = theNeta