BodyFilter.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/Core/NonCopyable.h>
#include <Jolt/Physics/Body/BodyID.h>
 
JPH_NAMESPACE_BEGIN
 
class Body;
 
class JPH_EXPORT BodyFilter : public NonCopyable
{
public:
    virtual                 ~BodyFilter() = default;
 
    virtual bool            ShouldCollide([[maybe_unused]] const BodyID &inBodyID) const
    {
        return true;
    }
 
    virtual bool            ShouldCollideLocked([[maybe_unused]] const Body &inBody) const
    {
        return true;
    }
};
 
class JPH_EXPORT IgnoreSingleBodyFilter : public BodyFilter
{
public:
    explicit                IgnoreSingleBodyFilter(const BodyID &inBodyID) :
        mBodyID(inBodyID)
    {
    }
 
    virtual bool            ShouldCollide(const BodyID &inBodyID) const override
    {
        return mBodyID != inBodyID;
    }
 
private:
    BodyID                  mBodyID;
};
 
class JPH_EXPORT IgnoreMultipleBodiesFilter : public BodyFilter
{
public:
    void                    Clear()
    {
        mBodyIDs.clear();
    }
 
    void                    Reserve(uint inSize)
    {
        mBodyIDs.reserve(inSize);
    }
 
    void                    IgnoreBody(const BodyID &inBodyID)
    {
        mBodyIDs.push_back(inBodyID);
    }
 
    virtual bool            ShouldCollide(const BodyID &inBodyID) const override
    {
        return std::find(mBodyIDs.begin(), mBodyIDs.end(), inBodyID) == mBodyIDs.end();
    }
 
private:
    Array<BodyID>           mBodyIDs;
};
 
class JPH_EXPORT IgnoreSingleBodyFilterChained : public BodyFilter
{
public:
    explicit                IgnoreSingleBodyFilterChained(const BodyID inBodyID, const BodyFilter &inFilter) :
        mBodyID(inBodyID),
        mFilter(inFilter)
    {
    }
 
    virtual bool            ShouldCollide(const BodyID &inBodyID) const override
    {
        return inBodyID != mBodyID && mFilter.ShouldCollide(inBodyID);
    }
 
    virtual bool            ShouldCollideLocked(const Body &inBody) const override
    {
        return mFilter.ShouldCollideLocked(inBody);
    }
 
private:
    BodyID                  mBodyID;
    const BodyFilter &      mFilter;
};
 
#ifdef JPH_DEBUG_RENDERER
class JPH_EXPORT BodyDrawFilter : public NonCopyable
{
public:
    virtual                 ~BodyDrawFilter() = default;
 
    virtual bool            ShouldDraw([[maybe_unused]] const Body& inBody) const
    {
        return true;
    }
};
#endif // JPH_DEBUG_RENDERER
 
JPH_NAMESPACE_END