Raytracer.cpp 7.5 KB
Newer Older
unknown's avatar
unknown committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#include "Raytracer.h"
#include <engine/Messaging.h>
#include <engine/Time.h>
#include <random>

namespace glare
{
	namespace advanced
	{
		Raytracer::Raytracer()
			: Raytracer(std::make_shared<RayGeneratorDefault>())
		{

		}

		Raytracer::Raytracer(std::shared_ptr<RayGeneratorBase> ray_generator)
			: Raytracer(std::shared_ptr<SceneCollector>(nullptr), ray_generator)
		{
		}

		Raytracer::Raytracer(std::shared_ptr<SceneCollector> collector)
			: Raytracer(collector, std::make_shared<RayGeneratorDefault>())
		{
		}

		Raytracer::Raytracer(std::shared_ptr<SceneCollector> collector, std::shared_ptr<RayGeneratorBase> ray_generator)
		{
28
29
30
31
32
33
34
			m_bounce_count_thresholds[unsigned(EffectType::eDiffuse)] = 4;
			m_bounce_count_thresholds[unsigned(EffectType::eTranslucent)] = 6;
			m_bounce_count_thresholds[unsigned(EffectType::eReflection)] = 6;
			m_bounce_count_thresholds[unsigned(EffectType::eRefraction)] = 12;
			m_bounce_count_thresholds[unsigned(EffectType::eTransparent)] = 12;
			m_bounce_count_thresholds[unsigned(EffectType::eEmissive)] = 0;

35
36
37
38
39
40
41
42
43
			m_ls_bounce_count_thresholds[unsigned(EffectType::eDiffuse)] = 1;
			m_ls_bounce_count_thresholds[unsigned(EffectType::eTranslucent)] = 3;
			m_ls_bounce_count_thresholds[unsigned(EffectType::eReflection)] = 3;
			m_ls_bounce_count_thresholds[unsigned(EffectType::eRefraction)] = 6;
			m_ls_bounce_count_thresholds[unsigned(EffectType::eTransparent)] = 8;
			m_ls_bounce_count_thresholds[unsigned(EffectType::eEmissive)] = 0;


			m_render_config.bounces = 16;
44
			m_render_shader = core::ShaderProgram::makeUnique("/raytracer/pathtracer.compute");
unknown's avatar
unknown committed
45
46
47
48
49
50
51
52
53
54
55
56
			m_ray_generator = ray_generator;
			m_trace_buffer = std::make_unique<core::Buffer<gl::BufferType::eShaderStorage>>();
			core::MessageHandler::getInstance().registerReceiver(core::GlobalMessages::c_scene_changed, *this);
			if (collector)
				initialize(collector);
		}

		Raytracer::~Raytracer()
		{

		}

57
58
59
60
61
62
63
64
65
66
		void Raytracer::setEffectBounces(EffectType type, uint8_t bounces)
		{
			uint8_t clamped = bounces & 0xf;

			if (clamped != bounces)
			{
				Log_Warn << bounces << " - value larger than 15 (0xf). Result of " << clamped << " may be unintended.";
			}

			m_bounce_count_thresholds[unsigned(type)] = clamped;
unknown's avatar
unknown committed
67
68
69
70
71
72
			reset();
		}

		uint8_t Raytracer::getEffectBounces(EffectType type)
		{
			return m_bounce_count_thresholds[unsigned(type)];
73
74
		}

75
76
77
78
79
80
81
82
83
84
		void Raytracer::setLinespaceBounceThresholds(EffectType type, uint8_t bounces)
		{
			uint8_t clamped = bounces & 0xf;

			if (clamped != bounces)
			{
				Log_Warn << bounces << " - value larger than 15 (0xf). Result of " << clamped << " may be unintended.";
			}

			m_ls_bounce_count_thresholds[unsigned(type)] = clamped;
unknown's avatar
unknown committed
85
86
87
88
89
90
			reset();
		}

