Host-side shape registry (Layer 1). More...

#include <shape_registry.hpp>

Inheritance diagram for peclet::dem::ShapeRegistry:

Public Member Functions
void	setSphereShape (float radius)

void	initializeShape (int shape_type, float radius, float height, float thickness)

int	addShape (int shape_type, float radius, float height, float thickness)
	Append an analytic shape and return its index, for simulations with a MIXTURE of shapes.

void	setShapeIds (const std::vector< int > &ids)
	Per-particle shape assignment.

int	addSdfShape (const std::vector< float > &grid, int nx, int ny, int nz, F3 origin, F3 spacing, const std::vector< float > &shellFlat, F3 invInertia, float boundingRadius)
	Append a grid-SDF shape (the general non-spherical particle) to the mixture and return its index.

int	numShapes () const

void	setSdfShape (const std::vector< float > &grid, int nx, int ny, int nz, F3 origin, F3 spacing, const std::vector< float > &shellFlat, F3 invInertia, float boundingRadius)

int	addSceneShape (const std::vector< int > &nodeInts, const std::vector< float > &nodeReals, int root, const std::vector< float > &shellFlat, F3 invInertia, float boundingRadius)

std::vector< float >	autoShell (const std::vector< float > &grid, int nx, int ny, int nz, F3 origin, F3 spacing, float boundingRadius) const
	Generate a collision shell for a grid SDF by sampling its own zero level set, so a caller does not have to supply one.

Protected Member Functions
void	ensureContactCapacity ()

void	clearShapes ()

void	appendShape (ShapeDesc sd, const std::vector< F3 > &shell, F3 invI, float baseRadius)
	Append one shape, taking ownership of its shell.

void	uploadShapes ()
	Rebuild the device Views from the host registry.

Protected Attributes
Particles	P_

float	baseRadius_ = 1.0f

std::vector< ShapeDesc >	shapesHost_

std::vector< F3 >	shellHost_

std::vector< F3 >	invIHost_

std::vector< float >	baseRadiusHost_

std::vector< float >	sdfSamplesHost_

int	shellPoints_ = 0

std::vector< peclet::core::geom::ShapeNode< float > >	shapeNodesHost_

F3	defaultInvI_ {2.5f, 2.5f, 2.5f}

Detailed Description

Host-side shape registry (Layer 1).

The device Views (P_.shapes / P_.shell / P_.sdfGrid) are rebuilt from the host vectors by uploadShapes(), which is what assigns each shape its shellOffset and grid offset into the shared pools. Keeping the authoritative copy on the host is what makes a MIXTURE of shapes possible at all: the old code wrote descriptor slot 0 directly and had nowhere to put a second shell.

Definition at line 32 of file shape_registry.hpp.

Member Function Documentation

◆ setSphereShape()

void peclet::dem::ShapeRegistry::setSphereShape ( float radius )

inline

Definition at line 34 of file shape_registry.hpp.

References initializeShape(), and peclet::dem::SPHERE.

Referenced by peclet::dem::Simulation::Simulation().

◆ initializeShape()

void peclet::dem::ShapeRegistry::initializeShape	(	int	shape_type,
		float	radius,
		float	height,
		float	thickness
	)

inline

Definition at line 47 of file shape_registry.hpp.

References appendShape(), peclet::dem::BOX, clearShapes(), ensureContactCapacity(), peclet::dem::genBoxShell(), peclet::dem::genCylinderShell(), peclet::dem::HOLLOW_CYLINDER, peclet::dem::SPHERE, and uploadShapes().

Referenced by addShape(), NB_MODULE(), and setSphereShape().

◆ addShape()

int peclet::dem::ShapeRegistry::addShape	(	int	shape_type,
		float	radius,
		float	height,
		float	thickness
	)

inline

Append an analytic shape and return its index, for simulations with a MIXTURE of shapes.

Same arguments as initializeShape, which stays the "single shape" entry point (it resets the registry). Assign the returned index to particles with setShapeIds().

Definition at line 115 of file shape_registry.hpp.

