add custom display options

src/lib.rs

@@ -36,6 +36,13 @@ pub struct RedrawArgs {
 	pub light_spread: f32,
 	pub accum_sigma: f32,
 	pub accum_scale: f32,
+	pub show_axes: bool,
+	pub show_shapes: bool,
+	pub show_hit_emission: bool,
+	pub show_miss_emission: bool,
+	pub show_direct_hit: bool,
+	pub show_indirect_hit: bool,
+	pub show_light: bool,
 }
 
 pub struct Gpu {
@@ -163,7 +170,9 @@ impl Core {
 			depth_stencil_attachment: None,
 			..Default::default()
 		});
-		self.pipeline.render(&mut pass, [&self.tripod]);
+		if args.show_axes {
+			self.pipeline.render(&mut pass, [&self.tripod]);
+		}
 
 		let source = Source {
 			position_yaw: args.light_position.yaw,
@@ -173,14 +182,16 @@ impl Core {
 			spread: args.light_spread,
 		};
 
-		let contour: Vec<Vertex> = loop_list(source.contour(17))
+		if args.show_shapes {
-			.map(|pos| Vertex {
+			let contour: Vec<Vertex> = loop_list(source.contour(17))
-				pos,
+				.map(|pos| Vertex {
-				color: vec3(1., 1., 1.),
+					pos,
-			})
+					color: vec3(1., 1., 1.),
-			.collect();
+				})
-		self.pipeline
+				.collect();
-			.render(&mut pass, [&Mesh::new(&self.device, &contour)]);
+			self.pipeline
+				.render(&mut pass, [&Mesh::new(&self.device, &contour)]);
+		}
 
 		const BASE_R: f32 = 2.;
 		const BASE_POS: Vec3 = vec3(0., 0., -BASE_R);
@@ -211,35 +222,43 @@ impl Core {
 		for ray in source_rays {
 			if let Some(hit) = scene.trace_ray(ray) {
 				hits.push(hit);
-				source_ray_display.extend([
+				if args.show_hit_emission {
-					Vertex {
+					source_ray_display.extend([
-						pos: ray.base,
+						Vertex {
-						color: vec3(1., 1., 1.),
+							pos: ray.base,
-					},
+							color: vec3(1., 1., 1.),
-					Vertex {
+						},
-						pos: ray.base + 0.1 * ray.dir,
+						Vertex {
-						color: vec3(0., 1., 0.),
+							pos: ray.base + 0.1 * ray.dir,
-					},
+							color: vec3(0., 1., 0.),
-					Vertex {
+						},
-						pos: hit.incident.base - 0.02 * hit.incident.dir,
+					]);
-						color: vec3(0., 0., 1.),
+				}
-					},
+				if args.show_direct_hit {
-					Vertex {
+					source_ray_display.extend([
-						pos: hit.incident.base,
+						Vertex {
-						color: vec3(1., 1., 1.),
+							pos: hit.incident.base - 0.02 * hit.incident.dir,
-					},
+							color: vec3(0., 0., 1.),
-				]);
+						},
+						Vertex {
+							pos: hit.incident.base,
+							color: vec3(1., 1., 1.),
+						},
+					]);
+				}
 			} else {
-				source_ray_display.extend([
+				if args.show_miss_emission {
-					Vertex {
+					source_ray_display.extend([
-						pos: ray.base,
+						Vertex {
-						color: vec3(1., 1., 1.),
+							pos: ray.base,
-					},
+							color: vec3(1., 1., 1.),
-					Vertex {
+						},
-						pos: ray.base + 0.1 * ray.dir,
+						Vertex {
-						color: vec3(1., 0., 0.),
+							pos: ray.base + 0.1 * ray.dir,
-					},
+							color: vec3(1., 0., 0.),
-				]);
+						},
+					]);
+				}
 			}
 		}
 		let mut hits2: Vec<Hit> = Vec::with_capacity(hits.len());
@@ -251,62 +270,70 @@ impl Core {
 				continue;
 			};
 			hits2.push(hit2);
-			source_ray_display.extend([
+			if args.show_indirect_hit {
-				Vertex {
+				source_ray_display.extend([
-					pos: hit2.incident.base - 0.02 * hit2.incident.dir,
+					Vertex {
-					color: vec3(1., 0., 1.),
+						pos: hit2.incident.base - 0.02 * hit2.incident.dir,
-				},
+						color: vec3(1., 0., 1.),
-				Vertex {
+					},
-					pos: hit2.incident.base,
+					Vertex {
-					color: vec3(1., 1., 1.),
+						pos: hit2.incident.base,
-				},
+						color: vec3(1., 1., 1.),
-			]);
+					},
+				]);
+			}
 		}
 		hits.extend(hits2);
 		let mut camera_ray_display: Vec<Vertex> = Vec::with_capacity(camera_rays.len());
-		let sigma2 = args.accum_sigma.powi(2);
+		if args.show_light {
-		let accum_normalizator = (2. * PI * sigma2).sqrt().recip();
+			let sigma2 = args.accum_sigma.powi(2);
-		for ray in camera_rays {
+			let accum_normalizator = (2. * PI * sigma2).sqrt().recip();
-			let Some(hit) = scene.trace_ray(ray) else {
+			for ray in camera_rays {
-				continue;
+				let Some(hit) = scene.trace_ray(ray) else {
-			};
-			let mut total_cd = 0.0f32;
-			for light_hit in &hits {
-				let d2 = hit.incident.base.distance_squared(light_hit.incident.base);
-				if d2 > 9. * sigma2 {
 					continue;
+				};
+				let mut total_cd = 0.0f32;
+				for light_hit in &hits {
+					let d2 = hit.incident.base.distance_squared(light_hit.incident.base);
+					if d2 > 9. * sigma2 {
+						continue;
+					}
+					assert!(hit.normal.is_normalized());
+					assert!(hit.incident.dir.is_normalized());
+					let reflector = Lambertian;
+					let in_lm = 1.0;
+					let out_cd = in_lm
+						* hit.normal.dot(-hit.incident.dir)
+						* reflector.brdf(hit.normal, hit.incident.dir, -ray.dir);
+					let weight = accum_normalizator * (-0.5 * d2 / sigma2).exp();
+					total_cd += weight * out_cd;
 				}
-				assert!(hit.normal.is_normalized());
+				let brightness = 3. * (1. - (1. + total_cd * args.accum_scale).recip());
-				assert!(hit.incident.dir.is_normalized());
+				let r = args.accum_sigma;
-				let reflector = Lambertian;
+				let color = vec3(brightness, brightness - 1., brightness - 2.)
-				let in_lm = 1.0;
+					.clamp(Vec3::splat(0.), Vec3::splat(1.));
-				let out_cd = in_lm
+				let vertex = |off: Vec3| Vertex {
-					* hit.normal.dot(-hit.incident.dir)
+					pos: hit.incident.base + r * off,
-					* reflector.brdf(hit.normal, hit.incident.dir, -ray.dir);
+					color,
-				let weight = accum_normalizator * (-0.5 * d2 / sigma2).exp();
+				};
-				total_cd += weight * out_cd;
+				camera_ray_display.extend([
+					vertex(-Vec3::X),
+					vertex(Vec3::X),
+					vertex(-Vec3::Y),
+					vertex(Vec3::Y),
+					vertex(-Vec3::Z),
+					vertex(Vec3::Z),
+				]);
 			}
-			let brightness = 3. * (1. - (1. + total_cd * args.accum_scale).recip());
-			let r = args.accum_sigma;
-			let color = vec3(brightness, brightness - 1., brightness - 2.)
-				.clamp(Vec3::splat(0.), Vec3::splat(1.));
-			let vertex = |off: Vec3| Vertex {
-				pos: hit.incident.base + r * off,
-				color,
-			};
-			camera_ray_display.extend([
-				vertex(-Vec3::X),
-				vertex(Vec3::X),
-				vertex(-Vec3::Y),
-				vertex(Vec3::Y),
-				vertex(-Vec3::Z),
-				vertex(Vec3::Z),
-			]);
 		}
-		self.pipeline
+		if !source_ray_display.is_empty() {
-			.render(&mut pass, [&Mesh::new(&self.device, &source_ray_display)]);
+			self.pipeline
-		self.pipeline
+				.render(&mut pass, [&Mesh::new(&self.device, &source_ray_display)]);
-			.render(&mut pass, [&Mesh::new(&self.device, &camera_ray_display)]);
+		}
+		if !camera_ray_display.is_empty() {
+			self.pipeline
+				.render(&mut pass, [&Mesh::new(&self.device, &camera_ray_display)]);
+		}
 
 		drop(pass);
 		self.queue.submit(std::iter::once(encoder.finish()));

ui/src/api.hxx

@@ -20,6 +20,13 @@ struct RedrawArgs {
 	float light_spread = 0.;
 	float accum_sigma = 1.;
 	float accum_scale = 1.;
+	bool show_axes = true;
+	bool show_shapes = true;
+	bool show_hit_emission = true;
+	bool show_miss_emission = true;
+	bool show_direct_hit = true;
+	bool show_indirect_hit = true;
+	bool show_light = true;
 };
 } // namespace ffi
 

ui/src/main_window.cxx

@@ -20,24 +20,33 @@ void PhotonLight::updateView() {
 		.camera_position = SphericalPosition{
 			.yaw = deg_to_rad(m_ui->cameraYaw->value()),
 			.pitch = deg_to_rad(m_ui->cameraPitch->value()),
-			.distance = 3.0,
+			.distance = m_ui->cameraDistance->value() / 10.0f,
 		},
 		.light_position = SphericalPosition{
 			.yaw = deg_to_rad(m_ui->lightYaw->value()),
 			.pitch = deg_to_rad(m_ui->lightPitch->value()),
-			.distance = 1.0,
+			.distance = m_ui->lightDistance->value() / 10.0f,
 		},
 		.light_radius = 0.125,
 		.light_spread = 0.125,
 		.accum_sigma = exp10f(m_ui->accumSigma->value() / 25.0),
 		.accum_scale = exp10f(m_ui->accumScale->value() / 25.0),
+		.show_axes = m_ui->displayAxes->isChecked(),
+		.show_shapes = m_ui->displayShapes->isChecked(),
+		.show_hit_emission = m_ui->displayEmitted->isChecked(),
+		.show_miss_emission = m_ui->displayEmitted->isChecked(),
+		.show_direct_hit = m_ui->displayDirectHits->isChecked(),
+		.show_indirect_hit = m_ui->displayIndirectHits->isChecked(),
+		.show_light = m_ui->displayResult->isChecked(),
 	};
 	m_ui->cameraYawLabel->setText(tr("Yaw: %1 deg").arg(QString::number(qRadiansToDegrees(args.camera_position.yaw))));
 	m_ui->cameraPitchLabel->setText(tr("Pitch: %1 deg").arg(QString::number(qRadiansToDegrees(args.camera_position.pitch))));
+	m_ui->cameraDistanceLabel->setText(tr("Distance: %1").arg(QString::number(args.camera_position.distance)));
 	m_ui->lightYawLabel->setText(tr("Yaw: %1 deg").arg(QString::number(qRadiansToDegrees(args.light_position.yaw))));
 	m_ui->lightPitchLabel->setText(tr("Pitch: %1 deg").arg(QString::number(qRadiansToDegrees(args.light_position.pitch))));
-	m_ui->accumSigmaLabel->setText(tr("Sigma: %1").arg(QString::number(args.accum_sigma, 'f', 3)));
+	m_ui->lightDistanceLabel->setText(tr("Distance: %1").arg(QString::number(args.light_position.distance)));
-	m_ui->accumScaleLabel->setText(tr("Scale: %1").arg(QString::number(args.accum_scale, 'f', 3)));
+	m_ui->accumSigmaLabel->setText(tr("Averaging radius: %1").arg(QString::number(args.accum_sigma, 'f', 3)));
+	m_ui->accumScaleLabel->setText(tr("Brightness: %1").arg(QString::number(args.accum_scale, 'f', 3)));
 	m_ui->viewport->setView(args);
 }
 

ui/src/main_window.ui

@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>800</width>
+    <width>1600</width>
-    <height>600</height>
+    <height>1200</height>
    </rect>
   </property>
   <property name="windowTitle">
@@ -25,7 +25,7 @@
     <rect>
      <x>0</x>
      <y>0</y>
-     <width>800</width>
+     <width>1600</width>
      <height>38</height>
     </rect>
    </property>
@@ -98,6 +98,26 @@
           </property>
          </widget>
         </item>
+        <item>
+         <widget class="QLabel" name="cameraDistanceLabel">
+          <property name="text">
+           <string>Distance</string>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QSlider" name="cameraDistance">
+          <property name="maximum">
+           <number>50</number>
+          </property>
+          <property name="value">
+           <number>30</number>
+          </property>
+          <property name="orientation">
+           <enum>Qt::Horizontal</enum>
+          </property>
+         </widget>
+        </item>
        </layout>
       </widget>
      </item>
@@ -159,19 +179,39 @@
           </property>
          </widget>
         </item>
+        <item>
+         <widget class="QLabel" name="lightDistanceLabel">
+          <property name="text">
+           <string>Distance</string>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QSlider" name="lightDistance">
+          <property name="maximum">
+           <number>50</number>
+          </property>
+          <property name="value">
+           <number>10</number>
+          </property>
+          <property name="orientation">
+           <enum>Qt::Horizontal</enum>
+          </property>
+         </widget>
+        </item>
        </layout>
       </widget>
      </item>
      <item>
       <widget class="QGroupBox" name="groupBox_3">
        <property name="title">
-        <string>Accumulating</string>
+        <string>Lighting</string>
        </property>
        <layout class="QVBoxLayout" name="verticalLayout_4">
         <item>
          <widget class="QLabel" name="accumSigmaLabel">
           <property name="text">
-           <string>Sigma</string>
+           <string>Averaging radius</string>
           </property>
          </widget>
         </item>
@@ -194,7 +234,7 @@
         <item>
          <widget class="QLabel" name="accumScaleLabel">
           <property name="text">
-           <string>Scale</string>
+           <string>Brightness</string>
           </property>
          </widget>
         </item>
@@ -217,6 +257,75 @@
        </layout>
       </widget>
      </item>
+     <item>
+      <widget class="QGroupBox" name="groupBox_4">
+       <property name="title">
+        <string>Show</string>
+       </property>
+       <layout class="QVBoxLayout" name="verticalLayout_5">
+        <item>
+         <widget class="QCheckBox" name="displayAxes">
+          <property name="text">
+           <string>Axes</string>
+          </property>
+          <property name="checked">
+           <bool>true</bool>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QCheckBox" name="displayShapes">
+          <property name="text">
+           <string>Shapes</string>
+          </property>
+          <property name="checked">
+           <bool>true</bool>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QCheckBox" name="displayEmitted">
+          <property name="text">
+           <string>Emitted rays</string>
+          </property>
+          <property name="checked">
+           <bool>false</bool>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QCheckBox" name="displayDirectHits">
+          <property name="text">
+           <string>Direct incident rays</string>
+          </property>
+          <property name="checked">
+           <bool>false</bool>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QCheckBox" name="displayIndirectHits">
+          <property name="text">
+           <string>Indirect incident rays</string>
+          </property>
+          <property name="checked">
+           <bool>false</bool>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <widget class="QCheckBox" name="displayResult">
+          <property name="text">
+           <string>Average light</string>
+          </property>
+          <property name="checked">
+           <bool>true</bool>
+          </property>
+         </widget>
+        </item>
+       </layout>
+      </widget>
+     </item>
      <item>
       <spacer name="verticalSpacer">
        <property name="orientation">
@@ -251,8 +360,8 @@
    <slot>updateView()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>688</x>
+     <x>1585</x>
-     <y>159</y>
+     <y>169</y>
     </hint>
     <hint type="destinationlabel">
      <x>403</x>
@@ -267,8 +376,8 @@
    <slot>updateView()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>688</x>
+     <x>1585</x>
-     <y>215</y>
+     <y>225</y>
     </hint>
     <hint type="destinationlabel">
      <x>403</x>
@@ -283,8 +392,8 @@
    <slot>updateView()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>688</x>
+     <x>1585</x>
-     <y>319</y>
+     <y>385</y>
     </hint>
     <hint type="destinationlabel">
      <x>403</x>
@@ -299,8 +408,8 @@
    <slot>updateView()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>688</x>
+     <x>1585</x>
-     <y>375</y>
+     <y>441</y>
     </hint>
     <hint type="destinationlabel">
      <x>403</x>
@@ -315,8 +424,8 @@
    <slot>updateView()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>729</x>
+     <x>1585</x>
-     <y>479</y>
+     <y>601</y>
     </hint>
     <hint type="destinationlabel">
      <x>399</x>
@@ -331,8 +440,8 @@
    <slot>updateView()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>729</x>
+     <x>1585</x>
-     <y>535</y>
+     <y>657</y>
     </hint>
     <hint type="destinationlabel">
      <x>399</x>
@@ -340,6 +449,134 @@
     </hint>
    </hints>
   </connection>
+  <connection>
+   <sender>cameraDistance</sender>
+   <signal>valueChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>271</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>lightDistance</sender>
+   <signal>valueChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>487</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>displayAxes</sender>
+   <signal>stateChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>799</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>displayDirectHits</sender>
+   <signal>stateChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>901</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>displayEmitted</sender>
+   <signal>stateChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>867</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>displayIndirectHits</sender>
+   <signal>stateChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>935</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>displayResult</sender>
+   <signal>stateChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>969</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>displayShapes</sender>
+   <signal>stateChanged(int)</signal>
+   <receiver>MainWindow</receiver>
+   <slot>updateView()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>1489</x>
+     <y>833</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>799</x>
+     <y>599</y>
+    </hint>
+   </hints>
+  </connection>
  </connections>
  <slots>
   <slot>updateView()</slot>