mirror of
https://github.com/esphome/esphome.git
synced 2024-11-27 09:18:00 +01:00
Merge branch 'dev' into nrf52_zigbee
This commit is contained in:
commit
b98cfd28d1
175 changed files with 3563 additions and 586 deletions
2
.github/actions/restore-python/action.yml
vendored
2
.github/actions/restore-python/action.yml
vendored
|
@ -17,7 +17,7 @@ runs:
|
||||||
steps:
|
steps:
|
||||||
- name: Set up Python ${{ inputs.python-version }}
|
- name: Set up Python ${{ inputs.python-version }}
|
||||||
id: python
|
id: python
|
||||||
uses: actions/setup-python@v5.1.1
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: ${{ inputs.python-version }}
|
python-version: ${{ inputs.python-version }}
|
||||||
- name: Restore Python virtual environment
|
- name: Restore Python virtual environment
|
||||||
|
|
2
.github/workflows/ci-api-proto.yml
vendored
2
.github/workflows/ci-api-proto.yml
vendored
|
@ -23,7 +23,7 @@ jobs:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4.1.7
|
uses: actions/checkout@v4.1.7
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5.1.0
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: "3.11"
|
python-version: "3.11"
|
||||||
|
|
||||||
|
|
2
.github/workflows/ci-docker.yml
vendored
2
.github/workflows/ci-docker.yml
vendored
|
@ -42,7 +42,7 @@ jobs:
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4.1.7
|
- uses: actions/checkout@v4.1.7
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5.1.0
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: "3.9"
|
python-version: "3.9"
|
||||||
- name: Set up Docker Buildx
|
- name: Set up Docker Buildx
|
||||||
|
|
2
.github/workflows/ci.yml
vendored
2
.github/workflows/ci.yml
vendored
|
@ -41,7 +41,7 @@ jobs:
|
||||||
run: echo key="${{ hashFiles('requirements.txt', 'requirements_optional.txt', 'requirements_test.txt') }}" >> $GITHUB_OUTPUT
|
run: echo key="${{ hashFiles('requirements.txt', 'requirements_optional.txt', 'requirements_test.txt') }}" >> $GITHUB_OUTPUT
|
||||||
- name: Set up Python ${{ env.DEFAULT_PYTHON }}
|
- name: Set up Python ${{ env.DEFAULT_PYTHON }}
|
||||||
id: python
|
id: python
|
||||||
uses: actions/setup-python@v5.1.0
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: ${{ env.DEFAULT_PYTHON }}
|
python-version: ${{ env.DEFAULT_PYTHON }}
|
||||||
- name: Restore Python virtual environment
|
- name: Restore Python virtual environment
|
||||||
|
|
6
.github/workflows/release.yml
vendored
6
.github/workflows/release.yml
vendored
|
@ -53,7 +53,7 @@ jobs:
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4.1.7
|
- uses: actions/checkout@v4.1.7
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5.1.0
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: "3.x"
|
python-version: "3.x"
|
||||||
- name: Set up python environment
|
- name: Set up python environment
|
||||||
|
@ -85,7 +85,7 @@ jobs:
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4.1.7
|
- uses: actions/checkout@v4.1.7
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5.1.0
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: "3.9"
|
python-version: "3.9"
|
||||||
|
|
||||||
|
@ -141,7 +141,7 @@ jobs:
|
||||||
echo name=$(cat /tmp/platform) >> $GITHUB_OUTPUT
|
echo name=$(cat /tmp/platform) >> $GITHUB_OUTPUT
|
||||||
|
|
||||||
- name: Upload digests
|
- name: Upload digests
|
||||||
uses: actions/upload-artifact@v4.3.4
|
uses: actions/upload-artifact@v4.4.0
|
||||||
with:
|
with:
|
||||||
name: digests-${{ steps.sanitize.outputs.name }}
|
name: digests-${{ steps.sanitize.outputs.name }}
|
||||||
path: /tmp/digests
|
path: /tmp/digests
|
||||||
|
|
2
.github/workflows/sync-device-classes.yml
vendored
2
.github/workflows/sync-device-classes.yml
vendored
|
@ -22,7 +22,7 @@ jobs:
|
||||||
path: lib/home-assistant
|
path: lib/home-assistant
|
||||||
|
|
||||||
- name: Setup Python
|
- name: Setup Python
|
||||||
uses: actions/setup-python@v5.1.0
|
uses: actions/setup-python@v5.2.0
|
||||||
with:
|
with:
|
||||||
python-version: 3.12
|
python-version: 3.12
|
||||||
|
|
||||||
|
|
|
@ -58,6 +58,7 @@ esphome/components/beken_spi_led_strip/* @Mat931
|
||||||
esphome/components/bh1750/* @OttoWinter
|
esphome/components/bh1750/* @OttoWinter
|
||||||
esphome/components/binary_sensor/* @esphome/core
|
esphome/components/binary_sensor/* @esphome/core
|
||||||
esphome/components/bk72xx/* @kuba2k2
|
esphome/components/bk72xx/* @kuba2k2
|
||||||
|
esphome/components/bl0906/* @athom-tech @jesserockz @tarontop
|
||||||
esphome/components/bl0939/* @ziceva
|
esphome/components/bl0939/* @ziceva
|
||||||
esphome/components/bl0940/* @tobias-
|
esphome/components/bl0940/* @tobias-
|
||||||
esphome/components/bl0942/* @dbuezas
|
esphome/components/bl0942/* @dbuezas
|
||||||
|
@ -82,6 +83,7 @@ esphome/components/cap1188/* @mreditor97
|
||||||
esphome/components/captive_portal/* @OttoWinter
|
esphome/components/captive_portal/* @OttoWinter
|
||||||
esphome/components/ccs811/* @habbie
|
esphome/components/ccs811/* @habbie
|
||||||
esphome/components/cd74hc4067/* @asoehlke
|
esphome/components/cd74hc4067/* @asoehlke
|
||||||
|
esphome/components/ch422g/* @jesterret
|
||||||
esphome/components/climate/* @esphome/core
|
esphome/components/climate/* @esphome/core
|
||||||
esphome/components/climate_ir/* @glmnet
|
esphome/components/climate_ir/* @glmnet
|
||||||
esphome/components/color_temperature/* @jesserockz
|
esphome/components/color_temperature/* @jesserockz
|
||||||
|
@ -422,6 +424,7 @@ esphome/components/tuya/switch/* @jesserockz
|
||||||
esphome/components/tuya/text_sensor/* @dentra
|
esphome/components/tuya/text_sensor/* @dentra
|
||||||
esphome/components/uart/* @esphome/core
|
esphome/components/uart/* @esphome/core
|
||||||
esphome/components/uart/button/* @ssieb
|
esphome/components/uart/button/* @ssieb
|
||||||
|
esphome/components/udp/* @clydebarrow
|
||||||
esphome/components/ufire_ec/* @pvizeli
|
esphome/components/ufire_ec/* @pvizeli
|
||||||
esphome/components/ufire_ise/* @pvizeli
|
esphome/components/ufire_ise/* @pvizeli
|
||||||
esphome/components/ultrasonic/* @OttoWinter
|
esphome/components/ultrasonic/* @OttoWinter
|
||||||
|
|
|
@ -34,8 +34,8 @@ RUN \
|
||||||
python3-wheel=0.38.4-2 \
|
python3-wheel=0.38.4-2 \
|
||||||
iputils-ping=3:20221126-1 \
|
iputils-ping=3:20221126-1 \
|
||||||
git=1:2.39.2-1.1 \
|
git=1:2.39.2-1.1 \
|
||||||
curl=7.88.1-10+deb12u6 \
|
curl=7.88.1-10+deb12u7 \
|
||||||
openssh-client=1:9.2p1-2+deb12u2 \
|
openssh-client=1:9.2p1-2+deb12u3 \
|
||||||
python3-cffi=1.15.1-5 \
|
python3-cffi=1.15.1-5 \
|
||||||
libcairo2=1.16.0-7 \
|
libcairo2=1.16.0-7 \
|
||||||
libmagic1=1:5.44-3 \
|
libmagic1=1:5.44-3 \
|
||||||
|
@ -49,7 +49,7 @@ RUN \
|
||||||
zlib1g-dev=1:1.2.13.dfsg-1 \
|
zlib1g-dev=1:1.2.13.dfsg-1 \
|
||||||
libjpeg-dev=1:2.1.5-2 \
|
libjpeg-dev=1:2.1.5-2 \
|
||||||
libfreetype-dev=2.12.1+dfsg-5+deb12u3 \
|
libfreetype-dev=2.12.1+dfsg-5+deb12u3 \
|
||||||
libssl-dev=3.0.13-1~deb12u1 \
|
libssl-dev=3.0.14-1~deb12u1 \
|
||||||
libffi-dev=3.4.4-1 \
|
libffi-dev=3.4.4-1 \
|
||||||
libopenjp2-7=2.5.0-2 \
|
libopenjp2-7=2.5.0-2 \
|
||||||
libtiff6=4.5.0-6+deb12u1 \
|
libtiff6=4.5.0-6+deb12u1 \
|
||||||
|
|
|
@ -1107,6 +1107,19 @@ enum MediaPlayerCommand {
|
||||||
MEDIA_PLAYER_COMMAND_MUTE = 3;
|
MEDIA_PLAYER_COMMAND_MUTE = 3;
|
||||||
MEDIA_PLAYER_COMMAND_UNMUTE = 4;
|
MEDIA_PLAYER_COMMAND_UNMUTE = 4;
|
||||||
}
|
}
|
||||||
|
enum MediaPlayerFormatPurpose {
|
||||||
|
MEDIA_PLAYER_FORMAT_PURPOSE_DEFAULT = 0;
|
||||||
|
MEDIA_PLAYER_FORMAT_PURPOSE_ANNOUNCEMENT = 1;
|
||||||
|
}
|
||||||
|
message MediaPlayerSupportedFormat {
|
||||||
|
option (id) = 119;
|
||||||
|
option (ifdef) = "USE_MEDIA_PLAYER";
|
||||||
|
|
||||||
|
string format = 1;
|
||||||
|
uint32 sample_rate = 2;
|
||||||
|
uint32 num_channels = 3;
|
||||||
|
MediaPlayerFormatPurpose purpose = 4;
|
||||||
|
}
|
||||||
message ListEntitiesMediaPlayerResponse {
|
message ListEntitiesMediaPlayerResponse {
|
||||||
option (id) = 63;
|
option (id) = 63;
|
||||||
option (source) = SOURCE_SERVER;
|
option (source) = SOURCE_SERVER;
|
||||||
|
@ -1122,6 +1135,8 @@ message ListEntitiesMediaPlayerResponse {
|
||||||
EntityCategory entity_category = 7;
|
EntityCategory entity_category = 7;
|
||||||
|
|
||||||
bool supports_pause = 8;
|
bool supports_pause = 8;
|
||||||
|
|
||||||
|
repeated MediaPlayerSupportedFormat supported_formats = 9;
|
||||||
}
|
}
|
||||||
message MediaPlayerStateResponse {
|
message MediaPlayerStateResponse {
|
||||||
option (id) = 64;
|
option (id) = 64;
|
||||||
|
|
|
@ -179,6 +179,7 @@ void APIConnection::loop() {
|
||||||
SubscribeHomeAssistantStateResponse resp;
|
SubscribeHomeAssistantStateResponse resp;
|
||||||
resp.entity_id = it.entity_id;
|
resp.entity_id = it.entity_id;
|
||||||
resp.attribute = it.attribute.value();
|
resp.attribute = it.attribute.value();
|
||||||
|
resp.once = it.once;
|
||||||
if (this->send_subscribe_home_assistant_state_response(resp)) {
|
if (this->send_subscribe_home_assistant_state_response(resp)) {
|
||||||
state_subs_at_++;
|
state_subs_at_++;
|
||||||
}
|
}
|
||||||
|
@ -1025,6 +1026,15 @@ bool APIConnection::send_media_player_info(media_player::MediaPlayer *media_play
|
||||||
auto traits = media_player->get_traits();
|
auto traits = media_player->get_traits();
|
||||||
msg.supports_pause = traits.get_supports_pause();
|
msg.supports_pause = traits.get_supports_pause();
|
||||||
|
|
||||||
|
for (auto &supported_format : traits.get_supported_formats()) {
|
||||||
|
MediaPlayerSupportedFormat media_format;
|
||||||
|
media_format.format = supported_format.format;
|
||||||
|
media_format.sample_rate = supported_format.sample_rate;
|
||||||
|
media_format.num_channels = supported_format.num_channels;
|
||||||
|
media_format.purpose = static_cast<enums::MediaPlayerFormatPurpose>(supported_format.purpose);
|
||||||
|
msg.supported_formats.push_back(media_format);
|
||||||
|
}
|
||||||
|
|
||||||
return this->send_list_entities_media_player_response(msg);
|
return this->send_list_entities_media_player_response(msg);
|
||||||
}
|
}
|
||||||
void APIConnection::media_player_command(const MediaPlayerCommandRequest &msg) {
|
void APIConnection::media_player_command(const MediaPlayerCommandRequest &msg) {
|
||||||
|
|
|
@ -387,6 +387,18 @@ template<> const char *proto_enum_to_string<enums::MediaPlayerCommand>(enums::Me
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
#ifdef HAS_PROTO_MESSAGE_DUMP
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
|
template<> const char *proto_enum_to_string<enums::MediaPlayerFormatPurpose>(enums::MediaPlayerFormatPurpose value) {
|
||||||
|
switch (value) {
|
||||||
|
case enums::MEDIA_PLAYER_FORMAT_PURPOSE_DEFAULT:
|
||||||
|
return "MEDIA_PLAYER_FORMAT_PURPOSE_DEFAULT";
|
||||||
|
case enums::MEDIA_PLAYER_FORMAT_PURPOSE_ANNOUNCEMENT:
|
||||||
|
return "MEDIA_PLAYER_FORMAT_PURPOSE_ANNOUNCEMENT";
|
||||||
|
default:
|
||||||
|
return "UNKNOWN";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
template<>
|
template<>
|
||||||
const char *proto_enum_to_string<enums::BluetoothDeviceRequestType>(enums::BluetoothDeviceRequestType value) {
|
const char *proto_enum_to_string<enums::BluetoothDeviceRequestType>(enums::BluetoothDeviceRequestType value) {
|
||||||
switch (value) {
|
switch (value) {
|
||||||
|
@ -5123,6 +5135,64 @@ void ButtonCommandRequest::dump_to(std::string &out) const {
|
||||||
out.append("}");
|
out.append("}");
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
bool MediaPlayerSupportedFormat::decode_varint(uint32_t field_id, ProtoVarInt value) {
|
||||||
|
switch (field_id) {
|
||||||
|
case 2: {
|
||||||
|
this->sample_rate = value.as_uint32();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
case 3: {
|
||||||
|
this->num_channels = value.as_uint32();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
case 4: {
|
||||||
|
this->purpose = value.as_enum<enums::MediaPlayerFormatPurpose>();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
bool MediaPlayerSupportedFormat::decode_length(uint32_t field_id, ProtoLengthDelimited value) {
|
||||||
|
switch (field_id) {
|
||||||
|
case 1: {
|
||||||
|
this->format = value.as_string();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void MediaPlayerSupportedFormat::encode(ProtoWriteBuffer buffer) const {
|
||||||
|
buffer.encode_string(1, this->format);
|
||||||
|
buffer.encode_uint32(2, this->sample_rate);
|
||||||
|
buffer.encode_uint32(3, this->num_channels);
|
||||||
|
buffer.encode_enum<enums::MediaPlayerFormatPurpose>(4, this->purpose);
|
||||||
|
}
|
||||||
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
|
void MediaPlayerSupportedFormat::dump_to(std::string &out) const {
|
||||||
|
__attribute__((unused)) char buffer[64];
|
||||||
|
out.append("MediaPlayerSupportedFormat {\n");
|
||||||
|
out.append(" format: ");
|
||||||
|
out.append("'").append(this->format).append("'");
|
||||||
|
out.append("\n");
|
||||||
|
|
||||||
|
out.append(" sample_rate: ");
|
||||||
|
sprintf(buffer, "%" PRIu32, this->sample_rate);
|
||||||
|
out.append(buffer);
|
||||||
|
out.append("\n");
|
||||||
|
|
||||||
|
out.append(" num_channels: ");
|
||||||
|
sprintf(buffer, "%" PRIu32, this->num_channels);
|
||||||
|
out.append(buffer);
|
||||||
|
out.append("\n");
|
||||||
|
|
||||||
|
out.append(" purpose: ");
|
||||||
|
out.append(proto_enum_to_string<enums::MediaPlayerFormatPurpose>(this->purpose));
|
||||||
|
out.append("\n");
|
||||||
|
out.append("}");
|
||||||
|
}
|
||||||
|
#endif
|
||||||
bool ListEntitiesMediaPlayerResponse::decode_varint(uint32_t field_id, ProtoVarInt value) {
|
bool ListEntitiesMediaPlayerResponse::decode_varint(uint32_t field_id, ProtoVarInt value) {
|
||||||
switch (field_id) {
|
switch (field_id) {
|
||||||
case 6: {
|
case 6: {
|
||||||
|
@ -5159,6 +5229,10 @@ bool ListEntitiesMediaPlayerResponse::decode_length(uint32_t field_id, ProtoLeng
|
||||||
this->icon = value.as_string();
|
this->icon = value.as_string();
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
case 9: {
|
||||||
|
this->supported_formats.push_back(value.as_message<MediaPlayerSupportedFormat>());
|
||||||
|
return true;
|
||||||
|
}
|
||||||
default:
|
default:
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
@ -5182,6 +5256,9 @@ void ListEntitiesMediaPlayerResponse::encode(ProtoWriteBuffer buffer) const {
|
||||||
buffer.encode_bool(6, this->disabled_by_default);
|
buffer.encode_bool(6, this->disabled_by_default);
|
||||||
buffer.encode_enum<enums::EntityCategory>(7, this->entity_category);
|
buffer.encode_enum<enums::EntityCategory>(7, this->entity_category);
|
||||||
buffer.encode_bool(8, this->supports_pause);
|
buffer.encode_bool(8, this->supports_pause);
|
||||||
|
for (auto &it : this->supported_formats) {
|
||||||
|
buffer.encode_message<MediaPlayerSupportedFormat>(9, it, true);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
#ifdef HAS_PROTO_MESSAGE_DUMP
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
void ListEntitiesMediaPlayerResponse::dump_to(std::string &out) const {
|
void ListEntitiesMediaPlayerResponse::dump_to(std::string &out) const {
|
||||||
|
@ -5219,6 +5296,12 @@ void ListEntitiesMediaPlayerResponse::dump_to(std::string &out) const {
|
||||||
out.append(" supports_pause: ");
|
out.append(" supports_pause: ");
|
||||||
out.append(YESNO(this->supports_pause));
|
out.append(YESNO(this->supports_pause));
|
||||||
out.append("\n");
|
out.append("\n");
|
||||||
|
|
||||||
|
for (const auto &it : this->supported_formats) {
|
||||||
|
out.append(" supported_formats: ");
|
||||||
|
it.dump_to(out);
|
||||||
|
out.append("\n");
|
||||||
|
}
|
||||||
out.append("}");
|
out.append("}");
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -156,6 +156,10 @@ enum MediaPlayerCommand : uint32_t {
|
||||||
MEDIA_PLAYER_COMMAND_MUTE = 3,
|
MEDIA_PLAYER_COMMAND_MUTE = 3,
|
||||||
MEDIA_PLAYER_COMMAND_UNMUTE = 4,
|
MEDIA_PLAYER_COMMAND_UNMUTE = 4,
|
||||||
};
|
};
|
||||||
|
enum MediaPlayerFormatPurpose : uint32_t {
|
||||||
|
MEDIA_PLAYER_FORMAT_PURPOSE_DEFAULT = 0,
|
||||||
|
MEDIA_PLAYER_FORMAT_PURPOSE_ANNOUNCEMENT = 1,
|
||||||
|
};
|
||||||
enum BluetoothDeviceRequestType : uint32_t {
|
enum BluetoothDeviceRequestType : uint32_t {
|
||||||
BLUETOOTH_DEVICE_REQUEST_TYPE_CONNECT = 0,
|
BLUETOOTH_DEVICE_REQUEST_TYPE_CONNECT = 0,
|
||||||
BLUETOOTH_DEVICE_REQUEST_TYPE_DISCONNECT = 1,
|
BLUETOOTH_DEVICE_REQUEST_TYPE_DISCONNECT = 1,
|
||||||
|
@ -1267,6 +1271,21 @@ class ButtonCommandRequest : public ProtoMessage {
|
||||||
protected:
|
protected:
|
||||||
bool decode_32bit(uint32_t field_id, Proto32Bit value) override;
|
bool decode_32bit(uint32_t field_id, Proto32Bit value) override;
|
||||||
};
|
};
|
||||||
|
class MediaPlayerSupportedFormat : public ProtoMessage {
|
||||||
|
public:
|
||||||
|
std::string format{};
|
||||||
|
uint32_t sample_rate{0};
|
||||||
|
uint32_t num_channels{0};
|
||||||
|
enums::MediaPlayerFormatPurpose purpose{};
|
||||||
|
void encode(ProtoWriteBuffer buffer) const override;
|
||||||
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
|
void dump_to(std::string &out) const override;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
protected:
|
||||||
|
bool decode_length(uint32_t field_id, ProtoLengthDelimited value) override;
|
||||||
|
bool decode_varint(uint32_t field_id, ProtoVarInt value) override;
|
||||||
|
};
|
||||||
class ListEntitiesMediaPlayerResponse : public ProtoMessage {
|
class ListEntitiesMediaPlayerResponse : public ProtoMessage {
|
||||||
public:
|
public:
|
||||||
std::string object_id{};
|
std::string object_id{};
|
||||||
|
@ -1277,6 +1296,7 @@ class ListEntitiesMediaPlayerResponse : public ProtoMessage {
|
||||||
bool disabled_by_default{false};
|
bool disabled_by_default{false};
|
||||||
enums::EntityCategory entity_category{};
|
enums::EntityCategory entity_category{};
|
||||||
bool supports_pause{false};
|
bool supports_pause{false};
|
||||||
|
std::vector<MediaPlayerSupportedFormat> supported_formats{};
|
||||||
void encode(ProtoWriteBuffer buffer) const override;
|
void encode(ProtoWriteBuffer buffer) const override;
|
||||||
#ifdef HAS_PROTO_MESSAGE_DUMP
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
void dump_to(std::string &out) const override;
|
void dump_to(std::string &out) const override;
|
||||||
|
|
|
@ -311,6 +311,14 @@ bool APIServerConnectionBase::send_list_entities_button_response(const ListEntit
|
||||||
#ifdef USE_BUTTON
|
#ifdef USE_BUTTON
|
||||||
#endif
|
#endif
|
||||||
#ifdef USE_MEDIA_PLAYER
|
#ifdef USE_MEDIA_PLAYER
|
||||||
|
bool APIServerConnectionBase::send_media_player_supported_format(const MediaPlayerSupportedFormat &msg) {
|
||||||
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
|
ESP_LOGVV(TAG, "send_media_player_supported_format: %s", msg.dump().c_str());
|
||||||
|
#endif
|
||||||
|
return this->send_message_<MediaPlayerSupportedFormat>(msg, 119);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifdef USE_MEDIA_PLAYER
|
||||||
bool APIServerConnectionBase::send_list_entities_media_player_response(const ListEntitiesMediaPlayerResponse &msg) {
|
bool APIServerConnectionBase::send_list_entities_media_player_response(const ListEntitiesMediaPlayerResponse &msg) {
|
||||||
#ifdef HAS_PROTO_MESSAGE_DUMP
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
ESP_LOGVV(TAG, "send_list_entities_media_player_response: %s", msg.dump().c_str());
|
ESP_LOGVV(TAG, "send_list_entities_media_player_response: %s", msg.dump().c_str());
|
||||||
|
@ -1135,6 +1143,17 @@ bool APIServerConnectionBase::read_message(uint32_t msg_size, uint32_t msg_type,
|
||||||
ESP_LOGVV(TAG, "on_update_command_request: %s", msg.dump().c_str());
|
ESP_LOGVV(TAG, "on_update_command_request: %s", msg.dump().c_str());
|
||||||
#endif
|
#endif
|
||||||
this->on_update_command_request(msg);
|
this->on_update_command_request(msg);
|
||||||
|
#endif
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case 119: {
|
||||||
|
#ifdef USE_MEDIA_PLAYER
|
||||||
|
MediaPlayerSupportedFormat msg;
|
||||||
|
msg.decode(msg_data, msg_size);
|
||||||
|
#ifdef HAS_PROTO_MESSAGE_DUMP
|
||||||
|
ESP_LOGVV(TAG, "on_media_player_supported_format: %s", msg.dump().c_str());
|
||||||
|
#endif
|
||||||
|
this->on_media_player_supported_format(msg);
|
||||||
#endif
|
#endif
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
|
@ -145,6 +145,10 @@ class APIServerConnectionBase : public ProtoService {
|
||||||
#ifdef USE_BUTTON
|
#ifdef USE_BUTTON
|
||||||
virtual void on_button_command_request(const ButtonCommandRequest &value){};
|
virtual void on_button_command_request(const ButtonCommandRequest &value){};
|
||||||
#endif
|
#endif
|
||||||
|
#ifdef USE_MEDIA_PLAYER
|
||||||
|
bool send_media_player_supported_format(const MediaPlayerSupportedFormat &msg);
|
||||||
|
virtual void on_media_player_supported_format(const MediaPlayerSupportedFormat &value){};
|
||||||
|
#endif
|
||||||
#ifdef USE_MEDIA_PLAYER
|
#ifdef USE_MEDIA_PLAYER
|
||||||
bool send_list_entities_media_player_response(const ListEntitiesMediaPlayerResponse &msg);
|
bool send_list_entities_media_player_response(const ListEntitiesMediaPlayerResponse &msg);
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -1,34 +1,34 @@
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
|
||||||
from esphome.components import sensor, spi
|
from esphome.components import sensor, spi
|
||||||
|
import esphome.config_validation as cv
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
CONF_ID,
|
|
||||||
CONF_REACTIVE_POWER,
|
|
||||||
CONF_VOLTAGE,
|
|
||||||
CONF_CURRENT,
|
CONF_CURRENT,
|
||||||
|
CONF_FORWARD_ACTIVE_ENERGY,
|
||||||
|
CONF_FREQUENCY,
|
||||||
|
CONF_ID,
|
||||||
|
CONF_LINE_FREQUENCY,
|
||||||
CONF_POWER,
|
CONF_POWER,
|
||||||
CONF_POWER_FACTOR,
|
CONF_POWER_FACTOR,
|
||||||
CONF_FREQUENCY,
|
CONF_REACTIVE_POWER,
|
||||||
CONF_FORWARD_ACTIVE_ENERGY,
|
|
||||||
CONF_REVERSE_ACTIVE_ENERGY,
|
CONF_REVERSE_ACTIVE_ENERGY,
|
||||||
|
CONF_VOLTAGE,
|
||||||
DEVICE_CLASS_CURRENT,
|
DEVICE_CLASS_CURRENT,
|
||||||
DEVICE_CLASS_ENERGY,
|
DEVICE_CLASS_ENERGY,
|
||||||
DEVICE_CLASS_POWER,
|
DEVICE_CLASS_POWER,
|
||||||
DEVICE_CLASS_POWER_FACTOR,
|
DEVICE_CLASS_POWER_FACTOR,
|
||||||
DEVICE_CLASS_VOLTAGE,
|
DEVICE_CLASS_VOLTAGE,
|
||||||
ICON_LIGHTBULB,
|
|
||||||
ICON_CURRENT_AC,
|
ICON_CURRENT_AC,
|
||||||
|
ICON_LIGHTBULB,
|
||||||
STATE_CLASS_MEASUREMENT,
|
STATE_CLASS_MEASUREMENT,
|
||||||
STATE_CLASS_TOTAL_INCREASING,
|
STATE_CLASS_TOTAL_INCREASING,
|
||||||
|
UNIT_AMPERE,
|
||||||
UNIT_HERTZ,
|
UNIT_HERTZ,
|
||||||
UNIT_VOLT,
|
UNIT_VOLT,
|
||||||
UNIT_AMPERE,
|
|
||||||
UNIT_WATT,
|
|
||||||
UNIT_VOLT_AMPS_REACTIVE,
|
UNIT_VOLT_AMPS_REACTIVE,
|
||||||
|
UNIT_WATT,
|
||||||
UNIT_WATT_HOURS,
|
UNIT_WATT_HOURS,
|
||||||
)
|
)
|
||||||
|
|
||||||
CONF_LINE_FREQUENCY = "line_frequency"
|
|
||||||
CONF_METER_CONSTANT = "meter_constant"
|
CONF_METER_CONSTANT = "meter_constant"
|
||||||
CONF_PL_CONST = "pl_const"
|
CONF_PL_CONST = "pl_const"
|
||||||
CONF_GAIN_PGA = "gain_pga"
|
CONF_GAIN_PGA = "gain_pga"
|
||||||
|
|
|
@ -7,6 +7,7 @@ from esphome.const import (
|
||||||
CONF_FORWARD_ACTIVE_ENERGY,
|
CONF_FORWARD_ACTIVE_ENERGY,
|
||||||
CONF_FREQUENCY,
|
CONF_FREQUENCY,
|
||||||
CONF_ID,
|
CONF_ID,
|
||||||
|
CONF_LINE_FREQUENCY,
|
||||||
CONF_PHASE_A,
|
CONF_PHASE_A,
|
||||||
CONF_PHASE_ANGLE,
|
CONF_PHASE_ANGLE,
|
||||||
CONF_PHASE_B,
|
CONF_PHASE_B,
|
||||||
|
@ -39,7 +40,6 @@ from esphome.const import (
|
||||||
|
|
||||||
from . import atm90e32_ns
|
from . import atm90e32_ns
|
||||||
|
|
||||||
CONF_LINE_FREQUENCY = "line_frequency"
|
|
||||||
CONF_CHIP_TEMPERATURE = "chip_temperature"
|
CONF_CHIP_TEMPERATURE = "chip_temperature"
|
||||||
CONF_GAIN_PGA = "gain_pga"
|
CONF_GAIN_PGA = "gain_pga"
|
||||||
CONF_CURRENT_PHASES = "current_phases"
|
CONF_CURRENT_PHASES = "current_phases"
|
||||||
|
|
1
esphome/components/bl0906/__init__.py
Normal file
1
esphome/components/bl0906/__init__.py
Normal file
|
@ -0,0 +1 @@
|
||||||
|
CODEOWNERS = ["@athom-tech", "@tarontop", "@jesserockz"]
|
238
esphome/components/bl0906/bl0906.cpp
Normal file
238
esphome/components/bl0906/bl0906.cpp
Normal file
|
@ -0,0 +1,238 @@
|
||||||
|
#include "bl0906.h"
|
||||||
|
#include "constants.h"
|
||||||
|
|
||||||
|
#include "esphome/core/log.h"
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace bl0906 {
|
||||||
|
|
||||||
|
static const char *const TAG = "bl0906";
|
||||||
|
|
||||||
|
constexpr uint32_t to_uint32_t(ube24_t input) { return input.h << 16 | input.m << 8 | input.l; }
|
||||||
|
|
||||||
|
constexpr int32_t to_int32_t(sbe24_t input) { return input.h << 16 | input.m << 8 | input.l; }
|
||||||
|
|
||||||
|
// The SUM byte is (Addr+Data_L+Data_M+Data_H)&0xFF negated;
|
||||||
|
constexpr uint8_t bl0906_checksum(const uint8_t address, const DataPacket *data) {
|
||||||
|
return (address + data->l + data->m + data->h) ^ 0xFF;
|
||||||
|
}
|
||||||
|
|
||||||
|
void BL0906::loop() {
|
||||||
|
if (this->current_channel_ == UINT8_MAX) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
while (this->available())
|
||||||
|
this->flush();
|
||||||
|
|
||||||
|
if (this->current_channel_ == 0) {
|
||||||
|
// Temperature
|
||||||
|
this->read_data_(BL0906_TEMPERATURE, BL0906_TREF, this->temperature_sensor_);
|
||||||
|
} else if (this->current_channel_ == 1) {
|
||||||
|
this->read_data_(BL0906_I_1_RMS, BL0906_IREF, this->current_1_sensor_);
|
||||||
|
this->read_data_(BL0906_WATT_1, BL0906_PREF, this->power_1_sensor_);
|
||||||
|
this->read_data_(BL0906_CF_1_CNT, BL0906_EREF, this->energy_1_sensor_);
|
||||||
|
} else if (this->current_channel_ == 2) {
|
||||||
|
this->read_data_(BL0906_I_2_RMS, BL0906_IREF, this->current_2_sensor_);
|
||||||
|
this->read_data_(BL0906_WATT_2, BL0906_PREF, this->power_2_sensor_);
|
||||||
|
this->read_data_(BL0906_CF_2_CNT, BL0906_EREF, this->energy_2_sensor_);
|
||||||
|
} else if (this->current_channel_ == 3) {
|
||||||
|
this->read_data_(BL0906_I_3_RMS, BL0906_IREF, this->current_3_sensor_);
|
||||||
|
this->read_data_(BL0906_WATT_3, BL0906_PREF, this->power_3_sensor_);
|
||||||
|
this->read_data_(BL0906_CF_3_CNT, BL0906_EREF, this->energy_3_sensor_);
|
||||||
|
} else if (this->current_channel_ == 4) {
|
||||||
|
this->read_data_(BL0906_I_4_RMS, BL0906_IREF, this->current_4_sensor_);
|
||||||
|
this->read_data_(BL0906_WATT_4, BL0906_PREF, this->power_4_sensor_);
|
||||||
|
this->read_data_(BL0906_CF_4_CNT, BL0906_EREF, this->energy_4_sensor_);
|
||||||
|
} else if (this->current_channel_ == 5) {
|
||||||
|
this->read_data_(BL0906_I_5_RMS, BL0906_IREF, this->current_5_sensor_);
|
||||||
|
this->read_data_(BL0906_WATT_5, BL0906_PREF, this->power_5_sensor_);
|
||||||
|
this->read_data_(BL0906_CF_5_CNT, BL0906_EREF, this->energy_5_sensor_);
|
||||||
|
} else if (this->current_channel_ == 6) {
|
||||||
|
this->read_data_(BL0906_I_6_RMS, BL0906_IREF, this->current_6_sensor_);
|
||||||
|
this->read_data_(BL0906_WATT_6, BL0906_PREF, this->power_6_sensor_);
|
||||||
|
this->read_data_(BL0906_CF_6_CNT, BL0906_EREF, this->energy_6_sensor_);
|
||||||
|
} else if (this->current_channel_ == UINT8_MAX - 2) {
|
||||||
|
// Frequency
|
||||||
|
this->read_data_(BL0906_FREQUENCY, BL0906_FREF, frequency_sensor_);
|
||||||
|
// Voltage
|
||||||
|
this->read_data_(BL0906_V_RMS, BL0906_UREF, voltage_sensor_);
|
||||||
|
} else if (this->current_channel_ == UINT8_MAX - 1) {
|
||||||
|
// Total power
|
||||||
|
this->read_data_(BL0906_WATT_SUM, BL0906_WATT, this->total_power_sensor_);
|
||||||
|
// Total Energy
|
||||||
|
this->read_data_(BL0906_CF_SUM_CNT, BL0906_CF, this->total_energy_sensor_);
|
||||||
|
} else {
|
||||||
|
this->current_channel_ = UINT8_MAX - 2; // Go to frequency and voltage
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
this->current_channel_++;
|
||||||
|
this->handle_actions_();
|
||||||
|
}
|
||||||
|
|
||||||
|
void BL0906::setup() {
|
||||||
|
while (this->available())
|
||||||
|
this->flush();
|
||||||
|
this->write_array(USR_WRPROT_WITABLE, sizeof(USR_WRPROT_WITABLE));
|
||||||
|
// Calibration (1: register address; 2: value before calibration; 3: value after calibration)
|
||||||
|
this->bias_correction_(BL0906_RMSOS_1, 0.01600, 0); // Calibration current_1
|
||||||
|
this->bias_correction_(BL0906_RMSOS_2, 0.01500, 0);
|
||||||
|
this->bias_correction_(BL0906_RMSOS_3, 0.01400, 0);
|
||||||
|
this->bias_correction_(BL0906_RMSOS_4, 0.01300, 0);
|
||||||
|
this->bias_correction_(BL0906_RMSOS_5, 0.01200, 0);
|
||||||
|
this->bias_correction_(BL0906_RMSOS_6, 0.01200, 0); // Calibration current_6
|
||||||
|
|
||||||
|
this->write_array(USR_WRPROT_ONLYREAD, sizeof(USR_WRPROT_ONLYREAD));
|
||||||
|
}
|
||||||
|
|
||||||
|
void BL0906::update() { this->current_channel_ = 0; }
|
||||||
|
|
||||||
|
size_t BL0906::enqueue_action_(ActionCallbackFuncPtr function) {
|
||||||
|
this->action_queue_.push_back(function);
|
||||||
|
return this->action_queue_.size();
|
||||||
|
}
|
||||||
|
|
||||||
|
void BL0906::handle_actions_() {
|
||||||
|
if (this->action_queue_.empty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
ActionCallbackFuncPtr ptr_func = nullptr;
|
||||||
|
for (int i = 0; i < this->action_queue_.size(); i++) {
|
||||||
|
ptr_func = this->action_queue_[i];
|
||||||
|
if (ptr_func) {
|
||||||
|
ESP_LOGI(TAG, "HandleActionCallback[%d]...", i);
|
||||||
|
(this->*ptr_func)();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
while (this->available()) {
|
||||||
|
this->read();
|
||||||
|
}
|
||||||
|
|
||||||
|
this->action_queue_.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reset energy
|
||||||
|
void BL0906::reset_energy_() {
|
||||||
|
this->write_array(BL0906_INIT[0], 6);
|
||||||
|
delay(1);
|
||||||
|
this->flush();
|
||||||
|
|
||||||
|
ESP_LOGW(TAG, "RMSOS:%02X%02X%02X%02X%02X%02X", BL0906_INIT[0][0], BL0906_INIT[0][1], BL0906_INIT[0][2],
|
||||||
|
BL0906_INIT[0][3], BL0906_INIT[0][4], BL0906_INIT[0][5]);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read data
|
||||||
|
void BL0906::read_data_(const uint8_t address, const float reference, sensor::Sensor *sensor) {
|
||||||
|
if (sensor == nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
DataPacket buffer;
|
||||||
|
ube24_t data_u24;
|
||||||
|
sbe24_t data_s24;
|
||||||
|
float value = 0;
|
||||||
|
|
||||||
|
bool signed_result = reference == BL0906_TREF || reference == BL0906_WATT || reference == BL0906_PREF;
|
||||||
|
|
||||||
|
this->write_byte(BL0906_READ_COMMAND);
|
||||||
|
this->write_byte(address);
|
||||||
|
if (this->read_array((uint8_t *) &buffer, sizeof(buffer) - 1)) {
|
||||||
|
if (bl0906_checksum(address, &buffer) == buffer.checksum) {
|
||||||
|
if (signed_result) {
|
||||||
|
data_s24.l = buffer.l;
|
||||||
|
data_s24.m = buffer.m;
|
||||||
|
data_s24.h = buffer.h;
|
||||||
|
} else {
|
||||||
|
data_u24.l = buffer.l;
|
||||||
|
data_u24.m = buffer.m;
|
||||||
|
data_u24.h = buffer.h;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
ESP_LOGW(TAG, "Junk on wire. Throwing away partial message");
|
||||||
|
while (read() >= 0)
|
||||||
|
;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Power
|
||||||
|
if (reference == BL0906_PREF) {
|
||||||
|
value = (float) to_int32_t(data_s24) * reference;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Total power
|
||||||
|
if (reference == BL0906_WATT) {
|
||||||
|
value = (float) to_int32_t(data_s24) * reference;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Voltage, current, power, total power
|
||||||
|
if (reference == BL0906_UREF || reference == BL0906_IREF || reference == BL0906_EREF || reference == BL0906_CF) {
|
||||||
|
value = (float) to_uint32_t(data_u24) * reference;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Frequency
|
||||||
|
if (reference == BL0906_FREF) {
|
||||||
|
value = reference / (float) to_uint32_t(data_u24);
|
||||||
|
}
|
||||||
|
// Chip temperature
|
||||||
|
if (reference == BL0906_TREF) {
|
||||||
|
value = (float) to_int32_t(data_s24);
|
||||||
|
value = (value - 64) * 12.5 / 59 - 40;
|
||||||
|
}
|
||||||
|
sensor->publish_state(value);
|
||||||
|
}
|
||||||
|
|
||||||
|
// RMS offset correction
|
||||||
|
void BL0906::bias_correction_(uint8_t address, float measurements, float correction) {
|
||||||
|
DataPacket data;
|
||||||
|
float ki = 12875 * 1 * (5.1 + 5.1) * 1000 / 2000 / 1.097; // Current coefficient
|
||||||
|
float i_rms0 = measurements * ki;
|
||||||
|
float i_rms = correction * ki;
|
||||||
|
int32_t value = (i_rms * i_rms - i_rms0 * i_rms0) / 256;
|
||||||
|
data.l = value << 24 >> 24;
|
||||||
|
data.m = value << 16 >> 24;
|
||||||
|
if (value < 0) {
|
||||||
|
data.h = (value << 8 >> 24) | 0b10000000;
|
||||||
|
}
|
||||||
|
data.address = bl0906_checksum(address, &data);
|
||||||
|
ESP_LOGV(TAG, "RMSOS:%02X%02X%02X%02X%02X%02X", BL0906_WRITE_COMMAND, address, data.l, data.m, data.h, data.address);
|
||||||
|
this->write_byte(BL0906_WRITE_COMMAND);
|
||||||
|
this->write_byte(address);
|
||||||
|
this->write_byte(data.l);
|
||||||
|
this->write_byte(data.m);
|
||||||
|
this->write_byte(data.h);
|
||||||
|
this->write_byte(data.address);
|
||||||
|
}
|
||||||
|
|
||||||
|
void BL0906::dump_config() {
|
||||||
|
ESP_LOGCONFIG(TAG, "BL0906:");
|
||||||
|
LOG_SENSOR(" ", "Voltage", this->voltage_sensor_);
|
||||||
|
|
||||||
|
LOG_SENSOR(" ", "Current1", this->current_1_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Current2", this->current_2_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Current3", this->current_3_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Current4", this->current_4_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Current5", this->current_5_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Current6", this->current_6_sensor_);
|
||||||
|
|
||||||
|
LOG_SENSOR(" ", "Power1", this->power_1_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Power2", this->power_2_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Power3", this->power_3_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Power4", this->power_4_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Power5", this->power_5_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Power6", this->power_6_sensor_);
|
||||||
|
|
||||||
|
LOG_SENSOR(" ", "Energy1", this->energy_1_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Energy2", this->energy_2_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Energy3", this->energy_3_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Energy4", this->energy_4_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Energy5", this->energy_5_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Energy6", this->energy_6_sensor_);
|
||||||
|
|
||||||
|
LOG_SENSOR(" ", "Total Power", this->total_power_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Total Energy", this->total_energy_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Frequency", this->frequency_sensor_);
|
||||||
|
LOG_SENSOR(" ", "Temperature", this->temperature_sensor_);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace bl0906
|
||||||
|
} // namespace esphome
|
96
esphome/components/bl0906/bl0906.h
Normal file
96
esphome/components/bl0906/bl0906.h
Normal file
|
@ -0,0 +1,96 @@
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "esphome/components/sensor/sensor.h"
|
||||||
|
#include "esphome/components/uart/uart.h"
|
||||||
|
#include "esphome/core/automation.h"
|
||||||
|
#include "esphome/core/component.h"
|
||||||
|
#include "esphome/core/datatypes.h"
|
||||||
|
|
||||||
|
// https://www.belling.com.cn/media/file_object/bel_product/BL0906/datasheet/BL0906_V1.02_cn.pdf
|
||||||
|
// https://www.belling.com.cn/media/file_object/bel_product/BL0906/guide/BL0906%20APP%20Note_V1.02.pdf
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace bl0906 {
|
||||||
|
|
||||||
|
struct DataPacket { // NOLINT(altera-struct-pack-align)
|
||||||
|
uint8_t l{0};
|
||||||
|
uint8_t m{0};
|
||||||
|
uint8_t h{0};
|
||||||
|
uint8_t checksum; // checksum
|
||||||
|
uint8_t address;
|
||||||
|
} __attribute__((packed));
|
||||||
|
|
||||||
|
struct ube24_t { // NOLINT(readability-identifier-naming,altera-struct-pack-align)
|
||||||
|
uint8_t l{0};
|
||||||
|
uint8_t m{0};
|
||||||
|
uint8_t h{0};
|
||||||
|
} __attribute__((packed));
|
||||||
|
|
||||||
|
struct sbe24_t { // NOLINT(readability-identifier-naming,altera-struct-pack-align)
|
||||||
|
uint8_t l{0};
|
||||||
|
uint8_t m{0};
|
||||||
|
int8_t h{0};
|
||||||
|
} __attribute__((packed));
|
||||||
|
|
||||||
|
template<typename... Ts> class ResetEnergyAction;
|
||||||
|
|
||||||
|
class BL0906;
|
||||||
|
|
||||||
|
using ActionCallbackFuncPtr = void (BL0906::*)();
|
||||||
|
|
||||||
|
class BL0906 : public PollingComponent, public uart::UARTDevice {
|
||||||
|
SUB_SENSOR(voltage)
|
||||||
|
SUB_SENSOR(current_1)
|
||||||
|
SUB_SENSOR(current_2)
|
||||||
|
SUB_SENSOR(current_3)
|
||||||
|
SUB_SENSOR(current_4)
|
||||||
|
SUB_SENSOR(current_5)
|
||||||
|
SUB_SENSOR(current_6)
|
||||||
|
SUB_SENSOR(power_1)
|
||||||
|
SUB_SENSOR(power_2)
|
||||||
|
SUB_SENSOR(power_3)
|
||||||
|
SUB_SENSOR(power_4)
|
||||||
|
SUB_SENSOR(power_5)
|
||||||
|
SUB_SENSOR(power_6)
|
||||||
|
SUB_SENSOR(total_power)
|
||||||
|
SUB_SENSOR(energy_1)
|
||||||
|
SUB_SENSOR(energy_2)
|
||||||
|
SUB_SENSOR(energy_3)
|
||||||
|
SUB_SENSOR(energy_4)
|
||||||
|
SUB_SENSOR(energy_5)
|
||||||
|
SUB_SENSOR(energy_6)
|
||||||
|
SUB_SENSOR(total_energy)
|
||||||
|
SUB_SENSOR(frequency)
|
||||||
|
SUB_SENSOR(temperature)
|
||||||
|
|
||||||
|
public:
|
||||||
|
void loop() override;
|
||||||
|
|
||||||
|
void update() override;
|
||||||
|
void setup() override;
|
||||||
|
void dump_config() override;
|
||||||
|
|
||||||
|
protected:
|
||||||
|
template<typename... Ts> friend class ResetEnergyAction;
|
||||||
|
|
||||||
|
void reset_energy_();
|
||||||
|
|
||||||
|
void read_data_(uint8_t address, float reference, sensor::Sensor *sensor);
|
||||||
|
|
||||||
|
void bias_correction_(uint8_t address, float measurements, float correction);
|
||||||
|
|
||||||
|
uint8_t current_channel_{0};
|
||||||
|
size_t enqueue_action_(ActionCallbackFuncPtr function);
|
||||||
|
void handle_actions_();
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::vector<ActionCallbackFuncPtr> action_queue_{};
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename... Ts> class ResetEnergyAction : public Action<Ts...>, public Parented<BL0906> {
|
||||||
|
public:
|
||||||
|
void play(Ts... x) override { this->parent_->enqueue_action_(&BL0906::reset_energy_); }
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace bl0906
|
||||||
|
} // namespace esphome
|
4
esphome/components/bl0906/const.py
Normal file
4
esphome/components/bl0906/const.py
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
# const.py
|
||||||
|
ICON_ENERGY = "mdi:lightning-bolt"
|
||||||
|
ICON_FREQUENCY = "mdi:cosine-wave"
|
||||||
|
ICON_VOLTAGE = "mdi:sine-wave"
|
122
esphome/components/bl0906/constants.h
Normal file
122
esphome/components/bl0906/constants.h
Normal file
|
@ -0,0 +1,122 @@
|
||||||
|
#pragma once
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace bl0906 {
|
||||||
|
|
||||||
|
// Total power conversion
|
||||||
|
static const float BL0906_WATT = 16 * 1.097 * 1.097 * (20000 + 20000 + 20000 + 20000 + 20000) /
|
||||||
|
(40.41259 * ((5.1 + 5.1) * 1000 / 2000) * 1 * 100 * 1 * 1000);
|
||||||
|
// Total Energy conversion
|
||||||
|
static const float BL0906_CF = 16 * 4194304 * 0.032768 * 16 /
|
||||||
|
(3600000 * 16 *
|
||||||
|
(40.4125 * ((5.1 + 5.1) * 1000 / 2000) * 1 * 100 * 1 * 1000 /
|
||||||
|
(1.097 * 1.097 * (20000 + 20000 + 20000 + 20000 + 20000))));
|
||||||
|
// Frequency conversion
|
||||||
|
static const float BL0906_FREF = 10000000;
|
||||||
|
// Temperature conversion
|
||||||
|
static const float BL0906_TREF = 12.5 / 59 - 40;
|
||||||
|
// Current conversion
|
||||||
|
static const float BL0906_IREF = 1.097 / (12875 * 1 * (5.1 + 5.1) * 1000 / 2000);
|
||||||
|
// Voltage conversion
|
||||||
|
static const float BL0906_UREF = 1.097 * (20000 + 20000 + 20000 + 20000 + 20000) / (13162 * 1 * 100 * 1000);
|
||||||
|
// Power conversion
|
||||||
|
static const float BL0906_PREF = 1.097 * 1.097 * (20000 + 20000 + 20000 + 20000 + 20000) /
|
||||||
|
(40.41259 * ((5.1 + 5.1) * 1000 / 2000) * 1 * 100 * 1 * 1000);
|
||||||
|
// Energy conversion
|
||||||
|
static const float BL0906_EREF = 4194304 * 0.032768 * 16 /
|
||||||
|
(3600000 * 16 *
|
||||||
|
(40.4125 * ((5.1 + 5.1) * 1000 / 2000) * 1 * 100 * 1 * 1000 /
|
||||||
|
(1.097 * 1.097 * (20000 + 20000 + 20000 + 20000 + 20000))));
|
||||||
|
// Current coefficient
|
||||||
|
static const float BL0906_KI = 12875 * 1 * (5.1 + 5.1) * 1000 / 2000 / 1.097;
|
||||||
|
// Power coefficient
|
||||||
|
static const float BL0906_KP = 40.4125 * ((5.1 + 5.1) * 1000 / 2000) * 1 * 100 * 1 * 1000 / 1.097 / 1.097 /
|
||||||
|
(20000 + 20000 + 20000 + 20000 + 20000);
|
||||||
|
|
||||||
|
static const uint8_t USR_WRPROT_WITABLE[6] = {0xCA, 0x9E, 0x55, 0x55, 0x00, 0xB7};
|
||||||
|
static const uint8_t USR_WRPROT_ONLYREAD[6] = {0xCA, 0x9E, 0x00, 0x00, 0x00, 0x61};
|
||||||
|
|
||||||
|
static const uint8_t BL0906_READ_COMMAND = 0x35;
|
||||||
|
static const uint8_t BL0906_WRITE_COMMAND = 0xCA;
|
||||||
|
|
||||||
|
// Register address
|
||||||
|
// Voltage
|
||||||
|
static const uint8_t BL0906_V_RMS = 0x16;
|
||||||
|
|
||||||
|
// Total power
|
||||||
|
static const uint8_t BL0906_WATT_SUM = 0X2C;
|
||||||
|
|
||||||
|
// Current1~6
|
||||||
|
static const uint8_t BL0906_I_1_RMS = 0x0D; // current_1
|
||||||
|
static const uint8_t BL0906_I_2_RMS = 0x0E;
|
||||||
|
static const uint8_t BL0906_I_3_RMS = 0x0F;
|
||||||
|
static const uint8_t BL0906_I_4_RMS = 0x10;
|
||||||
|
static const uint8_t BL0906_I_5_RMS = 0x13;
|
||||||
|
static const uint8_t BL0906_I_6_RMS = 0x14; // current_6
|
||||||
|
|
||||||
|
// Power1~6
|
||||||
|
static const uint8_t BL0906_WATT_1 = 0X23; // power_1
|
||||||
|
static const uint8_t BL0906_WATT_2 = 0X24;
|
||||||
|
static const uint8_t BL0906_WATT_3 = 0X25;
|
||||||
|
static const uint8_t BL0906_WATT_4 = 0X26;
|
||||||
|
static const uint8_t BL0906_WATT_5 = 0X29;
|
||||||
|
static const uint8_t BL0906_WATT_6 = 0X2A; // power_6
|
||||||
|
|
||||||
|
// Active pulse count, unsigned
|
||||||
|
static const uint8_t BL0906_CF_1_CNT = 0X30; // Channel_1
|
||||||
|
static const uint8_t BL0906_CF_2_CNT = 0X31;
|
||||||
|
static const uint8_t BL0906_CF_3_CNT = 0X32;
|
||||||
|
static const uint8_t BL0906_CF_4_CNT = 0X33;
|
||||||
|
static const uint8_t BL0906_CF_5_CNT = 0X36;
|
||||||
|
static const uint8_t BL0906_CF_6_CNT = 0X37; // Channel_6
|
||||||
|
|
||||||
|
// Total active pulse count, unsigned
|
||||||
|
static const uint8_t BL0906_CF_SUM_CNT = 0X39;
|
||||||
|
|
||||||
|
// Voltage frequency cycle
|
||||||
|
static const uint8_t BL0906_FREQUENCY = 0X4E;
|
||||||
|
|
||||||
|
// Internal temperature
|
||||||
|
static const uint8_t BL0906_TEMPERATURE = 0X5E;
|
||||||
|
|
||||||
|
// Calibration register
|
||||||
|
// RMS gain adjustment register
|
||||||
|
static const uint8_t BL0906_RMSGN_1 = 0x6D; // Channel_1
|
||||||
|
static const uint8_t BL0906_RMSGN_2 = 0x6E;
|
||||||
|
static const uint8_t BL0906_RMSGN_3 = 0x6F;
|
||||||
|
static const uint8_t BL0906_RMSGN_4 = 0x70;
|
||||||
|
static const uint8_t BL0906_RMSGN_5 = 0x73;
|
||||||
|
static const uint8_t BL0906_RMSGN_6 = 0x74; // Channel_6
|
||||||
|
|
||||||
|
// RMS offset correction register
|
||||||
|
static const uint8_t BL0906_RMSOS_1 = 0x78; // Channel_1
|
||||||
|
static const uint8_t BL0906_RMSOS_2 = 0x79;
|
||||||
|
static const uint8_t BL0906_RMSOS_3 = 0x7A;
|
||||||
|
static const uint8_t BL0906_RMSOS_4 = 0x7B;
|
||||||
|
static const uint8_t BL0906_RMSOS_5 = 0x7E;
|
||||||
|
static const uint8_t BL0906_RMSOS_6 = 0x7F; // Channel_6
|
||||||
|
|
||||||
|
// Active power gain adjustment register
|
||||||
|
static const uint8_t BL0906_WATTGN_1 = 0xB7; // Channel_1
|
||||||
|
static const uint8_t BL0906_WATTGN_2 = 0xB8;
|
||||||
|
static const uint8_t BL0906_WATTGN_3 = 0xB9;
|
||||||
|
static const uint8_t BL0906_WATTGN_4 = 0xBA;
|
||||||
|
static const uint8_t BL0906_WATTGN_5 = 0xBD;
|
||||||
|
static const uint8_t BL0906_WATTGN_6 = 0xBE; // Channel_6
|
||||||
|
|
||||||
|
// User write protection setting register,
|
||||||
|
// You must first write 0x5555 to the write protection setting register before writing to other registers.
|
||||||
|
static const uint8_t BL0906_USR_WRPROT = 0x9E;
|
||||||
|
|
||||||
|
// Reset Register
|
||||||
|
static const uint8_t BL0906_SOFT_RESET = 0x9F;
|
||||||
|
|
||||||
|
const uint8_t BL0906_INIT[2][6] = {
|
||||||
|
// Reset to default
|
||||||
|
{BL0906_WRITE_COMMAND, BL0906_SOFT_RESET, 0x5A, 0x5A, 0x5A, 0x52},
|
||||||
|
// Enable User Operation Write
|
||||||
|
{BL0906_WRITE_COMMAND, BL0906_USR_WRPROT, 0x55, 0x55, 0x00, 0xB7}};
|
||||||
|
|
||||||
|
} // namespace bl0906
|
||||||
|
} // namespace esphome
|
184
esphome/components/bl0906/sensor.py
Normal file
184
esphome/components/bl0906/sensor.py
Normal file
|
@ -0,0 +1,184 @@
|
||||||
|
from esphome import automation
|
||||||
|
from esphome.automation import maybe_simple_id
|
||||||
|
import esphome.codegen as cg
|
||||||
|
from esphome.components import sensor, uart
|
||||||
|
import esphome.config_validation as cv
|
||||||
|
from esphome.const import (
|
||||||
|
CONF_CHANNEL,
|
||||||
|
CONF_CURRENT,
|
||||||
|
CONF_ENERGY,
|
||||||
|
CONF_FREQUENCY,
|
||||||
|
CONF_ID,
|
||||||
|
CONF_NAME,
|
||||||
|
CONF_POWER,
|
||||||
|
CONF_TEMPERATURE,
|
||||||
|
CONF_TOTAL_POWER,
|
||||||
|
CONF_VOLTAGE,
|
||||||
|
DEVICE_CLASS_CURRENT,
|
||||||
|
DEVICE_CLASS_ENERGY,
|
||||||
|
DEVICE_CLASS_FREQUENCY,
|
||||||
|
DEVICE_CLASS_POWER,
|
||||||
|
DEVICE_CLASS_TEMPERATURE,
|
||||||
|
DEVICE_CLASS_VOLTAGE,
|
||||||
|
ICON_CURRENT_AC,
|
||||||
|
ICON_POWER,
|
||||||
|
ICON_THERMOMETER,
|
||||||
|
STATE_CLASS_MEASUREMENT,
|
||||||
|
STATE_CLASS_TOTAL_INCREASING,
|
||||||
|
UNIT_AMPERE,
|
||||||
|
UNIT_CELSIUS,
|
||||||
|
UNIT_HERTZ,
|
||||||
|
UNIT_KILOWATT_HOURS,
|
||||||
|
UNIT_VOLT,
|
||||||
|
UNIT_WATT,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Import ICONS not included in esphome's const.py, from the local components const.py
|
||||||
|
from .const import ICON_ENERGY, ICON_FREQUENCY, ICON_VOLTAGE
|
||||||
|
|
||||||
|
DEPENDENCIES = ["uart"]
|
||||||
|
AUTO_LOAD = ["bl0906"]
|
||||||
|
CONF_TOTAL_ENERGY = "total_energy"
|
||||||
|
|
||||||
|
bl0906_ns = cg.esphome_ns.namespace("bl0906")
|
||||||
|
BL0906 = bl0906_ns.class_("BL0906", cg.PollingComponent, uart.UARTDevice)
|
||||||
|
ResetEnergyAction = bl0906_ns.class_("ResetEnergyAction", automation.Action)
|
||||||
|
|
||||||
|
CONFIG_SCHEMA = (
|
||||||
|
cv.Schema(
|
||||||
|
{
|
||||||
|
cv.GenerateID(): cv.declare_id(BL0906),
|
||||||
|
cv.Optional(CONF_FREQUENCY): sensor.sensor_schema(
|
||||||
|
icon=ICON_FREQUENCY,
|
||||||
|
accuracy_decimals=0,
|
||||||
|
device_class=DEVICE_CLASS_FREQUENCY,
|
||||||
|
unit_of_measurement=UNIT_HERTZ,
|
||||||
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_TEMPERATURE): sensor.sensor_schema(
|
||||||
|
icon=ICON_THERMOMETER,
|
||||||
|
accuracy_decimals=0,
|
||||||
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
|
unit_of_measurement=UNIT_CELSIUS,
|
||||||
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_VOLTAGE): sensor.sensor_schema(
|
||||||
|
icon=ICON_VOLTAGE,
|
||||||
|
accuracy_decimals=1,
|
||||||
|
device_class=DEVICE_CLASS_VOLTAGE,
|
||||||
|
unit_of_measurement=UNIT_VOLT,
|
||||||
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_TOTAL_POWER): sensor.sensor_schema(
|
||||||
|
icon=ICON_POWER,
|
||||||
|
accuracy_decimals=3,
|
||||||
|
device_class=DEVICE_CLASS_POWER,
|
||||||
|
unit_of_measurement=UNIT_WATT,
|
||||||
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_TOTAL_ENERGY): sensor.sensor_schema(
|
||||||
|
icon=ICON_ENERGY,
|
||||||
|
accuracy_decimals=3,
|
||||||
|
device_class=DEVICE_CLASS_ENERGY,
|
||||||
|
state_class=STATE_CLASS_TOTAL_INCREASING,
|
||||||
|
unit_of_measurement=UNIT_KILOWATT_HOURS,
|
||||||
|
),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.extend(
|
||||||
|
cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Optional(f"{CONF_CHANNEL}_{i + 1}"): cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Optional(CONF_CURRENT): cv.maybe_simple_value(
|
||||||
|
sensor.sensor_schema(
|
||||||
|
icon=ICON_CURRENT_AC,
|
||||||
|
accuracy_decimals=3,
|
||||||
|
device_class=DEVICE_CLASS_CURRENT,
|
||||||
|
unit_of_measurement=UNIT_AMPERE,
|
||||||
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
|
),
|
||||||
|
key=CONF_NAME,
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_POWER): cv.maybe_simple_value(
|
||||||
|
sensor.sensor_schema(
|
||||||
|
icon=ICON_POWER,
|
||||||
|
accuracy_decimals=0,
|
||||||
|
device_class=DEVICE_CLASS_POWER,
|
||||||
|
unit_of_measurement=UNIT_WATT,
|
||||||
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
|
),
|
||||||
|
key=CONF_NAME,
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_ENERGY): cv.maybe_simple_value(
|
||||||
|
sensor.sensor_schema(
|
||||||
|
icon=ICON_ENERGY,
|
||||||
|
accuracy_decimals=3,
|
||||||
|
device_class=DEVICE_CLASS_ENERGY,
|
||||||
|
unit_of_measurement=UNIT_KILOWATT_HOURS,
|
||||||
|
state_class=STATE_CLASS_TOTAL_INCREASING,
|
||||||
|
),
|
||||||
|
key=CONF_NAME,
|
||||||
|
),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
for i in range(6)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
)
|
||||||
|
.extend(uart.UART_DEVICE_SCHEMA)
|
||||||
|
.extend(cv.polling_component_schema("60s"))
|
||||||
|
)
|
||||||
|
|
||||||
|
FINAL_VALIDATE_SCHEMA = uart.final_validate_device_schema(
|
||||||
|
"bl0906", baud_rate=19200, require_tx=True, require_rx=True
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@automation.register_action(
|
||||||
|
"bl0906.reset_energy",
|
||||||
|
ResetEnergyAction,
|
||||||
|
maybe_simple_id(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_ID): cv.use_id(BL0906),
|
||||||
|
}
|
||||||
|
),
|
||||||
|
)
|
||||||
|
async def reset_energy_to_code(config, action_id, template_arg, args):
|
||||||
|
paren = await cg.get_variable(config[CONF_ID])
|
||||||
|
return cg.new_Pvariable(action_id, template_arg, paren)
|
||||||
|
|
||||||
|
|
||||||
|
async def to_code(config):
|
||||||
|
var = cg.new_Pvariable(config[CONF_ID])
|
||||||
|
await cg.register_component(var, config)
|
||||||
|
await uart.register_uart_device(var, config)
|
||||||
|
if frequency_config := config.get(CONF_FREQUENCY):
|
||||||
|
sens = await sensor.new_sensor(frequency_config)
|
||||||
|
cg.add(var.set_frequency_sensor(sens))
|
||||||
|
if temperature_config := config.get(CONF_TEMPERATURE):
|
||||||
|
sens = await sensor.new_sensor(temperature_config)
|
||||||
|
cg.add(var.set_temperature_sensor(sens))
|
||||||
|
if voltage_config := config.get(CONF_VOLTAGE):
|
||||||
|
sens = await sensor.new_sensor(voltage_config)
|
||||||
|
cg.add(var.set_voltage_sensor(sens))
|
||||||
|
|
||||||
|
for i in range(6):
|
||||||
|
if channel_config := config.get(f"{CONF_CHANNEL}_{i + 1}"):
|
||||||
|
if current_config := channel_config.get(CONF_CURRENT):
|
||||||
|
sens = await sensor.new_sensor(current_config)
|
||||||
|
cg.add(getattr(var, f"set_current_{i + 1}_sensor")(sens))
|
||||||
|
if power_config := channel_config.get(CONF_POWER):
|
||||||
|
sens = await sensor.new_sensor(power_config)
|
||||||
|
cg.add(getattr(var, f"set_power_{i + 1}_sensor")(sens))
|
||||||
|
if energy_config := channel_config.get(CONF_ENERGY):
|
||||||
|
sens = await sensor.new_sensor(energy_config)
|
||||||
|
cg.add(getattr(var, f"set_energy_{i + 1}_sensor")(sens))
|
||||||
|
|
||||||
|
if total_power_config := config.get(CONF_TOTAL_POWER):
|
||||||
|
sens = await sensor.new_sensor(total_power_config)
|
||||||
|
cg.add(var.set_total_power_sensor(sens))
|
||||||
|
|
||||||
|
if total_energy_config := config.get(CONF_TOTAL_ENERGY):
|
||||||
|
sens = await sensor.new_sensor(total_energy_config)
|
||||||
|
cg.add(var.set_total_energy_sensor(sens))
|
|
@ -2,6 +2,8 @@
|
||||||
#include "esphome/core/log.h"
|
#include "esphome/core/log.h"
|
||||||
#include <cinttypes>
|
#include <cinttypes>
|
||||||
|
|
||||||
|
// Datasheet: https://www.belling.com.cn/media/file_object/bel_product/BL0942/datasheet/BL0942_V1.06_en.pdf
|
||||||
|
|
||||||
namespace esphome {
|
namespace esphome {
|
||||||
namespace bl0942 {
|
namespace bl0942 {
|
||||||
|
|
||||||
|
@ -12,33 +14,41 @@ static const uint8_t BL0942_FULL_PACKET = 0xAA;
|
||||||
static const uint8_t BL0942_PACKET_HEADER = 0x55;
|
static const uint8_t BL0942_PACKET_HEADER = 0x55;
|
||||||
|
|
||||||
static const uint8_t BL0942_WRITE_COMMAND = 0xA8;
|
static const uint8_t BL0942_WRITE_COMMAND = 0xA8;
|
||||||
static const uint8_t BL0942_REG_I_FAST_RMS_CTRL = 0x10;
|
|
||||||
static const uint8_t BL0942_REG_MODE = 0x18;
|
static const uint8_t BL0942_REG_I_RMSOS = 0x12;
|
||||||
static const uint8_t BL0942_REG_SOFT_RESET = 0x19;
|
static const uint8_t BL0942_REG_WA_CREEP = 0x14;
|
||||||
static const uint8_t BL0942_REG_USR_WRPROT = 0x1A;
|
static const uint8_t BL0942_REG_I_FAST_RMS_TH = 0x15;
|
||||||
|
static const uint8_t BL0942_REG_I_FAST_RMS_CYC = 0x16;
|
||||||
|
static const uint8_t BL0942_REG_FREQ_CYC = 0x17;
|
||||||
|
static const uint8_t BL0942_REG_OT_FUNX = 0x18;
|
||||||
|
static const uint8_t BL0942_REG_MODE = 0x19;
|
||||||
|
static const uint8_t BL0942_REG_SOFT_RESET = 0x1C;
|
||||||
|
static const uint8_t BL0942_REG_USR_WRPROT = 0x1D;
|
||||||
static const uint8_t BL0942_REG_TPS_CTRL = 0x1B;
|
static const uint8_t BL0942_REG_TPS_CTRL = 0x1B;
|
||||||
|
|
||||||
// TODO: Confirm insialisation works as intended
|
static const uint32_t BL0942_REG_MODE_RESV = 0x03;
|
||||||
const uint8_t BL0942_INIT[5][6] = {
|
static const uint32_t BL0942_REG_MODE_CF_EN = 0x04;
|
||||||
// Reset to default
|
static const uint32_t BL0942_REG_MODE_RMS_UPDATE_SEL = 0x08;
|
||||||
{BL0942_WRITE_COMMAND, BL0942_REG_SOFT_RESET, 0x5A, 0x5A, 0x5A, 0x38},
|
static const uint32_t BL0942_REG_MODE_FAST_RMS_SEL = 0x10;
|
||||||
// Enable User Operation Write
|
static const uint32_t BL0942_REG_MODE_AC_FREQ_SEL = 0x20;
|
||||||
{BL0942_WRITE_COMMAND, BL0942_REG_USR_WRPROT, 0x55, 0x00, 0x00, 0xF0},
|
static const uint32_t BL0942_REG_MODE_CF_CNT_CLR_SEL = 0x40;
|
||||||
// 0x0100 = CF_UNABLE energy pulse, AC_FREQ_SEL 50Hz, RMS_UPDATE_SEL 800mS
|
static const uint32_t BL0942_REG_MODE_CF_CNT_ADD_SEL = 0x80;
|
||||||
{BL0942_WRITE_COMMAND, BL0942_REG_MODE, 0x00, 0x10, 0x00, 0x37},
|
static const uint32_t BL0942_REG_MODE_UART_RATE_19200 = 0x200;
|
||||||
// 0x47FF = Over-current and leakage alarm on, Automatic temperature measurement, Interval 100mS
|
static const uint32_t BL0942_REG_MODE_UART_RATE_38400 = 0x300;
|
||||||
{BL0942_WRITE_COMMAND, BL0942_REG_TPS_CTRL, 0xFF, 0x47, 0x00, 0xFE},
|
static const uint32_t BL0942_REG_MODE_DEFAULT =
|
||||||
// 0x181C = Half cycle, Fast RMS threshold 6172
|
BL0942_REG_MODE_RESV | BL0942_REG_MODE_CF_EN | BL0942_REG_MODE_CF_CNT_ADD_SEL;
|
||||||
{BL0942_WRITE_COMMAND, BL0942_REG_I_FAST_RMS_CTRL, 0x1C, 0x18, 0x00, 0x1B}};
|
|
||||||
|
static const uint32_t BL0942_REG_SOFT_RESET_MAGIC = 0x5a5a5a;
|
||||||
|
static const uint32_t BL0942_REG_USR_WRPROT_MAGIC = 0x55;
|
||||||
|
|
||||||
void BL0942::loop() {
|
void BL0942::loop() {
|
||||||
DataPacket buffer;
|
DataPacket buffer;
|
||||||
if (!this->available()) {
|
if (!this->available()) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (read_array((uint8_t *) &buffer, sizeof(buffer))) {
|
if (this->read_array((uint8_t *) &buffer, sizeof(buffer))) {
|
||||||
if (validate_checksum(&buffer)) {
|
if (this->validate_checksum_(&buffer)) {
|
||||||
received_package_(&buffer);
|
this->received_package_(&buffer);
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
ESP_LOGW(TAG, "Junk on wire. Throwing away partial message");
|
ESP_LOGW(TAG, "Junk on wire. Throwing away partial message");
|
||||||
|
@ -47,8 +57,8 @@ void BL0942::loop() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool BL0942::validate_checksum(DataPacket *data) {
|
bool BL0942::validate_checksum_(DataPacket *data) {
|
||||||
uint8_t checksum = BL0942_READ_COMMAND;
|
uint8_t checksum = BL0942_READ_COMMAND | this->address_;
|
||||||
// Whole package but checksum
|
// Whole package but checksum
|
||||||
uint8_t *raw = (uint8_t *) data;
|
uint8_t *raw = (uint8_t *) data;
|
||||||
for (uint32_t i = 0; i < sizeof(*data) - 1; i++) {
|
for (uint32_t i = 0; i < sizeof(*data) - 1; i++) {
|
||||||
|
@ -61,17 +71,58 @@ bool BL0942::validate_checksum(DataPacket *data) {
|
||||||
return checksum == data->checksum;
|
return checksum == data->checksum;
|
||||||
}
|
}
|
||||||
|
|
||||||
void BL0942::update() {
|
void BL0942::write_reg_(uint8_t reg, uint32_t val) {
|
||||||
|
uint8_t pkt[6];
|
||||||
|
|
||||||
this->flush();
|
this->flush();
|
||||||
this->write_byte(BL0942_READ_COMMAND);
|
pkt[0] = BL0942_WRITE_COMMAND | this->address_;
|
||||||
|
pkt[1] = reg;
|
||||||
|
pkt[2] = (val & 0xff);
|
||||||
|
pkt[3] = (val >> 8) & 0xff;
|
||||||
|
pkt[4] = (val >> 16) & 0xff;
|
||||||
|
pkt[5] = (pkt[0] + pkt[1] + pkt[2] + pkt[3] + pkt[4]) ^ 0xff;
|
||||||
|
this->write_array(pkt, 6);
|
||||||
|
delay(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
int BL0942::read_reg_(uint8_t reg) {
|
||||||
|
union {
|
||||||
|
uint8_t b[4];
|
||||||
|
uint32_le_t le32;
|
||||||
|
} resp;
|
||||||
|
|
||||||
|
this->write_byte(BL0942_READ_COMMAND | this->address_);
|
||||||
|
this->write_byte(reg);
|
||||||
|
this->flush();
|
||||||
|
if (this->read_array(resp.b, 4) &&
|
||||||
|
resp.b[3] ==
|
||||||
|
(uint8_t) ((BL0942_READ_COMMAND + this->address_ + reg + resp.b[0] + resp.b[1] + resp.b[2]) ^ 0xff)) {
|
||||||
|
resp.b[3] = 0;
|
||||||
|
return resp.le32;
|
||||||
|
}
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void BL0942::update() {
|
||||||
|
this->write_byte(BL0942_READ_COMMAND | this->address_);
|
||||||
this->write_byte(BL0942_FULL_PACKET);
|
this->write_byte(BL0942_FULL_PACKET);
|
||||||
}
|
}
|
||||||
|
|
||||||
void BL0942::setup() {
|
void BL0942::setup() {
|
||||||
for (auto *i : BL0942_INIT) {
|
this->write_reg_(BL0942_REG_USR_WRPROT, BL0942_REG_USR_WRPROT_MAGIC);
|
||||||
this->write_array(i, 6);
|
this->write_reg_(BL0942_REG_SOFT_RESET, BL0942_REG_SOFT_RESET_MAGIC);
|
||||||
delay(1);
|
|
||||||
}
|
uint32_t mode = BL0942_REG_MODE_DEFAULT;
|
||||||
|
mode |= BL0942_REG_MODE_RMS_UPDATE_SEL; /* 800ms refresh time */
|
||||||
|
if (this->line_freq_ == LINE_FREQUENCY_60HZ)
|
||||||
|
mode |= BL0942_REG_MODE_AC_FREQ_SEL;
|
||||||
|
this->write_reg_(BL0942_REG_MODE, mode);
|
||||||
|
|
||||||
|
this->write_reg_(BL0942_REG_USR_WRPROT, 0);
|
||||||
|
|
||||||
|
if (this->read_reg_(BL0942_REG_MODE) != mode)
|
||||||
|
this->status_set_warning("BL0942 setup failed!");
|
||||||
|
|
||||||
this->flush();
|
this->flush();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -104,13 +155,15 @@ void BL0942::received_package_(DataPacket *data) {
|
||||||
if (frequency_sensor_ != nullptr) {
|
if (frequency_sensor_ != nullptr) {
|
||||||
frequency_sensor_->publish_state(frequency);
|
frequency_sensor_->publish_state(frequency);
|
||||||
}
|
}
|
||||||
|
this->status_clear_warning();
|
||||||
ESP_LOGV(TAG, "BL0942: U %fV, I %fA, P %fW, Cnt %" PRId32 ", ∫P %fkWh, frequency %fHz, status 0x%08X", v_rms, i_rms,
|
ESP_LOGV(TAG, "BL0942: U %fV, I %fA, P %fW, Cnt %" PRId32 ", ∫P %fkWh, frequency %fHz, status 0x%08X", v_rms, i_rms,
|
||||||
watt, cf_cnt, total_energy_consumption, frequency, data->status);
|
watt, cf_cnt, total_energy_consumption, frequency, data->status);
|
||||||
}
|
}
|
||||||
|
|
||||||
void BL0942::dump_config() { // NOLINT(readability-function-cognitive-complexity)
|
void BL0942::dump_config() { // NOLINT(readability-function-cognitive-complexity)
|
||||||
ESP_LOGCONFIG(TAG, "BL0942:");
|
ESP_LOGCONFIG(TAG, "BL0942:");
|
||||||
|
ESP_LOGCONFIG(TAG, " Address: %d", this->address_);
|
||||||
|
ESP_LOGCONFIG(TAG, " Nominal line frequency: %d Hz", this->line_freq_);
|
||||||
LOG_SENSOR("", "Voltage", this->voltage_sensor_);
|
LOG_SENSOR("", "Voltage", this->voltage_sensor_);
|
||||||
LOG_SENSOR("", "Current", this->current_sensor_);
|
LOG_SENSOR("", "Current", this->current_sensor_);
|
||||||
LOG_SENSOR("", "Power", this->power_sensor_);
|
LOG_SENSOR("", "Power", this->power_sensor_);
|
||||||
|
|
|
@ -28,6 +28,11 @@ struct DataPacket {
|
||||||
uint8_t checksum;
|
uint8_t checksum;
|
||||||
} __attribute__((packed));
|
} __attribute__((packed));
|
||||||
|
|
||||||
|
enum LineFrequency : uint8_t {
|
||||||
|
LINE_FREQUENCY_50HZ = 50,
|
||||||
|
LINE_FREQUENCY_60HZ = 60,
|
||||||
|
};
|
||||||
|
|
||||||
class BL0942 : public PollingComponent, public uart::UARTDevice {
|
class BL0942 : public PollingComponent, public uart::UARTDevice {
|
||||||
public:
|
public:
|
||||||
void set_voltage_sensor(sensor::Sensor *voltage_sensor) { voltage_sensor_ = voltage_sensor; }
|
void set_voltage_sensor(sensor::Sensor *voltage_sensor) { voltage_sensor_ = voltage_sensor; }
|
||||||
|
@ -35,9 +40,10 @@ class BL0942 : public PollingComponent, public uart::UARTDevice {
|
||||||
void set_power_sensor(sensor::Sensor *power_sensor) { power_sensor_ = power_sensor; }
|
void set_power_sensor(sensor::Sensor *power_sensor) { power_sensor_ = power_sensor; }
|
||||||
void set_energy_sensor(sensor::Sensor *energy_sensor) { energy_sensor_ = energy_sensor; }
|
void set_energy_sensor(sensor::Sensor *energy_sensor) { energy_sensor_ = energy_sensor; }
|
||||||
void set_frequency_sensor(sensor::Sensor *frequency_sensor) { frequency_sensor_ = frequency_sensor; }
|
void set_frequency_sensor(sensor::Sensor *frequency_sensor) { frequency_sensor_ = frequency_sensor; }
|
||||||
|
void set_line_freq(LineFrequency freq) { this->line_freq_ = freq; }
|
||||||
|
void set_address(uint8_t address) { this->address_ = address; }
|
||||||
|
|
||||||
void loop() override;
|
void loop() override;
|
||||||
|
|
||||||
void update() override;
|
void update() override;
|
||||||
void setup() override;
|
void setup() override;
|
||||||
void dump_config() override;
|
void dump_config() override;
|
||||||
|
@ -59,9 +65,12 @@ class BL0942 : public PollingComponent, public uart::UARTDevice {
|
||||||
float current_reference_ = BL0942_IREF;
|
float current_reference_ = BL0942_IREF;
|
||||||
// Divide by this to turn into kWh
|
// Divide by this to turn into kWh
|
||||||
float energy_reference_ = BL0942_EREF;
|
float energy_reference_ = BL0942_EREF;
|
||||||
|
uint8_t address_ = 0;
|
||||||
|
LineFrequency line_freq_ = LINE_FREQUENCY_50HZ;
|
||||||
|
|
||||||
static bool validate_checksum(DataPacket *data);
|
bool validate_checksum_(DataPacket *data);
|
||||||
|
int read_reg_(uint8_t reg);
|
||||||
|
void write_reg_(uint8_t reg, uint32_t val);
|
||||||
void received_package_(DataPacket *data);
|
void received_package_(DataPacket *data);
|
||||||
};
|
};
|
||||||
} // namespace bl0942
|
} // namespace bl0942
|
||||||
|
|
|
@ -1,25 +1,27 @@
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
|
||||||
from esphome.components import sensor, uart
|
from esphome.components import sensor, uart
|
||||||
|
import esphome.config_validation as cv
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
|
CONF_ADDRESS,
|
||||||
CONF_CURRENT,
|
CONF_CURRENT,
|
||||||
CONF_ENERGY,
|
CONF_ENERGY,
|
||||||
|
CONF_FREQUENCY,
|
||||||
CONF_ID,
|
CONF_ID,
|
||||||
|
CONF_LINE_FREQUENCY,
|
||||||
CONF_POWER,
|
CONF_POWER,
|
||||||
CONF_VOLTAGE,
|
CONF_VOLTAGE,
|
||||||
CONF_FREQUENCY,
|
|
||||||
DEVICE_CLASS_CURRENT,
|
DEVICE_CLASS_CURRENT,
|
||||||
DEVICE_CLASS_ENERGY,
|
DEVICE_CLASS_ENERGY,
|
||||||
|
DEVICE_CLASS_FREQUENCY,
|
||||||
DEVICE_CLASS_POWER,
|
DEVICE_CLASS_POWER,
|
||||||
DEVICE_CLASS_VOLTAGE,
|
DEVICE_CLASS_VOLTAGE,
|
||||||
DEVICE_CLASS_FREQUENCY,
|
|
||||||
STATE_CLASS_MEASUREMENT,
|
STATE_CLASS_MEASUREMENT,
|
||||||
|
STATE_CLASS_TOTAL_INCREASING,
|
||||||
UNIT_AMPERE,
|
UNIT_AMPERE,
|
||||||
|
UNIT_HERTZ,
|
||||||
UNIT_KILOWATT_HOURS,
|
UNIT_KILOWATT_HOURS,
|
||||||
UNIT_VOLT,
|
UNIT_VOLT,
|
||||||
UNIT_WATT,
|
UNIT_WATT,
|
||||||
UNIT_HERTZ,
|
|
||||||
STATE_CLASS_TOTAL_INCREASING,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
DEPENDENCIES = ["uart"]
|
DEPENDENCIES = ["uart"]
|
||||||
|
@ -27,6 +29,12 @@ DEPENDENCIES = ["uart"]
|
||||||
bl0942_ns = cg.esphome_ns.namespace("bl0942")
|
bl0942_ns = cg.esphome_ns.namespace("bl0942")
|
||||||
BL0942 = bl0942_ns.class_("BL0942", cg.PollingComponent, uart.UARTDevice)
|
BL0942 = bl0942_ns.class_("BL0942", cg.PollingComponent, uart.UARTDevice)
|
||||||
|
|
||||||
|
LineFrequency = bl0942_ns.enum("LineFrequency")
|
||||||
|
LINE_FREQS = {
|
||||||
|
50: LineFrequency.LINE_FREQUENCY_50HZ,
|
||||||
|
60: LineFrequency.LINE_FREQUENCY_60HZ,
|
||||||
|
}
|
||||||
|
|
||||||
CONFIG_SCHEMA = (
|
CONFIG_SCHEMA = (
|
||||||
cv.Schema(
|
cv.Schema(
|
||||||
{
|
{
|
||||||
|
@ -61,6 +69,14 @@ CONFIG_SCHEMA = (
|
||||||
device_class=DEVICE_CLASS_FREQUENCY,
|
device_class=DEVICE_CLASS_FREQUENCY,
|
||||||
state_class=STATE_CLASS_MEASUREMENT,
|
state_class=STATE_CLASS_MEASUREMENT,
|
||||||
),
|
),
|
||||||
|
cv.Optional(CONF_LINE_FREQUENCY, default="50HZ"): cv.All(
|
||||||
|
cv.frequency,
|
||||||
|
cv.enum(
|
||||||
|
LINE_FREQS,
|
||||||
|
int=True,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_ADDRESS, default=0): cv.int_range(min=0, max=3),
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
.extend(cv.polling_component_schema("60s"))
|
.extend(cv.polling_component_schema("60s"))
|
||||||
|
@ -88,3 +104,5 @@ async def to_code(config):
|
||||||
if frequency_config := config.get(CONF_FREQUENCY):
|
if frequency_config := config.get(CONF_FREQUENCY):
|
||||||
sens = await sensor.new_sensor(frequency_config)
|
sens = await sensor.new_sensor(frequency_config)
|
||||||
cg.add(var.set_frequency_sensor(sens))
|
cg.add(var.set_frequency_sensor(sens))
|
||||||
|
cg.add(var.set_line_freq(config[CONF_LINE_FREQUENCY]))
|
||||||
|
cg.add(var.set_address(config[CONF_ADDRESS]))
|
||||||
|
|
|
@ -54,6 +54,9 @@ bool BluetoothProxy::parse_devices(esp_ble_gap_cb_param_t::ble_scan_result_evt_p
|
||||||
}
|
}
|
||||||
|
|
||||||
resp.advertisements.push_back(std::move(adv));
|
resp.advertisements.push_back(std::move(adv));
|
||||||
|
|
||||||
|
ESP_LOGV(TAG, "Proxying raw packet from %02X:%02X:%02X:%02X:%02X:%02X, length %d. RSSI: %d dB", result.bda[0],
|
||||||
|
result.bda[1], result.bda[2], result.bda[3], result.bda[4], result.bda[5], length, result.rssi);
|
||||||
}
|
}
|
||||||
ESP_LOGV(TAG, "Proxying %d packets", count);
|
ESP_LOGV(TAG, "Proxying %d packets", count);
|
||||||
this->api_connection_->send_bluetooth_le_raw_advertisements_response(resp);
|
this->api_connection_->send_bluetooth_le_raw_advertisements_response(resp);
|
||||||
|
@ -87,6 +90,8 @@ void BluetoothProxy::send_api_packet_(const esp32_ble_tracker::ESPBTDevice &devi
|
||||||
void BluetoothProxy::dump_config() {
|
void BluetoothProxy::dump_config() {
|
||||||
ESP_LOGCONFIG(TAG, "Bluetooth Proxy:");
|
ESP_LOGCONFIG(TAG, "Bluetooth Proxy:");
|
||||||
ESP_LOGCONFIG(TAG, " Active: %s", YESNO(this->active_));
|
ESP_LOGCONFIG(TAG, " Active: %s", YESNO(this->active_));
|
||||||
|
ESP_LOGCONFIG(TAG, " Connections: %d", this->connections_.size());
|
||||||
|
ESP_LOGCONFIG(TAG, " Raw advertisements: %s", YESNO(this->raw_advertisements_));
|
||||||
}
|
}
|
||||||
|
|
||||||
int BluetoothProxy::get_bluetooth_connections_free() {
|
int BluetoothProxy::get_bluetooth_connections_free() {
|
||||||
|
|
67
esphome/components/ch422g/__init__.py
Normal file
67
esphome/components/ch422g/__init__.py
Normal file
|
@ -0,0 +1,67 @@
|
||||||
|
from esphome import pins
|
||||||
|
import esphome.codegen as cg
|
||||||
|
from esphome.components import i2c
|
||||||
|
import esphome.config_validation as cv
|
||||||
|
from esphome.const import (
|
||||||
|
CONF_ID,
|
||||||
|
CONF_INPUT,
|
||||||
|
CONF_INVERTED,
|
||||||
|
CONF_MODE,
|
||||||
|
CONF_NUMBER,
|
||||||
|
CONF_OUTPUT,
|
||||||
|
CONF_RESTORE_VALUE,
|
||||||
|
)
|
||||||
|
|
||||||
|
CODEOWNERS = ["@jesterret"]
|
||||||
|
DEPENDENCIES = ["i2c"]
|
||||||
|
MULTI_CONF = True
|
||||||
|
ch422g_ns = cg.esphome_ns.namespace("ch422g")
|
||||||
|
|
||||||
|
CH422GComponent = ch422g_ns.class_("CH422GComponent", cg.Component, i2c.I2CDevice)
|
||||||
|
CH422GGPIOPin = ch422g_ns.class_(
|
||||||
|
"CH422GGPIOPin", cg.GPIOPin, cg.Parented.template(CH422GComponent)
|
||||||
|
)
|
||||||
|
|
||||||
|
CONF_CH422G = "ch422g"
|
||||||
|
CONFIG_SCHEMA = (
|
||||||
|
cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_ID): cv.declare_id(CH422GComponent),
|
||||||
|
cv.Optional(CONF_RESTORE_VALUE, default=False): cv.boolean,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.extend(cv.COMPONENT_SCHEMA)
|
||||||
|
.extend(i2c.i2c_device_schema(0x24))
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def to_code(config):
|
||||||
|
var = cg.new_Pvariable(config[CONF_ID])
|
||||||
|
cg.add(var.set_restore_value(config[CONF_RESTORE_VALUE]))
|
||||||
|
await cg.register_component(var, config)
|
||||||
|
await i2c.register_i2c_device(var, config)
|
||||||
|
|
||||||
|
|
||||||
|
CH422G_PIN_SCHEMA = pins.gpio_base_schema(
|
||||||
|
CH422GGPIOPin,
|
||||||
|
cv.int_range(min=0, max=7),
|
||||||
|
modes=[CONF_INPUT, CONF_OUTPUT],
|
||||||
|
).extend(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_CH422G): cv.use_id(CH422GComponent),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pins.PIN_SCHEMA_REGISTRY.register(CONF_CH422G, CH422G_PIN_SCHEMA)
|
||||||
|
async def ch422g_pin_to_code(config):
|
||||||
|
var = cg.new_Pvariable(config[CONF_ID])
|
||||||
|
parent = await cg.get_variable(config[CONF_CH422G])
|
||||||
|
|
||||||
|
cg.add(var.set_parent(parent))
|
||||||
|
|
||||||
|
num = config[CONF_NUMBER]
|
||||||
|
cg.add(var.set_pin(num))
|
||||||
|
cg.add(var.set_inverted(config[CONF_INVERTED]))
|
||||||
|
cg.add(var.set_flags(pins.gpio_flags_expr(config[CONF_MODE])))
|
||||||
|
return var
|
122
esphome/components/ch422g/ch422g.cpp
Normal file
122
esphome/components/ch422g/ch422g.cpp
Normal file
|
@ -0,0 +1,122 @@
|
||||||
|
#include "ch422g.h"
|
||||||
|
#include "esphome/core/log.h"
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace ch422g {
|
||||||
|
|
||||||
|
const uint8_t CH422G_REG_IN = 0x26;
|
||||||
|
const uint8_t CH422G_REG_OUT = 0x38;
|
||||||
|
const uint8_t OUT_REG_DEFAULT_VAL = 0xdf;
|
||||||
|
|
||||||
|
static const char *const TAG = "ch422g";
|
||||||
|
|
||||||
|
void CH422GComponent::setup() {
|
||||||
|
ESP_LOGCONFIG(TAG, "Setting up CH422G...");
|
||||||
|
// Test to see if device exists
|
||||||
|
if (!this->read_inputs_()) {
|
||||||
|
ESP_LOGE(TAG, "CH422G not detected at 0x%02X", this->address_);
|
||||||
|
this->mark_failed();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// restore defaults over whatever got saved on last boot
|
||||||
|
if (!this->restore_value_) {
|
||||||
|
this->write_output_(OUT_REG_DEFAULT_VAL);
|
||||||
|
}
|
||||||
|
|
||||||
|
ESP_LOGD(TAG, "Initialization complete. Warning: %d, Error: %d", this->status_has_warning(),
|
||||||
|
this->status_has_error());
|
||||||
|
}
|
||||||
|
|
||||||
|
void CH422GComponent::loop() {
|
||||||
|
// Clear all the previously read flags.
|
||||||
|
this->pin_read_cache_ = 0x00;
|
||||||
|
}
|
||||||
|
|
||||||
|
void CH422GComponent::dump_config() {
|
||||||
|
ESP_LOGCONFIG(TAG, "CH422G:");
|
||||||
|
LOG_I2C_DEVICE(this)
|
||||||
|
if (this->is_failed()) {
|
||||||
|
ESP_LOGE(TAG, "Communication with CH422G failed!");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ch422g doesn't have any flag support (needs docs?)
|
||||||
|
void CH422GComponent::pin_mode(uint8_t pin, gpio::Flags flags) {}
|
||||||
|
|
||||||
|
bool CH422GComponent::digital_read(uint8_t pin) {
|
||||||
|
if (this->pin_read_cache_ == 0 || this->pin_read_cache_ & (1 << pin)) {
|
||||||
|
// Read values on first access or in case it's being read again in the same loop
|
||||||
|
this->read_inputs_();
|
||||||
|
}
|
||||||
|
|
||||||
|
this->pin_read_cache_ |= (1 << pin);
|
||||||
|
return this->state_mask_ & (1 << pin);
|
||||||
|
}
|
||||||
|
|
||||||
|
void CH422GComponent::digital_write(uint8_t pin, bool value) {
|
||||||
|
if (value) {
|
||||||
|
this->write_output_(this->state_mask_ | (1 << pin));
|
||||||
|
} else {
|
||||||
|
this->write_output_(this->state_mask_ & ~(1 << pin));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool CH422GComponent::read_inputs_() {
|
||||||
|
if (this->is_failed()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t temp = 0;
|
||||||
|
if ((this->last_error_ = this->read(&temp, 1)) != esphome::i2c::ERROR_OK) {
|
||||||
|
this->status_set_warning(str_sprintf("read_inputs_(): I2C I/O error: %d", (int) this->last_error_).c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t output = 0;
|
||||||
|
if ((this->last_error_ = this->bus_->read(CH422G_REG_IN, &output, 1)) != esphome::i2c::ERROR_OK) {
|
||||||
|
this->status_set_warning(str_sprintf("read_inputs_(): I2C I/O error: %d", (int) this->last_error_).c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
this->state_mask_ = output;
|
||||||
|
this->status_clear_warning();
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool CH422GComponent::write_output_(uint8_t value) {
|
||||||
|
const uint8_t temp = 1;
|
||||||
|
if ((this->last_error_ = this->write(&temp, 1, false)) != esphome::i2c::ERROR_OK) {
|
||||||
|
this->status_set_warning(str_sprintf("write_output_(): I2C I/O error: %d", (int) this->last_error_).c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t write_mask = value;
|
||||||
|
if ((this->last_error_ = this->bus_->write(CH422G_REG_OUT, &write_mask, 1)) != esphome::i2c::ERROR_OK) {
|
||||||
|
this->status_set_warning(
|
||||||
|
str_sprintf("write_output_(): I2C I/O error: %d for write_mask: %d", (int) this->last_error_, (int) write_mask)
|
||||||
|
.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
this->state_mask_ = value;
|
||||||
|
this->status_clear_warning();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
float CH422GComponent::get_setup_priority() const { return setup_priority::IO; }
|
||||||
|
|
||||||
|
// Run our loop() method very early in the loop, so that we cache read values
|
||||||
|
// before other components call our digital_read() method.
|
||||||
|
float CH422GComponent::get_loop_priority() const { return 9.0f; } // Just after WIFI
|
||||||
|
|
||||||
|
void CH422GGPIOPin::setup() { pin_mode(flags_); }
|
||||||
|
void CH422GGPIOPin::pin_mode(gpio::Flags flags) { this->parent_->pin_mode(this->pin_, flags); }
|
||||||
|
bool CH422GGPIOPin::digital_read() { return this->parent_->digital_read(this->pin_) != this->inverted_; }
|
||||||
|
|
||||||
|
void CH422GGPIOPin::digital_write(bool value) { this->parent_->digital_write(this->pin_, value != this->inverted_); }
|
||||||
|
std::string CH422GGPIOPin::dump_summary() const { return str_sprintf("EXIO%u via CH422G", pin_); }
|
||||||
|
|
||||||
|
} // namespace ch422g
|
||||||
|
} // namespace esphome
|
70
esphome/components/ch422g/ch422g.h
Normal file
70
esphome/components/ch422g/ch422g.h
Normal file
|
@ -0,0 +1,70 @@
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "esphome/core/component.h"
|
||||||
|
#include "esphome/core/hal.h"
|
||||||
|
#include "esphome/components/i2c/i2c.h"
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace ch422g {
|
||||||
|
|
||||||
|
class CH422GComponent : public Component, public i2c::I2CDevice {
|
||||||
|
public:
|
||||||
|
CH422GComponent() = default;
|
||||||
|
|
||||||
|
/// Check i2c availability and setup masks
|
||||||
|
void setup() override;
|
||||||
|
/// Poll for input changes periodically
|
||||||
|
void loop() override;
|
||||||
|
/// Helper function to read the value of a pin.
|
||||||
|
bool digital_read(uint8_t pin);
|
||||||
|
/// Helper function to write the value of a pin.
|
||||||
|
void digital_write(uint8_t pin, bool value);
|
||||||
|
/// Helper function to set the pin mode of a pin.
|
||||||
|
void pin_mode(uint8_t pin, gpio::Flags flags);
|
||||||
|
|
||||||
|
float get_setup_priority() const override;
|
||||||
|
|
||||||
|
float get_loop_priority() const override;
|
||||||
|
|
||||||
|
void dump_config() override;
|
||||||
|
|
||||||
|
void set_restore_value(bool restore_value) { this->restore_value_ = restore_value; }
|
||||||
|
|
||||||
|
protected:
|
||||||
|
bool read_inputs_();
|
||||||
|
|
||||||
|
bool write_output_(uint8_t value);
|
||||||
|
|
||||||
|
/// The mask to write as output state - 1 means HIGH, 0 means LOW
|
||||||
|
uint8_t state_mask_{0x00};
|
||||||
|
/// Flags to check if read previously during this loop
|
||||||
|
uint8_t pin_read_cache_ = {0x00};
|
||||||
|
/// Storage for last I2C error seen
|
||||||
|
esphome::i2c::ErrorCode last_error_;
|
||||||
|
/// Whether we want to override stored values on expander
|
||||||
|
bool restore_value_{false};
|
||||||
|
};
|
||||||
|
|
||||||
|
/// Helper class to expose a CH422G pin as an internal input GPIO pin.
|
||||||
|
class CH422GGPIOPin : public GPIOPin {
|
||||||
|
public:
|
||||||
|
void setup() override;
|
||||||
|
void pin_mode(gpio::Flags flags) override;
|
||||||
|
bool digital_read() override;
|
||||||
|
void digital_write(bool value) override;
|
||||||
|
std::string dump_summary() const override;
|
||||||
|
|
||||||
|
void set_parent(CH422GComponent *parent) { parent_ = parent; }
|
||||||
|
void set_pin(uint8_t pin) { pin_ = pin; }
|
||||||
|
void set_inverted(bool inverted) { inverted_ = inverted; }
|
||||||
|
void set_flags(gpio::Flags flags) { flags_ = flags; }
|
||||||
|
|
||||||
|
protected:
|
||||||
|
CH422GComponent *parent_;
|
||||||
|
uint8_t pin_;
|
||||||
|
bool inverted_;
|
||||||
|
gpio::Flags flags_;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace ch422g
|
||||||
|
} // namespace esphome
|
|
@ -186,7 +186,7 @@ async def datetime_date_set_to_code(config, action_id, template_arg, args):
|
||||||
|
|
||||||
date_config = config[CONF_DATE]
|
date_config = config[CONF_DATE]
|
||||||
if cg.is_template(date_config):
|
if cg.is_template(date_config):
|
||||||
template_ = await cg.templatable(date_config, [], cg.ESPTime)
|
template_ = await cg.templatable(date_config, args, cg.ESPTime)
|
||||||
cg.add(action_var.set_date(template_))
|
cg.add(action_var.set_date(template_))
|
||||||
else:
|
else:
|
||||||
date_struct = cg.StructInitializer(
|
date_struct = cg.StructInitializer(
|
||||||
|
@ -217,7 +217,7 @@ async def datetime_time_set_to_code(config, action_id, template_arg, args):
|
||||||
|
|
||||||
time_config = config[CONF_TIME]
|
time_config = config[CONF_TIME]
|
||||||
if cg.is_template(time_config):
|
if cg.is_template(time_config):
|
||||||
template_ = await cg.templatable(time_config, [], cg.ESPTime)
|
template_ = await cg.templatable(time_config, args, cg.ESPTime)
|
||||||
cg.add(action_var.set_time(template_))
|
cg.add(action_var.set_time(template_))
|
||||||
else:
|
else:
|
||||||
time_struct = cg.StructInitializer(
|
time_struct = cg.StructInitializer(
|
||||||
|
@ -248,7 +248,7 @@ async def datetime_datetime_set_to_code(config, action_id, template_arg, args):
|
||||||
|
|
||||||
datetime_config = config[CONF_DATETIME]
|
datetime_config = config[CONF_DATETIME]
|
||||||
if cg.is_template(datetime_config):
|
if cg.is_template(datetime_config):
|
||||||
template_ = await cg.templatable(datetime_config, [], cg.ESPTime)
|
template_ = await cg.templatable(datetime_config, args, cg.ESPTime)
|
||||||
cg.add(action_var.set_datetime(template_))
|
cg.add(action_var.set_datetime(template_))
|
||||||
else:
|
else:
|
||||||
datetime_struct = cg.StructInitializer(
|
datetime_struct = cg.StructInitializer(
|
||||||
|
|
|
@ -1,15 +1,15 @@
|
||||||
|
from esphome import automation, core
|
||||||
|
from esphome.automation import maybe_simple_id
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
from esphome import core, automation
|
|
||||||
from esphome.automation import maybe_simple_id
|
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
CONF_AUTO_CLEAR_ENABLED,
|
CONF_AUTO_CLEAR_ENABLED,
|
||||||
|
CONF_FROM,
|
||||||
CONF_ID,
|
CONF_ID,
|
||||||
CONF_LAMBDA,
|
CONF_LAMBDA,
|
||||||
CONF_PAGES,
|
|
||||||
CONF_PAGE_ID,
|
CONF_PAGE_ID,
|
||||||
|
CONF_PAGES,
|
||||||
CONF_ROTATION,
|
CONF_ROTATION,
|
||||||
CONF_FROM,
|
|
||||||
CONF_TO,
|
CONF_TO,
|
||||||
CONF_TRIGGER_ID,
|
CONF_TRIGGER_ID,
|
||||||
)
|
)
|
||||||
|
@ -195,3 +195,4 @@ async def display_is_displaying_page_to_code(config, condition_id, template_arg,
|
||||||
@coroutine_with_priority(100.0)
|
@coroutine_with_priority(100.0)
|
||||||
async def to_code(config):
|
async def to_code(config):
|
||||||
cg.add_global(display_ns.using)
|
cg.add_global(display_ns.using)
|
||||||
|
cg.add_define("USE_DISPLAY")
|
||||||
|
|
|
@ -1,18 +1,23 @@
|
||||||
import esphome.codegen as cg
|
|
||||||
import esphome.config_validation as cv
|
|
||||||
from esphome import pins
|
from esphome import pins
|
||||||
|
import esphome.codegen as cg
|
||||||
from esphome.components import canbus
|
from esphome.components import canbus
|
||||||
from esphome.const import CONF_ID, CONF_RX_PIN, CONF_TX_PIN
|
from esphome.components.canbus import CONF_BIT_RATE, CanbusComponent, CanSpeed
|
||||||
from esphome.components.canbus import CanbusComponent, CanSpeed, CONF_BIT_RATE
|
|
||||||
|
|
||||||
from esphome.components.esp32 import get_esp32_variant
|
from esphome.components.esp32 import get_esp32_variant
|
||||||
from esphome.components.esp32.const import (
|
from esphome.components.esp32.const import (
|
||||||
VARIANT_ESP32,
|
VARIANT_ESP32,
|
||||||
VARIANT_ESP32S2,
|
|
||||||
VARIANT_ESP32S3,
|
|
||||||
VARIANT_ESP32C3,
|
VARIANT_ESP32C3,
|
||||||
VARIANT_ESP32C6,
|
VARIANT_ESP32C6,
|
||||||
VARIANT_ESP32H2,
|
VARIANT_ESP32H2,
|
||||||
|
VARIANT_ESP32S2,
|
||||||
|
VARIANT_ESP32S3,
|
||||||
|
)
|
||||||
|
import esphome.config_validation as cv
|
||||||
|
from esphome.const import (
|
||||||
|
CONF_ID,
|
||||||
|
CONF_RX_PIN,
|
||||||
|
CONF_RX_QUEUE_LEN,
|
||||||
|
CONF_TX_PIN,
|
||||||
|
CONF_TX_QUEUE_LEN,
|
||||||
)
|
)
|
||||||
|
|
||||||
CODEOWNERS = ["@Sympatron"]
|
CODEOWNERS = ["@Sympatron"]
|
||||||
|
@ -77,6 +82,8 @@ CONFIG_SCHEMA = canbus.CANBUS_SCHEMA.extend(
|
||||||
cv.Optional(CONF_BIT_RATE, default="125KBPS"): validate_bit_rate,
|
cv.Optional(CONF_BIT_RATE, default="125KBPS"): validate_bit_rate,
|
||||||
cv.Required(CONF_RX_PIN): pins.internal_gpio_input_pin_number,
|
cv.Required(CONF_RX_PIN): pins.internal_gpio_input_pin_number,
|
||||||
cv.Required(CONF_TX_PIN): pins.internal_gpio_output_pin_number,
|
cv.Required(CONF_TX_PIN): pins.internal_gpio_output_pin_number,
|
||||||
|
cv.Optional(CONF_RX_QUEUE_LEN): cv.uint32_t,
|
||||||
|
cv.Optional(CONF_TX_QUEUE_LEN): cv.uint32_t,
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -87,3 +94,7 @@ async def to_code(config):
|
||||||
|
|
||||||
cg.add(var.set_rx(config[CONF_RX_PIN]))
|
cg.add(var.set_rx(config[CONF_RX_PIN]))
|
||||||
cg.add(var.set_tx(config[CONF_TX_PIN]))
|
cg.add(var.set_tx(config[CONF_TX_PIN]))
|
||||||
|
if (rx_queue_len := config.get(CONF_RX_QUEUE_LEN)) is not None:
|
||||||
|
cg.add(var.set_rx_queue_len(rx_queue_len))
|
||||||
|
if (tx_queue_len := config.get(CONF_TX_QUEUE_LEN)) is not None:
|
||||||
|
cg.add(var.set_tx_queue_len(tx_queue_len))
|
||||||
|
|
|
@ -69,6 +69,13 @@ static bool get_bitrate(canbus::CanSpeed bitrate, twai_timing_config_t *t_config
|
||||||
bool ESP32Can::setup_internal() {
|
bool ESP32Can::setup_internal() {
|
||||||
twai_general_config_t g_config =
|
twai_general_config_t g_config =
|
||||||
TWAI_GENERAL_CONFIG_DEFAULT((gpio_num_t) this->tx_, (gpio_num_t) this->rx_, TWAI_MODE_NORMAL);
|
TWAI_GENERAL_CONFIG_DEFAULT((gpio_num_t) this->tx_, (gpio_num_t) this->rx_, TWAI_MODE_NORMAL);
|
||||||
|
if (this->tx_queue_len_.has_value()) {
|
||||||
|
g_config.tx_queue_len = this->tx_queue_len_.value();
|
||||||
|
}
|
||||||
|
if (this->rx_queue_len_.has_value()) {
|
||||||
|
g_config.rx_queue_len = this->rx_queue_len_.value();
|
||||||
|
}
|
||||||
|
|
||||||
twai_filter_config_t f_config = TWAI_FILTER_CONFIG_ACCEPT_ALL();
|
twai_filter_config_t f_config = TWAI_FILTER_CONFIG_ACCEPT_ALL();
|
||||||
twai_timing_config_t t_config;
|
twai_timing_config_t t_config;
|
||||||
|
|
||||||
|
@ -111,6 +118,7 @@ canbus::Error ESP32Can::send_message(struct canbus::CanFrame *frame) {
|
||||||
.flags = flags,
|
.flags = flags,
|
||||||
.identifier = frame->can_id,
|
.identifier = frame->can_id,
|
||||||
.data_length_code = frame->can_data_length_code,
|
.data_length_code = frame->can_data_length_code,
|
||||||
|
.data = {}, // to suppress warning, data is initialized properly below
|
||||||
};
|
};
|
||||||
if (!frame->remote_transmission_request) {
|
if (!frame->remote_transmission_request) {
|
||||||
memcpy(message.data, frame->data, frame->can_data_length_code);
|
memcpy(message.data, frame->data, frame->can_data_length_code);
|
||||||
|
|
|
@ -12,6 +12,8 @@ class ESP32Can : public canbus::Canbus {
|
||||||
public:
|
public:
|
||||||
void set_rx(int rx) { rx_ = rx; }
|
void set_rx(int rx) { rx_ = rx; }
|
||||||
void set_tx(int tx) { tx_ = tx; }
|
void set_tx(int tx) { tx_ = tx; }
|
||||||
|
void set_tx_queue_len(uint32_t tx_queue_len) { this->tx_queue_len_ = tx_queue_len; }
|
||||||
|
void set_rx_queue_len(uint32_t rx_queue_len) { this->rx_queue_len_ = rx_queue_len; }
|
||||||
ESP32Can(){};
|
ESP32Can(){};
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
|
@ -21,6 +23,8 @@ class ESP32Can : public canbus::Canbus {
|
||||||
|
|
||||||
int rx_{-1};
|
int rx_{-1};
|
||||||
int tx_{-1};
|
int tx_{-1};
|
||||||
|
optional<uint32_t> tx_queue_len_{};
|
||||||
|
optional<uint32_t> rx_queue_len_{};
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace esp32_can
|
} // namespace esp32_can
|
||||||
|
|
|
@ -38,7 +38,8 @@ void ESP32RMTLEDStripLightOutput::setup() {
|
||||||
}
|
}
|
||||||
|
|
||||||
ExternalRAMAllocator<rmt_item32_t> rmt_allocator(ExternalRAMAllocator<rmt_item32_t>::ALLOW_FAILURE);
|
ExternalRAMAllocator<rmt_item32_t> rmt_allocator(ExternalRAMAllocator<rmt_item32_t>::ALLOW_FAILURE);
|
||||||
this->rmt_buf_ = rmt_allocator.allocate(buffer_size * 8); // 8 bits per byte, 1 rmt_item32_t per bit
|
this->rmt_buf_ = rmt_allocator.allocate(buffer_size * 8 +
|
||||||
|
1); // 8 bits per byte, 1 rmt_item32_t per bit + 1 rmt_item32_t for reset
|
||||||
|
|
||||||
rmt_config_t config;
|
rmt_config_t config;
|
||||||
memset(&config, 0, sizeof(config));
|
memset(&config, 0, sizeof(config));
|
||||||
|
@ -66,7 +67,7 @@ void ESP32RMTLEDStripLightOutput::setup() {
|
||||||
}
|
}
|
||||||
|
|
||||||
void ESP32RMTLEDStripLightOutput::set_led_params(uint32_t bit0_high, uint32_t bit0_low, uint32_t bit1_high,
|
void ESP32RMTLEDStripLightOutput::set_led_params(uint32_t bit0_high, uint32_t bit0_low, uint32_t bit1_high,
|
||||||
uint32_t bit1_low) {
|
uint32_t bit1_low, uint32_t reset_time_high, uint32_t reset_time_low) {
|
||||||
float ratio = (float) RMT_CLK_FREQ / RMT_CLK_DIV / 1e09f;
|
float ratio = (float) RMT_CLK_FREQ / RMT_CLK_DIV / 1e09f;
|
||||||
|
|
||||||
// 0-bit
|
// 0-bit
|
||||||
|
@ -79,6 +80,11 @@ void ESP32RMTLEDStripLightOutput::set_led_params(uint32_t bit0_high, uint32_t bi
|
||||||
this->bit1_.level0 = 1;
|
this->bit1_.level0 = 1;
|
||||||
this->bit1_.duration1 = (uint32_t) (ratio * bit1_low);
|
this->bit1_.duration1 = (uint32_t) (ratio * bit1_low);
|
||||||
this->bit1_.level1 = 0;
|
this->bit1_.level1 = 0;
|
||||||
|
// reset
|
||||||
|
this->reset_.duration0 = (uint32_t) (ratio * reset_time_high);
|
||||||
|
this->reset_.level0 = 1;
|
||||||
|
this->reset_.duration1 = (uint32_t) (ratio * reset_time_low);
|
||||||
|
this->reset_.level1 = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void ESP32RMTLEDStripLightOutput::write_state(light::LightState *state) {
|
void ESP32RMTLEDStripLightOutput::write_state(light::LightState *state) {
|
||||||
|
@ -118,6 +124,12 @@ void ESP32RMTLEDStripLightOutput::write_state(light::LightState *state) {
|
||||||
psrc++;
|
psrc++;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (this->reset_.duration0 > 0 || this->reset_.duration1 > 0) {
|
||||||
|
pdest->val = this->reset_.val;
|
||||||
|
pdest++;
|
||||||
|
len++;
|
||||||
|
}
|
||||||
|
|
||||||
if (rmt_write_items(this->channel_, this->rmt_buf_, len, false) != ESP_OK) {
|
if (rmt_write_items(this->channel_, this->rmt_buf_, len, false) != ESP_OK) {
|
||||||
ESP_LOGE(TAG, "RMT TX error");
|
ESP_LOGE(TAG, "RMT TX error");
|
||||||
this->status_set_warning();
|
this->status_set_warning();
|
||||||
|
|
|
@ -49,7 +49,8 @@ class ESP32RMTLEDStripLightOutput : public light::AddressableLight {
|
||||||
/// Set a maximum refresh rate in µs as some lights do not like being updated too often.
|
/// Set a maximum refresh rate in µs as some lights do not like being updated too often.
|
||||||
void set_max_refresh_rate(uint32_t interval_us) { this->max_refresh_rate_ = interval_us; }
|
void set_max_refresh_rate(uint32_t interval_us) { this->max_refresh_rate_ = interval_us; }
|
||||||
|
|
||||||
void set_led_params(uint32_t bit0_high, uint32_t bit0_low, uint32_t bit1_high, uint32_t bit1_low);
|
void set_led_params(uint32_t bit0_high, uint32_t bit0_low, uint32_t bit1_high, uint32_t bit1_low,
|
||||||
|
uint32_t reset_time_high, uint32_t reset_time_low);
|
||||||
|
|
||||||
void set_rgb_order(RGBOrder rgb_order) { this->rgb_order_ = rgb_order; }
|
void set_rgb_order(RGBOrder rgb_order) { this->rgb_order_ = rgb_order; }
|
||||||
void set_rmt_channel(rmt_channel_t channel) { this->channel_ = channel; }
|
void set_rmt_channel(rmt_channel_t channel) { this->channel_ = channel; }
|
||||||
|
@ -75,7 +76,7 @@ class ESP32RMTLEDStripLightOutput : public light::AddressableLight {
|
||||||
bool is_rgbw_;
|
bool is_rgbw_;
|
||||||
bool is_wrgb_;
|
bool is_wrgb_;
|
||||||
|
|
||||||
rmt_item32_t bit0_, bit1_;
|
rmt_item32_t bit0_, bit1_, reset_;
|
||||||
RGBOrder rgb_order_;
|
RGBOrder rgb_order_;
|
||||||
rmt_channel_t channel_;
|
rmt_channel_t channel_;
|
||||||
|
|
||||||
|
|
|
@ -43,13 +43,16 @@ class LEDStripTimings:
|
||||||
bit0_low: int
|
bit0_low: int
|
||||||
bit1_high: int
|
bit1_high: int
|
||||||
bit1_low: int
|
bit1_low: int
|
||||||
|
reset_high: int
|
||||||
|
reset_low: int
|
||||||
|
|
||||||
|
|
||||||
CHIPSETS = {
|
CHIPSETS = {
|
||||||
"WS2812": LEDStripTimings(400, 1000, 1000, 400),
|
"WS2811": LEDStripTimings(300, 1090, 1090, 320, 0, 300000),
|
||||||
"SK6812": LEDStripTimings(300, 900, 600, 600),
|
"WS2812": LEDStripTimings(400, 1000, 1000, 400, 0, 0),
|
||||||
"APA106": LEDStripTimings(350, 1360, 1360, 350),
|
"SK6812": LEDStripTimings(300, 900, 600, 600, 0, 0),
|
||||||
"SM16703": LEDStripTimings(300, 900, 900, 300),
|
"APA106": LEDStripTimings(350, 1360, 1360, 350, 0, 0),
|
||||||
|
"SM16703": LEDStripTimings(300, 900, 900, 300, 0, 0),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -58,6 +61,8 @@ CONF_BIT0_HIGH = "bit0_high"
|
||||||
CONF_BIT0_LOW = "bit0_low"
|
CONF_BIT0_LOW = "bit0_low"
|
||||||
CONF_BIT1_HIGH = "bit1_high"
|
CONF_BIT1_HIGH = "bit1_high"
|
||||||
CONF_BIT1_LOW = "bit1_low"
|
CONF_BIT1_LOW = "bit1_low"
|
||||||
|
CONF_RESET_HIGH = "reset_high"
|
||||||
|
CONF_RESET_LOW = "reset_low"
|
||||||
|
|
||||||
|
|
||||||
CONFIG_SCHEMA = cv.All(
|
CONFIG_SCHEMA = cv.All(
|
||||||
|
@ -88,6 +93,14 @@ CONFIG_SCHEMA = cv.All(
|
||||||
CONF_BIT1_LOW,
|
CONF_BIT1_LOW,
|
||||||
"custom",
|
"custom",
|
||||||
): cv.positive_time_period_nanoseconds,
|
): cv.positive_time_period_nanoseconds,
|
||||||
|
cv.Optional(
|
||||||
|
CONF_RESET_HIGH,
|
||||||
|
default="0 us",
|
||||||
|
): cv.positive_time_period_nanoseconds,
|
||||||
|
cv.Optional(
|
||||||
|
CONF_RESET_LOW,
|
||||||
|
default="0 us",
|
||||||
|
): cv.positive_time_period_nanoseconds,
|
||||||
}
|
}
|
||||||
),
|
),
|
||||||
cv.has_exactly_one_key(CONF_CHIPSET, CONF_BIT0_HIGH),
|
cv.has_exactly_one_key(CONF_CHIPSET, CONF_BIT0_HIGH),
|
||||||
|
@ -113,6 +126,8 @@ async def to_code(config):
|
||||||
chipset.bit0_low,
|
chipset.bit0_low,
|
||||||
chipset.bit1_high,
|
chipset.bit1_high,
|
||||||
chipset.bit1_low,
|
chipset.bit1_low,
|
||||||
|
chipset.reset_high,
|
||||||
|
chipset.reset_low,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
|
@ -122,6 +137,8 @@ async def to_code(config):
|
||||||
config[CONF_BIT0_LOW],
|
config[CONF_BIT0_LOW],
|
||||||
config[CONF_BIT1_HIGH],
|
config[CONF_BIT1_HIGH],
|
||||||
config[CONF_BIT1_LOW],
|
config[CONF_BIT1_LOW],
|
||||||
|
config[CONF_RESET_HIGH],
|
||||||
|
config[CONF_RESET_LOW],
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -472,13 +472,13 @@ void EthernetComponent::start_connect_() {
|
||||||
if (err != ESP_ERR_ESP_NETIF_DHCP_ALREADY_STARTED) {
|
if (err != ESP_ERR_ESP_NETIF_DHCP_ALREADY_STARTED) {
|
||||||
ESPHL_ERROR_CHECK(err, "DHCPC start error");
|
ESPHL_ERROR_CHECK(err, "DHCPC start error");
|
||||||
}
|
}
|
||||||
#if USE_NETWORK_IPV6
|
|
||||||
err = esp_netif_create_ip6_linklocal(this->eth_netif_);
|
|
||||||
if (err != ESP_OK) {
|
|
||||||
ESPHL_ERROR_CHECK(err, "Enable IPv6 link local failed");
|
|
||||||
}
|
|
||||||
#endif /* USE_NETWORK_IPV6 */
|
|
||||||
}
|
}
|
||||||
|
#if USE_NETWORK_IPV6
|
||||||
|
err = esp_netif_create_ip6_linklocal(this->eth_netif_);
|
||||||
|
if (err != ESP_OK) {
|
||||||
|
ESPHL_ERROR_CHECK(err, "Enable IPv6 link local failed");
|
||||||
|
}
|
||||||
|
#endif /* USE_NETWORK_IPV6 */
|
||||||
|
|
||||||
this->connect_begin_ = millis();
|
this->connect_begin_ = millis();
|
||||||
this->status_set_warning();
|
this->status_set_warning();
|
||||||
|
|
|
@ -1,43 +1,35 @@
|
||||||
|
import functools
|
||||||
import hashlib
|
import hashlib
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
import functools
|
|
||||||
from pathlib import Path
|
|
||||||
import os
|
import os
|
||||||
|
from pathlib import Path
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from packaging import version
|
from packaging import version
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
from esphome import core
|
from esphome import core, external_files
|
||||||
from esphome import external_files
|
|
||||||
import esphome.config_validation as cv
|
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
from esphome.helpers import (
|
import esphome.config_validation as cv
|
||||||
copy_file_if_changed,
|
|
||||||
cpp_string_escape,
|
|
||||||
)
|
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
CONF_FAMILY,
|
CONF_FAMILY,
|
||||||
CONF_FILE,
|
CONF_FILE,
|
||||||
CONF_GLYPHS,
|
CONF_GLYPHS,
|
||||||
CONF_ID,
|
CONF_ID,
|
||||||
|
CONF_PATH,
|
||||||
CONF_RAW_DATA_ID,
|
CONF_RAW_DATA_ID,
|
||||||
CONF_TYPE,
|
|
||||||
CONF_REFRESH,
|
CONF_REFRESH,
|
||||||
CONF_SIZE,
|
CONF_SIZE,
|
||||||
CONF_PATH,
|
CONF_TYPE,
|
||||||
CONF_WEIGHT,
|
|
||||||
CONF_URL,
|
CONF_URL,
|
||||||
|
CONF_WEIGHT,
|
||||||
)
|
)
|
||||||
from esphome.core import (
|
from esphome.core import CORE, HexInt
|
||||||
CORE,
|
from esphome.helpers import copy_file_if_changed, cpp_string_escape
|
||||||
HexInt,
|
|
||||||
)
|
|
||||||
|
|
||||||
_LOGGER = logging.getLogger(__name__)
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
DOMAIN = "font"
|
DOMAIN = "font"
|
||||||
DEPENDENCIES = ["display"]
|
|
||||||
MULTI_CONF = True
|
MULTI_CONF = True
|
||||||
|
|
||||||
CODEOWNERS = ["@esphome/core", "@clydebarrow"]
|
CODEOWNERS = ["@esphome/core", "@clydebarrow"]
|
||||||
|
@ -400,10 +392,7 @@ class EFont:
|
||||||
|
|
||||||
|
|
||||||
def convert_bitmap_to_pillow_font(filepath):
|
def convert_bitmap_to_pillow_font(filepath):
|
||||||
from PIL import (
|
from PIL import BdfFontFile, PcfFontFile
|
||||||
PcfFontFile,
|
|
||||||
BdfFontFile,
|
|
||||||
)
|
|
||||||
|
|
||||||
local_bitmap_font_file = external_files.compute_local_file_dir(
|
local_bitmap_font_file = external_files.compute_local_file_dir(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
|
|
|
@ -1,9 +1,8 @@
|
||||||
#include "font.h"
|
#include "font.h"
|
||||||
|
|
||||||
|
#include "esphome/core/color.h"
|
||||||
#include "esphome/core/hal.h"
|
#include "esphome/core/hal.h"
|
||||||
#include "esphome/core/log.h"
|
#include "esphome/core/log.h"
|
||||||
#include "esphome/core/color.h"
|
|
||||||
#include "esphome/components/display/display_buffer.h"
|
|
||||||
|
|
||||||
namespace esphome {
|
namespace esphome {
|
||||||
namespace font {
|
namespace font {
|
||||||
|
@ -68,6 +67,7 @@ int Font::match_next_glyph(const uint8_t *str, int *match_length) {
|
||||||
return -1;
|
return -1;
|
||||||
return lo;
|
return lo;
|
||||||
}
|
}
|
||||||
|
#ifdef USE_DISPLAY
|
||||||
void Font::measure(const char *str, int *width, int *x_offset, int *baseline, int *height) {
|
void Font::measure(const char *str, int *width, int *x_offset, int *baseline, int *height) {
|
||||||
*baseline = this->baseline_;
|
*baseline = this->baseline_;
|
||||||
*height = this->height_;
|
*height = this->height_;
|
||||||
|
@ -164,6 +164,7 @@ void Font::print(int x_start, int y_start, display::Display *display, Color colo
|
||||||
i += match_length;
|
i += match_length;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
} // namespace font
|
} // namespace font
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
|
|
@ -1,8 +1,11 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include "esphome/core/datatypes.h"
|
|
||||||
#include "esphome/core/color.h"
|
#include "esphome/core/color.h"
|
||||||
#include "esphome/components/display/display_buffer.h"
|
#include "esphome/core/datatypes.h"
|
||||||
|
#include "esphome/core/defines.h"
|
||||||
|
#ifdef USE_DISPLAY
|
||||||
|
#include "esphome/components/display/display.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
namespace esphome {
|
namespace esphome {
|
||||||
namespace font {
|
namespace font {
|
||||||
|
@ -38,7 +41,11 @@ class Glyph {
|
||||||
const GlyphData *glyph_data_;
|
const GlyphData *glyph_data_;
|
||||||
};
|
};
|
||||||
|
|
||||||
class Font : public display::BaseFont {
|
class Font
|
||||||
|
#ifdef USE_DISPLAY
|
||||||
|
: public display::BaseFont
|
||||||
|
#endif
|
||||||
|
{
|
||||||
public:
|
public:
|
||||||
/** Construct the font with the given glyphs.
|
/** Construct the font with the given glyphs.
|
||||||
*
|
*
|
||||||
|
@ -50,9 +57,11 @@ class Font : public display::BaseFont {
|
||||||
|
|
||||||
int match_next_glyph(const uint8_t *str, int *match_length);
|
int match_next_glyph(const uint8_t *str, int *match_length);
|
||||||
|
|
||||||
|
#ifdef USE_DISPLAY
|
||||||
void print(int x_start, int y_start, display::Display *display, Color color, const char *text,
|
void print(int x_start, int y_start, display::Display *display, Color color, const char *text,
|
||||||
Color background) override;
|
Color background) override;
|
||||||
void measure(const char *str, int *width, int *x_offset, int *baseline, int *height) override;
|
void measure(const char *str, int *width, int *x_offset, int *baseline, int *height) override;
|
||||||
|
#endif
|
||||||
inline int get_baseline() { return this->baseline_; }
|
inline int get_baseline() { return this->baseline_; }
|
||||||
inline int get_height() { return this->height_; }
|
inline int get_height() { return this->height_; }
|
||||||
inline int get_bpp() { return this->bpp_; }
|
inline int get_bpp() { return this->bpp_; }
|
||||||
|
|
|
@ -1,11 +1,10 @@
|
||||||
import esphome.codegen as cg
|
|
||||||
import esphome.config_validation as cv
|
|
||||||
|
|
||||||
from esphome import pins
|
from esphome import pins
|
||||||
|
import esphome.codegen as cg
|
||||||
from esphome.components import i2c, touchscreen
|
from esphome.components import i2c, touchscreen
|
||||||
from esphome.const import CONF_INTERRUPT_PIN, CONF_ID
|
import esphome.config_validation as cv
|
||||||
from .. import gt911_ns
|
from esphome.const import CONF_ID, CONF_INTERRUPT_PIN, CONF_RESET_PIN
|
||||||
|
|
||||||
|
from .. import gt911_ns
|
||||||
|
|
||||||
GT911ButtonListener = gt911_ns.class_("GT911ButtonListener")
|
GT911ButtonListener = gt911_ns.class_("GT911ButtonListener")
|
||||||
GT911Touchscreen = gt911_ns.class_(
|
GT911Touchscreen = gt911_ns.class_(
|
||||||
|
@ -18,6 +17,7 @@ CONFIG_SCHEMA = touchscreen.TOUCHSCREEN_SCHEMA.extend(
|
||||||
{
|
{
|
||||||
cv.GenerateID(): cv.declare_id(GT911Touchscreen),
|
cv.GenerateID(): cv.declare_id(GT911Touchscreen),
|
||||||
cv.Optional(CONF_INTERRUPT_PIN): pins.internal_gpio_input_pin_schema,
|
cv.Optional(CONF_INTERRUPT_PIN): pins.internal_gpio_input_pin_schema,
|
||||||
|
cv.Optional(CONF_RESET_PIN): pins.gpio_output_pin_schema,
|
||||||
}
|
}
|
||||||
).extend(i2c.i2c_device_schema(0x5D))
|
).extend(i2c.i2c_device_schema(0x5D))
|
||||||
|
|
||||||
|
@ -29,3 +29,5 @@ async def to_code(config):
|
||||||
|
|
||||||
if interrupt_pin := config.get(CONF_INTERRUPT_PIN):
|
if interrupt_pin := config.get(CONF_INTERRUPT_PIN):
|
||||||
cg.add(var.set_interrupt_pin(await cg.gpio_pin_expression(interrupt_pin)))
|
cg.add(var.set_interrupt_pin(await cg.gpio_pin_expression(interrupt_pin)))
|
||||||
|
if reset_pin := config.get(CONF_RESET_PIN):
|
||||||
|
cg.add(var.set_reset_pin(await cg.gpio_pin_expression(reset_pin)))
|
||||||
|
|
|
@ -26,6 +26,23 @@ static const size_t MAX_BUTTONS = 4; // max number of buttons scanned
|
||||||
void GT911Touchscreen::setup() {
|
void GT911Touchscreen::setup() {
|
||||||
i2c::ErrorCode err;
|
i2c::ErrorCode err;
|
||||||
ESP_LOGCONFIG(TAG, "Setting up GT911 Touchscreen...");
|
ESP_LOGCONFIG(TAG, "Setting up GT911 Touchscreen...");
|
||||||
|
if (this->reset_pin_ != nullptr) {
|
||||||
|
this->reset_pin_->setup();
|
||||||
|
this->reset_pin_->digital_write(false);
|
||||||
|
if (this->interrupt_pin_ != nullptr) {
|
||||||
|
// The interrupt pin is used as an input during reset to select the I2C address.
|
||||||
|
this->interrupt_pin_->pin_mode(gpio::FLAG_OUTPUT);
|
||||||
|
this->interrupt_pin_->setup();
|
||||||
|
this->interrupt_pin_->digital_write(false);
|
||||||
|
}
|
||||||
|
delay(2);
|
||||||
|
this->reset_pin_->digital_write(true);
|
||||||
|
delay(50); // NOLINT
|
||||||
|
if (this->interrupt_pin_ != nullptr) {
|
||||||
|
this->interrupt_pin_->pin_mode(gpio::FLAG_INPUT);
|
||||||
|
this->interrupt_pin_->setup();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// check the configuration of the int line.
|
// check the configuration of the int line.
|
||||||
uint8_t data[4];
|
uint8_t data[4];
|
||||||
|
|
|
@ -19,12 +19,14 @@ class GT911Touchscreen : public touchscreen::Touchscreen, public i2c::I2CDevice
|
||||||
void dump_config() override;
|
void dump_config() override;
|
||||||
|
|
||||||
void set_interrupt_pin(InternalGPIOPin *pin) { this->interrupt_pin_ = pin; }
|
void set_interrupt_pin(InternalGPIOPin *pin) { this->interrupt_pin_ = pin; }
|
||||||
|
void set_reset_pin(GPIOPin *pin) { this->reset_pin_ = pin; }
|
||||||
void register_button_listener(GT911ButtonListener *listener) { this->button_listeners_.push_back(listener); }
|
void register_button_listener(GT911ButtonListener *listener) { this->button_listeners_.push_back(listener); }
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
void update_touches() override;
|
void update_touches() override;
|
||||||
|
|
||||||
InternalGPIOPin *interrupt_pin_{};
|
InternalGPIOPin *interrupt_pin_{};
|
||||||
|
GPIOPin *reset_pin_{};
|
||||||
std::vector<GT911ButtonListener *> button_listeners_;
|
std::vector<GT911ButtonListener *> button_listeners_;
|
||||||
uint8_t button_state_{0xFF}; // last button state. Initial FF guarantees first update.
|
uint8_t button_state_{0xFF}; // last button state. Initial FF guarantees first update.
|
||||||
};
|
};
|
||||||
|
|
|
@ -1,31 +1,31 @@
|
||||||
import esphome.codegen as cg
|
|
||||||
import esphome.config_validation as cv
|
|
||||||
from esphome import core, pins
|
from esphome import core, pins
|
||||||
from esphome.components import display, spi, font
|
import esphome.codegen as cg
|
||||||
|
from esphome.components import display, font, spi
|
||||||
from esphome.components.display import validate_rotation
|
from esphome.components.display import validate_rotation
|
||||||
from esphome.core import CORE, HexInt
|
import esphome.config_validation as cv
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
|
CONF_COLOR_ORDER,
|
||||||
CONF_COLOR_PALETTE,
|
CONF_COLOR_PALETTE,
|
||||||
CONF_DC_PIN,
|
CONF_DC_PIN,
|
||||||
CONF_ID,
|
|
||||||
CONF_LAMBDA,
|
|
||||||
CONF_MODEL,
|
|
||||||
CONF_RAW_DATA_ID,
|
|
||||||
CONF_PAGES,
|
|
||||||
CONF_RESET_PIN,
|
|
||||||
CONF_DIMENSIONS,
|
CONF_DIMENSIONS,
|
||||||
CONF_WIDTH,
|
|
||||||
CONF_HEIGHT,
|
CONF_HEIGHT,
|
||||||
CONF_ROTATION,
|
CONF_ID,
|
||||||
|
CONF_INVERT_COLORS,
|
||||||
|
CONF_LAMBDA,
|
||||||
CONF_MIRROR_X,
|
CONF_MIRROR_X,
|
||||||
CONF_MIRROR_Y,
|
CONF_MIRROR_Y,
|
||||||
CONF_SWAP_XY,
|
CONF_MODEL,
|
||||||
CONF_COLOR_ORDER,
|
|
||||||
CONF_OFFSET_HEIGHT,
|
CONF_OFFSET_HEIGHT,
|
||||||
CONF_OFFSET_WIDTH,
|
CONF_OFFSET_WIDTH,
|
||||||
|
CONF_PAGES,
|
||||||
|
CONF_RAW_DATA_ID,
|
||||||
|
CONF_RESET_PIN,
|
||||||
|
CONF_ROTATION,
|
||||||
|
CONF_SWAP_XY,
|
||||||
CONF_TRANSFORM,
|
CONF_TRANSFORM,
|
||||||
CONF_INVERT_COLORS,
|
CONF_WIDTH,
|
||||||
)
|
)
|
||||||
|
from esphome.core import CORE, HexInt
|
||||||
|
|
||||||
DEPENDENCIES = ["spi"]
|
DEPENDENCIES = ["spi"]
|
||||||
|
|
||||||
|
@ -177,7 +177,7 @@ CONFIG_SCHEMA = cv.All(
|
||||||
cv.Optional(CONF_INVERT_DISPLAY): cv.invalid(
|
cv.Optional(CONF_INVERT_DISPLAY): cv.invalid(
|
||||||
"'invert_display' has been replaced by 'invert_colors'"
|
"'invert_display' has been replaced by 'invert_colors'"
|
||||||
),
|
),
|
||||||
cv.Optional(CONF_INVERT_COLORS): cv.boolean,
|
cv.Required(CONF_INVERT_COLORS): cv.boolean,
|
||||||
cv.Optional(CONF_COLOR_ORDER): cv.one_of(*COLOR_ORDERS.keys(), upper=True),
|
cv.Optional(CONF_COLOR_ORDER): cv.one_of(*COLOR_ORDERS.keys(), upper=True),
|
||||||
cv.Exclusive(CONF_ROTATION, CONF_ROTATION): validate_rotation,
|
cv.Exclusive(CONF_ROTATION, CONF_ROTATION): validate_rotation,
|
||||||
cv.Exclusive(CONF_TRANSFORM, CONF_ROTATION): cv.Schema(
|
cv.Exclusive(CONF_TRANSFORM, CONF_ROTATION): cv.Schema(
|
||||||
|
@ -287,5 +287,4 @@ async def to_code(config):
|
||||||
prog_arr = cg.progmem_array(config[CONF_RAW_DATA_ID], rhs)
|
prog_arr = cg.progmem_array(config[CONF_RAW_DATA_ID], rhs)
|
||||||
cg.add(var.set_palette(prog_arr))
|
cg.add(var.set_palette(prog_arr))
|
||||||
|
|
||||||
if CONF_INVERT_COLORS in config:
|
cg.add(var.invert_colors(config[CONF_INVERT_COLORS]))
|
||||||
cg.add(var.invert_colors(config[CONF_INVERT_COLORS]))
|
|
||||||
|
|
|
@ -118,6 +118,7 @@ void ILI9XXXDisplay::dump_config() {
|
||||||
ESP_LOGCONFIG(TAG, " Swap_xy: %s", YESNO(this->swap_xy_));
|
ESP_LOGCONFIG(TAG, " Swap_xy: %s", YESNO(this->swap_xy_));
|
||||||
ESP_LOGCONFIG(TAG, " Mirror_x: %s", YESNO(this->mirror_x_));
|
ESP_LOGCONFIG(TAG, " Mirror_x: %s", YESNO(this->mirror_x_));
|
||||||
ESP_LOGCONFIG(TAG, " Mirror_y: %s", YESNO(this->mirror_y_));
|
ESP_LOGCONFIG(TAG, " Mirror_y: %s", YESNO(this->mirror_y_));
|
||||||
|
ESP_LOGCONFIG(TAG, " Invert colors: %s", YESNO(this->pre_invertcolors_));
|
||||||
|
|
||||||
if (this->is_failed()) {
|
if (this->is_failed()) {
|
||||||
ESP_LOGCONFIG(TAG, " => Failed to init Memory: YES!");
|
ESP_LOGCONFIG(TAG, " => Failed to init Memory: YES!");
|
||||||
|
@ -154,7 +155,6 @@ void ILI9XXXDisplay::fill(Color color) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
break;
|
|
||||||
default:
|
default:
|
||||||
new_color = display::ColorUtil::color_to_332(color, display::ColorOrder::COLOR_ORDER_RGB);
|
new_color = display::ColorUtil::color_to_332(color, display::ColorOrder::COLOR_ORDER_RGB);
|
||||||
break;
|
break;
|
||||||
|
|
|
@ -28,8 +28,8 @@ class ILI9XXXDisplay : public display::DisplayBuffer,
|
||||||
spi::CLOCK_PHASE_LEADING, spi::DATA_RATE_40MHZ> {
|
spi::CLOCK_PHASE_LEADING, spi::DATA_RATE_40MHZ> {
|
||||||
public:
|
public:
|
||||||
ILI9XXXDisplay() = default;
|
ILI9XXXDisplay() = default;
|
||||||
ILI9XXXDisplay(uint8_t const *init_sequence, int16_t width, int16_t height, bool invert_colors)
|
ILI9XXXDisplay(uint8_t const *init_sequence, int16_t width, int16_t height)
|
||||||
: init_sequence_{init_sequence}, width_{width}, height_{height}, pre_invertcolors_{invert_colors} {
|
: init_sequence_{init_sequence}, width_{width}, height_{height} {
|
||||||
uint8_t cmd, num_args, bits;
|
uint8_t cmd, num_args, bits;
|
||||||
const uint8_t *addr = init_sequence;
|
const uint8_t *addr = init_sequence;
|
||||||
while ((cmd = *addr++) != 0) {
|
while ((cmd = *addr++) != 0) {
|
||||||
|
@ -144,7 +144,7 @@ class ILI9XXXDisplay : public display::DisplayBuffer,
|
||||||
bool need_update_ = false;
|
bool need_update_ = false;
|
||||||
bool is_18bitdisplay_ = false;
|
bool is_18bitdisplay_ = false;
|
||||||
PixelMode pixel_mode_{};
|
PixelMode pixel_mode_{};
|
||||||
bool pre_invertcolors_ = false;
|
bool pre_invertcolors_{};
|
||||||
display::ColorOrder color_order_{display::COLOR_ORDER_BGR};
|
display::ColorOrder color_order_{display::COLOR_ORDER_BGR};
|
||||||
bool swap_xy_{};
|
bool swap_xy_{};
|
||||||
bool mirror_x_{};
|
bool mirror_x_{};
|
||||||
|
@ -154,54 +154,54 @@ class ILI9XXXDisplay : public display::DisplayBuffer,
|
||||||
//----------- M5Stack display --------------
|
//----------- M5Stack display --------------
|
||||||
class ILI9XXXM5Stack : public ILI9XXXDisplay {
|
class ILI9XXXM5Stack : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXM5Stack() : ILI9XXXDisplay(INITCMD_M5STACK, 320, 240, true) {}
|
ILI9XXXM5Stack() : ILI9XXXDisplay(INITCMD_M5STACK, 320, 240) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- M5Stack display --------------
|
//----------- M5Stack display --------------
|
||||||
class ILI9XXXM5CORE : public ILI9XXXDisplay {
|
class ILI9XXXM5CORE : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXM5CORE() : ILI9XXXDisplay(INITCMD_M5CORE, 320, 240, true) {}
|
ILI9XXXM5CORE() : ILI9XXXDisplay(INITCMD_M5CORE, 320, 240) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ST7789V display --------------
|
//----------- ST7789V display --------------
|
||||||
class ILI9XXXST7789V : public ILI9XXXDisplay {
|
class ILI9XXXST7789V : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXST7789V() : ILI9XXXDisplay(INITCMD_ST7789V, 240, 320, false) {}
|
ILI9XXXST7789V() : ILI9XXXDisplay(INITCMD_ST7789V, 240, 320) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9XXX_24_TFT display --------------
|
//----------- ILI9XXX_24_TFT display --------------
|
||||||
class ILI9XXXILI9341 : public ILI9XXXDisplay {
|
class ILI9XXXILI9341 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI9341() : ILI9XXXDisplay(INITCMD_ILI9341, 240, 320, false) {}
|
ILI9XXXILI9341() : ILI9XXXDisplay(INITCMD_ILI9341, 240, 320) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9XXX_24_TFT rotated display --------------
|
//----------- ILI9XXX_24_TFT rotated display --------------
|
||||||
class ILI9XXXILI9342 : public ILI9XXXDisplay {
|
class ILI9XXXILI9342 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI9342() : ILI9XXXDisplay(INITCMD_ILI9341, 320, 240, false) {}
|
ILI9XXXILI9342() : ILI9XXXDisplay(INITCMD_ILI9341, 320, 240) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9XXX_??_TFT rotated display --------------
|
//----------- ILI9XXX_??_TFT rotated display --------------
|
||||||
class ILI9XXXILI9481 : public ILI9XXXDisplay {
|
class ILI9XXXILI9481 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI9481() : ILI9XXXDisplay(INITCMD_ILI9481, 480, 320, false) {}
|
ILI9XXXILI9481() : ILI9XXXDisplay(INITCMD_ILI9481, 480, 320) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9481 in 18 bit mode --------------
|
//----------- ILI9481 in 18 bit mode --------------
|
||||||
class ILI9XXXILI948118 : public ILI9XXXDisplay {
|
class ILI9XXXILI948118 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI948118() : ILI9XXXDisplay(INITCMD_ILI9481_18, 320, 480, true) {}
|
ILI9XXXILI948118() : ILI9XXXDisplay(INITCMD_ILI9481_18, 320, 480) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9XXX_35_TFT rotated display --------------
|
//----------- ILI9XXX_35_TFT rotated display --------------
|
||||||
class ILI9XXXILI9486 : public ILI9XXXDisplay {
|
class ILI9XXXILI9486 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI9486() : ILI9XXXDisplay(INITCMD_ILI9486, 480, 320, false) {}
|
ILI9XXXILI9486() : ILI9XXXDisplay(INITCMD_ILI9486, 480, 320) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class ILI9XXXILI9488 : public ILI9XXXDisplay {
|
class ILI9XXXILI9488 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI9488(const uint8_t *seq = INITCMD_ILI9488) : ILI9XXXDisplay(seq, 480, 320, true) {}
|
ILI9XXXILI9488(const uint8_t *seq = INITCMD_ILI9488) : ILI9XXXDisplay(seq, 480, 320) {}
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
void set_madctl() override {
|
void set_madctl() override {
|
||||||
|
@ -246,34 +246,34 @@ class WAVESHARERES35 : public ILI9XXXILI9488 {
|
||||||
//----------- ILI9XXX_35_TFT origin colors rotated display --------------
|
//----------- ILI9XXX_35_TFT origin colors rotated display --------------
|
||||||
class ILI9XXXILI9488A : public ILI9XXXDisplay {
|
class ILI9XXXILI9488A : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXILI9488A() : ILI9XXXDisplay(INITCMD_ILI9488_A, 480, 320, true) {}
|
ILI9XXXILI9488A() : ILI9XXXDisplay(INITCMD_ILI9488_A, 480, 320) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9XXX_35_TFT rotated display --------------
|
//----------- ILI9XXX_35_TFT rotated display --------------
|
||||||
class ILI9XXXST7796 : public ILI9XXXDisplay {
|
class ILI9XXXST7796 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXST7796() : ILI9XXXDisplay(INITCMD_ST7796, 320, 480, false) {}
|
ILI9XXXST7796() : ILI9XXXDisplay(INITCMD_ST7796, 320, 480) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class ILI9XXXS3Box : public ILI9XXXDisplay {
|
class ILI9XXXS3Box : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXS3Box() : ILI9XXXDisplay(INITCMD_S3BOX, 320, 240, false) {}
|
ILI9XXXS3Box() : ILI9XXXDisplay(INITCMD_S3BOX, 320, 240) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class ILI9XXXS3BoxLite : public ILI9XXXDisplay {
|
class ILI9XXXS3BoxLite : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXS3BoxLite() : ILI9XXXDisplay(INITCMD_S3BOXLITE, 320, 240, true) {}
|
ILI9XXXS3BoxLite() : ILI9XXXDisplay(INITCMD_S3BOXLITE, 320, 240) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class ILI9XXXGC9A01A : public ILI9XXXDisplay {
|
class ILI9XXXGC9A01A : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXGC9A01A() : ILI9XXXDisplay(INITCMD_GC9A01A, 240, 240, true) {}
|
ILI9XXXGC9A01A() : ILI9XXXDisplay(INITCMD_GC9A01A, 240, 240) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
//----------- ILI9XXX_24_TFT display --------------
|
//----------- ILI9XXX_24_TFT display --------------
|
||||||
class ILI9XXXST7735 : public ILI9XXXDisplay {
|
class ILI9XXXST7735 : public ILI9XXXDisplay {
|
||||||
public:
|
public:
|
||||||
ILI9XXXST7735() : ILI9XXXDisplay(INITCMD_ST7735, 128, 160, false) {}
|
ILI9XXXST7735() : ILI9XXXDisplay(INITCMD_ST7735, 128, 160) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace ili9xxx
|
} // namespace ili9xxx
|
||||||
|
|
|
@ -101,7 +101,6 @@ static const uint8_t PROGMEM INITCMD_ILI9481[] = {
|
||||||
ILI9XXX_MADCTL , 1, MADCTL_MV | MADCTL_BGR, // Memory Access Control
|
ILI9XXX_MADCTL , 1, MADCTL_MV | MADCTL_BGR, // Memory Access Control
|
||||||
ILI9XXX_CSCON , 1, 0x01,
|
ILI9XXX_CSCON , 1, 0x01,
|
||||||
ILI9XXX_PIXFMT, 1, 0x55, // 16 bit mode
|
ILI9XXX_PIXFMT, 1, 0x55, // 16 bit mode
|
||||||
ILI9XXX_INVON, 0,
|
|
||||||
ILI9XXX_DISPON, 0x80, // Set display on
|
ILI9XXX_DISPON, 0x80, // Set display on
|
||||||
0x00 // end
|
0x00 // end
|
||||||
};
|
};
|
||||||
|
@ -121,7 +120,6 @@ static const uint8_t PROGMEM INITCMD_ILI9481_18[] = {
|
||||||
ILI9XXX_MADCTL , 1, MADCTL_MX| MADCTL_BGR, // Memory Access Control
|
ILI9XXX_MADCTL , 1, MADCTL_MX| MADCTL_BGR, // Memory Access Control
|
||||||
ILI9XXX_CSCON , 1, 0x01,
|
ILI9XXX_CSCON , 1, 0x01,
|
||||||
ILI9XXX_PIXFMT, 1, 0x66, // 18 bit mode
|
ILI9XXX_PIXFMT, 1, 0x66, // 18 bit mode
|
||||||
ILI9XXX_INVON, 0,
|
|
||||||
ILI9XXX_DISPON, 0x80, // Set display on
|
ILI9XXX_DISPON, 0x80, // Set display on
|
||||||
0x00 // end
|
0x00 // end
|
||||||
};
|
};
|
||||||
|
@ -204,7 +202,6 @@ static const uint8_t PROGMEM INITCMD_ILI9488_A[] = {
|
||||||
|
|
||||||
|
|
||||||
ILI9XXX_SLPOUT, 0x80, // Exit sleep mode
|
ILI9XXX_SLPOUT, 0x80, // Exit sleep mode
|
||||||
//ILI9XXX_INVON , 0,
|
|
||||||
ILI9XXX_DISPON, 0x80, // Set display on
|
ILI9XXX_DISPON, 0x80, // Set display on
|
||||||
0x00 // end
|
0x00 // end
|
||||||
};
|
};
|
||||||
|
|
|
@ -8,6 +8,8 @@
|
||||||
#endif
|
#endif
|
||||||
#include <driver/ledc.h>
|
#include <driver/ledc.h>
|
||||||
|
|
||||||
|
#include <cinttypes>
|
||||||
|
|
||||||
#define CLOCK_FREQUENCY 80e6f
|
#define CLOCK_FREQUENCY 80e6f
|
||||||
|
|
||||||
#ifdef USE_ARDUINO
|
#ifdef USE_ARDUINO
|
||||||
|
@ -115,20 +117,22 @@ void LEDCOutput::write_state(float state) {
|
||||||
const uint32_t max_duty = (uint32_t(1) << this->bit_depth_) - 1;
|
const uint32_t max_duty = (uint32_t(1) << this->bit_depth_) - 1;
|
||||||
const float duty_rounded = roundf(state * max_duty);
|
const float duty_rounded = roundf(state * max_duty);
|
||||||
auto duty = static_cast<uint32_t>(duty_rounded);
|
auto duty = static_cast<uint32_t>(duty_rounded);
|
||||||
|
ESP_LOGV(TAG, "Setting duty: %" PRIu32 " on channel %u", duty, this->channel_);
|
||||||
#ifdef USE_ARDUINO
|
#ifdef USE_ARDUINO
|
||||||
ESP_LOGV(TAG, "Setting duty: %u on channel %u", duty, this->channel_);
|
|
||||||
ledcWrite(this->channel_, duty);
|
ledcWrite(this->channel_, duty);
|
||||||
#endif
|
#endif
|
||||||
#ifdef USE_ESP_IDF
|
#ifdef USE_ESP_IDF
|
||||||
// ensure that 100% on is not 99.975% on
|
|
||||||
if ((duty == max_duty) && (max_duty != 1)) {
|
|
||||||
duty = max_duty + 1;
|
|
||||||
}
|
|
||||||
auto speed_mode = get_speed_mode(channel_);
|
auto speed_mode = get_speed_mode(channel_);
|
||||||
auto chan_num = static_cast<ledc_channel_t>(channel_ % 8);
|
auto chan_num = static_cast<ledc_channel_t>(channel_ % 8);
|
||||||
int hpoint = ledc_angle_to_htop(this->phase_angle_, this->bit_depth_);
|
int hpoint = ledc_angle_to_htop(this->phase_angle_, this->bit_depth_);
|
||||||
ledc_set_duty_with_hpoint(speed_mode, chan_num, duty, hpoint);
|
if (duty == max_duty) {
|
||||||
ledc_update_duty(speed_mode, chan_num);
|
ledc_stop(speed_mode, chan_num, 1);
|
||||||
|
} else if (duty == 0) {
|
||||||
|
ledc_stop(speed_mode, chan_num, 0);
|
||||||
|
} else {
|
||||||
|
ledc_set_duty_with_hpoint(speed_mode, chan_num, duty, hpoint);
|
||||||
|
ledc_update_duty(speed_mode, chan_num);
|
||||||
|
}
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -21,8 +21,8 @@ from esphome.final_validate import full_config
|
||||||
from esphome.helpers import write_file_if_changed
|
from esphome.helpers import write_file_if_changed
|
||||||
|
|
||||||
from . import defines as df, helpers, lv_validation as lvalid
|
from . import defines as df, helpers, lv_validation as lvalid
|
||||||
from .automation import disp_update, update_to_code
|
from .automation import disp_update, focused_widgets, update_to_code
|
||||||
from .defines import CONF_SKIP
|
from .defines import CONF_ADJUSTABLE, CONF_SKIP
|
||||||
from .encoders import ENCODERS_CONFIG, encoders_to_code, initial_focus_to_code
|
from .encoders import ENCODERS_CONFIG, encoders_to_code, initial_focus_to_code
|
||||||
from .lv_validation import lv_bool, lv_images_used
|
from .lv_validation import lv_bool, lv_images_used
|
||||||
from .lvcode import LvContext, LvglComponent
|
from .lvcode import LvContext, LvglComponent
|
||||||
|
@ -67,7 +67,7 @@ from .widgets.lv_bar import bar_spec
|
||||||
from .widgets.meter import meter_spec
|
from .widgets.meter import meter_spec
|
||||||
from .widgets.msgbox import MSGBOX_SCHEMA, msgboxes_to_code
|
from .widgets.msgbox import MSGBOX_SCHEMA, msgboxes_to_code
|
||||||
from .widgets.obj import obj_spec
|
from .widgets.obj import obj_spec
|
||||||
from .widgets.page import add_pages, page_spec
|
from .widgets.page import add_pages, generate_page_triggers, page_spec
|
||||||
from .widgets.roller import roller_spec
|
from .widgets.roller import roller_spec
|
||||||
from .widgets.slider import slider_spec
|
from .widgets.slider import slider_spec
|
||||||
from .widgets.spinbox import spinbox_spec
|
from .widgets.spinbox import spinbox_spec
|
||||||
|
@ -182,6 +182,14 @@ def final_validation(config):
|
||||||
raise cv.Invalid(
|
raise cv.Invalid(
|
||||||
"Using RGBA or RGB24 in image config not compatible with LVGL", path
|
"Using RGBA or RGB24 in image config not compatible with LVGL", path
|
||||||
)
|
)
|
||||||
|
for w in focused_widgets:
|
||||||
|
path = global_config.get_path_for_id(w)
|
||||||
|
widget_conf = global_config.get_config_for_path(path[:-1])
|
||||||
|
if CONF_ADJUSTABLE in widget_conf and not widget_conf[CONF_ADJUSTABLE]:
|
||||||
|
raise cv.Invalid(
|
||||||
|
"A non adjustable arc may not be focused",
|
||||||
|
path,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
async def to_code(config):
|
async def to_code(config):
|
||||||
|
@ -266,8 +274,12 @@ async def to_code(config):
|
||||||
await add_top_layer(config)
|
await add_top_layer(config)
|
||||||
await msgboxes_to_code(config)
|
await msgboxes_to_code(config)
|
||||||
await disp_update(f"{lv_component}->get_disp()", config)
|
await disp_update(f"{lv_component}->get_disp()", config)
|
||||||
Widget.set_completed()
|
# At this point only the setup code should be generated
|
||||||
|
assert LvContext.added_lambda_count == 1
|
||||||
|
Widget.set_completed()
|
||||||
|
async with LvContext(lv_component):
|
||||||
await generate_triggers(lv_component)
|
await generate_triggers(lv_component)
|
||||||
|
await generate_page_triggers(lv_component, config)
|
||||||
for conf in config.get(CONF_ON_IDLE, ()):
|
for conf in config.get(CONF_ON_IDLE, ()):
|
||||||
templ = await cg.templatable(conf[CONF_TIMEOUT], [], cg.uint32)
|
templ = await cg.templatable(conf[CONF_TIMEOUT], [], cg.uint32)
|
||||||
idle_trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], lv_component, templ)
|
idle_trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], lv_component, templ)
|
||||||
|
@ -315,6 +327,8 @@ CONFIG_SCHEMA = (
|
||||||
{
|
{
|
||||||
cv.Optional(df.CONF_GRID_CELL_X_ALIGN): grid_alignments,
|
cv.Optional(df.CONF_GRID_CELL_X_ALIGN): grid_alignments,
|
||||||
cv.Optional(df.CONF_GRID_CELL_Y_ALIGN): grid_alignments,
|
cv.Optional(df.CONF_GRID_CELL_Y_ALIGN): grid_alignments,
|
||||||
|
cv.Optional(df.CONF_PAD_ROW): lvalid.pixels,
|
||||||
|
cv.Optional(df.CONF_PAD_COLUMN): lvalid.pixels,
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
|
|
|
@ -4,12 +4,15 @@ from typing import Callable
|
||||||
from esphome import automation
|
from esphome import automation
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
from esphome.const import CONF_ID, CONF_TIMEOUT
|
from esphome.const import CONF_ACTION, CONF_GROUP, CONF_ID, CONF_TIMEOUT
|
||||||
|
from esphome.cpp_generator import RawExpression, get_variable
|
||||||
from esphome.cpp_types import nullptr
|
from esphome.cpp_types import nullptr
|
||||||
|
|
||||||
from .defines import (
|
from .defines import (
|
||||||
CONF_DISP_BG_COLOR,
|
CONF_DISP_BG_COLOR,
|
||||||
CONF_DISP_BG_IMAGE,
|
CONF_DISP_BG_IMAGE,
|
||||||
|
CONF_EDITING,
|
||||||
|
CONF_FREEZE,
|
||||||
CONF_LVGL_ID,
|
CONF_LVGL_ID,
|
||||||
CONF_SHOW_SNOW,
|
CONF_SHOW_SNOW,
|
||||||
literal,
|
literal,
|
||||||
|
@ -26,8 +29,10 @@ from .lvcode import (
|
||||||
add_line_marks,
|
add_line_marks,
|
||||||
lv,
|
lv,
|
||||||
lv_add,
|
lv_add,
|
||||||
|
lv_expr,
|
||||||
lv_obj,
|
lv_obj,
|
||||||
lvgl_comp,
|
lvgl_comp,
|
||||||
|
static_cast,
|
||||||
)
|
)
|
||||||
from .schemas import DISP_BG_SCHEMA, LIST_ACTION_SCHEMA, LVGL_SCHEMA
|
from .schemas import DISP_BG_SCHEMA, LIST_ACTION_SCHEMA, LVGL_SCHEMA
|
||||||
from .types import (
|
from .types import (
|
||||||
|
@ -36,9 +41,20 @@ from .types import (
|
||||||
LvglCondition,
|
LvglCondition,
|
||||||
ObjUpdateAction,
|
ObjUpdateAction,
|
||||||
lv_disp_t,
|
lv_disp_t,
|
||||||
|
lv_group_t,
|
||||||
lv_obj_t,
|
lv_obj_t,
|
||||||
|
lv_pseudo_button_t,
|
||||||
)
|
)
|
||||||
from .widgets import Widget, get_widgets, lv_scr_act, set_obj_properties
|
from .widgets import (
|
||||||
|
Widget,
|
||||||
|
get_widgets,
|
||||||
|
lv_scr_act,
|
||||||
|
set_obj_properties,
|
||||||
|
wait_for_widgets,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Record widgets that are used in a focused action here
|
||||||
|
focused_widgets = set()
|
||||||
|
|
||||||
|
|
||||||
async def action_to_code(
|
async def action_to_code(
|
||||||
|
@ -48,10 +64,12 @@ async def action_to_code(
|
||||||
template_arg,
|
template_arg,
|
||||||
args,
|
args,
|
||||||
):
|
):
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext(parameters=args, where=action_id) as context:
|
async with LambdaContext(parameters=args, where=action_id) as context:
|
||||||
|
with LvConditional(lv_expr.is_pre_initialise()):
|
||||||
|
context.add(RawExpression("return"))
|
||||||
for widget in widgets:
|
for widget in widgets:
|
||||||
with LvConditional(widget.obj != nullptr):
|
await action(widget)
|
||||||
await action(widget)
|
|
||||||
var = cg.new_Pvariable(action_id, template_arg, await context.get_lambda())
|
var = cg.new_Pvariable(action_id, template_arg, await context.get_lambda())
|
||||||
return var
|
return var
|
||||||
|
|
||||||
|
@ -147,7 +165,7 @@ async def lvgl_update_to_code(config, action_id, template_arg, args):
|
||||||
widgets = await get_widgets(config)
|
widgets = await get_widgets(config)
|
||||||
w = widgets[0]
|
w = widgets[0]
|
||||||
disp = f"{w.obj}->get_disp()"
|
disp = f"{w.obj}->get_disp()"
|
||||||
async with LambdaContext(parameters=args, where=action_id) as context:
|
async with LambdaContext(LVGL_COMP_ARG, where=action_id) as context:
|
||||||
await disp_update(disp, config)
|
await disp_update(disp, config)
|
||||||
var = cg.new_Pvariable(action_id, template_arg, await context.get_lambda())
|
var = cg.new_Pvariable(action_id, template_arg, await context.get_lambda())
|
||||||
await cg.register_parented(var, w.var)
|
await cg.register_parented(var, w.var)
|
||||||
|
@ -224,3 +242,72 @@ async def obj_show_to_code(config, action_id, template_arg, args):
|
||||||
return await action_to_code(
|
return await action_to_code(
|
||||||
await get_widgets(config), do_show, action_id, template_arg, args
|
await get_widgets(config), do_show, action_id, template_arg, args
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def focused_id(value):
|
||||||
|
value = cv.use_id(lv_pseudo_button_t)(value)
|
||||||
|
focused_widgets.add(value)
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
@automation.register_action(
|
||||||
|
"lvgl.widget.focus",
|
||||||
|
ObjUpdateAction,
|
||||||
|
cv.Any(
|
||||||
|
cv.maybe_simple_value(
|
||||||
|
{
|
||||||
|
cv.Optional(CONF_GROUP): cv.use_id(lv_group_t),
|
||||||
|
cv.Required(CONF_ACTION): cv.one_of(
|
||||||
|
"MARK", "RESTORE", "NEXT", "PREVIOUS", upper=True
|
||||||
|
),
|
||||||
|
cv.GenerateID(CONF_LVGL_ID): cv.use_id(LvglComponent),
|
||||||
|
cv.Optional(CONF_FREEZE, default=False): cv.boolean,
|
||||||
|
},
|
||||||
|
key=CONF_ACTION,
|
||||||
|
),
|
||||||
|
cv.maybe_simple_value(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_ID): focused_id,
|
||||||
|
cv.Optional(CONF_FREEZE, default=False): cv.boolean,
|
||||||
|
cv.Optional(CONF_EDITING, default=False): cv.boolean,
|
||||||
|
},
|
||||||
|
key=CONF_ID,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
async def widget_focus(config, action_id, template_arg, args):
|
||||||
|
widget = await get_widgets(config)
|
||||||
|
if widget:
|
||||||
|
widget = widget[0]
|
||||||
|
group = static_cast(
|
||||||
|
lv_group_t.operator("ptr"), lv_expr.obj_get_group(widget.obj)
|
||||||
|
)
|
||||||
|
elif group := config.get(CONF_GROUP):
|
||||||
|
group = await get_variable(group)
|
||||||
|
else:
|
||||||
|
group = lv_expr.group_get_default()
|
||||||
|
|
||||||
|
async with LambdaContext(parameters=args, where=action_id) as context:
|
||||||
|
if widget:
|
||||||
|
lv.group_focus_freeze(group, False)
|
||||||
|
lv.group_focus_obj(widget.obj)
|
||||||
|
if config[CONF_EDITING]:
|
||||||
|
lv.group_set_editing(group, True)
|
||||||
|
else:
|
||||||
|
action = config[CONF_ACTION]
|
||||||
|
lv_comp = await get_variable(config[CONF_LVGL_ID])
|
||||||
|
if action == "MARK":
|
||||||
|
context.add(lv_comp.set_focus_mark(group))
|
||||||
|
else:
|
||||||
|
lv.group_focus_freeze(group, False)
|
||||||
|
if action == "RESTORE":
|
||||||
|
context.add(lv_comp.restore_focus_mark(group))
|
||||||
|
elif action == "NEXT":
|
||||||
|
lv.group_focus_next(group)
|
||||||
|
else:
|
||||||
|
lv.group_focus_prev(group)
|
||||||
|
|
||||||
|
if config[CONF_FREEZE]:
|
||||||
|
lv.group_focus_freeze(group, True)
|
||||||
|
var = cg.new_Pvariable(action_id, template_arg, await context.get_lambda())
|
||||||
|
return var
|
||||||
|
|
|
@ -10,7 +10,7 @@ from ..defines import CONF_LVGL_ID, CONF_WIDGET
|
||||||
from ..lvcode import EVENT_ARG, LambdaContext, LvContext
|
from ..lvcode import EVENT_ARG, LambdaContext, LvContext
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, lv_pseudo_button_t
|
from ..types import LV_EVENT, lv_pseudo_button_t
|
||||||
from ..widgets import Widget, get_widgets
|
from ..widgets import Widget, get_widgets, wait_for_widgets
|
||||||
|
|
||||||
CONFIG_SCHEMA = (
|
CONFIG_SCHEMA = (
|
||||||
binary_sensor_schema(BinarySensor)
|
binary_sensor_schema(BinarySensor)
|
||||||
|
@ -29,6 +29,7 @@ async def to_code(config):
|
||||||
widget = await get_widgets(config, CONF_WIDGET)
|
widget = await get_widgets(config, CONF_WIDGET)
|
||||||
widget = widget[0]
|
widget = widget[0]
|
||||||
assert isinstance(widget, Widget)
|
assert isinstance(widget, Widget)
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext(EVENT_ARG) as pressed_ctx:
|
async with LambdaContext(EVENT_ARG) as pressed_ctx:
|
||||||
pressed_ctx.add(sensor.publish_state(widget.is_pressed()))
|
pressed_ctx.add(sensor.publish_state(widget.is_pressed()))
|
||||||
async with LvContext(paren) as ctx:
|
async with LvContext(paren) as ctx:
|
||||||
|
|
|
@ -148,6 +148,7 @@ LV_EVENT_MAP = {
|
||||||
"DEFOCUS": "DEFOCUSED",
|
"DEFOCUS": "DEFOCUSED",
|
||||||
"READY": "READY",
|
"READY": "READY",
|
||||||
"CANCEL": "CANCEL",
|
"CANCEL": "CANCEL",
|
||||||
|
"ALL_EVENTS": "ALL",
|
||||||
}
|
}
|
||||||
|
|
||||||
LV_EVENT_TRIGGERS = tuple(f"on_{x.lower()}" for x in LV_EVENT_MAP)
|
LV_EVENT_TRIGGERS = tuple(f"on_{x.lower()}" for x in LV_EVENT_MAP)
|
||||||
|
@ -390,6 +391,7 @@ CONF_DEFAULT_FONT = "default_font"
|
||||||
CONF_DEFAULT_GROUP = "default_group"
|
CONF_DEFAULT_GROUP = "default_group"
|
||||||
CONF_DIR = "dir"
|
CONF_DIR = "dir"
|
||||||
CONF_DISPLAYS = "displays"
|
CONF_DISPLAYS = "displays"
|
||||||
|
CONF_EDITING = "editing"
|
||||||
CONF_ENCODERS = "encoders"
|
CONF_ENCODERS = "encoders"
|
||||||
CONF_END_ANGLE = "end_angle"
|
CONF_END_ANGLE = "end_angle"
|
||||||
CONF_END_VALUE = "end_value"
|
CONF_END_VALUE = "end_value"
|
||||||
|
@ -401,6 +403,7 @@ CONF_FLEX_ALIGN_MAIN = "flex_align_main"
|
||||||
CONF_FLEX_ALIGN_CROSS = "flex_align_cross"
|
CONF_FLEX_ALIGN_CROSS = "flex_align_cross"
|
||||||
CONF_FLEX_ALIGN_TRACK = "flex_align_track"
|
CONF_FLEX_ALIGN_TRACK = "flex_align_track"
|
||||||
CONF_FLEX_GROW = "flex_grow"
|
CONF_FLEX_GROW = "flex_grow"
|
||||||
|
CONF_FREEZE = "freeze"
|
||||||
CONF_FULL_REFRESH = "full_refresh"
|
CONF_FULL_REFRESH = "full_refresh"
|
||||||
CONF_GRID_CELL_ROW_POS = "grid_cell_row_pos"
|
CONF_GRID_CELL_ROW_POS = "grid_cell_row_pos"
|
||||||
CONF_GRID_CELL_COLUMN_POS = "grid_cell_column_pos"
|
CONF_GRID_CELL_COLUMN_POS = "grid_cell_column_pos"
|
||||||
|
@ -428,9 +431,9 @@ CONF_MSGBOXES = "msgboxes"
|
||||||
CONF_OBJ = "obj"
|
CONF_OBJ = "obj"
|
||||||
CONF_OFFSET_X = "offset_x"
|
CONF_OFFSET_X = "offset_x"
|
||||||
CONF_OFFSET_Y = "offset_y"
|
CONF_OFFSET_Y = "offset_y"
|
||||||
|
CONF_ONE_CHECKED = "one_checked"
|
||||||
CONF_ONE_LINE = "one_line"
|
CONF_ONE_LINE = "one_line"
|
||||||
CONF_ON_SELECT = "on_select"
|
CONF_ON_SELECT = "on_select"
|
||||||
CONF_ONE_CHECKED = "one_checked"
|
|
||||||
CONF_NEXT = "next"
|
CONF_NEXT = "next"
|
||||||
CONF_PAD_ROW = "pad_row"
|
CONF_PAD_ROW = "pad_row"
|
||||||
CONF_PAD_COLUMN = "pad_column"
|
CONF_PAD_COLUMN = "pad_column"
|
||||||
|
@ -505,4 +508,10 @@ DEFAULT_ESPHOME_FONT = "esphome_lv_default_font"
|
||||||
|
|
||||||
|
|
||||||
def join_enums(enums, prefix=""):
|
def join_enums(enums, prefix=""):
|
||||||
return literal("|".join(f"(int){prefix}{e.upper()}" for e in enums))
|
enums = list(enums)
|
||||||
|
enums.sort()
|
||||||
|
# If a prefix is provided, prepend each constant with the prefix, and assume that all the constants are within the
|
||||||
|
# same namespace, otherwise cast to int to avoid triggering warnings about mixing enum types.
|
||||||
|
if prefix:
|
||||||
|
return literal("|".join(f"{prefix}{e.upper()}" for e in enums))
|
||||||
|
return literal("|".join(f"(int){e.upper()}" for e in enums))
|
||||||
|
|
|
@ -8,7 +8,7 @@ from ..defines import CONF_LVGL_ID
|
||||||
from ..lvcode import LvContext
|
from ..lvcode import LvContext
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LvType, lvgl_ns
|
from ..types import LvType, lvgl_ns
|
||||||
from ..widgets import get_widgets
|
from ..widgets import get_widgets, wait_for_widgets
|
||||||
|
|
||||||
lv_led_t = LvType("lv_led_t")
|
lv_led_t = LvType("lv_led_t")
|
||||||
LVLight = lvgl_ns.class_("LVLight", LightOutput)
|
LVLight = lvgl_ns.class_("LVLight", LightOutput)
|
||||||
|
@ -28,5 +28,6 @@ async def to_code(config):
|
||||||
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
||||||
widget = await get_widgets(config, CONF_LED)
|
widget = await get_widgets(config, CONF_LED)
|
||||||
widget = widget[0]
|
widget = widget[0]
|
||||||
|
await wait_for_widgets()
|
||||||
async with LvContext(paren) as ctx:
|
async with LvContext(paren) as ctx:
|
||||||
ctx.add(var.set_obj(widget.obj))
|
ctx.add(var.set_obj(widget.obj))
|
||||||
|
|
|
@ -52,9 +52,7 @@ opacity = LValidator(opacity_validator, uint32, retmapper=literal)
|
||||||
def color(value):
|
def color(value):
|
||||||
if value == SCHEMA_EXTRACT:
|
if value == SCHEMA_EXTRACT:
|
||||||
return ["hex color value", "color ID"]
|
return ["hex color value", "color ID"]
|
||||||
if isinstance(value, int):
|
return cv.Any(cv.int_, cv.use_id(ColorStruct))(value)
|
||||||
return value
|
|
||||||
return cv.use_id(ColorStruct)(value)
|
|
||||||
|
|
||||||
|
|
||||||
def color_retmapper(value):
|
def color_retmapper(value):
|
||||||
|
@ -82,10 +80,10 @@ def pixels_or_percent_validator(value):
|
||||||
"""A length in one axis - either a number (pixels) or a percentage"""
|
"""A length in one axis - either a number (pixels) or a percentage"""
|
||||||
if value == SCHEMA_EXTRACT:
|
if value == SCHEMA_EXTRACT:
|
||||||
return ["pixels", "..%"]
|
return ["pixels", "..%"]
|
||||||
|
value = cv.Any(cv.int_, cv.percentage)(value)
|
||||||
if isinstance(value, int):
|
if isinstance(value, int):
|
||||||
return cv.int_(value)
|
return value
|
||||||
# Will throw an exception if not a percentage.
|
return f"lv_pct({int(value * 100)})"
|
||||||
return f"lv_pct({int(cv.percentage(value) * 100)})"
|
|
||||||
|
|
||||||
|
|
||||||
pixels_or_percent = LValidator(pixels_or_percent_validator, uint32, retmapper=literal)
|
pixels_or_percent = LValidator(pixels_or_percent_validator, uint32, retmapper=literal)
|
||||||
|
@ -116,10 +114,7 @@ def size_validator(value):
|
||||||
if value.upper() == "SIZE_CONTENT":
|
if value.upper() == "SIZE_CONTENT":
|
||||||
return "LV_SIZE_CONTENT"
|
return "LV_SIZE_CONTENT"
|
||||||
raise cv.Invalid("must be 'size_content', a percentage or an integer (pixels)")
|
raise cv.Invalid("must be 'size_content', a percentage or an integer (pixels)")
|
||||||
if isinstance(value, int):
|
return pixels_or_percent_validator(value)
|
||||||
return cv.int_(value)
|
|
||||||
# Will throw an exception if not a percentage.
|
|
||||||
return f"lv_pct({int(cv.percentage(value) * 100)})"
|
|
||||||
|
|
||||||
|
|
||||||
size = LValidator(size_validator, uint32, retmapper=literal)
|
size = LValidator(size_validator, uint32, retmapper=literal)
|
||||||
|
|
|
@ -28,7 +28,7 @@ LVGL_COMP = "lv_component" # used as a lambda argument in lvgl_comp()
|
||||||
LvglComponent = lvgl_ns.class_("LvglComponent", cg.PollingComponent)
|
LvglComponent = lvgl_ns.class_("LvglComponent", cg.PollingComponent)
|
||||||
LVGL_COMP_ARG = [(LvglComponent.operator("ptr"), LVGL_COMP)]
|
LVGL_COMP_ARG = [(LvglComponent.operator("ptr"), LVGL_COMP)]
|
||||||
lv_event_t_ptr = cg.global_ns.namespace("lv_event_t").operator("ptr")
|
lv_event_t_ptr = cg.global_ns.namespace("lv_event_t").operator("ptr")
|
||||||
EVENT_ARG = [(lv_event_t_ptr, "ev")]
|
EVENT_ARG = [(lv_event_t_ptr, "event")]
|
||||||
# Two custom events; API_EVENT is fired when an entity is updated remotely by an API interaction;
|
# Two custom events; API_EVENT is fired when an entity is updated remotely by an API interaction;
|
||||||
# UPDATE_EVENT is fired when an entity is programmatically updated locally.
|
# UPDATE_EVENT is fired when an entity is programmatically updated locally.
|
||||||
# VALUE_CHANGED is the event generated by LVGL when an entity's value changes through user interaction.
|
# VALUE_CHANGED is the event generated by LVGL when an entity's value changes through user interaction.
|
||||||
|
@ -176,6 +176,8 @@ class LvContext(LambdaContext):
|
||||||
Code generation into the LVGL initialisation code (called in `setup()`)
|
Code generation into the LVGL initialisation code (called in `setup()`)
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
added_lambda_count = 0
|
||||||
|
|
||||||
def __init__(self, lv_component, args=None):
|
def __init__(self, lv_component, args=None):
|
||||||
self.args = args or LVGL_COMP_ARG
|
self.args = args or LVGL_COMP_ARG
|
||||||
super().__init__(parameters=self.args)
|
super().__init__(parameters=self.args)
|
||||||
|
@ -183,6 +185,7 @@ class LvContext(LambdaContext):
|
||||||
|
|
||||||
async def add_init_lambda(self):
|
async def add_init_lambda(self):
|
||||||
cg.add(self.lv_component.add_init_lambda(await self.get_lambda()))
|
cg.add(self.lv_component.add_init_lambda(await self.get_lambda()))
|
||||||
|
LvContext.added_lambda_count += 1
|
||||||
|
|
||||||
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||||
await super().__aexit__(exc_type, exc_val, exc_tb)
|
await super().__aexit__(exc_type, exc_val, exc_tb)
|
||||||
|
@ -288,6 +291,10 @@ class LvExpr(MockLv):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def static_cast(type, value):
|
||||||
|
return literal(f"static_cast<{type}>({value})")
|
||||||
|
|
||||||
|
|
||||||
# Top level mock for generic lv_ calls to be recorded
|
# Top level mock for generic lv_ calls to be recorded
|
||||||
lv = MockLv("lv_")
|
lv = MockLv("lv_")
|
||||||
# Just generate an expression
|
# Just generate an expression
|
||||||
|
|
|
@ -15,6 +15,60 @@ static void log_cb(const char *buf) {
|
||||||
}
|
}
|
||||||
#endif // LV_USE_LOG
|
#endif // LV_USE_LOG
|
||||||
|
|
||||||
|
static const char *const EVENT_NAMES[] = {
|
||||||
|
"NONE",
|
||||||
|
"PRESSED",
|
||||||
|
"PRESSING",
|
||||||
|
"PRESS_LOST",
|
||||||
|
"SHORT_CLICKED",
|
||||||
|
"LONG_PRESSED",
|
||||||
|
"LONG_PRESSED_REPEAT",
|
||||||
|
"CLICKED",
|
||||||
|
"RELEASED",
|
||||||
|
"SCROLL_BEGIN",
|
||||||
|
"SCROLL_END",
|
||||||
|
"SCROLL",
|
||||||
|
"GESTURE",
|
||||||
|
"KEY",
|
||||||
|
"FOCUSED",
|
||||||
|
"DEFOCUSED",
|
||||||
|
"LEAVE",
|
||||||
|
"HIT_TEST",
|
||||||
|
"COVER_CHECK",
|
||||||
|
"REFR_EXT_DRAW_SIZE",
|
||||||
|
"DRAW_MAIN_BEGIN",
|
||||||
|
"DRAW_MAIN",
|
||||||
|
"DRAW_MAIN_END",
|
||||||
|
"DRAW_POST_BEGIN",
|
||||||
|
"DRAW_POST",
|
||||||
|
"DRAW_POST_END",
|
||||||
|
"DRAW_PART_BEGIN",
|
||||||
|
"DRAW_PART_END",
|
||||||
|
"VALUE_CHANGED",
|
||||||
|
"INSERT",
|
||||||
|
"REFRESH",
|
||||||
|
"READY",
|
||||||
|
"CANCEL",
|
||||||
|
"DELETE",
|
||||||
|
"CHILD_CHANGED",
|
||||||
|
"CHILD_CREATED",
|
||||||
|
"CHILD_DELETED",
|
||||||
|
"SCREEN_UNLOAD_START",
|
||||||
|
"SCREEN_LOAD_START",
|
||||||
|
"SCREEN_LOADED",
|
||||||
|
"SCREEN_UNLOADED",
|
||||||
|
"SIZE_CHANGED",
|
||||||
|
"STYLE_CHANGED",
|
||||||
|
"LAYOUT_CHANGED",
|
||||||
|
"GET_SELF_SIZE",
|
||||||
|
};
|
||||||
|
|
||||||
|
std::string lv_event_code_name_for(uint8_t event_code) {
|
||||||
|
if (event_code < sizeof(EVENT_NAMES) / sizeof(EVENT_NAMES[0])) {
|
||||||
|
return EVENT_NAMES[event_code];
|
||||||
|
}
|
||||||
|
return str_sprintf("%2d", event_code);
|
||||||
|
}
|
||||||
static void rounder_cb(lv_disp_drv_t *disp_drv, lv_area_t *area) {
|
static void rounder_cb(lv_disp_drv_t *disp_drv, lv_area_t *area) {
|
||||||
// make sure all coordinates are even
|
// make sure all coordinates are even
|
||||||
if (area->x1 & 1)
|
if (area->x1 & 1)
|
||||||
|
@ -294,6 +348,13 @@ void LvglComponent::loop() {
|
||||||
}
|
}
|
||||||
lv_timer_handler_run_in_period(5);
|
lv_timer_handler_run_in_period(5);
|
||||||
}
|
}
|
||||||
|
bool lv_is_pre_initialise() {
|
||||||
|
if (!lv_is_initialized()) {
|
||||||
|
ESP_LOGE(TAG, "LVGL call before component is initialised");
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
#ifdef USE_LVGL_IMAGE
|
#ifdef USE_LVGL_IMAGE
|
||||||
lv_img_dsc_t *lv_img_from(image::Image *src, lv_img_dsc_t *img_dsc) {
|
lv_img_dsc_t *lv_img_from(image::Image *src, lv_img_dsc_t *img_dsc) {
|
||||||
|
|
|
@ -40,6 +40,8 @@ namespace lvgl {
|
||||||
|
|
||||||
extern lv_event_code_t lv_api_event; // NOLINT
|
extern lv_event_code_t lv_api_event; // NOLINT
|
||||||
extern lv_event_code_t lv_update_event; // NOLINT
|
extern lv_event_code_t lv_update_event; // NOLINT
|
||||||
|
extern std::string lv_event_code_name_for(uint8_t event_code);
|
||||||
|
extern bool lv_is_pre_initialise();
|
||||||
#ifdef USE_LVGL_COLOR
|
#ifdef USE_LVGL_COLOR
|
||||||
inline lv_color_t lv_color_from(Color color) { return lv_color_make(color.red, color.green, color.blue); }
|
inline lv_color_t lv_color_from(Color color) { return lv_color_make(color.red, color.green, color.blue); }
|
||||||
#endif // USE_LVGL_COLOR
|
#endif // USE_LVGL_COLOR
|
||||||
|
@ -142,6 +144,13 @@ class LvglComponent : public PollingComponent {
|
||||||
void show_next_page(lv_scr_load_anim_t anim, uint32_t time);
|
void show_next_page(lv_scr_load_anim_t anim, uint32_t time);
|
||||||
void show_prev_page(lv_scr_load_anim_t anim, uint32_t time);
|
void show_prev_page(lv_scr_load_anim_t anim, uint32_t time);
|
||||||
void set_page_wrap(bool wrap) { this->page_wrap_ = wrap; }
|
void set_page_wrap(bool wrap) { this->page_wrap_ = wrap; }
|
||||||
|
void set_focus_mark(lv_group_t *group) { this->focus_marks_[group] = lv_group_get_focused(group); }
|
||||||
|
void restore_focus_mark(lv_group_t *group) {
|
||||||
|
auto *mark = this->focus_marks_[group];
|
||||||
|
if (mark != nullptr) {
|
||||||
|
lv_group_focus_obj(mark);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
void write_random_();
|
void write_random_();
|
||||||
|
@ -157,6 +166,7 @@ class LvglComponent : public PollingComponent {
|
||||||
bool show_snow_{};
|
bool show_snow_{};
|
||||||
lv_coord_t snow_line_{};
|
lv_coord_t snow_line_{};
|
||||||
bool page_wrap_{true};
|
bool page_wrap_{true};
|
||||||
|
std::map<lv_group_t *, lv_obj_t *> focus_marks_{};
|
||||||
|
|
||||||
std::vector<std::function<void(LvglComponent *lv_component)>> init_lambdas_;
|
std::vector<std::function<void(LvglComponent *lv_component)>> init_lambdas_;
|
||||||
CallbackManager<void(uint32_t)> idle_callbacks_{};
|
CallbackManager<void(uint32_t)> idle_callbacks_{};
|
||||||
|
|
|
@ -16,7 +16,7 @@ from ..lvcode import (
|
||||||
)
|
)
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, LvNumber, lvgl_ns
|
from ..types import LV_EVENT, LvNumber, lvgl_ns
|
||||||
from ..widgets import get_widgets
|
from ..widgets import get_widgets, wait_for_widgets
|
||||||
|
|
||||||
LVGLNumber = lvgl_ns.class_("LVGLNumber", number.Number)
|
LVGLNumber = lvgl_ns.class_("LVGLNumber", number.Number)
|
||||||
|
|
||||||
|
@ -44,11 +44,13 @@ async def to_code(config):
|
||||||
step=widget.get_step(),
|
step=widget.get_step(),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext([(cg.float_, "v")]) as control:
|
async with LambdaContext([(cg.float_, "v")]) as control:
|
||||||
await widget.set_property(
|
await widget.set_property(
|
||||||
"value", MockObj("v") * MockObj(widget.get_scale()), config[CONF_ANIMATED]
|
"value", MockObj("v") * MockObj(widget.get_scale()), config[CONF_ANIMATED]
|
||||||
)
|
)
|
||||||
lv.event_send(widget.obj, API_EVENT, cg.nullptr)
|
lv.event_send(widget.obj, API_EVENT, cg.nullptr)
|
||||||
|
control.add(var.publish_state(widget.get_value()))
|
||||||
async with LambdaContext(EVENT_ARG) as event:
|
async with LambdaContext(EVENT_ARG) as event:
|
||||||
event.add(var.publish_state(widget.get_value()))
|
event.add(var.publish_state(widget.get_value()))
|
||||||
event_code = (
|
event_code = (
|
||||||
|
|
|
@ -20,7 +20,7 @@ from . import defines as df, lv_validation as lvalid
|
||||||
from .defines import CONF_TIME_FORMAT
|
from .defines import CONF_TIME_FORMAT
|
||||||
from .helpers import add_lv_use, requires_component, validate_printf
|
from .helpers import add_lv_use, requires_component, validate_printf
|
||||||
from .lv_validation import lv_color, lv_font, lv_image
|
from .lv_validation import lv_color, lv_font, lv_image
|
||||||
from .lvcode import LvglComponent
|
from .lvcode import LvglComponent, lv_event_t_ptr
|
||||||
from .types import (
|
from .types import (
|
||||||
LVEncoderListener,
|
LVEncoderListener,
|
||||||
LvType,
|
LvType,
|
||||||
|
@ -215,14 +215,12 @@ def automation_schema(typ: LvType):
|
||||||
events = df.LV_EVENT_TRIGGERS + (CONF_ON_VALUE,)
|
events = df.LV_EVENT_TRIGGERS + (CONF_ON_VALUE,)
|
||||||
else:
|
else:
|
||||||
events = df.LV_EVENT_TRIGGERS
|
events = df.LV_EVENT_TRIGGERS
|
||||||
if isinstance(typ, LvType):
|
args = [typ.get_arg_type()] if isinstance(typ, LvType) else []
|
||||||
template = Trigger.template(typ.get_arg_type())
|
args.append(lv_event_t_ptr)
|
||||||
else:
|
|
||||||
template = Trigger.template()
|
|
||||||
return {
|
return {
|
||||||
cv.Optional(event): validate_automation(
|
cv.Optional(event): validate_automation(
|
||||||
{
|
{
|
||||||
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(template),
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(Trigger.template(*args)),
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
for event in events
|
for event in events
|
||||||
|
@ -361,7 +359,13 @@ LVGL_SCHEMA = cv.Schema(
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
ALL_STYLES = {**STYLE_PROPS, **GRID_CELL_SCHEMA, **FLEX_OBJ_SCHEMA}
|
ALL_STYLES = {
|
||||||
|
**STYLE_PROPS,
|
||||||
|
**GRID_CELL_SCHEMA,
|
||||||
|
**FLEX_OBJ_SCHEMA,
|
||||||
|
cv.Optional(df.CONF_PAD_ROW): lvalid.pixels,
|
||||||
|
cv.Optional(df.CONF_PAD_COLUMN): lvalid.pixels,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def container_validator(schema, widget_type: WidgetType):
|
def container_validator(schema, widget_type: WidgetType):
|
||||||
|
|
|
@ -15,7 +15,7 @@ from ..lvcode import (
|
||||||
)
|
)
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, LvSelect, lvgl_ns
|
from ..types import LV_EVENT, LvSelect, lvgl_ns
|
||||||
from ..widgets import get_widgets
|
from ..widgets import get_widgets, wait_for_widgets
|
||||||
|
|
||||||
LVGLSelect = lvgl_ns.class_("LVGLSelect", select.Select)
|
LVGLSelect = lvgl_ns.class_("LVGLSelect", select.Select)
|
||||||
|
|
||||||
|
@ -37,11 +37,13 @@ async def to_code(config):
|
||||||
options = widget.config.get(CONF_OPTIONS, [])
|
options = widget.config.get(CONF_OPTIONS, [])
|
||||||
selector = await select.new_select(config, options=options)
|
selector = await select.new_select(config, options=options)
|
||||||
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext(EVENT_ARG) as pub_ctx:
|
async with LambdaContext(EVENT_ARG) as pub_ctx:
|
||||||
pub_ctx.add(selector.publish_index(widget.get_value()))
|
pub_ctx.add(selector.publish_index(widget.get_value()))
|
||||||
async with LambdaContext([(cg.uint16, "v")]) as control:
|
async with LambdaContext([(cg.uint16, "v")]) as control:
|
||||||
await widget.set_property("selected", "v", animated=config[CONF_ANIMATED])
|
await widget.set_property("selected", "v", animated=config[CONF_ANIMATED])
|
||||||
lv.event_send(widget.obj, API_EVENT, cg.nullptr)
|
lv.event_send(widget.obj, API_EVENT, cg.nullptr)
|
||||||
|
control.add(selector.publish_index(widget.get_value()))
|
||||||
async with LvContext(paren) as ctx:
|
async with LvContext(paren) as ctx:
|
||||||
lv_add(selector.set_control_lambda(await control.get_lambda()))
|
lv_add(selector.set_control_lambda(await control.get_lambda()))
|
||||||
ctx.add(
|
ctx.add(
|
||||||
|
|
|
@ -14,7 +14,7 @@ from ..lvcode import (
|
||||||
)
|
)
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, LvNumber
|
from ..types import LV_EVENT, LvNumber
|
||||||
from ..widgets import Widget, get_widgets
|
from ..widgets import Widget, get_widgets, wait_for_widgets
|
||||||
|
|
||||||
CONFIG_SCHEMA = (
|
CONFIG_SCHEMA = (
|
||||||
sensor_schema(Sensor)
|
sensor_schema(Sensor)
|
||||||
|
@ -33,6 +33,7 @@ async def to_code(config):
|
||||||
widget = await get_widgets(config, CONF_WIDGET)
|
widget = await get_widgets(config, CONF_WIDGET)
|
||||||
widget = widget[0]
|
widget = widget[0]
|
||||||
assert isinstance(widget, Widget)
|
assert isinstance(widget, Widget)
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext(EVENT_ARG) as lamb:
|
async with LambdaContext(EVENT_ARG) as lamb:
|
||||||
lv_add(sensor.publish_state(widget.get_value()))
|
lv_add(sensor.publish_state(widget.get_value()))
|
||||||
async with LvContext(paren, LVGL_COMP_ARG):
|
async with LvContext(paren, LVGL_COMP_ARG):
|
||||||
|
|
|
@ -3,7 +3,7 @@ from esphome.components.switch import Switch, new_switch, switch_schema
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
from esphome.cpp_generator import MockObj
|
from esphome.cpp_generator import MockObj
|
||||||
|
|
||||||
from ..defines import CONF_LVGL_ID, CONF_WIDGET
|
from ..defines import CONF_LVGL_ID, CONF_WIDGET, literal
|
||||||
from ..lvcode import (
|
from ..lvcode import (
|
||||||
API_EVENT,
|
API_EVENT,
|
||||||
EVENT_ARG,
|
EVENT_ARG,
|
||||||
|
@ -16,7 +16,7 @@ from ..lvcode import (
|
||||||
)
|
)
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, LV_STATE, lv_pseudo_button_t, lvgl_ns
|
from ..types import LV_EVENT, LV_STATE, lv_pseudo_button_t, lvgl_ns
|
||||||
from ..widgets import get_widgets
|
from ..widgets import get_widgets, wait_for_widgets
|
||||||
|
|
||||||
LVGLSwitch = lvgl_ns.class_("LVGLSwitch", Switch)
|
LVGLSwitch = lvgl_ns.class_("LVGLSwitch", Switch)
|
||||||
CONFIG_SCHEMA = (
|
CONFIG_SCHEMA = (
|
||||||
|
@ -35,6 +35,7 @@ async def to_code(config):
|
||||||
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
||||||
widget = await get_widgets(config, CONF_WIDGET)
|
widget = await get_widgets(config, CONF_WIDGET)
|
||||||
widget = widget[0]
|
widget = widget[0]
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext(EVENT_ARG) as checked_ctx:
|
async with LambdaContext(EVENT_ARG) as checked_ctx:
|
||||||
checked_ctx.add(switch.publish_state(widget.get_value()))
|
checked_ctx.add(switch.publish_state(widget.get_value()))
|
||||||
async with LambdaContext([(cg.bool_, "v")]) as control:
|
async with LambdaContext([(cg.bool_, "v")]) as control:
|
||||||
|
@ -43,6 +44,7 @@ async def to_code(config):
|
||||||
cond.else_()
|
cond.else_()
|
||||||
widget.clear_state(LV_STATE.CHECKED)
|
widget.clear_state(LV_STATE.CHECKED)
|
||||||
lv.event_send(widget.obj, API_EVENT, cg.nullptr)
|
lv.event_send(widget.obj, API_EVENT, cg.nullptr)
|
||||||
|
control.add(switch.publish_state(literal("v")))
|
||||||
async with LvContext(paren) as ctx:
|
async with LvContext(paren) as ctx:
|
||||||
lv_add(switch.set_control_lambda(await control.get_lambda()))
|
lv_add(switch.set_control_lambda(await control.get_lambda()))
|
||||||
ctx.add(
|
ctx.add(
|
||||||
|
|
|
@ -15,7 +15,7 @@ from ..lvcode import (
|
||||||
)
|
)
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, LvText, lvgl_ns
|
from ..types import LV_EVENT, LvText, lvgl_ns
|
||||||
from ..widgets import get_widgets
|
from ..widgets import get_widgets, wait_for_widgets
|
||||||
|
|
||||||
LVGLText = lvgl_ns.class_("LVGLText", text.Text)
|
LVGLText = lvgl_ns.class_("LVGLText", text.Text)
|
||||||
|
|
||||||
|
@ -32,9 +32,11 @@ async def to_code(config):
|
||||||
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
||||||
widget = await get_widgets(config, CONF_WIDGET)
|
widget = await get_widgets(config, CONF_WIDGET)
|
||||||
widget = widget[0]
|
widget = widget[0]
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext([(cg.std_string, "text_value")]) as control:
|
async with LambdaContext([(cg.std_string, "text_value")]) as control:
|
||||||
await widget.set_property("text", "text_value.c_str())")
|
await widget.set_property("text", "text_value.c_str())")
|
||||||
lv.event_send(widget.obj, API_EVENT, None)
|
lv.event_send(widget.obj, API_EVENT, None)
|
||||||
|
control.add(textvar.publish_state(widget.get_value()))
|
||||||
async with LambdaContext(EVENT_ARG) as lamb:
|
async with LambdaContext(EVENT_ARG) as lamb:
|
||||||
lv_add(textvar.publish_state(widget.get_value()))
|
lv_add(textvar.publish_state(widget.get_value()))
|
||||||
async with LvContext(paren):
|
async with LvContext(paren):
|
||||||
|
|
|
@ -10,7 +10,7 @@ from ..defines import CONF_LVGL_ID, CONF_WIDGET
|
||||||
from ..lvcode import API_EVENT, EVENT_ARG, UPDATE_EVENT, LambdaContext, LvContext
|
from ..lvcode import API_EVENT, EVENT_ARG, UPDATE_EVENT, LambdaContext, LvContext
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LV_EVENT, LvText
|
from ..types import LV_EVENT, LvText
|
||||||
from ..widgets import get_widgets
|
from ..widgets import get_widgets, wait_for_widgets
|
||||||
|
|
||||||
CONFIG_SCHEMA = (
|
CONFIG_SCHEMA = (
|
||||||
text_sensor_schema(TextSensor)
|
text_sensor_schema(TextSensor)
|
||||||
|
@ -28,6 +28,7 @@ async def to_code(config):
|
||||||
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
paren = await cg.get_variable(config[CONF_LVGL_ID])
|
||||||
widget = await get_widgets(config, CONF_WIDGET)
|
widget = await get_widgets(config, CONF_WIDGET)
|
||||||
widget = widget[0]
|
widget = widget[0]
|
||||||
|
await wait_for_widgets()
|
||||||
async with LambdaContext(EVENT_ARG) as pressed_ctx:
|
async with LambdaContext(EVENT_ARG) as pressed_ctx:
|
||||||
pressed_ctx.add(sensor.publish_state(widget.get_value()))
|
pressed_ctx.add(sensor.publish_state(widget.get_value()))
|
||||||
async with LvContext(paren) as ctx:
|
async with LvContext(paren) as ctx:
|
||||||
|
|
|
@ -19,6 +19,7 @@ from .lvcode import (
|
||||||
LvConditional,
|
LvConditional,
|
||||||
lv,
|
lv,
|
||||||
lv_add,
|
lv_add,
|
||||||
|
lv_event_t_ptr,
|
||||||
)
|
)
|
||||||
from .types import LV_EVENT
|
from .types import LV_EVENT
|
||||||
from .widgets import widget_map
|
from .widgets import widget_map
|
||||||
|
@ -65,10 +66,10 @@ async def generate_triggers(lv_component):
|
||||||
async def add_trigger(conf, lv_component, w, *events):
|
async def add_trigger(conf, lv_component, w, *events):
|
||||||
tid = conf[CONF_TRIGGER_ID]
|
tid = conf[CONF_TRIGGER_ID]
|
||||||
trigger = cg.new_Pvariable(tid)
|
trigger = cg.new_Pvariable(tid)
|
||||||
args = w.get_args()
|
args = w.get_args() + [(lv_event_t_ptr, "event")]
|
||||||
value = w.get_value()
|
value = w.get_value()
|
||||||
await automation.build_automation(trigger, args, conf)
|
await automation.build_automation(trigger, args, conf)
|
||||||
async with LambdaContext(EVENT_ARG, where=tid) as context:
|
async with LambdaContext(EVENT_ARG, where=tid) as context:
|
||||||
with LvConditional(w.is_selected()):
|
with LvConditional(w.is_selected()):
|
||||||
lv_add(trigger.trigger(value))
|
lv_add(trigger.trigger(value, literal("event")))
|
||||||
lv_add(lv_component.add_event_cb(w.obj, await context.get_lambda(), *events))
|
lv_add(lv_component.add_event_cb(w.obj, await context.get_lambda(), *events))
|
||||||
|
|
|
@ -57,7 +57,7 @@ lv_group_t = cg.global_ns.struct("lv_group_t")
|
||||||
LVTouchListener = lvgl_ns.class_("LVTouchListener")
|
LVTouchListener = lvgl_ns.class_("LVTouchListener")
|
||||||
LVEncoderListener = lvgl_ns.class_("LVEncoderListener")
|
LVEncoderListener = lvgl_ns.class_("LVEncoderListener")
|
||||||
lv_obj_t = LvType("lv_obj_t")
|
lv_obj_t = LvType("lv_obj_t")
|
||||||
lv_page_t = cg.global_ns.class_("LvPageType", LvCompound)
|
lv_page_t = LvType("LvPageType", parents=(LvCompound,))
|
||||||
lv_img_t = LvType("lv_img_t")
|
lv_img_t = LvType("lv_img_t")
|
||||||
|
|
||||||
LV_EVENT = MockObj(base="LV_EVENT_", op="")
|
LV_EVENT = MockObj(base="LV_EVENT_", op="")
|
||||||
|
|
|
@ -118,7 +118,14 @@ class Widget:
|
||||||
def clear_flag(self, flag):
|
def clear_flag(self, flag):
|
||||||
return lv_obj.clear_flag(self.obj, literal(flag))
|
return lv_obj.clear_flag(self.obj, literal(flag))
|
||||||
|
|
||||||
async def set_property(self, prop, value, animated: bool = None):
|
async def set_property(self, prop, value, animated: bool = None, lv_name=None):
|
||||||
|
"""
|
||||||
|
Set a property of the widget.
|
||||||
|
:param prop: The property name
|
||||||
|
:param value: The value
|
||||||
|
:param animated: If the change should be animated
|
||||||
|
:param lv_name: The base type of the widget e.g. "obj"
|
||||||
|
"""
|
||||||
if isinstance(value, dict):
|
if isinstance(value, dict):
|
||||||
value = value.get(prop)
|
value = value.get(prop)
|
||||||
if isinstance(ALL_STYLES.get(prop), LValidator):
|
if isinstance(ALL_STYLES.get(prop), LValidator):
|
||||||
|
@ -131,11 +138,12 @@ class Widget:
|
||||||
value = value.total_milliseconds
|
value = value.total_milliseconds
|
||||||
if isinstance(value, str):
|
if isinstance(value, str):
|
||||||
value = literal(value)
|
value = literal(value)
|
||||||
|
lv_name = lv_name or self.type.lv_name
|
||||||
if animated is None or self.type.animated is not True:
|
if animated is None or self.type.animated is not True:
|
||||||
lv.call(f"{self.type.lv_name}_set_{prop}", self.obj, value)
|
lv.call(f"{lv_name}_set_{prop}", self.obj, value)
|
||||||
else:
|
else:
|
||||||
lv.call(
|
lv.call(
|
||||||
f"{self.type.lv_name}_set_{prop}",
|
f"{lv_name}_set_{prop}",
|
||||||
self.obj,
|
self.obj,
|
||||||
value,
|
value,
|
||||||
literal("LV_ANIM_ON" if animated else "LV_ANIM_OFF"),
|
literal("LV_ANIM_ON" if animated else "LV_ANIM_OFF"),
|
||||||
|
@ -217,12 +225,25 @@ def get_widget_generator(wid):
|
||||||
yield
|
yield
|
||||||
|
|
||||||
|
|
||||||
async def get_widget_(wid: Widget):
|
async def get_widget_(wid):
|
||||||
if obj := widget_map.get(wid):
|
if obj := widget_map.get(wid):
|
||||||
return obj
|
return obj
|
||||||
return await FakeAwaitable(get_widget_generator(wid))
|
return await FakeAwaitable(get_widget_generator(wid))
|
||||||
|
|
||||||
|
|
||||||
|
def widgets_wait_generator():
|
||||||
|
while True:
|
||||||
|
if Widget.widgets_completed:
|
||||||
|
return
|
||||||
|
yield
|
||||||
|
|
||||||
|
|
||||||
|
async def wait_for_widgets():
|
||||||
|
if Widget.widgets_completed:
|
||||||
|
return
|
||||||
|
await FakeAwaitable(widgets_wait_generator())
|
||||||
|
|
||||||
|
|
||||||
async def get_widgets(config: Union[dict, list], id: str = CONF_ID) -> list[Widget]:
|
async def get_widgets(config: Union[dict, list], id: str = CONF_ID) -> list[Widget]:
|
||||||
if not config:
|
if not config:
|
||||||
return []
|
return []
|
||||||
|
@ -306,8 +327,15 @@ async def set_obj_properties(w: Widget, config):
|
||||||
lv_obj.set_flex_align(w.obj, main, cross, track)
|
lv_obj.set_flex_align(w.obj, main, cross, track)
|
||||||
parts = collect_parts(config)
|
parts = collect_parts(config)
|
||||||
for part, states in parts.items():
|
for part, states in parts.items():
|
||||||
|
part = "LV_PART_" + part.upper()
|
||||||
for state, props in states.items():
|
for state, props in states.items():
|
||||||
lv_state = join_enums((f"LV_STATE_{state}", f"LV_PART_{part}"))
|
state = "LV_STATE_" + state.upper()
|
||||||
|
if state == "LV_STATE_DEFAULT":
|
||||||
|
lv_state = literal(part)
|
||||||
|
elif part == "LV_PART_MAIN":
|
||||||
|
lv_state = literal(state)
|
||||||
|
else:
|
||||||
|
lv_state = join_enums((state, part))
|
||||||
for style_id in props.get(CONF_STYLES, ()):
|
for style_id in props.get(CONF_STYLES, ()):
|
||||||
lv_obj.add_style(w.obj, MockObj(style_id), lv_state)
|
lv_obj.add_style(w.obj, MockObj(style_id), lv_state)
|
||||||
for prop, value in {
|
for prop, value in {
|
||||||
|
@ -320,8 +348,6 @@ async def set_obj_properties(w: Widget, config):
|
||||||
if group := config.get(CONF_GROUP):
|
if group := config.get(CONF_GROUP):
|
||||||
group = await cg.get_variable(group)
|
group = await cg.get_variable(group)
|
||||||
lv.group_add_obj(group, w.obj)
|
lv.group_add_obj(group, w.obj)
|
||||||
flag_clr = set()
|
|
||||||
flag_set = set()
|
|
||||||
props = parts[CONF_MAIN][CONF_DEFAULT]
|
props = parts[CONF_MAIN][CONF_DEFAULT]
|
||||||
lambs = {}
|
lambs = {}
|
||||||
flag_set = set()
|
flag_set = set()
|
||||||
|
@ -371,7 +397,7 @@ async def set_obj_properties(w: Widget, config):
|
||||||
w.add_state(state)
|
w.add_state(state)
|
||||||
cond.else_()
|
cond.else_()
|
||||||
w.clear_state(state)
|
w.clear_state(state)
|
||||||
await w.set_property(CONF_SCROLLBAR_MODE, config)
|
await w.set_property(CONF_SCROLLBAR_MODE, config, lv_name="obj")
|
||||||
|
|
||||||
|
|
||||||
async def add_widgets(parent: Widget, config: dict):
|
async def add_widgets(parent: Widget, config: dict):
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
|
CONF_GROUP,
|
||||||
CONF_MAX_VALUE,
|
CONF_MAX_VALUE,
|
||||||
CONF_MIN_VALUE,
|
CONF_MIN_VALUE,
|
||||||
CONF_MODE,
|
CONF_MODE,
|
||||||
|
@ -20,7 +21,7 @@ from ..defines import (
|
||||||
literal,
|
literal,
|
||||||
)
|
)
|
||||||
from ..lv_validation import angle, get_start_value, lv_float
|
from ..lv_validation import angle, get_start_value, lv_float
|
||||||
from ..lvcode import lv, lv_obj
|
from ..lvcode import lv, lv_expr, lv_obj
|
||||||
from ..types import LvNumber, NumberType
|
from ..types import LvNumber, NumberType
|
||||||
from . import Widget
|
from . import Widget
|
||||||
|
|
||||||
|
@ -69,6 +70,9 @@ class ArcType(NumberType):
|
||||||
if config.get(CONF_ADJUSTABLE) is False:
|
if config.get(CONF_ADJUSTABLE) is False:
|
||||||
lv_obj.remove_style(w.obj, nullptr, literal("LV_PART_KNOB"))
|
lv_obj.remove_style(w.obj, nullptr, literal("LV_PART_KNOB"))
|
||||||
w.clear_flag("LV_OBJ_FLAG_CLICKABLE")
|
w.clear_flag("LV_OBJ_FLAG_CLICKABLE")
|
||||||
|
elif CONF_GROUP not in config:
|
||||||
|
# For some reason arc does not get automatically added to the default group
|
||||||
|
lv.group_add_obj(lv_expr.group_get_default(), w.obj)
|
||||||
|
|
||||||
value = await get_start_value(config)
|
value = await get_start_value(config)
|
||||||
if value is not None:
|
if value is not None:
|
||||||
|
|
|
@ -13,11 +13,13 @@ from ..defines import (
|
||||||
CONF_KEY_CODE,
|
CONF_KEY_CODE,
|
||||||
CONF_MAIN,
|
CONF_MAIN,
|
||||||
CONF_ONE_CHECKED,
|
CONF_ONE_CHECKED,
|
||||||
|
CONF_PAD_COLUMN,
|
||||||
|
CONF_PAD_ROW,
|
||||||
CONF_ROWS,
|
CONF_ROWS,
|
||||||
CONF_SELECTED,
|
CONF_SELECTED,
|
||||||
)
|
)
|
||||||
from ..helpers import lvgl_components_required
|
from ..helpers import lvgl_components_required
|
||||||
from ..lv_validation import key_code, lv_bool
|
from ..lv_validation import key_code, lv_bool, pixels
|
||||||
from ..lvcode import lv, lv_add, lv_expr
|
from ..lvcode import lv, lv_add, lv_expr
|
||||||
from ..schemas import automation_schema
|
from ..schemas import automation_schema
|
||||||
from ..types import (
|
from ..types import (
|
||||||
|
@ -57,6 +59,8 @@ BUTTONMATRIX_BUTTON_SCHEMA = cv.Schema(
|
||||||
BUTTONMATRIX_SCHEMA = cv.Schema(
|
BUTTONMATRIX_SCHEMA = cv.Schema(
|
||||||
{
|
{
|
||||||
cv.Optional(CONF_ONE_CHECKED, default=False): lv_bool,
|
cv.Optional(CONF_ONE_CHECKED, default=False): lv_bool,
|
||||||
|
cv.Optional(CONF_PAD_ROW): pixels,
|
||||||
|
cv.Optional(CONF_PAD_COLUMN): pixels,
|
||||||
cv.GenerateID(CONF_BUTTON_TEXT_LIST_ID): cv.declare_id(char_ptr),
|
cv.GenerateID(CONF_BUTTON_TEXT_LIST_ID): cv.declare_id(char_ptr),
|
||||||
cv.Required(CONF_ROWS): cv.ensure_list(
|
cv.Required(CONF_ROWS): cv.ensure_list(
|
||||||
cv.Schema(
|
cv.Schema(
|
||||||
|
|
|
@ -1,7 +1,8 @@
|
||||||
|
from esphome.config_validation import Optional
|
||||||
from esphome.const import CONF_TEXT
|
from esphome.const import CONF_TEXT
|
||||||
|
|
||||||
from ..defines import CONF_INDICATOR, CONF_MAIN
|
from ..defines import CONF_INDICATOR, CONF_MAIN, CONF_PAD_COLUMN
|
||||||
from ..lv_validation import lv_text
|
from ..lv_validation import lv_text, pixels
|
||||||
from ..lvcode import lv
|
from ..lvcode import lv
|
||||||
from ..schemas import TEXT_SCHEMA
|
from ..schemas import TEXT_SCHEMA
|
||||||
from ..types import LvBoolean
|
from ..types import LvBoolean
|
||||||
|
@ -16,7 +17,11 @@ class CheckboxType(WidgetType):
|
||||||
CONF_CHECKBOX,
|
CONF_CHECKBOX,
|
||||||
LvBoolean("lv_checkbox_t"),
|
LvBoolean("lv_checkbox_t"),
|
||||||
(CONF_MAIN, CONF_INDICATOR),
|
(CONF_MAIN, CONF_INDICATOR),
|
||||||
TEXT_SCHEMA,
|
TEXT_SCHEMA.extend(
|
||||||
|
{
|
||||||
|
Optional(CONF_PAD_COLUMN): pixels,
|
||||||
|
}
|
||||||
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
async def to_code(self, w: Widget, config):
|
async def to_code(self, w: Widget, config):
|
||||||
|
|
|
@ -3,7 +3,7 @@ import functools
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
|
|
||||||
from ..defines import CONF_MAIN, literal
|
from ..defines import CONF_MAIN
|
||||||
from ..lvcode import lv
|
from ..lvcode import lv
|
||||||
from ..types import LvType
|
from ..types import LvType
|
||||||
from . import Widget, WidgetType
|
from . import Widget, WidgetType
|
||||||
|
@ -38,13 +38,15 @@ LINE_SCHEMA = {
|
||||||
|
|
||||||
class LineType(WidgetType):
|
class LineType(WidgetType):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super().__init__(CONF_LINE, LvType("lv_line_t"), (CONF_MAIN,), LINE_SCHEMA)
|
super().__init__(
|
||||||
|
CONF_LINE, LvType("lv_line_t"), (CONF_MAIN,), LINE_SCHEMA, modify_schema={}
|
||||||
|
)
|
||||||
|
|
||||||
async def to_code(self, w: Widget, config):
|
async def to_code(self, w: Widget, config):
|
||||||
"""For a line object, create and add the points"""
|
"""For a line object, create and add the points"""
|
||||||
data = literal(config[CONF_POINTS])
|
if data := config.get(CONF_POINTS):
|
||||||
points = cg.static_const_array(config[CONF_POINT_LIST_ID], data)
|
points = cg.static_const_array(config[CONF_POINT_LIST_ID], data)
|
||||||
lv.line_set_points(w.obj, points, len(data))
|
lv.line_set_points(w.obj, points, len(data))
|
||||||
|
|
||||||
|
|
||||||
line_spec = LineType()
|
line_spec = LineType()
|
||||||
|
|
|
@ -13,7 +13,7 @@ from ..defines import (
|
||||||
TYPE_FLEX,
|
TYPE_FLEX,
|
||||||
literal,
|
literal,
|
||||||
)
|
)
|
||||||
from ..helpers import add_lv_use
|
from ..helpers import add_lv_use, lvgl_components_required
|
||||||
from ..lv_validation import lv_bool, lv_pct, lv_text
|
from ..lv_validation import lv_bool, lv_pct, lv_text
|
||||||
from ..lvcode import (
|
from ..lvcode import (
|
||||||
EVENT_ARG,
|
EVENT_ARG,
|
||||||
|
@ -72,6 +72,7 @@ async def msgbox_to_code(conf):
|
||||||
*buttonmatrix_spec.get_uses(),
|
*buttonmatrix_spec.get_uses(),
|
||||||
*button_spec.get_uses(),
|
*button_spec.get_uses(),
|
||||||
)
|
)
|
||||||
|
lvgl_components_required.add("BUTTONMATRIX")
|
||||||
messagebox_id = conf[CONF_ID]
|
messagebox_id = conf[CONF_ID]
|
||||||
outer = lv_Pvariable(lv_obj_t, messagebox_id.id)
|
outer = lv_Pvariable(lv_obj_t, messagebox_id.id)
|
||||||
buttonmatrix = new_Pvariable(
|
buttonmatrix = new_Pvariable(
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
from esphome import automation, codegen as cg
|
from esphome import automation, codegen as cg
|
||||||
|
from esphome.automation import Trigger
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
from esphome.const import CONF_ID, CONF_PAGES, CONF_TIME
|
from esphome.const import CONF_ID, CONF_PAGES, CONF_TIME, CONF_TRIGGER_ID
|
||||||
|
|
||||||
from ..defines import (
|
from ..defines import (
|
||||||
CONF_ANIMATION,
|
CONF_ANIMATION,
|
||||||
|
@ -9,12 +10,39 @@ from ..defines import (
|
||||||
CONF_PAGE_WRAP,
|
CONF_PAGE_WRAP,
|
||||||
CONF_SKIP,
|
CONF_SKIP,
|
||||||
LV_ANIM,
|
LV_ANIM,
|
||||||
|
literal,
|
||||||
)
|
)
|
||||||
from ..lv_validation import lv_bool, lv_milliseconds
|
from ..lv_validation import lv_bool, lv_milliseconds
|
||||||
from ..lvcode import LVGL_COMP_ARG, LambdaContext, add_line_marks, lv_add, lvgl_comp
|
from ..lvcode import (
|
||||||
|
EVENT_ARG,
|
||||||
|
LVGL_COMP_ARG,
|
||||||
|
LambdaContext,
|
||||||
|
add_line_marks,
|
||||||
|
lv_add,
|
||||||
|
lvgl_comp,
|
||||||
|
)
|
||||||
from ..schemas import LVGL_SCHEMA
|
from ..schemas import LVGL_SCHEMA
|
||||||
from ..types import LvglAction, lv_page_t
|
from ..types import LvglAction, lv_page_t
|
||||||
from . import Widget, WidgetType, add_widgets, set_obj_properties
|
from . import Widget, WidgetType, add_widgets, get_widgets, set_obj_properties
|
||||||
|
|
||||||
|
CONF_ON_LOAD = "on_load"
|
||||||
|
CONF_ON_UNLOAD = "on_unload"
|
||||||
|
|
||||||
|
PAGE_SCHEMA = cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Optional(CONF_SKIP, default=False): lv_bool,
|
||||||
|
cv.Optional(CONF_ON_LOAD): automation.validate_automation(
|
||||||
|
{
|
||||||
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(Trigger.template()),
|
||||||
|
}
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_ON_UNLOAD): automation.validate_automation(
|
||||||
|
{
|
||||||
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(Trigger.template()),
|
||||||
|
}
|
||||||
|
),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class PageType(WidgetType):
|
class PageType(WidgetType):
|
||||||
|
@ -23,9 +51,8 @@ class PageType(WidgetType):
|
||||||
CONF_PAGE,
|
CONF_PAGE,
|
||||||
lv_page_t,
|
lv_page_t,
|
||||||
(),
|
(),
|
||||||
{
|
PAGE_SCHEMA,
|
||||||
cv.Optional(CONF_SKIP, default=False): lv_bool,
|
modify_schema={},
|
||||||
},
|
|
||||||
)
|
)
|
||||||
|
|
||||||
async def to_code(self, w: Widget, config: dict):
|
async def to_code(self, w: Widget, config: dict):
|
||||||
|
@ -39,7 +66,6 @@ SHOW_SCHEMA = LVGL_SCHEMA.extend(
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
page_spec = PageType()
|
page_spec = PageType()
|
||||||
|
|
||||||
|
|
||||||
|
@ -111,3 +137,21 @@ async def add_pages(lv_component, config):
|
||||||
await set_obj_properties(page, config)
|
await set_obj_properties(page, config)
|
||||||
await set_obj_properties(page, pconf)
|
await set_obj_properties(page, pconf)
|
||||||
await add_widgets(page, pconf)
|
await add_widgets(page, pconf)
|
||||||
|
|
||||||
|
|
||||||
|
async def generate_page_triggers(lv_component, config):
|
||||||
|
for pconf in config.get(CONF_PAGES, ()):
|
||||||
|
page = (await get_widgets(pconf))[0]
|
||||||
|
for ev in (CONF_ON_LOAD, CONF_ON_UNLOAD):
|
||||||
|
for loaded in pconf.get(ev, ()):
|
||||||
|
trigger = cg.new_Pvariable(loaded[CONF_TRIGGER_ID])
|
||||||
|
await automation.build_automation(trigger, [], loaded)
|
||||||
|
async with LambdaContext(EVENT_ARG, where=id) as context:
|
||||||
|
lv_add(trigger.trigger())
|
||||||
|
lv_add(
|
||||||
|
lv_component.add_event_cb(
|
||||||
|
page.obj,
|
||||||
|
await context.get_lambda(),
|
||||||
|
literal(f"LV_EVENT_SCREEN_{ev[3:].upper()}_START"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
from esphome import automation
|
from esphome import automation
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
import esphome.config_validation as cv
|
||||||
from esphome.const import CONF_ID, CONF_ON_VALUE, CONF_ROW, CONF_TRIGGER_ID
|
from esphome.const import CONF_ID, CONF_ROW
|
||||||
|
|
||||||
from ..automation import action_to_code
|
from ..automation import action_to_code
|
||||||
from ..defines import (
|
from ..defines import (
|
||||||
|
@ -29,6 +29,7 @@ lv_tileview_t = LvType(
|
||||||
"lv_tileview_t",
|
"lv_tileview_t",
|
||||||
largs=[(lv_obj_t_ptr, "tile")],
|
largs=[(lv_obj_t_ptr, "tile")],
|
||||||
lvalue=lambda w: w.get_property("tile_act"),
|
lvalue=lambda w: w.get_property("tile_act"),
|
||||||
|
has_on_value=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
tile_spec = WidgetType("lv_tileview_tile_t", lv_tile_t, (CONF_MAIN,), {})
|
tile_spec = WidgetType("lv_tileview_tile_t", lv_tile_t, (CONF_MAIN,), {})
|
||||||
|
@ -46,13 +47,6 @@ TILEVIEW_SCHEMA = cv.Schema(
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
cv.Optional(CONF_ON_VALUE): automation.validate_automation(
|
|
||||||
{
|
|
||||||
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(
|
|
||||||
automation.Trigger.template(lv_obj_t_ptr)
|
|
||||||
)
|
|
||||||
}
|
|
||||||
),
|
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -27,6 +27,18 @@ enum MediaPlayerCommand : uint8_t {
|
||||||
};
|
};
|
||||||
const char *media_player_command_to_string(MediaPlayerCommand command);
|
const char *media_player_command_to_string(MediaPlayerCommand command);
|
||||||
|
|
||||||
|
enum class MediaPlayerFormatPurpose : uint8_t {
|
||||||
|
PURPOSE_DEFAULT = 0,
|
||||||
|
PURPOSE_ANNOUNCEMENT = 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
struct MediaPlayerSupportedFormat {
|
||||||
|
std::string format;
|
||||||
|
uint32_t sample_rate;
|
||||||
|
uint32_t num_channels;
|
||||||
|
MediaPlayerFormatPurpose purpose;
|
||||||
|
};
|
||||||
|
|
||||||
class MediaPlayer;
|
class MediaPlayer;
|
||||||
|
|
||||||
class MediaPlayerTraits {
|
class MediaPlayerTraits {
|
||||||
|
@ -37,8 +49,11 @@ class MediaPlayerTraits {
|
||||||
|
|
||||||
bool get_supports_pause() const { return this->supports_pause_; }
|
bool get_supports_pause() const { return this->supports_pause_; }
|
||||||
|
|
||||||
|
std::vector<MediaPlayerSupportedFormat> &get_supported_formats() { return this->supported_formats_; }
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
bool supports_pause_{false};
|
bool supports_pause_{false};
|
||||||
|
std::vector<MediaPlayerSupportedFormat> supported_formats_{};
|
||||||
};
|
};
|
||||||
|
|
||||||
class MediaPlayerCall {
|
class MediaPlayerCall {
|
||||||
|
|
|
@ -70,72 +70,62 @@ void MICS4514Component::update() {
|
||||||
|
|
||||||
if (this->carbon_monoxide_sensor_ != nullptr) {
|
if (this->carbon_monoxide_sensor_ != nullptr) {
|
||||||
float co = 0.0f;
|
float co = 0.0f;
|
||||||
if (red_f <= 0.425f) {
|
if (red_f > 3.4f) {
|
||||||
co = (0.425f - red_f) / 0.000405f;
|
co = 0.0;
|
||||||
if (co < 1.0f)
|
} else if (red_f < 0.01) {
|
||||||
co = 0.0f;
|
co = 1000.0;
|
||||||
if (co > 1000.0f)
|
} else {
|
||||||
co = 1000.0f;
|
co = 4.2 / pow(red_f, 1.2);
|
||||||
}
|
}
|
||||||
this->carbon_monoxide_sensor_->publish_state(co);
|
this->carbon_monoxide_sensor_->publish_state(co);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this->nitrogen_dioxide_sensor_ != nullptr) {
|
if (this->nitrogen_dioxide_sensor_ != nullptr) {
|
||||||
float nitrogendioxide = 0.0f;
|
float nitrogendioxide = 0.0f;
|
||||||
if (ox_f >= 1.1f) {
|
if (ox_f < 0.3f) {
|
||||||
nitrogendioxide = (ox_f - 0.045f) / 6.13f;
|
nitrogendioxide = 0.0;
|
||||||
if (nitrogendioxide < 0.1f)
|
} else {
|
||||||
nitrogendioxide = 0.0f;
|
nitrogendioxide = 0.164 * pow(ox_f, 0.975);
|
||||||
if (nitrogendioxide > 10.0f)
|
|
||||||
nitrogendioxide = 10.0f;
|
|
||||||
}
|
}
|
||||||
this->nitrogen_dioxide_sensor_->publish_state(nitrogendioxide);
|
this->nitrogen_dioxide_sensor_->publish_state(nitrogendioxide);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this->methane_sensor_ != nullptr) {
|
if (this->methane_sensor_ != nullptr) {
|
||||||
float methane = 0.0f;
|
float methane = 0.0f;
|
||||||
if (red_f <= 0.786f) {
|
if (red_f > 0.9f || red_f < 0.5) { // outside the range->unlikely
|
||||||
methane = (0.786f - red_f) / 0.000023f;
|
methane = 0.0;
|
||||||
if (methane < 1000.0f)
|
} else {
|
||||||
methane = 0.0f;
|
methane = 630 / pow(red_f, 4.4);
|
||||||
if (methane > 25000.0f)
|
|
||||||
methane = 25000.0f;
|
|
||||||
}
|
}
|
||||||
this->methane_sensor_->publish_state(methane);
|
this->methane_sensor_->publish_state(methane);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this->ethanol_sensor_ != nullptr) {
|
if (this->ethanol_sensor_ != nullptr) {
|
||||||
float ethanol = 0.0f;
|
float ethanol = 0.0f;
|
||||||
if (red_f <= 0.306f) {
|
if (red_f > 1.0f || red_f < 0.02) { // outside the range->unlikely
|
||||||
ethanol = (0.306f - red_f) / 0.00057f;
|
ethanol = 0.0;
|
||||||
if (ethanol < 10.0f)
|
} else {
|
||||||
ethanol = 0.0f;
|
ethanol = 1.52 / pow(red_f, 1.55);
|
||||||
if (ethanol > 500.0f)
|
|
||||||
ethanol = 500.0f;
|
|
||||||
}
|
}
|
||||||
this->ethanol_sensor_->publish_state(ethanol);
|
this->ethanol_sensor_->publish_state(ethanol);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this->hydrogen_sensor_ != nullptr) {
|
if (this->hydrogen_sensor_ != nullptr) {
|
||||||
float hydrogen = 0.0f;
|
float hydrogen = 0.0f;
|
||||||
if (red_f <= 0.279f) {
|
if (red_f > 0.9f || red_f < 0.02) { // outside the range->unlikely
|
||||||
hydrogen = (0.279f - red_f) / 0.00026f;
|
hydrogen = 0.0;
|
||||||
if (hydrogen < 1.0f)
|
} else {
|
||||||
hydrogen = 0.0f;
|
hydrogen = 0.85 / pow(red_f, 1.75);
|
||||||
if (hydrogen > 1000.0f)
|
|
||||||
hydrogen = 1000.0f;
|
|
||||||
}
|
}
|
||||||
this->hydrogen_sensor_->publish_state(hydrogen);
|
this->hydrogen_sensor_->publish_state(hydrogen);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this->ammonia_sensor_ != nullptr) {
|
if (this->ammonia_sensor_ != nullptr) {
|
||||||
float ammonia = 0.0f;
|
float ammonia = 0.0f;
|
||||||
if (red_f <= 0.8f) {
|
if (red_f > 0.98f || red_f < 0.2532) { // outside the ammonia range->unlikely
|
||||||
ammonia = (0.8f - red_f) / 0.0015f;
|
ammonia = 0.0;
|
||||||
if (ammonia < 1.0f)
|
} else {
|
||||||
ammonia = 0.0f;
|
ammonia = 0.9 / pow(red_f, 4.6);
|
||||||
if (ammonia > 500.0f)
|
|
||||||
ammonia = 500.0f;
|
|
||||||
}
|
}
|
||||||
this->ammonia_sensor_->publish_state(ammonia);
|
this->ammonia_sensor_->publish_state(ammonia);
|
||||||
}
|
}
|
||||||
|
|
|
@ -3,6 +3,7 @@ import esphome.config_validation as cv
|
||||||
from esphome import pins
|
from esphome import pins
|
||||||
from esphome.components import display
|
from esphome.components import display
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
|
CONF_ENABLE_PIN,
|
||||||
CONF_HSYNC_PIN,
|
CONF_HSYNC_PIN,
|
||||||
CONF_RESET_PIN,
|
CONF_RESET_PIN,
|
||||||
CONF_DATA_PINS,
|
CONF_DATA_PINS,
|
||||||
|
@ -112,6 +113,7 @@ CONFIG_SCHEMA = cv.All(
|
||||||
cv.Required(CONF_PCLK_PIN): pins.internal_gpio_output_pin_schema,
|
cv.Required(CONF_PCLK_PIN): pins.internal_gpio_output_pin_schema,
|
||||||
cv.Required(CONF_HSYNC_PIN): pins.internal_gpio_output_pin_schema,
|
cv.Required(CONF_HSYNC_PIN): pins.internal_gpio_output_pin_schema,
|
||||||
cv.Required(CONF_VSYNC_PIN): pins.internal_gpio_output_pin_schema,
|
cv.Required(CONF_VSYNC_PIN): pins.internal_gpio_output_pin_schema,
|
||||||
|
cv.Optional(CONF_ENABLE_PIN): pins.gpio_output_pin_schema,
|
||||||
cv.Optional(CONF_RESET_PIN): pins.gpio_output_pin_schema,
|
cv.Optional(CONF_RESET_PIN): pins.gpio_output_pin_schema,
|
||||||
cv.Optional(CONF_HSYNC_PULSE_WIDTH, default=10): cv.int_,
|
cv.Optional(CONF_HSYNC_PULSE_WIDTH, default=10): cv.int_,
|
||||||
cv.Optional(CONF_HSYNC_BACK_PORCH, default=10): cv.int_,
|
cv.Optional(CONF_HSYNC_BACK_PORCH, default=10): cv.int_,
|
||||||
|
@ -164,6 +166,10 @@ async def to_code(config):
|
||||||
cg.add(var.add_data_pin(data_pin, index))
|
cg.add(var.add_data_pin(data_pin, index))
|
||||||
index += 1
|
index += 1
|
||||||
|
|
||||||
|
if enable_pin := config.get(CONF_ENABLE_PIN):
|
||||||
|
enable = await cg.gpio_pin_expression(enable_pin)
|
||||||
|
cg.add(var.set_enable_pin(enable))
|
||||||
|
|
||||||
if reset_pin := config.get(CONF_RESET_PIN):
|
if reset_pin := config.get(CONF_RESET_PIN):
|
||||||
reset = await cg.gpio_pin_expression(reset_pin)
|
reset = await cg.gpio_pin_expression(reset_pin)
|
||||||
cg.add(var.set_reset_pin(reset))
|
cg.add(var.set_reset_pin(reset))
|
||||||
|
|
|
@ -104,12 +104,30 @@ void RpiDpiRgb::dump_config() {
|
||||||
ESP_LOGCONFIG(TAG, " Height: %u", this->height_);
|
ESP_LOGCONFIG(TAG, " Height: %u", this->height_);
|
||||||
ESP_LOGCONFIG(TAG, " Width: %u", this->width_);
|
ESP_LOGCONFIG(TAG, " Width: %u", this->width_);
|
||||||
LOG_PIN(" DE Pin: ", this->de_pin_);
|
LOG_PIN(" DE Pin: ", this->de_pin_);
|
||||||
|
LOG_PIN(" Enable Pin: ", this->enable_pin_);
|
||||||
LOG_PIN(" Reset Pin: ", this->reset_pin_);
|
LOG_PIN(" Reset Pin: ", this->reset_pin_);
|
||||||
size_t data_pin_count = sizeof(this->data_pins_) / sizeof(this->data_pins_[0]);
|
size_t data_pin_count = sizeof(this->data_pins_) / sizeof(this->data_pins_[0]);
|
||||||
for (size_t i = 0; i != data_pin_count; i++)
|
for (size_t i = 0; i != data_pin_count; i++)
|
||||||
ESP_LOGCONFIG(TAG, " Data pin %d: %s", i, (this->data_pins_[i])->dump_summary().c_str());
|
ESP_LOGCONFIG(TAG, " Data pin %d: %s", i, (this->data_pins_[i])->dump_summary().c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void RpiDpiRgb::reset_display_() const {
|
||||||
|
if (this->reset_pin_ != nullptr) {
|
||||||
|
this->reset_pin_->setup();
|
||||||
|
this->reset_pin_->digital_write(false);
|
||||||
|
if (this->enable_pin_ != nullptr) {
|
||||||
|
this->enable_pin_->setup();
|
||||||
|
this->enable_pin_->digital_write(false);
|
||||||
|
}
|
||||||
|
delay(1);
|
||||||
|
this->reset_pin_->digital_write(true);
|
||||||
|
if (this->enable_pin_ != nullptr) {
|
||||||
|
delay(11);
|
||||||
|
this->enable_pin_->digital_write(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace rpi_dpi_rgb
|
} // namespace rpi_dpi_rgb
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
|
||||||
|
|
|
@ -36,6 +36,7 @@ class RpiDpiRgb : public display::Display {
|
||||||
void set_pclk_pin(InternalGPIOPin *pclk_pin) { this->pclk_pin_ = pclk_pin; }
|
void set_pclk_pin(InternalGPIOPin *pclk_pin) { this->pclk_pin_ = pclk_pin; }
|
||||||
void set_vsync_pin(InternalGPIOPin *vsync_pin) { this->vsync_pin_ = vsync_pin; }
|
void set_vsync_pin(InternalGPIOPin *vsync_pin) { this->vsync_pin_ = vsync_pin; }
|
||||||
void set_hsync_pin(InternalGPIOPin *hsync_pin) { this->hsync_pin_ = hsync_pin; }
|
void set_hsync_pin(InternalGPIOPin *hsync_pin) { this->hsync_pin_ = hsync_pin; }
|
||||||
|
void set_enable_pin(GPIOPin *enable_pin) { this->enable_pin_ = enable_pin; }
|
||||||
void set_reset_pin(GPIOPin *reset_pin) { this->reset_pin_ = reset_pin; }
|
void set_reset_pin(GPIOPin *reset_pin) { this->reset_pin_ = reset_pin; }
|
||||||
void set_width(uint16_t width) { this->width_ = width; }
|
void set_width(uint16_t width) { this->width_ = width; }
|
||||||
void set_dimensions(uint16_t width, uint16_t height) {
|
void set_dimensions(uint16_t width, uint16_t height) {
|
||||||
|
@ -62,10 +63,12 @@ class RpiDpiRgb : public display::Display {
|
||||||
protected:
|
protected:
|
||||||
int get_width_internal() override { return this->width_; }
|
int get_width_internal() override { return this->width_; }
|
||||||
int get_height_internal() override { return this->height_; }
|
int get_height_internal() override { return this->height_; }
|
||||||
|
void reset_display_() const;
|
||||||
InternalGPIOPin *de_pin_{nullptr};
|
InternalGPIOPin *de_pin_{nullptr};
|
||||||
InternalGPIOPin *pclk_pin_{nullptr};
|
InternalGPIOPin *pclk_pin_{nullptr};
|
||||||
InternalGPIOPin *hsync_pin_{nullptr};
|
InternalGPIOPin *hsync_pin_{nullptr};
|
||||||
InternalGPIOPin *vsync_pin_{nullptr};
|
InternalGPIOPin *vsync_pin_{nullptr};
|
||||||
|
GPIOPin *enable_pin_{nullptr};
|
||||||
GPIOPin *reset_pin_{nullptr};
|
GPIOPin *reset_pin_{nullptr};
|
||||||
InternalGPIOPin *data_pins_[16] = {};
|
InternalGPIOPin *data_pins_[16] = {};
|
||||||
uint16_t hsync_front_porch_ = 8;
|
uint16_t hsync_front_porch_ = 8;
|
||||||
|
|
|
@ -1,47 +1,39 @@
|
||||||
import esphome.codegen as cg
|
|
||||||
import esphome.config_validation as cv
|
|
||||||
from esphome import pins
|
from esphome import pins
|
||||||
from esphome.components import (
|
import esphome.codegen as cg
|
||||||
spi,
|
from esphome.components import display, spi
|
||||||
display,
|
from esphome.components.esp32 import const, only_on_variant
|
||||||
)
|
|
||||||
from esphome.const import (
|
|
||||||
CONF_DC_PIN,
|
|
||||||
CONF_HSYNC_PIN,
|
|
||||||
CONF_RESET_PIN,
|
|
||||||
CONF_DATA_PINS,
|
|
||||||
CONF_ID,
|
|
||||||
CONF_DIMENSIONS,
|
|
||||||
CONF_VSYNC_PIN,
|
|
||||||
CONF_WIDTH,
|
|
||||||
CONF_HEIGHT,
|
|
||||||
CONF_LAMBDA,
|
|
||||||
CONF_MIRROR_X,
|
|
||||||
CONF_MIRROR_Y,
|
|
||||||
CONF_COLOR_ORDER,
|
|
||||||
CONF_TRANSFORM,
|
|
||||||
CONF_OFFSET_HEIGHT,
|
|
||||||
CONF_OFFSET_WIDTH,
|
|
||||||
CONF_INVERT_COLORS,
|
|
||||||
CONF_RED,
|
|
||||||
CONF_GREEN,
|
|
||||||
CONF_BLUE,
|
|
||||||
CONF_NUMBER,
|
|
||||||
CONF_IGNORE_STRAPPING_WARNING,
|
|
||||||
)
|
|
||||||
|
|
||||||
from esphome.components.esp32 import (
|
|
||||||
only_on_variant,
|
|
||||||
const,
|
|
||||||
)
|
|
||||||
from esphome.components.rpi_dpi_rgb.display import (
|
from esphome.components.rpi_dpi_rgb.display import (
|
||||||
CONF_PCLK_FREQUENCY,
|
CONF_PCLK_FREQUENCY,
|
||||||
CONF_PCLK_INVERTED,
|
CONF_PCLK_INVERTED,
|
||||||
)
|
)
|
||||||
from .init_sequences import (
|
import esphome.config_validation as cv
|
||||||
ST7701S_INITS,
|
from esphome.const import (
|
||||||
cmd,
|
CONF_BLUE,
|
||||||
|
CONF_COLOR_ORDER,
|
||||||
|
CONF_DATA_PINS,
|
||||||
|
CONF_DC_PIN,
|
||||||
|
CONF_DIMENSIONS,
|
||||||
|
CONF_GREEN,
|
||||||
|
CONF_HEIGHT,
|
||||||
|
CONF_HSYNC_PIN,
|
||||||
|
CONF_ID,
|
||||||
|
CONF_IGNORE_STRAPPING_WARNING,
|
||||||
|
CONF_INVERT_COLORS,
|
||||||
|
CONF_LAMBDA,
|
||||||
|
CONF_MIRROR_X,
|
||||||
|
CONF_MIRROR_Y,
|
||||||
|
CONF_NUMBER,
|
||||||
|
CONF_OFFSET_HEIGHT,
|
||||||
|
CONF_OFFSET_WIDTH,
|
||||||
|
CONF_RED,
|
||||||
|
CONF_RESET_PIN,
|
||||||
|
CONF_TRANSFORM,
|
||||||
|
CONF_VSYNC_PIN,
|
||||||
|
CONF_WIDTH,
|
||||||
)
|
)
|
||||||
|
from esphome.core import TimePeriod
|
||||||
|
|
||||||
|
from .init_sequences import ST7701S_INITS, cmd
|
||||||
|
|
||||||
CONF_INIT_SEQUENCE = "init_sequence"
|
CONF_INIT_SEQUENCE = "init_sequence"
|
||||||
CONF_DE_PIN = "de_pin"
|
CONF_DE_PIN = "de_pin"
|
||||||
|
@ -59,6 +51,7 @@ DEPENDENCIES = ["spi", "esp32"]
|
||||||
st7701s_ns = cg.esphome_ns.namespace("st7701s")
|
st7701s_ns = cg.esphome_ns.namespace("st7701s")
|
||||||
ST7701S = st7701s_ns.class_("ST7701S", display.Display, cg.Component, spi.SPIDevice)
|
ST7701S = st7701s_ns.class_("ST7701S", display.Display, cg.Component, spi.SPIDevice)
|
||||||
ColorOrder = display.display_ns.enum("ColorMode")
|
ColorOrder = display.display_ns.enum("ColorMode")
|
||||||
|
ST7701S_DELAY_FLAG = 0xFF
|
||||||
|
|
||||||
COLOR_ORDERS = {
|
COLOR_ORDERS = {
|
||||||
"RGB": ColorOrder.COLOR_ORDER_RGB,
|
"RGB": ColorOrder.COLOR_ORDER_RGB,
|
||||||
|
@ -93,18 +86,23 @@ def map_sequence(value):
|
||||||
"""
|
"""
|
||||||
An initialisation sequence can be selected from one of the pre-defined sequences in init_sequences.py,
|
An initialisation sequence can be selected from one of the pre-defined sequences in init_sequences.py,
|
||||||
or can be a literal array of data bytes.
|
or can be a literal array of data bytes.
|
||||||
The format is a repeated sequence of [CMD, LEN, <data>] where <data> is LEN bytes.
|
The format is a repeated sequence of [CMD, <data>] where <data> is s a sequence of bytes. The length is inferred
|
||||||
|
from the length of the sequence and should not be explicit.
|
||||||
|
A delay can be inserted by specifying "- delay N" where N is in ms
|
||||||
"""
|
"""
|
||||||
|
if isinstance(value, str) and value.lower().startswith("delay "):
|
||||||
|
value = value.lower()[6:]
|
||||||
|
delay = cv.All(
|
||||||
|
cv.positive_time_period_milliseconds,
|
||||||
|
cv.Range(TimePeriod(milliseconds=1), TimePeriod(milliseconds=255)),
|
||||||
|
)(value)
|
||||||
|
return [delay, ST7701S_DELAY_FLAG]
|
||||||
if not isinstance(value, list):
|
if not isinstance(value, list):
|
||||||
value = cv.int_(value)
|
value = cv.int_(value)
|
||||||
value = cv.one_of(*ST7701S_INITS)(value)
|
value = cv.one_of(*ST7701S_INITS)(value)
|
||||||
return ST7701S_INITS[value]
|
return ST7701S_INITS[value]
|
||||||
# value = cv.ensure_list(cv.uint8_t)(value)
|
value = cv.Length(min=1, max=254)(value)
|
||||||
data_length = len(value)
|
return cmd(*value)
|
||||||
if data_length == 0:
|
|
||||||
raise cv.Invalid("Empty sequence")
|
|
||||||
value = cmd(*value)
|
|
||||||
return value
|
|
||||||
|
|
||||||
|
|
||||||
CONFIG_SCHEMA = cv.All(
|
CONFIG_SCHEMA = cv.All(
|
||||||
|
|
|
@ -138,11 +138,16 @@ void ST7701S::write_init_sequence_() {
|
||||||
for (size_t i = 0; i != this->init_sequence_.size();) {
|
for (size_t i = 0; i != this->init_sequence_.size();) {
|
||||||
uint8_t cmd = this->init_sequence_[i++];
|
uint8_t cmd = this->init_sequence_[i++];
|
||||||
size_t len = this->init_sequence_[i++];
|
size_t len = this->init_sequence_[i++];
|
||||||
this->write_sequence_(cmd, len, &this->init_sequence_[i]);
|
if (len == ST7701S_DELAY_FLAG) {
|
||||||
i += len;
|
ESP_LOGV(TAG, "Delay %dms", cmd);
|
||||||
esph_log_v(TAG, "Command %X, %d bytes", cmd, len);
|
delay(cmd);
|
||||||
if (cmd == SW_RESET_CMD)
|
} else {
|
||||||
delay(6);
|
this->write_sequence_(cmd, len, &this->init_sequence_[i]);
|
||||||
|
i += len;
|
||||||
|
ESP_LOGV(TAG, "Command %X, %d bytes", cmd, len);
|
||||||
|
if (cmd == SW_RESET_CMD)
|
||||||
|
delay(6);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// st7701 does not appear to support axis swapping
|
// st7701 does not appear to support axis swapping
|
||||||
this->write_sequence_(CMD2_BKSEL, sizeof(CMD2_BK0), CMD2_BK0);
|
this->write_sequence_(CMD2_BKSEL, sizeof(CMD2_BK0), CMD2_BK0);
|
||||||
|
@ -153,7 +158,7 @@ void ST7701S::write_init_sequence_() {
|
||||||
val |= 0x10;
|
val |= 0x10;
|
||||||
this->write_command_(MADCTL_CMD);
|
this->write_command_(MADCTL_CMD);
|
||||||
this->write_data_(val);
|
this->write_data_(val);
|
||||||
esph_log_d(TAG, "write MADCTL %X", val);
|
ESP_LOGD(TAG, "write MADCTL %X", val);
|
||||||
this->write_command_(this->invert_colors_ ? INVERT_ON : INVERT_OFF);
|
this->write_command_(this->invert_colors_ ? INVERT_ON : INVERT_OFF);
|
||||||
this->set_timeout(120, [this] {
|
this->set_timeout(120, [this] {
|
||||||
this->write_command_(SLEEP_OUT);
|
this->write_command_(SLEEP_OUT);
|
||||||
|
|
|
@ -25,6 +25,7 @@ const uint8_t INVERT_ON = 0x21;
|
||||||
const uint8_t DISPLAY_ON = 0x29;
|
const uint8_t DISPLAY_ON = 0x29;
|
||||||
const uint8_t CMD2_BKSEL = 0xFF;
|
const uint8_t CMD2_BKSEL = 0xFF;
|
||||||
const uint8_t CMD2_BK0[5] = {0x77, 0x01, 0x00, 0x00, 0x10};
|
const uint8_t CMD2_BK0[5] = {0x77, 0x01, 0x00, 0x00, 0x10};
|
||||||
|
const uint8_t ST7701S_DELAY_FLAG = 0xFF;
|
||||||
|
|
||||||
class ST7701S : public display::Display,
|
class ST7701S : public display::Display,
|
||||||
public spi::SPIDevice<spi::BIT_ORDER_MSB_FIRST, spi::CLOCK_POLARITY_LOW, spi::CLOCK_PHASE_LEADING,
|
public spi::SPIDevice<spi::BIT_ORDER_MSB_FIRST, spi::CLOCK_POLARITY_LOW, spi::CLOCK_PHASE_LEADING,
|
||||||
|
|
|
@ -9,6 +9,7 @@ from esphome.const import (
|
||||||
CONF_MULTIPLY,
|
CONF_MULTIPLY,
|
||||||
CONF_STEP,
|
CONF_STEP,
|
||||||
CONF_INITIAL_VALUE,
|
CONF_INITIAL_VALUE,
|
||||||
|
CONF_RESTORE_VALUE,
|
||||||
)
|
)
|
||||||
from .. import tuya_ns, CONF_TUYA_ID, Tuya, TuyaDatapointType
|
from .. import tuya_ns, CONF_TUYA_ID, Tuya, TuyaDatapointType
|
||||||
|
|
||||||
|
@ -58,6 +59,7 @@ CONFIG_SCHEMA = cv.All(
|
||||||
DATAPOINT_TYPES, lower=True
|
DATAPOINT_TYPES, lower=True
|
||||||
),
|
),
|
||||||
cv.Optional(CONF_INITIAL_VALUE): cv.float_,
|
cv.Optional(CONF_INITIAL_VALUE): cv.float_,
|
||||||
|
cv.Optional(CONF_RESTORE_VALUE, default=False): cv.boolean,
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
|
@ -90,3 +92,4 @@ async def to_code(config):
|
||||||
hidden_init_value := hidden_config.get(CONF_INITIAL_VALUE, None)
|
hidden_init_value := hidden_config.get(CONF_INITIAL_VALUE, None)
|
||||||
) is not None:
|
) is not None:
|
||||||
cg.add(var.set_datapoint_initial_value(hidden_init_value))
|
cg.add(var.set_datapoint_initial_value(hidden_init_value))
|
||||||
|
cg.add(var.set_restore_value(hidden_config[CONF_RESTORE_VALUE]))
|
||||||
|
|
|
@ -7,14 +7,28 @@ namespace tuya {
|
||||||
static const char *const TAG = "tuya.number";
|
static const char *const TAG = "tuya.number";
|
||||||
|
|
||||||
void TuyaNumber::setup() {
|
void TuyaNumber::setup() {
|
||||||
|
if (this->restore_value_) {
|
||||||
|
this->pref_ = global_preferences->make_preference<float>(this->get_object_id_hash());
|
||||||
|
}
|
||||||
|
|
||||||
this->parent_->register_listener(this->number_id_, [this](const TuyaDatapoint &datapoint) {
|
this->parent_->register_listener(this->number_id_, [this](const TuyaDatapoint &datapoint) {
|
||||||
if (datapoint.type == TuyaDatapointType::INTEGER) {
|
if (datapoint.type == TuyaDatapointType::INTEGER) {
|
||||||
ESP_LOGV(TAG, "MCU reported number %u is: %d", datapoint.id, datapoint.value_int);
|
ESP_LOGV(TAG, "MCU reported number %u is: %d", datapoint.id, datapoint.value_int);
|
||||||
this->publish_state(datapoint.value_int / multiply_by_);
|
float value = datapoint.value_int / multiply_by_;
|
||||||
|
this->publish_state(value);
|
||||||
|
if (this->restore_value_)
|
||||||
|
this->pref_.save(&value);
|
||||||
} else if (datapoint.type == TuyaDatapointType::ENUM) {
|
} else if (datapoint.type == TuyaDatapointType::ENUM) {
|
||||||
ESP_LOGV(TAG, "MCU reported number %u is: %u", datapoint.id, datapoint.value_enum);
|
ESP_LOGV(TAG, "MCU reported number %u is: %u", datapoint.id, datapoint.value_enum);
|
||||||
this->publish_state(datapoint.value_enum);
|
float value = datapoint.value_enum;
|
||||||
|
this->publish_state(value);
|
||||||
|
if (this->restore_value_)
|
||||||
|
this->pref_.save(&value);
|
||||||
|
} else {
|
||||||
|
ESP_LOGW(TAG, "Reported type (%d) is not a number!", static_cast<int>(datapoint.type));
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if ((this->type_) && (this->type_ != datapoint.type)) {
|
if ((this->type_) && (this->type_ != datapoint.type)) {
|
||||||
ESP_LOGW(TAG, "Reported type (%d) different than previously set (%d)!", static_cast<int>(datapoint.type),
|
ESP_LOGW(TAG, "Reported type (%d) different than previously set (%d)!", static_cast<int>(datapoint.type),
|
||||||
static_cast<int>(*this->type_));
|
static_cast<int>(*this->type_));
|
||||||
|
@ -23,8 +37,26 @@ void TuyaNumber::setup() {
|
||||||
});
|
});
|
||||||
|
|
||||||
this->parent_->add_on_initialized_callback([this] {
|
this->parent_->add_on_initialized_callback([this] {
|
||||||
if ((this->initial_value_) && (this->type_)) {
|
if (this->type_) {
|
||||||
this->control(*this->initial_value_);
|
float value;
|
||||||
|
if (!this->restore_value_) {
|
||||||
|
if (this->initial_value_) {
|
||||||
|
value = *this->initial_value_;
|
||||||
|
} else {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (!this->pref_.load(&value)) {
|
||||||
|
if (this->initial_value_) {
|
||||||
|
value = *this->initial_value_;
|
||||||
|
} else {
|
||||||
|
value = this->traits.get_min_value();
|
||||||
|
ESP_LOGW(TAG, "Failed to restore and there is no initial value defined. Setting min_value (%f)", value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
this->control(value);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
@ -38,6 +70,9 @@ void TuyaNumber::control(float value) {
|
||||||
this->parent_->set_enum_datapoint_value(this->number_id_, value);
|
this->parent_->set_enum_datapoint_value(this->number_id_, value);
|
||||||
}
|
}
|
||||||
this->publish_state(value);
|
this->publish_state(value);
|
||||||
|
|
||||||
|
if (this->restore_value_)
|
||||||
|
this->pref_.save(&value);
|
||||||
}
|
}
|
||||||
|
|
||||||
void TuyaNumber::dump_config() {
|
void TuyaNumber::dump_config() {
|
||||||
|
@ -52,6 +87,8 @@ void TuyaNumber::dump_config() {
|
||||||
if (this->initial_value_) {
|
if (this->initial_value_) {
|
||||||
ESP_LOGCONFIG(TAG, " Initial Value: %f", *this->initial_value_);
|
ESP_LOGCONFIG(TAG, " Initial Value: %f", *this->initial_value_);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ESP_LOGCONFIG(TAG, " Restore Value: %s", YESNO(this->restore_value_));
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace tuya
|
} // namespace tuya
|
||||||
|
|
|
@ -1,9 +1,10 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include "esphome/core/component.h"
|
|
||||||
#include "esphome/components/tuya/tuya.h"
|
|
||||||
#include "esphome/components/number/number.h"
|
#include "esphome/components/number/number.h"
|
||||||
|
#include "esphome/components/tuya/tuya.h"
|
||||||
|
#include "esphome/core/component.h"
|
||||||
#include "esphome/core/optional.h"
|
#include "esphome/core/optional.h"
|
||||||
|
#include "esphome/core/preferences.h"
|
||||||
|
|
||||||
namespace esphome {
|
namespace esphome {
|
||||||
namespace tuya {
|
namespace tuya {
|
||||||
|
@ -16,6 +17,7 @@ class TuyaNumber : public number::Number, public Component {
|
||||||
void set_write_multiply(float factor) { multiply_by_ = factor; }
|
void set_write_multiply(float factor) { multiply_by_ = factor; }
|
||||||
void set_datapoint_type(TuyaDatapointType type) { type_ = type; }
|
void set_datapoint_type(TuyaDatapointType type) { type_ = type; }
|
||||||
void set_datapoint_initial_value(float value) { this->initial_value_ = value; }
|
void set_datapoint_initial_value(float value) { this->initial_value_ = value; }
|
||||||
|
void set_restore_value(bool restore_value) { this->restore_value_ = restore_value; }
|
||||||
|
|
||||||
void set_tuya_parent(Tuya *parent) { this->parent_ = parent; }
|
void set_tuya_parent(Tuya *parent) { this->parent_ = parent; }
|
||||||
|
|
||||||
|
@ -27,6 +29,9 @@ class TuyaNumber : public number::Number, public Component {
|
||||||
float multiply_by_{1.0};
|
float multiply_by_{1.0};
|
||||||
optional<TuyaDatapointType> type_{};
|
optional<TuyaDatapointType> type_{};
|
||||||
optional<float> initial_value_{};
|
optional<float> initial_value_{};
|
||||||
|
bool restore_value_{false};
|
||||||
|
|
||||||
|
ESPPreferenceObject pref_;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace tuya
|
} // namespace tuya
|
||||||
|
|
158
esphome/components/udp/__init__.py
Normal file
158
esphome/components/udp/__init__.py
Normal file
|
@ -0,0 +1,158 @@
|
||||||
|
import hashlib
|
||||||
|
|
||||||
|
import esphome.codegen as cg
|
||||||
|
from esphome.components.api import CONF_ENCRYPTION
|
||||||
|
from esphome.components.binary_sensor import BinarySensor
|
||||||
|
from esphome.components.sensor import Sensor
|
||||||
|
import esphome.config_validation as cv
|
||||||
|
from esphome.const import (
|
||||||
|
CONF_BINARY_SENSORS,
|
||||||
|
CONF_ID,
|
||||||
|
CONF_INTERNAL,
|
||||||
|
CONF_KEY,
|
||||||
|
CONF_NAME,
|
||||||
|
CONF_PORT,
|
||||||
|
CONF_SENSORS,
|
||||||
|
)
|
||||||
|
from esphome.cpp_generator import MockObjClass
|
||||||
|
|
||||||
|
CODEOWNERS = ["@clydebarrow"]
|
||||||
|
DEPENDENCIES = ["network"]
|
||||||
|
AUTO_LOAD = ["socket"]
|
||||||
|
MULTI_CONF = True
|
||||||
|
|
||||||
|
udp_ns = cg.esphome_ns.namespace("udp")
|
||||||
|
UDPComponent = udp_ns.class_("UDPComponent", cg.PollingComponent)
|
||||||
|
|
||||||
|
CONF_BROADCAST = "broadcast"
|
||||||
|
CONF_BROADCAST_ID = "broadcast_id"
|
||||||
|
CONF_ADDRESSES = "addresses"
|
||||||
|
CONF_PROVIDER = "provider"
|
||||||
|
CONF_PROVIDERS = "providers"
|
||||||
|
CONF_REMOTE_ID = "remote_id"
|
||||||
|
CONF_UDP_ID = "udp_id"
|
||||||
|
CONF_PING_PONG_ENABLE = "ping_pong_enable"
|
||||||
|
CONF_PING_PONG_RECYCLE_TIME = "ping_pong_recycle_time"
|
||||||
|
CONF_ROLLING_CODE_ENABLE = "rolling_code_enable"
|
||||||
|
|
||||||
|
|
||||||
|
def sensor_validation(cls: MockObjClass):
|
||||||
|
return cv.maybe_simple_value(
|
||||||
|
cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_ID): cv.use_id(cls),
|
||||||
|
cv.Optional(CONF_BROADCAST_ID): cv.validate_id_name,
|
||||||
|
}
|
||||||
|
),
|
||||||
|
key=CONF_ID,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
ENCRYPTION_SCHEMA = {
|
||||||
|
cv.Optional(CONF_ENCRYPTION): cv.maybe_simple_value(
|
||||||
|
cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_KEY): cv.string,
|
||||||
|
}
|
||||||
|
),
|
||||||
|
key=CONF_KEY,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
PROVIDER_SCHEMA = cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Required(CONF_NAME): cv.valid_name,
|
||||||
|
}
|
||||||
|
).extend(ENCRYPTION_SCHEMA)
|
||||||
|
|
||||||
|
|
||||||
|
def validate_(config):
|
||||||
|
if CONF_ENCRYPTION in config:
|
||||||
|
if CONF_SENSORS not in config and CONF_BINARY_SENSORS not in config:
|
||||||
|
raise cv.Invalid("No sensors or binary sensors to encrypt")
|
||||||
|
elif config[CONF_ROLLING_CODE_ENABLE]:
|
||||||
|
raise cv.Invalid("Rolling code requires an encryption key")
|
||||||
|
if config[CONF_PING_PONG_ENABLE]:
|
||||||
|
if not any(CONF_ENCRYPTION in p for p in config.get(CONF_PROVIDERS) or ()):
|
||||||
|
raise cv.Invalid("Ping-pong requires at least one encrypted provider")
|
||||||
|
return config
|
||||||
|
|
||||||
|
|
||||||
|
CONFIG_SCHEMA = cv.All(
|
||||||
|
cv.polling_component_schema("15s")
|
||||||
|
.extend(
|
||||||
|
{
|
||||||
|
cv.GenerateID(): cv.declare_id(UDPComponent),
|
||||||
|
cv.Optional(CONF_PORT, default=18511): cv.port,
|
||||||
|
cv.Optional(CONF_ADDRESSES, default=["255.255.255.255"]): cv.ensure_list(
|
||||||
|
cv.ipv4
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_ROLLING_CODE_ENABLE, default=False): cv.boolean,
|
||||||
|
cv.Optional(CONF_PING_PONG_ENABLE, default=False): cv.boolean,
|
||||||
|
cv.Optional(
|
||||||
|
CONF_PING_PONG_RECYCLE_TIME, default="600s"
|
||||||
|
): cv.positive_time_period_seconds,
|
||||||
|
cv.Optional(CONF_SENSORS): cv.ensure_list(sensor_validation(Sensor)),
|
||||||
|
cv.Optional(CONF_BINARY_SENSORS): cv.ensure_list(
|
||||||
|
sensor_validation(BinarySensor)
|
||||||
|
),
|
||||||
|
cv.Optional(CONF_PROVIDERS): cv.ensure_list(PROVIDER_SCHEMA),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.extend(ENCRYPTION_SCHEMA),
|
||||||
|
validate_,
|
||||||
|
)
|
||||||
|
|
||||||
|
SENSOR_SCHEMA = cv.Schema(
|
||||||
|
{
|
||||||
|
cv.Optional(CONF_REMOTE_ID): cv.string_strict,
|
||||||
|
cv.Required(CONF_PROVIDER): cv.valid_name,
|
||||||
|
cv.GenerateID(CONF_UDP_ID): cv.use_id(UDPComponent),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def require_internal_with_name(config):
|
||||||
|
if CONF_NAME in config and CONF_INTERNAL not in config:
|
||||||
|
raise cv.Invalid("Must provide internal: config when using name:")
|
||||||
|
return config
|
||||||
|
|
||||||
|
|
||||||
|
def hash_encryption_key(config: dict):
|
||||||
|
return list(hashlib.sha256(config[CONF_KEY].encode()).digest())
|
||||||
|
|
||||||
|
|
||||||
|
async def to_code(config):
|
||||||
|
cg.add_define("USE_UDP")
|
||||||
|
cg.add_global(udp_ns.using)
|
||||||
|
var = cg.new_Pvariable(config[CONF_ID])
|
||||||
|
await cg.register_component(var, config)
|
||||||
|
cg.add(var.set_port(config[CONF_PORT]))
|
||||||
|
cg.add(var.set_rolling_code_enable(config[CONF_ROLLING_CODE_ENABLE]))
|
||||||
|
cg.add(var.set_ping_pong_enable(config[CONF_PING_PONG_ENABLE]))
|
||||||
|
cg.add(
|
||||||
|
var.set_ping_pong_recycle_time(
|
||||||
|
config[CONF_PING_PONG_RECYCLE_TIME].total_seconds
|
||||||
|
)
|
||||||
|
)
|
||||||
|
for sens_conf in config.get(CONF_SENSORS, ()):
|
||||||
|
sens_id = sens_conf[CONF_ID]
|
||||||
|
sensor = await cg.get_variable(sens_id)
|
||||||
|
bcst_id = sens_conf.get(CONF_BROADCAST_ID, sens_id.id)
|
||||||
|
cg.add(var.add_sensor(bcst_id, sensor))
|
||||||
|
for sens_conf in config.get(CONF_BINARY_SENSORS, ()):
|
||||||
|
sens_id = sens_conf[CONF_ID]
|
||||||
|
sensor = await cg.get_variable(sens_id)
|
||||||
|
bcst_id = sens_conf.get(CONF_BROADCAST_ID, sens_id.id)
|
||||||
|
cg.add(var.add_binary_sensor(bcst_id, sensor))
|
||||||
|
for address in config[CONF_ADDRESSES]:
|
||||||
|
cg.add(var.add_address(str(address)))
|
||||||
|
|
||||||
|
if encryption := config.get(CONF_ENCRYPTION):
|
||||||
|
cg.add(var.set_encryption_key(hash_encryption_key(encryption)))
|
||||||
|
|
||||||
|
for provider in config.get(CONF_PROVIDERS, ()):
|
||||||
|
name = provider[CONF_NAME]
|
||||||
|
cg.add(var.add_provider(name))
|
||||||
|
if encryption := provider.get(CONF_ENCRYPTION):
|
||||||
|
cg.add(var.set_provider_encryption(name, hash_encryption_key(encryption)))
|
27
esphome/components/udp/binary_sensor.py
Normal file
27
esphome/components/udp/binary_sensor.py
Normal file
|
@ -0,0 +1,27 @@
|
||||||
|
import esphome.codegen as cg
|
||||||
|
from esphome.components import binary_sensor
|
||||||
|
from esphome.config_validation import All, has_at_least_one_key
|
||||||
|
from esphome.const import CONF_ID
|
||||||
|
|
||||||
|
from . import (
|
||||||
|
CONF_PROVIDER,
|
||||||
|
CONF_REMOTE_ID,
|
||||||
|
CONF_UDP_ID,
|
||||||
|
SENSOR_SCHEMA,
|
||||||
|
require_internal_with_name,
|
||||||
|
)
|
||||||
|
|
||||||
|
DEPENDENCIES = ["udp"]
|
||||||
|
|
||||||
|
CONFIG_SCHEMA = All(
|
||||||
|
binary_sensor.binary_sensor_schema().extend(SENSOR_SCHEMA),
|
||||||
|
has_at_least_one_key(CONF_ID, CONF_REMOTE_ID),
|
||||||
|
require_internal_with_name,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def to_code(config):
|
||||||
|
var = await binary_sensor.new_binary_sensor(config)
|
||||||
|
comp = await cg.get_variable(config[CONF_UDP_ID])
|
||||||
|
remote_id = str(config.get(CONF_REMOTE_ID) or config.get(CONF_ID))
|
||||||
|
cg.add(comp.add_remote_binary_sensor(config[CONF_PROVIDER], remote_id, var))
|
27
esphome/components/udp/sensor.py
Normal file
27
esphome/components/udp/sensor.py
Normal file
|
@ -0,0 +1,27 @@
|
||||||
|
import esphome.codegen as cg
|
||||||
|
from esphome.components.sensor import new_sensor, sensor_schema
|
||||||
|
from esphome.config_validation import All, has_at_least_one_key
|
||||||
|
from esphome.const import CONF_ID
|
||||||
|
|
||||||
|
from . import (
|
||||||
|
CONF_PROVIDER,
|
||||||
|
CONF_REMOTE_ID,
|
||||||
|
CONF_UDP_ID,
|
||||||
|
SENSOR_SCHEMA,
|
||||||
|
require_internal_with_name,
|
||||||
|
)
|
||||||
|
|
||||||
|
DEPENDENCIES = ["udp"]
|
||||||
|
|
||||||
|
CONFIG_SCHEMA = All(
|
||||||
|
sensor_schema().extend(SENSOR_SCHEMA),
|
||||||
|
has_at_least_one_key(CONF_ID, CONF_REMOTE_ID),
|
||||||
|
require_internal_with_name,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def to_code(config):
|
||||||
|
var = await new_sensor(config)
|
||||||
|
comp = await cg.get_variable(config[CONF_UDP_ID])
|
||||||
|
remote_id = str(config.get(CONF_REMOTE_ID) or config.get(CONF_ID))
|
||||||
|
cg.add(comp.add_remote_sensor(config[CONF_PROVIDER], remote_id, var))
|
616
esphome/components/udp/udp_component.cpp
Normal file
616
esphome/components/udp/udp_component.cpp
Normal file
|
@ -0,0 +1,616 @@
|
||||||
|
#include "esphome/core/log.h"
|
||||||
|
#include "esphome/core/application.h"
|
||||||
|
#include "esphome/components/network/util.h"
|
||||||
|
#include "udp_component.h"
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace udp {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Structure of a data packet; everything is little-endian
|
||||||
|
*
|
||||||
|
* --- In clear text ---
|
||||||
|
* MAGIC_NUMBER: 16 bits
|
||||||
|
* host name length: 1 byte
|
||||||
|
* host name: (length) bytes
|
||||||
|
* padding: 0 or more null bytes to a 4 byte boundary
|
||||||
|
*
|
||||||
|
* --- Encrypted (if key set) ----
|
||||||
|
* DATA_KEY: 1 byte: OR ROLLING_CODE_KEY:
|
||||||
|
* Rolling code (if enabled): 8 bytes
|
||||||
|
* Ping keys: if any
|
||||||
|
* repeat:
|
||||||
|
* PING_KEY: 1 byte
|
||||||
|
* ping code: 4 bytes
|
||||||
|
* Sensors:
|
||||||
|
* repeat:
|
||||||
|
* SENSOR_KEY: 1 byte
|
||||||
|
* float value: 4 bytes
|
||||||
|
* name length: 1 byte
|
||||||
|
* name
|
||||||
|
* Binary Sensors:
|
||||||
|
* repeat:
|
||||||
|
* BINARY_SENSOR_KEY: 1 byte
|
||||||
|
* bool value: 1 bytes
|
||||||
|
* name length: 1 byte
|
||||||
|
* name
|
||||||
|
*
|
||||||
|
* Padded to a 4 byte boundary with nulls
|
||||||
|
*
|
||||||
|
* Structure of a ping request packet:
|
||||||
|
* --- In clear text ---
|
||||||
|
* MAGIC_PING: 16 bits
|
||||||
|
* host name length: 1 byte
|
||||||
|
* host name: (length) bytes
|
||||||
|
* Ping key (4 bytes)
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
static const char *const TAG = "udp";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* XXTEA implementation, using 256 bit key.
|
||||||
|
*/
|
||||||
|
|
||||||
|
static const uint32_t DELTA = 0x9e3779b9;
|
||||||
|
#define MX ((((z >> 5) ^ (y << 2)) + ((y >> 3) ^ (z << 4))) ^ ((sum ^ y) + (k[(p ^ e) & 7] ^ z)))
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Encrypt a block of data in-place
|
||||||
|
*/
|
||||||
|
|
||||||
|
static void xxtea_encrypt(uint32_t *v, size_t n, const uint32_t *k) {
|
||||||
|
uint32_t z, y, sum, e;
|
||||||
|
size_t p;
|
||||||
|
size_t q = 6 + 52 / n;
|
||||||
|
sum = 0;
|
||||||
|
z = v[n - 1];
|
||||||
|
while (q-- != 0) {
|
||||||
|
sum += DELTA;
|
||||||
|
e = (sum >> 2);
|
||||||
|
for (p = 0; p != n - 1; p++) {
|
||||||
|
y = v[p + 1];
|
||||||
|
z = v[p] += MX;
|
||||||
|
}
|
||||||
|
y = v[0];
|
||||||
|
z = v[n - 1] += MX;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void xxtea_decrypt(uint32_t *v, size_t n, const uint32_t *k) {
|
||||||
|
uint32_t z, y, sum, e;
|
||||||
|
size_t p;
|
||||||
|
size_t q = 6 + 52 / n;
|
||||||
|
sum = q * DELTA;
|
||||||
|
y = v[0];
|
||||||
|
while (q-- != 0) {
|
||||||
|
e = (sum >> 2);
|
||||||
|
for (p = n - 1; p != 0; p--) {
|
||||||
|
z = v[p - 1];
|
||||||
|
y = v[p] -= MX;
|
||||||
|
}
|
||||||
|
z = v[n - 1];
|
||||||
|
y = v[0] -= MX;
|
||||||
|
sum -= DELTA;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
inline static size_t round4(size_t value) { return (value + 3) & ~3; }
|
||||||
|
|
||||||
|
union FuData {
|
||||||
|
uint32_t u32;
|
||||||
|
float f32;
|
||||||
|
};
|
||||||
|
|
||||||
|
static const size_t MAX_PACKET_SIZE = 508;
|
||||||
|
static const uint16_t MAGIC_NUMBER = 0x4553;
|
||||||
|
static const uint16_t MAGIC_PING = 0x5048;
|
||||||
|
static const uint32_t PREF_HASH = 0x45535043;
|
||||||
|
enum DataKey {
|
||||||
|
ZERO_FILL_KEY,
|
||||||
|
DATA_KEY,
|
||||||
|
SENSOR_KEY,
|
||||||
|
BINARY_SENSOR_KEY,
|
||||||
|
PING_KEY,
|
||||||
|
ROLLING_CODE_KEY,
|
||||||
|
};
|
||||||
|
|
||||||
|
static const size_t MAX_PING_KEYS = 4;
|
||||||
|
|
||||||
|
static inline void add(std::vector<uint8_t> &vec, uint32_t data) {
|
||||||
|
vec.push_back(data & 0xFF);
|
||||||
|
vec.push_back((data >> 8) & 0xFF);
|
||||||
|
vec.push_back((data >> 16) & 0xFF);
|
||||||
|
vec.push_back((data >> 24) & 0xFF);
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline uint32_t get_uint32(uint8_t *&buf) {
|
||||||
|
uint32_t data = *buf++;
|
||||||
|
data += *buf++ << 8;
|
||||||
|
data += *buf++ << 16;
|
||||||
|
data += *buf++ << 24;
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline uint16_t get_uint16(uint8_t *&buf) {
|
||||||
|
uint16_t data = *buf++;
|
||||||
|
data += *buf++ << 8;
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void add(std::vector<uint8_t> &vec, uint8_t data) { vec.push_back(data); }
|
||||||
|
static inline void add(std::vector<uint8_t> &vec, uint16_t data) {
|
||||||
|
vec.push_back((uint8_t) data);
|
||||||
|
vec.push_back((uint8_t) (data >> 8));
|
||||||
|
}
|
||||||
|
static inline void add(std::vector<uint8_t> &vec, DataKey data) { vec.push_back(data); }
|
||||||
|
static void add(std::vector<uint8_t> &vec, const char *str) {
|
||||||
|
auto len = strlen(str);
|
||||||
|
vec.push_back(len);
|
||||||
|
for (size_t i = 0; i != len; i++) {
|
||||||
|
vec.push_back(*str++);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::setup() {
|
||||||
|
this->name_ = App.get_name().c_str();
|
||||||
|
if (strlen(this->name_) > 255) {
|
||||||
|
this->mark_failed();
|
||||||
|
this->status_set_error("Device name exceeds 255 chars");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
this->resend_ping_key_ = this->ping_pong_enable_;
|
||||||
|
// restore the upper 32 bits of the rolling code, increment and save.
|
||||||
|
this->pref_ = global_preferences->make_preference<uint32_t>(PREF_HASH, true);
|
||||||
|
this->pref_.load(&this->rolling_code_[1]);
|
||||||
|
this->rolling_code_[1]++;
|
||||||
|
this->pref_.save(&this->rolling_code_[1]);
|
||||||
|
this->ping_key_ = random_uint32();
|
||||||
|
ESP_LOGV(TAG, "Rolling code incremented, upper part now %u", (unsigned) this->rolling_code_[1]);
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
for (auto &sensor : this->sensors_) {
|
||||||
|
sensor.sensor->add_on_state_callback([this, &sensor](float x) {
|
||||||
|
this->updated_ = true;
|
||||||
|
sensor.updated = true;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
for (auto &sensor : this->binary_sensors_) {
|
||||||
|
sensor.sensor->add_on_state_callback([this, &sensor](bool value) {
|
||||||
|
this->updated_ = true;
|
||||||
|
sensor.updated = true;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
this->should_send_ = this->ping_pong_enable_;
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
this->should_send_ |= !this->sensors_.empty();
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
this->should_send_ |= !this->binary_sensors_.empty();
|
||||||
|
#endif
|
||||||
|
this->should_listen_ = !this->providers_.empty() || this->is_encrypted_();
|
||||||
|
// initialise the header. This is invariant.
|
||||||
|
add(this->header_, MAGIC_NUMBER);
|
||||||
|
add(this->header_, this->name_);
|
||||||
|
// pad to a multiple of 4 bytes
|
||||||
|
while (this->header_.size() & 0x3)
|
||||||
|
this->header_.push_back(0);
|
||||||
|
#if defined(USE_SOCKET_IMPL_BSD_SOCKETS) || defined(USE_SOCKET_IMPL_LWIP_SOCKETS)
|
||||||
|
for (const auto &address : this->addresses_) {
|
||||||
|
struct sockaddr saddr {};
|
||||||
|
socket::set_sockaddr(&saddr, sizeof(saddr), address, this->port_);
|
||||||
|
this->sockaddrs_.push_back(saddr);
|
||||||
|
}
|
||||||
|
// set up broadcast socket
|
||||||
|
if (this->should_send_) {
|
||||||
|
this->broadcast_socket_ = socket::socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
|
||||||
|
if (this->broadcast_socket_ == nullptr) {
|
||||||
|
this->mark_failed();
|
||||||
|
this->status_set_error("Could not create socket");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
int enable = 1;
|
||||||
|
auto err = this->broadcast_socket_->setsockopt(SOL_SOCKET, SO_REUSEADDR, &enable, sizeof(int));
|
||||||
|
if (err != 0) {
|
||||||
|
this->status_set_warning("Socket unable to set reuseaddr");
|
||||||
|
// we can still continue
|
||||||
|
}
|
||||||
|
err = this->broadcast_socket_->setsockopt(SOL_SOCKET, SO_BROADCAST, &enable, sizeof(int));
|
||||||
|
if (err != 0) {
|
||||||
|
this->status_set_warning("Socket unable to set broadcast");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// create listening socket if we either want to subscribe to providers, or need to listen
|
||||||
|
// for ping key broadcasts.
|
||||||
|
if (this->should_listen_) {
|
||||||
|
this->listen_socket_ = socket::socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
|
||||||
|
if (this->listen_socket_ == nullptr) {
|
||||||
|
this->mark_failed();
|
||||||
|
this->status_set_error("Could not create socket");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
auto err = this->listen_socket_->setblocking(false);
|
||||||
|
if (err < 0) {
|
||||||
|
ESP_LOGE(TAG, "Unable to set nonblocking: errno %d", errno);
|
||||||
|
this->mark_failed();
|
||||||
|
this->status_set_error("Unable to set nonblocking");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
int enable = 1;
|
||||||
|
err = this->listen_socket_->setsockopt(SOL_SOCKET, SO_REUSEADDR, &enable, sizeof(enable));
|
||||||
|
if (err != 0) {
|
||||||
|
this->status_set_warning("Socket unable to set reuseaddr");
|
||||||
|
// we can still continue
|
||||||
|
}
|
||||||
|
struct sockaddr_in server {};
|
||||||
|
|
||||||
|
socklen_t sl = socket::set_sockaddr_any((struct sockaddr *) &server, sizeof(server), this->port_);
|
||||||
|
if (sl == 0) {
|
||||||
|
ESP_LOGE(TAG, "Socket unable to set sockaddr: errno %d", errno);
|
||||||
|
this->mark_failed();
|
||||||
|
this->status_set_error("Unable to set sockaddr");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
err = this->listen_socket_->bind((struct sockaddr *) &server, sizeof(server));
|
||||||
|
if (err != 0) {
|
||||||
|
ESP_LOGE(TAG, "Socket unable to bind: errno %d", errno);
|
||||||
|
this->mark_failed();
|
||||||
|
this->status_set_error("Unable to bind socket");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
// 8266 and RP2040 `Duino
|
||||||
|
for (const auto &address : this->addresses_) {
|
||||||
|
auto ipaddr = IPAddress();
|
||||||
|
ipaddr.fromString(address.c_str());
|
||||||
|
this->ipaddrs_.push_back(ipaddr);
|
||||||
|
}
|
||||||
|
if (this->should_listen_)
|
||||||
|
this->udp_client_.begin(this->port_);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::init_data_() {
|
||||||
|
this->data_.clear();
|
||||||
|
if (this->rolling_code_enable_) {
|
||||||
|
add(this->data_, ROLLING_CODE_KEY);
|
||||||
|
add(this->data_, this->rolling_code_[0]);
|
||||||
|
add(this->data_, this->rolling_code_[1]);
|
||||||
|
this->increment_code_();
|
||||||
|
} else {
|
||||||
|
add(this->data_, DATA_KEY);
|
||||||
|
}
|
||||||
|
for (auto pkey : this->ping_keys_) {
|
||||||
|
add(this->data_, PING_KEY);
|
||||||
|
add(this->data_, pkey.second);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::flush_() {
|
||||||
|
if (!network::is_connected() || this->data_.empty())
|
||||||
|
return;
|
||||||
|
uint32_t buffer[MAX_PACKET_SIZE / 4];
|
||||||
|
memset(buffer, 0, sizeof buffer);
|
||||||
|
// len must be a multiple of 4
|
||||||
|
auto header_len = round4(this->header_.size()) / 4;
|
||||||
|
auto len = round4(data_.size()) / 4;
|
||||||
|
memcpy(buffer, this->header_.data(), this->header_.size());
|
||||||
|
memcpy(buffer + header_len, this->data_.data(), this->data_.size());
|
||||||
|
if (this->is_encrypted_()) {
|
||||||
|
xxtea_encrypt(buffer + header_len, len, (uint32_t *) this->encryption_key_.data());
|
||||||
|
}
|
||||||
|
auto total_len = (header_len + len) * 4;
|
||||||
|
this->send_packet_(buffer, total_len);
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::add_binary_data_(uint8_t key, const char *id, bool data) {
|
||||||
|
auto len = 1 + 1 + 1 + strlen(id);
|
||||||
|
if (len + this->header_.size() + this->data_.size() > MAX_PACKET_SIZE) {
|
||||||
|
this->flush_();
|
||||||
|
}
|
||||||
|
add(this->data_, key);
|
||||||
|
add(this->data_, (uint8_t) data);
|
||||||
|
add(this->data_, id);
|
||||||
|
}
|
||||||
|
void UDPComponent::add_data_(uint8_t key, const char *id, float data) {
|
||||||
|
FuData udata{.f32 = data};
|
||||||
|
this->add_data_(key, id, udata.u32);
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::add_data_(uint8_t key, const char *id, uint32_t data) {
|
||||||
|
auto len = 4 + 1 + 1 + strlen(id);
|
||||||
|
if (len + this->header_.size() + this->data_.size() > MAX_PACKET_SIZE) {
|
||||||
|
this->flush_();
|
||||||
|
}
|
||||||
|
add(this->data_, key);
|
||||||
|
add(this->data_, data);
|
||||||
|
add(this->data_, id);
|
||||||
|
}
|
||||||
|
void UDPComponent::send_data_(bool all) {
|
||||||
|
if (!this->should_send_ || !network::is_connected())
|
||||||
|
return;
|
||||||
|
this->init_data_();
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
for (auto &sensor : this->sensors_) {
|
||||||
|
if (all || sensor.updated) {
|
||||||
|
sensor.updated = false;
|
||||||
|
this->add_data_(SENSOR_KEY, sensor.id, sensor.sensor->get_state());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
for (auto &sensor : this->binary_sensors_) {
|
||||||
|
if (all || sensor.updated) {
|
||||||
|
sensor.updated = false;
|
||||||
|
this->add_binary_data_(BINARY_SENSOR_KEY, sensor.id, sensor.sensor->state);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
this->flush_();
|
||||||
|
this->updated_ = false;
|
||||||
|
this->resend_data_ = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::update() {
|
||||||
|
this->updated_ = true;
|
||||||
|
this->resend_data_ = this->should_send_;
|
||||||
|
auto now = millis() / 1000;
|
||||||
|
if (this->last_key_time_ + this->ping_pong_recyle_time_ < now) {
|
||||||
|
this->resend_ping_key_ = this->ping_pong_enable_;
|
||||||
|
this->last_key_time_ = now;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::loop() {
|
||||||
|
uint8_t buf[MAX_PACKET_SIZE];
|
||||||
|
if (this->should_listen_) {
|
||||||
|
for (;;) {
|
||||||
|
#if defined(USE_SOCKET_IMPL_BSD_SOCKETS) || defined(USE_SOCKET_IMPL_LWIP_SOCKETS)
|
||||||
|
auto len = this->listen_socket_->read(buf, sizeof(buf));
|
||||||
|
#else
|
||||||
|
auto len = this->udp_client_.parsePacket();
|
||||||
|
if (len > 0)
|
||||||
|
len = this->udp_client_.read(buf, sizeof(buf));
|
||||||
|
#endif
|
||||||
|
if (len > 0) {
|
||||||
|
this->process_(buf, len);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (this->resend_ping_key_)
|
||||||
|
this->send_ping_pong_request_();
|
||||||
|
if (this->updated_) {
|
||||||
|
this->send_data_(this->resend_data_);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::add_key_(const char *name, uint32_t key) {
|
||||||
|
if (!this->is_encrypted_())
|
||||||
|
return;
|
||||||
|
if (this->ping_keys_.count(name) == 0 && this->ping_keys_.size() == MAX_PING_KEYS) {
|
||||||
|
ESP_LOGW(TAG, "Ping key from %s discarded", name);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
this->ping_keys_[name] = key;
|
||||||
|
this->resend_data_ = true;
|
||||||
|
ESP_LOGV(TAG, "Ping key from %s now %X", name, (unsigned) key);
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::process_ping_request_(const char *name, uint8_t *ptr, size_t len) {
|
||||||
|
if (len != 4) {
|
||||||
|
ESP_LOGW(TAG, "Bad ping request");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
auto key = get_uint32(ptr);
|
||||||
|
this->add_key_(name, key);
|
||||||
|
ESP_LOGV(TAG, "Updated ping key for %s to %08X", name, (unsigned) key);
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool process_rolling_code(Provider &provider, uint8_t *&buf, const uint8_t *end) {
|
||||||
|
if (end - buf < 8)
|
||||||
|
return false;
|
||||||
|
auto code0 = get_uint32(buf);
|
||||||
|
auto code1 = get_uint32(buf);
|
||||||
|
if (code1 < provider.last_code[1] || (code1 == provider.last_code[1] && code0 <= provider.last_code[0])) {
|
||||||
|
ESP_LOGW(TAG, "Rolling code for %s %08lX:%08lX is old", provider.name, (unsigned long) code1,
|
||||||
|
(unsigned long) code0);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
provider.last_code[0] = code0;
|
||||||
|
provider.last_code[1] = code1;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Process a received packet
|
||||||
|
*/
|
||||||
|
void UDPComponent::process_(uint8_t *buf, const size_t len) {
|
||||||
|
auto ping_key_seen = !this->ping_pong_enable_;
|
||||||
|
if (len < 8) {
|
||||||
|
return ESP_LOGV(TAG, "Bad length %zu", len);
|
||||||
|
}
|
||||||
|
char namebuf[256]{};
|
||||||
|
uint8_t byte;
|
||||||
|
uint8_t *start_ptr = buf;
|
||||||
|
const uint8_t *end = buf + len;
|
||||||
|
FuData rdata{};
|
||||||
|
auto magic = get_uint16(buf);
|
||||||
|
if (magic != MAGIC_NUMBER && magic != MAGIC_PING)
|
||||||
|
return ESP_LOGV(TAG, "Bad magic %X", magic);
|
||||||
|
|
||||||
|
auto hlen = *buf++;
|
||||||
|
if (hlen > len - 3) {
|
||||||
|
return ESP_LOGV(TAG, "Bad hostname length %u > %zu", hlen, len - 3);
|
||||||
|
}
|
||||||
|
memcpy(namebuf, buf, hlen);
|
||||||
|
if (strcmp(this->name_, namebuf) == 0) {
|
||||||
|
return ESP_LOGV(TAG, "Ignoring our own data");
|
||||||
|
}
|
||||||
|
buf += hlen;
|
||||||
|
if (magic == MAGIC_PING)
|
||||||
|
return this->process_ping_request_(namebuf, buf, end - buf);
|
||||||
|
if (round4(len) != len) {
|
||||||
|
return ESP_LOGW(TAG, "Bad length %zu", len);
|
||||||
|
}
|
||||||
|
hlen = round4(hlen + 3);
|
||||||
|
buf = start_ptr + hlen;
|
||||||
|
if (buf == end) {
|
||||||
|
return ESP_LOGV(TAG, "No data after header");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this->providers_.count(namebuf) == 0) {
|
||||||
|
return ESP_LOGVV(TAG, "Unknown hostname %s", namebuf);
|
||||||
|
}
|
||||||
|
auto &provider = this->providers_[namebuf];
|
||||||
|
// if encryption not used with this host, ping check is pointless since it would be easily spoofed.
|
||||||
|
if (provider.encryption_key.empty())
|
||||||
|
ping_key_seen = true;
|
||||||
|
|
||||||
|
ESP_LOGV(TAG, "Found hostname %s", namebuf);
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
auto &sensors = this->remote_sensors_[namebuf];
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
auto &binary_sensors = this->remote_binary_sensors_[namebuf];
|
||||||
|
#endif
|
||||||
|
|
||||||
|
if (!provider.encryption_key.empty()) {
|
||||||
|
xxtea_decrypt((uint32_t *) buf, (end - buf) / 4, (uint32_t *) provider.encryption_key.data());
|
||||||
|
}
|
||||||
|
byte = *buf++;
|
||||||
|
if (byte == ROLLING_CODE_KEY) {
|
||||||
|
if (!process_rolling_code(provider, buf, end))
|
||||||
|
return;
|
||||||
|
} else if (byte != DATA_KEY) {
|
||||||
|
return ESP_LOGV(TAG, "Expected rolling_key or data_key, got %X", byte);
|
||||||
|
}
|
||||||
|
while (buf < end) {
|
||||||
|
byte = *buf++;
|
||||||
|
if (byte == ZERO_FILL_KEY)
|
||||||
|
continue;
|
||||||
|
if (byte == PING_KEY) {
|
||||||
|
if (end - buf < 4) {
|
||||||
|
return ESP_LOGV(TAG, "PING_KEY requires 4 more bytes");
|
||||||
|
}
|
||||||
|
auto key = get_uint32(buf);
|
||||||
|
if (key == this->ping_key_) {
|
||||||
|
ping_key_seen = true;
|
||||||
|
ESP_LOGV(TAG, "Found good ping key %X", (unsigned) key);
|
||||||
|
} else {
|
||||||
|
ESP_LOGV(TAG, "Unknown ping key %X", (unsigned) key);
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (!ping_key_seen) {
|
||||||
|
ESP_LOGW(TAG, "Ping key not seen");
|
||||||
|
this->resend_ping_key_ = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (byte == BINARY_SENSOR_KEY) {
|
||||||
|
if (end - buf < 3) {
|
||||||
|
return ESP_LOGV(TAG, "Binary sensor key requires at least 3 more bytes");
|
||||||
|
}
|
||||||
|
rdata.u32 = *buf++;
|
||||||
|
} else if (byte == SENSOR_KEY) {
|
||||||
|
if (end - buf < 6) {
|
||||||
|
return ESP_LOGV(TAG, "Sensor key requires at least 6 more bytes");
|
||||||
|
}
|
||||||
|
rdata.u32 = get_uint32(buf);
|
||||||
|
} else {
|
||||||
|
return ESP_LOGW(TAG, "Unknown key byte %X", byte);
|
||||||
|
}
|
||||||
|
|
||||||
|
hlen = *buf++;
|
||||||
|
if (end - buf < hlen) {
|
||||||
|
return ESP_LOGV(TAG, "Name length of %u not available", hlen);
|
||||||
|
}
|
||||||
|
memset(namebuf, 0, sizeof namebuf);
|
||||||
|
memcpy(namebuf, buf, hlen);
|
||||||
|
ESP_LOGV(TAG, "Found sensor key %d, id %s, data %lX", byte, namebuf, (unsigned long) rdata.u32);
|
||||||
|
buf += hlen;
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
if (byte == SENSOR_KEY && sensors.count(namebuf) != 0)
|
||||||
|
sensors[namebuf]->publish_state(rdata.f32);
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
if (byte == BINARY_SENSOR_KEY && binary_sensors.count(namebuf) != 0)
|
||||||
|
binary_sensors[namebuf]->publish_state(rdata.u32 != 0);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::dump_config() {
|
||||||
|
ESP_LOGCONFIG(TAG, "UDP:");
|
||||||
|
ESP_LOGCONFIG(TAG, " Port: %u", this->port_);
|
||||||
|
ESP_LOGCONFIG(TAG, " Encrypted: %s", YESNO(this->is_encrypted_()));
|
||||||
|
ESP_LOGCONFIG(TAG, " Ping-pong: %s", YESNO(this->ping_pong_enable_));
|
||||||
|
for (const auto &address : this->addresses_)
|
||||||
|
ESP_LOGCONFIG(TAG, " Address: %s", address.c_str());
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
for (auto sensor : this->sensors_)
|
||||||
|
ESP_LOGCONFIG(TAG, " Sensor: %s", sensor.id);
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
for (auto sensor : this->binary_sensors_)
|
||||||
|
ESP_LOGCONFIG(TAG, " Binary Sensor: %s", sensor.id);
|
||||||
|
#endif
|
||||||
|
for (const auto &host : this->providers_) {
|
||||||
|
ESP_LOGCONFIG(TAG, " Remote host: %s", host.first.c_str());
|
||||||
|
ESP_LOGCONFIG(TAG, " Encrypted: %s", YESNO(!host.second.encryption_key.empty()));
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
for (const auto &sensor : this->remote_sensors_[host.first.c_str()])
|
||||||
|
ESP_LOGCONFIG(TAG, " Sensor: %s", sensor.first.c_str());
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
for (const auto &sensor : this->remote_binary_sensors_[host.first.c_str()])
|
||||||
|
ESP_LOGCONFIG(TAG, " Binary Sensor: %s", sensor.first.c_str());
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void UDPComponent::increment_code_() {
|
||||||
|
if (this->rolling_code_enable_) {
|
||||||
|
if (++this->rolling_code_[0] == 0) {
|
||||||
|
this->rolling_code_[1]++;
|
||||||
|
this->pref_.save(&this->rolling_code_[1]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void UDPComponent::send_packet_(void *data, size_t len) {
|
||||||
|
#if defined(USE_SOCKET_IMPL_BSD_SOCKETS) || defined(USE_SOCKET_IMPL_LWIP_SOCKETS)
|
||||||
|
for (const auto &saddr : this->sockaddrs_) {
|
||||||
|
auto result = this->broadcast_socket_->sendto(data, len, 0, &saddr, sizeof(saddr));
|
||||||
|
if (result < 0)
|
||||||
|
ESP_LOGW(TAG, "sendto() error %d", errno);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
auto iface = IPAddress(0, 0, 0, 0);
|
||||||
|
for (const auto &saddr : this->ipaddrs_) {
|
||||||
|
if (this->udp_client_.beginPacketMulticast(saddr, this->port_, iface, 128) != 0) {
|
||||||
|
this->udp_client_.write((const uint8_t *) data, len);
|
||||||
|
auto result = this->udp_client_.endPacket();
|
||||||
|
if (result == 0)
|
||||||
|
ESP_LOGW(TAG, "udp.write() error");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void UDPComponent::send_ping_pong_request_() {
|
||||||
|
if (!this->ping_pong_enable_ || !network::is_connected())
|
||||||
|
return;
|
||||||
|
this->ping_key_ = random_uint32();
|
||||||
|
this->ping_header_.clear();
|
||||||
|
add(this->ping_header_, MAGIC_PING);
|
||||||
|
add(this->ping_header_, this->name_);
|
||||||
|
add(this->ping_header_, this->ping_key_);
|
||||||
|
this->send_packet_(this->ping_header_.data(), this->ping_header_.size());
|
||||||
|
this->resend_ping_key_ = false;
|
||||||
|
ESP_LOGV(TAG, "Sent new ping request %08X", (unsigned) this->ping_key_);
|
||||||
|
}
|
||||||
|
} // namespace udp
|
||||||
|
} // namespace esphome
|
158
esphome/components/udp/udp_component.h
Normal file
158
esphome/components/udp/udp_component.h
Normal file
|
@ -0,0 +1,158 @@
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "esphome/core/component.h"
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
#include "esphome/components/sensor/sensor.h"
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
#include "esphome/components/binary_sensor/binary_sensor.h"
|
||||||
|
#endif
|
||||||
|
#if defined(USE_SOCKET_IMPL_BSD_SOCKETS) || defined(USE_SOCKET_IMPL_LWIP_SOCKETS)
|
||||||
|
#include "esphome/components/socket/socket.h"
|
||||||
|
#else
|
||||||
|
#include <WiFiUdp.h>
|
||||||
|
#endif
|
||||||
|
#include <vector>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
namespace esphome {
|
||||||
|
namespace udp {
|
||||||
|
|
||||||
|
struct Provider {
|
||||||
|
std::vector<uint8_t> encryption_key;
|
||||||
|
const char *name;
|
||||||
|
uint32_t last_code[2];
|
||||||
|
};
|
||||||
|
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
struct Sensor {
|
||||||
|
sensor::Sensor *sensor;
|
||||||
|
const char *id;
|
||||||
|
bool updated;
|
||||||
|
};
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
struct BinarySensor {
|
||||||
|
binary_sensor::BinarySensor *sensor;
|
||||||
|
const char *id;
|
||||||
|
bool updated;
|
||||||
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
|
class UDPComponent : public PollingComponent {
|
||||||
|
public:
|
||||||
|
void setup() override;
|
||||||
|
void loop() override;
|
||||||
|
void update() override;
|
||||||
|
void dump_config() override;
|
||||||
|
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
void add_sensor(const char *id, sensor::Sensor *sensor) {
|
||||||
|
Sensor st{sensor, id, true};
|
||||||
|
this->sensors_.push_back(st);
|
||||||
|
}
|
||||||
|
void add_remote_sensor(const char *hostname, const char *remote_id, sensor::Sensor *sensor) {
|
||||||
|
this->add_provider(hostname);
|
||||||
|
this->remote_sensors_[hostname][remote_id] = sensor;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
void add_binary_sensor(const char *id, binary_sensor::BinarySensor *sensor) {
|
||||||
|
BinarySensor st{sensor, id, true};
|
||||||
|
this->binary_sensors_.push_back(st);
|
||||||
|
}
|
||||||
|
|
||||||
|
void add_remote_binary_sensor(const char *hostname, const char *remote_id, binary_sensor::BinarySensor *sensor) {
|
||||||
|
this->add_provider(hostname);
|
||||||
|
this->remote_binary_sensors_[hostname][remote_id] = sensor;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
void add_address(const char *addr) { this->addresses_.emplace_back(addr); }
|
||||||
|
void set_port(uint16_t port) { this->port_ = port; }
|
||||||
|
float get_setup_priority() const override { return setup_priority::AFTER_WIFI; }
|
||||||
|
|
||||||
|
void add_provider(const char *hostname) {
|
||||||
|
if (this->providers_.count(hostname) == 0) {
|
||||||
|
Provider provider;
|
||||||
|
provider.encryption_key = std::vector<uint8_t>{};
|
||||||
|
provider.last_code[0] = 0;
|
||||||
|
provider.last_code[1] = 0;
|
||||||
|
provider.name = hostname;
|
||||||
|
this->providers_[hostname] = provider;
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
this->remote_sensors_[hostname] = std::map<std::string, sensor::Sensor *>();
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
this->remote_binary_sensors_[hostname] = std::map<std::string, binary_sensor::BinarySensor *>();
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void set_encryption_key(std::vector<uint8_t> key) { this->encryption_key_ = std::move(key); }
|
||||||
|
void set_rolling_code_enable(bool enable) { this->rolling_code_enable_ = enable; }
|
||||||
|
void set_ping_pong_enable(bool enable) { this->ping_pong_enable_ = enable; }
|
||||||
|
void set_ping_pong_recycle_time(uint32_t recycle_time) { this->ping_pong_recyle_time_ = recycle_time; }
|
||||||
|
void set_provider_encryption(const char *name, std::vector<uint8_t> key) {
|
||||||
|
this->providers_[name].encryption_key = std::move(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
void send_data_(bool all);
|
||||||
|
void process_(uint8_t *buf, size_t len);
|
||||||
|
void flush_();
|
||||||
|
void add_data_(uint8_t key, const char *id, float data);
|
||||||
|
void add_data_(uint8_t key, const char *id, uint32_t data);
|
||||||
|
void increment_code_();
|
||||||
|
void add_binary_data_(uint8_t key, const char *id, bool data);
|
||||||
|
void init_data_();
|
||||||
|
|
||||||
|
bool updated_{};
|
||||||
|
uint16_t port_{18511};
|
||||||
|
uint32_t ping_key_{};
|
||||||
|
uint32_t rolling_code_[2]{};
|
||||||
|
bool rolling_code_enable_{};
|
||||||
|
bool ping_pong_enable_{};
|
||||||
|
uint32_t ping_pong_recyle_time_{};
|
||||||
|
uint32_t last_key_time_{};
|
||||||
|
bool resend_ping_key_{};
|
||||||
|
bool resend_data_{};
|
||||||
|
bool should_send_{};
|
||||||
|
const char *name_{};
|
||||||
|
bool should_listen_{};
|
||||||
|
ESPPreferenceObject pref_;
|
||||||
|
|
||||||
|
#if defined(USE_SOCKET_IMPL_BSD_SOCKETS) || defined(USE_SOCKET_IMPL_LWIP_SOCKETS)
|
||||||
|
std::unique_ptr<socket::Socket> broadcast_socket_ = nullptr;
|
||||||
|
std::unique_ptr<socket::Socket> listen_socket_ = nullptr;
|
||||||
|
std::vector<struct sockaddr> sockaddrs_{};
|
||||||
|
#else
|
||||||
|
std::vector<IPAddress> ipaddrs_{};
|
||||||
|
WiFiUDP udp_client_{};
|
||||||
|
#endif
|
||||||
|
std::vector<uint8_t> encryption_key_{};
|
||||||
|
std::vector<std::string> addresses_{};
|
||||||
|
|
||||||
|
#ifdef USE_SENSOR
|
||||||
|
std::vector<Sensor> sensors_{};
|
||||||
|
std::map<std::string, std::map<std::string, sensor::Sensor *>> remote_sensors_{};
|
||||||
|
#endif
|
||||||
|
#ifdef USE_BINARY_SENSOR
|
||||||
|
std::vector<BinarySensor> binary_sensors_{};
|
||||||
|
std::map<std::string, std::map<std::string, binary_sensor::BinarySensor *>> remote_binary_sensors_{};
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::map<std::string, Provider> providers_{};
|
||||||
|
std::vector<uint8_t> ping_header_{};
|
||||||
|
std::vector<uint8_t> header_{};
|
||||||
|
std::vector<uint8_t> data_{};
|
||||||
|
std::map<const char *, uint32_t> ping_keys_{};
|
||||||
|
void add_key_(const char *name, uint32_t key);
|
||||||
|
void send_ping_pong_request_();
|
||||||
|
void send_packet_(void *data, size_t len);
|
||||||
|
void process_ping_request_(const char *name, uint8_t *ptr, size_t len);
|
||||||
|
|
||||||
|
inline bool is_encrypted_() { return !this->encryption_key_.empty(); }
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace udp
|
||||||
|
} // namespace esphome
|
|
@ -1,35 +1,36 @@
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import gzip
|
import gzip
|
||||||
|
|
||||||
import esphome.codegen as cg
|
import esphome.codegen as cg
|
||||||
import esphome.config_validation as cv
|
|
||||||
import esphome.final_validate as fv
|
|
||||||
from esphome.components import web_server_base
|
from esphome.components import web_server_base
|
||||||
from esphome.components.web_server_base import CONF_WEB_SERVER_BASE_ID
|
from esphome.components.web_server_base import CONF_WEB_SERVER_BASE_ID
|
||||||
|
import esphome.config_validation as cv
|
||||||
from esphome.const import (
|
from esphome.const import (
|
||||||
|
CONF_AUTH,
|
||||||
CONF_CSS_INCLUDE,
|
CONF_CSS_INCLUDE,
|
||||||
CONF_CSS_URL,
|
CONF_CSS_URL,
|
||||||
|
CONF_ENABLE_PRIVATE_NETWORK_ACCESS,
|
||||||
CONF_ID,
|
CONF_ID,
|
||||||
|
CONF_INCLUDE_INTERNAL,
|
||||||
CONF_JS_INCLUDE,
|
CONF_JS_INCLUDE,
|
||||||
CONF_JS_URL,
|
CONF_JS_URL,
|
||||||
CONF_ENABLE_PRIVATE_NETWORK_ACCESS,
|
|
||||||
CONF_PORT,
|
|
||||||
CONF_AUTH,
|
|
||||||
CONF_USERNAME,
|
|
||||||
CONF_PASSWORD,
|
|
||||||
CONF_INCLUDE_INTERNAL,
|
|
||||||
CONF_OTA,
|
|
||||||
CONF_LOG,
|
|
||||||
CONF_VERSION,
|
|
||||||
CONF_LOCAL,
|
CONF_LOCAL,
|
||||||
|
CONF_LOG,
|
||||||
|
CONF_OTA,
|
||||||
|
CONF_PASSWORD,
|
||||||
|
CONF_PORT,
|
||||||
|
CONF_USERNAME,
|
||||||
|
CONF_VERSION,
|
||||||
CONF_WEB_SERVER_ID,
|
CONF_WEB_SERVER_ID,
|
||||||
CONF_WEB_SERVER_SORTING_WEIGHT,
|
CONF_WEB_SERVER_SORTING_WEIGHT,
|
||||||
|
PLATFORM_BK72XX,
|
||||||
PLATFORM_ESP32,
|
PLATFORM_ESP32,
|
||||||
PLATFORM_ESP8266,
|
PLATFORM_ESP8266,
|
||||||
PLATFORM_BK72XX,
|
|
||||||
PLATFORM_RTL87XX,
|
PLATFORM_RTL87XX,
|
||||||
)
|
)
|
||||||
from esphome.core import CORE, coroutine_with_priority
|
from esphome.core import CORE, coroutine_with_priority
|
||||||
|
import esphome.final_validate as fv
|
||||||
|
|
||||||
AUTO_LOAD = ["json", "web_server_base"]
|
AUTO_LOAD = ["json", "web_server_base"]
|
||||||
|
|
||||||
|
@ -208,7 +209,6 @@ async def to_code(config):
|
||||||
var = cg.new_Pvariable(config[CONF_ID], paren)
|
var = cg.new_Pvariable(config[CONF_ID], paren)
|
||||||
await cg.register_component(var, config)
|
await cg.register_component(var, config)
|
||||||
|
|
||||||
cg.add_define("USE_WEBSERVER")
|
|
||||||
version = config[CONF_VERSION]
|
version = config[CONF_VERSION]
|
||||||
|
|
||||||
cg.add(paren.set_port(config[CONF_PORT]))
|
cg.add(paren.set_port(config[CONF_PORT]))
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
#include "list_entities.h"
|
#include "list_entities.h"
|
||||||
|
#ifdef USE_WEBSERVER
|
||||||
#include "esphome/core/application.h"
|
#include "esphome/core/application.h"
|
||||||
#include "esphome/core/log.h"
|
#include "esphome/core/log.h"
|
||||||
#include "esphome/core/util.h"
|
#include "esphome/core/util.h"
|
||||||
|
@ -188,3 +189,4 @@ bool ListEntitiesIterator::on_update(update::UpdateEntity *update) {
|
||||||
|
|
||||||
} // namespace web_server
|
} // namespace web_server
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
#endif
|
||||||
|
|
|
@ -1,8 +1,9 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
#include "esphome/core/defines.h"
|
||||||
|
#ifdef USE_WEBSERVER
|
||||||
#include "esphome/core/component.h"
|
#include "esphome/core/component.h"
|
||||||
#include "esphome/core/component_iterator.h"
|
#include "esphome/core/component_iterator.h"
|
||||||
#include "esphome/core/defines.h"
|
|
||||||
namespace esphome {
|
namespace esphome {
|
||||||
namespace web_server {
|
namespace web_server {
|
||||||
|
|
||||||
|
@ -78,3 +79,4 @@ class ListEntitiesIterator : public ComponentIterator {
|
||||||
|
|
||||||
} // namespace web_server
|
} // namespace web_server
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
#endif
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
#include "web_server.h"
|
#include "web_server.h"
|
||||||
|
#ifdef USE_WEBSERVER
|
||||||
#include "esphome/components/json/json_util.h"
|
#include "esphome/components/json/json_util.h"
|
||||||
#include "esphome/components/network/util.h"
|
#include "esphome/components/network/util.h"
|
||||||
#include "esphome/core/application.h"
|
#include "esphome/core/application.h"
|
||||||
|
@ -1659,3 +1659,4 @@ void WebServer::schedule_(std::function<void()> &&f) {
|
||||||
|
|
||||||
} // namespace web_server
|
} // namespace web_server
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
#endif
|
||||||
|
|
|
@ -3,6 +3,7 @@
|
||||||
#include "list_entities.h"
|
#include "list_entities.h"
|
||||||
|
|
||||||
#include "esphome/components/web_server_base/web_server_base.h"
|
#include "esphome/components/web_server_base/web_server_base.h"
|
||||||
|
#ifdef USE_WEBSERVER
|
||||||
#include "esphome/core/component.h"
|
#include "esphome/core/component.h"
|
||||||
#include "esphome/core/controller.h"
|
#include "esphome/core/controller.h"
|
||||||
#include "esphome/core/entity_base.h"
|
#include "esphome/core/entity_base.h"
|
||||||
|
@ -366,3 +367,4 @@ class WebServer : public Controller, public Component, public AsyncWebHandler {
|
||||||
|
|
||||||
} // namespace web_server
|
} // namespace web_server
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
#endif
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
#include "web_server_base.h"
|
#include "web_server_base.h"
|
||||||
|
#ifdef USE_NETWORK
|
||||||
#include "esphome/core/log.h"
|
#include "esphome/core/log.h"
|
||||||
#include "esphome/core/application.h"
|
#include "esphome/core/application.h"
|
||||||
#include "esphome/core/helpers.h"
|
#include "esphome/core/helpers.h"
|
||||||
|
@ -121,3 +122,4 @@ float WebServerBase::get_setup_priority() const {
|
||||||
|
|
||||||
} // namespace web_server_base
|
} // namespace web_server_base
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
#endif
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
#include "esphome/core/defines.h"
|
||||||
|
#ifdef USE_NETWORK
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
@ -145,3 +146,4 @@ class OTARequestHandler : public AsyncWebHandler {
|
||||||
|
|
||||||
} // namespace web_server_base
|
} // namespace web_server_base
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
#endif
|
||||||
|
|
|
@ -431,6 +431,7 @@ CONF_LIGHT_ID = "light_id"
|
||||||
CONF_LIGHTNING_ENERGY = "lightning_energy"
|
CONF_LIGHTNING_ENERGY = "lightning_energy"
|
||||||
CONF_LIGHTNING_THRESHOLD = "lightning_threshold"
|
CONF_LIGHTNING_THRESHOLD = "lightning_threshold"
|
||||||
CONF_LIMIT_MODE = "limit_mode"
|
CONF_LIMIT_MODE = "limit_mode"
|
||||||
|
CONF_LINE_FREQUENCY = "line_frequency"
|
||||||
CONF_LINE_THICKNESS = "line_thickness"
|
CONF_LINE_THICKNESS = "line_thickness"
|
||||||
CONF_LINE_TYPE = "line_type"
|
CONF_LINE_TYPE = "line_type"
|
||||||
CONF_LOADED_INTEGRATIONS = "loaded_integrations"
|
CONF_LOADED_INTEGRATIONS = "loaded_integrations"
|
||||||
|
@ -729,6 +730,7 @@ CONF_RW_PIN = "rw_pin"
|
||||||
CONF_RX_BUFFER_SIZE = "rx_buffer_size"
|
CONF_RX_BUFFER_SIZE = "rx_buffer_size"
|
||||||
CONF_RX_ONLY = "rx_only"
|
CONF_RX_ONLY = "rx_only"
|
||||||
CONF_RX_PIN = "rx_pin"
|
CONF_RX_PIN = "rx_pin"
|
||||||
|
CONF_RX_QUEUE_LEN = "rx_queue_len"
|
||||||
CONF_SAFE_MODE = "safe_mode"
|
CONF_SAFE_MODE = "safe_mode"
|
||||||
CONF_SAMPLE_RATE = "sample_rate"
|
CONF_SAMPLE_RATE = "sample_rate"
|
||||||
CONF_SAMSUNG = "samsung"
|
CONF_SAMSUNG = "samsung"
|
||||||
|
@ -880,6 +882,7 @@ CONF_TVOC = "tvoc"
|
||||||
CONF_TX_BUFFER_SIZE = "tx_buffer_size"
|
CONF_TX_BUFFER_SIZE = "tx_buffer_size"
|
||||||
CONF_TX_PIN = "tx_pin"
|
CONF_TX_PIN = "tx_pin"
|
||||||
CONF_TX_POWER = "tx_power"
|
CONF_TX_POWER = "tx_power"
|
||||||
|
CONF_TX_QUEUE_LEN = "tx_queue_len"
|
||||||
CONF_TYPE = "type"
|
CONF_TYPE = "type"
|
||||||
CONF_TYPE_ID = "type_id"
|
CONF_TYPE_ID = "type_id"
|
||||||
CONF_UART_ID = "uart_id"
|
CONF_UART_ID = "uart_id"
|
||||||
|
|
|
@ -1,19 +1,67 @@
|
||||||
#include "bytebuffer.h"
|
#include "bytebuffer.h"
|
||||||
#include <cassert>
|
#include <cassert>
|
||||||
|
#include "esphome/core/helpers.h"
|
||||||
|
|
||||||
|
#include <list>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
namespace esphome {
|
namespace esphome {
|
||||||
|
|
||||||
ByteBuffer ByteBuffer::create(size_t capacity) {
|
ByteBuffer ByteBuffer::wrap(const uint8_t *ptr, size_t len, Endian endianness) {
|
||||||
std::vector<uint8_t> data(capacity);
|
// there is a double copy happening here, could be optimized but at cost of clarity.
|
||||||
return {data};
|
|
||||||
}
|
|
||||||
|
|
||||||
ByteBuffer ByteBuffer::wrap(uint8_t *ptr, size_t len) {
|
|
||||||
std::vector<uint8_t> data(ptr, ptr + len);
|
std::vector<uint8_t> data(ptr, ptr + len);
|
||||||
return {data};
|
ByteBuffer buffer = {data};
|
||||||
|
buffer.endianness_ = endianness;
|
||||||
|
return buffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
ByteBuffer ByteBuffer::wrap(std::vector<uint8_t> data) { return {std::move(data)}; }
|
ByteBuffer ByteBuffer::wrap(std::vector<uint8_t> const &data, Endian endianness) {
|
||||||
|
ByteBuffer buffer = {data};
|
||||||
|
buffer.endianness_ = endianness;
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ByteBuffer ByteBuffer::wrap(uint8_t value) {
|
||||||
|
ByteBuffer buffer = ByteBuffer(1);
|
||||||
|
buffer.put_uint8(value);
|
||||||
|
buffer.flip();
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ByteBuffer ByteBuffer::wrap(uint16_t value, Endian endianness) {
|
||||||
|
ByteBuffer buffer = ByteBuffer(2, endianness);
|
||||||
|
buffer.put_uint16(value);
|
||||||
|
buffer.flip();
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ByteBuffer ByteBuffer::wrap(uint32_t value, Endian endianness) {
|
||||||
|
ByteBuffer buffer = ByteBuffer(4, endianness);
|
||||||
|
buffer.put_uint32(value);
|
||||||
|
buffer.flip();
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ByteBuffer ByteBuffer::wrap(uint64_t value, Endian endianness) {
|
||||||
|
ByteBuffer buffer = ByteBuffer(8, endianness);
|
||||||
|
buffer.put_uint64(value);
|
||||||
|
buffer.flip();
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ByteBuffer ByteBuffer::wrap(float value, Endian endianness) {
|
||||||
|
ByteBuffer buffer = ByteBuffer(sizeof(float), endianness);
|
||||||
|
buffer.put_float(value);
|
||||||
|
buffer.flip();
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ByteBuffer ByteBuffer::wrap(double value, Endian endianness) {
|
||||||
|
ByteBuffer buffer = ByteBuffer(sizeof(double), endianness);
|
||||||
|
buffer.put_double(value);
|
||||||
|
buffer.flip();
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
void ByteBuffer::set_limit(size_t limit) {
|
void ByteBuffer::set_limit(size_t limit) {
|
||||||
assert(limit <= this->get_capacity());
|
assert(limit <= this->get_capacity());
|
||||||
|
@ -27,108 +75,93 @@ void ByteBuffer::clear() {
|
||||||
this->limit_ = this->get_capacity();
|
this->limit_ = this->get_capacity();
|
||||||
this->position_ = 0;
|
this->position_ = 0;
|
||||||
}
|
}
|
||||||
uint16_t ByteBuffer::get_uint16() {
|
void ByteBuffer::flip() {
|
||||||
assert(this->get_remaining() >= 2);
|
this->limit_ = this->position_;
|
||||||
uint16_t value;
|
this->position_ = 0;
|
||||||
if (endianness_ == LITTLE) {
|
|
||||||
value = this->data_[this->position_++];
|
|
||||||
value |= this->data_[this->position_++] << 8;
|
|
||||||
} else {
|
|
||||||
value = this->data_[this->position_++] << 8;
|
|
||||||
value |= this->data_[this->position_++];
|
|
||||||
}
|
|
||||||
return value;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
uint32_t ByteBuffer::get_uint32() {
|
/// Getters
|
||||||
assert(this->get_remaining() >= 4);
|
|
||||||
uint32_t value;
|
|
||||||
if (endianness_ == LITTLE) {
|
|
||||||
value = this->data_[this->position_++];
|
|
||||||
value |= this->data_[this->position_++] << 8;
|
|
||||||
value |= this->data_[this->position_++] << 16;
|
|
||||||
value |= this->data_[this->position_++] << 24;
|
|
||||||
} else {
|
|
||||||
value = this->data_[this->position_++] << 24;
|
|
||||||
value |= this->data_[this->position_++] << 16;
|
|
||||||
value |= this->data_[this->position_++] << 8;
|
|
||||||
value |= this->data_[this->position_++];
|
|
||||||
}
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
uint32_t ByteBuffer::get_uint24() {
|
|
||||||
assert(this->get_remaining() >= 3);
|
|
||||||
uint32_t value;
|
|
||||||
if (endianness_ == LITTLE) {
|
|
||||||
value = this->data_[this->position_++];
|
|
||||||
value |= this->data_[this->position_++] << 8;
|
|
||||||
value |= this->data_[this->position_++] << 16;
|
|
||||||
} else {
|
|
||||||
value = this->data_[this->position_++] << 16;
|
|
||||||
value |= this->data_[this->position_++] << 8;
|
|
||||||
value |= this->data_[this->position_++];
|
|
||||||
}
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
uint32_t ByteBuffer::get_int24() {
|
|
||||||
auto value = this->get_uint24();
|
|
||||||
uint32_t mask = (~(uint32_t) 0) << 23;
|
|
||||||
if ((value & mask) != 0)
|
|
||||||
value |= mask;
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
uint8_t ByteBuffer::get_uint8() {
|
uint8_t ByteBuffer::get_uint8() {
|
||||||
assert(this->get_remaining() >= 1);
|
assert(this->get_remaining() >= 1);
|
||||||
return this->data_[this->position_++];
|
return this->data_[this->position_++];
|
||||||
}
|
}
|
||||||
float ByteBuffer::get_float() {
|
uint64_t ByteBuffer::get_uint(size_t length) {
|
||||||
auto value = this->get_uint32();
|
assert(this->get_remaining() >= length);
|
||||||
return *(float *) &value;
|
uint64_t value = 0;
|
||||||
|
if (this->endianness_ == LITTLE) {
|
||||||
|
this->position_ += length;
|
||||||
|
auto index = this->position_;
|
||||||
|
while (length-- != 0) {
|
||||||
|
value <<= 8;
|
||||||
|
value |= this->data_[--index];
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
while (length-- != 0) {
|
||||||
|
value <<= 8;
|
||||||
|
value |= this->data_[this->position_++];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return value;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
uint32_t ByteBuffer::get_int24() {
|
||||||
|
auto value = this->get_uint24();
|
||||||
|
uint32_t mask = (~static_cast<uint32_t>(0)) << 23;
|
||||||
|
if ((value & mask) != 0)
|
||||||
|
value |= mask;
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
float ByteBuffer::get_float() {
|
||||||
|
assert(this->get_remaining() >= sizeof(float));
|
||||||
|
return bit_cast<float>(this->get_uint32());
|
||||||
|
}
|
||||||
|
double ByteBuffer::get_double() {
|
||||||
|
assert(this->get_remaining() >= sizeof(double));
|
||||||
|
return bit_cast<double>(this->get_uint64());
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<uint8_t> ByteBuffer::get_vector(size_t length) {
|
||||||
|
assert(this->get_remaining() >= length);
|
||||||
|
auto start = this->data_.begin() + this->position_;
|
||||||
|
this->position_ += length;
|
||||||
|
return {start, start + length};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Putters
|
||||||
void ByteBuffer::put_uint8(uint8_t value) {
|
void ByteBuffer::put_uint8(uint8_t value) {
|
||||||
assert(this->get_remaining() >= 1);
|
assert(this->get_remaining() >= 1);
|
||||||
this->data_[this->position_++] = value;
|
this->data_[this->position_++] = value;
|
||||||
}
|
}
|
||||||
|
|
||||||
void ByteBuffer::put_uint16(uint16_t value) {
|
void ByteBuffer::put_uint(uint64_t value, size_t length) {
|
||||||
assert(this->get_remaining() >= 2);
|
assert(this->get_remaining() >= length);
|
||||||
if (this->endianness_ == LITTLE) {
|
if (this->endianness_ == LITTLE) {
|
||||||
this->data_[this->position_++] = (uint8_t) value;
|
while (length-- != 0) {
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 8);
|
this->data_[this->position_++] = static_cast<uint8_t>(value);
|
||||||
|
value >>= 8;
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 8);
|
this->position_ += length;
|
||||||
this->data_[this->position_++] = (uint8_t) value;
|
auto index = this->position_;
|
||||||
|
while (length-- != 0) {
|
||||||
|
this->data_[--index] = static_cast<uint8_t>(value);
|
||||||
|
value >>= 8;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
void ByteBuffer::put_uint24(uint32_t value) {
|
void ByteBuffer::put_float(float value) {
|
||||||
assert(this->get_remaining() >= 3);
|
static_assert(sizeof(float) == sizeof(uint32_t), "Float sizes other than 32 bit not supported");
|
||||||
if (this->endianness_ == LITTLE) {
|
assert(this->get_remaining() >= sizeof(float));
|
||||||
this->data_[this->position_++] = (uint8_t) value;
|
this->put_uint32(bit_cast<uint32_t>(value));
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 8);
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 16);
|
|
||||||
} else {
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 16);
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 8);
|
|
||||||
this->data_[this->position_++] = (uint8_t) value;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
void ByteBuffer::put_uint32(uint32_t value) {
|
void ByteBuffer::put_double(double value) {
|
||||||
assert(this->get_remaining() >= 4);
|
static_assert(sizeof(double) == sizeof(uint64_t), "Double sizes other than 64 bit not supported");
|
||||||
if (this->endianness_ == LITTLE) {
|
assert(this->get_remaining() >= sizeof(double));
|
||||||
this->data_[this->position_++] = (uint8_t) value;
|
this->put_uint64(bit_cast<uint64_t>(value));
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 8);
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 16);
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 24);
|
|
||||||
} else {
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 24);
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 16);
|
|
||||||
this->data_[this->position_++] = (uint8_t) (value >> 8);
|
|
||||||
this->data_[this->position_++] = (uint8_t) value;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
void ByteBuffer::put_float(float value) { this->put_uint32(*(uint32_t *) &value); }
|
void ByteBuffer::put_vector(const std::vector<uint8_t> &value) {
|
||||||
void ByteBuffer::flip() {
|
assert(this->get_remaining() >= value.size());
|
||||||
this->limit_ = this->position_;
|
std::copy(value.begin(), value.end(), this->data_.begin() + this->position_);
|
||||||
this->position_ = 0;
|
this->position_ += value.size();
|
||||||
}
|
}
|
||||||
} // namespace esphome
|
} // namespace esphome
|
||||||
|
|
Some files were not shown because too many files have changed in this diff Show more
Loading…
Reference in a new issue