From 517ad1151742f11585055f32c51056c2c9c987bd Mon Sep 17 00:00:00 2001
From: Anatoly <akopyl@radner.ru>
Date: Sat, 23 Jan 2021 04:11:09 +0300
Subject: [PATCH] Added a tiny oled display

---
 cat.png  | Bin 0 -> 1905 bytes
 image.py |  23 +++++++++++++++++++++++
 main.py  |   8 +++++++-
 3 files changed, 30 insertions(+), 1 deletion(-)
 create mode 100644 cat.png
 create mode 100644 image.py

diff --git a/cat.png b/cat.png
new file mode 100644
index 0000000000000000000000000000000000000000..d730aad3098748f2eb8cb3cfa2cb35e24ecf784d
GIT binary patch
literal 1905
zcmcIlTWs4@7<Q_%ZlfI=f^}VsW?WXN!0|0{Vy{`#CSi$0O4X*Lyg*KlPZHC_c5Syw
zGoh7@fe>gE2~-LaJWy%C5aJ>s&?X8@6BE)H5-(6BiZ)SrK?5<6*no1J8x`8BzF^7r
z@j2i5{_nq>-%kt<>}%P)b2C8@EwN~L2*0QCO>S(!&ynB{CVp))qW4>bOL*3~UGLm*
z^hSc%K&XjPdo+HJ4E4MlD0&LHi+Ka138JUBXaG2dY%+y1surLwo>-trRS8h{dE;!{
z2%)SREtzPzG?0L$F(@fi?_RQ}C}V;=vH@Al=QK+$2B<QxjK|J2Ly={OJr<z)9E0R&
ze2@(3CL+CVk%pYdOZp@?Ckh@(x{Ksl&d;!Zh7)K`ls&A>^JMj+ur*Uj%R}M2tG4i0
zfXdpoAu~*&P;eInw{B(_PLd>s<r$u*F@m<HG#eCY&FZW%gpmbJ)v#4vBOOMN(kJWy
zg_TxZ$QyN9&8mh82aG8KgW=q)<5CS6kFO2Q=j&+8j!a@JRlQdwwh~hYVup~VPnZxz
zCXr@$){H5zzSo#Ab7fBz$e<j`W0-}7adlHhR=0I4t6#&9`f}YHLk^h;*t(g}^;|7c
zgEba16mk+Hk$1!aR5gdX+mTh>g~Gr_0g7jNj%GQU=My|DbArry?qmg-Wh>A)E(QhI
zV0~a+QPWfFfyU!<OtWmDK@<xIC@jjYs){V6eLm<HAq^BSOM8I8(|!pFw8yVN!6yk4
zk33~vcz;-j6HW}B{gv8Pbci|XVUh)o<rNQtv`_RxnooIBv;@6r+Rym}s0hf%rT7}#
zy{3wf56G=%byO*sF@@3|B&Jd{C$VYTqwo^#N61Hueh=WhfQ4QFC{lrPS~v3m7mJz)
z8N?V`h9dt}6Vh|K8P^pYb7A#-EEF0v^|YG93)axSelivbiJT;gH1Fn|+Q%}myHVA`
ziJq#SreQQ*9pzNAT%t07&dCT+&?yO|P}OPmDq&uqLlxRW7Gcsgs-yz5^t4?7ChE)J
zl&r55nQM7qfyuS?udhu0Z~fH(!z|D;2sd1Yay(+3)>KIk^WV&s*H*Q$8XWgVXIO8_
z_@my|k%p;F+_WG3($z~4uGUz%FOmKBr@4=}oFbb;vnwmv(vRT9u2FKgm^^y(Nc%#v
z|E<{#_Vc&-8-m^6{}qGtfB*6L+=}`Y`F8Xnb-Z_J`$u|mF1a|y4VNeawq3rReb)uQ
z?g(}qK8T)rc=u8>xB~WkcTCJLyz!~))gyt>Msexbo+f4cYtot7#+h!{xep$_DLBvm
zcJj%ymrw5x>hG17uP2UlH1B8(5{<+IrPT3lxk$LM<xJlvhhJVi^wQb*W6QsG9SELi
zU;HMwb(-1q=g7%j!2`E+w2<ddkMGK4qdWFrdZG1Pw)4ux3*y4eFYV8){Qi33!_Qr}
zK9cTjot}vlmyfqKxtp#dx>}yTG*6s=Zm)F#v=Q5S#*+=k7h8#+-zUUZcCrV3pPh1k
Oim}K*_|5+0p??5N18v>_

literal 0
HcmV?d00001

diff --git a/image.py b/image.py
new file mode 100644
index 0000000..2999689
--- /dev/null
+++ b/image.py
@@ -0,0 +1,23 @@
+import time
+
+import Adafruit_GPIO.SPI as SPI
+import Adafruit_SSD1306
+
+from PIL import Image
+
+RST = 24
+disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST)
+
+disp.begin()
+disp.clear()
+disp.display()
+
+image = Image.open('cat.png').convert('1')
+
+def cat():
+	disp.image(image)
+	disp.display()
+
+def no_cat():
+	disp.clear()
+	disp.display()
diff --git a/main.py b/main.py
index e0e5f6d..5c4d3d9 100644
--- a/main.py
+++ b/main.py
@@ -5,6 +5,8 @@ import importlib.util
 from threading import Thread
 import time
 
+from image import *
+
 IM_WIDTH = 1280
 IM_HEIGHT = 720
 
@@ -110,7 +112,11 @@ def pet_detector(frame, detection_time):
                 detection_time += 1
             else:
                 detection_time = 0
-            print(detection_time)   
+                no_cat()
+            print(detection_time)
+            if detection_time >= 2:
+                cat()   
+
 
     return frame, detection_time