| 1 | // Copyright 2009-2021 Intel Corporation | 
| 2 | // SPDX-License-Identifier: Apache-2.0 | 
| 3 |  | 
| 4 | #pragma once | 
| 5 |  | 
| 6 | #include "primitive.h" | 
| 7 |  | 
| 8 | namespace embree | 
| 9 | { | 
| 10 |   /* Stores the vertices of M triangles in struct of array layout */ | 
| 11 |   template <int M> | 
| 12 |   struct TriangleMv | 
| 13 |   {  | 
| 14 |   public: | 
| 15 |     struct Type : public PrimitiveType  | 
| 16 |     { | 
| 17 |       const char* name() const; | 
| 18 |       size_t sizeActive(const char* This) const; | 
| 19 |       size_t sizeTotal(const char* This) const; | 
| 20 |       size_t getBytes(const char* This) const; | 
| 21 |     }; | 
| 22 |     static Type type; | 
| 23 |  | 
| 24 |   public: | 
| 25 |  | 
| 26 |     /* Returns maximum number of stored triangles */ | 
| 27 |     static __forceinline size_t max_size() { return M; } | 
| 28 |      | 
| 29 |     /* Returns required number of primitive blocks for N primitives */ | 
| 30 |     static __forceinline size_t blocks(size_t N) { return (N+max_size()-1)/max_size(); } | 
| 31 |     | 
| 32 |   public: | 
| 33 |  | 
| 34 |     /* Default constructor */ | 
| 35 |     __forceinline TriangleMv() {} | 
| 36 |  | 
| 37 |     /* Construction from vertices and IDs */ | 
| 38 |     __forceinline TriangleMv(const Vec3vf<M>& v0, const Vec3vf<M>& v1, const Vec3vf<M>& v2, const vuint<M>& geomIDs, const vuint<M>& primIDs) | 
| 39 |       : v0(v0), v1(v1), v2(v2), geomIDs(geomIDs), primIDs(primIDs) {} | 
| 40 |      | 
| 41 |     /* Returns a mask that tells which triangles are valid */ | 
| 42 |     __forceinline vbool<M> valid() const { return geomIDs != vuint<M>(-1); } | 
| 43 |  | 
| 44 |     /* Returns true if the specified triangle is valid */ | 
| 45 |     __forceinline bool valid(const size_t i) const { assert(i<M); return geomIDs[i] != -1; } | 
| 46 |  | 
| 47 |     /* Returns the number of stored triangles */ | 
| 48 |     __forceinline size_t size() const { return bsf(~movemask(valid())); } | 
| 49 |  | 
| 50 |     /* Returns the geometry IDs */ | 
| 51 |     __forceinline       vuint<M>& geomID()       { return geomIDs; } | 
| 52 |     __forceinline const vuint<M>& geomID() const { return geomIDs; } | 
| 53 |     __forceinline unsigned int geomID(const size_t i) const { assert(i<M); return geomIDs[i]; } | 
| 54 |  | 
| 55 |     /* Returns the primitive IDs */ | 
| 56 |     __forceinline       vuint<M>& primID()       { return primIDs; } | 
| 57 |     __forceinline const vuint<M>& primID() const { return primIDs; } | 
| 58 |     __forceinline unsigned int primID(const size_t i) const { assert(i<M); return primIDs[i]; } | 
| 59 |  | 
| 60 |     /* Calculate the bounds of the triangles */ | 
| 61 |     __forceinline BBox3fa bounds() const  | 
| 62 |     { | 
| 63 |       Vec3vf<M> lower = min(v0,v1,v2); | 
| 64 |       Vec3vf<M> upper = max(v0,v1,v2); | 
| 65 |       vbool<M> mask = valid(); | 
| 66 |       lower.x = select(mask,lower.x,vfloat<M>(pos_inf)); | 
| 67 |       lower.y = select(mask,lower.y,vfloat<M>(pos_inf)); | 
| 68 |       lower.z = select(mask,lower.z,vfloat<M>(pos_inf)); | 
| 69 |       upper.x = select(mask,upper.x,vfloat<M>(neg_inf)); | 
| 70 |       upper.y = select(mask,upper.y,vfloat<M>(neg_inf)); | 
| 71 |       upper.z = select(mask,upper.z,vfloat<M>(neg_inf)); | 
| 72 |       return BBox3fa(Vec3fa(reduce_min(lower.x),reduce_min(lower.y),reduce_min(lower.z)), | 
| 73 |                      Vec3fa(reduce_max(upper.x),reduce_max(upper.y),reduce_max(upper.z))); | 
| 74 |     } | 
| 75 |      | 
| 76 |     /* Non temporal store */ | 
| 77 |     __forceinline static void store_nt(TriangleMv* dst, const TriangleMv& src) | 
| 78 |     { | 
| 79 |       vfloat<M>::store_nt(&dst->v0.x,src.v0.x); | 
| 80 |       vfloat<M>::store_nt(&dst->v0.y,src.v0.y); | 
| 81 |       vfloat<M>::store_nt(&dst->v0.z,src.v0.z); | 
| 82 |       vfloat<M>::store_nt(&dst->v1.x,src.v1.x); | 
| 83 |       vfloat<M>::store_nt(&dst->v1.y,src.v1.y); | 
| 84 |       vfloat<M>::store_nt(&dst->v1.z,src.v1.z); | 
| 85 |       vfloat<M>::store_nt(&dst->v2.x,src.v2.x); | 
| 86 |       vfloat<M>::store_nt(&dst->v2.y,src.v2.y); | 
| 87 |       vfloat<M>::store_nt(&dst->v2.z,src.v2.z); | 
| 88 |       vuint<M>::store_nt(&dst->geomIDs,src.geomIDs); | 
| 89 |       vuint<M>::store_nt(&dst->primIDs,src.primIDs); | 
| 90 |     } | 
| 91 |  | 
| 92 |     /* Fill triangle from triangle list */ | 
| 93 |     __forceinline void fill(const PrimRef* prims, size_t& begin, size_t end, Scene* scene) | 
| 94 |     { | 
| 95 |       vuint<M> vgeomID = -1, vprimID = -1; | 
| 96 |       Vec3vf<M> v0 = zero, v1 = zero, v2 = zero; | 
| 97 |        | 
| 98 |       for (size_t i=0; i<M && begin<end; i++, begin++) | 
| 99 |       { | 
| 100 | 	const PrimRef& prim = prims[begin]; | 
| 101 |         const unsigned geomID = prim.geomID(); | 
| 102 |         const unsigned primID = prim.primID(); | 
| 103 |         const TriangleMesh* __restrict__ const mesh = scene->get<TriangleMesh>(i: geomID); | 
| 104 |         const TriangleMesh::Triangle& tri = mesh->triangle(i: primID); | 
| 105 |         const Vec3fa& p0 = mesh->vertex(i: tri.v[0]); | 
| 106 |         const Vec3fa& p1 = mesh->vertex(i: tri.v[1]); | 
| 107 |         const Vec3fa& p2 = mesh->vertex(i: tri.v[2]); | 
| 108 |         vgeomID [i] = geomID; | 
| 109 |         vprimID [i] = primID; | 
| 110 |         v0.x[i] = p0.x; v0.y[i] = p0.y; v0.z[i] = p0.z; | 
| 111 |         v1.x[i] = p1.x; v1.y[i] = p1.y; v1.z[i] = p1.z; | 
| 112 |         v2.x[i] = p2.x; v2.y[i] = p2.y; v2.z[i] = p2.z; | 
| 113 |       } | 
| 114 |       TriangleMv::store_nt(dst: this,src: TriangleMv(v0,v1,v2,vgeomID,vprimID)); | 
| 115 |     } | 
| 116 |  | 
| 117 |     /* Updates the primitive */ | 
| 118 |     __forceinline BBox3fa update(TriangleMesh* mesh) | 
| 119 |     { | 
| 120 |       BBox3fa bounds = empty; | 
| 121 |       vuint<M> vgeomID = -1, vprimID = -1; | 
| 122 |       Vec3vf<M> v0 = zero, v1 = zero, v2 = zero; | 
| 123 |        | 
| 124 |       for (size_t i=0; i<M; i++) | 
| 125 |       { | 
| 126 |         if (primID(i) == -1) break; | 
| 127 |         const unsigned geomId = geomID(i); | 
| 128 |         const unsigned primId = primID(i); | 
| 129 |         const TriangleMesh::Triangle& tri = mesh->triangle(i: primId); | 
| 130 |         const Vec3fa p0 = mesh->vertex(i: tri.v[0]); | 
| 131 |         const Vec3fa p1 = mesh->vertex(i: tri.v[1]); | 
| 132 |         const Vec3fa p2 = mesh->vertex(i: tri.v[2]); | 
| 133 |         bounds.extend(other: merge(a: BBox3fa(p0),b: BBox3fa(p1),c: BBox3fa(p2))); | 
| 134 |         vgeomID [i] = geomId; | 
| 135 |         vprimID [i] = primId; | 
| 136 |         v0.x[i] = p0.x; v0.y[i] = p0.y; v0.z[i] = p0.z; | 
| 137 |         v1.x[i] = p1.x; v1.y[i] = p1.y; v1.z[i] = p1.z; | 
| 138 |         v2.x[i] = p2.x; v2.y[i] = p2.y; v2.z[i] = p2.z; | 
| 139 |       } | 
| 140 |       new (this) TriangleMv(v0,v1,v2,vgeomID,vprimID); | 
| 141 |       return bounds; | 
| 142 |     } | 
| 143 |     | 
| 144 |   public: | 
| 145 |     Vec3vf<M> v0;      // 1st vertex of the triangles | 
| 146 |     Vec3vf<M> v1;      // 2nd vertex of the triangles | 
| 147 |     Vec3vf<M> v2;      // 3rd vertex of the triangles | 
| 148 |   private: | 
| 149 |     vuint<M> geomIDs; // geometry ID | 
| 150 |     vuint<M> primIDs; // primitive ID | 
| 151 |   }; | 
| 152 |  | 
| 153 |   template<int M> | 
| 154 |   typename TriangleMv<M>::Type TriangleMv<M>::type; | 
| 155 |  | 
| 156 |   typedef TriangleMv<4> Triangle4v; | 
| 157 | } | 
| 158 |  |