From 5e86145822d9690a54ab91d6199e0c9a6035d8eb Mon Sep 17 00:00:00 2001
From: NikolajDanger <nikolajgade@live.dk>
Date: Wed, 7 Sep 2022 14:39:55 +0200
Subject: [PATCH] :sparkles:

---
 wiggle.py | 76 ++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 70 insertions(+), 6 deletions(-)

diff --git a/wiggle.py b/wiggle.py
index e3dd98b..c5d9509 100644
--- a/wiggle.py
+++ b/wiggle.py
@@ -10,7 +10,55 @@ RIGHT_WHEEL_OFFSET = 4
 
 CLOCKWISE_OFFSET = 0.82
 
-def loop(arlo):
+MORSE = {
+  "0": "-----",
+  "1": ".----",
+  "2": "..---",
+  "3": "...--",
+  "4": "....-",
+  "5": ".....",
+  "6": "-....",
+  "7": "--...",
+  "8": "---..",
+  "9": "----.",
+  "a": ".-",
+  "b": "-...",
+  "c": "-.-.",
+  "d": "-..",
+  "e": ".",
+  "f": "..-.",
+  "g": "--.",
+  "h": "....",
+  "i": "..",
+  "j": ".---",
+  "k": "-.-",
+  "l": ".-..",
+  "m": "--",
+  "n": "-.",
+  "o": "---",
+  "p": ".--.",
+  "q": "--.-",
+  "r": ".-.",
+  "s": "...",
+  "t": "-",
+  "u": "..-",
+  "v": "...-",
+  "w": ".--",
+  "x": "-..-",
+  "y": "-.--",
+  "z": "--..",
+  ".": ".-.-.-",
+  ",": "--..--",
+  "?": "..--..",
+  "!": "-.-.--",
+  "-": "-....-",
+  "/": "-..-.",
+  "@": ".--.-.",
+  "(": "-.--.",
+  ")": "-.--.-"
+}
+
+def short_wiggle(arlo):
     arlo.go_diff(POWER, POWER + RIGHT_WHEEL_OFFSET, 1, 0)
     sleep(TURN_T * CLOCKWISE_OFFSET)
     arlo.stop()
@@ -19,15 +67,31 @@ def loop(arlo):
     sleep(TURN_T)
     arlo.stop()
 
+def long_wiggle(arlo):
+    arlo.go_diff(POWER, POWER + RIGHT_WHEEL_OFFSET, 1, 0)
+    sleep(TURN_T * CLOCKWISE_OFFSET * 2)
+    arlo.stop()
+    sleep(0.3)
+    arlo.go_diff(POWER, POWER + RIGHT_WHEEL_OFFSET, 0, 1)
+    sleep(TURN_T * 2)
+    arlo.stop()
+
 def main():
     # Initializes the robot and runs the
     arlo = robot.Robot()
 
-    try:
-        loop(arlo)
-
-    except KeyboardInterrupt:
-        arlo.stop()
+    text = input()
+    for word in text.split(" "):
+        for character in word:
+            morse_code = MORSE[character]
+            for morse_character in morse_code:
+                if morse_character == ".":
+                    short_wiggle(arlo)
+                else:
+                    long_wiggle(arlo)
+                sleep(0.2)
+            sleep(0.5)
+        sleep(1)
 
 
 if __name__ == "__main__":