Shape.h

Go to the documentation of this file.

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2021 Jorrit Rouwe
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <Jolt/Physics/Body/MassProperties.h>
#include <Jolt/Physics/Collision/BackFaceMode.h>
#include <Jolt/Physics/Collision/CollisionCollector.h>
#include <Jolt/Physics/Collision/ShapeFilter.h>
#include <Jolt/Geometry/AABox.h>
#include <Jolt/Core/Reference.h>
#include <Jolt/Core/Color.h>
#include <Jolt/Core/Result.h>
#include <Jolt/Core/NonCopyable.h>
#include <Jolt/Core/UnorderedMap.h>
#include <Jolt/Core/UnorderedSet.h>
#include <Jolt/ObjectStream/SerializableObject.h>
 
JPH_NAMESPACE_BEGIN
 
struct RayCast;
class RayCastSettings;
struct ShapeCast;
class ShapeCastSettings;
class RayCastResult;
class ShapeCastResult;
class CollidePointResult;
class CollideShapeResult;
class SubShapeIDCreator;
class SubShapeID;
class PhysicsMaterial;
class TransformedShape;
class Plane;
class Shape;
class StreamOut;
class StreamIn;
#ifdef JPH_DEBUG_RENDERER
class DebugRenderer;
#endif // JPH_DEBUG_RENDERER
 
using CastRayCollector = CollisionCollector<RayCastResult, CollisionCollectorTraitsCastRay>;
using CastShapeCollector = CollisionCollector<ShapeCastResult, CollisionCollectorTraitsCastShape>;
using CollidePointCollector = CollisionCollector<CollidePointResult, CollisionCollectorTraitsCollidePoint>;
using CollideShapeCollector = CollisionCollector<CollideShapeResult, CollisionCollectorTraitsCollideShape>;
using TransformedShapeCollector = CollisionCollector<TransformedShape, CollisionCollectorTraitsCollideShape>;
 
using ShapeRefC = RefConst<Shape>;
using ShapeList = Array<ShapeRefC>;
using PhysicsMaterialRefC = RefConst<PhysicsMaterial>;
using PhysicsMaterialList = Array<PhysicsMaterialRefC>;
 
enum class EShapeType : uint8
{
    Convex,                         
    Compound,                       
    Decorated,                      
    Mesh,                           
    HeightField,                    
    
    // User defined shapes
    User1,
    User2,
    User3,
    User4,
};
 
enum class EShapeSubType : uint8
{
    // Convex shapes
    Sphere,
    Box,
    Triangle,
    Capsule,
    TaperedCapsule,
    Cylinder,
    ConvexHull,
    
    // Compound shapes
    StaticCompound,
    MutableCompound,
    
    // Decorated shapes
    RotatedTranslated,
    Scaled,
    OffsetCenterOfMass,
 
    // Other shapes
    Mesh,
    HeightField,
    
    // User defined shapes
    User1,
    User2,
    User3,
    User4,
    User5,
    User6,
    User7,
    User8,
 
    // User defined convex shapes
    UserConvex1,
    UserConvex2,
    UserConvex3,
    UserConvex4,
    UserConvex5,
    UserConvex6,
    UserConvex7,
    UserConvex8,
};
 
