yeelight灯泡wifi控制器

vany5921

      一个简单的Yeelight灯泡无线M5控制器，可替代手机进行操作，如果手机连接Yeelight时设置了自动光线补偿，那么会对控制器的调光产生一定影响。控制器闲置3秒自动休眠，按B键唤醒，通过陀螺仪调整灯泡亮度，C键开关灯控制。开机黑屏，接入wifi后屏幕会显示内容。   

1. #define M5STACK_MPU6886
   
2. #include <ArduinoJson.h>
   
3. #include <M5Stack.h>
   
4. #include <WiFi.h>
   
5. 
   
6. #include "yeelight.h"
   
7. 
   
8. #define SSID "SSID"
   
9. #define WIFI_PWD "PASSWORD"
   
10. #define SLEEP_WAIT 3000
    
11. #define ACCEL_THRESHOLD 200
    
12. #define MAX_DIMMER_VALUE 50000
    
13. #define DIMMER_TO_PERCENT(param) ((100 * (param)) / MAX_DIMMER_VALUE)
    
14. 
    
15. void angleDimmer(void* xParam);
    
16. void sleepWait(void* xParam);
    
17. 
    
18. void waitForWiFi();
    
19. void lightPowerToggle();
    
20. 
    
21. void lightAdjustBrightness();
    
22. 
    
23. void power_on_rutine();
    
24. 
    
25. TaskHandle_t sleepTaskHandle;
    
26. TaskHandle_t angleTaskHandle;
    
27. Yeelight* yeelight;
    
28. 
    
29. bool initialized = false;
    
30. bool inactive = false;
    
31. 
    
32. time_t last_activity = millis();
    
33. int32_t dimmer_value = 0;
    
34. int current_bright = 1;
    
35. int toggle_counter = 1;
    
36. 
    
37. String brightness_params;
    
38. 
    
39. void setup() {
    
40.   M5.begin();
    
41. 
    
42.   M5.Power.begin();
    
43.   M5.Power.setWakeupButton(BUTTON_B_PIN);
    
44.   M5.IMU.Init();
    
45. 
    
46.   WiFi.begin(SSID, WIFI_PWD);
    
47. 
    
48.   waitForWiFi();
    
49. 
    
50.   M5.Lcd.fillScreen(BLACK);
    
51.   M5.Lcd.setTextColor(GREEN, BLACK);
    
52.   M5.Lcd.setTextSize(2);
    
53. 
    
54.   yeelight = new Yeelight();
    
55.   yeelight->lookup();
    
56. }
    
57. 
    
58. void loop() {
    
59.   M5.update();
    
60. 
    
61.   if (!initialized) {
    
62.     power_on_rutine();
    
63.     initialized = true;
    
64.     inactive = false;
    
65.   }
    
66. 
    
67.   if (M5.BtnC.wasPressed()) {
    
68.     lightPowerToggle();
    
69.   }
    
70. 
    
71.   Serial.println(current_bright);
    
72. 
    
73.   lightAdjustBrightness();
    
74. 
    
75.   if (inactive) {
    
76.     vTaskDelete(angleTaskHandle);
    
77.     vTaskDelete(sleepTaskHandle);
    
78.     initialized = false;
    
79.     M5.Power.lightSleep();
    
80.   }
    
81.   delay(100);
    
82. }
    
83. 
    
84. void lightPowerToggle() {
    
85.   last_activity = millis();
    
86. 
    
87.   M5.Lcd.setCursor(0, 60);
    
88.   M5.Lcd.print("Sending power toggle ");
    
89.   M5.Lcd.println(toggle_counter++);
    
90.   yeelight->sendCommand("toggle", "[]");
    
91. }
    
92. 
    
93. void lightAdjustBrightness() {
    
94.   int new_bright = (int)(dimmer_value / (MAX_DIMMER_VALUE / 100));
    
95.   String brightness_params;
    
96. 
    
97.   if (new_bright < 1) {
    
98.     new_bright = 1;
    
99.   }
    
100. 
     
101.   if (new_bright != current_bright) {
     
102.     current_bright = new_bright;
     
103.     brightness_params = String("[") + new_bright + ", "smooth", 500]";
     
104. 
     
105.     Serial.println(yeelight->sendCommand("set_bright", brightness_params));
     
106.   }
     
107. }
     
108. 
     
109. void power_on_rutine() {
     
110.   M5.Lcd.fillScreen(BLACK);
     
111. 
     
112.   last_activity = millis();
     
113.   xTaskCreate(angleDimmer, "angleDimmer", 4096, NULL, 10, &angleTaskHandle);
     
114.   xTaskCreate(sleepWait, "sleepWait", XT_STACK_MIN_SIZE, NULL, 9,
     
115.               &sleepTaskHandle);
     
116. 
     
117.   waitForWiFi();
     
118.   M5.Lcd.setCursor(0, 0);
     
119.   M5.Lcd.println("Connected to Wifi");
     
120. 
     
121.   while (!yeelight->feedback()) {
     
122.     yeelight->lookup();
     
123.     delay(50);
     
124.   }
     
125.   M5.Lcd.setCursor(0, 20);
     
126.   M5.Lcd.println("Light detected");
     
127. }
     
128. 
     
129. void waitForWiFi() {
     
130.   while (WiFi.status() != WL_CONNECTED) {
     
131.     delay(100);
     
132.     Serial.print(F("."));
     
133.   }
     
134. }
     
135. 
     
