o3de/Code/Sandbox/Editor/GameResourcesExporter.cpp

/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#include "EditorDefs.h"

#include "GameResourcesExporter.h"

// Qt
#include <QFileDialog>

// Editor
#include "UsedResources.h"
#include "GameEngine.h"
#include "Include/IObjectManager.h"
#include "WaitProgress.h"


//////////////////////////////////////////////////////////////////////////
// Static data.
//////////////////////////////////////////////////////////////////////////
CGameResourcesExporter::Files CGameResourcesExporter::m_files;

//////////////////////////////////////////////////////////////////////////
void CGameResourcesExporter::ChooseDirectoryAndSave()
{
    ChooseDirectory();
    if (!m_path.isEmpty())
    {
        Save(m_path);
    }
}

//////////////////////////////////////////////////////////////////////////
void CGameResourcesExporter::ChooseDirectory()
{
    m_path = QFileDialog::getExistingDirectory(nullptr, QObject::tr("Choose Target (root/PrimaryCD) Folder"));
}

//////////////////////////////////////////////////////////////////////////
void CGameResourcesExporter::GatherAllLoadedResources()
{
    m_files.clear();
    m_files.reserve(100000);        // count from GetResourceList, GetFilesFromObjects ...  is unknown

    auto pResList = gEnv->pCryPak->GetResourceList(AZ::IO::IArchive::RFOM_Level);
    {
        for (const char* filename = pResList->GetFirst(); filename; filename = pResList->GetNext())
        {
            m_files.push_back(filename);
        }
    }

    GetFilesFromObjects();
}

//////////////////////////////////////////////////////////////////////////
void CGameResourcesExporter::SetUsedResources(CUsedResources& resources)
{
    for (CUsedResources::TResourceFiles::const_iterator it = resources.files.begin(); it != resources.files.end(); it++)
    {
        m_files.push_back(*it);
    }
}

//////////////////////////////////////////////////////////////////////////
void CGameResourcesExporter::Save(const QString& outputDirectory)
{
    CMemoryBlock data;

    int numFiles = m_files.size();

    CLogFile::WriteLine("===========================================================================");
    CLogFile::FormatLine("Exporting Level %s resources, %d files", GetIEditor()->GetGameEngine()->GetLevelName().toUtf8().data(), numFiles);
    CLogFile::WriteLine("===========================================================================");

    // Needed files.
    CWaitProgress wait("Exporting Resources");
    for (int i = 0; i < numFiles; i++)
    {
        QString srcFilename = m_files[i];
        if (!wait.Step((i * 100) / numFiles))
        {
            break;
        }
        wait.SetText(srcFilename.toUtf8().data());

        CLogFile::WriteLine(srcFilename.toUtf8().data());

        CCryFile file;
        if (file.Open(srcFilename.toUtf8().data(), "rb"))
        {
            // Save this file in target folder.
            QString trgFilename = Path::Make(outputDirectory, srcFilename);
            int fsize = file.GetLength();
            if (fsize > data.GetSize())
            {
                data.Allocate(fsize + 16);
            }
            // Read data.
            file.ReadRaw(data.GetBuffer(), fsize);

            // Save this data to target file.
            QString trgFileDir = Path::GetPath(trgFilename);
            CFileUtil::CreateDirectory(trgFileDir.toUtf8().data());
            // Create a file.
            FILE* trgFile = nullptr;
            azfopen(&trgFile, trgFilename.toUtf8().data(), "wb");
            if (trgFile)
            {
                // Save data to new file.
                fwrite(data.GetBuffer(), fsize, 1, trgFile);
                fclose(trgFile);
            }
        }
    }
    CLogFile::WriteLine("===========================================================================");
    m_files.clear();
}

#if defined(WIN64) || defined(APPLE) || defined(AZ_PLATFORM_LINUX)
template <class Container1, class Container2>
void Append(Container1& a, const Container2& b)
{
    a.reserve (a.size() + b.size());
    for (auto it = b.begin(); it != b.end(); ++it)
    {
        a.insert(a.end(), *it);
    }
}
#else
template <class Container1, class Container2>
void Append(Container1& a, const Container2& b)
{
    a.insert (a.end(), b.begin(), b.end());
}
#endif
//////////////////////////////////////////////////////////////////////////
//
// Go through all editor objects and gathers files from thier properties.
//
//////////////////////////////////////////////////////////////////////////
void CGameResourcesExporter::GetFilesFromObjects()
{
    CUsedResources rs;
    GetIEditor()->GetObjectManager()->GatherUsedResources(rs);

    Append(m_files, rs.files);
}