108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package d2common
|
|
|
|
import (
|
|
"log"
|
|
"strconv"
|
|
)
|
|
|
|
type textDictionaryHashEntry struct {
|
|
IsActive bool
|
|
Index uint16
|
|
HashValue uint32
|
|
IndexString uint32
|
|
NameString uint32
|
|
NameLength uint16
|
|
}
|
|
|
|
var lookupTable map[string]string
|
|
|
|
func TranslateString(key string) string {
|
|
result, ok := lookupTable[key]
|
|
if !ok {
|
|
log.Panicf("Could not find a string for the key '%s'", key)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func GetDictionaryEntryCount() int {
|
|
if lookupTable == nil {
|
|
return 0
|
|
}
|
|
return len(lookupTable)
|
|
}
|
|
|
|
func LoadDictionary(dictionaryData []byte) {
|
|
if lookupTable == nil {
|
|
lookupTable = make(map[string]string)
|
|
}
|
|
br := CreateStreamReader(dictionaryData)
|
|
// CRC
|
|
if _, err := br.ReadBytes(2); err != nil {
|
|
log.Fatal("Error reading CRC")
|
|
}
|
|
numberOfElements := br.GetUInt16()
|
|
hashTableSize := br.GetUInt32()
|
|
// Version (always 0)
|
|
if _, err := br.ReadByte(); err != nil {
|
|
log.Fatal("Error reading Version record")
|
|
}
|
|
br.GetUInt32() // StringOffset
|
|
br.GetUInt32() // When the number of times you have missed a match with a hash key equals this value, you give up because it is not there.
|
|
br.GetUInt32() // FileSize
|
|
|
|
elementIndex := make([]uint16, numberOfElements)
|
|
for i := 0; i < int(numberOfElements); i++ {
|
|
elementIndex[i] = br.GetUInt16()
|
|
}
|
|
|
|
hashEntries := make([]textDictionaryHashEntry, hashTableSize)
|
|
for i := 0; i < int(hashTableSize); i++ {
|
|
hashEntries[i] = textDictionaryHashEntry{
|
|
br.GetByte() == 1,
|
|
br.GetUInt16(),
|
|
br.GetUInt32(),
|
|
br.GetUInt32(),
|
|
br.GetUInt32(),
|
|
br.GetUInt16(),
|
|
}
|
|
}
|
|
|
|
for idx, hashEntry := range hashEntries {
|
|
if !hashEntry.IsActive {
|
|
continue
|
|
}
|
|
br.SetPosition(uint64(hashEntry.NameString))
|
|
nameVal, _ := br.ReadBytes(int(hashEntry.NameLength - 1))
|
|
value := string(nameVal)
|
|
br.SetPosition(uint64(hashEntry.IndexString))
|
|
key := ""
|
|
for {
|
|
b := br.GetByte()
|
|
if b == 0 {
|
|
break
|
|
}
|
|
key += string(b)
|
|
}
|
|
if key == "x" || key == "X" {
|
|
key = "#" + strconv.Itoa(idx)
|
|
}
|
|
_, exists := lookupTable[key]
|
|
if !exists {
|
|
lookupTable[key] = value
|
|
|
|
}
|
|
// Use the following code to write out the values
|
|
/*=
|
|
f, err := os.OpenFile(`C:\Users\lunat\Desktop\D2\langdict.txt`,
|
|
os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
defer f.Close()
|
|
if _, err := f.WriteString("\n[" + key + "] " + value); err != nil {
|
|
log.Println(err)
|
|
}
|
|
*/
|
|
}
|
|
}
|