// Sets of shape sub types
static constexpr EShapeSubType sAllSubShapeTypes[] = { EShapeSubType::Sphere, EShapeSubType::Box, EShapeSubType::Triangle, EShapeSubType::Capsule, EShapeSubType::TaperedCapsule, EShapeSubType::Cylinder, EShapeSubType::ConvexHull, EShapeSubType::StaticCompound, EShapeSubType::MutableCompound, EShapeSubType::RotatedTranslated, EShapeSubType::Scaled, EShapeSubType::OffsetCenterOfMass, EShapeSubType::Mesh, EShapeSubType::HeightField, EShapeSubType::User1, EShapeSubType::User2, EShapeSubType::User3, EShapeSubType::User4, EShapeSubType::User5, EShapeSubType::User6, EShapeSubType::User7, EShapeSubType::User8, EShapeSubType::UserConvex1, EShapeSubType::UserConvex2, EShapeSubType::UserConvex3, EShapeSubType::UserConvex4, EShapeSubType::UserConvex5, EShapeSubType::UserConvex6, EShapeSubType::UserConvex7, EShapeSubType::UserConvex8 };
static constexpr EShapeSubType sConvexSubShapeTypes[] = { EShapeSubType::Sphere, EShapeSubType::Box, EShapeSubType::Triangle, EShapeSubType::Capsule, EShapeSubType::TaperedCapsule, EShapeSubType::Cylinder, EShapeSubType::ConvexHull, EShapeSubType::UserConvex1, EShapeSubType::UserConvex2, EShapeSubType::UserConvex3, EShapeSubType::UserConvex4, EShapeSubType::UserConvex5, EShapeSubType::UserConvex6, EShapeSubType::UserConvex7, EShapeSubType::UserConvex8 };
static constexpr EShapeSubType sCompoundSubShapeTypes[] = { EShapeSubType::StaticCompound, EShapeSubType::MutableCompound };
static constexpr EShapeSubType sDecoratorSubShapeTypes[] = { EShapeSubType::RotatedTranslated, EShapeSubType::Scaled, EShapeSubType::OffsetCenterOfMass };
 
static constexpr uint NumSubShapeTypes = (uint)size(sAllSubShapeTypes);
 
static constexpr const char *sSubShapeTypeNames[] = { "Sphere", "Box", "Triangle", "Capsule", "TaperedCapsule", "Cylinder", "ConvexHull", "StaticCompound", "MutableCompound", "RotatedTranslated", "Scaled", "OffsetCenterOfMass", "Mesh", "HeightField", "User1", "User2", "User3", "User4", "User5", "User6", "User7", "User8", "UserConvex1", "UserConvex2", "UserConvex3", "UserConvex4", "UserConvex5", "UserConvex6", "UserConvex7", "UserConvex8" };
static_assert(size(sSubShapeTypeNames) == NumSubShapeTypes);
 
class ShapeSettings : public SerializableObject, public RefTarget<ShapeSettings>
{
public:
    JPH_DECLARE_SERIALIZABLE_ABSTRACT(ShapeSettings)
 
    using ShapeResult = Result<Ref<Shape>>;
 
    virtual ShapeResult             Create() const = 0;
 
    uint64                          mUserData = 0;
 
protected:
    mutable ShapeResult             mCachedResult;
};
 
class ShapeFunctions
{
public:
    Shape *                         (*mConstruct)() = nullptr;
 
    Color                           mColor = Color::sBlack;
 
    static inline ShapeFunctions &  sGet(EShapeSubType inSubType)                                       { return sRegistry[(int)inSubType]; }
 
private:
    static ShapeFunctions           sRegistry[NumSubShapeTypes];
};
 
class Shape : public RefTarget<Shape>, public NonCopyable
{
public:
    JPH_OVERRIDE_NEW_DELETE
 
    using ShapeResult = ShapeSettings::ShapeResult;
 
                                    Shape(EShapeType inType, EShapeSubType inSubType) : mShapeType(inType), mShapeSubType(inSubType) { }
                                    Shape(EShapeType inType, EShapeSubType inSubType, const ShapeSettings &inSettings, [[maybe_unused]] ShapeResult &outResult) : mUserData(inSettings.mUserData), mShapeType(inType), mShapeSubType(inSubType) { }
 
    virtual                         ~Shape() = default;
 
    inline EShapeType               GetType() const                                                     { return mShapeType; }
    inline EShapeSubType            GetSubType() const                                                  { return mShapeSubType; }
 
    uint64                          GetUserData() const                                                 { return mUserData; }
    void                            SetUserData(uint64 inUserData)                                      { mUserData = inUserData; }
 
