mirrors/LibHac
1
0
Fork 0
mirror of https://github.com/Thealexbarney/LibHac.git synced 2024-11-14 10:49:41 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow adding individual files to the RomFS builder

Browse source
This commit is contained in:
Alex Barney 2019-02-03 18:27:04 -06:00
parent c1beb9d1fd
commit cdc3800e8d
1 changed files with 26 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
src/LibHac/IO/RomFs/RomFsBuilder.cs
View file

@ -12,38 +12,44 @@ namespace LibHac.IO.RomFs
private const int HeaderWithPaddingSize = 0x200; private const int HeaderWithPaddingSize = 0x200;
public List<IStorage> Sources { get; } = new List<IStorage>(); public List<IStorage> Sources { get; } = new List<IStorage>();
public HierarchicalRomFileTable FileTable { get; } public HierarchicalRomFileTable FileTable { get; } = new HierarchicalRomFileTable();
private long CurrentOffset { get; set; }
public RomFsBuilder() { }
public RomFsBuilder(IFileSystem input) public RomFsBuilder(IFileSystem input)
{ {
FileTable = new HierarchicalRomFileTable();
var fileInfo = new RomFileInfo();
long offset = 0;
foreach (DirectoryEntry file in input.EnumerateEntries().Where(x => x.Type == DirectoryEntryType.File) foreach (DirectoryEntry file in input.EnumerateEntries().Where(x => x.Type == DirectoryEntryType.File)
.OrderBy(x => x.FullPath, StringComparer.Ordinal)) .OrderBy(x => x.FullPath, StringComparer.Ordinal))
{ {
fileInfo.Offset = offset; AddFile(file.FullPath, input.OpenFile(file.FullPath, OpenMode.Read));
fileInfo.Length = file.Size;
IStorage fileStorage = input.OpenFile(file.FullPath, OpenMode.Read).AsStorage();
Sources.Add(fileStorage);
long newOffset = offset + file.Size;
offset = Util.AlignUp(newOffset, FileAlignment);
var padding = new NullStorage(offset - newOffset);
Sources.Add(padding);
FileTable.CreateFile(file.FullPath, ref fileInfo);
} }
}
FileTable.TrimExcess(); public void AddFile(string path, IFile file)
{
var fileInfo = new RomFileInfo();
long fileSize = file.GetSize();
fileInfo.Offset = CurrentOffset;
fileInfo.Length = fileSize;
IStorage fileStorage = file.AsStorage();
Sources.Add(fileStorage);
long newOffset = CurrentOffset + fileSize;
CurrentOffset = Util.AlignUp(newOffset, FileAlignment);
var padding = new NullStorage(CurrentOffset - newOffset);
Sources.Add(padding);
FileTable.CreateFile(path, ref fileInfo);
} }
public IStorage Build() public IStorage Build()
{ {
FileTable.TrimExcess();
var header = new byte[HeaderWithPaddingSize]; var header = new byte[HeaderWithPaddingSize];
var headerWriter = new BinaryWriter(new MemoryStream(header)); var headerWriter = new BinaryWriter(new MemoryStream(header));