413 lines
16 KiB
C++
413 lines
16 KiB
C++
// Copyright 2009-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
#pragma once
|
|
|
|
#include "primitive.h"
|
|
|
|
namespace embree
|
|
{
|
|
template<int M>
|
|
struct PointMi
|
|
{
|
|
/* Virtual interface to query information about the line segment type */
|
|
struct Type : public PrimitiveType
|
|
{
|
|
const char* name() const;
|
|
size_t sizeActive(const char* This) const;
|
|
size_t sizeTotal(const char* This) const;
|
|
size_t getBytes(const char* This) const;
|
|
};
|
|
static Type type;
|
|
|
|
public:
|
|
/* primitive supports multiple time segments */
|
|
static const bool singleTimeSegment = false;
|
|
|
|
/* Returns maximum number of stored line segments */
|
|
static __forceinline size_t max_size()
|
|
{
|
|
return M;
|
|
}
|
|
|
|
/* Returns required number of primitive blocks for N line segments */
|
|
static __forceinline size_t blocks(size_t N)
|
|
{
|
|
return (N + max_size() - 1) / max_size();
|
|
}
|
|
|
|
/* Returns required number of bytes for N line segments */
|
|
static __forceinline size_t bytes(size_t N)
|
|
{
|
|
return blocks(N) * sizeof(PointMi);
|
|
}
|
|
|
|
public:
|
|
/* Default constructor */
|
|
__forceinline PointMi() {}
|
|
|
|
/* Construction from vertices and IDs */
|
|
__forceinline PointMi(const vuint<M>& geomIDs, const vuint<M>& primIDs, Geometry::GType gtype, uint32_t numPrimitives)
|
|
: gtype((unsigned char)gtype),
|
|
numPrimitives(numPrimitives),
|
|
sharedGeomID(geomIDs[0]),
|
|
primIDs(primIDs)
|
|
{
|
|
assert(all(vuint<M>(geomID()) == geomIDs));
|
|
}
|
|
|
|
/* Returns a mask that tells which line segments are valid */
|
|
__forceinline vbool<M> valid() const {
|
|
return vint<M>(step) < vint<M>(numPrimitives);
|
|
}
|
|
|
|
/* Returns if the specified line segment is valid */
|
|
__forceinline bool valid(const size_t i) const
|
|
{
|
|
assert(i < M);
|
|
return i < numPrimitives;
|
|
}
|
|
|
|
/* Returns the number of stored line segments */
|
|
__forceinline size_t size() const {
|
|
return numPrimitives;
|
|
}
|
|
|
|
__forceinline unsigned int geomID(unsigned int i = 0) const {
|
|
return sharedGeomID;
|
|
}
|
|
|
|
__forceinline vuint<M>& primID() {
|
|
return primIDs;
|
|
}
|
|
__forceinline const vuint<M>& primID() const {
|
|
return primIDs;
|
|
}
|
|
__forceinline unsigned int primID(const size_t i) const {
|
|
assert(i < M);
|
|
return primIDs[i];
|
|
}
|
|
|
|
/* gather the line segments */
|
|
__forceinline void gather(Vec4vf<M>& p0, const Points* geom) const;
|
|
__forceinline void gather(Vec4vf<M>& p0, Vec3vf<M>& n0, const Points* geom) const;
|
|
|
|
__forceinline void gatheri(Vec4vf<M>& p0, const Points* geom, const int itime) const;
|
|
__forceinline void gatheri(Vec4vf<M>& p0, Vec3vf<M>& n0, const Points* geom, const int itime) const;
|
|
|
|
__forceinline void gather(Vec4vf<M>& p0, const Points* geom, float time) const;
|
|
__forceinline void gather(Vec4vf<M>& p0, Vec3vf<M>& n0, const Points* geom, float time) const;
|
|
|
|
/* Calculate the bounds of the line segments */
|
|
__forceinline const BBox3fa bounds(const Scene* scene, size_t itime = 0) const
|
|
{
|
|
BBox3fa bounds = empty;
|
|
for (size_t i = 0; i < M && valid(i); i++) {
|
|
const Points* geom = scene->get<Points>(geomID(i));
|
|
bounds.extend(geom->bounds(primID(i),itime));
|
|
}
|
|
return bounds;
|
|
}
|
|
|
|
/* Calculate the linear bounds of the primitive */
|
|
__forceinline LBBox3fa linearBounds(const Scene* scene, size_t itime) {
|
|
return LBBox3fa(bounds(scene, itime + 0), bounds(scene, itime + 1));
|
|
}
|
|
|
|
__forceinline LBBox3fa linearBounds(const Scene* const scene, size_t itime, size_t numTimeSteps)
|
|
{
|
|
LBBox3fa allBounds = empty;
|
|
for (size_t i = 0; i < M && valid(i); i++) {
|
|
const Points* geom = scene->get<Points>(geomID(i));
|
|
allBounds.extend(geom->linearBounds(primID(i), itime, numTimeSteps));
|
|
}
|
|
return allBounds;
|
|
}
|
|
|
|
__forceinline LBBox3fa linearBounds(const Scene* const scene, const BBox1f time_range)
|
|
{
|
|
LBBox3fa allBounds = empty;
|
|
for (size_t i = 0; i < M && valid(i); i++) {
|
|
const Points* geom = scene->get<Points>(geomID((unsigned int)i));
|
|
allBounds.extend(geom->linearBounds(primID(i), time_range));
|
|
}
|
|
return allBounds;
|
|
}
|
|
|
|
/* Fill line segment from line segment list */
|
|
template<typename PrimRefT>
|
|
__forceinline void fill(const PrimRefT* prims, size_t& begin, size_t end, Scene* scene)
|
|
{
|
|
Geometry::GType gty = scene->get(prims[begin].geomID())->getType();
|
|
vuint<M> geomID, primID;
|
|
vuint<M> v0;
|
|
const PrimRefT* prim = &prims[begin];
|
|
|
|
int numPrimitives = 0;
|
|
for (size_t i = 0; i < M; i++) {
|
|
if (begin < end) {
|
|
geomID[i] = prim->geomID();
|
|
primID[i] = prim->primID();
|
|
begin++;
|
|
numPrimitives++;
|
|
} else {
|
|
assert(i);
|
|
if (i > 0) {
|
|
geomID[i] = geomID[i - 1];
|
|
primID[i] = primID[i - 1];
|
|
}
|
|
}
|
|
if (begin < end)
|
|
prim = &prims[begin]; // FIXME: remove this line
|
|
}
|
|
new (this) PointMi(geomID, primID, gty, numPrimitives); // FIXME: use non temporal store
|
|
}
|
|
|
|
template<typename BVH, typename Allocator>
|
|
__forceinline static typename BVH::NodeRef createLeaf(BVH* bvh,
|
|
const PrimRef* prims,
|
|
const range<size_t>& set,
|
|
const Allocator& alloc)
|
|
{
|
|
size_t start = set.begin();
|
|
size_t items = PointMi::blocks(set.size());
|
|
size_t numbytes = PointMi::bytes(set.size());
|
|
PointMi* accel = (PointMi*)alloc.malloc1(numbytes, M * sizeof(float));
|
|
for (size_t i = 0; i < items; i++) {
|
|
accel[i].fill(prims, start, set.end(), bvh->scene);
|
|
}
|
|
return bvh->encodeLeaf((char*)accel, items);
|
|
};
|
|
|
|
__forceinline LBBox3fa fillMB(const PrimRef* prims, size_t& begin, size_t end, Scene* scene, size_t itime)
|
|
{
|
|
fill(prims, begin, end, scene);
|
|
return linearBounds(scene, itime);
|
|
}
|
|
|
|
__forceinline LBBox3fa fillMB(
|
|
const PrimRefMB* prims, size_t& begin, size_t end, Scene* scene, const BBox1f time_range)
|
|
{
|
|
fill(prims, begin, end, scene);
|
|
return linearBounds(scene, time_range);
|
|
}
|
|
|
|
template<typename BVH, typename SetMB, typename Allocator>
|
|
__forceinline static typename BVH::NodeRecordMB4D createLeafMB(BVH* bvh, const SetMB& prims, const Allocator& alloc)
|
|
{
|
|
size_t start = prims.object_range.begin();
|
|
size_t end = prims.object_range.end();
|
|
size_t items = PointMi::blocks(prims.object_range.size());
|
|
size_t numbytes = PointMi::bytes(prims.object_range.size());
|
|
PointMi* accel = (PointMi*)alloc.malloc1(numbytes, M * sizeof(float));
|
|
const typename BVH::NodeRef node = bvh->encodeLeaf((char*)accel, items);
|
|
|
|
LBBox3fa bounds = empty;
|
|
for (size_t i = 0; i < items; i++)
|
|
bounds.extend(accel[i].fillMB(prims.prims->data(), start, end, bvh->scene, prims.time_range));
|
|
|
|
return typename BVH::NodeRecordMB4D(node, bounds, prims.time_range);
|
|
};
|
|
|
|
/*! output operator */
|
|
friend __forceinline embree_ostream operator<<(embree_ostream cout, const PointMi& point)
|
|
{
|
|
return cout << "Point" << M << "i {" << point.geomID() << ", " << point.primID() << "}";
|
|
}
|
|
|
|
public:
|
|
unsigned char gtype;
|
|
unsigned char numPrimitives;
|
|
unsigned int sharedGeomID;
|
|
|
|
private:
|
|
vuint<M> primIDs; // primitive ID
|
|
};
|
|
|
|
template<>
|
|
__forceinline void PointMi<4>::gather(Vec4vf4& p0, const Points* geom) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0)));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1)));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2)));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3)));
|
|
transpose(a0, a1, a2, a3, p0.x, p0.y, p0.z, p0.w);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<4>::gather(Vec4vf4& p0, Vec3vf4& n0, const Points* geom) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0)));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1)));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2)));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3)));
|
|
transpose(a0, a1, a2, a3, p0.x, p0.y, p0.z, p0.w);
|
|
const vfloat4 b0 = vfloat4(geom->normal(primID(0)));
|
|
const vfloat4 b1 = vfloat4(geom->normal(primID(1)));
|
|
const vfloat4 b2 = vfloat4(geom->normal(primID(2)));
|
|
const vfloat4 b3 = vfloat4(geom->normal(primID(3)));
|
|
transpose(b0, b1, b2, b3, n0.x, n0.y, n0.z);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<4>::gatheri(Vec4vf4& p0, const Points* geom, const int itime) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0), itime));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1), itime));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2), itime));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3), itime));
|
|
transpose(a0, a1, a2, a3, p0.x, p0.y, p0.z, p0.w);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<4>::gatheri(Vec4vf4& p0, Vec3vf4& n0, const Points* geom, const int itime) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0), itime));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1), itime));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2), itime));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3), itime));
|
|
transpose(a0, a1, a2, a3, p0.x, p0.y, p0.z, p0.w);
|
|
const vfloat4 b0 = vfloat4(geom->normal((size_t)primID(0), (size_t)itime));
|
|
const vfloat4 b1 = vfloat4(geom->normal((size_t)primID(1), (size_t)itime));
|
|
const vfloat4 b2 = vfloat4(geom->normal((size_t)primID(2), (size_t)itime));
|
|
const vfloat4 b3 = vfloat4(geom->normal((size_t)primID(3), (size_t)itime));
|
|
transpose(b0, b1, b2, b3, n0.x, n0.y, n0.z);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<4>::gather(Vec4vf4& p0, const Points* geom, float time) const
|
|
{
|
|
float ftime;
|
|
const int itime = geom->timeSegment(time, ftime);
|
|
|
|
Vec4vf4 a0; gatheri(a0, geom, itime);
|
|
Vec4vf4 b0; gatheri(b0, geom, itime + 1);
|
|
p0 = lerp(a0, b0, vfloat4(ftime));
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<4>::gather(Vec4vf4& p0, Vec3vf4& n0, const Points* geom, float time) const
|
|
{
|
|
float ftime;
|
|
const int itime = geom->timeSegment(time, ftime);
|
|
|
|
Vec4vf4 a0, b0;
|
|
Vec3vf4 norm0, norm1;
|
|
gatheri(a0, norm0, geom, itime);
|
|
gatheri(b0, norm1, geom, itime + 1);
|
|
p0 = lerp(a0, b0, vfloat4(ftime));
|
|
n0 = lerp(norm0, norm1, vfloat4(ftime));
|
|
}
|
|
|
|
#if defined(__AVX__)
|
|
|
|
template<>
|
|
__forceinline void PointMi<8>::gather(Vec4vf8& p0, const Points* geom) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0)));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1)));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2)));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3)));
|
|
const vfloat4 a4 = vfloat4::loadu(geom->vertexPtr(primID(4)));
|
|
const vfloat4 a5 = vfloat4::loadu(geom->vertexPtr(primID(5)));
|
|
const vfloat4 a6 = vfloat4::loadu(geom->vertexPtr(primID(6)));
|
|
const vfloat4 a7 = vfloat4::loadu(geom->vertexPtr(primID(7)));
|
|
transpose(a0, a1, a2, a3, a4, a5, a6, a7, p0.x, p0.y, p0.z, p0.w);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<8>::gather(Vec4vf8& p0, Vec3vf8& n0, const Points* geom) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0)));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1)));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2)));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3)));
|
|
const vfloat4 a4 = vfloat4::loadu(geom->vertexPtr(primID(4)));
|
|
const vfloat4 a5 = vfloat4::loadu(geom->vertexPtr(primID(5)));
|
|
const vfloat4 a6 = vfloat4::loadu(geom->vertexPtr(primID(6)));
|
|
const vfloat4 a7 = vfloat4::loadu(geom->vertexPtr(primID(7)));
|
|
transpose(a0, a1, a2, a3, a4, a5, a6, a7, p0.x, p0.y, p0.z, p0.w);
|
|
const vfloat4 b0 = vfloat4(geom->normal(primID(0)));
|
|
const vfloat4 b1 = vfloat4(geom->normal(primID(1)));
|
|
const vfloat4 b2 = vfloat4(geom->normal(primID(2)));
|
|
const vfloat4 b3 = vfloat4(geom->normal(primID(3)));
|
|
const vfloat4 b4 = vfloat4(geom->normal(primID(4)));
|
|
const vfloat4 b5 = vfloat4(geom->normal(primID(5)));
|
|
const vfloat4 b6 = vfloat4(geom->normal(primID(6)));
|
|
const vfloat4 b7 = vfloat4(geom->normal(primID(7)));
|
|
transpose(b0, b1, b2, b3, b4, b5, b6, b7, n0.x, n0.y, n0.z);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<8>::gatheri(Vec4vf8& p0, const Points* geom, const int itime) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0), itime));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1), itime));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2), itime));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3), itime));
|
|
const vfloat4 a4 = vfloat4::loadu(geom->vertexPtr(primID(4), itime));
|
|
const vfloat4 a5 = vfloat4::loadu(geom->vertexPtr(primID(5), itime));
|
|
const vfloat4 a6 = vfloat4::loadu(geom->vertexPtr(primID(6), itime));
|
|
const vfloat4 a7 = vfloat4::loadu(geom->vertexPtr(primID(7), itime));
|
|
transpose(a0, a1, a2, a3, a4, a5, a6, a7, p0.x, p0.y, p0.z, p0.w);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<8>::gatheri(Vec4vf8& p0, Vec3vf8& n0, const Points* geom, const int itime) const
|
|
{
|
|
const vfloat4 a0 = vfloat4::loadu(geom->vertexPtr(primID(0), itime));
|
|
const vfloat4 a1 = vfloat4::loadu(geom->vertexPtr(primID(1), itime));
|
|
const vfloat4 a2 = vfloat4::loadu(geom->vertexPtr(primID(2), itime));
|
|
const vfloat4 a3 = vfloat4::loadu(geom->vertexPtr(primID(3), itime));
|
|
const vfloat4 a4 = vfloat4::loadu(geom->vertexPtr(primID(4), itime));
|
|
const vfloat4 a5 = vfloat4::loadu(geom->vertexPtr(primID(5), itime));
|
|
const vfloat4 a6 = vfloat4::loadu(geom->vertexPtr(primID(6), itime));
|
|
const vfloat4 a7 = vfloat4::loadu(geom->vertexPtr(primID(7), itime));
|
|
transpose(a0, a1, a2, a3, a4, a5, a6, a7, p0.x, p0.y, p0.z, p0.w);
|
|
const vfloat4 b0 = vfloat4(geom->normal((size_t)primID(0), (size_t)itime));
|
|
const vfloat4 b1 = vfloat4(geom->normal((size_t)primID(1), (size_t)itime));
|
|
const vfloat4 b2 = vfloat4(geom->normal((size_t)primID(2), (size_t)itime));
|
|
const vfloat4 b3 = vfloat4(geom->normal((size_t)primID(3), (size_t)itime));
|
|
const vfloat4 b4 = vfloat4(geom->normal((size_t)primID(4), (size_t)itime));
|
|
const vfloat4 b5 = vfloat4(geom->normal((size_t)primID(5), (size_t)itime));
|
|
const vfloat4 b6 = vfloat4(geom->normal((size_t)primID(6), (size_t)itime));
|
|
const vfloat4 b7 = vfloat4(geom->normal((size_t)primID(7), (size_t)itime));
|
|
transpose(b0, b1, b2, b3, b4, b5, b6, b7, n0.x, n0.y, n0.z);
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<8>::gather(Vec4vf8& p0, const Points* geom, float time) const
|
|
{
|
|
float ftime;
|
|
const int itime = geom->timeSegment(time, ftime);
|
|
|
|
Vec4vf8 a0;
|
|
gatheri(a0, geom, itime);
|
|
Vec4vf8 b0;
|
|
gatheri(b0, geom, itime + 1);
|
|
p0 = lerp(a0, b0, vfloat8(ftime));
|
|
}
|
|
|
|
template<>
|
|
__forceinline void PointMi<8>::gather(Vec4vf8& p0, Vec3vf8& n0, const Points* geom, float time) const
|
|
{
|
|
float ftime;
|
|
const int itime = geom->timeSegment(time, ftime);
|
|
|
|
Vec4vf8 a0, b0;
|
|
Vec3vf8 norm0, norm1;
|
|
gatheri(a0, norm0, geom, itime);
|
|
gatheri(b0, norm1, geom, itime + 1);
|
|
p0 = lerp(a0, b0, vfloat8(ftime));
|
|
n0 = lerp(norm0, norm1, vfloat8(ftime));
|
|
}
|
|
#endif
|
|
|
|
template<int M>
|
|
typename PointMi<M>::Type PointMi<M>::type;
|
|
|
|
typedef PointMi<4> Point4i;
|
|
typedef PointMi<8> Point8i;
|
|
|
|
} // namespace embree
|