From 82fca7d325ebe1e2459195139bac8fdcbd17eb0d Mon Sep 17 00:00:00 2001
From: baldurk <baldurk@baldurk.org>
Date: Wed, 15 Jan 2020 09:29:46 +0000
Subject: [PATCH] Make the display_window.py example quit after a few loops

---
 .../examples/renderdoc/display_window.py      | 44 ++++++++++++-------
 1 file changed, 28 insertions(+), 16 deletions(-)

diff --git a/docs/python_api/examples/renderdoc/display_window.py b/docs/python_api/examples/renderdoc/display_window.py
index de3d108d2..390c5942c 100644
--- a/docs/python_api/examples/renderdoc/display_window.py
+++ b/docs/python_api/examples/renderdoc/display_window.py
@@ -84,38 +84,50 @@ def paint():
 	window.after(33, paint)
 
 # Start on the first drawcall
-curdraw = 0
+curdraw = draws[0]
 
-# The advance function will be called every 150ms, to move to the next draw
+loopcount = 0
+
+# The advance function will be called every 50ms, to move to the next draw
 def advance():
-	global out, window, curdraw
+	global out, window, curdraw, draws, loopcount
 
 	# Move to the current drawcall
-	controller.SetFrameEvent(draws[curdraw].eventId, False)
+	controller.SetFrameEvent(curdraw.eventId, False)
 
 	# Initialise a default TextureDisplay object
 	disp = rd.TextureDisplay()
 
 	# Set the first colour output as the texture to display
-	disp.resourceId = draws[curdraw].outputs[0]
+	disp.resourceId = curdraw.outputs[0]
 
-	# Get the details of this texture
-	texDetails = getTexture(disp.resourceId)
+	if disp.resourceId != rd.ResourceId.Null():
+		# Get the details of this texture
+		texDetails = getTexture(disp.resourceId)
 
-	# Calculate the scale required in width and height
-	widthScale = window.winfo_width() / texDetails.width
-	heightScale = window.winfo_height() / texDetails.height
+		# Calculate the scale required in width and height
+		widthScale = window.winfo_width() / texDetails.width
+		heightScale = window.winfo_height() / texDetails.height
 
-	# Use the lower scale to fit the texture on the window
-	disp.scale = min(widthScale, heightScale)
+		# Use the lower scale to fit the texture on the window
+		disp.scale = min(widthScale, heightScale)
 
-	# Update the texture display
-	out.SetTextureDisplay(disp)
+		# Update the texture display
+		out.SetTextureDisplay(disp)
 
 	# Set the next drawcall
-	curdraw = (curdraw + 1) % len(draws)
+	curdraw = curdraw.next
 
-	window.after(150, advance)
+	# If we have no next draw, start again from the first
+	if curdraw is None:
+		loopcount = loopcount + 1
+		curdraw = draws[0]
+
+	# after 3 loops, quit
+	if loopcount == 3:
+		window.quit()
+	else:
+		window.after(50, advance)
 
 # Start the callbacks
 advance()