#决赛#学霸助手 简单

#  -*- coding: UTF-8 -*-

# MindPlus
# Python
import base64
from io import BytesIO
from PIL import Image
from pinpong.libs.dfrobot_ens160 import Ens160
from pinpong.libs.dfrobot_sht31 import SHT31
from pinpong.extension.unihiker import *
from pinpong.board import Board,Pin
from dfrobot_rp2040_sci import *
from pinpong.board import Board
from unihiker import Audio
from unihiker import GUI
import math
import siot
import time
import cv2


Board().begin()
siot.init(client_id="7626107718319322",server="192.168.31.128",port=1883,user="siot",password="dfrobot")
u_gui=GUI()
u_audio = Audio()
siot.connect()
siot.loop()
p_p21_in=Pin(Pin.P21, Pin.IN)
SCI1 = DFRobot_RP2040_SCI_IIC(addr=0x21)
while SCI1.begin() != 0:
    print("Initialization Sensor Universal Adapter Board failed.")
    time.sleep(1)
print("Initialization Sensor Universal Adapter Board done.")
p_ens160 = Ens160(0x52)
p_sht31 = SHT31(0x45)
ZuoWeiKaiGuan = False
BiGanZiTai = False
ShuXieShiChang = 0
ShuXieKaiShi = 0
YueDuKaiShi = 0
YueDuShiChang = 0
YueDuQiZhuangTai = False
YouXiaoShiChang = 0
XueXiShiChang = 0
XueXiZhuangTai = False
XueXiKaiShi = 0
ZheCiDeLiKaiShiChang = 0
KaiShiLiKai = 0
LiKaiShiChang = 0
XueXiJiHua = (str("6:00~6:30 朗读并背诵I have  a dream      6:30~7:10 听写CNN 10      7:10~7:30 早餐      7:30~8:30  临习米芾尺牍《至知府大夫尺牍》      8：30~9:30 学习B站宋浩的偏导数第一课      9：30~10：00 coffe break      10:00~11:00 跟粒子姐姐学mind+python编程      11：00~12：00上浦育innolab继续autobilling项目      12：00~12：30午餐      12：30~13：30午休      13：30~14:30 练习古琴《广陵散》      14：30~15：30陶艺课      15：30~16：00 coffe break      16:00~17:00 1000米游泳练习      17：00~17：30 晚餐      17：30~19：30 英语社团辩论活动      19：30~20：30 街舞课       20：30~21：00 营养点心      21：00~22：00 阅读与日记      22：00~22：40 家庭分享      22：40~23：00 睡前准备      23：00 休息 "))
ShiDu = 78
QiWen = 26
ZaoYin = 18
GuangZhao = 85
YangQi = 95

def frame2base64(frame):
    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    img = Image.fromarray(frame) #将每一帧转为Image
    output_buffer = BytesIO() #创建一个BytesIO
    img.save(output_buffer, format='JPEG') #写入output_buffer
    byte_data = output_buffer.getvalue() #在内存中读取
    base64_data = base64.b64encode(byte_data) #转为BASE64
    return base64_data #转码成功 返回base64编码

def base642base64(frame):
    data=str('data:image/png;base64,')
    base64data = str(frame2base64(frame))
    framedata = base64data[2:(len(base64data)-1)]
    base642base64_data = data + str(framedata)
    return base642base64_data
siot.getsubscribe(topic="siot/学习状态")
siot.getsubscribe(topic="siot/学习画面")
img2 = cv2.imread("/root/xueba/bizhi.png", cv2.IMREAD_UNCHANGED)
siot.publish(topic="siot/学习画面", data=base642base64(img2))
vd = cv2.VideoCapture()
vd.open(0)
while not (vd.isOpened()):
    print("摄像头初始化..")
print("摄像头初始化成功")
txt=u_gui.draw_text(text="休息....",x=70,y=220,font_size=20, color="#0000FF")
bq=u_gui.draw_emoji(emoji="Smile",x=0,y=0,duration=0.2)
img0 = cv2.imread("/root/xueba/0.jpg", cv2.IMREAD_UNCHANGED)
siot.publish(topic="siot/学习状态", data=base642base64(img0))
while not ((not (p_p21_in.read_digital()==True))):
    pass
