import log
import utime as time
from machine import I2C, Pin, UART


# | 参数      | 参数类型 | 说明                | 类型      |
# | -------- | ------- | ------------------ | -------- |
# | CRITICAL | 常量     | 日志记录级别的数值 50 | critical |
# | ERROR    | 常量     | 日志记录级别的数值 40 | error    |
# | WARNING  | 常量     | 日志记录级别的数值 30 | warning  |
# | INFO     | 常量     | 日志记录级别的数值 20 | info     |
# | DEBUG    | 常量     | 日志记录级别的数值 10 | debug    |
# | NOTSET   | 常量     | 日志记录级别的数值 0  | notset   |
log.basicConfig(level=log.NOTSET)   # 设置日志输出级别
# uart_x = UART(UART.UART0, 115200, 8, 0, 1, 0)
# log.set_output(uart_x)
Grey_log = log.getLogger("Grey")


class aht10class:
    i2c_dev = None
    i2c_addre = None

    AHT10_CALIBRATION_CMD = 0xE1  # 初始化命令
    AHT10_START_MEASURMENT_CMD = 0xAC  # 触发测量
    AHT10_RESET_CMD = 0xBA  # 复位

    def write_data(self, reg, data):
        self.i2c_dev.write(self.i2c_addre, bytearray(reg), len(reg), bytearray(data), len(data))
        # Grey_log.debug('[write]addr:0x{:X} deta:{}'.format((self.i2c_addre << 1)+0, data))
        pass

    def read_data(self, data, length):
        r_data = [0x00 for _ in range(length)]
        r_data = bytearray(r_data)
        self.i2c_dev.read(self.i2c_addre, bytearray(data), len(data), r_data, length, 50)
        # Grey_log.debug('[read]addr:0x{:X} deta:{}'.format((self.i2c_addre << 1)+1, r_data))
        return list(r_data)

    def aht10_init(self, addre=0x38):
        # self.i2c_dev = I2C(I2C.I2C1, I2C.STANDARD_MODE)  # EC600U
        self.i2c_dev = I2C(I2C.I2C0, I2C.STANDARD_MODE)  # EC200U
        # i2c_x = I2C(I2C.I2C0, I2C.STANDARD_MODE)  # 此处我将IIC0跟IIC1进行了并联, 未屏蔽干扰需要此句代码.
        self.i2c_addre = addre
        # 校准
        self.write_data([self.AHT10_CALIBRATION_CMD, 0x08], [0x00])
        time.sleep_ms(300)  # at last 300ms

    @staticmethod
    def aht10_transformation_temperature(data):
        r_data = data
        # 根据数据手册的描述来转化温度
        humidity = (r_data[0] << 12) | (
            r_data[1] << 4) | ((r_data[2] & 0xF0) >> 4)
        humidity = (humidity/(1 << 20)) * 100.0
        temperature = ((r_data[2] & 0xf) << 16) | (
            r_data[3] << 8) | r_data[4]
        temperature = (temperature * 200.0 / (1 << 20)) - 50
        Grey_log.info("当前温度: {:.2f} ℃".format(temperature))
        Grey_log.info("当前湿度: {:.2f} ％".format(humidity))
        time.sleep_ms(500)

    def ath10_reset(self):
        self.write_data([self.AHT10_RESET_CMD])
        time.sleep_ms(20)  # at last 20ms

    def trigger_measurement(self):
        self.write_data([self.AHT10_START_MEASURMENT_CMD], [0x33, 0x00])
        time.sleep_ms(200)
        r_data = self.read_data([0x00, 0x00], 6)
        # check bit7
        if (r_data[0] >> 7) != 0x0:
            # Grey_log.error("换算错误")
            pass
        else:
            self.aht10_transformation_temperature(r_data[1:6])


def i2c_aht10_test():
    ath_dev = aht10class()
    ath_dev.aht10_init(addre=0x38)

    # 测试十次
    for i in range(100):
    # while True:
        ath_dev.trigger_measurement()


if __name__ == "__main__":
    i2c_aht10_test()