Merge branch 'master' of https://github.com/BPI-STEAM/BPI-BIT-MicroPython

Author: junhuanchen

10.microbit/Sensegesture.py

@@ -1,31 +1,106 @@
+from time import sleep_ms
+
 
 class LightIntensity():
     def __init__(self, pin):
         from machine import ADC, Pin
         self.adc = ADC(Pin(pin, Pin.IN))
-        self.adc.atten(ADC.ATTN_11DB) # 0-3.9V
+        self.adc.atten(ADC.ATTN_11DB)  # 0-3.9V
+
     def read(self):
-        self.last_val=self.new_val
+        self.last_val = self.new_val
         self.new_val = self.adc.read()/5
         return int(self.new_val*5/40.95)
 
+
+class Sensegesture():
+    idle, running, finish = 0, 1, 2
+
+    def __init__(self, leftPin=36, rightpin=39):
+        from light1 import LightIntensity
+        self.LightL = LightIntensity(leftPin)
+        self.LightR = LightIntensity(rightpin)
+
+    def get_brightness(self):
+        self.LightR.read()
+        self.LightL.read()
+
+    def set_threshold(self, val):  # 设置阈值
+        self.threshold = self.LightR.new_val*val
+
+    def is_right(self, runtime=10, interval=0.5):
+        status = 0
+        self.get_brightness()
+        self.set_threshold(0.1)
+        for t in range(runtime*40):
+            if status == Sensegesture.idle:
+                count = 0
+                self.get_brightness()
+                # print(self.threshold)
+                # print('right_new=%d' % self.LightR.new_val)
+                sleep_ms(20)
+                if self.LightL.last_val-self.LightL.new_val > self.threshold and self.LightR.new_val-self.LightL.new_val > self.threshold:
+                    status = self.running
+            elif status == Sensegesture.running:
+                self.get_brightness()
+                # print(self.threshold)
+                # print(self.LightL.last_val)
+                # print('left_new=%d' % self.LightL.new_val)
+                sleep_ms(20)
+                count += 1
+                if count > interval*40:
+                    status = Sensegesture.idle
+                if self.LightR.last_val-self.LightR.new_val > self.threshold:
+                    status = self.finish
+            elif status == Sensegesture.finish:
+                return True
+        return False
+
+    def is_left(self, runtime=10, interval=0.5):
+        status = 0
+        self.get_brightness()
+        self.set_threshold(0.1)
+        for t in range(runtime*40):
+            if status == Sensegesture.idle:
+                count = 0
+                self.get_brightness()
+                # print('right_new=%d' % self.LightR.new_val)
+                sleep_ms(20)
+                if self.LightR.last_val-self.LightR.new_val > self.threshold and self.LightL.new_val-self.LightR.new_val > self.threshold:
+                    status = Sensegesture.running
+            elif status == Sensegesture.running:
+                self.get_brightness()
+                # print(self.threshold)
+                # print(self.LightL.last_val)
+                # print('left_new=%d' % self.LightL.new_val)
+                sleep_ms(20)
+                count += 1
+                if count > interval*40:
+                    status = Sensegesture.idle
+                if self.LightL.last_val-self.LightL.new_val > self.threshold:
+                    status = Sensegesture.finish
+
+            elif status == Sensegesture.finish:
+                return True
+        return False
+
+
 def unit_test():
     print('\n\
-        from time import sleep\n\
-        LightL = LightIntensity(36)\n\
-        LightR = LightIntensity(39)\n\
-        while True:\n\
-            print("LightL", LightL.read())\n\
-            print("LightR", LightR.read())\n\
-            sleep(0.5)\n\
+        from machine import Pin\n\
+        t = Sensegesture()\n\
+        LED = Pin(18, Pin.OUT)\n\
+        LED.value(0)\n\
+        if(t.is_right(runtime=15)):\n\
+            LED.value(1)\n\
     ')
-    from time import sleep
-    LightL = LightIntensity(36)
-    LightR = LightIntensity(39)
-    while True:
-        print('LightL', LightL.read())
-        print('LightR', LightR.read())
-        sleep(0.5)
+    from machine import Pin
+    t = Sensegesture()
+    LED = Pin(18, Pin.OUT)
+    LED.value(0)
+    if(t.is_right(runtime=15)):
+        LED.value(1)
+
 
 if __name__ == '__main__':
     unit_test()