"""
MicroPython IoT Weather Station Example for Wokwi.com

To view the data:

1. Go to http://www.hivemq.com/demos/websocket-client/
2. Click "Connect"
3. Under Subscriptions, click "Add New Topic Subscription"
4. In the Topic field, type "wokwi-weather" then click "Subscribe"

Now click on the DHT22 sensor in the simulation,
change the temperature/humidity, and you should see
the message appear on the MQTT Broker, in the "Messages" pane.

Copyright (C) 2022, Uri Shaked

https://wokwi.com/arduino/projects/322577683855704658
"""

import network
import time
from machine import Pin,I2C
import dht
import ujson
from umqtt.simple import MQTTClient
import ssd1306
import _thread
from machine import Timer
import machine
# MQTT Server Parameters
MQTT_CLIENT_ID = "zhengyihan2206054074"
MQTT_BROKER    = "broker.emqx.io"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "zhengyihan2206054074"
TOPIC1 = b"ledctl2013206032"  #灯控主题，为了避免主题和其他同学一样，末尾需要添加自己的学号
TOPIC2 = b"pwmled2013206032"
TOPIC3 = b"ledstatus2013206032"


i2c=I2C(0,scl=Pin(22),sda=Pin(21))
oled_width=128
oled_height=64
oled=ssd1306.SSD1306_I2C(oled_width,oled_height,i2c)
oled.show()
sensor = dht.DHT22(Pin(15))
# 选择gpio2引脚
led_blue = machine.PWM(machine.Pin(2), freq=1000)  # 设置 GPIO2 为输出
led_blue.duty(1023)

print("Connecting to WiFi", end="")
sta_if = network.WLAN(network.STA_IF)
sta_if.active(True)
sta_if.connect('Wokwi-GUEST', '')
while not sta_if.isconnected():
  print(".", end="")
  time.sleep(0.1)
print(" Connected!")
#配置定义回调函数
def sub_cb(topic, msg):   #定义收到mqtt主题信息之后的回调函数
    global pwm_old, pwmval
    if topic == TOPIC1:
        print((topic, msg))
        if msg == b"on":
            pwmval = 0
            led_blue.duty(pwmval)
        elif msg == b"off":
            pwmval = 1023;
            led_blue.duty(pwmval)

    if topic == TOPIC2:
        pwmval = int(((100 - int(msg)) / 100) * 1023)
        if pwmval>1023 : pwmval=1023
        if pwmval<0:    pwmval=0
        print((topic, pwmval))
        led_blue.duty(pwmval)

print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(sub_cb)    #定义好client参数之后，就需要在connect之前
client.connect()
client.subscribe(TOPIC1)     #connect之后就需要马上订阅主题，才能收到mqtt服务器送过来的信息
client.subscribe(TOPIC2)     
print("Connected!")

def sample_light(name):
    while True:
        client.wait_msg()
        time.sleep(3)
prevr_weather = ""
def thread1(name):
    global count,oled,prevr_weather,temp,humidity
    print('%s is running'%name) 
    while True:
        sensor.measure()
        temp = sensor.temperature()
        humidity = sensor.humidity()
        msg_data={
            "temp": sensor.temperature(),
            "humidity": sensor.humidity(),
        }
        message=ujson.dumps(msg_data)
        if message != prevr_weather:
            print("Updated!")
            print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))
            client.publish(MQTT_TOPIC, message)
            prevr_weather = message
            oled.fill(0)
            oled.show()
            oled.text("%.1f dc"%(msg_data["temp"]),10,30)
            oled.text("%.1f %%"%(msg_data["humidity"]),10,40)
            oled.text('yihan',10,10)
            oled.text('2206054074',10,20)
            oled.show()
            time.sleep(1)
        else:
            print("%s:No change"%name)
        time.sleep(2)
_thread.start_new_thread(sample_light,("Measuring weather",))
_thread.start_new_thread(thread1,("Measuring weather",))