KaiShiLiKai = time.time()
XueXiZhuangTai = True
txt.config(text="加油！")
bq.config(emoji="Think")
siot.publish_save(topic="siot/学习计划", data=XueXiJiHua)
img1 = cv2.imread("/root/xueba/1.jpg", cv2.IMREAD_UNCHANGED)
siot.publish(topic="siot/学习状态", data=base642base64(img1))
siot.publish_save(topic="siot/正在做什么", data="学霸准备中...")
time.sleep(1)
while not ((XueXiZhuangTai == False)):
    if vd.grab():
        ret, grab = vd.read()
        siot.publish(topic="siot/学习画面", data=base642base64(grab))
    ShiDu = p_sht31.humidity()
    QiWen = p_sht31.temp_c()
    ZaoYin = u_audio.sound_level()
    GuangZhao = light.read()
    YangQi = p_ens160.get_aqi()
    siot.publish_save(topic="siot/书房湿度", data=ShiDu)
    siot.publish_save(topic="siot/书房温度", data=QiWen)
    siot.publish_save(topic="siot/书房噪音", data=ZaoYin)
    siot.publish_save(topic="siot/书房光线", data=GuangZhao)
    siot.publish_save(topic="siot/空气质量指数", data=YangQi)
    # 这两个如果的功能是获取书写时间，用x轴的加速度是否<0.5做为判定是否处于书写状态。
    if (((int(float(SCI1.get_value1(SCI1.eALL,"Mag_Z")))) > 0) and (BiGanZiTai == True)):
        siot.publish_save(topic="siot/正在做什么", data="停止书写")
        BiGanZiTai = False
        ShuXieShiChang = (ShuXieShiChang + (time.time() - ShuXieKaiShi))
    if (((int(float(SCI1.get_value1(SCI1.eALL,"Mag_Z")))) < 0) and (BiGanZiTai == False)):
        siot.publish_save(topic="siot/正在做什么", data="正在书写")
        BiGanZiTai = True
        ShuXieKaiShi = time.time()
    # 这两个如果的功能是获取阅读时间，用光线传感器的值做为判定是否处于阅读状态。
    if (((int(float(SCI1.get_value1(SCI1.eALL,"Light")))) < 40) and (YueDuQiZhuangTai == False)):
        siot.publish_save(topic="siot/正在做什么", data="正在阅读")
        YueDuKaiShi = time.time()
        YueDuQiZhuangTai = True
    if (((int(float(SCI1.get_value1(SCI1.eALL,"Light")))) >= 40) and (YueDuQiZhuangTai == True)):
        siot.publish_save(topic="siot/正在做什么", data="停止阅读")
        YueDuShiChang = (YueDuShiChang + (time.time() - YueDuKaiShi))
        YueDuQiZhuangTai = False
    # 开关一松开，人离开，开始持续计算离开的时长。
    if ((p_p21_in.read_digital()==True) == True):
        if (ZuoWeiKaiGuan == True):
            ZuoWeiKaiGuan = False
            siot.publish_save(topic="siot/正在做什么", data="离开椅子")
            KaiShiLiKai = time.time()
            XueXiShiChang = (XueXiShiChang + (time.time() - XueXiKaiShi))
        ZheCiDeLiKaiShiChang = (time.time() - KaiShiLiKai)
        if (ZheCiDeLiKaiShiChang > 10):
            XueXiZhuangTai = False
    else:
        # 开关刚按下，开始学习，学习时间开始计时。
        # 结算离开的时长。
        if (ZuoWeiKaiGuan == False):
            LiKaiShiChang = (LiKaiShiChang + (time.time() - KaiShiLiKai))
            siot.publish_save(topic="siot/正在做什么", data="回到座位")
            XueXiKaiShi = time.time()
            ZuoWeiKaiGuan = True
siot.publish_save(topic="siot/学习时长", data=(str((int(ShuXieShiChang))) + str((str("，") + str((str((int(YueDuShiChang))) + str((str("，") + str((int(LiKaiShiChang)))))))))))
if ((ShuXieShiChang + YueDuShiChang) < 100):
    img2 = cv2.imread("/root/xueba/bizhi.png", cv2.IMREAD_UNCHANGED)
    siot.publish(topic="siot/学习画面", data=base642base64(img2))
else:
    img2 = cv2.imread("/root/xueba/bizhi.png", cv2.IMREAD_UNCHANGED)
    siot.publish(topic="siot/学习画面", data=base642base64(img2))
txt.config(text="休息...")
bq.config(emoji="Sleep")
img0 = cv2.imread("/root/xueba/0.jpg", cv2.IMREAD_UNCHANGED)
siot.publish(topic="siot/学习状态", data=base642base64(img0))