CollisionCollectorImpl.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/Collision/CollisionCollector.h>
#include <Jolt/Core/QuickSort.h>
 
JPH_NAMESPACE_BEGIN
 
template <class CollectorType>
class AllHitCollisionCollector : public CollectorType
{
public:
    using ResultType = typename CollectorType::ResultType;
 
    // See: CollectorType::Reset
    virtual void        Reset() override
    { 
        CollectorType::Reset();
 
        mHits.clear();
    }
 
    // See: CollectorType::AddHit
    virtual void        AddHit(const ResultType &inResult) override
    {
        mHits.push_back(inResult);
    }
 
    void                Sort()
    {
        QuickSort(mHits.begin(), mHits.end(), [](const ResultType &inLHS, const ResultType &inRHS) { return inLHS.GetEarlyOutFraction() < inRHS.GetEarlyOutFraction(); });
    }
 
    inline bool         HadHit() const
    {
        return !mHits.empty();
    }
 
    Array<ResultType>   mHits;
};
 
template <class CollectorType>
class ClosestHitCollisionCollector : public CollectorType
{
public:
    using ResultType = typename CollectorType::ResultType;
 
    // See: CollectorType::Reset
    virtual void        Reset() override
    { 
        CollectorType::Reset();
 
        mHadHit = false;
    }
 
    // See: CollectorType::AddHit
    virtual void        AddHit(const ResultType &inResult) override
    {
        float early_out = inResult.GetEarlyOutFraction();
        if (!mHadHit || early_out < mHit.GetEarlyOutFraction())
        {
            // Update early out fraction
            CollectorType::UpdateEarlyOutFraction(early_out);
 
            // Store hit
            mHit = inResult;
            mHadHit = true;
        }
    }
 
    inline bool         HadHit() const
    {
        return mHadHit;
    }
 
    ResultType          mHit;
 
private:
    bool                mHadHit = false;
};
 
template <class CollectorType>
class AnyHitCollisionCollector : public CollectorType
{
public:
    using ResultType = typename CollectorType::ResultType;
 
    // See: CollectorType::Reset
    virtual void        Reset() override
    { 
        CollectorType::Reset();
 
        mHadHit = false;
    }
 
    // See: CollectorType::AddHit
    virtual void        AddHit(const ResultType &inResult) override
    {
        // Test that the collector is not collecting more hits after forcing an early out
        JPH_ASSERT(!mHadHit);
 
        // Abort any further testing
        CollectorType::ForceEarlyOut();
 
        // Store hit
        mHit = inResult;
        mHadHit = true;
    }
 
    inline bool         HadHit() const
    {
        return mHadHit;
    }
 
    ResultType          mHit;
 
private:
    bool                mHadHit = false;
};
 
JPH_NAMESPACE_END