mirror of
https://github.com/OpenDiablo2/OpenDiablo2
synced 2024-10-31 16:27:18 -04:00
cc678ba747
* Add asset manager * Fix rebase * Update asset manager to support mpq hash caching * Update vendoring
144 lines
3.1 KiB
Go
144 lines
3.1 KiB
Go
package d2core
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/OpenDiablo2/D2Shared/d2common/d2resource"
|
|
"github.com/OpenDiablo2/D2Shared/d2data/d2mpq"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2corecommon"
|
|
)
|
|
|
|
type archiveEntry struct {
|
|
archivePath string
|
|
hashEntryMap d2mpq.HashEntryMap
|
|
}
|
|
|
|
type assetManager struct {
|
|
fileCache *cache
|
|
archiveCache *cache
|
|
archiveEntries []archiveEntry
|
|
config *d2corecommon.Configuration
|
|
mutex sync.Mutex
|
|
}
|
|
|
|
func createAssetManager(config *d2corecommon.Configuration) *assetManager {
|
|
return &assetManager{
|
|
fileCache: createCache(1024 * 1024 * 32),
|
|
archiveCache: createCache(1024 * 1024 * 128),
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
func (am *assetManager) LoadFile(filePath string) []byte {
|
|
data, err := am.loadFile(am.fixupFilePath(filePath))
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
return data
|
|
}
|
|
|
|
func (am *assetManager) loadFile(filePath string) ([]byte, error) {
|
|
if value, found := am.fileCache.retrieve(filePath); found {
|
|
return value.([]byte), nil
|
|
}
|
|
|
|
archive, err := am.loadArchiveForFilePath(filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data, err := archive.ReadFile(filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := am.fileCache.insert(filePath, data, len(data)); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
func (am *assetManager) loadArchiveForFilePath(filePath string) (*d2mpq.MPQ, error) {
|
|
am.mutex.Lock()
|
|
defer am.mutex.Unlock()
|
|
|
|
if err := am.cacheArchiveEntries(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, archiveEntry := range am.archiveEntries {
|
|
if archiveEntry.hashEntryMap.Contains(filePath) {
|
|
return am.loadArchive(archiveEntry.archivePath)
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("file not found: %s", filePath)
|
|
}
|
|
|
|
func (am *assetManager) loadArchive(archivePath string) (*d2mpq.MPQ, error) {
|
|
if archive, found := am.archiveCache.retrieve(archivePath); found {
|
|
return archive.(*d2mpq.MPQ), nil
|
|
}
|
|
|
|
archive, err := d2mpq.Load(archivePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
stat, err := os.Stat(archivePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := am.archiveCache.insert(archivePath, archive, int(stat.Size())); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return archive, nil
|
|
}
|
|
|
|
func (am *assetManager) cacheArchiveEntries() error {
|
|
if len(am.archiveEntries) == len(am.config.MpqLoadOrder) {
|
|
return nil
|
|
}
|
|
|
|
am.archiveEntries = nil
|
|
|
|
for _, archiveName := range am.config.MpqLoadOrder {
|
|
archivePath := path.Join(am.config.MpqPath, archiveName)
|
|
archive, err := am.loadArchive(archivePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
am.archiveEntries = append(
|
|
am.archiveEntries,
|
|
archiveEntry{archivePath, archive.HashEntryMap},
|
|
)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (am *assetManager) fixupFilePath(filePath string) string {
|
|
filePath = strings.ReplaceAll(filePath, "{LANG}", am.config.Language)
|
|
if strings.ToUpper(d2resource.LanguageCode) == "CHI" {
|
|
filePath = strings.ReplaceAll(filePath, "{LANG_FONT}", am.config.Language)
|
|
} else {
|
|
filePath = strings.ReplaceAll(filePath, "{LANG_FONT}", "latin")
|
|
}
|
|
|
|
filePath = strings.ToLower(filePath)
|
|
filePath = strings.ReplaceAll(filePath, `/`, "\\")
|
|
filePath = strings.TrimPrefix(filePath, "\\")
|
|
|
|
return filePath
|
|
}
|