hiprt.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2011-2023 Blender Foundation
 *
 * SPDX-License-Identifier: Apache-2.0 */
 
#ifdef WITH_HIPRT
 
#  pragma once
 
#  include "bvh/bvh.h"
#  include "bvh/params.h"
 
#  ifdef WITH_HIP_DYNLOAD
#    include <hiprtew.h>
#  else
#    include <hiprt/hiprt_types.h>
#  endif
 
#  include "device/memory.h"
 
CCL_NAMESPACE_BEGIN
 
class BVHHIPRT : public BVH {
 public:
  friend class HIPDevice;
 
  bool is_tlas()
  {
    return params.top_level;
  }
 
  hiprtGeometry hiprt_geom;
  hiprtTriangleMeshPrimitive triangle_mesh;
  hiprtAABBListPrimitive custom_prim_aabb;
  hiprtGeometryBuildInput geom_input;
 
  vector<int2> custom_prim_info; /* x: prim_id, y: prim_type */
  vector<float2> prims_time;
 
  /* Custom primitives. */
  device_vector<BoundBox> custom_primitive_bound;
  device_vector<int> triangle_index;
  device_vector<float> vertex_data;
 
  BVHHIPRT(const BVHParams &params,
           const vector<Geometry *> &geometry,
           const vector<Object *> &objects,
           Device *in_device);
 
  ~BVHHIPRT() override;
 
 private:
  Device *device;
};
 
CCL_NAMESPACE_END
 
#endif