53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
type Edge struct {
|
||
|
|
name string
|
||
|
|
weight float64
|
||
|
|
target *Node
|
||
|
|
}
|
||
|
|
|
||
|
|
type Node struct {
|
||
|
|
edges map[string]Edge
|
||
|
|
name string
|
||
|
|
start int
|
||
|
|
end int
|
||
|
|
weight float64
|
||
|
|
scentenceId uint32
|
||
|
|
}
|
||
|
|
|
||
|
|
func makeNode(name string, scentenceId uint32) *Node {
|
||
|
|
node := &Node{name: name, edges: make(map[string]Edge), start: 0, end: 0, weight: 0, scentenceId: scentenceId}
|
||
|
|
return node
|
||
|
|
}
|
||
|
|
|
||
|
|
type Thesaurus struct {
|
||
|
|
pntrmap map[string]*Node
|
||
|
|
}
|
||
|
|
|
||
|
|
func (thes *Thesaurus) addEntry(start string, targets []string, scentenceId uint32) {
|
||
|
|
_, exists := thes.pntrmap[start]
|
||
|
|
if !exists {
|
||
|
|
thes.pntrmap[start] = makeNode(start, scentenceId)
|
||
|
|
}
|
||
|
|
thes.pntrmap[start].weight += 1
|
||
|
|
thes.addEdges(start, targets, scentenceId)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (thes *Thesaurus) addEdges(start string, targets []string, scentenceId uint32) {
|
||
|
|
val := thes.pntrmap[start]
|
||
|
|
for _, s := range targets {
|
||
|
|
edgeVal, edgeExists := val.edges[s]
|
||
|
|
if edgeExists {
|
||
|
|
edgeVal.weight += 1
|
||
|
|
val.edges[s] = edgeVal
|
||
|
|
} else {
|
||
|
|
targetVal, targetExists := thes.pntrmap[s]
|
||
|
|
if !targetExists {
|
||
|
|
thes.pntrmap[s] = makeNode(s, scentenceId)
|
||
|
|
targetVal = thes.pntrmap[s]
|
||
|
|
}
|
||
|
|
val.edges[s] = Edge{name: s, weight: 1, target: targetVal}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|