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

IFolder

Browse Source
This commit is contained in:
Justin Swanson 2020-08-11 10:24:02 -05:00
parent b32f12f978
commit a6169988c1
3 changed files with 23 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,7 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Wabbajack.Common;
@ -28,15 +29,9 @@ namespace Compression.BSA
public FileFlags FileFlags { get; private set; }
public IEnumerable<IFile> Files
{
get
{
foreach (var folder in _folders.Value)
foreach (var file in folder._files.Value)
yield return file;
}
}
public IEnumerable<IFile> Files => _folders.Value.SelectMany(f => f.Files);
public IEnumerable<IFolder> Folders => _folders.Value;
public ArchiveStateObject State => new BSAStateObject(this);

16
Compression.BSA/BSA/Reader/FolderRecord.cs
View File

@ -9,34 +9,30 @@ using File = Alphaleonis.Win32.Filesystem.File;
namespace Compression.BSA
{
public class FolderRecord
public class FolderRecord : IFolder
{
internal readonly BSAReader BSA;
private readonly ReadOnlyMemorySlice<byte> _data;
internal Lazy<FileRecord[]> _files;
private ReadOnlyMemorySlice<byte>? _nameData;
private int _prevFileCount;
internal FileNameBlock FileNameBlock;
private readonly Lazy<string> _name;
internal int Index { get; }
public string Name { get; private set; }
public int Index { get; }
public string Name => _name.Value;
public IEnumerable<IFile> Files => _files.Value;
internal FolderRecord(BSAReader bsa, ReadOnlyMemorySlice<byte> data, int index)
{
BSA = bsa;
_data = data;
Index = index;
_name = new Lazy<string>(
() => _nameData.HasValue ? _nameData.Value.ReadStringTerm(BSA.HeaderType) : string.Empty,
isThreadSafe: true);
}
private bool IsLongform => BSA.HeaderType == VersionType.SSE;
public ulong Hash => BinaryPrimitives.ReadUInt64LittleEndian(_data);
public uint FileCount => BinaryPrimitives.ReadUInt32LittleEndian(_data.Slice(0x8));
public int FileCount => checked((int)BinaryPrimitives.ReadUInt32LittleEndian(_data.Slice(0x8)));
public uint Unknown => IsLongform ?
BinaryPrimitives.ReadUInt32LittleEndian(_data.Slice(0xC)) :
@ -65,7 +61,7 @@ namespace Compression.BSA
{
var len = rdr.ReadByte();
data = rdr.ReadBytes(len + totalFileLen);
_nameData = data.Slice(0, len);
Name = data.Slice(0, len).ReadStringTerm(BSA.HeaderType);
data = data.Slice(len);
}
else

13
Compression.BSA/Interfaces/IFolder.cs Normal file
View File

@ -0,0 +1,13 @@
using System;
using System.Collections.Generic;
using System.Text;
namespace Compression.BSA
{
public interface IFolder
{
string Name { get; }
IEnumerable<IFile> Files { get; }
int FileCount { get; }
}
}