HackUTOKYOv2Pim

commit 57e859a9581a3310f5f9761b0d35b0cf11d12916
parent 1fdaba87a64c63a7ae8b0e4112e10a0f66d5ffb8
Author: 諒太 林 <ryouturn@gmail.com>
Date:   Fri, 30 Aug 2024 20:34:11 +0900

update test

Diffstat:
M
test.py
 | 
113
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++----------------------

1 file changed, 82 insertions(+), 31 deletions(-)
diff --git a/test.py b/test.py
@@ -1,51 +1,98 @@
+#!/usr/bin/python
+# coding: utf-8
 import RPi.GPIO as GPIO
 import time
+import sys
+import cv2
+import os
 
 stepPin = 22
 dircPin = 17
 enabPin = 23
-num = 100
 
-GPIO.setmode(GPIO.BCM)  # GPIOで指定
-GPIO.setup(enabPin, GPIO.OUT)   # 2:Enableに定義
-GPIO.setup(dircPin, GPIO.OUT)   # 3:Dir
-GPIO.setup(stepPin, GPIO.OUT)   # 4:Step
-print("setup was completed")
+camLimit = 5
+
+#GPIO.setmode(GPIO.BCM)  # GPIOで指定
+#GPIO.setup(enabPin, GPIO.OUT)   # 2:Enableに定義
+#GPIO.setup(dircPin, GPIO.OUT)   # 3:Dir
+#GPIO.setup(stepPin, GPIO.OUT)   # 4:Step
 
 #GPIO.setup(17, GPIO.OUT)#17:MS1
 #GPIO.setup(27, GPIO.OUT)#27:MS2
 #GPIO.setup(22, GPIO.OUT)#22:MS3
 
 def main():
-    GPIO.output(enabPin, 0)
-    print("enable setted 1")
-    for i in range(0,10000):
-        print(i)
-        GPIO.output(stepPin, 1)
-        time.sleep(0.001)
-        GPIO.output(stepPin, 0)
-        time.sleep(0.001)
-    GPIO.output(enabPin, 1)
-    GPIO.cleanup()
+    #GPIO.output(enabPin, 0)
+    while(True):
+        ret, frame = cap.read()
+
+        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) 
+        mos = mosaic(gray)
+        count = dump(gray, keep)
+             
+        if count[0] >= camLimit:
+            if count[1] < camLimit:
+                escapeR
+            elif count[0] > count[1]:
+                escapeR
+            else:
+                escapeL
+        elif count[1] >= camLimit:
+            escapeL
+
+        cv2.imshow('frame',frame)
+        cv2.imshow('mos',mos)
+
+        if cv2.waitKey(1) != -1:
+            break
+    
+    #GPIO.cleanup()
 
 #逆かもしれない
 def escapeR():
-    GPIO.output(dircPin, 0)
-    for i in range(0,200):
-            GPIO.output(stepPin, 1)
-            time.sleep(0.001)
-            GPIO.output(stepPin, 0)
-            time.sleep(0.001)
-    time.sleep(1)
-    
+    print("escapeR")
 def escapeL():
-    GPIO.output(dircPin, 1)
-    for i in range(0,200):
-            GPIO.output(stepPin, 1)
-            time.sleep(0.001)
-            GPIO.output(stepPin, 0)
-            time.sleep(0.001)
-    time.sleep(1)
+    print("escapeL")
+
+def mosaic(src):
+    dst = cv2.resize(src, None, fx=0.1, fy=0.1, interpolation=cv2.INTER_NEAREST)
+    return cv2.resize(dst, src.shape[:2][::-1], interpolation=cv2.INTER_NEAREST)
+
+def difference(a, b):
+    if(a < b):
+        return b - a
+    return a - b
+
+def dump(src, keep):
+    os.system('clear')
+    RightCount = 0
+    LeftCount = 0
+    for x in range(0, 32):
+        str = ''
+        for y in range(0, 24):
+            target = src[y*10][x*10]    
+            diff = difference(keep[y][x],target)
+            if diff > 50:
+                str += '{:02} '.format(diff) 
+                if(y >= 12):
+                     RightCount += 1
+                else:
+                     LeftCount += 1
+            else:
+                str += '-- '
+            keep[y][x] = target
+        print(str)
+    count=[RightCount,LeftCount]
+    return count
+# 仕様メモcount=[RightCount,LeftCount]
+
+cap = cv2.VideoCapture(1)
+
+cap.set(3,320) # WIDTH
+cap.set(4,240) # HEIGHT
+cap.set(5,30) # FPS
+
+keep = [[0] * 32 for i in range(24)]
 
 busy = True
 
@@ -54,3 +101,6 @@ try:
 except KeyboardInterrupt:
         print('interrupted!')
         GPIO.cleanup()
+
+cap.release()
+cv2.destroyAllWindows()+
\ No newline at end of file