Method segment

src/IECoreScene/MeshAlgoSegment.cpp:48–102

using namespace IECoreScene;

std::vector<MeshPrimitivePtr> IECoreScene::MeshAlgo::segment( const MeshPrimitive *mesh, const PrimitiveVariable &primitiveVariable, const IECore::Data *segmentValues, const Canceller *canceller )
{
	MeshSplitter ms( mesh, primitiveVariable, canceller );
	int numMeshes = ms.numMeshes();
	std::vector<MeshPrimitivePtr> ret;
	ret.reserve( numMeshes );

	if( !segmentValues )
	{
		for( int i = 0; i < numMeshes; i++ )
		{
			ret.push_back( ms.mesh( i ) );
		}
	}
	else
	{
		IECore::dispatch( segmentValues,
			[ ms, numMeshes, &ret, canceller ]( auto *typedSegmentValues )
			{
				using DataType = typename std::remove_pointer_t< decltype( typedSegmentValues ) >;
				if constexpr ( TypeTraits::IsVectorTypedData<DataType>::value )
				{
					using ElementType = typename DataType::ValueType::value_type;

					std::unordered_map< ElementType, int > idMap;
					Canceller::check( canceller );
					for( int i = 0; i < numMeshes; i++ )
					{
						idMap[ ms.value<ElementType>( i ) ] = i;
					}

					Canceller::check( canceller );
					for( const auto &i : typedSegmentValues->readable() )
					{
						auto f = idMap.find( i );
						if( f == idMap.end() )
						{
							ret.push_back( nullptr );
						}
						else
						{
							ret.push_back( ms.mesh( idMap[i] ) );
						}
					}
				}
				else
				{
					throw IECore::Exception( "Invalid Primitive Variable with non-vector typed data." );
				}
			}
		);
	}

	return ret;
}