136. void angleDimmer(void* xParam) {
     
137.   int16_t gx, gy, gz;
     
138. 
     
139.   while (1) {
     
140.     M5.IMU.getGyroAdc(&gx, &gy, &gz);
     
141. 
     
142.     if (gz >= ACCEL_THRESHOLD || gz <= -ACCEL_THRESHOLD) {
     
143.       dimmer_value -= gz;  // I want right turns to increment the brightness
     
144.       last_activity = millis();
     
145.     }
     
146. 
     
147.     if (dimmer_value >= MAX_DIMMER_VALUE) {
     
148.       dimmer_value = MAX_DIMMER_VALUE;
     
149.     } else if (dimmer_value <= 0) {
     
150.       dimmer_value = 0;
     
151.     }
     
152.     vTaskDelay(100 * portTICK_PERIOD_MS);
     
153.   }
     
154. }
     
155. 
     
156. void sleepWait(void* xParam) {
     
157.   while (1) {
     
158.     if (millis() > last_activity + SLEEP_WAIT) {
     
159.       inactive = true;
     
160.     }
     
161.     M5.Lcd.setCursor(0, 100);
     
162.     M5.Lcd.print("Brightness:    %");
     
163.     M5.Lcd.setCursor(140, 100);
     
164.     M5.Lcd.println(current_bright);
     
165.     vTaskDelay(500 * portTICK_PERIOD_MS);
     
166.   }
     
167. }

复制代码

yeelight.cpp

1. #ifdef ESP32
   
2. #include <WiFi.h>
   
3. #else
   
4. #include <ESP8266WiFi.h>
   
5. #endif
   
6. 
   
7. #include <WiFiUdp.h>
   
8. 
   
9. #include "yeelight.h"
   
10. 
    
11. WiFiUDP _udp;
    
12. WiFiClient _client;
    
13. IPAddress _ipMulti(239, 255, 255, 250);
    
14. char _packetBuffer[550];
    
15. 
    
16. String _location, _support;
    
17. bool _powered;
    
18. char _server[15];
    
19. 
    
20. uint16_t _cmdid = 0, _port = 0;
    
21. 
    
22. Yeelight::Yeelight() {}
    
23. 
    
24. Yeelight::~Yeelight() { _udp.stop(); }
    
25. 
    
26. void Yeelight::lookup() {
    
27. #ifdef ESP32
    
28.   _udp.beginMulticast(_ipMulti, 1982);
    
29.   _udp.beginMulticastPacket();
    
30. #else
    
31.   _udp.beginMulticast(WiFi.localIP(), _ipMulti, 1982);
    
32.   _udp.beginPacketMulticast(_ipMulti, 1982, WiFi.localIP());
    
33. #endif
    
34.   _udp.print(
    
35.       "M-SEARCH * HTTP/1.1\r\nHOST: 239.255.255.250:1982\r\nMAN: "
    
36.       ""ssdp:discover"\r\nST: wifi_bulb");
    
37.   _udp.endPacket();
    
38.   _udp.begin(1982);
    
39. }
    
40. 
    
41. int Yeelight::feedback() {
    
42.   int packetSize = _udp.parsePacket();
    
43.   if (packetSize) {
    
44.     int len = _udp.read(_packetBuffer, 550);
    
45.     if (len > 0) {
    
46.       _packetBuffer[len] = 0;
    
47.     }
    
48.     parseFeedback(_packetBuffer, len);
    
49.   }
    
50.   return packetSize;
    
51. }
    
52. 
    
53. void Yeelight::parseFeedback(char* buffer, size_t len) {
    
54.   int i = 0, _i = 0;
    
55.   char _b[255];
    
56.   while (i < len) {
    
57.     if (buffer[i] == '\r' && i + 1 <= len && buffer[i + 1] == '\n') {
    
58.       _b[_i] = 0;
    
59. 
    
60.       // ----
    
61.       String _str = String(_b);
    
62.       if (_str.startsWith("Location: yeelight://")) {
    
63.         int colon = _str.indexOf(':', 21) + 1;
    
64.         _location = _str.substring(10);
    
65.         _str.substring(21, colon).toCharArray(_server, colon - 21);
    
66.         _port = _str.substring(colon).toInt();
    
67.       }
    
68.       if (_str.startsWith("support: ")) {
    
69.         _support = _str.substring(9);
    
70.       }
    
71.       if (_str.startsWith("power: ")) {
    
72.         _powered = _str.substring(7) == "on";
    
73.       }
    
74.       // ----
    
75. 
    
76.       i = i + 2;
    
77.       _i = 0;
    
78.     } else {
    
79.       _b[_i] = buffer[i];
    
80.       i++;
    
81.       _i++;
    
82.     }
    
83.   }
    
84. }
    
85. 
    
86. String Yeelight::sendCommand(String method, String params) {
    
87.   if (_client.connect(_server, _port)) {
    
88.     String payload = String("") + "{"id":" + (++_cmdid) + ","method":"" +
    
89.                      method + "","params":" + params + "}";
    
90.     _client.println(payload);
    
91.   }
    
92. 
    
93.   String result = "";
    
94.   while (_client.connected()) {
    
95.     result = _client.readStringUntil('\r');
    
96.     _client.stop();
    
97.   }
    
98.   return result;
    
99. }
    
100. 
     
101. bool Yeelight::isPowered() { return _powered; }
     
102. 
     
103. String Yeelight::getSupport() { return _support; }
     
104. 
     
105. String Yeelight::getLocation() { return _location; }

复制代码

yeelight.h

1. #ifndef Yeelight_h
   
2. #define Yeelight_h
   
3. 
   
4. class Yeelight {
   
5. public:
   
6.   Yeelight();
   
7.   ~Yeelight();
   
8.   void lookup();
   
9.   int feedback();
   
10.   String getLocation();
    
11.   String getSupport();
    
12.   bool isPowered();
    
13.   String sendCommand(String method, String params);
    
14. 
    
15. private:
    
16.   void parseFeedback(char* buffer, size_t len);
    
17. };
    
18. 
    
19. #endif

复制代码