o3de/Code/Sandbox/Editor/Alembic/AlembicCompiler.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 "AlembicCompiler.h"

// Editor
#include "AlembicCompileDialog.h"
#include "Util/EditorUtils.h"
#include "Util/FileUtil.h"
#include "Util/PathUtil.h"

// AzCore
#include <AzCore/std/string/wildcard.h>

// AzToolsFramework
#include <AzToolsFramework/AssetBrowser/AssetBrowserEntry.h>



namespace Internal
{
    // Attempt to add the file to source control if it is available
    bool TryAddFileToSourceControl(const QString& filename)
    {
        if (!CFileUtil::CheckoutFile(filename.toUtf8().data(), nullptr))
        {
        CryWarning(VALIDATOR_MODULE_EDITOR, VALIDATOR_ERROR, "Failed to add file %s to the source control provider", filename.toUtf8().constData());
            return false;
        }

        return true;
    }
} // namespace Internal

CAlembicCompiler::CAlembicCompiler()
{
    AzToolsFramework::AssetBrowser::AssetBrowserInteractionNotificationBus::Handler::BusConnect();
}

CAlembicCompiler::~CAlembicCompiler()
{
    AzToolsFramework::AssetBrowser::AssetBrowserInteractionNotificationBus::Handler::BusDisconnect();
}

bool CAlembicCompiler::CompileAlembic(const QString& fullPath)
{
    bool compileConfigFileSaved = false;
    const QString configPath = Path::ReplaceExtension(fullPath, "cbc");
    XmlNodeRef config = XmlHelpers::LoadXmlFromFile(configPath.toUtf8().data());

    CAlembicCompileDialog dialog(config);

    if (dialog.exec() == QDialog::Accepted)
    {
        bool          configChanged = false;
        const QString upAxis = dialog.GetUpAxis();
        const QString playbackFromMemory = dialog.GetPlaybackFromMemory();
        const QString blockCompressionFormat = dialog.GetBlockCompressionFormat();
        const QString meshPrediction = dialog.GetMeshPrediction();
        const QString useBFrames = dialog.GetUseBFrames();
        const uint    indexFrameDistance = dialog.GetIndexFrameDistance();
        const double  positionPrecision = dialog.GetPositionPrecision();
        const float   uvMax = dialog.GetUVmax();

        if (!config)
        {
            CryLog("Build configuration file not found, writing new one");
            config = XmlHelpers::CreateXmlNode("CacheBuildConfiguration");
            configChanged = true;
        }

        if (strcmp(config->getAttr("UpAxis"), upAxis.toUtf8().data()) != 0)
        {
            config->setAttr("UpAxis", upAxis.toUtf8().data());
            configChanged = true;
        }
        if (strcmp(config->getAttr("MeshPrediction"), meshPrediction.toUtf8().data()) != 0)
        {
            config->setAttr("MeshPrediction", meshPrediction.toUtf8().data());
            configChanged = true;
        }
        if (strcmp(config->getAttr("UseBFrames"), useBFrames.toUtf8().data()) != 0)
        {
            config->setAttr("UseBFrames", useBFrames.toUtf8().data());
            configChanged = true;
        }
        if (atoi(config->getAttr("IndexFrameDistance")) != indexFrameDistance)
        {
            config->setAttr("IndexFrameDistance", indexFrameDistance);
            configChanged = true;
        }
        if (strcmp(config->getAttr("BlockCompressionFormat"), blockCompressionFormat.toUtf8().data()) != 0)
        {
            config->setAttr("BlockCompressionFormat", blockCompressionFormat.toUtf8().data());
            configChanged = true;
        }
        if (strcmp(config->getAttr("PlaybackFromMemory"), playbackFromMemory.toUtf8().data()) != 0)
        {
            config->setAttr("PlaybackFromMemory", playbackFromMemory.toUtf8().data());
            configChanged = true;
        }
        if (atof(config->getAttr("PositionPrecision")) != positionPrecision)
        {
            config->setAttr("PositionPrecision", positionPrecision);
            configChanged = true;
        }
        if (atof(config->getAttr("UVmax")) != uvMax)
        {
            config->setAttr("UVmax", uvMax);
            configChanged = true;
        }

        if (configChanged)
        {
            compileConfigFileSaved = XmlHelpers::SaveXmlNode(GetIEditor()->GetFileUtil(), config, configPath.toUtf8().data());
            if (compileConfigFileSaved)
            {
                // If we just created the file above, or the cbc file was not previously managed, attempt to add it to perforce now.
                // Note that XmlHelpers::SaveXmlNode will prompt the user to checkout or overwrite the file
                Internal::TryAddFileToSourceControl(configPath);
            }
        }       
    }

    return compileConfigFileSaved;
}

void CAlembicCompiler::AddSourceFileOpeners(const char* fullSourceFileName, [[maybe_unused]] const AZ::Uuid& sourceUUID, AzToolsFramework::AssetBrowser::SourceFileOpenerList& openers)
{
    using namespace AzToolsFramework;
    using namespace AzToolsFramework::AssetBrowser;

    if (AZStd::wildcard_match("*.abc", fullSourceFileName))
    {
        auto alembicCallback = [this]([[maybe_unused]] const char* fullSourceFileNameInCall, const AZ::Uuid& sourceUUIDInCall)
        {
            const SourceAssetBrowserEntry* fullDetails = SourceAssetBrowserEntry::GetSourceByUuid(sourceUUIDInCall);
            if (fullDetails)
            {
                CompileAlembic(fullDetails->GetRelativePath().c_str());
            }
        };

        openers.push_back({ "Lumberyard_AlembicCompiler", "Open In Alembic Compiler...", QIcon(), alembicCallback });
    }
}