References appendShape(), baseRadiusHost_, ensureContactCapacity(), initializeShape(), invIHost_, sdfSamplesHost_, shapesHost_, shellHost_, and uploadShapes().

Referenced by NB_MODULE().

◆ setShapeIds()

void peclet::dem::ShapeRegistry::setShapeIds ( const std::vector< int > & ids )

inline

Per-particle shape assignment.

Also refreshes each particle's inverse inertia from its new shape, so the order of setPositions/setShapeIds does not matter.

Definition at line 144 of file shape_registry.hpp.

References invIHost_, peclet::dem::Particles::invInertia, peclet::dem::Particles::numReal, P_, peclet::dem::Particles::shapeId, and shapesHost_.

Referenced by NB_MODULE().

◆ addSdfShape()

int peclet::dem::ShapeRegistry::addSdfShape	(	const std::vector< float > &	grid,
		int	nx,
		int	ny,
		int	nz,
		F3	origin,
		F3	spacing,
		const std::vector< float > &	shellFlat,
		F3	invInertia,
		float	boundingRadius
	)

inline

Append a grid-SDF shape (the general non-spherical particle) to the mixture and return its index.

Same arguments as set_sdf_shape, which stays the single-shape entry point. The samples are appended to the shared pool and the descriptor's offset set accordingly.

Definition at line 168 of file shape_registry.hpp.

References appendShape(), autoShell(), ensureContactCapacity(), sdfSamplesHost_, peclet::dem::SHAPE_GRID_SDF, shapesHost_, peclet::dem::toCoreVec(), peclet::dem::ShapeDesc::type, uploadShapes(), peclet::dem::F3::x, peclet::dem::F3::y, and peclet::dem::F3::z.

Referenced by NB_MODULE().

◆ numShapes()

int peclet::dem::ShapeRegistry::numShapes ( ) const

inline

Definition at line 206 of file shape_registry.hpp.

References shapesHost_.

Referenced by NB_MODULE().

◆ setSdfShape()

void peclet::dem::ShapeRegistry::setSdfShape	(	const std::vector< float > &	grid,
		int	nx,
		int	ny,
		int	nz,
		F3	origin,
		F3	spacing,
		const std::vector< float > &	shellFlat,
		F3	invInertia,
		float	boundingRadius
	)

inline

Definition at line 219 of file shape_registry.hpp.

References appendShape(), autoShell(), clearShapes(), ensureContactCapacity(), P_, peclet::dem::Particles::sdfGrid, sdfSamplesHost_, peclet::dem::SHAPE_GRID_SDF, peclet::dem::Particles::shell, peclet::dem::toCoreVec(), peclet::dem::ShapeDesc::type, uploadShapes(), peclet::dem::F3::x, peclet::dem::F3::y, and peclet::dem::F3::z.

Referenced by NB_MODULE().

◆ addSceneShape()

int peclet::dem::ShapeRegistry::addSceneShape	(	const std::vector< int > &	nodeInts,
		const std::vector< float > &	nodeReals,
		int	root,
		const std::vector< float > &	shellFlat,
		F3	invInertia,
		float	boundingRadius
	)

inline

Definition at line 285 of file shape_registry.hpp.

References appendShape(), ensureContactCapacity(), peclet::dem::SHAPE_SCENE, shapeNodesHost_, shapesHost_, peclet::dem::ShapeDesc::type, and uploadShapes().

Referenced by NB_MODULE().

◆ autoShell()

std::vector< float > peclet::dem::ShapeRegistry::autoShell	(	const std::vector< float > &	grid,
		int	nx,
		int	ny,
		int	nz,
		F3	origin,
		F3	spacing,
		float	boundingRadius
	)		const

inline

Generate a collision shell for a grid SDF by sampling its own zero level set, so a caller does not have to supply one.

Layer 1: previously every shape needed a hand-written generator (there were only two, for the cylinder and the box) or a marching-cubes shell computed in Python; core's surfacePoints() is driven by the SDF itself and works for any geometry.

Definition at line 334 of file shape_registry.hpp.

References peclet::dem::F3::x, peclet::dem::F3::y, and peclet::dem::F3::z.

Referenced by addSdfShape(), and setSdfShape().