test

README.md

@@ -67,3 +67,4 @@ cp -r ../spike-prime-api/hub ./
 cp -r ../spike-prime-api/spike ./
 ```
 
+test

iqrobot.py

@@ -1,9 +1,9 @@
-# LEGO type:standard slot:6 autostart
+# LEGO type:standard slot:7 autostart
 
-from spike import PrimeHub, Motor, MotorPair, ColorSensor
+from spike import PrimeHub, Motor, MotorPair, ColorSensor, MotionSensor
 from spike.control import wait_for_seconds
 
-HELLO = "HELLO IQ"
+HELLO = "HELLO IQ 2"
 
 '''
 Wir nutzen "Duck typing", dh wir schreiben hinter jede Variabel mit ':' die Klasse, zB `leftMotor: Motor`
@@ -11,12 +11,29 @@ damit man dann später auch wieder Code Completion hat bei Nutzung der Variablen
 '''
 class IQRobot:
 
-    def __init__(self, hub: PrimeHub, leftMotorPort: str, rightMotorPort: str, colorSensorPort: str):
+    def __init__(self, hub: PrimeHub, colorSensorPort: str, typ: str):
         self.hub: PrimeHub = hub
-        self.leftMotor: Motor = Motor(leftMotorPort)
-        self.rightMotor: Motor = Motor(rightMotorPort)
-        self.movementMotors: MotorPair = MotorPair(leftMotorPort, rightMotorPort) 
-        self.colorSensor: ColorSensor = ColorSensor(colorSensorPort)
+        self.typ=typ
+        if self.typ=="backstein":
+            LEFT_MOTOR_PORT = 'F'
+            RIGHT_MOTOR_PORT = 'B'
+            FRONT_MOTOR_RIGHT_PORT = "E"
+            self.frontMotorRight: Motor = Motor(FRONT_MOTOR_RIGHT_PORT)
+        elif self.typ=="brickies":
+            LEFT_MOTOR_PORT = 'E'
+            RIGHT_MOTOR_PORT = 'F'
+            FRONT_MOTOR_RIGHT_PORT = "B"
+            FRONT_MOTOR_LEFT_PORT = "A"
+            self.frontMotorRight: Motor = Motor(FRONT_MOTOR_RIGHT_PORT)
+            self.frontMotorLeft: Motor = Motor(FRONT_MOTOR_LEFT_PORT)
+            self.bothFrontMotors: MotorPair = MotorPair(FRONT_MOTOR_LEFT_PORT, FRONT_MOTOR_RIGHT_PORT)
+
+        self.leftMotor: Motor = Motor(LEFT_MOTOR_PORT)
+        self.rightMotor: Motor = Motor(RIGHT_MOTOR_PORT)
+        self.movementMotors: MotorPair = MotorPair(LEFT_MOTOR_PORT, RIGHT_MOTOR_PORT) 
+        #self.colorSensor: ColorSensor = ColorSensor(colorSensorPort)
+        #self.frontMotorLeft: Motor = Motor("C")
+        self.motionSensor: MotionSensor = MotionSensor()
 
 
     def show(self, image: str):
@@ -38,9 +55,128 @@ class IQRobot:
         (red, green, blue, colorIntensity) = self.colorSensor.get_rgb_intensity()
         return colorIntensity
 
+    def drehe(self, grad=90, with_reset=True):
+        if with_reset:
+            self.motionSensor.reset_yaw_angle()
+        steering = 100 if grad > 0 else -100
+        self.movementMotors.start(steering=steering, speed=10)
+        while abs(self.motionSensor.get_yaw_angle()) < abs(grad):
+            pass
+        self.movementMotors.stop()
 
 
+    def drehe_robot(self, grad=90):
+        if self.typ == "backstein":
+            radius=9.5
+            stift_versatz=2.2
+        if self.typ == "brickies":
+            radius=17.4
+            stift_versatz=0.3
+        self.fahre_gerade(-radius - stift_versatz)
+        self.drehe(grad)
+        self.fahre_gerade(radius - stift_versatz)
 
+    def fahre_gerade(self, cm, zeichne=False):
+        if zeichne:
+            self.bewege_stift(1) # Stift runter
+        self.motionSensor.reset_yaw_angle()
+        if self.typ == "brickies":
+            cm = -cm
+        self.movementMotors.move(cm)
+        if zeichne:
+            self.bewege_stift(-1) # Stift hoch
+        versatz = self.motionSensor.get_yaw_angle()
+        self.drehe(grad=-versatz)
+
+    def buchstabe_zu_segmenten(self, buchstabe):
+        # Segmente um Buchstaben zu schreiben
+        #          4_
+        #       5 |__|3
+        #       0 |6_|2
+        #          1
+        #
+        buchstabe_zu_segmenten = {"L": [1,1,0,0,0,1,0], "E": [1,1,0,0,1,1,1], "G": [1,1,1,0,1,1,0], "O": [1,1,1,1,1,1,0]}
+        return buchstabe_zu_segmenten[buchstabe]
+
+
+    def bewege_stift(self, richtung):
+        if self.typ == "backstein":
+            self.frontMotorRight.run_for_rotations(richtung*0.4)
+        if self.typ == "brickies":
+            #print("bewege stift brickies")
+            self.bothFrontMotors.move(-richtung*0.2, unit='rotations', speed=5)
+
+    def schreibe_buchstabe(self, buchstabe):
+        print("Schreibe " + buchstabe)
+        segmente = self.buchstabe_zu_segmenten(buchstabe)
+        grad_drehung=-90
+        self.fahre_gerade(2)
+        self.drehe_robot(-grad_drehung) # drehe rechts
+        for segment_nummer, segment in enumerate(segmente):
+            print("Segment: " + str(segment) + " , Segment Nummer: " + str(segment_nummer))
+            if segment==1:
+                self.fahre_gerade(5, zeichne=True)
+            else:
+                self.fahre_gerade(5)
+            if segment_nummer != 2 and segment_nummer != 6:
+                self.drehe_robot(grad_drehung) # drehe links
+            
+
+    def schreibeL(self, schreibe=True, zurueck=False):
+        if zurueck: 
+            step = 5
+            faktor = -1
+        else:
+            step = 1
+            faktor = 1
+        print("Schreibe L")
+        #self.frontMotorRight.run_for_rotations(-0.4)
+        radius=9.5
+        stift_versatz=2.2
+        if schreibe:
+            self.frontMotorRight.run_for_rotations(0.4)
+        self.movementMotors.set_default_speed(10)
+
+        while (True):
+            if step == 0:
+                break
+            if step == 1:
+                self.movementMotors.move(faktor * 5)
+                if schreibe:
+                    self.frontMotorRight.run_for_rotations(-0.4)
+            if step == 2:
+                self.movementMotors.move(faktor * (-radius - stift_versatz))
+            if step == 3:
+                self.drehe(faktor * -90)
+            if step == 4:
+                self.movementMotors.move(faktor*(radius - stift_versatz))
+                if schreibe:
+                    self.frontMotorRight.run_for_rotations(0.4)
+            if step == 5:
+                self.movementMotors.move(faktor * 2)
+                if schreibe:
+                    self.frontMotorRight.run_for_rotations(-0.4)
+            if step == 6:
+                break
+            step += faktor
+        # Fahre 5 cm rückwerts  
+        # dann drehe nach rechts 90°
+        # und fahre 2cm fohrwärts 
+        #stift hoch
+
+
+    def schreibeLego(self):
+        #self.schreibeL()
+        #self.schreibeL(schreibe=False, zurueck=True)
+        self.movementMotors.set_default_speed(10)
+        self.bewege_stift(-1)
+        self.fahre_gerade(4, zeichne=True)
+        self.drehe_robot()
+        self.fahre_gerade(4, zeichne=True)
+        #self.schreibe_buchstabe("L")
+        #self.schreibe_buchstabe("E")
+        #self.schreibe_buchstabe("G")
+        #self.schreibe_buchstabe("O")
 
 print("successfully loaded the IQ Lego teams code :)")
 

main.py

@@ -1,4 +1,4 @@
-# LEGO type:standard slot:5 autostart
+# LEGO type:standard slot:4 autostart
 
 import os, sys
 
@@ -33,9 +33,10 @@ def importFile(slotid=0, precompiled=False, module_name='importFile'):
         program = f.read()
         print(program)
     try:
-        os.remove("/"+module_name+suffix)
+        os.remove("/"+module_name+".py")
+        os.remove("/"+module_name+".mpy")
     except:
-        pass
+        print("Couldn't remove old module")
     with open("/"+module_name+suffix,"w+") as f:
         print("trying to write import program")
         f.write(program)
@@ -50,7 +51,7 @@ def importFile(slotid=0, precompiled=False, module_name='importFile'):
 
 # Importiere Code aus der Datei "iqrobot.py"
 # Dateiname und Modulname sollten gleich sein, dann kann man Code Completion nutzen
-importFile(slotid=6, precompiled=True, module_name="iqrobot")
+importFile(slotid=7, precompiled=True, module_name="iqrobot")
 import iqrobot as iq
 print(iq.HELLO) 
 
@@ -71,22 +72,19 @@ dh auch an die Funktionen im importierten Code übergeben werde
 '''
 
 # Definiere an welchen Ports die Sensoren angeschlossen sind 
-COLOR_SENSOR_PORT = 'E'
-LEFT_MOTOR_PORT = 'A'
-RIGHT_MOTOR_PORT = 'B'
+COLOR_SENSOR_PORT = 'E' #not implemented yet
+
 
 # Initialisieren des Hubs, der Aktoren und Sensoren
 hub = PrimeHub()
 
 # Initialisiere Robot Klasse mit unseren Funktionen
-iqRobot: iq.IQRobot = iq.IQRobot(hub, LEFT_MOTOR_PORT, RIGHT_MOTOR_PORT, COLOR_SENSOR_PORT)
+iqRobot: iq.IQRobot = iq.IQRobot(hub, COLOR_SENSOR_PORT, typ="brickies")
 
 # Führe Funktionen aus unser Robot Klasse aus:
 iqRobot.show('HAPPY')
-iqRobot.driveForward_for_sec(2.0)
-colorIntensity = iqRobot.getColorIntensity()
-print("Farbintensität: " + str(colorIntensity))
-
+iqRobot.schreibeLego()
+#iqRobot.schreibeL()