From 1573508f757588761008899594a91652fecab686 Mon Sep 17 00:00:00 2001
From: Laurens Valk <laurens@pybricks.com>
Date: Wed, 20 Jan 2021 11:05:09 +0100
Subject: [PATCH] examples/pup/primehub: auto rotate display

---
 doc/api/hubs/primehub.rst                     |  3 +++
 .../hub_primehub/display_orientation_imu.py   | 19 +++++++++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100644 examples/pup/hub_primehub/display_orientation_imu.py

diff --git a/doc/api/hubs/primehub.rst b/doc/api/hubs/primehub.rst
index 7e8d718..0e04e5f 100644
--- a/doc/api/hubs/primehub.rst
+++ b/doc/api/hubs/primehub.rst
@@ -145,6 +145,9 @@ Changing the display orientation
 .. literalinclude::
     ../../../examples/pup/hub_primehub/display_orientation.py
 
+.. literalinclude::
+    ../../../examples/pup/hub_primehub/display_orientation_imu.py
+
 Making your own images
 **********************
 
diff --git a/examples/pup/hub_primehub/display_orientation_imu.py b/examples/pup/hub_primehub/display_orientation_imu.py
new file mode 100644
index 0000000..e16e9b3
--- /dev/null
+++ b/examples/pup/hub_primehub/display_orientation_imu.py
@@ -0,0 +1,19 @@
+from pybricks.hubs import PrimeHub
+from pybricks.parameters import Icon
+from pybricks.tools import wait
+
+# Initialize the hub.
+hub = PrimeHub()
+
+while True:
+
+    # Check which side of the hub is up.
+    up_side = hub.imu.up()
+
+    # Use this side to set the display orientation.
+    hub.display.orientation(up_side)
+
+    # Display something, like an arrow.
+    hub.display.image(Icon.UP)
+
+    wait(10)