    virtual bool                    MustBeStatic() const                                                { return false; }
 
    virtual Vec3                    GetCenterOfMass() const                                             { return Vec3::sZero(); }
 
    virtual AABox                   GetLocalBounds() const = 0;
 
    virtual uint                    GetSubShapeIDBitsRecursive() const = 0;
 
    virtual AABox                   GetWorldSpaceBounds(Mat44Arg inCenterOfMassTransform, Vec3Arg inScale) const { return GetLocalBounds().Scaled(inScale).Transformed(inCenterOfMassTransform); }
 
    AABox                           GetWorldSpaceBounds(DMat44Arg inCenterOfMassTransform, Vec3Arg inScale) const
    {
        // Use single precision version using the rotation only
        AABox bounds = GetWorldSpaceBounds(inCenterOfMassTransform.GetRotation(), inScale);
 
        // Apply translation
        bounds.Translate(inCenterOfMassTransform.GetTranslation());
 
        return bounds;
    }
 
    virtual float                   GetInnerRadius() const = 0;
 
    virtual MassProperties          GetMassProperties() const = 0;
 
    virtual const PhysicsMaterial * GetMaterial(const SubShapeID &inSubShapeID) const = 0;
 
    virtual Vec3                    GetSurfaceNormal(const SubShapeID &inSubShapeID, Vec3Arg inLocalSurfacePosition) const = 0;
 
    using SupportingFace = StaticArray<Vec3, 32>;
 
    virtual void                    GetSupportingFace(const SubShapeID &inSubShapeID, Vec3Arg inDirection, Vec3Arg inScale, Mat44Arg inCenterOfMassTransform, SupportingFace &outVertices) const { /* Nothing */ }
 
    virtual uint64                  GetSubShapeUserData(const SubShapeID &inSubShapeID) const           { return mUserData; }
 
    virtual TransformedShape        GetSubShapeTransformedShape(const SubShapeID &inSubShapeID, Vec3Arg inPositionCOM, QuatArg inRotation, Vec3Arg inScale, SubShapeID &outRemainder) const;
 
#ifdef JPH_DEBUG_RENDERER
#endif
    virtual void                    GetSubmergedVolume(Mat44Arg inCenterOfMassTransform, Vec3Arg inScale, const Plane &inSurface, float &outTotalVolume, float &outSubmergedVolume, Vec3 &outCenterOfBuoyancy
#ifdef JPH_DEBUG_RENDERER // Not using JPH_IF_DEBUG_RENDERER for Doxygen
        , RVec3Arg inBaseOffset
#endif
        ) const = 0;
    
#ifdef JPH_DEBUG_RENDERER
    virtual void                    Draw(DebugRenderer *inRenderer, RMat44Arg inCenterOfMassTransform, Vec3Arg inScale, ColorArg inColor, bool inUseMaterialColors, bool inDrawWireframe) const = 0;
 
    virtual void                    DrawGetSupportFunction(DebugRenderer *inRenderer, RMat44Arg inCenterOfMassTransform, Vec3Arg inScale, ColorArg inColor, bool inDrawSupportDirection) const { /* Only implemented for convex shapes */ }
 
    virtual void                    DrawGetSupportingFace(DebugRenderer *inRenderer, RMat44Arg inCenterOfMassTransform, Vec3Arg inScale) const { /* Only implemented for convex shapes */ }
#endif // JPH_DEBUG_RENDERER
 
    virtual bool                    CastRay(const RayCast &inRay, const SubShapeIDCreator &inSubShapeIDCreator, RayCastResult &ioHit) const = 0;
 
    virtual void                    CastRay(const RayCast &inRay, const RayCastSettings &inRayCastSettings, const SubShapeIDCreator &inSubShapeIDCreator, CastRayCollector &ioCollector, const ShapeFilter &inShapeFilter = { }) const = 0;
 
