140 lines
6.1 KiB
C++
140 lines
6.1 KiB
C++
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions
|
|
// are met:
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above copyright
|
|
// notice, this list of conditions and the following disclaimer in the
|
|
// documentation and/or other materials provided with the distribution.
|
|
// * Neither the name of NVIDIA CORPORATION nor the names of its
|
|
// contributors may be used to endorse or promote products derived
|
|
// from this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ''AS IS'' AND ANY
|
|
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
|
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
|
// OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
// Copyright (c) 2008-2025 NVIDIA Corporation. All rights reserved.
|
|
// Copyright (c) 2004-2008 AGEIA Technologies, Inc. All rights reserved.
|
|
// Copyright (c) 2001-2004 NovodeX AG. All rights reserved.
|
|
|
|
#ifndef EXT_SQ_MANAGER_H
|
|
#define EXT_SQ_MANAGER_H
|
|
|
|
#include "common/PxPhysXCommonConfig.h"
|
|
|
|
#include "foundation/PxBitMap.h"
|
|
#include "foundation/PxArray.h"
|
|
#include "SqPruner.h"
|
|
#include "SqManager.h"
|
|
|
|
#include "foundation/PxHashSet.h"
|
|
namespace physx
|
|
{
|
|
namespace Sq
|
|
{
|
|
class CompoundPruner;
|
|
}
|
|
}
|
|
|
|
#include "foundation/PxMutex.h"
|
|
|
|
namespace physx
|
|
{
|
|
class PxRenderOutput;
|
|
class PxBVH;
|
|
class PxSceneLimits;
|
|
|
|
namespace Gu
|
|
{
|
|
class BVH;
|
|
}
|
|
|
|
namespace Sq
|
|
{
|
|
// PT: this is a customized version of physx::Sq::PrunerManager that supports more than 2 hardcoded pruners.
|
|
// It might not be possible to support the whole PxSceneQuerySystem API with an arbitrary number of pruners.
|
|
class ExtPrunerManager : public PxUserAllocated
|
|
{
|
|
public:
|
|
ExtPrunerManager(PxU64 contextID, float inflation, const Adapter& adapter, bool usesTreeOfPruners);
|
|
~ExtPrunerManager();
|
|
|
|
PxU32 addPruner(Gu::Pruner* pruner, PxU32 preallocated);
|
|
|
|
Gu::PrunerHandle addPrunerShape(const Gu::PrunerPayload& payload, PxU32 prunerIndex, bool dynamic, PrunerCompoundId compoundId, const PxBounds3& bounds, const PxTransform& transform, bool hasPruningStructure=false);
|
|
void addCompoundShape(const PxBVH& bvh, PrunerCompoundId compoundId, const PxTransform& compoundTransform, Gu::PrunerHandle* prunerHandle, const Gu::PrunerPayload* payloads, const PxTransform* transforms, bool isDynamic);
|
|
void markForUpdate(PxU32 prunerIndex, bool dynamic, PrunerCompoundId compoundId, Gu::PrunerHandle shapeHandle, const PxTransform& transform);
|
|
void removePrunerShape(PxU32 prunerIndex, bool dynamic, PrunerCompoundId compoundId, Gu::PrunerHandle shapeHandle, Gu::PrunerPayloadRemovalCallback* removalCallback);
|
|
|
|
PX_FORCE_INLINE PxU32 getNbPruners() const { return mPrunerExt.size(); }
|
|
PX_FORCE_INLINE const Gu::Pruner* getPruner(PxU32 index) const { return mPrunerExt[index]->mPruner; }
|
|
PX_FORCE_INLINE Gu::Pruner* getPruner(PxU32 index) { return mPrunerExt[index]->mPruner; }
|
|
PX_FORCE_INLINE const CompoundPruner* getCompoundPruner() const { return mCompoundPrunerExt.mPruner; }
|
|
PX_FORCE_INLINE PxU64 getContextId() const { return mContextID; }
|
|
|
|
void preallocate(PxU32 prunerIndex, PxU32 nbShapes);
|
|
|
|
void setDynamicTreeRebuildRateHint(PxU32 dynTreeRebuildRateHint);
|
|
PX_FORCE_INLINE PxU32 getDynamicTreeRebuildRateHint() const { return mRebuildRateHint; }
|
|
|
|
void flushUpdates();
|
|
void forceRebuildDynamicTree(PxU32 prunerIndex);
|
|
|
|
void updateCompoundActor(PrunerCompoundId compoundId, const PxTransform& compoundTransform);
|
|
void removeCompoundActor(PrunerCompoundId compoundId, Gu::PrunerPayloadRemovalCallback* removalCallback);
|
|
|
|
void* prepareSceneQueriesUpdate(PxU32 prunerIndex);
|
|
void sceneQueryBuildStep(void* handle);
|
|
|
|
void sync(PxU32 prunerIndex, const Gu::PrunerHandle* handles, const PxU32* boundsIndices, const PxBounds3* bounds, const PxTransform32* transforms, PxU32 count, const PxBitMap& ignoredIndices);
|
|
void afterSync(bool buildStep, bool commit);
|
|
void shiftOrigin(const PxVec3& shift);
|
|
void visualize(PxU32 prunerIndex, PxRenderOutput& out) const;
|
|
|
|
void flushMemory();
|
|
PX_FORCE_INLINE PxU32 getStaticTimestamp() const { return mStaticTimestamp; }
|
|
PX_FORCE_INLINE const Adapter& getAdapter() const { return mAdapter; }
|
|
PX_FORCE_INLINE const Gu::BVH* getTreeOfPruners() const { return mTreeOfPruners; }
|
|
|
|
PxU32 startCustomBuildstep();
|
|
void customBuildstep(PxU32 index);
|
|
void finishCustomBuildstep();
|
|
|
|
void createTreeOfPruners();
|
|
private:
|
|
const Adapter& mAdapter;
|
|
PxArray<PrunerExt*> mPrunerExt;
|
|
CompoundPrunerExt mCompoundPrunerExt;
|
|
|
|
Gu::BVH* mTreeOfPruners;
|
|
|
|
const PxU64 mContextID;
|
|
PxU32 mStaticTimestamp;
|
|
PxU32 mRebuildRateHint;
|
|
const float mInflation; // SQ_PRUNER_EPSILON
|
|
|
|
PxMutex mSQLock; // to make sure only one query updates the dirty pruner structure if multiple queries run in parallel
|
|
|
|
volatile bool mPrunerNeedsUpdating;
|
|
volatile bool mTimestampNeedsUpdating;
|
|
const bool mUsesTreeOfPruners;
|
|
|
|
void flushShapes();
|
|
PX_FORCE_INLINE void invalidateStaticTimestamp() { mStaticTimestamp++; }
|
|
|
|
PX_NOCOPY(ExtPrunerManager)
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif
|