winamp/Src/external_dependencies/openmpt-trunk/unarchiver/unarchiver.cpp
2024-09-24 14:54:57 +02:00

174 lines
3.2 KiB
C++

/*
* unarchiver.cpp
* --------------
* Purpose: archive loader
* Notes : (currently none)
* Authors: OpenMPT Devs
* The OpenMPT source code is released under the BSD license. Read LICENSE for more details.
*/
#include "stdafx.h"
#include "unarchiver.h"
#include "../common/FileReader.h"
OPENMPT_NAMESPACE_BEGIN
CUnarchiver::CUnarchiver(FileReader &file)
: impl(nullptr)
, inFile(file)
, emptyArchive(inFile)
#if (defined(MPT_WITH_ZLIB) && defined(MPT_WITH_MINIZIP)) || defined(MPT_WITH_MINIZ)
, zipArchive(inFile)
#endif
#ifdef MPT_WITH_LHASA
, lhaArchive(inFile)
#endif
#if defined(MPT_WITH_ZLIB) || defined(MPT_WITH_MINIZ)
, gzipArchive(inFile)
#endif
#ifdef MPT_WITH_UNRAR
, rarArchive(inFile)
#endif
#ifdef MPT_WITH_ANCIENT
, ancientArchive(inFile)
#endif
{
inFile.Rewind();
#if (defined(MPT_WITH_ZLIB) && defined(MPT_WITH_MINIZIP)) || defined(MPT_WITH_MINIZ)
if(zipArchive.IsArchive()) { impl = &zipArchive; return; }
#endif
#ifdef MPT_WITH_LHASA
if(lhaArchive.IsArchive()) { impl = &lhaArchive; return; }
#endif
#if defined(MPT_WITH_ZLIB) || defined(MPT_WITH_MINIZ)
if(gzipArchive.IsArchive()) { impl = &gzipArchive; return; }
#endif
#ifdef MPT_WITH_UNRAR
if(rarArchive.IsArchive()) { impl = &rarArchive; return; }
#endif
#ifdef MPT_WITH_ANCIENT
if(ancientArchive.IsArchive()) { impl = &ancientArchive; return; }
#endif
impl = &emptyArchive;
}
CUnarchiver::~CUnarchiver()
{
return;
}
static inline std::string GetExtension(const std::string &filename)
{
if(filename.find_last_of(".") != std::string::npos)
{
return mpt::ToLowerCaseAscii(filename.substr(filename.find_last_of(".") + 1));
}
return std::string();
}
std::size_t CUnarchiver::FindBestFile(const std::vector<const char *> &extensions)
{
if(!IsArchive())
{
return failIndex;
}
uint64 biggestSize = 0;
std::size_t bestIndex = failIndex;
for(std::size_t i = 0; i < size(); ++i)
{
if(operator[](i).type != ArchiveFileType::Normal)
{
continue;
}
const std::string ext = GetExtension(operator[](i).name.ToUTF8());
if(ext == "diz" || ext == "nfo" || ext == "txt")
{
// we do not want these
continue;
}
// Compare with list of preferred extensions
if(mpt::contains(extensions, ext))
{
bestIndex = i;
break;
}
if(operator[](i).size >= biggestSize)
{
biggestSize = operator[](i).size;
bestIndex = i;
}
}
return bestIndex;
}
bool CUnarchiver::ExtractBestFile(const std::vector<const char *> &extensions)
{
std::size_t bestFile = FindBestFile(extensions);
if(bestFile == failIndex)
{
return false;
}
return ExtractFile(bestFile);
}
bool CUnarchiver::IsArchive() const
{
return impl->IsArchive();
}
mpt::ustring CUnarchiver::GetComment() const
{
return impl->GetComment();
}
bool CUnarchiver::ExtractFile(std::size_t index)
{
return impl->ExtractFile(index);
}
FileReader CUnarchiver::GetOutputFile() const
{
return impl->GetOutputFile();
}
std::size_t CUnarchiver::size() const
{
return impl->size();
}
IArchive::const_iterator CUnarchiver::begin() const
{
return impl->begin();
}
IArchive::const_iterator CUnarchiver::end() const
{
return impl->end();
}
const ArchiveFileInfo & CUnarchiver::operator [] (std::size_t index) const
{
return impl->operator[](index);
}
OPENMPT_NAMESPACE_END