MatrixInterfaces.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "VectorTypes.h"
 
 
namespace UE
{
    namespace Solvers
    {
        using namespace UE::Geometry;
 
        template<typename RealType>
        class TSparseMatrixAssembler
        {
        public:
            TUniqueFunction<void(int32)> ReserveEntriesFunc;
 
            TUniqueFunction<void(int32, int32, RealType)> AddEntryFunc;
 
            struct FTupleData
            {
                FTupleData(int32 IIn, int32 JIn, RealType ValueIn)
                    : I(IIn)
                    , J(JIn)
                    , Value(ValueIn)
                {}
                int32 I;
                int32 J;
                RealType Value;
            };
 
            TUniqueFunction<void(const FTupleData*, int32)> AddEntriesFunc = [this](const FTupleData* DataStart, int32 Num)
            {
                if (!AddEntryFunc)
                {
                    return;
                }
                for (int32 i = 0; i < Num; ++i)
                {
                    const FTupleData& Tuple = DataStart[Num];
                    AddEntryFunc(Tuple.I, Tuple.J, Tuple.Value);
                }
            };
        };
 
        typedef TSparseMatrixAssembler<double> FSparseMatrixAssemblerd;
        typedef TSparseMatrixAssembler<float> FSparseMatrixAssemblerf;
 
 
 
        struct FPositionConstraint
        {
            int32 ElementID = -1;
 
            int32 ConstraintIndex = -1;
 
            FVector3d Position = FVector3d::Zero();
 
            bool     bPostFix = false;
 
            double Weight = 1.0;
        };
 
 
 
        struct FUVConstraint
        {
            int32 ElementID = -1;
 
            int32 ConstraintIndex = -1;
 
            FVector2d Position = FVector2d::Zero();
 
            bool bPostFix = false;
 
            double Weight = 1.0;
        };
 
 
    }
}