302 lines
9.1 KiB
C
302 lines
9.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 PXG_ISOSURFACE_EXTRACTION_H
|
||
|
|
#define PXG_ISOSURFACE_EXTRACTION_H
|
||
|
|
|
||
|
|
|
||
|
|
#include "foundation/PxSimpleTypes.h"
|
||
|
|
#include "foundation/PxVec4.h"
|
||
|
|
|
||
|
|
#include "PxSparseGridParams.h"
|
||
|
|
|
||
|
|
#include "foundation/PxArray.h"
|
||
|
|
|
||
|
|
#include "PxIsosurfaceExtraction.h"
|
||
|
|
|
||
|
|
#include "PxgSparseGridStandalone.h"
|
||
|
|
#include "PxgAlgorithms.h"
|
||
|
|
#include "PxgIsosurfaceData.h"
|
||
|
|
#include "PxgKernelLauncher.h"
|
||
|
|
|
||
|
|
#if !PX_DOXYGEN
|
||
|
|
namespace physx
|
||
|
|
{
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#if PX_SUPPORT_GPU_PHYSX
|
||
|
|
|
||
|
|
class PxgSharedIsosurfaceExtractor
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
bool mEnabled;
|
||
|
|
PxIsosurfaceParams mIsosurfaceParams;
|
||
|
|
|
||
|
|
PxGpuScan mScan;
|
||
|
|
|
||
|
|
PxgKernelLauncher mKernelLauncher;
|
||
|
|
PxU32* mNumVerticesNumIndices;
|
||
|
|
|
||
|
|
bool mOwnsOutputGPUBuffers;
|
||
|
|
PxVec4* mVertices;
|
||
|
|
PxVec4* mNormals;
|
||
|
|
PxU32* mTriIndices;
|
||
|
|
|
||
|
|
//public:
|
||
|
|
PxgSharedIsosurfaceExtractor() : mEnabled(true), mKernelLauncher(), mNumVerticesNumIndices(NULL), mOwnsOutputGPUBuffers(false),
|
||
|
|
mVertices(NULL), mNormals(NULL), mTriIndices(NULL)
|
||
|
|
{}
|
||
|
|
|
||
|
|
virtual ~PxgSharedIsosurfaceExtractor() {}
|
||
|
|
|
||
|
|
template<typename DenseOrSparseGpuDataPackage>
|
||
|
|
void extractIso(DenseOrSparseGpuDataPackage& mData, PxVec4* deviceParticlePos, const PxU32 numParticles, CUstream stream, PxU32* phases, PxU32 validPhaseMask,
|
||
|
|
PxU32* activeIndices = NULL, PxVec4* anisotropy1 = NULL, PxVec4* anisotropy2 = NULL, PxVec4* anisotropy3 = NULL, PxReal anisotropyFactor = 1.0f);
|
||
|
|
|
||
|
|
template<typename DenseOrSparseGpuDataPackage>
|
||
|
|
void meshFromDensity(DenseOrSparseGpuDataPackage& mData, CUstream stream);
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
\brief GPU based isosurface extractor operating on a sparse grid
|
||
|
|
*/
|
||
|
|
class PxgSparseGridIsosurfaceExtractor : public PxSparseGridIsosurfaceExtractor, public PxUserAllocated
|
||
|
|
{
|
||
|
|
protected:
|
||
|
|
PxgSharedIsosurfaceExtractor mShared;
|
||
|
|
PxSparseIsosurfaceExtractionData mData;
|
||
|
|
PxSparseGridBuilder mSparseGrid;
|
||
|
|
|
||
|
|
void paramsToMCData();
|
||
|
|
|
||
|
|
virtual void setMaxVerticesAndTriangles(PxU32 maxIsosurfaceVertices, PxU32 maxIsosurfaceTriangles);
|
||
|
|
|
||
|
|
virtual void releaseGPUBuffers();
|
||
|
|
|
||
|
|
virtual void allocateGPUBuffers();
|
||
|
|
|
||
|
|
public:
|
||
|
|
PxgSparseGridIsosurfaceExtractor() : mShared()
|
||
|
|
{}
|
||
|
|
|
||
|
|
PxgSparseGridIsosurfaceExtractor(PxgKernelLauncher& cudaContextManager, const PxSparseGridParams sparseGridParams,
|
||
|
|
const PxIsosurfaceParams& isosurfaceParams, PxU32 maxNumParticles, PxU32 maxNumVertices, PxU32 maxNumTriangles) : mShared()
|
||
|
|
{
|
||
|
|
initialize(cudaContextManager, sparseGridParams, isosurfaceParams, maxNumParticles, maxNumVertices, maxNumTriangles);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
virtual void setResultBufferDevice(PxVec4* vertices, PxU32* triIndices, PxVec4* normals);
|
||
|
|
|
||
|
|
virtual ~PxgSparseGridIsosurfaceExtractor() { }
|
||
|
|
|
||
|
|
void initialize(PxgKernelLauncher& cudaContextManager, const PxSparseGridParams sparseGridParams,
|
||
|
|
const PxIsosurfaceParams& isosurfaceParams, PxU32 maxNumParticles, PxU32 maxNumVertices, PxU32 maxNumTriangles);
|
||
|
|
|
||
|
|
virtual void release();
|
||
|
|
|
||
|
|
virtual void setIsosurfaceParams(const PxIsosurfaceParams& params)
|
||
|
|
{
|
||
|
|
mShared.mIsosurfaceParams = params;
|
||
|
|
paramsToMCData();
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void clearDensity(CUstream stream);
|
||
|
|
|
||
|
|
virtual PxU32 getMaxParticles() const
|
||
|
|
{
|
||
|
|
return mSparseGrid.getMaxParticles();
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getMaxVertices() const
|
||
|
|
{
|
||
|
|
return mData.maxVerts;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getMaxTriangles() const
|
||
|
|
{
|
||
|
|
return mData.maxTriIds / 3;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void setMaxParticles(PxU32 maxParticles);
|
||
|
|
|
||
|
|
virtual void extractIsosurface(PxVec4* deviceParticlePos, const PxU32 numParticles, CUstream stream, PxU32* phases = NULL, PxU32 validPhaseMask = PxParticlePhaseFlag::eParticlePhaseFluid,
|
||
|
|
PxU32* activeIndices = NULL, PxVec4* anisotropy1 = NULL, PxVec4* anisotropy2 = NULL, PxVec4* anisotropy3 = NULL, PxReal anisotropyFactor = 1.0f);
|
||
|
|
|
||
|
|
virtual void setResultBufferHost(PxVec4* vertices, PxU32* triIndices, PxVec4* normals);
|
||
|
|
|
||
|
|
virtual PxIsosurfaceParams getIsosurfaceParams() const
|
||
|
|
{
|
||
|
|
return mShared.mIsosurfaceParams;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getNumVertices() const
|
||
|
|
{
|
||
|
|
if (!mShared.mNumVerticesNumIndices)
|
||
|
|
return 0;
|
||
|
|
return mShared.mNumVerticesNumIndices[0];
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getNumTriangles() const
|
||
|
|
{
|
||
|
|
if (!mShared.mNumVerticesNumIndices)
|
||
|
|
return 0;
|
||
|
|
return mShared.mNumVerticesNumIndices[1] / 3;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void setEnabled(bool enabled)
|
||
|
|
{
|
||
|
|
mShared.mEnabled = enabled;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual bool isEnabled() const
|
||
|
|
{
|
||
|
|
return mShared.mEnabled;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxSparseGridParams getSparseGridParams() const
|
||
|
|
{
|
||
|
|
return mSparseGrid.getGridParameters();
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void setSparseGridParams(const PxSparseGridParams& params)
|
||
|
|
{
|
||
|
|
mSparseGrid.setGridParameters(params);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
\brief GPU based isosurface extractor operating on a dense grid
|
||
|
|
*/
|
||
|
|
class PxgDenseGridIsosurfaceExtractor : public PxIsosurfaceExtractor, public PxUserAllocated
|
||
|
|
{
|
||
|
|
protected:
|
||
|
|
PxgSharedIsosurfaceExtractor mShared;
|
||
|
|
PxIsosurfaceExtractionData mData;
|
||
|
|
|
||
|
|
PxU32 mMaxParticles; //For compatibility with sparse grid isosurface extractor. There is no upper particle limit on the dense grid extractor.
|
||
|
|
|
||
|
|
void paramsToMCData();
|
||
|
|
|
||
|
|
virtual void setMaxVerticesAndTriangles(PxU32 maxIsosurfaceVertices, PxU32 maxIsosurfaceTriangles);
|
||
|
|
|
||
|
|
virtual void releaseGPUBuffers();
|
||
|
|
|
||
|
|
virtual void allocateGPUBuffers();
|
||
|
|
|
||
|
|
public:
|
||
|
|
PxgDenseGridIsosurfaceExtractor() : mShared()
|
||
|
|
{}
|
||
|
|
|
||
|
|
PxgDenseGridIsosurfaceExtractor(PxgKernelLauncher& cudaContextManager, const PxBounds3& worldBounds,
|
||
|
|
PxReal cellSize, const PxIsosurfaceParams& isosurfaceParams, PxU32 maxNumParticles, PxU32 maxNumVertices, PxU32 maxNumTriangles) : mShared()
|
||
|
|
{
|
||
|
|
initialize(cudaContextManager, worldBounds, cellSize, isosurfaceParams, maxNumParticles, maxNumVertices, maxNumTriangles);
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void setResultBufferDevice(PxVec4* vertices, PxU32* triIndices, PxVec4* normals);
|
||
|
|
|
||
|
|
virtual ~PxgDenseGridIsosurfaceExtractor() { }
|
||
|
|
|
||
|
|
void initialize(PxgKernelLauncher& cudaContextManager, const PxBounds3& worldBounds,
|
||
|
|
PxReal cellSize, const PxIsosurfaceParams& isosurfaceParams, PxU32 maxNumParticles, PxU32 maxNumVertices, PxU32 maxNumTriangles);
|
||
|
|
|
||
|
|
virtual void release();
|
||
|
|
|
||
|
|
virtual void setIsosurfaceParams(const PxIsosurfaceParams& params)
|
||
|
|
{
|
||
|
|
mShared.mIsosurfaceParams = params;
|
||
|
|
paramsToMCData();
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void clearDensity(CUstream stream);
|
||
|
|
|
||
|
|
virtual PxU32 getMaxParticles() const
|
||
|
|
{
|
||
|
|
return mMaxParticles;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getMaxVertices() const
|
||
|
|
{
|
||
|
|
return mData.maxVerts;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getMaxTriangles() const
|
||
|
|
{
|
||
|
|
return mData.maxTriIds / 3;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void setMaxParticles(PxU32 maxParticles)
|
||
|
|
{
|
||
|
|
//No need to resize internal buffers on the dense grid isosurface;
|
||
|
|
mMaxParticles = maxParticles;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void extractIsosurface(PxVec4* deviceParticlePos, const PxU32 numParticles, CUstream stream, PxU32* phases = NULL, PxU32 validPhaseMask = PxParticlePhaseFlag::eParticlePhaseFluid,
|
||
|
|
PxU32* activeIndices = NULL, PxVec4* anisotropy1 = NULL, PxVec4* anisotropy2 = NULL, PxVec4* anisotropy3 = NULL, PxReal anisotropyFactor = 1.0f);
|
||
|
|
|
||
|
|
virtual void setResultBufferHost(PxVec4* vertices, PxU32* triIndices, PxVec4* normals);
|
||
|
|
|
||
|
|
virtual PxIsosurfaceParams getIsosurfaceParams() const
|
||
|
|
{
|
||
|
|
return mShared.mIsosurfaceParams;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getNumVertices() const
|
||
|
|
{
|
||
|
|
if (!mShared.mNumVerticesNumIndices)
|
||
|
|
return 0;
|
||
|
|
return mShared.mNumVerticesNumIndices[0];
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual PxU32 getNumTriangles() const
|
||
|
|
{
|
||
|
|
if (!mShared.mNumVerticesNumIndices)
|
||
|
|
return 0;
|
||
|
|
return mShared.mNumVerticesNumIndices[1] / 3;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void setEnabled(bool enabled)
|
||
|
|
{
|
||
|
|
mShared.mEnabled = enabled;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual bool isEnabled() const
|
||
|
|
{
|
||
|
|
return mShared.mEnabled;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#if !PX_DOXYGEN
|
||
|
|
} // namespace physx
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#endif
|