    virtual void                    CollidePoint(Vec3Arg inPoint, const SubShapeIDCreator &inSubShapeIDCreator, CollidePointCollector &ioCollector, const ShapeFilter &inShapeFilter = { }) const = 0;
 
    virtual void                    CollectTransformedShapes(const AABox &inBox, Vec3Arg inPositionCOM, QuatArg inRotation, Vec3Arg inScale, const SubShapeIDCreator &inSubShapeIDCreator, TransformedShapeCollector &ioCollector, const ShapeFilter &inShapeFilter) const;
       
    virtual void                    TransformShape(Mat44Arg inCenterOfMassTransform, TransformedShapeCollector &ioCollector) const;
 
    ShapeResult                     ScaleShape(Vec3Arg inScale) const;
 
    struct alignas(16)              GetTrianglesContext { uint8 mData[4288]; };
 
    static constexpr int            cGetTrianglesMinTrianglesRequested = 32;
 
    virtual void                    GetTrianglesStart(GetTrianglesContext &ioContext, const AABox &inBox, Vec3Arg inPositionCOM, QuatArg inRotation, Vec3Arg inScale) const = 0;
 
    virtual int                     GetTrianglesNext(GetTrianglesContext &ioContext, int inMaxTrianglesRequested, Float3 *outTriangleVertices, const PhysicsMaterial **outMaterials = nullptr) const = 0;
 
 
    virtual void                    SaveBinaryState(StreamOut &inStream) const;
 
    static ShapeResult              sRestoreFromBinaryState(StreamIn &inStream);
 
    virtual void                    SaveMaterialState(PhysicsMaterialList &outMaterials) const          { /* By default do nothing */ }
 
    virtual void                    RestoreMaterialState(const PhysicsMaterialRefC *inMaterials, uint inNumMaterials) { JPH_ASSERT(inNumMaterials == 0); }
 
    virtual void                    SaveSubShapeState(ShapeList &outSubShapes) const                    { /* By default do nothing */ }
 
    virtual void                    RestoreSubShapeState(const ShapeRefC *inSubShapes, uint inNumShapes) { JPH_ASSERT(inNumShapes == 0); }
 
    using ShapeToIDMap = UnorderedMap<const Shape *, uint32>;
    using MaterialToIDMap = UnorderedMap<const PhysicsMaterial *, uint32>;
    using IDToShapeMap = Array<Ref<Shape>>;
    using IDToMaterialMap = Array<Ref<PhysicsMaterial>>;
 
    void                            SaveWithChildren(StreamOut &inStream, ShapeToIDMap &ioShapeMap, MaterialToIDMap &ioMaterialMap) const;
 
    static ShapeResult              sRestoreWithChildren(StreamIn &inStream, IDToShapeMap &ioShapeMap, IDToMaterialMap &ioMaterialMap);
 
 
    struct Stats
    {
                                    Stats(size_t inSizeBytes, uint inNumTriangles) : mSizeBytes(inSizeBytes), mNumTriangles(inNumTriangles) { }
 
        size_t                      mSizeBytes;             
        uint                        mNumTriangles;          
    };
 
    virtual Stats                   GetStats() const = 0;
 
    using VisitedShapes = UnorderedSet<const Shape *>;
 
    virtual Stats                   GetStatsRecursive(VisitedShapes &ioVisitedShapes) const;
 
    virtual float                   GetVolume() const = 0;
 
    virtual bool                    IsValidScale(Vec3Arg inScale) const                                 { return !inScale.IsNearZero(); }
 
#ifdef JPH_DEBUG_RENDERER
    static bool                     sDrawSubmergedVolumes;
#endif // JPH_DEBUG_RENDERER
 
protected:
    virtual void                    RestoreBinaryState(StreamIn &inStream);
 
private:
    uint64                          mUserData = 0;
    EShapeType                      mShapeType;
    EShapeSubType                   mShapeSubType;
};
 
JPH_NAMESPACE_END