1 files changed, 85 insertions, 0 deletions
diff --git a/thirdparty/embree/kernels/geometry/instance_array.h b/thirdparty/embree/kernels/geometry/instance_array.h
new file mode 100644
index 0000000000..fdbe79b3a8
--- /dev/null
+++ b/thirdparty/embree/kernels/geometry/instance_array.h
@@ -0,0 +1,85 @@
+// Copyright 2009-2021 Intel Corporation
+// SPDX-License-Identifier: Apache-2.0
+
+#pragma once
+
+#include "primitive.h"
+#include "../common/scene_instance_array.h"
+
+namespace embree
+{
+  //template<int M>
+  struct InstanceArrayPrimitive
+  {
+    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 primitives */
+    static __forceinline size_t max_size() { return 1; }
+
+    /* Returns required number of primitive blocks for N primitives */
+    static __forceinline size_t blocks(size_t N) { return N; }
+
+  public:
+
+    InstanceArrayPrimitive (const uint32_t geomID, const uint32_t primID)
+    : primID_(primID)
+    , instID_(geomID)
+    {}
+
+    __forceinline bool valid() const { 
+      return primID_ != -1;
+    }
+
+    void fill(const PrimRef* prims, size_t& i, size_t end, Scene* scene)
+    {
+      assert(end-i == 1);
+      const PrimRef& prim = prims[i]; i++;
+      const unsigned int geomID = prim.geomID();
+      const unsigned int primID = prim.primID();
+      new (this) InstanceArrayPrimitive(geomID, primID);
+    }
+
+    __forceinline LBBox3fa fillMB(const PrimRef* prims, size_t& i, size_t end, Scene* scene, size_t itime)
+    {
+      assert(end-i == 1);
+      const PrimRef& prim = prims[i]; i++;
+      const unsigned int geomID = prim.geomID();
+      const size_t primID = prim.primID();
+      new (this) InstanceArrayPrimitive(geomID, primID);
+      const InstanceArray* instanceArray = scene->get<InstanceArray>(geomID);
+      return instanceArray->linearBounds(primID,itime);
+    }
+
+    __forceinline LBBox3fa fillMB(const PrimRefMB* prims, size_t& i, size_t end, Scene* scene, const BBox1f time_range)
+    {
+      assert(end-i == 1);
+      const PrimRefMB& prim = prims[i]; i++;
+      const unsigned int geomID = prim.geomID();
+      const size_t primID = prim.primID();
+      new (this) InstanceArrayPrimitive(geomID, primID);
+      const InstanceArray* instanceArray = scene->get<InstanceArray>(geomID);
+      return instanceArray->linearBounds(primID,time_range);
+    }
+
+    /* Updates the primitive */
+    __forceinline BBox3fa update(InstanceArray* instanceArray) {
+      return instanceArray->bounds(0);
+    }
+
+  public:
+    unsigned int primID_;
+    unsigned int instID_;
+  };
+}