o3de/Code/Tools/SceneAPI/SceneData/GraphData/MeshVertexBitangentData.cpp

/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#include <SceneAPI/SceneData/GraphData/MeshVertexBitangentData.h>
#include <AzCore/Serialization/SerializeContext.h>
#include <AzCore/RTTI/BehaviorContext.h>

namespace AZ::SceneData::GraphData
{
    void MeshVertexBitangentData::Reflect(ReflectContext* context)
    {
        SerializeContext* serializeContext = azrtti_cast<SerializeContext*>(context);
        if (serializeContext)
        {
            serializeContext->Class<MeshVertexBitangentData>()->Version(2);
        }

        BehaviorContext* behaviorContext = azrtti_cast<BehaviorContext*>(context);
        if (behaviorContext)
        {
            behaviorContext->Class<MeshVertexBitangentData>()
                ->Attribute(AZ::Script::Attributes::Scope, AZ::Script::Attributes::ScopeFlags::Common)
                ->Attribute(AZ::Script::Attributes::Module, "scene")
                ->Method("GetCount", &MeshVertexBitangentData::GetCount)
                ->Method("GetBitangent", &MeshVertexBitangentData::GetBitangent)
                ->Method("GetBitangentSetIndex", &MeshVertexBitangentData::GetBitangentSetIndex)
                ->Method("GetGenerationMethod", &MeshVertexBitangentData::GetGenerationMethod)
                ->Enum<(int)SceneAPI::DataTypes::TangentGenerationMethod::FromSourceScene>("FromSourceScene")
                ->Enum<(int)SceneAPI::DataTypes::TangentGenerationMethod::MikkT>("MikkT");
        }
    }

    void MeshVertexBitangentData::CloneAttributesFrom(const IGraphObject* sourceObject)
    {
        IMeshVertexBitangentData::CloneAttributesFrom(sourceObject);
        if (const auto* typedSource = azrtti_cast<const MeshVertexBitangentData*>(sourceObject))
        {
            SetGenerationMethod(typedSource->GetGenerationMethod());
            SetBitangentSetIndex(typedSource->GetBitangentSetIndex());
        }
    }

    size_t MeshVertexBitangentData::GetCount() const
    {
        return m_bitangents.size();
    }

    const AZ::Vector3& MeshVertexBitangentData::GetBitangent(size_t index) const
    {
        AZ_Assert(index < m_bitangents.size(), "Invalid index %i for mesh bitangents.", index);
        return m_bitangents[index];
    }

    void MeshVertexBitangentData::ReserveContainerSpace(size_t numVerts)
    {
        m_bitangents.reserve(numVerts);
    }

    void MeshVertexBitangentData::Resize(size_t numVerts)
    {
        m_bitangents.resize(numVerts);
    }

    void MeshVertexBitangentData::AppendBitangent(const AZ::Vector3& bitangent)
    {
        m_bitangents.push_back(bitangent);
    }

    void MeshVertexBitangentData::SetBitangent(size_t vertexIndex, const AZ::Vector3& bitangent)
    {
        m_bitangents[vertexIndex] = bitangent;
    }

    void MeshVertexBitangentData::SetBitangentSetIndex(size_t setIndex)
    {
        m_setIndex = setIndex;
    }

    size_t MeshVertexBitangentData::GetBitangentSetIndex() const
    {
        return m_setIndex;
    }

    AZ::SceneAPI::DataTypes::TangentGenerationMethod MeshVertexBitangentData::GetGenerationMethod() const
    { 
        return m_generationMethod;
    }

    void MeshVertexBitangentData::SetGenerationMethod(AZ::SceneAPI::DataTypes::TangentGenerationMethod method)
    { 
        m_generationMethod = method;
    }

    void MeshVertexBitangentData::GetDebugOutput(AZ::SceneAPI::Utilities::DebugOutput& output) const
    {
        output.Write("Bitangents", m_bitangents);
        output.Write("GenerationMethod", aznumeric_cast<int64_t>(m_generationMethod));
    }
} // AZ::SceneData::GraphData