// Copyright 2009-2021 Intel Corporation // SPDX-License-Identifier: Apache-2.0 #pragma once #include "priminfo.h" #include "priminfo_mb.h" #include "../../common/algorithms/parallel_reduce.h" #include "../../common/algorithms/parallel_partition.h" namespace embree { namespace isa { /*! mapping into bins */ template struct BinMapping { public: __forceinline BinMapping() {} /*! calculates the mapping */ __forceinline BinMapping(size_t N, const BBox3fa& centBounds) { num = min(BINS,size_t(4.0f + 0.05f*N)); assert(num >= 1); const vfloat4 eps = 1E-34f; const vfloat4 diag = max(eps, (vfloat4) centBounds.size()); scale = select(diag > eps,vfloat4(0.99f*num)/diag,vfloat4(0.0f)); ofs = (vfloat4) centBounds.lower; } /*! calculates the mapping */ __forceinline BinMapping(const BBox3fa& centBounds) { num = BINS; const vfloat4 eps = 1E-34f; const vfloat4 diag = max(eps, (vfloat4) centBounds.size()); scale = select(diag > eps,vfloat4(0.99f*num)/diag,vfloat4(0.0f)); ofs = (vfloat4) centBounds.lower; } /*! calculates the mapping */ template __forceinline BinMapping(const PrimInfo& pinfo) { const vfloat4 eps = 1E-34f; num = min(BINS,size_t(4.0f + 0.05f*pinfo.size())); const vfloat4 diag = max(eps,(vfloat4) pinfo.centBounds.size()); scale = select(diag > eps,vfloat4(0.99f*num)/diag,vfloat4(0.0f)); ofs = (vfloat4) pinfo.centBounds.lower; } /*! returns number of bins */ __forceinline size_t size() const { return num; } /*! slower but safe binning */ __forceinline Vec3ia bin(const Vec3fa& p) const { const vint4 i = floori((vfloat4(p)-ofs)*scale); assert(i[0] >= 0 && (size_t)i[0] < num); assert(i[1] >= 0 && (size_t)i[1] < num); assert(i[2] >= 0 && (size_t)i[2] < num); // we clamp to handle corner cases that could calculate out of bounds bin return Vec3ia(clamp(i,vint4(0),vint4(num-1))); } /*! faster but unsafe binning */ __forceinline Vec3ia bin_unsafe(const Vec3fa& p) const { return Vec3ia(floori((vfloat4(p)-ofs)*scale)); } /*! faster but unsafe binning */ template __forceinline Vec3ia bin_unsafe(const PrimRef& p) const { return bin_unsafe(p.binCenter()); } /*! faster but unsafe binning */ template __forceinline Vec3ia bin_unsafe(const PrimRef& p, const BinBoundsAndCenter& binBoundsAndCenter) const { return bin_unsafe(binBoundsAndCenter.binCenter(p)); } template __forceinline bool bin_unsafe(const PrimRef& ref, const vint4& vSplitPos, const vbool4& splitDimMask) const // FIXME: rename to isLeft { return any(((vint4)bin_unsafe(center2(ref.bounds())) < vSplitPos) & splitDimMask); } /*! calculates left spatial position of bin */ __forceinline float pos(const size_t bin, const size_t dim) const { return madd(float(bin),1.0f / scale[dim],ofs[dim]); } /*! returns true if the mapping is invalid in some dimension */ __forceinline bool invalid(const size_t dim) const { return scale[dim] == 0.0f; } /*! stream output */ friend embree_ostream operator<<(embree_ostream cout, const BinMapping& mapping) { return cout << "BinMapping { num = " << mapping.num << ", ofs = " << mapping.ofs << ", scale = " << mapping.scale << "}"; } public: size_t num; vfloat4 ofs,scale; //!< linear function that maps to bin ID }; /*! stores all information to perform some split */ template struct BinSplit { enum { SPLIT_OBJECT = 0, SPLIT_FALLBACK = 1, SPLIT_ENFORCE = 2, // splits with larger ID are enforced in createLargeLeaf even if we could create a leaf already SPLIT_TEMPORAL = 2, SPLIT_GEOMID = 3, }; /*! construct an invalid split by default */ __forceinline BinSplit() : sah(inf), dim(-1), pos(0), data(0) {} __forceinline BinSplit(float sah, unsigned data, int dim = 0, float fpos = 0) : sah(sah), dim(dim), fpos(fpos), data(data) {} /*! constructs specified split */ __forceinline BinSplit(float sah, int dim, int pos, const BinMapping& mapping) : sah(sah), dim(dim), pos(pos), data(0), mapping(mapping) {} /*! tests if this split is valid */ __forceinline bool valid() const { return dim != -1; } /*! calculates surface area heuristic for performing the split */ __forceinline float splitSAH() const { return sah; } /*! stream output */ friend embree_ostream operator<<(embree_ostream cout, const BinSplit& split) { return cout << "BinSplit { sah = " << split.sah << ", dim = " << split.dim << ", pos = " << split.pos << "}"; } public: float sah; //!< SAH cost of the split int dim; //!< split dimension union { int pos; float fpos; }; //!< bin index for splitting unsigned int data; //!< extra optional split data BinMapping mapping; //!< mapping into bins }; /*! stores extended information about the split */ template struct SplitInfoT { __forceinline SplitInfoT () {} __forceinline SplitInfoT (size_t leftCount, const BBox& leftBounds, size_t rightCount, const BBox& rightBounds) : leftCount(leftCount), rightCount(rightCount), leftBounds(leftBounds), rightBounds(rightBounds) {} public: size_t leftCount,rightCount; BBox leftBounds,rightBounds; }; typedef SplitInfoT SplitInfo; typedef SplitInfoT SplitInfo2; /*! stores all binning information */ template struct __aligned(64) BinInfoT { typedef BinSplit Split; typedef vbool4 vbool; typedef vint4 vint; typedef vfloat4 vfloat; __forceinline BinInfoT() { } __forceinline BinInfoT(EmptyTy) { clear(); } /*! bin access function */ __forceinline BBox &bounds(const size_t binID, const size_t dimID) { return _bounds[binID][dimID]; } __forceinline const BBox &bounds(const size_t binID, const size_t dimID) const { return _bounds[binID][dimID]; } __forceinline unsigned int &counts(const size_t binID, const size_t dimID) { return _counts[binID][dimID]; } __forceinline const unsigned int &counts(const size_t binID, const size_t dimID) const { return _counts[binID][dimID]; } __forceinline vuint4 &counts(const size_t binID) { return _counts[binID]; } __forceinline const vuint4 &counts(const size_t binID) const { return _counts[binID]; } /*! clears the bin info */ __forceinline void clear() { for (size_t i=0; i& mapping) { if (unlikely(N == 0)) return; size_t i; for (i=0; i(bin0); bounds(b00,0).extend(prim0); const unsigned int b01 = extract<1>(bin0); bounds(b01,1).extend(prim0); const unsigned int b02 = extract<2>(bin0); bounds(b02,2).extend(prim0); const unsigned int s0 = (unsigned int)prims[i+0].size(); counts(b00,0)+=s0; counts(b01,1)+=s0; counts(b02,2)+=s0; /*! increase bounds of bins for odd primitive */ const unsigned int b10 = extract<0>(bin1); bounds(b10,0).extend(prim1); const unsigned int b11 = extract<1>(bin1); bounds(b11,1).extend(prim1); const unsigned int b12 = extract<2>(bin1); bounds(b12,2).extend(prim1); const unsigned int s1 = (unsigned int)prims[i+1].size(); counts(b10,0)+=s1; counts(b11,1)+=s1; counts(b12,2)+=s1; } /*! for uneven number of primitives */ if (i < N) { /*! map primitive to bin */ BBox prim0; Vec3fa center0; prims[i].binBoundsAndCenter(prim0,center0); const vint4 bin0 = (vint4)mapping.bin(center0); /*! increase bounds of bins */ const unsigned int s0 = (unsigned int)prims[i].size(); const int b00 = extract<0>(bin0); counts(b00,0)+=s0; bounds(b00,0).extend(prim0); const int b01 = extract<1>(bin0); counts(b01,1)+=s0; bounds(b01,1).extend(prim0); const int b02 = extract<2>(bin0); counts(b02,2)+=s0; bounds(b02,2).extend(prim0); } } /*! bins an array of primitives */ template __forceinline void bin (const PrimRef* prims, size_t N, const BinMapping& mapping, const BinBoundsAndCenter& binBoundsAndCenter) { if (N == 0) return; size_t i; for (i=0; i(bin0); counts(b00,0)+=s0; bounds(b00,0).extend(prim0); const int b01 = extract<1>(bin0); counts(b01,1)+=s0; bounds(b01,1).extend(prim0); const int b02 = extract<2>(bin0); counts(b02,2)+=s0; bounds(b02,2).extend(prim0); /*! increase bounds of bins for odd primitive */ const unsigned int s1 = prims[i+1].size(); const int b10 = extract<0>(bin1); counts(b10,0)+=s1; bounds(b10,0).extend(prim1); const int b11 = extract<1>(bin1); counts(b11,1)+=s1; bounds(b11,1).extend(prim1); const int b12 = extract<2>(bin1); counts(b12,2)+=s1; bounds(b12,2).extend(prim1); } /*! for uneven number of primitives */ if (i < N) { /*! map primitive to bin */ BBox prim0; Vec3fa center0; binBoundsAndCenter.binBoundsAndCenter(prims[i+0],prim0,center0); const vint4 bin0 = (vint4)mapping.bin(center0); /*! increase bounds of bins */ const unsigned int s0 = prims[i+0].size(); const int b00 = extract<0>(bin0); counts(b00,0)+=s0; bounds(b00,0).extend(prim0); const int b01 = extract<1>(bin0); counts(b01,1)+=s0; bounds(b01,1).extend(prim0); const int b02 = extract<2>(bin0); counts(b02,2)+=s0; bounds(b02,2).extend(prim0); } } __forceinline void bin(const PrimRef* prims, size_t begin, size_t end, const BinMapping& mapping) { bin(prims+begin,end-begin,mapping); } template __forceinline void bin(const PrimRef* prims, size_t begin, size_t end, const BinMapping& mapping, const BinBoundsAndCenter& binBoundsAndCenter) { bin(prims+begin,end-begin,mapping,binBoundsAndCenter); } /*! merges in other binning information */ __forceinline void merge (const BinInfoT& other, size_t numBins) { for (size_t i=0; i& mapping, const size_t blocks_shift) const { /* sweep from right to left and compute parallel prefix of merged bounds */ vfloat4 rAreas[BINS]; vuint4 rCounts[BINS]; vuint4 count = 0; BBox bx = empty; BBox by = empty; BBox bz = empty; for (size_t i=mapping.size()-1; i>0; i--) { count += counts(i); rCounts[i] = count; bx.extend(bounds(i,0)); rAreas[i][0] = expectedApproxHalfArea(bx); by.extend(bounds(i,1)); rAreas[i][1] = expectedApproxHalfArea(by); bz.extend(bounds(i,2)); rAreas[i][2] = expectedApproxHalfArea(bz); rAreas[i][3] = 0.0f; } /* sweep from left to right and compute SAH */ vuint4 blocks_add = (1 << blocks_shift)-1; vuint4 ii = 1; vfloat4 vbestSAH = pos_inf; vuint4 vbestPos = 0; count = 0; bx = empty; by = empty; bz = empty; for (size_t i=1; i> (unsigned int)(blocks_shift); // if blocks_shift >=1 then lCount < 4B and could be represented with an vint4, which would allow for faster vfloat4 conversions. const vuint4 rCount = (rCounts[i]+blocks_add) >> (unsigned int)(blocks_shift); const vfloat4 sah = madd(lArea,vfloat4(lCount),rArea*vfloat4(rCount)); //const vfloat4 sah = madd(lArea,vfloat4(vint4(lCount)),rArea*vfloat4(vint4(rCount))); vbestPos = select(sah < vbestSAH,ii ,vbestPos); vbestSAH = select(sah < vbestSAH,sah,vbestSAH); } /* find best dimension */ float bestSAH = inf; int bestDim = -1; int bestPos = 0; for (int dim=0; dim<3; dim++) { /* ignore zero sized dimensions */ if (unlikely(mapping.invalid(dim))) continue; /* test if this is a better dimension */ if (vbestSAH[dim] < bestSAH && vbestPos[dim] != 0) { bestDim = dim; bestPos = vbestPos[dim]; bestSAH = vbestSAH[dim]; } } return Split(bestSAH,bestDim,bestPos,mapping); } /*! finds the best split by scanning binning information */ __forceinline Split best_block_size(const BinMapping& mapping, const size_t blockSize) const { /* sweep from right to left and compute parallel prefix of merged bounds */ vfloat4 rAreas[BINS]; vuint4 rCounts[BINS]; vuint4 count = 0; BBox bx = empty; BBox by = empty; BBox bz = empty; for (size_t i=mapping.size()-1; i>0; i--) { count += counts(i); rCounts[i] = count; bx.extend(bounds(i,0)); rAreas[i][0] = expectedApproxHalfArea(bx); by.extend(bounds(i,1)); rAreas[i][1] = expectedApproxHalfArea(by); bz.extend(bounds(i,2)); rAreas[i][2] = expectedApproxHalfArea(bz); rAreas[i][3] = 0.0f; } /* sweep from left to right and compute SAH */ vuint4 blocks_add = blockSize-1; vfloat4 blocks_factor = 1.0f/float(blockSize); vuint4 ii = 1; vfloat4 vbestSAH = pos_inf; vuint4 vbestPos = 0; count = 0; bx = empty; by = empty; bz = empty; for (size_t i=1; i& mapping, const Split& split, SplitInfoT& info) const { if (split.dim == -1) { new (&info) SplitInfoT(0,empty,0,empty); return; } size_t leftCount = 0; BBox leftBounds = empty; for (size_t i=0; i<(size_t)split.pos; i++) { leftCount += counts(i,split.dim); leftBounds.extend(bounds(i,split.dim)); } size_t rightCount = 0; BBox rightBounds = empty; for (size_t i=split.pos; i(leftCount,leftBounds,rightCount,rightBounds); } /*! gets the number of primitives left of the split */ __forceinline size_t getLeftCount(const BinMapping& mapping, const Split& split) const { if (unlikely(split.dim == -1)) return -1; size_t leftCount = 0; for (size_t i = 0; i < (size_t)split.pos; i++) { leftCount += counts(i, split.dim); } return leftCount; } /*! gets the number of primitives right of the split */ __forceinline size_t getRightCount(const BinMapping& mapping, const Split& split) const { if (unlikely(split.dim == -1)) return -1; size_t rightCount = 0; for (size_t i = (size_t)split.pos; i __forceinline void bin_parallel(BinInfoT& binner, const PrimRef* prims, size_t begin, size_t end, size_t blockSize, size_t parallelThreshold, const BinMapping& mapping) { if (likely(end-begin < parallelThreshold)) { binner.bin(prims,begin,end,mapping); } else { binner = parallel_reduce(begin,end,blockSize,binner, [&](const range& r) -> BinInfoT { BinInfoT binner(empty); binner.bin(prims + r.begin(), r.size(), mapping); return binner; }, [&](const BinInfoT& b0, const BinInfoT& b1) -> BinInfoT { BinInfoT r = b0; r.merge(b1, mapping.size()); return r; }); } } template __forceinline void bin_parallel(BinInfoT& binner, const PrimRef* prims, size_t begin, size_t end, size_t blockSize, size_t parallelThreshold, const BinMapping& mapping, const BinBoundsAndCenter& binBoundsAndCenter) { if (likely(end-begin < parallelThreshold)) { binner.bin(prims,begin,end,mapping,binBoundsAndCenter); } else { binner = parallel_reduce(begin,end,blockSize,binner, [&](const range& r) -> BinInfoT { BinInfoT binner(empty); binner.bin(prims + r.begin(), r.size(), mapping, binBoundsAndCenter); return binner; }, [&](const BinInfoT& b0, const BinInfoT& b1) -> BinInfoT { BinInfoT r = b0; r.merge(b1, mapping.size()); return r; }); } } template __forceinline void bin_serial_or_parallel(BinInfoT& binner, const PrimRef* prims, size_t begin, size_t end, size_t blockSize, const BinMapping& mapping) { if (!parallel) { binner.bin(prims,begin,end,mapping); } else { binner = parallel_reduce(begin,end,blockSize,binner, [&](const range& r) -> BinInfoT { BinInfoT binner(empty); binner.bin(prims + r.begin(), r.size(), mapping); return binner; }, [&](const BinInfoT& b0, const BinInfoT& b1) -> BinInfoT { BinInfoT r = b0; r.merge(b1, mapping.size()); return r; }); } } template __forceinline void bin_serial_or_parallel(BinInfoT& binner, const PrimRef* prims, size_t begin, size_t end, size_t blockSize, const BinMapping& mapping, const BinBoundsAndCenter& binBoundsAndCenter) { if (!parallel) { binner.bin(prims,begin,end,mapping,binBoundsAndCenter); } else { binner = parallel_reduce(begin,end,blockSize,binner, [&](const range& r) -> BinInfoT { BinInfoT binner(empty); binner.bin(prims + r.begin(), r.size(), mapping, binBoundsAndCenter); return binner; }, [&](const BinInfoT& b0, const BinInfoT& b1) -> BinInfoT { BinInfoT r = b0; r.merge(b1, mapping.size()); return r; }); } } }