SplineCurve.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
 
#include "Geo/Curves/Curve.h"
 
#include "Math/Point.h"
#include "Math/InterpCurve.h"
 
DEFINE_INTERPCURVE_WRAPPER_STRUCT(FInterpCurveDouble, double)
DEFINE_INTERPCURVE_WRAPPER_STRUCT(FInterpCurveFVector, FVector)
 
namespace UE::CADKernel
{
 
class CADKERNEL_API FSplineCurve : public FCurve
{
    friend class FEntity;
 
protected:
 
    FInterpCurveFVector Position;
 
    //FInterpCurveQuat Rotation;
 
    //FInterpCurveFPoint Scale;
 
    FInterpCurveDouble ReparamTable;
 
    bool bClosedLoop;
 
    //bool bLoopPositionOverride;
    //double LoopPosition;
 
    FSplineCurve(const TArray<FVector>& InPoles)
    {
        SetSplinePoints(InPoles);
    }
 
    FSplineCurve(const TArray<FVector>& InPoles, const TArray<FVector>& InTangents)
    {
        SetSplinePoints(InPoles, InTangents);
    }
 
    FSplineCurve(const TArray<FVector>& InPoles, const TArray<FVector>& InArriveTangents, const TArray<FVector>& InLeaveTangents)
    {
        SetSplinePoints(InPoles, InArriveTangents, InLeaveTangents);
    }
 
    FSplineCurve() = default;
 
    FVector GetLocationAtSplineInputKey(double InCoordinate) const;
 
    FVector GetTangentAtSplineInputKey(double InCoordinate) const;
 
    FVector GetDirectionAtSplineInputKey(double InCoordinate) const;
 
public:
 
    virtual void Serialize(FCADKernelArchive& Ar) override
    {
        FCurve::Serialize(Ar);
        Ar << Position;
        Ar << ReparamTable;
        Ar << bClosedLoop;
    }
 
#ifdef CADKERNEL_DEV
    virtual FInfoEntity& GetInfo(FInfoEntity&) const override;
#endif
 
    virtual ECurve GetCurveType() const override
    {
        return ECurve::Spline;
    }
 
    virtual TSharedPtr<FEntityGeom> ApplyMatrix(const FMatrixH& InMatrix) const override;
    virtual void Offset(const FVector& OffsetDirection) override;
 
    virtual void EvaluatePoint(double Coordinate, FCurvePoint& OutPoint, int32 DerivativeOrder = 0) const override;
 
    virtual void ExtendTo(const FVector& Point) override;
 
    void UpdateSpline(bool bClosedLoop = false, bool bStationaryEndpoints = false, int32 ReparamStepsPerSegment = 10);
 
    double GetSegmentLength(const int32 Index, const double Param, bool bClosedLoop = false) const;
 
    double GetSplineLength() const;
 
    void SetSplinePoints(const TArray<FVector>& InPoints);
    void SetSplinePoints(const TArray<FVector>& InPoints, const TArray<FVector>& InTangents);
    void SetSplinePoints(const TArray<FVector>& InPoints, const TArray<FVector>& InArriveTangents, const TArray<FVector>& InLeaveTangents);
 
    const FInterpCurveFVector& GetSplinePointsPosition() const { return Position; }
};
 
} // namespace UE::CADKernel