BodyLockMulti.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/BodyLockInterface.h>
 
JPH_NAMESPACE_BEGIN

template <bool Write, class BodyType>
class BodyLockMultiBase : public NonCopyable
{
public:
    using MutexMask = BodyLockInterface::MutexMask;

                                BodyLockMultiBase(const BodyLockInterface &inBodyLockInterface, const BodyID *inBodyIDs, int inNumber) :
        mBodyLockInterface(inBodyLockInterface),
        mMutexMask(inBodyLockInterface.GetMutexMask(inBodyIDs, inNumber)),
        mBodyIDs(inBodyIDs),
        mNumBodyIDs(inNumber)
    {
        if (mMutexMask != 0)
        {
            // Get mutex
            if (Write)
                inBodyLockInterface.LockWrite(mMutexMask);
            else
                inBodyLockInterface.LockRead(mMutexMask);
        }
    }

    inline void                 ReleaseLocks()
    {
        if (mMutexMask != 0)
        {
            if (Write)
                mBodyLockInterface.UnlockWrite(mMutexMask);
            else
                mBodyLockInterface.UnlockRead(mMutexMask);
 
            mMutexMask = 0;
            mBodyIDs = nullptr;
            mNumBodyIDs = 0;
        }
    }

                                ~BodyLockMultiBase()
    {
        ReleaseLocks();
    }

    inline int                  GetNumBodies() const
    {
        return mNumBodyIDs;
    }

    inline BodyType *           GetBody(int inBodyIndex) const
    {
        // Range check
        JPH_ASSERT(inBodyIndex >= 0 && inBodyIndex < mNumBodyIDs);
 
        // Get body ID
        const BodyID &body_id = mBodyIDs[inBodyIndex];
        if (body_id.IsInvalid())
            return nullptr;
 
        // Get a reference to the body or nullptr when it is no longer valid
        return mBodyLockInterface.TryGetBody(body_id);
    }
 
private:
    const BodyLockInterface &   mBodyLockInterface;
    MutexMask                   mMutexMask;
    const BodyID *              mBodyIDs;
    int                         mNumBodyIDs;
};

class BodyLockMultiRead : public BodyLockMultiBase<false, const Body>
{
    using BodyLockMultiBase::BodyLockMultiBase;
};

class BodyLockMultiWrite : public BodyLockMultiBase<true, Body>
{
    using BodyLockMultiBase::BodyLockMultiBase;
};
 
JPH_NAMESPACE_END