Refactoring

ray_tracing_specialization/shaders/host_device.h

@@ -0,0 +1,118 @@
+/*
+ * Copyright (c) 2019-2021, NVIDIA CORPORATION.  All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * SPDX-FileCopyrightText: Copyright (c) 2019-2021 NVIDIA CORPORATION
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+
+#ifndef COMMON_HOST_DEVICE
+#define COMMON_HOST_DEVICE
+
+#ifdef __cplusplus
+#include "nvmath/nvmath.h"
+// GLSL Type
+using vec2 = nvmath::vec2f;
+using vec3 = nvmath::vec3f;
+using vec4 = nvmath::vec4f;
+using mat4 = nvmath::mat4f;
+using uint = unsigned int;
+#endif
+
+// clang-format off
+#ifdef __cplusplus // Descriptor binding helper for C++ and GLSL
+ #define START_BINDING(a) enum a {
+ #define END_BINDING() }
+#else
+ #define START_BINDING(a)  const uint
+ #define END_BINDING() 
+#endif
+
+START_BINDING(SceneBindings)
+  eGlobals  = 0,  // Global uniform containing camera matrices
+  eObjDescs = 1,  // Access to the object descriptions
+  eTextures = 2   // Access to textures
+END_BINDING();
+
+START_BINDING(RtxBindings)
+  eTlas     = 0,  // Top-level acceleration structure
+  eOutImage = 1   // Ray tracer output image
+END_BINDING();
+// clang-format on
+
+
+// Information of a obj model when referenced in a shader
+struct ObjDesc
+{
+  int      txtOffset;             // Texture index offset in the array of textures
+  uint64_t vertexAddress;         // Address of the Vertex buffer
+  uint64_t indexAddress;          // Address of the index buffer
+  uint64_t materialAddress;       // Address of the material buffer
+  uint64_t materialIndexAddress;  // Address of the triangle material index buffer
+};
+
+// Uniform buffer set at each frame
+struct GlobalUniforms
+{
+  mat4 viewProj;     // Camera view * projection
+  mat4 viewInverse;  // Camera inverse view matrix
+  mat4 projInverse;  // Camera inverse projection matrix
+};
+
+// Push constant structure for the raster
+struct PushConstantRaster
+{
+  mat4  modelMatrix;  // matrix of the instance
+  vec3  lightPosition;
+  uint  objIndex;
+  float lightIntensity;
+  int   lightType;
+};
+
+
+// Push constant structure for the ray tracer
+struct PushConstantRay
+{
+  vec4  clearColor;
+  vec3  lightPosition;
+  float lightIntensity;
+  int   lightType;
+  int   specialization;
+};
+
+struct Vertex  // See ObjLoader, copy of VertexObj, could be compressed for device
+{
+  vec3 pos;
+  vec3 nrm;
+  vec3 color;
+  vec2 texCoord;
+};
+
+struct WaveFrontMaterial  // See ObjLoader, copy of MaterialObj, could be compressed for device
+{
+  vec3  ambient;
+  vec3  diffuse;
+  vec3  specular;
+  vec3  transmittance;
+  vec3  emission;
+  float shininess;
+  float ior;       // index of refraction
+  float dissolve;  // 1 == opaque; 0 == fully transparent
+  int   illum;     // illumination model (see http://www.fileformat.info/format/material/)
+  int   textureId;
+};
+
+
+#endif