int LafsFlat::getDirectoryEntryCount(const DirectoryEntry& dir)
{
int iEntryCount = 2; //2 for root entry 1 (root directory) + 1 (jump directory)
std::vector<DirectoryEntry> it = dir.getDirectories();
for(std::vector<DirectoryEntry>::const_iterator it = dir.getDirectories().begin();
it != dir.getDirectories().end();
it++)
iEntryCount += getDirectoryEntryCount(*it);
return iEntryCount;
}
WORD RsfsFormatter::getFileIndexSize(const DirectoryEntry& rootDir)
{
const std::vector<DirectoryEntry>& dirs = rootDir.getDirectories();
WORD wDataRegionOffset = rootDir.getFiles().size() * RSFS_FILEENTRY_SIZE + dirs.size() * RSFS_DIRECTORYENTRY_SIZE;
for(std::vector<DirectoryEntry>::const_iterator it = dirs.begin(); it != dirs.end(); it++)
wDataRegionOffset += getFileIndexSize(*it);
return wDataRegionOffset;
}
void RsfsFormatter::create(std::ostream& out, const DirectoryEntry& rootDir)
{
std::stringstream ssFileIndexBuffer;
std::stringstream ssFileRawBuffer;
WORD wDataRegionOffset = getFileIndexSize(rootDir) + RSFS_HEADER_SIZE;
writeDirectory(rootDir, wDataRegionOffset, ssFileIndexBuffer, ssFileRawBuffer);
//Write out header
out.put(1);
out.put(0);
WORD numRootEntries = rootDir.getDirectories().size() + rootDir.getFiles().size();
out.write(reinterpret_cast<char*>(&numRootEntries), sizeof(WORD));
//write file directories
for(unsigned char b = ssFileIndexBuffer.get(); !ssFileIndexBuffer.eof(); b = ssFileIndexBuffer.get())
out.put(b);
for(unsigned char b = ssFileRawBuffer.get(); !ssFileRawBuffer.eof(); b = ssFileRawBuffer.get())
out.put(b);
}
void RsfsFormatter::writeDirectory(const DirectoryEntry& dir, const WORD wDataRegionOffset, std::ostream& fileIndexBuffer, std::ostream& fileRawDataBuffer)
{
char fileReadBuffer[100];
const std::vector<DirectoryEntry>& dirs = dir.getDirectories();
const std::vector<FileEntry>& files = dir.getFiles();
WORD wBuffer = 0;
if(dir.getName().length() > 0)
{
wBuffer = RSFS_ENTRYTYPE_DIRECTORY;
fileIndexBuffer.write(reinterpret_cast<char*>(&wBuffer), sizeof(WORD));
fileIndexBuffer.write(dir.getName().c_str(), dir.getName().length());
for(unsigned int i = RSFS_NAME_SIZE - dir.getName().length(); i > 0; i--)
fileIndexBuffer.put(0);
if(dir.getDirectories().size() > 0xFFFF)
throw FormattingException("The number of sub-directories exceeds the max permittable number of sub-directories.");
wBuffer = static_cast<WORD>(dir.getDirectories().size() + dir.getFiles().size());
fileIndexBuffer.write(reinterpret_cast<char*>(&wBuffer), sizeof(WORD));
fileIndexBuffer.put(0);
fileIndexBuffer.put(0);
}
for(std::vector<FileEntry>::const_iterator it = files.begin(); it != files.end(); it++)
{
if(it->getName().length() >= RSFS_NAME_SIZE)
{
throw new FormattingException(std::string("Skipped due to file name exceeding max length: ") + it->getName());
continue;
}
wBuffer = RSFS_ENTRYTYPE_FILE;
fileIndexBuffer.write(reinterpret_cast<char*>(&wBuffer), sizeof(WORD));
fileIndexBuffer.write(it->getName().c_str(), it->getName().length());
//Padd the file name with nulls.
for(unsigned int i = RSFS_NAME_SIZE - it->getName().length(); i > 0; i--)
fileIndexBuffer.put(0);
if(it->getSize() > 0xFFFF)
throw FormattingException("Size of file exceeded permitable size.");
wBuffer = static_cast<WORD>(it->getSize());
fileIndexBuffer.write(reinterpret_cast<char*>(&wBuffer), sizeof(WORD));
//Make sure data is aligned in words
std::streamoff offset = 0;
for(offset = wDataRegionOffset + fileRawDataBuffer.tellp(); offset % 2; offset++)
fileRawDataBuffer.put(0);
if(offset + it->getSize() > 0xFFFF)
throw FormattingException("Size of medium exceeded permitable size.");
WORD wFileDataOffset = static_cast<WORD>(offset) / 2; //Offset in terms of words.
fileIndexBuffer.write(reinterpret_cast<char*>(&wFileDataOffset), sizeof(WORD));
//Write our file's data to the fileRawBuffer.
std::ifstream inFile(it->getSourceFile(), std::ifstream::binary | std::ifstream::in);
if(!inFile.good())
throw FormattingException(std::string("Error reading from source file") + it->getName());
while(!inFile.eof())
{
inFile.read(fileReadBuffer, sizeof(fileReadBuffer));
fileRawDataBuffer.write(fileReadBuffer, inFile.gcount());
}
}
for(std::vector<DirectoryEntry>::const_iterator it = dirs.begin(); it != dirs.end(); it++)
writeDirectory(*it, wDataRegionOffset, fileIndexBuffer, fileRawDataBuffer);
}