		uint8_t Raytracer::getLinespaceBounceThreshold(EffectType type)
		{
			return m_ls_bounce_count_thresholds[unsigned(type)];
91
92
		}

unknown's avatar
unknown committed
93
94
		void Raytracer::reset()
		{
95
			m_render_config.current_sample = 0;
unknown's avatar
unknown committed
96
97
98
99
			if (m_color_store)
				gl::clearTextureImage(m_color_store->getID(), 0, gl::TextureFormat::eRGBA, gl::Type::eFloat, glm::value_ptr(glm::vec4(0)));
		}

100
		const core::TextureRGBA_32F &Raytracer::render(uint32_t width, uint32_t height)
unknown's avatar
unknown committed
101
102
103
104
105
106
107
		{
			if (width != m_last_width || height != m_last_height) {
				m_last_width = width;
				m_last_height = height;
				m_trace_buffer->setData(sizeof(Trace) * width * height, gl::BufferUsage::eDynamicCopy);
				m_render_target = std::make_unique<core::TextureRGBA_32F>(width, height);
				m_color_store = std::make_unique<core::TextureRGBA_32F>(width, height);
108
				m_render_config.current_sample = 0;
unknown's avatar
unknown committed
109
110
			}
			
111
			for (unsigned sample = 0; sample < m_render_config.samples_per_frame; sample++) {
unknown's avatar
unknown committed
112

113
				if (m_render_config.current_sample >= m_render_config.max_samples)
unknown's avatar
unknown committed
114
115
116
117
118
119
120
121
122
123
124
125
					return *m_render_target;

				m_ray_generator->generate(*this);

				m_render_shader->use();

				m_render_shader->updateUniformImageRGBA32F("u_render_target", *m_render_target, gl::Access::eReadWrite);
				m_render_shader->updateUniformImageRGBA32F("u_color_store", *m_color_store, gl::Access::eReadWrite);

				m_render_shader->updateStorageBuffer("lights_buffer", m_collector->lightBuffer());
				m_render_shader->updateStorageBuffer("trace_buffer", m_trace_buffer);
				m_render_shader->updateStorageBuffer("mesh_buffer", m_collector->meshBuffer());
unknown's avatar
unknown committed
126
127
				m_render_shader->updateStorageBuffer("object_bvh_buffer", m_collector->objectBVHNodeBuffer());
				m_render_shader->updateStorageBuffer("object_bvh_id_buffer", m_collector->objectBVHIDBuffer());
unknown's avatar
unknown committed
128
129
130
131
132

				m_render_shader->updateStorageBuffer("material_buffer", m_collector->materialBuffer());

				m_render_shader->updateUniformStruct("u_camera", m_collector->getActiveCamera());

133
134
135
136
137
				m_render_shader->updateUniformUInt("u_render_config.current_sample", m_render_config.current_sample);
				m_render_shader->updateUniformUInt("u_render_config.max_samples", m_render_config.max_samples);
				m_render_shader->updateUniformUInt("u_render_config.num_bounces", m_render_config.bounces);
				m_render_shader->updateUniformFloat("u_render_config.clamp_direct", m_render_config.clamp_direct);
				m_render_shader->updateUniformFloat("u_render_config.clamp_indirect", m_render_config.clamp_indirect);
unknown's avatar
unknown committed
138
139
140
141
142
143
144
145
146
147
148
149

				if (m_skybox)
				{
					m_render_shader->updateUniformUInt64("u_environment.cubemap", m_skybox->getBindlessID());
					m_render_shader->updateUniformInt("u_environment.has", 1);
				}
				else
				{
					m_render_shader->updateUniformVec4("u_environment.color", glm::vec4(0.7f, 0.9f, 0.97f, 1));
					m_render_shader->updateUniformInt("u_environment.has", 0);
				}

150
151
				m_render_shader->updateUniformFloat("u_linespace_properties.accuracy_quality", m_linespace_config.accuracy_threshold);
				m_render_shader->updateUniformFloat("u_linespace_properties.shadow_quality", m_linespace_config.shadow_threshold);
unknown's avatar
unknown committed
152
153
154
155
156
157
				
				unsigned bounce_thresholds  = 0;
				for (int i = 0; i < 8; ++i)
				{
					bounce_thresholds |= (m_bounce_count_thresholds[i] & 0xf) << (4 * i);
				}
158
159
				m_render_shader->updateUniformUInt("u_render_config.bounce_thresholds", bounce_thresholds);

160
				m_linespace_config.bounce_thresholds = 0;
161
162
163
164
165
				for (int i = 0; i < 8; ++i)
				{
					m_linespace_config.bounce_thresholds |= (m_ls_bounce_count_thresholds[i] & 0xf) << (4 * i);
				}
				m_render_shader->updateUniformUInt("u_linespace_properties.bounce_thresholds", m_linespace_config.bounce_thresholds);
unknown's avatar
unknown committed
166

unknown's avatar
unknown committed
167
				m_render_shader->dispatchCompute2D(width, 4, height, 4);
unknown's avatar
unknown committed
168

169
				++m_render_config.current_sample;
unknown's avatar
unknown committed
170
171
172
173
174
			}

			return *m_render_target;
		}

unknown's avatar
unknown committed
175
176
		void Raytracer::saveRender(fs::path target) const
		{
177
			io::image::saveTexture(target, *m_render_target);
unknown's avatar
unknown committed
178
179
		}

unknown's avatar
unknown committed
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
		void Raytracer::setGenerator(std::shared_ptr<RayGeneratorBase> generator)
		{
			m_ray_generator = generator;
			reset();
		}

		void Raytracer::initialize(std::shared_ptr<SceneCollector> collector)
		{
			m_collector = collector;
			reset();
		}

		void Raytracer::handle(core::message_t& message)
		{
			switch (message.name)
			{
			case core::GlobalMessages::c_scene_changed:
				reset();
				break;

			}
		}

		const core::ShaderProgram& Raytracer::getShader() const
		{
			return *m_render_shader;
		}

		void Raytracer::setSkybox(std::shared_ptr<core::Skybox> skybox)
		{
			m_skybox = skybox;
			reset();
		}

		uint32_t Raytracer::width() const
		{
			return m_last_width;
		}

		uint32_t Raytracer::height() const
		{
			return m_last_height;
		}

		core::Buffer<gl::BufferType::eShaderStorage>& Raytracer::traceBuffer() const
		{
			return *m_trace_buffer;
		}

		const core::TextureRGBA_32F& Raytracer::renderTarget() const
		{
			return *m_render_target;
		}

		std::shared_ptr<SceneCollector> Raytracer::collector() const
		{
			return m_collector;
		}
	}
}