mirror of
https://github.com/OpenDiablo2/OpenDiablo2
synced 2024-09-06 11:34:15 -04:00
* move music path enumerations into d2resource * move text dictionary (.tbl) loader into d2fileformats sub-package d2tbl * lint fix, add doc file for d2tbl * moved data_dictionary.go into d2fileformats sub-package d2txt, added doc file * added sub-packages d2geom for geometry-related things, and d2path for path-related things * moved calcstring.go to d2calculation * move bitmuncher, bitstream, stream reader/writer from d2common into sub-package d2datautils * fix lint errors in d2datadict loaders (caused by moving stuf around in d2common) * move size.go into d2geom * move d2common/cache.go into sub-package d2common/d2cache * renamed d2debugutil to d2util, moved utility functions from d2common into d2util
105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
package d2asset
|
|
|
|
import (
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2cache"
|
|
"strings"
|
|
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2interface"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2resource"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2core/d2config"
|
|
)
|
|
|
|
const (
|
|
fileBudget = 1024 * 1024 * 32
|
|
)
|
|
|
|
// Static checks to confirm struct conforms to interface
|
|
var _ d2interface.FileManager = &fileManager{}
|
|
var _ d2interface.Cacher = &fileManager{}
|
|
|
|
type fileManager struct {
|
|
cache d2interface.Cache
|
|
archiveManager d2interface.ArchiveManager
|
|
config *d2config.Configuration
|
|
}
|
|
|
|
func createFileManager(config *d2config.Configuration,
|
|
archiveManager d2interface.ArchiveManager) d2interface.FileManager {
|
|
return &fileManager{
|
|
d2cache.CreateCache(fileBudget),
|
|
archiveManager,
|
|
config,
|
|
}
|
|
}
|
|
|
|
// LoadFileStream loads a file as a stream automatically from an archive
|
|
func (fm *fileManager) LoadFileStream(filePath string) (d2interface.ArchiveDataStream, error) {
|
|
filePath = fm.fixupFilePath(filePath)
|
|
|
|
archive, err := fm.archiveManager.LoadArchiveForFile(filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return archive.ReadFileStream(filePath)
|
|
}
|
|
|
|
// LoadFile loads a file automatically from a managed archive
|
|
func (fm *fileManager) LoadFile(filePath string) ([]byte, error) {
|
|
filePath = fm.fixupFilePath(filePath)
|
|
if value, found := fm.cache.Retrieve(filePath); found {
|
|
return value.([]byte), nil
|
|
}
|
|
|
|
archive, err := fm.archiveManager.LoadArchiveForFile(filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data, err := archive.ReadFile(filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := fm.cache.Insert(filePath, data, len(data)); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
// FileExists checks if a file exists in an archive
|
|
func (fm *fileManager) FileExists(filePath string) (bool, error) {
|
|
filePath = fm.fixupFilePath(filePath)
|
|
return fm.archiveManager.FileExistsInArchive(filePath)
|
|
}
|
|
|
|
func (fm *fileManager) ClearCache() {
|
|
fm.cache.Clear()
|
|
}
|
|
|
|
func (fm *fileManager) GetCache() d2interface.Cache {
|
|
return fm.cache
|
|
}
|
|
|
|
func (fm *fileManager) fixupFilePath(filePath string) string {
|
|
filePath = fm.removeLocaleTokens(filePath)
|
|
filePath = strings.ToLower(filePath)
|
|
filePath = strings.ReplaceAll(filePath, `/`, "\\")
|
|
filePath = strings.TrimPrefix(filePath, "\\")
|
|
|
|
return filePath
|
|
}
|
|
|
|
func (fm *fileManager) removeLocaleTokens(filePath string) string {
|
|
tableToken := d2resource.LanguageTableToken
|
|
fontToken := d2resource.LanguageFontToken
|
|
|
|
filePath = strings.ReplaceAll(filePath, tableToken, fm.config.Language)
|
|
|
|
// fixme: not all languages==latin
|
|
filePath = strings.ReplaceAll(filePath, fontToken, "latin")
|
|
|
|
return filePath
|
|
}
|