Mirrored_Repos/wabbajack
1
0
Fork 0
mirror of https://github.com/wabbajack-tools/wabbajack.git synced 2024-08-30 18:42:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

BSAReader.TryGetFolder

Browse Source
This commit is contained in:
Justin Swanson 2020-08-11 10:29:37 -05:00
parent 56fe093618
commit dbc868a363
1 changed files with 31 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
Compression.BSA/BSA/Reader/BSAReader.cs
View File

@ -1,5 +1,6 @@
using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.IO;
using System.Linq;
using System.Text;
@ -19,6 +20,7 @@ namespace Compression.BSA
internal uint _folderCount;
internal uint _folderRecordOffset;
private Lazy<FolderRecord[]> _folders = null!;
private Lazy<Dictionary<string, FolderRecord>> _foldersByName = null!;
internal string _magic = string.Empty;
internal uint _totalFileNameLength;
internal uint _totalFolderNameLength;
@ -107,6 +109,9 @@ namespace Compression.BSA
_folders = new Lazy<FolderRecord[]>(
isThreadSafe: true,
valueFactory: () => LoadFolderRecords());
_foldersByName = new Lazy<Dictionary<string, FolderRecord>>(
isThreadSafe: true,
valueFactory: GetFolderDictionary);
}
private FolderRecord[] LoadFolderRecords()
@ -139,5 +144,31 @@ namespace Compression.BSA
return ret;
}
private Dictionary<string, FolderRecord> GetFolderDictionary()
{
if (!HasFolderNames)
{
throw new ArgumentException("Cannot get folders by name if the BSA does not have folder names.");
}
var ret = new Dictionary<string, FolderRecord>();
foreach (var folder in _folders.Value)
{
ret.Add(folder.Name!, folder);
}
return ret;
}
public bool TryGetFolder(string path, [MaybeNullWhen(false)] out IFolder folder)
{
if (!HasFolderNames
|| !_foldersByName.Value.TryGetValue(path, out var folderRec))
{
folder = default;
return false;
}
folder = folderRec;
return true;
}
}
}