#新物联网挑战二#水培蔬菜智能温室系统 简单

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

# MindPlus
# Python
import base64
from io import BytesIO
from PIL import Image
from pinpong.libs.dfrobot_sht31 import SHT31
from pinpong.libs.dfrobot_tds import TDS
from pinpong.extension.unihiker import *
from pinpong.board import Board,Pin
from pinpong.board import DS18B20
from dfrobot_rp2040_sci import *
from pinpong.board import Board
from pinpong.board import Pin
from unihiker import GUI
import siot
import time
import cv2

# 事件回调函数
def u_thread2_function():
    while True:
        if (light.read() < 200):
            p_p23_out=Pin(Pin.P23, Pin.OUT)
            p_p23_out.write_digital(1)
            siot.publish_save(topic="siot/补光开关", data="on")
            time.sleep((60 * (60 * 0.5)))
            p_p23_out=Pin(Pin.P23, Pin.OUT)
            p_p23_out.write_digital(0)
            siot.publish_save(topic="siot/补光开关", data="off")
            time.sleep(600)
def u_thread1_function():
    while True:
        p_p22_out=Pin(Pin.P22, Pin.OUT)
        p_p22_out.write_digital(1)
        siot.publish_save(topic="siot/水泵开关", data="off")
        time.sleep((60 * (60 * 2)))
        p_p22_out=Pin(Pin.P22, Pin.OUT)
        p_p22_out.write_digital(0)
        siot.publish_save(topic="siot/水泵开关", data="on")
        time.sleep((60 * (60 * 0.5)))
def on_message_callback(client, userdata, msg):
    if (msg.topic.find("siot/水泵开关")!=-1):
        if (msg.payload.decode().find("on")!=-1):
            p_p22_out=Pin(Pin.P22, Pin.OUT)
            p_p22_out.write_digital(1)
        elif (msg.payload.decode().find("off")!=-1):
            p_p22_out=Pin(Pin.P22, Pin.OUT)
            p_p22_out.write_digital(0)
    elif (msg.topic.find("siot/补光开关")!=-1):
        if (msg.payload.decode().find("on")!=-1):
            p_p23_out=Pin(Pin.P23, Pin.OUT)
            p_p23_out.write_digital(1)
        elif (msg.payload.decode().find("off")!=-1):
            p_p23_out=Pin(Pin.P23, Pin.OUT)
            p_p23_out.write_digital(0)


Board().begin()
u_gui=GUI()
siot.init(client_id="938137717436466",server="10.1.2.3",port=1883,user="siot",password="dfrobot")
siot.connect()
siot.loop()
siot.set_callback(on_message_callback)
p_sht31 = SHT31(0x45)
tds1 = TDS(Pin((Pin.P21)))
ds1 = DS18B20(Pin((Pin.P24)))
x = 20
y = 0
size = 18
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.")
t0=u_gui.draw_text(text="行空板",x=x,y=0,font_size=25, color="#FF0000")
t1=u_gui.draw_text(text="行空板",x=x,y=60,font_size=size, color="#0000FF")
t2=u_gui.draw_text(text="行空板",x=x,y=90,font_size=size, color="#0000FF")
t3=u_gui.draw_text(text="行空板",x=x,y=120,font_size=size, color="#0000FF")
t4=u_gui.draw_text(text="行空板",x=x,y=150,font_size=size, color="#0000FF")
t5=u_gui.draw_text(text="行空板",x=x,y=180,font_size=size, color="#0000FF")
t6=u_gui.draw_text(text="行空板",x=x,y=210,font_size=size, color="#0000FF")
t7=u_gui.draw_text(text="行空板",x=x,y=240,font_size=size, color="#0000FF")
t8=u_gui.draw_text(text="行空板",x=x,y=270,font_size=size, color="#0000FF")

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
vd = cv2.VideoCapture()
vd.open(0)
while not (vd.isOpened()):
    print("摄像头初始化..")
print("摄像头初始化成功")
siot.getsubscribe(topic="siot/补光开关")
siot.getsubscribe(topic="siot/水泵开关")
mytime = 0
thread1=u_gui.start_thread(u_thread1_function)
thread2=u_gui.start_thread(u_thread2_function)

while True:
    YingYangYeFeiLi = tds1.get_value()
    QiWen = p_sht31.temp_c()
    ShiDu = p_sht31.humidity()
    GuangZhao = light.read()
    YangQi = SCI1.get_value0("O2")
    ShuiWen = ds1.temp_c()
    CO2 = SCI1.get_value0("ECO2")
    ShuiWei = SCI1.get_value0("Moisture")
    t0.config(text="智能水培系统")
    t1.config(text=(str("肥力：") + str(YingYangYeFeiLi)))
    t2.config(text=(str("气温：") + str(QiWen)))
    t3.config(text=(str("湿度：") + str(ShiDu)))
    t4.config(text=(str("光照：") + str(GuangZhao)))
    t5.config(text=(str("氧气：") + str(YangQi)))
    t6.config(text=(str("水温：") + str(ShuiWen)))
    t7.config(text=(str("CO2:  ") + str(CO2)))
    t8.config(text=(str("水位：") + str(ShuiWei)))
    siot.publish_save(topic="siot/温室温度", data=QiWen)
    siot.publish_save(topic="siot/氧气指数", data=YangQi)
    siot.publish_save(topic="siot/温室光照", data=GuangZhao)
    siot.publish_save(topic="siot/营养液水位", data=ShuiWei)
    siot.publish_save(topic="siot/温室湿度", data=ShiDu)
    siot.publish_save(topic="siot/营养液肥力", data=YingYangYeFeiLi)
    siot.publish_save(topic="siot/营养液水温", data=ShuiWen)
    ret, grab = vd.read()
    siot.publish(topic="siot/温室画面", data=base642base64(grab))