mirror of
				https://github.com/esphome/esphome.git
				synced 2025-11-03 16:41:50 +00:00 
			
		
		
		
	Compare commits
	
		
			122 Commits
		
	
	
		
			jesserockz
			...
			2024.9.0b3
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					8bd182d96c | ||
| 
						 | 
					a63b9a9e0c | ||
| 
						 | 
					6483ceb6eb | ||
| 
						 | 
					e7fe9b374f | ||
| 
						 | 
					3e7161ad41 | ||
| 
						 | 
					5d8fb7cdf4 | ||
| 
						 | 
					80e3de94d3 | ||
| 
						 | 
					7f00b5eb65 | ||
| 
						 | 
					9ed2ed8177 | ||
| 
						 | 
					63cda412f9 | ||
| 
						 | 
					625726c650 | ||
| 
						 | 
					955a909846 | ||
| 
						 | 
					e3ae8cd31e | ||
| 
						 | 
					04248b6840 | ||
| 
						 | 
					dbecade122 | ||
| 
						 | 
					ffc2b58714 | ||
| 
						 | 
					39ad358b51 | ||
| 
						 | 
					7b90bfaec6 | ||
| 
						 | 
					7abbb0fb97 | ||
| 
						 | 
					de7d2c33e1 | ||
| 
						 | 
					c8aed15157 | ||
| 
						 | 
					dcfad31770 | ||
| 
						 | 
					f5c2921b85 | ||
| 
						 | 
					b5e5741ffd | ||
| 
						 | 
					d10feafa9b | ||
| 
						 | 
					9f42b76de3 | ||
| 
						 | 
					198bd3b41a | ||
| 
						 | 
					c90dcfc0ca | ||
| 
						 | 
					7a93dde5d4 | ||
| 
						 | 
					32995a352b | ||
| 
						 | 
					9722876ef6 | ||
| 
						 | 
					8bd46a43b9 | ||
| 
						 | 
					18a1191e03 | ||
| 
						 | 
					1548fa0811 | ||
| 
						 | 
					b496233425 | ||
| 
						 | 
					dc4e60526c | ||
| 
						 | 
					71a7f6383f | ||
| 
						 | 
					e882cea47e | ||
| 
						 | 
					a7fd3b34aa | ||
| 
						 | 
					a96de54d46 | ||
| 
						 | 
					188faa6530 | ||
| 
						 | 
					1a71cc3047 | ||
| 
						 | 
					10ccc5f125 | ||
| 
						 | 
					c6e64a9ed3 | ||
| 
						 | 
					565835cb59 | ||
| 
						 | 
					f2d8ab62dd | ||
| 
						 | 
					cb4bede6d8 | ||
| 
						 | 
					39b2f30b16 | ||
| 
						 | 
					e18ae84031 | ||
| 
						 | 
					5cb1d18574 | ||
| 
						 | 
					f8ec5242c9 | ||
| 
						 | 
					d6eeac0619 | ||
| 
						 | 
					3b14b0efce | ||
| 
						 | 
					29f0b504b9 | ||
| 
						 | 
					01c50432c9 | ||
| 
						 | 
					e5e06a12ef | ||
| 
						 | 
					c9c5ca28d2 | ||
| 
						 | 
					04ec6c5677 | ||
| 
						 | 
					816b060edc | ||
| 
						 | 
					91c7c43682 | ||
| 
						 | 
					ca8e45cf4c | ||
| 
						 | 
					854bafbd4a | ||
| 
						 | 
					094c867fba | ||
| 
						 | 
					fc930327b4 | ||
| 
						 | 
					6490fc9c62 | ||
| 
						 | 
					3a7aabb2eb | ||
| 
						 | 
					61223a3cc9 | ||
| 
						 | 
					ca2f25e73b | ||
| 
						 | 
					ba6963cf72 | ||
| 
						 | 
					721b532d71 | ||
| 
						 | 
					a5d46ae9e5 | ||
| 
						 | 
					f8e8bd2c24 | ||
| 
						 | 
					69f98e0f87 | ||
| 
						 | 
					87d801721b | ||
| 
						 | 
					d754bdde1b | ||
| 
						 | 
					725e50348b | ||
| 
						 | 
					acb00c9c59 | ||
| 
						 | 
					bb6693a255 | ||
| 
						 | 
					c09df3c05d | ||
| 
						 | 
					f28418d0b4 | ||
| 
						 | 
					1922f2bbee | ||
| 
						 | 
					0375072bdf | ||
| 
						 | 
					b3f03c07c6 | ||
| 
						 | 
					4b2032a98e | ||
| 
						 | 
					d6df466237 | ||
| 
						 | 
					458a8970b6 | ||
| 
						 | 
					8b6c95f723 | ||
| 
						 | 
					28eda4b220 | ||
| 
						 | 
					9975e8b544 | ||
| 
						 | 
					c1774c42c2 | ||
| 
						 | 
					8677763492 | ||
| 
						 | 
					388abaf09f | ||
| 
						 | 
					92ae506ffb | ||
| 
						 | 
					34cce0e920 | ||
| 
						 | 
					7e18a5c44f | ||
| 
						 | 
					5a707b558d | ||
| 
						 | 
					e10f8128c8 | ||
| 
						 | 
					0f2064193f | ||
| 
						 | 
					dc9c001056 | ||
| 
						 | 
					60fced53c2 | ||
| 
						 | 
					71d6bbc7e6 | ||
| 
						 | 
					caaae59ea9 | ||
| 
						 | 
					a01fea54a0 | ||
| 
						 | 
					43f8f2fd2e | ||
| 
						 | 
					3c65cabe1d | ||
| 
						 | 
					5cc8dbace4 | ||
| 
						 | 
					ab620acd4f | ||
| 
						 | 
					11e155d866 | ||
| 
						 | 
					68272c39c0 | ||
| 
						 | 
					da72bae94a | ||
| 
						 | 
					1f21e419aa | ||
| 
						 | 
					5d4bf5f8e5 | ||
| 
						 | 
					b5a6d3aa9d | ||
| 
						 | 
					813d517076 | ||
| 
						 | 
					4ed6a64869 | ||
| 
						 | 
					aaae8f4a87 | ||
| 
						 | 
					436c6282da | ||
| 
						 | 
					c043bbe598 | ||
| 
						 | 
					8fae609316 | ||
| 
						 | 
					848fd0442d | ||
| 
						 | 
					bd3d065a23 | ||
| 
						 | 
					fa497d06b0 | 
							
								
								
									
										6
									
								
								.github/actions/build-image/action.yaml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										6
									
								
								.github/actions/build-image/action.yaml
									
									
									
									
										vendored
									
									
								
							@@ -47,6 +47,9 @@ runs:
 | 
			
		||||
    - name: Build and push to ghcr by digest
 | 
			
		||||
      id: build-ghcr
 | 
			
		||||
      uses: docker/build-push-action@v6.7.0
 | 
			
		||||
      env:
 | 
			
		||||
        DOCKER_BUILD_SUMMARY: false
 | 
			
		||||
        DOCKER_BUILD_RECORD_UPLOAD: false
 | 
			
		||||
      with:
 | 
			
		||||
        context: .
 | 
			
		||||
        file: ./docker/Dockerfile
 | 
			
		||||
@@ -70,6 +73,9 @@ runs:
 | 
			
		||||
    - name: Build and push to dockerhub by digest
 | 
			
		||||
      id: build-dockerhub
 | 
			
		||||
      uses: docker/build-push-action@v6.7.0
 | 
			
		||||
      env:
 | 
			
		||||
        DOCKER_BUILD_SUMMARY: false
 | 
			
		||||
        DOCKER_BUILD_RECORD_UPLOAD: false
 | 
			
		||||
      with:
 | 
			
		||||
        context: .
 | 
			
		||||
        file: ./docker/Dockerfile
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										2
									
								
								.github/actions/restore-python/action.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/actions/restore-python/action.yml
									
									
									
									
										vendored
									
									
								
							@@ -17,7 +17,7 @@ runs:
 | 
			
		||||
  steps:
 | 
			
		||||
    - name: Set up Python ${{ inputs.python-version }}
 | 
			
		||||
      id: python
 | 
			
		||||
      uses: actions/setup-python@v5.1.1
 | 
			
		||||
      uses: actions/setup-python@v5.2.0
 | 
			
		||||
      with:
 | 
			
		||||
        python-version: ${{ inputs.python-version }}
 | 
			
		||||
    - 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
 | 
			
		||||
        uses: actions/checkout@v4.1.7
 | 
			
		||||
      - name: Set up Python
 | 
			
		||||
        uses: actions/setup-python@v5.1.0
 | 
			
		||||
        uses: actions/setup-python@v5.2.0
 | 
			
		||||
        with:
 | 
			
		||||
          python-version: "3.11"
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										2
									
								
								.github/workflows/ci-docker.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/workflows/ci-docker.yml
									
									
									
									
										vendored
									
									
								
							@@ -42,7 +42,7 @@ jobs:
 | 
			
		||||
    steps:
 | 
			
		||||
      - uses: actions/checkout@v4.1.7
 | 
			
		||||
      - name: Set up Python
 | 
			
		||||
        uses: actions/setup-python@v5.1.0
 | 
			
		||||
        uses: actions/setup-python@v5.2.0
 | 
			
		||||
        with:
 | 
			
		||||
          python-version: "3.9"
 | 
			
		||||
      - 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
 | 
			
		||||
      - name: Set up Python ${{ env.DEFAULT_PYTHON }}
 | 
			
		||||
        id: python
 | 
			
		||||
        uses: actions/setup-python@v5.1.0
 | 
			
		||||
        uses: actions/setup-python@v5.2.0
 | 
			
		||||
        with:
 | 
			
		||||
          python-version: ${{ env.DEFAULT_PYTHON }}
 | 
			
		||||
      - name: Restore Python virtual environment
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										8
									
								
								.github/workflows/release.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								.github/workflows/release.yml
									
									
									
									
										vendored
									
									
								
							@@ -53,7 +53,7 @@ jobs:
 | 
			
		||||
    steps:
 | 
			
		||||
      - uses: actions/checkout@v4.1.7
 | 
			
		||||
      - name: Set up Python
 | 
			
		||||
        uses: actions/setup-python@v5.1.0
 | 
			
		||||
        uses: actions/setup-python@v5.2.0
 | 
			
		||||
        with:
 | 
			
		||||
          python-version: "3.x"
 | 
			
		||||
      - name: Set up python environment
 | 
			
		||||
@@ -65,7 +65,7 @@ jobs:
 | 
			
		||||
          pip3 install build
 | 
			
		||||
          python3 -m build
 | 
			
		||||
      - name: Publish
 | 
			
		||||
        uses: pypa/gh-action-pypi-publish@v1.9.0
 | 
			
		||||
        uses: pypa/gh-action-pypi-publish@v1.10.1
 | 
			
		||||
 | 
			
		||||
  deploy-docker:
 | 
			
		||||
    name: Build ESPHome ${{ matrix.platform }}
 | 
			
		||||
@@ -85,7 +85,7 @@ jobs:
 | 
			
		||||
    steps:
 | 
			
		||||
      - uses: actions/checkout@v4.1.7
 | 
			
		||||
      - name: Set up Python
 | 
			
		||||
        uses: actions/setup-python@v5.1.0
 | 
			
		||||
        uses: actions/setup-python@v5.2.0
 | 
			
		||||
        with:
 | 
			
		||||
          python-version: "3.9"
 | 
			
		||||
 | 
			
		||||
@@ -141,7 +141,7 @@ jobs:
 | 
			
		||||
          echo name=$(cat /tmp/platform) >> $GITHUB_OUTPUT
 | 
			
		||||
 | 
			
		||||
      - name: Upload digests
 | 
			
		||||
        uses: actions/upload-artifact@v4.3.4
 | 
			
		||||
        uses: actions/upload-artifact@v4.4.0
 | 
			
		||||
        with:
 | 
			
		||||
          name: digests-${{ steps.sanitize.outputs.name }}
 | 
			
		||||
          path: /tmp/digests
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										4
									
								
								.github/workflows/sync-device-classes.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								.github/workflows/sync-device-classes.yml
									
									
									
									
										vendored
									
									
								
							@@ -22,7 +22,7 @@ jobs:
 | 
			
		||||
          path: lib/home-assistant
 | 
			
		||||
 | 
			
		||||
      - name: Setup Python
 | 
			
		||||
        uses: actions/setup-python@v5.1.0
 | 
			
		||||
        uses: actions/setup-python@v5.2.0
 | 
			
		||||
        with:
 | 
			
		||||
          python-version: 3.12
 | 
			
		||||
 | 
			
		||||
@@ -36,7 +36,7 @@ jobs:
 | 
			
		||||
          python ./script/sync-device_class.py
 | 
			
		||||
 | 
			
		||||
      - name: Commit changes
 | 
			
		||||
        uses: peter-evans/create-pull-request@v6.1.0
 | 
			
		||||
        uses: peter-evans/create-pull-request@v7.0.0
 | 
			
		||||
        with:
 | 
			
		||||
          commit-message: "Synchronise Device Classes from Home Assistant"
 | 
			
		||||
          committer: esphomebot <esphome@nabucasa.com>
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										10
									
								
								CODEOWNERS
									
									
									
									
									
								
							
							
						
						
									
										10
									
								
								CODEOWNERS
									
									
									
									
									
								
							@@ -58,9 +58,10 @@ esphome/components/beken_spi_led_strip/* @Mat931
 | 
			
		||||
esphome/components/bh1750/* @OttoWinter
 | 
			
		||||
esphome/components/binary_sensor/* @esphome/core
 | 
			
		||||
esphome/components/bk72xx/* @kuba2k2
 | 
			
		||||
esphome/components/bl0906/* @athom-tech @jesserockz @tarontop
 | 
			
		||||
esphome/components/bl0939/* @ziceva
 | 
			
		||||
esphome/components/bl0940/* @tobias-
 | 
			
		||||
esphome/components/bl0942/* @dbuezas
 | 
			
		||||
esphome/components/bl0942/* @dbuezas @dwmw2
 | 
			
		||||
esphome/components/ble_client/* @buxtronix @clydebarrow
 | 
			
		||||
esphome/components/bluetooth_proxy/* @jesserockz
 | 
			
		||||
esphome/components/bme280_base/* @esphome/core
 | 
			
		||||
@@ -69,6 +70,9 @@ esphome/components/bme680_bsec/* @trvrnrth
 | 
			
		||||
esphome/components/bme68x_bsec2/* @kbx81 @neffs
 | 
			
		||||
esphome/components/bme68x_bsec2_i2c/* @kbx81 @neffs
 | 
			
		||||
esphome/components/bmi160/* @flaviut
 | 
			
		||||
esphome/components/bmp280_base/* @ademuri
 | 
			
		||||
esphome/components/bmp280_i2c/* @ademuri
 | 
			
		||||
esphome/components/bmp280_spi/* @ademuri
 | 
			
		||||
esphome/components/bmp3xx/* @latonita
 | 
			
		||||
esphome/components/bmp3xx_base/* @latonita @martgras
 | 
			
		||||
esphome/components/bmp3xx_i2c/* @latonita
 | 
			
		||||
@@ -82,6 +86,7 @@ esphome/components/cap1188/* @mreditor97
 | 
			
		||||
esphome/components/captive_portal/* @OttoWinter
 | 
			
		||||
esphome/components/ccs811/* @habbie
 | 
			
		||||
esphome/components/cd74hc4067/* @asoehlke
 | 
			
		||||
esphome/components/ch422g/* @jesterret
 | 
			
		||||
esphome/components/climate/* @esphome/core
 | 
			
		||||
esphome/components/climate_ir/* @glmnet
 | 
			
		||||
esphome/components/color_temperature/* @jesserockz
 | 
			
		||||
@@ -222,6 +227,7 @@ esphome/components/lilygo_t5_47/touchscreen/* @jesserockz
 | 
			
		||||
esphome/components/lock/* @esphome/core
 | 
			
		||||
esphome/components/logger/* @esphome/core
 | 
			
		||||
esphome/components/ltr390/* @latonita @sjtrny
 | 
			
		||||
esphome/components/ltr501/* @latonita
 | 
			
		||||
esphome/components/ltr_als_ps/* @latonita
 | 
			
		||||
esphome/components/lvgl/* @clydebarrow
 | 
			
		||||
esphome/components/m5stack_8angle/* @rnauber
 | 
			
		||||
@@ -383,6 +389,7 @@ esphome/components/st7701s/* @clydebarrow
 | 
			
		||||
esphome/components/st7735/* @SenexCrenshaw
 | 
			
		||||
esphome/components/st7789v/* @kbx81
 | 
			
		||||
esphome/components/st7920/* @marsjan155
 | 
			
		||||
esphome/components/statsd/* @Links2004
 | 
			
		||||
esphome/components/substitutions/* @esphome/core
 | 
			
		||||
esphome/components/sun/* @OttoWinter
 | 
			
		||||
esphome/components/sun_gtil2/* @Mat931
 | 
			
		||||
@@ -422,6 +429,7 @@ esphome/components/tuya/switch/* @jesserockz
 | 
			
		||||
esphome/components/tuya/text_sensor/* @dentra
 | 
			
		||||
esphome/components/uart/* @esphome/core
 | 
			
		||||
esphome/components/uart/button/* @ssieb
 | 
			
		||||
esphome/components/udp/* @clydebarrow
 | 
			
		||||
esphome/components/ufire_ec/* @pvizeli
 | 
			
		||||
esphome/components/ufire_ise/* @pvizeli
 | 
			
		||||
esphome/components/ultrasonic/* @OttoWinter
 | 
			
		||||
 
 | 
			
		||||
@@ -33,9 +33,9 @@ RUN \
 | 
			
		||||
        python3-venv=3.11.2-1+b1 \
 | 
			
		||||
        python3-wheel=0.38.4-2 \
 | 
			
		||||
        iputils-ping=3:20221126-1 \
 | 
			
		||||
        git=1:2.39.2-1.1 \
 | 
			
		||||
        curl=7.88.1-10+deb12u6 \
 | 
			
		||||
        openssh-client=1:9.2p1-2+deb12u2 \
 | 
			
		||||
        git=1:2.39.5-0+deb12u1 \
 | 
			
		||||
        curl=7.88.1-10+deb12u7 \
 | 
			
		||||
        openssh-client=1:9.2p1-2+deb12u3 \
 | 
			
		||||
        python3-cffi=1.15.1-5 \
 | 
			
		||||
        libcairo2=1.16.0-7 \
 | 
			
		||||
        libmagic1=1:5.44-3 \
 | 
			
		||||
@@ -49,7 +49,7 @@ RUN \
 | 
			
		||||
                zlib1g-dev=1:1.2.13.dfsg-1 \
 | 
			
		||||
                libjpeg-dev=1:2.1.5-2 \
 | 
			
		||||
                libfreetype-dev=2.12.1+dfsg-5+deb12u3 \
 | 
			
		||||
                libssl-dev=3.0.13-1~deb12u1 \
 | 
			
		||||
                libssl-dev=3.0.14-1~deb12u2 \
 | 
			
		||||
                libffi-dev=3.4.4-1 \
 | 
			
		||||
                libopenjp2-7=2.5.0-2 \
 | 
			
		||||
                libtiff6=4.5.0-6+deb12u1 \
 | 
			
		||||
@@ -96,14 +96,19 @@ RUN \
 | 
			
		||||
# First install requirements to leverage caching when requirements don't change
 | 
			
		||||
# tmpfs is for https://github.com/rust-lang/cargo/issues/8719
 | 
			
		||||
 | 
			
		||||
COPY requirements.txt requirements_optional.txt script/platformio_install_deps.py platformio.ini /
 | 
			
		||||
COPY requirements.txt requirements_optional.txt /
 | 
			
		||||
RUN --mount=type=tmpfs,target=/root/.cargo if [ "$TARGETARCH$TARGETVARIANT" = "armv7" ]; then \
 | 
			
		||||
        export PIP_EXTRA_INDEX_URL="https://www.piwheels.org/simple"; \
 | 
			
		||||
        curl -L https://www.piwheels.org/cp311/cryptography-43.0.0-cp37-abi3-linux_armv7l.whl -o /tmp/cryptography-43.0.0-cp37-abi3-linux_armv7l.whl \
 | 
			
		||||
        && pip3 install --break-system-packages --no-cache-dir /tmp/cryptography-43.0.0-cp37-abi3-linux_armv7l.whl \
 | 
			
		||||
        && rm /tmp/cryptography-43.0.0-cp37-abi3-linux_armv7l.whl \
 | 
			
		||||
        && export PIP_EXTRA_INDEX_URL="https://www.piwheels.org/simple"; \
 | 
			
		||||
    fi; \
 | 
			
		||||
    CARGO_REGISTRIES_CRATES_IO_PROTOCOL=sparse CARGO_HOME=/root/.cargo \
 | 
			
		||||
    pip3 install \
 | 
			
		||||
    --break-system-packages --no-cache-dir -r /requirements.txt -r /requirements_optional.txt \
 | 
			
		||||
    && /platformio_install_deps.py /platformio.ini --libraries
 | 
			
		||||
    --break-system-packages --no-cache-dir -r /requirements.txt -r /requirements_optional.txt
 | 
			
		||||
 | 
			
		||||
COPY script/platformio_install_deps.py platformio.ini /
 | 
			
		||||
RUN /platformio_install_deps.py /platformio.ini --libraries
 | 
			
		||||
 | 
			
		||||
# Avoid unsafe git error when container user and file config volume permissions don't match
 | 
			
		||||
RUN git config --system --add safe.directory '*'
 | 
			
		||||
 
 | 
			
		||||
@@ -38,7 +38,7 @@ from esphome.const import (
 | 
			
		||||
    SECRETS_FILES,
 | 
			
		||||
)
 | 
			
		||||
from esphome.core import CORE, EsphomeError, coroutine
 | 
			
		||||
from esphome.helpers import indent, is_ip_address
 | 
			
		||||
from esphome.helpers import indent, is_ip_address, get_bool_env
 | 
			
		||||
from esphome.log import Fore, color, setup_log
 | 
			
		||||
from esphome.util import (
 | 
			
		||||
    get_serial_ports,
 | 
			
		||||
@@ -731,7 +731,11 @@ POST_CONFIG_ACTIONS = {
 | 
			
		||||
def parse_args(argv):
 | 
			
		||||
    options_parser = argparse.ArgumentParser(add_help=False)
 | 
			
		||||
    options_parser.add_argument(
 | 
			
		||||
        "-v", "--verbose", help="Enable verbose ESPHome logs.", action="store_true"
 | 
			
		||||
        "-v",
 | 
			
		||||
        "--verbose",
 | 
			
		||||
        help="Enable verbose ESPHome logs.",
 | 
			
		||||
        action="store_true",
 | 
			
		||||
        default=get_bool_env("ESPHOME_VERBOSE"),
 | 
			
		||||
    )
 | 
			
		||||
    options_parser.add_argument(
 | 
			
		||||
        "-q", "--quiet", help="Disable all ESPHome logs.", action="store_true"
 | 
			
		||||
 
 | 
			
		||||
@@ -1107,6 +1107,19 @@ enum MediaPlayerCommand {
 | 
			
		||||
  MEDIA_PLAYER_COMMAND_MUTE = 3;
 | 
			
		||||
  MEDIA_PLAYER_COMMAND_UNMUTE = 4;
 | 
			
		||||
}
 | 
			
		||||
enum MediaPlayerFormatPurpose {
 | 
			
		||||
  MEDIA_PLAYER_FORMAT_PURPOSE_DEFAULT = 0;
 | 
			
		||||
  MEDIA_PLAYER_FORMAT_PURPOSE_ANNOUNCEMENT = 1;
 | 
			
		||||
}
 | 
			
		||||
message MediaPlayerSupportedFormat {
 | 
			
		||||
  option (ifdef) = "USE_MEDIA_PLAYER";
 | 
			
		||||
 | 
			
		||||
  string format = 1;
 | 
			
		||||
  uint32 sample_rate = 2;
 | 
			
		||||
  uint32 num_channels = 3;
 | 
			
		||||
  MediaPlayerFormatPurpose purpose = 4;
 | 
			
		||||
  uint32 sample_bytes = 5;
 | 
			
		||||
}
 | 
			
		||||
message ListEntitiesMediaPlayerResponse {
 | 
			
		||||
  option (id) = 63;
 | 
			
		||||
  option (source) = SOURCE_SERVER;
 | 
			
		||||
@@ -1122,6 +1135,8 @@ message ListEntitiesMediaPlayerResponse {
 | 
			
		||||
  EntityCategory entity_category = 7;
 | 
			
		||||
 | 
			
		||||
  bool supports_pause = 8;
 | 
			
		||||
 | 
			
		||||
  repeated MediaPlayerSupportedFormat supported_formats = 9;
 | 
			
		||||
}
 | 
			
		||||
message MediaPlayerStateResponse {
 | 
			
		||||
  option (id) = 64;
 | 
			
		||||
@@ -1539,6 +1554,23 @@ message VoiceAssistantTimerEventResponse {
 | 
			
		||||
  bool is_active = 6;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
message VoiceAssistantAnnounceRequest {
 | 
			
		||||
  option (id) = 119;
 | 
			
		||||
  option (source) = SOURCE_CLIENT;
 | 
			
		||||
  option (ifdef) = "USE_VOICE_ASSISTANT";
 | 
			
		||||
 | 
			
		||||
  string media_id = 1;
 | 
			
		||||
  string text = 2;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
message VoiceAssistantAnnounceFinished {
 | 
			
		||||
  option (id) = 120;
 | 
			
		||||
  option (source) = SOURCE_SERVER;
 | 
			
		||||
  option (ifdef) = "USE_VOICE_ASSISTANT";
 | 
			
		||||
 | 
			
		||||
  bool success = 1;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ==================== ALARM CONTROL PANEL ====================
 | 
			
		||||
enum AlarmControlPanelState {
 | 
			
		||||
  ALARM_STATE_DISARMED = 0;
 | 
			
		||||
 
 | 
			
		||||
@@ -179,6 +179,7 @@ void APIConnection::loop() {
 | 
			
		||||
      SubscribeHomeAssistantStateResponse resp;
 | 
			
		||||
      resp.entity_id = it.entity_id;
 | 
			
		||||
      resp.attribute = it.attribute.value();
 | 
			
		||||
      resp.once = it.once;
 | 
			
		||||
      if (this->send_subscribe_home_assistant_state_response(resp)) {
 | 
			
		||||
        state_subs_at_++;
 | 
			
		||||
      }
 | 
			
		||||
@@ -1025,6 +1026,16 @@ bool APIConnection::send_media_player_info(media_player::MediaPlayer *media_play
 | 
			
		||||
  auto traits = media_player->get_traits();
 | 
			
		||||
  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);
 | 
			
		||||
    media_format.sample_bytes = supported_format.sample_bytes;
 | 
			
		||||
    msg.supported_formats.push_back(media_format);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  return this->send_list_entities_media_player_response(msg);
 | 
			
		||||
}
 | 
			
		||||
void APIConnection::media_player_command(const MediaPlayerCommandRequest &msg) {
 | 
			
		||||
@@ -1203,6 +1214,16 @@ void APIConnection::on_voice_assistant_timer_event_response(const VoiceAssistant
 | 
			
		||||
  }
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
void APIConnection::on_voice_assistant_announce_request(const VoiceAssistantAnnounceRequest &msg) {
 | 
			
		||||
  if (voice_assistant::global_voice_assistant != nullptr) {
 | 
			
		||||
    if (voice_assistant::global_voice_assistant->get_api_connection() != this) {
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    voice_assistant::global_voice_assistant->on_announce(msg);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
#ifdef USE_ALARM_CONTROL_PANEL
 | 
			
		||||
 
 | 
			
		||||
@@ -151,6 +151,7 @@ class APIConnection : public APIServerConnection {
 | 
			
		||||
  void on_voice_assistant_event_response(const VoiceAssistantEventResponse &msg) override;
 | 
			
		||||
  void on_voice_assistant_audio(const VoiceAssistantAudio &msg) override;
 | 
			
		||||
  void on_voice_assistant_timer_event_response(const VoiceAssistantTimerEventResponse &msg) override;
 | 
			
		||||
  void on_voice_assistant_announce_request(const VoiceAssistantAnnounceRequest &msg) override;
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
#ifdef USE_ALARM_CONTROL_PANEL
 | 
			
		||||
 
 | 
			
		||||
@@ -387,6 +387,18 @@ template<> const char *proto_enum_to_string<enums::MediaPlayerCommand>(enums::Me
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
#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<>
 | 
			
		||||
const char *proto_enum_to_string<enums::BluetoothDeviceRequestType>(enums::BluetoothDeviceRequestType value) {
 | 
			
		||||
  switch (value) {
 | 
			
		||||
@@ -5123,6 +5135,74 @@ void ButtonCommandRequest::dump_to(std::string &out) const {
 | 
			
		||||
  out.append("}");
 | 
			
		||||
}
 | 
			
		||||
#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;
 | 
			
		||||
    }
 | 
			
		||||
    case 5: {
 | 
			
		||||
      this->sample_bytes = value.as_uint32();
 | 
			
		||||
      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);
 | 
			
		||||
  buffer.encode_uint32(5, this->sample_bytes);
 | 
			
		||||
}
 | 
			
		||||
#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("  sample_bytes: ");
 | 
			
		||||
  sprintf(buffer, "%" PRIu32, this->sample_bytes);
 | 
			
		||||
  out.append(buffer);
 | 
			
		||||
  out.append("\n");
 | 
			
		||||
  out.append("}");
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
bool ListEntitiesMediaPlayerResponse::decode_varint(uint32_t field_id, ProtoVarInt value) {
 | 
			
		||||
  switch (field_id) {
 | 
			
		||||
    case 6: {
 | 
			
		||||
@@ -5159,6 +5239,10 @@ bool ListEntitiesMediaPlayerResponse::decode_length(uint32_t field_id, ProtoLeng
 | 
			
		||||
      this->icon = value.as_string();
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
    case 9: {
 | 
			
		||||
      this->supported_formats.push_back(value.as_message<MediaPlayerSupportedFormat>());
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
    default:
 | 
			
		||||
      return false;
 | 
			
		||||
  }
 | 
			
		||||
@@ -5182,6 +5266,9 @@ void ListEntitiesMediaPlayerResponse::encode(ProtoWriteBuffer buffer) const {
 | 
			
		||||
  buffer.encode_bool(6, this->disabled_by_default);
 | 
			
		||||
  buffer.encode_enum<enums::EntityCategory>(7, this->entity_category);
 | 
			
		||||
  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
 | 
			
		||||
void ListEntitiesMediaPlayerResponse::dump_to(std::string &out) const {
 | 
			
		||||
@@ -5219,6 +5306,12 @@ void ListEntitiesMediaPlayerResponse::dump_to(std::string &out) const {
 | 
			
		||||
  out.append("  supports_pause: ");
 | 
			
		||||
  out.append(YESNO(this->supports_pause));
 | 
			
		||||
  out.append("\n");
 | 
			
		||||
 | 
			
		||||
  for (const auto &it : this->supported_formats) {
 | 
			
		||||
    out.append("  supported_formats: ");
 | 
			
		||||
    it.dump_to(out);
 | 
			
		||||
    out.append("\n");
 | 
			
		||||
  }
 | 
			
		||||
  out.append("}");
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
@@ -6978,6 +7071,59 @@ void VoiceAssistantTimerEventResponse::dump_to(std::string &out) const {
 | 
			
		||||
  out.append("}");
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
bool VoiceAssistantAnnounceRequest::decode_length(uint32_t field_id, ProtoLengthDelimited value) {
 | 
			
		||||
  switch (field_id) {
 | 
			
		||||
    case 1: {
 | 
			
		||||
      this->media_id = value.as_string();
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
    case 2: {
 | 
			
		||||
      this->text = value.as_string();
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
    default:
 | 
			
		||||
      return false;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
void VoiceAssistantAnnounceRequest::encode(ProtoWriteBuffer buffer) const {
 | 
			
		||||
  buffer.encode_string(1, this->media_id);
 | 
			
		||||
  buffer.encode_string(2, this->text);
 | 
			
		||||
}
 | 
			
		||||
#ifdef HAS_PROTO_MESSAGE_DUMP
 | 
			
		||||
void VoiceAssistantAnnounceRequest::dump_to(std::string &out) const {
 | 
			
		||||
  __attribute__((unused)) char buffer[64];
 | 
			
		||||
  out.append("VoiceAssistantAnnounceRequest {\n");
 | 
			
		||||
  out.append("  media_id: ");
 | 
			
		||||
  out.append("'").append(this->media_id).append("'");
 | 
			
		||||
  out.append("\n");
 | 
			
		||||
 | 
			
		||||
  out.append("  text: ");
 | 
			
		||||
  out.append("'").append(this->text).append("'");
 | 
			
		||||
  out.append("\n");
 | 
			
		||||
  out.append("}");
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
bool VoiceAssistantAnnounceFinished::decode_varint(uint32_t field_id, ProtoVarInt value) {
 | 
			
		||||
  switch (field_id) {
 | 
			
		||||
    case 1: {
 | 
			
		||||
      this->success = value.as_bool();
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
    default:
 | 
			
		||||
      return false;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
void VoiceAssistantAnnounceFinished::encode(ProtoWriteBuffer buffer) const { buffer.encode_bool(1, this->success); }
 | 
			
		||||
#ifdef HAS_PROTO_MESSAGE_DUMP
 | 
			
		||||
void VoiceAssistantAnnounceFinished::dump_to(std::string &out) const {
 | 
			
		||||
  __attribute__((unused)) char buffer[64];
 | 
			
		||||
  out.append("VoiceAssistantAnnounceFinished {\n");
 | 
			
		||||
  out.append("  success: ");
 | 
			
		||||
  out.append(YESNO(this->success));
 | 
			
		||||
  out.append("\n");
 | 
			
		||||
  out.append("}");
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
bool ListEntitiesAlarmControlPanelResponse::decode_varint(uint32_t field_id, ProtoVarInt value) {
 | 
			
		||||
  switch (field_id) {
 | 
			
		||||
    case 6: {
 | 
			
		||||
 
 | 
			
		||||
@@ -156,6 +156,10 @@ enum MediaPlayerCommand : uint32_t {
 | 
			
		||||
  MEDIA_PLAYER_COMMAND_MUTE = 3,
 | 
			
		||||
  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 {
 | 
			
		||||
  BLUETOOTH_DEVICE_REQUEST_TYPE_CONNECT = 0,
 | 
			
		||||
  BLUETOOTH_DEVICE_REQUEST_TYPE_DISCONNECT = 1,
 | 
			
		||||
@@ -1267,6 +1271,22 @@ class ButtonCommandRequest : public ProtoMessage {
 | 
			
		||||
 protected:
 | 
			
		||||
  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{};
 | 
			
		||||
  uint32_t sample_bytes{0};
 | 
			
		||||
  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 {
 | 
			
		||||
 public:
 | 
			
		||||
  std::string object_id{};
 | 
			
		||||
@@ -1277,6 +1297,7 @@ class ListEntitiesMediaPlayerResponse : public ProtoMessage {
 | 
			
		||||
  bool disabled_by_default{false};
 | 
			
		||||
  enums::EntityCategory entity_category{};
 | 
			
		||||
  bool supports_pause{false};
 | 
			
		||||
  std::vector<MediaPlayerSupportedFormat> supported_formats{};
 | 
			
		||||
  void encode(ProtoWriteBuffer buffer) const override;
 | 
			
		||||
#ifdef HAS_PROTO_MESSAGE_DUMP
 | 
			
		||||
  void dump_to(std::string &out) const override;
 | 
			
		||||
@@ -1805,6 +1826,29 @@ class VoiceAssistantTimerEventResponse : public ProtoMessage {
 | 
			
		||||
  bool decode_length(uint32_t field_id, ProtoLengthDelimited value) override;
 | 
			
		||||
  bool decode_varint(uint32_t field_id, ProtoVarInt value) override;
 | 
			
		||||
};
 | 
			
		||||
class VoiceAssistantAnnounceRequest : public ProtoMessage {
 | 
			
		||||
 public:
 | 
			
		||||
  std::string media_id{};
 | 
			
		||||
  std::string text{};
 | 
			
		||||
  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;
 | 
			
		||||
};
 | 
			
		||||
class VoiceAssistantAnnounceFinished : public ProtoMessage {
 | 
			
		||||
 public:
 | 
			
		||||
  bool success{false};
 | 
			
		||||
  void encode(ProtoWriteBuffer buffer) const override;
 | 
			
		||||
#ifdef HAS_PROTO_MESSAGE_DUMP
 | 
			
		||||
  void dump_to(std::string &out) const override;
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  bool decode_varint(uint32_t field_id, ProtoVarInt value) override;
 | 
			
		||||
};
 | 
			
		||||
class ListEntitiesAlarmControlPanelResponse : public ProtoMessage {
 | 
			
		||||
 public:
 | 
			
		||||
  std::string object_id{};
 | 
			
		||||
 
 | 
			
		||||
@@ -486,6 +486,16 @@ bool APIServerConnectionBase::send_voice_assistant_audio(const VoiceAssistantAud
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
bool APIServerConnectionBase::send_voice_assistant_announce_finished(const VoiceAssistantAnnounceFinished &msg) {
 | 
			
		||||
#ifdef HAS_PROTO_MESSAGE_DUMP
 | 
			
		||||
  ESP_LOGVV(TAG, "send_voice_assistant_announce_finished: %s", msg.dump().c_str());
 | 
			
		||||
#endif
 | 
			
		||||
  return this->send_message_<VoiceAssistantAnnounceFinished>(msg, 120);
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_ALARM_CONTROL_PANEL
 | 
			
		||||
bool APIServerConnectionBase::send_list_entities_alarm_control_panel_response(
 | 
			
		||||
    const ListEntitiesAlarmControlPanelResponse &msg) {
 | 
			
		||||
@@ -1135,6 +1145,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());
 | 
			
		||||
#endif
 | 
			
		||||
      this->on_update_command_request(msg);
 | 
			
		||||
#endif
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
    case 119: {
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
      VoiceAssistantAnnounceRequest msg;
 | 
			
		||||
      msg.decode(msg_data, msg_size);
 | 
			
		||||
#ifdef HAS_PROTO_MESSAGE_DUMP
 | 
			
		||||
      ESP_LOGVV(TAG, "on_voice_assistant_announce_request: %s", msg.dump().c_str());
 | 
			
		||||
#endif
 | 
			
		||||
      this->on_voice_assistant_announce_request(msg);
 | 
			
		||||
#endif
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -247,6 +247,12 @@ class APIServerConnectionBase : public ProtoService {
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
  virtual void on_voice_assistant_timer_event_response(const VoiceAssistantTimerEventResponse &value){};
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
  virtual void on_voice_assistant_announce_request(const VoiceAssistantAnnounceRequest &value){};
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_VOICE_ASSISTANT
 | 
			
		||||
  bool send_voice_assistant_announce_finished(const VoiceAssistantAnnounceFinished &msg);
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_ALARM_CONTROL_PANEL
 | 
			
		||||
  bool send_list_entities_alarm_control_panel_response(const ListEntitiesAlarmControlPanelResponse &msg);
 | 
			
		||||
#endif
 | 
			
		||||
 
 | 
			
		||||
@@ -1,13 +1,13 @@
 | 
			
		||||
# Dummy integration to allow relying on AsyncTCP
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.core import CORE, coroutine_with_priority
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    PLATFORM_BK72XX,
 | 
			
		||||
    PLATFORM_ESP32,
 | 
			
		||||
    PLATFORM_ESP8266,
 | 
			
		||||
    PLATFORM_BK72XX,
 | 
			
		||||
    PLATFORM_RTL87XX,
 | 
			
		||||
)
 | 
			
		||||
from esphome.core import CORE, coroutine_with_priority
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@OttoWinter"]
 | 
			
		||||
 | 
			
		||||
@@ -22,7 +22,7 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    if CORE.is_esp32 or CORE.is_libretiny:
 | 
			
		||||
        # https://github.com/esphome/AsyncTCP/blob/master/library.json
 | 
			
		||||
        cg.add_library("esphome/AsyncTCP-esphome", "2.1.3")
 | 
			
		||||
        cg.add_library("esphome/AsyncTCP-esphome", "2.1.4")
 | 
			
		||||
    elif CORE.is_esp8266:
 | 
			
		||||
        # https://github.com/esphome/ESPAsyncTCP
 | 
			
		||||
        cg.add_library("esphome/ESPAsyncTCP-esphome", "2.0.0")
 | 
			
		||||
 
 | 
			
		||||
@@ -1,34 +1,34 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import sensor, spi
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_REACTIVE_POWER,
 | 
			
		||||
    CONF_VOLTAGE,
 | 
			
		||||
    CONF_CURRENT,
 | 
			
		||||
    CONF_FORWARD_ACTIVE_ENERGY,
 | 
			
		||||
    CONF_FREQUENCY,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_LINE_FREQUENCY,
 | 
			
		||||
    CONF_POWER,
 | 
			
		||||
    CONF_POWER_FACTOR,
 | 
			
		||||
    CONF_FREQUENCY,
 | 
			
		||||
    CONF_FORWARD_ACTIVE_ENERGY,
 | 
			
		||||
    CONF_REACTIVE_POWER,
 | 
			
		||||
    CONF_REVERSE_ACTIVE_ENERGY,
 | 
			
		||||
    CONF_VOLTAGE,
 | 
			
		||||
    DEVICE_CLASS_CURRENT,
 | 
			
		||||
    DEVICE_CLASS_ENERGY,
 | 
			
		||||
    DEVICE_CLASS_POWER,
 | 
			
		||||
    DEVICE_CLASS_POWER_FACTOR,
 | 
			
		||||
    DEVICE_CLASS_VOLTAGE,
 | 
			
		||||
    ICON_LIGHTBULB,
 | 
			
		||||
    ICON_CURRENT_AC,
 | 
			
		||||
    ICON_LIGHTBULB,
 | 
			
		||||
    STATE_CLASS_MEASUREMENT,
 | 
			
		||||
    STATE_CLASS_TOTAL_INCREASING,
 | 
			
		||||
    UNIT_AMPERE,
 | 
			
		||||
    UNIT_HERTZ,
 | 
			
		||||
    UNIT_VOLT,
 | 
			
		||||
    UNIT_AMPERE,
 | 
			
		||||
    UNIT_WATT,
 | 
			
		||||
    UNIT_VOLT_AMPS_REACTIVE,
 | 
			
		||||
    UNIT_WATT,
 | 
			
		||||
    UNIT_WATT_HOURS,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONF_LINE_FREQUENCY = "line_frequency"
 | 
			
		||||
CONF_METER_CONSTANT = "meter_constant"
 | 
			
		||||
CONF_PL_CONST = "pl_const"
 | 
			
		||||
CONF_GAIN_PGA = "gain_pga"
 | 
			
		||||
 
 | 
			
		||||
@@ -7,6 +7,7 @@ from esphome.const import (
 | 
			
		||||
    CONF_FORWARD_ACTIVE_ENERGY,
 | 
			
		||||
    CONF_FREQUENCY,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_LINE_FREQUENCY,
 | 
			
		||||
    CONF_PHASE_A,
 | 
			
		||||
    CONF_PHASE_ANGLE,
 | 
			
		||||
    CONF_PHASE_B,
 | 
			
		||||
@@ -39,7 +40,6 @@ from esphome.const import (
 | 
			
		||||
 | 
			
		||||
from . import atm90e32_ns
 | 
			
		||||
 | 
			
		||||
CONF_LINE_FREQUENCY = "line_frequency"
 | 
			
		||||
CONF_CHIP_TEMPERATURE = "chip_temperature"
 | 
			
		||||
CONF_GAIN_PGA = "gain_pga"
 | 
			
		||||
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))
 | 
			
		||||
@@ -1 +1 @@
 | 
			
		||||
CODEOWNERS = ["@dbuezas"]
 | 
			
		||||
CODEOWNERS = ["@dbuezas", "@dwmw2"]
 | 
			
		||||
 
 | 
			
		||||
@@ -2,6 +2,8 @@
 | 
			
		||||
#include "esphome/core/log.h"
 | 
			
		||||
#include <cinttypes>
 | 
			
		||||
 | 
			
		||||
// Datasheet: https://www.belling.com.cn/media/file_object/bel_product/BL0942/datasheet/BL0942_V1.06_en.pdf
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bl0942 {
 | 
			
		||||
 | 
			
		||||
@@ -12,43 +14,64 @@ static const uint8_t BL0942_FULL_PACKET = 0xAA;
 | 
			
		||||
static const uint8_t BL0942_PACKET_HEADER = 0x55;
 | 
			
		||||
 | 
			
		||||
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_SOFT_RESET = 0x19;
 | 
			
		||||
static const uint8_t BL0942_REG_USR_WRPROT = 0x1A;
 | 
			
		||||
 | 
			
		||||
static const uint8_t BL0942_REG_I_RMSOS = 0x12;
 | 
			
		||||
static const uint8_t BL0942_REG_WA_CREEP = 0x14;
 | 
			
		||||
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;
 | 
			
		||||
 | 
			
		||||
// TODO: Confirm insialisation works as intended
 | 
			
		||||
const uint8_t BL0942_INIT[5][6] = {
 | 
			
		||||
    // Reset to default
 | 
			
		||||
    {BL0942_WRITE_COMMAND, BL0942_REG_SOFT_RESET, 0x5A, 0x5A, 0x5A, 0x38},
 | 
			
		||||
    // Enable User Operation Write
 | 
			
		||||
    {BL0942_WRITE_COMMAND, BL0942_REG_USR_WRPROT, 0x55, 0x00, 0x00, 0xF0},
 | 
			
		||||
    // 0x0100 = CF_UNABLE energy pulse, AC_FREQ_SEL 50Hz, RMS_UPDATE_SEL 800mS
 | 
			
		||||
    {BL0942_WRITE_COMMAND, BL0942_REG_MODE, 0x00, 0x10, 0x00, 0x37},
 | 
			
		||||
    // 0x47FF = Over-current and leakage alarm on, Automatic temperature measurement, Interval 100mS
 | 
			
		||||
    {BL0942_WRITE_COMMAND, BL0942_REG_TPS_CTRL, 0xFF, 0x47, 0x00, 0xFE},
 | 
			
		||||
    // 0x181C = Half cycle, Fast RMS threshold 6172
 | 
			
		||||
    {BL0942_WRITE_COMMAND, BL0942_REG_I_FAST_RMS_CTRL, 0x1C, 0x18, 0x00, 0x1B}};
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_RESV = 0x03;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_CF_EN = 0x04;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_RMS_UPDATE_SEL = 0x08;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_FAST_RMS_SEL = 0x10;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_AC_FREQ_SEL = 0x20;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_CF_CNT_CLR_SEL = 0x40;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_CF_CNT_ADD_SEL = 0x80;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_UART_RATE_19200 = 0x200;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_UART_RATE_38400 = 0x300;
 | 
			
		||||
static const uint32_t BL0942_REG_MODE_DEFAULT =
 | 
			
		||||
    BL0942_REG_MODE_RESV | BL0942_REG_MODE_CF_EN | BL0942_REG_MODE_CF_CNT_ADD_SEL;
 | 
			
		||||
 | 
			
		||||
static const uint32_t BL0942_REG_SOFT_RESET_MAGIC = 0x5a5a5a;
 | 
			
		||||
static const uint32_t BL0942_REG_USR_WRPROT_MAGIC = 0x55;
 | 
			
		||||
 | 
			
		||||
// 23-byte packet, 11 bits per byte, 2400 baud: about 105ms
 | 
			
		||||
static const uint32_t PKT_TIMEOUT_MS = 200;
 | 
			
		||||
 | 
			
		||||
void BL0942::loop() {
 | 
			
		||||
  DataPacket buffer;
 | 
			
		||||
  if (!this->available()) {
 | 
			
		||||
  int avail = this->available();
 | 
			
		||||
 | 
			
		||||
  if (!avail) {
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
  if (read_array((uint8_t *) &buffer, sizeof(buffer))) {
 | 
			
		||||
    if (validate_checksum(&buffer)) {
 | 
			
		||||
      received_package_(&buffer);
 | 
			
		||||
  if (avail < sizeof(buffer)) {
 | 
			
		||||
    if (!this->rx_start_) {
 | 
			
		||||
      this->rx_start_ = millis();
 | 
			
		||||
    } else if (millis() > this->rx_start_ + PKT_TIMEOUT_MS) {
 | 
			
		||||
      ESP_LOGW(TAG, "Junk on wire. Throwing away partial message (%d bytes)", avail);
 | 
			
		||||
      this->read_array((uint8_t *) &buffer, avail);
 | 
			
		||||
      this->rx_start_ = 0;
 | 
			
		||||
    }
 | 
			
		||||
  } else {
 | 
			
		||||
    ESP_LOGW(TAG, "Junk on wire. Throwing away partial message");
 | 
			
		||||
    while (read() >= 0)
 | 
			
		||||
      ;
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  if (this->read_array((uint8_t *) &buffer, sizeof(buffer))) {
 | 
			
		||||
    if (this->validate_checksum_(&buffer)) {
 | 
			
		||||
      this->received_package_(&buffer);
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  this->rx_start_ = 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool BL0942::validate_checksum(DataPacket *data) {
 | 
			
		||||
  uint8_t checksum = BL0942_READ_COMMAND;
 | 
			
		||||
bool BL0942::validate_checksum_(DataPacket *data) {
 | 
			
		||||
  uint8_t checksum = BL0942_READ_COMMAND | this->address_;
 | 
			
		||||
  // Whole package but checksum
 | 
			
		||||
  uint8_t *raw = (uint8_t *) data;
 | 
			
		||||
  for (uint32_t i = 0; i < sizeof(*data) - 1; i++) {
 | 
			
		||||
@@ -61,17 +84,73 @@ bool BL0942::validate_checksum(DataPacket *data) {
 | 
			
		||||
  return checksum == data->checksum;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void BL0942::update() {
 | 
			
		||||
void BL0942::write_reg_(uint8_t reg, uint32_t val) {
 | 
			
		||||
  uint8_t pkt[6];
 | 
			
		||||
 | 
			
		||||
  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);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void BL0942::setup() {
 | 
			
		||||
  for (auto *i : BL0942_INIT) {
 | 
			
		||||
    this->write_array(i, 6);
 | 
			
		||||
    delay(1);
 | 
			
		||||
  // If either current or voltage references are set explicitly by the user,
 | 
			
		||||
  // calculate the power reference from it unless that is also explicitly set.
 | 
			
		||||
  if ((this->current_reference_set_ || this->voltage_reference_set_) && !this->power_reference_set_) {
 | 
			
		||||
    this->power_reference_ = (this->voltage_reference_ * this->current_reference_ * 3537.0 / 305978.0) / 73989.0;
 | 
			
		||||
    this->power_reference_set_ = true;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  // Similarly for energy reference, if the power reference was set by the user
 | 
			
		||||
  // either implicitly or explicitly.
 | 
			
		||||
  if (this->power_reference_set_ && !this->energy_reference_set_) {
 | 
			
		||||
    this->energy_reference_ = this->power_reference_ * 3600000 / 419430.4;
 | 
			
		||||
    this->energy_reference_set_ = true;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  this->write_reg_(BL0942_REG_USR_WRPROT, BL0942_REG_USR_WRPROT_MAGIC);
 | 
			
		||||
  if (this->reset_)
 | 
			
		||||
    this->write_reg_(BL0942_REG_SOFT_RESET, BL0942_REG_SOFT_RESET_MAGIC);
 | 
			
		||||
 | 
			
		||||
  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();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -82,10 +161,17 @@ void BL0942::received_package_(DataPacket *data) {
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  // cf_cnt is only 24 bits, so track overflows
 | 
			
		||||
  uint32_t cf_cnt = (uint24_t) data->cf_cnt;
 | 
			
		||||
  cf_cnt |= this->prev_cf_cnt_ & 0xff000000;
 | 
			
		||||
  if (cf_cnt < this->prev_cf_cnt_) {
 | 
			
		||||
    cf_cnt += 0x1000000;
 | 
			
		||||
  }
 | 
			
		||||
  this->prev_cf_cnt_ = cf_cnt;
 | 
			
		||||
 | 
			
		||||
  float v_rms = (uint24_t) data->v_rms / voltage_reference_;
 | 
			
		||||
  float i_rms = (uint24_t) data->i_rms / current_reference_;
 | 
			
		||||
  float watt = (int24_t) data->watt / power_reference_;
 | 
			
		||||
  uint32_t cf_cnt = (uint24_t) data->cf_cnt;
 | 
			
		||||
  float total_energy_consumption = cf_cnt / energy_reference_;
 | 
			
		||||
  float frequency = 1000000.0f / data->frequency;
 | 
			
		||||
 | 
			
		||||
@@ -104,18 +190,25 @@ void BL0942::received_package_(DataPacket *data) {
 | 
			
		||||
  if (frequency_sensor_ != nullptr) {
 | 
			
		||||
    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,
 | 
			
		||||
           watt, cf_cnt, total_energy_consumption, frequency, data->status);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void BL0942::dump_config() {  // NOLINT(readability-function-cognitive-complexity)
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "BL0942:");
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Reset: %s", TRUEFALSE(this->reset_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Address: %d", this->address_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Nominal line frequency: %d Hz", this->line_freq_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Current reference: %f", this->current_reference_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Energy reference: %f", this->energy_reference_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Power reference: %f", this->power_reference_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Voltage reference: %f", this->voltage_reference_);
 | 
			
		||||
  LOG_SENSOR("", "Voltage", this->voltage_sensor_);
 | 
			
		||||
  LOG_SENSOR("", "Current", this->current_sensor_);
 | 
			
		||||
  LOG_SENSOR("", "Power", this->power_sensor_);
 | 
			
		||||
  LOG_SENSOR("", "Energy", this->energy_sensor_);
 | 
			
		||||
  LOG_SENSOR("", "frequency", this->frequency_sensor_);
 | 
			
		||||
  LOG_SENSOR("", "Frequency", this->frequency_sensor_);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}  // namespace bl0942
 | 
			
		||||
 
 | 
			
		||||
@@ -8,6 +8,57 @@
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bl0942 {
 | 
			
		||||
 | 
			
		||||
// The BL0942 IC is "calibration-free", which means that it doesn't care
 | 
			
		||||
// at all about calibration, and that's left to software. It measures a
 | 
			
		||||
// voltage differential on its IP/IN pins which linearly proportional to
 | 
			
		||||
// the current flow, and another on its VP pin which is proportional to
 | 
			
		||||
// the line voltage. It never knows the actual calibration; the values
 | 
			
		||||
// it reports are solely in terms of those inputs.
 | 
			
		||||
//
 | 
			
		||||
// The datasheet refers to the input voltages as I(A) and V(V), both
 | 
			
		||||
// in millivolts. It measures them against a reference voltage Vref,
 | 
			
		||||
// which is typically 1.218V (but that absolute value is meaningless
 | 
			
		||||
// without the actual calibration anyway).
 | 
			
		||||
//
 | 
			
		||||
// The reported I_RMS value is 305978 I(A)/Vref, and the reported V_RMS
 | 
			
		||||
// value is 73989 V(V)/Vref. So we can calibrate those by applying a
 | 
			
		||||
// simple meter with a resistive load.
 | 
			
		||||
//
 | 
			
		||||
// The chip also measures the phase difference between voltage and
 | 
			
		||||
// current, and uses it to calculate the power factor (cos φ). It
 | 
			
		||||
// reports the WATT value of 3537 * I_RMS * V_RMS * cos φ).
 | 
			
		||||
//
 | 
			
		||||
// It also integrates total energy based on the WATT value. The time for
 | 
			
		||||
// one CF_CNT pulse is 1638.4*256 / WATT.
 | 
			
		||||
//
 | 
			
		||||
// So... how do we calibrate that?
 | 
			
		||||
//
 | 
			
		||||
// Using a simple resistive load and an external meter, we can measure
 | 
			
		||||
// the true voltage and current for a given V_RMS and I_RMS reading,
 | 
			
		||||
// to calculate BL0942_UREF and BL0942_IREF. Those are in units of
 | 
			
		||||
// "305978 counts per amp" or "73989 counts per volt" respectively.
 | 
			
		||||
//
 | 
			
		||||
// We can derive BL0942_PREF from those. Let's eliminate the weird
 | 
			
		||||
// factors and express the calibration in plain counts per volt/amp:
 | 
			
		||||
// UREF1 = UREF/73989, IREF1 = IREF/305978.
 | 
			
		||||
//
 | 
			
		||||
// Next... the true power in Watts is V * I * cos φ, so that's equal
 | 
			
		||||
// to WATT/3537 * IREF1 * UREF1. Which means
 | 
			
		||||
// BL0942_PREF = BL0942_UREF * BL0942_IREF * 3537 / 305978 / 73989.
 | 
			
		||||
//
 | 
			
		||||
// Finally the accumulated energy. The period of a CF_CNT count is
 | 
			
		||||
// 1638.4*256 / WATT seconds, or 419230.4 / WATT seconds. Which means
 | 
			
		||||
// the energy represented by a CN_CNT pulse is 419230.4 WATT-seconds.
 | 
			
		||||
// Factoring in the calibration, that's 419230.4 / BL0942_PREF actual
 | 
			
		||||
// Watt-seconds (or Joules, as the physicists like to call them).
 | 
			
		||||
//
 | 
			
		||||
// But we're not being physicists today; we we're being engineers, so
 | 
			
		||||
// we want to convert to kWh instead. Which we do by dividing by 1000
 | 
			
		||||
// and then by 3600, so the energy in kWh is
 | 
			
		||||
// CF_CNT * 419230.4 / BL0942_PREF / 3600000
 | 
			
		||||
//
 | 
			
		||||
// Which makes BL0952_EREF = BL0942_PREF * 3600000 / 419430.4
 | 
			
		||||
 | 
			
		||||
static const float BL0942_PREF = 596;              // taken from tasmota
 | 
			
		||||
static const float BL0942_UREF = 15873.35944299;   // should be 73989/1.218
 | 
			
		||||
static const float BL0942_IREF = 251213.46469622;  // 305978/1.218
 | 
			
		||||
@@ -28,6 +79,11 @@ struct DataPacket {
 | 
			
		||||
  uint8_t checksum;
 | 
			
		||||
} __attribute__((packed));
 | 
			
		||||
 | 
			
		||||
enum LineFrequency : uint8_t {
 | 
			
		||||
  LINE_FREQUENCY_50HZ = 50,
 | 
			
		||||
  LINE_FREQUENCY_60HZ = 60,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class BL0942 : public PollingComponent, public uart::UARTDevice {
 | 
			
		||||
 public:
 | 
			
		||||
  void set_voltage_sensor(sensor::Sensor *voltage_sensor) { voltage_sensor_ = voltage_sensor; }
 | 
			
		||||
@@ -35,9 +91,27 @@ class BL0942 : public PollingComponent, public uart::UARTDevice {
 | 
			
		||||
  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_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 set_reset(bool reset) { this->reset_ = reset; }
 | 
			
		||||
  void set_current_reference(float current_ref) {
 | 
			
		||||
    this->current_reference_ = current_ref;
 | 
			
		||||
    this->current_reference_set_ = true;
 | 
			
		||||
  }
 | 
			
		||||
  void set_energy_reference(float energy_ref) {
 | 
			
		||||
    this->energy_reference_ = energy_ref;
 | 
			
		||||
    this->energy_reference_set_ = true;
 | 
			
		||||
  }
 | 
			
		||||
  void set_power_reference(float power_ref) {
 | 
			
		||||
    this->power_reference_ = power_ref;
 | 
			
		||||
    this->power_reference_set_ = true;
 | 
			
		||||
  }
 | 
			
		||||
  void set_voltage_reference(float voltage_ref) {
 | 
			
		||||
    this->voltage_reference_ = voltage_ref;
 | 
			
		||||
    this->voltage_reference_set_ = true;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  void loop() override;
 | 
			
		||||
 | 
			
		||||
  void update() override;
 | 
			
		||||
  void setup() override;
 | 
			
		||||
  void dump_config() override;
 | 
			
		||||
@@ -53,15 +127,25 @@ class BL0942 : public PollingComponent, public uart::UARTDevice {
 | 
			
		||||
 | 
			
		||||
  // Divide by this to turn into Watt
 | 
			
		||||
  float power_reference_ = BL0942_PREF;
 | 
			
		||||
  bool power_reference_set_ = false;
 | 
			
		||||
  // Divide by this to turn into Volt
 | 
			
		||||
  float voltage_reference_ = BL0942_UREF;
 | 
			
		||||
  bool voltage_reference_set_ = false;
 | 
			
		||||
  // Divide by this to turn into Ampere
 | 
			
		||||
  float current_reference_ = BL0942_IREF;
 | 
			
		||||
  bool current_reference_set_ = false;
 | 
			
		||||
  // Divide by this to turn into kWh
 | 
			
		||||
  float energy_reference_ = BL0942_EREF;
 | 
			
		||||
  bool energy_reference_set_ = false;
 | 
			
		||||
  uint8_t address_ = 0;
 | 
			
		||||
  bool reset_ = false;
 | 
			
		||||
  LineFrequency line_freq_ = LINE_FREQUENCY_50HZ;
 | 
			
		||||
  uint32_t rx_start_ = 0;
 | 
			
		||||
  uint32_t prev_cf_cnt_ = 0;
 | 
			
		||||
 | 
			
		||||
  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);
 | 
			
		||||
};
 | 
			
		||||
}  // namespace bl0942
 | 
			
		||||
 
 | 
			
		||||
@@ -1,32 +1,46 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import sensor, uart
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_ADDRESS,
 | 
			
		||||
    CONF_CURRENT,
 | 
			
		||||
    CONF_ENERGY,
 | 
			
		||||
    CONF_FREQUENCY,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_LINE_FREQUENCY,
 | 
			
		||||
    CONF_POWER,
 | 
			
		||||
    CONF_VOLTAGE,
 | 
			
		||||
    CONF_FREQUENCY,
 | 
			
		||||
    DEVICE_CLASS_CURRENT,
 | 
			
		||||
    DEVICE_CLASS_ENERGY,
 | 
			
		||||
    DEVICE_CLASS_FREQUENCY,
 | 
			
		||||
    DEVICE_CLASS_POWER,
 | 
			
		||||
    DEVICE_CLASS_VOLTAGE,
 | 
			
		||||
    DEVICE_CLASS_FREQUENCY,
 | 
			
		||||
    STATE_CLASS_MEASUREMENT,
 | 
			
		||||
    STATE_CLASS_TOTAL_INCREASING,
 | 
			
		||||
    UNIT_AMPERE,
 | 
			
		||||
    UNIT_HERTZ,
 | 
			
		||||
    UNIT_KILOWATT_HOURS,
 | 
			
		||||
    UNIT_VOLT,
 | 
			
		||||
    UNIT_WATT,
 | 
			
		||||
    UNIT_HERTZ,
 | 
			
		||||
    STATE_CLASS_TOTAL_INCREASING,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONF_CURRENT_REFERENCE = "current_reference"
 | 
			
		||||
CONF_ENERGY_REFERENCE = "energy_reference"
 | 
			
		||||
CONF_POWER_REFERENCE = "power_reference"
 | 
			
		||||
CONF_RESET = "reset"
 | 
			
		||||
CONF_VOLTAGE_REFERENCE = "voltage_reference"
 | 
			
		||||
 | 
			
		||||
DEPENDENCIES = ["uart"]
 | 
			
		||||
 | 
			
		||||
bl0942_ns = cg.esphome_ns.namespace("bl0942")
 | 
			
		||||
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 = (
 | 
			
		||||
    cv.Schema(
 | 
			
		||||
        {
 | 
			
		||||
@@ -45,22 +59,35 @@ CONFIG_SCHEMA = (
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_POWER): sensor.sensor_schema(
 | 
			
		||||
                unit_of_measurement=UNIT_WATT,
 | 
			
		||||
                accuracy_decimals=0,
 | 
			
		||||
                accuracy_decimals=1,
 | 
			
		||||
                device_class=DEVICE_CLASS_POWER,
 | 
			
		||||
                state_class=STATE_CLASS_MEASUREMENT,
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_ENERGY): sensor.sensor_schema(
 | 
			
		||||
                unit_of_measurement=UNIT_KILOWATT_HOURS,
 | 
			
		||||
                accuracy_decimals=0,
 | 
			
		||||
                accuracy_decimals=3,
 | 
			
		||||
                device_class=DEVICE_CLASS_ENERGY,
 | 
			
		||||
                state_class=STATE_CLASS_TOTAL_INCREASING,
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_FREQUENCY): sensor.sensor_schema(
 | 
			
		||||
                unit_of_measurement=UNIT_HERTZ,
 | 
			
		||||
                accuracy_decimals=0,
 | 
			
		||||
                accuracy_decimals=2,
 | 
			
		||||
                device_class=DEVICE_CLASS_FREQUENCY,
 | 
			
		||||
                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),
 | 
			
		||||
            cv.Optional(CONF_RESET, default=True): cv.boolean,
 | 
			
		||||
            cv.Optional(CONF_CURRENT_REFERENCE): cv.float_,
 | 
			
		||||
            cv.Optional(CONF_ENERGY_REFERENCE): cv.float_,
 | 
			
		||||
            cv.Optional(CONF_POWER_REFERENCE): cv.float_,
 | 
			
		||||
            cv.Optional(CONF_VOLTAGE_REFERENCE): cv.float_,
 | 
			
		||||
        }
 | 
			
		||||
    )
 | 
			
		||||
    .extend(cv.polling_component_schema("60s"))
 | 
			
		||||
@@ -88,3 +115,14 @@ async def to_code(config):
 | 
			
		||||
    if frequency_config := config.get(CONF_FREQUENCY):
 | 
			
		||||
        sens = await sensor.new_sensor(frequency_config)
 | 
			
		||||
        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]))
 | 
			
		||||
    cg.add(var.set_reset(config[CONF_RESET]))
 | 
			
		||||
    if (current_reference := config.get(CONF_CURRENT_REFERENCE, None)) is not None:
 | 
			
		||||
        cg.add(var.set_current_reference(current_reference))
 | 
			
		||||
    if (voltage_reference := config.get(CONF_VOLTAGE_REFERENCE, None)) is not None:
 | 
			
		||||
        cg.add(var.set_voltage_reference(voltage_reference))
 | 
			
		||||
    if (power_reference := config.get(CONF_POWER_REFERENCE, None)) is not None:
 | 
			
		||||
        cg.add(var.set_power_reference(power_reference))
 | 
			
		||||
    if (energy_reference := config.get(CONF_ENERGY_REFERENCE, None)) is not None:
 | 
			
		||||
        cg.add(var.set_energy_reference(energy_reference))
 | 
			
		||||
 
 | 
			
		||||
@@ -65,9 +65,7 @@ CONF_ON_PASSKEY_NOTIFICATION = "on_passkey_notification"
 | 
			
		||||
CONF_ON_NUMERIC_COMPARISON_REQUEST = "on_numeric_comparison_request"
 | 
			
		||||
CONF_AUTO_CONNECT = "auto_connect"
 | 
			
		||||
 | 
			
		||||
# Espressif platformio framework is built with MAX_BLE_CONN to 3, so
 | 
			
		||||
# enforce this in yaml checks.
 | 
			
		||||
MULTI_CONF = 3
 | 
			
		||||
MULTI_CONF = True
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = (
 | 
			
		||||
    cv.Schema(
 | 
			
		||||
 
 | 
			
		||||
@@ -41,7 +41,7 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
            cv.Optional(CONF_SERVICE_UUID): esp32_ble_tracker.bt_uuid,
 | 
			
		||||
            cv.Optional(CONF_IBEACON_MAJOR): cv.uint16_t,
 | 
			
		||||
            cv.Optional(CONF_IBEACON_MINOR): cv.uint16_t,
 | 
			
		||||
            cv.Optional(CONF_IBEACON_UUID): cv.uuid,
 | 
			
		||||
            cv.Optional(CONF_IBEACON_UUID): esp32_ble_tracker.bt_uuid,
 | 
			
		||||
            cv.Optional(CONF_TIMEOUT, default="5min"): cv.positive_time_period,
 | 
			
		||||
            cv.Optional(CONF_MIN_RSSI): cv.All(
 | 
			
		||||
                cv.decibel, cv.int_range(min=-100, max=-30)
 | 
			
		||||
@@ -83,7 +83,7 @@ async def to_code(config):
 | 
			
		||||
            cg.add(var.set_service_uuid128(uuid128))
 | 
			
		||||
 | 
			
		||||
    if ibeacon_uuid := config.get(CONF_IBEACON_UUID):
 | 
			
		||||
        ibeacon_uuid = esp32_ble_tracker.as_hex_array(str(ibeacon_uuid))
 | 
			
		||||
        ibeacon_uuid = esp32_ble_tracker.as_reversed_hex_array(ibeacon_uuid)
 | 
			
		||||
        cg.add(var.set_ibeacon_uuid(ibeacon_uuid))
 | 
			
		||||
 | 
			
		||||
        if (ibeacon_major := config.get(CONF_IBEACON_MAJOR)) is not None:
 | 
			
		||||
 
 | 
			
		||||
@@ -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));
 | 
			
		||||
 | 
			
		||||
    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);
 | 
			
		||||
  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() {
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "Bluetooth Proxy:");
 | 
			
		||||
  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() {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,96 +1,5 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import i2c, sensor
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_PRESSURE,
 | 
			
		||||
    CONF_TEMPERATURE,
 | 
			
		||||
    DEVICE_CLASS_PRESSURE,
 | 
			
		||||
    DEVICE_CLASS_TEMPERATURE,
 | 
			
		||||
    STATE_CLASS_MEASUREMENT,
 | 
			
		||||
    UNIT_CELSIUS,
 | 
			
		||||
    UNIT_HECTOPASCAL,
 | 
			
		||||
    CONF_IIR_FILTER,
 | 
			
		||||
    CONF_OVERSAMPLING,
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = cv.invalid(
 | 
			
		||||
    "The bmp280 sensor component has been renamed to bmp280_i2c."
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
DEPENDENCIES = ["i2c"]
 | 
			
		||||
 | 
			
		||||
bmp280_ns = cg.esphome_ns.namespace("bmp280")
 | 
			
		||||
BMP280Oversampling = bmp280_ns.enum("BMP280Oversampling")
 | 
			
		||||
OVERSAMPLING_OPTIONS = {
 | 
			
		||||
    "NONE": BMP280Oversampling.BMP280_OVERSAMPLING_NONE,
 | 
			
		||||
    "1X": BMP280Oversampling.BMP280_OVERSAMPLING_1X,
 | 
			
		||||
    "2X": BMP280Oversampling.BMP280_OVERSAMPLING_2X,
 | 
			
		||||
    "4X": BMP280Oversampling.BMP280_OVERSAMPLING_4X,
 | 
			
		||||
    "8X": BMP280Oversampling.BMP280_OVERSAMPLING_8X,
 | 
			
		||||
    "16X": BMP280Oversampling.BMP280_OVERSAMPLING_16X,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
BMP280IIRFilter = bmp280_ns.enum("BMP280IIRFilter")
 | 
			
		||||
IIR_FILTER_OPTIONS = {
 | 
			
		||||
    "OFF": BMP280IIRFilter.BMP280_IIR_FILTER_OFF,
 | 
			
		||||
    "2X": BMP280IIRFilter.BMP280_IIR_FILTER_2X,
 | 
			
		||||
    "4X": BMP280IIRFilter.BMP280_IIR_FILTER_4X,
 | 
			
		||||
    "8X": BMP280IIRFilter.BMP280_IIR_FILTER_8X,
 | 
			
		||||
    "16X": BMP280IIRFilter.BMP280_IIR_FILTER_16X,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
BMP280Component = bmp280_ns.class_(
 | 
			
		||||
    "BMP280Component", cg.PollingComponent, i2c.I2CDevice
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = (
 | 
			
		||||
    cv.Schema(
 | 
			
		||||
        {
 | 
			
		||||
            cv.GenerateID(): cv.declare_id(BMP280Component),
 | 
			
		||||
            cv.Optional(CONF_TEMPERATURE): sensor.sensor_schema(
 | 
			
		||||
                unit_of_measurement=UNIT_CELSIUS,
 | 
			
		||||
                accuracy_decimals=1,
 | 
			
		||||
                device_class=DEVICE_CLASS_TEMPERATURE,
 | 
			
		||||
                state_class=STATE_CLASS_MEASUREMENT,
 | 
			
		||||
            ).extend(
 | 
			
		||||
                {
 | 
			
		||||
                    cv.Optional(CONF_OVERSAMPLING, default="16X"): cv.enum(
 | 
			
		||||
                        OVERSAMPLING_OPTIONS, upper=True
 | 
			
		||||
                    ),
 | 
			
		||||
                }
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_PRESSURE): sensor.sensor_schema(
 | 
			
		||||
                unit_of_measurement=UNIT_HECTOPASCAL,
 | 
			
		||||
                accuracy_decimals=1,
 | 
			
		||||
                device_class=DEVICE_CLASS_PRESSURE,
 | 
			
		||||
                state_class=STATE_CLASS_MEASUREMENT,
 | 
			
		||||
            ).extend(
 | 
			
		||||
                {
 | 
			
		||||
                    cv.Optional(CONF_OVERSAMPLING, default="16X"): cv.enum(
 | 
			
		||||
                        OVERSAMPLING_OPTIONS, upper=True
 | 
			
		||||
                    ),
 | 
			
		||||
                }
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_IIR_FILTER, default="OFF"): cv.enum(
 | 
			
		||||
                IIR_FILTER_OPTIONS, upper=True
 | 
			
		||||
            ),
 | 
			
		||||
        }
 | 
			
		||||
    )
 | 
			
		||||
    .extend(cv.polling_component_schema("60s"))
 | 
			
		||||
    .extend(i2c.i2c_device_schema(0x77))
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    var = cg.new_Pvariable(config[CONF_ID])
 | 
			
		||||
    await cg.register_component(var, config)
 | 
			
		||||
    await i2c.register_i2c_device(var, config)
 | 
			
		||||
 | 
			
		||||
    if temperature_config := config.get(CONF_TEMPERATURE):
 | 
			
		||||
        sens = await sensor.new_sensor(temperature_config)
 | 
			
		||||
        cg.add(var.set_temperature_sensor(sens))
 | 
			
		||||
        cg.add(var.set_temperature_oversampling(temperature_config[CONF_OVERSAMPLING]))
 | 
			
		||||
 | 
			
		||||
    if pressure_config := config.get(CONF_PRESSURE):
 | 
			
		||||
        sens = await sensor.new_sensor(pressure_config)
 | 
			
		||||
        cg.add(var.set_pressure_sensor(sens))
 | 
			
		||||
        cg.add(var.set_pressure_oversampling(pressure_config[CONF_OVERSAMPLING]))
 | 
			
		||||
 | 
			
		||||
    cg.add(var.set_iir_filter(config[CONF_IIR_FILTER]))
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										88
									
								
								esphome/components/bmp280_base/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										88
									
								
								esphome/components/bmp280_base/__init__.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,88 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import sensor
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_IIR_FILTER,
 | 
			
		||||
    CONF_OVERSAMPLING,
 | 
			
		||||
    CONF_PRESSURE,
 | 
			
		||||
    CONF_TEMPERATURE,
 | 
			
		||||
    DEVICE_CLASS_PRESSURE,
 | 
			
		||||
    DEVICE_CLASS_TEMPERATURE,
 | 
			
		||||
    STATE_CLASS_MEASUREMENT,
 | 
			
		||||
    UNIT_CELSIUS,
 | 
			
		||||
    UNIT_HECTOPASCAL,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@ademuri"]
 | 
			
		||||
 | 
			
		||||
bmp280_ns = cg.esphome_ns.namespace("bmp280_base")
 | 
			
		||||
BMP280Oversampling = bmp280_ns.enum("BMP280Oversampling")
 | 
			
		||||
OVERSAMPLING_OPTIONS = {
 | 
			
		||||
    "NONE": BMP280Oversampling.BMP280_OVERSAMPLING_NONE,
 | 
			
		||||
    "1X": BMP280Oversampling.BMP280_OVERSAMPLING_1X,
 | 
			
		||||
    "2X": BMP280Oversampling.BMP280_OVERSAMPLING_2X,
 | 
			
		||||
    "4X": BMP280Oversampling.BMP280_OVERSAMPLING_4X,
 | 
			
		||||
    "8X": BMP280Oversampling.BMP280_OVERSAMPLING_8X,
 | 
			
		||||
    "16X": BMP280Oversampling.BMP280_OVERSAMPLING_16X,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
BMP280IIRFilter = bmp280_ns.enum("BMP280IIRFilter")
 | 
			
		||||
IIR_FILTER_OPTIONS = {
 | 
			
		||||
    "OFF": BMP280IIRFilter.BMP280_IIR_FILTER_OFF,
 | 
			
		||||
    "2X": BMP280IIRFilter.BMP280_IIR_FILTER_2X,
 | 
			
		||||
    "4X": BMP280IIRFilter.BMP280_IIR_FILTER_4X,
 | 
			
		||||
    "8X": BMP280IIRFilter.BMP280_IIR_FILTER_8X,
 | 
			
		||||
    "16X": BMP280IIRFilter.BMP280_IIR_FILTER_16X,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA_BASE = cv.Schema(
 | 
			
		||||
    {
 | 
			
		||||
        cv.Optional(CONF_TEMPERATURE): sensor.sensor_schema(
 | 
			
		||||
            unit_of_measurement=UNIT_CELSIUS,
 | 
			
		||||
            accuracy_decimals=1,
 | 
			
		||||
            device_class=DEVICE_CLASS_TEMPERATURE,
 | 
			
		||||
            state_class=STATE_CLASS_MEASUREMENT,
 | 
			
		||||
        ).extend(
 | 
			
		||||
            {
 | 
			
		||||
                cv.Optional(CONF_OVERSAMPLING, default="16X"): cv.enum(
 | 
			
		||||
                    OVERSAMPLING_OPTIONS, upper=True
 | 
			
		||||
                ),
 | 
			
		||||
            }
 | 
			
		||||
        ),
 | 
			
		||||
        cv.Optional(CONF_PRESSURE): sensor.sensor_schema(
 | 
			
		||||
            unit_of_measurement=UNIT_HECTOPASCAL,
 | 
			
		||||
            accuracy_decimals=1,
 | 
			
		||||
            device_class=DEVICE_CLASS_PRESSURE,
 | 
			
		||||
            state_class=STATE_CLASS_MEASUREMENT,
 | 
			
		||||
        ).extend(
 | 
			
		||||
            {
 | 
			
		||||
                cv.Optional(CONF_OVERSAMPLING, default="16X"): cv.enum(
 | 
			
		||||
                    OVERSAMPLING_OPTIONS, upper=True
 | 
			
		||||
                ),
 | 
			
		||||
            }
 | 
			
		||||
        ),
 | 
			
		||||
        cv.Optional(CONF_IIR_FILTER, default="OFF"): cv.enum(
 | 
			
		||||
            IIR_FILTER_OPTIONS, upper=True
 | 
			
		||||
        ),
 | 
			
		||||
    }
 | 
			
		||||
).extend(cv.polling_component_schema("60s"))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
async def to_code_base(config):
 | 
			
		||||
    var = cg.new_Pvariable(config[CONF_ID])
 | 
			
		||||
    await cg.register_component(var, config)
 | 
			
		||||
 | 
			
		||||
    if temperature_config := config.get(CONF_TEMPERATURE):
 | 
			
		||||
        sens = await sensor.new_sensor(temperature_config)
 | 
			
		||||
        cg.add(var.set_temperature_sensor(sens))
 | 
			
		||||
        cg.add(var.set_temperature_oversampling(temperature_config[CONF_OVERSAMPLING]))
 | 
			
		||||
 | 
			
		||||
    if pressure_config := config.get(CONF_PRESSURE):
 | 
			
		||||
        sens = await sensor.new_sensor(pressure_config)
 | 
			
		||||
        cg.add(var.set_pressure_sensor(sens))
 | 
			
		||||
        cg.add(var.set_pressure_oversampling(pressure_config[CONF_OVERSAMPLING]))
 | 
			
		||||
 | 
			
		||||
    cg.add(var.set_iir_filter(config[CONF_IIR_FILTER]))
 | 
			
		||||
 | 
			
		||||
    return var
 | 
			
		||||
@@ -1,9 +1,9 @@
 | 
			
		||||
#include "bmp280.h"
 | 
			
		||||
#include "bmp280_base.h"
 | 
			
		||||
#include "esphome/core/hal.h"
 | 
			
		||||
#include "esphome/core/log.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bmp280 {
 | 
			
		||||
namespace bmp280_base {
 | 
			
		||||
 | 
			
		||||
static const char *const TAG = "bmp280.sensor";
 | 
			
		||||
 | 
			
		||||
@@ -59,6 +59,14 @@ static const char *iir_filter_to_str(BMP280IIRFilter filter) {
 | 
			
		||||
void BMP280Component::setup() {
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "Setting up BMP280...");
 | 
			
		||||
  uint8_t chip_id = 0;
 | 
			
		||||
 | 
			
		||||
  // Read the chip id twice, to work around a bug where the first read is 0.
 | 
			
		||||
  // https://community.st.com/t5/stm32-mcus-products/issue-with-reading-bmp280-chip-id-using-spi/td-p/691855
 | 
			
		||||
  if (!this->read_byte(0xD0, &chip_id)) {
 | 
			
		||||
    this->error_code_ = COMMUNICATION_FAILED;
 | 
			
		||||
    this->mark_failed();
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
  if (!this->read_byte(0xD0, &chip_id)) {
 | 
			
		||||
    this->error_code_ = COMMUNICATION_FAILED;
 | 
			
		||||
    this->mark_failed();
 | 
			
		||||
@@ -122,7 +130,6 @@ void BMP280Component::setup() {
 | 
			
		||||
}
 | 
			
		||||
void BMP280Component::dump_config() {
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "BMP280:");
 | 
			
		||||
  LOG_I2C_DEVICE(this);
 | 
			
		||||
  switch (this->error_code_) {
 | 
			
		||||
    case COMMUNICATION_FAILED:
 | 
			
		||||
      ESP_LOGE(TAG, "Communication with BMP280 failed!");
 | 
			
		||||
@@ -262,5 +269,5 @@ uint16_t BMP280Component::read_u16_le_(uint8_t a_register) {
 | 
			
		||||
}
 | 
			
		||||
int16_t BMP280Component::read_s16_le_(uint8_t a_register) { return this->read_u16_le_(a_register); }
 | 
			
		||||
 | 
			
		||||
}  // namespace bmp280
 | 
			
		||||
}  // namespace bmp280_base
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
@@ -2,10 +2,9 @@
 | 
			
		||||
 | 
			
		||||
#include "esphome/core/component.h"
 | 
			
		||||
#include "esphome/components/sensor/sensor.h"
 | 
			
		||||
#include "esphome/components/i2c/i2c.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bmp280 {
 | 
			
		||||
namespace bmp280_base {
 | 
			
		||||
 | 
			
		||||
/// Internal struct storing the calibration values of an BMP280.
 | 
			
		||||
struct BMP280CalibrationData {
 | 
			
		||||
@@ -50,8 +49,8 @@ enum BMP280IIRFilter {
 | 
			
		||||
  BMP280_IIR_FILTER_16X = 0b100,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
/// This class implements support for the BMP280 Temperature+Pressure i2c sensor.
 | 
			
		||||
class BMP280Component : public PollingComponent, public i2c::I2CDevice {
 | 
			
		||||
/// This class implements support for the BMP280 Temperature+Pressure sensor.
 | 
			
		||||
class BMP280Component : public PollingComponent {
 | 
			
		||||
 public:
 | 
			
		||||
  void set_temperature_sensor(sensor::Sensor *temperature_sensor) { temperature_sensor_ = temperature_sensor; }
 | 
			
		||||
  void set_pressure_sensor(sensor::Sensor *pressure_sensor) { pressure_sensor_ = pressure_sensor; }
 | 
			
		||||
@@ -68,6 +67,11 @@ class BMP280Component : public PollingComponent, public i2c::I2CDevice {
 | 
			
		||||
  float get_setup_priority() const override;
 | 
			
		||||
  void update() override;
 | 
			
		||||
 | 
			
		||||
  virtual bool read_byte(uint8_t a_register, uint8_t *data) = 0;
 | 
			
		||||
  virtual bool write_byte(uint8_t a_register, uint8_t data) = 0;
 | 
			
		||||
  virtual bool read_bytes(uint8_t a_register, uint8_t *data, size_t len) = 0;
 | 
			
		||||
  virtual bool read_byte_16(uint8_t a_register, uint16_t *data) = 0;
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  /// Read the temperature value and store the calculated ambient temperature in t_fine.
 | 
			
		||||
  float read_temperature_(int32_t *t_fine);
 | 
			
		||||
@@ -90,5 +94,5 @@ class BMP280Component : public PollingComponent, public i2c::I2CDevice {
 | 
			
		||||
  } error_code_{NONE};
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace bmp280
 | 
			
		||||
}  // namespace bmp280_base
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										0
									
								
								esphome/components/bmp280_i2c/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										0
									
								
								esphome/components/bmp280_i2c/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
								
								
									
										27
									
								
								esphome/components/bmp280_i2c/bmp280_i2c.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								esphome/components/bmp280_i2c/bmp280_i2c.cpp
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,27 @@
 | 
			
		||||
#include "bmp280_i2c.h"
 | 
			
		||||
#include "esphome/core/hal.h"
 | 
			
		||||
#include "esphome/core/log.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bmp280_i2c {
 | 
			
		||||
 | 
			
		||||
bool BMP280I2CComponent::read_byte(uint8_t a_register, uint8_t *data) {
 | 
			
		||||
  return I2CDevice::read_byte(a_register, data);
 | 
			
		||||
};
 | 
			
		||||
bool BMP280I2CComponent::write_byte(uint8_t a_register, uint8_t data) {
 | 
			
		||||
  return I2CDevice::write_byte(a_register, data);
 | 
			
		||||
};
 | 
			
		||||
bool BMP280I2CComponent::read_bytes(uint8_t a_register, uint8_t *data, size_t len) {
 | 
			
		||||
  return I2CDevice::read_bytes(a_register, data, len);
 | 
			
		||||
};
 | 
			
		||||
bool BMP280I2CComponent::read_byte_16(uint8_t a_register, uint16_t *data) {
 | 
			
		||||
  return I2CDevice::read_byte_16(a_register, data);
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
void BMP280I2CComponent::dump_config() {
 | 
			
		||||
  LOG_I2C_DEVICE(this);
 | 
			
		||||
  BMP280Component::dump_config();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}  // namespace bmp280_i2c
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										22
									
								
								esphome/components/bmp280_i2c/bmp280_i2c.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								esphome/components/bmp280_i2c/bmp280_i2c.h
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,22 @@
 | 
			
		||||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "esphome/components/bmp280_base/bmp280_base.h"
 | 
			
		||||
#include "esphome/components/i2c/i2c.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bmp280_i2c {
 | 
			
		||||
 | 
			
		||||
static const char *const TAG = "bmp280_i2c.sensor";
 | 
			
		||||
 | 
			
		||||
/// This class implements support for the BMP280 Temperature+Pressure i2c sensor.
 | 
			
		||||
class BMP280I2CComponent : public esphome::bmp280_base::BMP280Component, public i2c::I2CDevice {
 | 
			
		||||
 public:
 | 
			
		||||
  bool read_byte(uint8_t a_register, uint8_t *data) override;
 | 
			
		||||
  bool write_byte(uint8_t a_register, uint8_t data) override;
 | 
			
		||||
  bool read_bytes(uint8_t a_register, uint8_t *data, size_t len) override;
 | 
			
		||||
  bool read_byte_16(uint8_t a_register, uint16_t *data) override;
 | 
			
		||||
  void dump_config() override;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace bmp280_i2c
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										22
									
								
								esphome/components/bmp280_i2c/sensor.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								esphome/components/bmp280_i2c/sensor.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,22 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import i2c
 | 
			
		||||
from ..bmp280_base import to_code_base, CONFIG_SCHEMA_BASE
 | 
			
		||||
 | 
			
		||||
AUTO_LOAD = ["bmp280_base"]
 | 
			
		||||
CODEOWNERS = ["@ademuri"]
 | 
			
		||||
DEPENDENCIES = ["i2c"]
 | 
			
		||||
 | 
			
		||||
bmp280_ns = cg.esphome_ns.namespace("bmp280_i2c")
 | 
			
		||||
BMP280I2CComponent = bmp280_ns.class_(
 | 
			
		||||
    "BMP280I2CComponent", cg.PollingComponent, i2c.I2CDevice
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = CONFIG_SCHEMA_BASE.extend(
 | 
			
		||||
    i2c.i2c_device_schema(default_address=0x77)
 | 
			
		||||
).extend({cv.GenerateID(): cv.declare_id(BMP280I2CComponent)})
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    var = await to_code_base(config)
 | 
			
		||||
    await i2c.register_i2c_device(var, config)
 | 
			
		||||
							
								
								
									
										0
									
								
								esphome/components/bmp280_spi/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										0
									
								
								esphome/components/bmp280_spi/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
								
								
									
										65
									
								
								esphome/components/bmp280_spi/bmp280_spi.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										65
									
								
								esphome/components/bmp280_spi/bmp280_spi.cpp
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,65 @@
 | 
			
		||||
#include <cstdint>
 | 
			
		||||
#include <cstddef>
 | 
			
		||||
 | 
			
		||||
#include "bmp280_spi.h"
 | 
			
		||||
#include <esphome/components/bmp280_base/bmp280_base.h>
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bmp280_spi {
 | 
			
		||||
 | 
			
		||||
uint8_t set_bit(uint8_t num, uint8_t position) {
 | 
			
		||||
  uint8_t mask = 1 << position;
 | 
			
		||||
  return num | mask;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
uint8_t clear_bit(uint8_t num, uint8_t position) {
 | 
			
		||||
  uint8_t mask = 1 << position;
 | 
			
		||||
  return num & ~mask;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void BMP280SPIComponent::setup() {
 | 
			
		||||
  this->spi_setup();
 | 
			
		||||
  BMP280Component::setup();
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// In SPI mode, only 7 bits of the register addresses are used; the MSB of register address is not used
 | 
			
		||||
// and replaced by a read/write bit (RW = ‘0’ for write and RW = ‘1’ for read).
 | 
			
		||||
// Example: address 0xF7 is accessed by using SPI register address 0x77. For write access, the byte
 | 
			
		||||
// 0x77 is transferred, for read access, the byte 0xF7 is transferred.
 | 
			
		||||
// https://www.bosch-sensortec.com/media/boschsensortec/downloads/datasheets/bst-bmp280-ds001.pdf
 | 
			
		||||
 | 
			
		||||
bool BMP280SPIComponent::read_byte(uint8_t a_register, uint8_t *data) {
 | 
			
		||||
  this->enable();
 | 
			
		||||
  this->transfer_byte(set_bit(a_register, 7));
 | 
			
		||||
  *data = this->transfer_byte(0);
 | 
			
		||||
  this->disable();
 | 
			
		||||
  return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool BMP280SPIComponent::write_byte(uint8_t a_register, uint8_t data) {
 | 
			
		||||
  this->enable();
 | 
			
		||||
  this->transfer_byte(clear_bit(a_register, 7));
 | 
			
		||||
  this->transfer_byte(data);
 | 
			
		||||
  this->disable();
 | 
			
		||||
  return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool BMP280SPIComponent::read_bytes(uint8_t a_register, uint8_t *data, size_t len) {
 | 
			
		||||
  this->enable();
 | 
			
		||||
  this->transfer_byte(set_bit(a_register, 7));
 | 
			
		||||
  this->read_array(data, len);
 | 
			
		||||
  this->disable();
 | 
			
		||||
  return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool BMP280SPIComponent::read_byte_16(uint8_t a_register, uint16_t *data) {
 | 
			
		||||
  this->enable();
 | 
			
		||||
  this->transfer_byte(set_bit(a_register, 7));
 | 
			
		||||
  ((uint8_t *) data)[1] = this->transfer_byte(0);
 | 
			
		||||
  ((uint8_t *) data)[0] = this->transfer_byte(0);
 | 
			
		||||
  this->disable();
 | 
			
		||||
  return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}  // namespace bmp280_spi
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										20
									
								
								esphome/components/bmp280_spi/bmp280_spi.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								esphome/components/bmp280_spi/bmp280_spi.h
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,20 @@
 | 
			
		||||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "esphome/components/bmp280_base/bmp280_base.h"
 | 
			
		||||
#include "esphome/components/spi/spi.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace bmp280_spi {
 | 
			
		||||
 | 
			
		||||
class BMP280SPIComponent : public esphome::bmp280_base::BMP280Component,
 | 
			
		||||
                           public spi::SPIDevice<spi::BIT_ORDER_MSB_FIRST, spi::CLOCK_POLARITY_LOW,
 | 
			
		||||
                                                 spi::CLOCK_PHASE_LEADING, spi::DATA_RATE_200KHZ> {
 | 
			
		||||
  void setup() override;
 | 
			
		||||
  bool read_byte(uint8_t a_register, uint8_t *data) override;
 | 
			
		||||
  bool write_byte(uint8_t a_register, uint8_t data) override;
 | 
			
		||||
  bool read_bytes(uint8_t a_register, uint8_t *data, size_t len) override;
 | 
			
		||||
  bool read_byte_16(uint8_t a_register, uint16_t *data) override;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace bmp280_spi
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										22
									
								
								esphome/components/bmp280_spi/sensor.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								esphome/components/bmp280_spi/sensor.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,22 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import spi
 | 
			
		||||
from ..bmp280_base import to_code_base, CONFIG_SCHEMA_BASE
 | 
			
		||||
 | 
			
		||||
AUTO_LOAD = ["bmp280_base"]
 | 
			
		||||
CODEOWNERS = ["@ademuri"]
 | 
			
		||||
DEPENDENCIES = ["spi"]
 | 
			
		||||
 | 
			
		||||
bmp280_ns = cg.esphome_ns.namespace("bmp280_spi")
 | 
			
		||||
BMP280SPIComponent = bmp280_ns.class_(
 | 
			
		||||
    "BMP280SPIComponent", cg.PollingComponent, spi.SPIDevice
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = CONFIG_SCHEMA_BASE.extend(
 | 
			
		||||
    spi.spi_device_schema(default_mode="mode3")
 | 
			
		||||
).extend({cv.GenerateID(): cv.declare_id(BMP280SPIComponent)})
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    var = await to_code_base(config)
 | 
			
		||||
    await spi.register_spi_device(var, config)
 | 
			
		||||
							
								
								
									
										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]
 | 
			
		||||
    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_))
 | 
			
		||||
    else:
 | 
			
		||||
        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]
 | 
			
		||||
    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_))
 | 
			
		||||
    else:
 | 
			
		||||
        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]
 | 
			
		||||
    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_))
 | 
			
		||||
    else:
 | 
			
		||||
        datetime_struct = cg.StructInitializer(
 | 
			
		||||
 
 | 
			
		||||
@@ -16,6 +16,8 @@
 | 
			
		||||
#include <esp32s2/rom/rtc.h>
 | 
			
		||||
#elif defined(USE_ESP32_VARIANT_ESP32S3)
 | 
			
		||||
#include <esp32s3/rom/rtc.h>
 | 
			
		||||
#elif defined(USE_ESP32_VARIANT_ESP32H2)
 | 
			
		||||
#include <esp32h2/rom/rtc.h>
 | 
			
		||||
#endif
 | 
			
		||||
#ifdef USE_ARDUINO
 | 
			
		||||
#include <Esp.h>
 | 
			
		||||
@@ -61,7 +63,7 @@ std::string DebugComponent::get_reset_reason_() {
 | 
			
		||||
    case RTCWDT_SYS_RESET:
 | 
			
		||||
      reset_reason = "RTC Watch Dog Reset Digital Core";
 | 
			
		||||
      break;
 | 
			
		||||
#if !defined(USE_ESP32_VARIANT_ESP32C6)
 | 
			
		||||
#if !defined(USE_ESP32_VARIANT_ESP32C6) && !defined(USE_ESP32_VARIANT_ESP32H2)
 | 
			
		||||
    case INTRUSION_RESET:
 | 
			
		||||
      reset_reason = "Intrusion Reset CPU";
 | 
			
		||||
      break;
 | 
			
		||||
 
 | 
			
		||||
@@ -1,15 +1,15 @@
 | 
			
		||||
from esphome import automation, core
 | 
			
		||||
from esphome.automation import maybe_simple_id
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome import core, automation
 | 
			
		||||
from esphome.automation import maybe_simple_id
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_AUTO_CLEAR_ENABLED,
 | 
			
		||||
    CONF_FROM,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_LAMBDA,
 | 
			
		||||
    CONF_PAGES,
 | 
			
		||||
    CONF_PAGE_ID,
 | 
			
		||||
    CONF_PAGES,
 | 
			
		||||
    CONF_ROTATION,
 | 
			
		||||
    CONF_FROM,
 | 
			
		||||
    CONF_TO,
 | 
			
		||||
    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)
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    cg.add_global(display_ns.using)
 | 
			
		||||
    cg.add_define("USE_DISPLAY")
 | 
			
		||||
 
 | 
			
		||||
@@ -256,6 +256,7 @@ bool Dsmr::parse_telegram() {
 | 
			
		||||
  MyData data;
 | 
			
		||||
  ESP_LOGV(TAG, "Trying to parse telegram");
 | 
			
		||||
  this->stop_requesting_data_();
 | 
			
		||||
 | 
			
		||||
  ::dsmr::ParseResult<void> res =
 | 
			
		||||
      ::dsmr::P1Parser::parse(&data, this->telegram_, this->bytes_read_, false,
 | 
			
		||||
                              this->crc_check_);  // Parse telegram according to data definition. Ignore unknown values.
 | 
			
		||||
@@ -267,6 +268,11 @@ bool Dsmr::parse_telegram() {
 | 
			
		||||
  } else {
 | 
			
		||||
    this->status_clear_warning();
 | 
			
		||||
    this->publish_sensors(data);
 | 
			
		||||
 | 
			
		||||
    // publish the telegram, after publishing the sensors so it can also trigger action based on latest values
 | 
			
		||||
    if (this->s_telegram_ != nullptr) {
 | 
			
		||||
      this->s_telegram_->publish_state(std::string(this->telegram_, this->bytes_read_));
 | 
			
		||||
    }
 | 
			
		||||
    return true;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -85,6 +85,9 @@ class Dsmr : public Component, public uart::UARTDevice {
 | 
			
		||||
  void set_##s(text_sensor::TextSensor *sensor) { s_##s##_ = sensor; }
 | 
			
		||||
  DSMR_TEXT_SENSOR_LIST(DSMR_SET_TEXT_SENSOR, )
 | 
			
		||||
 | 
			
		||||
  // handled outside dsmr
 | 
			
		||||
  void set_telegram(text_sensor::TextSensor *sensor) { s_telegram_ = sensor; }
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  void receive_telegram_();
 | 
			
		||||
  void receive_encrypted_telegram_();
 | 
			
		||||
@@ -124,6 +127,9 @@ class Dsmr : public Component, public uart::UARTDevice {
 | 
			
		||||
  bool header_found_{false};
 | 
			
		||||
  bool footer_found_{false};
 | 
			
		||||
 | 
			
		||||
  // handled outside dsmr
 | 
			
		||||
  text_sensor::TextSensor *s_telegram_{nullptr};
 | 
			
		||||
 | 
			
		||||
// Sensor member pointers
 | 
			
		||||
#define DSMR_DECLARE_SENSOR(s) sensor::Sensor *s_##s##_{nullptr};
 | 
			
		||||
  DSMR_SENSOR_LIST(DSMR_DECLARE_SENSOR, )
 | 
			
		||||
 
 | 
			
		||||
@@ -1,7 +1,7 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import text_sensor
 | 
			
		||||
 | 
			
		||||
from esphome.const import CONF_INTERNAL
 | 
			
		||||
from . import Dsmr, CONF_DSMR_ID
 | 
			
		||||
 | 
			
		||||
AUTO_LOAD = ["dsmr"]
 | 
			
		||||
@@ -22,6 +22,9 @@ CONFIG_SCHEMA = cv.Schema(
 | 
			
		||||
        cv.Optional("water_equipment_id"): text_sensor.text_sensor_schema(),
 | 
			
		||||
        cv.Optional("sub_equipment_id"): text_sensor.text_sensor_schema(),
 | 
			
		||||
        cv.Optional("gas_delivered_text"): text_sensor.text_sensor_schema(),
 | 
			
		||||
        cv.Optional("telegram"): text_sensor.text_sensor_schema().extend(
 | 
			
		||||
            {cv.Optional(CONF_INTERNAL, default=True): cv.boolean}
 | 
			
		||||
        ),
 | 
			
		||||
    }
 | 
			
		||||
).extend(cv.COMPONENT_SCHEMA)
 | 
			
		||||
 | 
			
		||||
@@ -37,7 +40,9 @@ async def to_code(config):
 | 
			
		||||
        if id and id.type == text_sensor.TextSensor:
 | 
			
		||||
            var = await text_sensor.new_text_sensor(conf)
 | 
			
		||||
            cg.add(getattr(hub, f"set_{key}")(var))
 | 
			
		||||
            text_sensors.append(f"F({key})")
 | 
			
		||||
            if key != "telegram":
 | 
			
		||||
                # telegram is not handled by dsmr
 | 
			
		||||
                text_sensors.append(f"F({key})")
 | 
			
		||||
 | 
			
		||||
    if text_sensors:
 | 
			
		||||
        cg.add_define(
 | 
			
		||||
 
 | 
			
		||||
@@ -31,6 +31,13 @@ ESPBTUUID ESPBTUUID::from_raw(const uint8_t *data) {
 | 
			
		||||
  memcpy(ret.uuid_.uuid.uuid128, data, ESP_UUID_LEN_128);
 | 
			
		||||
  return ret;
 | 
			
		||||
}
 | 
			
		||||
ESPBTUUID ESPBTUUID::from_raw_reversed(const uint8_t *data) {
 | 
			
		||||
  ESPBTUUID ret;
 | 
			
		||||
  ret.uuid_.len = ESP_UUID_LEN_128;
 | 
			
		||||
  for (int i = 0; i < ESP_UUID_LEN_128; i++)
 | 
			
		||||
    ret.uuid_.uuid.uuid128[ESP_UUID_LEN_128 - 1 - i] = data[i];
 | 
			
		||||
  return ret;
 | 
			
		||||
}
 | 
			
		||||
ESPBTUUID ESPBTUUID::from_raw(const std::string &data) {
 | 
			
		||||
  ESPBTUUID ret;
 | 
			
		||||
  if (data.length() == 4) {
 | 
			
		||||
 
 | 
			
		||||
@@ -20,6 +20,7 @@ class ESPBTUUID {
 | 
			
		||||
  static ESPBTUUID from_uint32(uint32_t uuid);
 | 
			
		||||
 | 
			
		||||
  static ESPBTUUID from_raw(const uint8_t *data);
 | 
			
		||||
  static ESPBTUUID from_raw_reversed(const uint8_t *data);
 | 
			
		||||
 | 
			
		||||
  static ESPBTUUID from_raw(const std::string &data);
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -462,14 +462,16 @@ void ESPBTDevice::parse_scan_rst(const esp_ble_gap_cb_param_t::ble_scan_result_e
 | 
			
		||||
    ESP_LOGVV(TAG, "  Service UUID: %s", uuid.to_string().c_str());
 | 
			
		||||
  }
 | 
			
		||||
  for (auto &data : this->manufacturer_datas_) {
 | 
			
		||||
    ESP_LOGVV(TAG, "  Manufacturer data: %s", format_hex_pretty(data.data).c_str());
 | 
			
		||||
    if (this->get_ibeacon().has_value()) {
 | 
			
		||||
      auto ibeacon = this->get_ibeacon().value();
 | 
			
		||||
      ESP_LOGVV(TAG, "    iBeacon data:");
 | 
			
		||||
      ESP_LOGVV(TAG, "      UUID: %s", ibeacon.get_uuid().to_string().c_str());
 | 
			
		||||
      ESP_LOGVV(TAG, "      Major: %u", ibeacon.get_major());
 | 
			
		||||
      ESP_LOGVV(TAG, "      Minor: %u", ibeacon.get_minor());
 | 
			
		||||
      ESP_LOGVV(TAG, "      TXPower: %d", ibeacon.get_signal_power());
 | 
			
		||||
    auto ibeacon = ESPBLEiBeacon::from_manufacturer_data(data);
 | 
			
		||||
    if (ibeacon.has_value()) {
 | 
			
		||||
      ESP_LOGVV(TAG, "  Manufacturer iBeacon:");
 | 
			
		||||
      ESP_LOGVV(TAG, "    UUID: %s", ibeacon.value().get_uuid().to_string().c_str());
 | 
			
		||||
      ESP_LOGVV(TAG, "    Major: %u", ibeacon.value().get_major());
 | 
			
		||||
      ESP_LOGVV(TAG, "    Minor: %u", ibeacon.value().get_minor());
 | 
			
		||||
      ESP_LOGVV(TAG, "    TXPower: %d", ibeacon.value().get_signal_power());
 | 
			
		||||
    } else {
 | 
			
		||||
      ESP_LOGVV(TAG, "  Manufacturer ID: %s, data: %s", data.uuid.to_string().c_str(),
 | 
			
		||||
                format_hex_pretty(data.data).c_str());
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  for (auto &data : this->service_datas_) {
 | 
			
		||||
@@ -478,7 +480,7 @@ void ESPBTDevice::parse_scan_rst(const esp_ble_gap_cb_param_t::ble_scan_result_e
 | 
			
		||||
    ESP_LOGVV(TAG, "    Data: %s", format_hex_pretty(data.data).c_str());
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  ESP_LOGVV(TAG, "Adv data: %s", format_hex_pretty(param.ble_adv, param.adv_data_len + param.scan_rsp_len).c_str());
 | 
			
		||||
  ESP_LOGVV(TAG, "  Adv data: %s", format_hex_pretty(param.ble_adv, param.adv_data_len + param.scan_rsp_len).c_str());
 | 
			
		||||
#endif
 | 
			
		||||
}
 | 
			
		||||
void ESPBTDevice::parse_adv_(const esp_ble_gap_cb_param_t::ble_scan_result_evt_param ¶m) {
 | 
			
		||||
 
 | 
			
		||||
@@ -44,10 +44,10 @@ class ESPBLEiBeacon {
 | 
			
		||||
  ESPBLEiBeacon(const uint8_t *data);
 | 
			
		||||
  static optional<ESPBLEiBeacon> from_manufacturer_data(const ServiceData &data);
 | 
			
		||||
 | 
			
		||||
  uint16_t get_major() { return ((this->beacon_data_.major & 0xFF) << 8) | (this->beacon_data_.major >> 8); }
 | 
			
		||||
  uint16_t get_minor() { return ((this->beacon_data_.minor & 0xFF) << 8) | (this->beacon_data_.minor >> 8); }
 | 
			
		||||
  uint16_t get_major() { return byteswap(this->beacon_data_.major); }
 | 
			
		||||
  uint16_t get_minor() { return byteswap(this->beacon_data_.minor); }
 | 
			
		||||
  int8_t get_signal_power() { return this->beacon_data_.signal_power; }
 | 
			
		||||
  ESPBTUUID get_uuid() { return ESPBTUUID::from_raw(this->beacon_data_.proximity_uuid); }
 | 
			
		||||
  ESPBTUUID get_uuid() { return ESPBTUUID::from_raw_reversed(this->beacon_data_.proximity_uuid); }
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -140,6 +140,8 @@ CONF_TEST_PATTERN = "test_pattern"
 | 
			
		||||
# framerates
 | 
			
		||||
CONF_MAX_FRAMERATE = "max_framerate"
 | 
			
		||||
CONF_IDLE_FRAMERATE = "idle_framerate"
 | 
			
		||||
# frame buffer
 | 
			
		||||
CONF_FRAME_BUFFER_COUNT = "frame_buffer_count"
 | 
			
		||||
 | 
			
		||||
# stream trigger
 | 
			
		||||
CONF_ON_STREAM_START = "on_stream_start"
 | 
			
		||||
@@ -213,6 +215,7 @@ CONFIG_SCHEMA = cv.ENTITY_BASE_SCHEMA.extend(
 | 
			
		||||
        cv.Optional(CONF_IDLE_FRAMERATE, default="0.1 fps"): cv.All(
 | 
			
		||||
            cv.framerate, cv.Range(min=0, max=1)
 | 
			
		||||
        ),
 | 
			
		||||
        cv.Optional(CONF_FRAME_BUFFER_COUNT, default=1): cv.int_range(min=1, max=2),
 | 
			
		||||
        cv.Optional(CONF_ON_STREAM_START): automation.validate_automation(
 | 
			
		||||
            {
 | 
			
		||||
                cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(
 | 
			
		||||
@@ -285,6 +288,7 @@ async def to_code(config):
 | 
			
		||||
        cg.add(var.set_idle_update_interval(0))
 | 
			
		||||
    else:
 | 
			
		||||
        cg.add(var.set_idle_update_interval(1000 / config[CONF_IDLE_FRAMERATE]))
 | 
			
		||||
    cg.add(var.set_frame_buffer_count(config[CONF_FRAME_BUFFER_COUNT]))
 | 
			
		||||
    cg.add(var.set_frame_size(config[CONF_RESOLUTION]))
 | 
			
		||||
 | 
			
		||||
    cg.add_define("USE_ESP32_CAMERA")
 | 
			
		||||
 
 | 
			
		||||
@@ -127,7 +127,7 @@ void ESP32Camera::dump_config() {
 | 
			
		||||
  sensor_t *s = esp_camera_sensor_get();
 | 
			
		||||
  auto st = s->status;
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  JPEG Quality: %u", st.quality);
 | 
			
		||||
  // ESP_LOGCONFIG(TAG, "  Framebuffer Count: %u", conf.fb_count);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Framebuffer Count: %u", conf.fb_count);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Contrast: %d", st.contrast);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Brightness: %d", st.brightness);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Saturation: %d", st.saturation);
 | 
			
		||||
@@ -212,6 +212,8 @@ ESP32Camera::ESP32Camera() {
 | 
			
		||||
  this->config_.frame_size = FRAMESIZE_VGA;  // 640x480
 | 
			
		||||
  this->config_.jpeg_quality = 10;
 | 
			
		||||
  this->config_.fb_count = 1;
 | 
			
		||||
  this->config_.grab_mode = CAMERA_GRAB_WHEN_EMPTY;
 | 
			
		||||
  this->config_.fb_location = CAMERA_FB_IN_PSRAM;
 | 
			
		||||
 | 
			
		||||
  global_esp32_camera = this;
 | 
			
		||||
}
 | 
			
		||||
@@ -333,6 +335,12 @@ void ESP32Camera::set_max_update_interval(uint32_t max_update_interval) {
 | 
			
		||||
void ESP32Camera::set_idle_update_interval(uint32_t idle_update_interval) {
 | 
			
		||||
  this->idle_update_interval_ = idle_update_interval;
 | 
			
		||||
}
 | 
			
		||||
/* set frame buffer parameters */
 | 
			
		||||
void ESP32Camera::set_frame_buffer_mode(camera_grab_mode_t mode) { this->config_.grab_mode = mode; }
 | 
			
		||||
void ESP32Camera::set_frame_buffer_count(uint8_t fb_count) {
 | 
			
		||||
  this->config_.fb_count = fb_count;
 | 
			
		||||
  this->set_frame_buffer_mode(fb_count > 1 ? CAMERA_GRAB_LATEST : CAMERA_GRAB_WHEN_EMPTY);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/* ---------------- public API (specific) ---------------- */
 | 
			
		||||
void ESP32Camera::add_image_callback(std::function<void(std::shared_ptr<CameraImage>)> &&callback) {
 | 
			
		||||
 
 | 
			
		||||
@@ -145,6 +145,9 @@ class ESP32Camera : public Component, public EntityBase {
 | 
			
		||||
  /* -- framerates */
 | 
			
		||||
  void set_max_update_interval(uint32_t max_update_interval);
 | 
			
		||||
  void set_idle_update_interval(uint32_t idle_update_interval);
 | 
			
		||||
  /* -- frame buffer */
 | 
			
		||||
  void set_frame_buffer_mode(camera_grab_mode_t mode);
 | 
			
		||||
  void set_frame_buffer_count(uint8_t fb_count);
 | 
			
		||||
 | 
			
		||||
  /* public API (derivated) */
 | 
			
		||||
  void setup() override;
 | 
			
		||||
 
 | 
			
		||||
@@ -1,18 +1,23 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome import pins
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
from esphome.components import canbus
 | 
			
		||||
from esphome.const import CONF_ID, CONF_RX_PIN, CONF_TX_PIN
 | 
			
		||||
from esphome.components.canbus import CanbusComponent, CanSpeed, CONF_BIT_RATE
 | 
			
		||||
 | 
			
		||||
from esphome.components.canbus import CONF_BIT_RATE, CanbusComponent, CanSpeed
 | 
			
		||||
from esphome.components.esp32 import get_esp32_variant
 | 
			
		||||
from esphome.components.esp32.const import (
 | 
			
		||||
    VARIANT_ESP32,
 | 
			
		||||
    VARIANT_ESP32S2,
 | 
			
		||||
    VARIANT_ESP32S3,
 | 
			
		||||
    VARIANT_ESP32C3,
 | 
			
		||||
    VARIANT_ESP32C6,
 | 
			
		||||
    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"]
 | 
			
		||||
@@ -77,6 +82,8 @@ CONFIG_SCHEMA = canbus.CANBUS_SCHEMA.extend(
 | 
			
		||||
        cv.Optional(CONF_BIT_RATE, default="125KBPS"): validate_bit_rate,
 | 
			
		||||
        cv.Required(CONF_RX_PIN): pins.internal_gpio_input_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_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() {
 | 
			
		||||
  twai_general_config_t g_config =
 | 
			
		||||
      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_timing_config_t t_config;
 | 
			
		||||
 | 
			
		||||
@@ -111,6 +118,7 @@ canbus::Error ESP32Can::send_message(struct canbus::CanFrame *frame) {
 | 
			
		||||
      .flags = flags,
 | 
			
		||||
      .identifier = frame->can_id,
 | 
			
		||||
      .data_length_code = frame->can_data_length_code,
 | 
			
		||||
      .data = {},  // to suppress warning, data is initialized properly below
 | 
			
		||||
  };
 | 
			
		||||
  if (!frame->remote_transmission_request) {
 | 
			
		||||
    memcpy(message.data, frame->data, frame->can_data_length_code);
 | 
			
		||||
 
 | 
			
		||||
@@ -12,6 +12,8 @@ class ESP32Can : public canbus::Canbus {
 | 
			
		||||
 public:
 | 
			
		||||
  void set_rx(int rx) { rx_ = rx; }
 | 
			
		||||
  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(){};
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
@@ -21,6 +23,8 @@ class ESP32Can : public canbus::Canbus {
 | 
			
		||||
 | 
			
		||||
  int rx_{-1};
 | 
			
		||||
  int tx_{-1};
 | 
			
		||||
  optional<uint32_t> tx_queue_len_{};
 | 
			
		||||
  optional<uint32_t> rx_queue_len_{};
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace esp32_can
 | 
			
		||||
 
 | 
			
		||||
@@ -38,7 +38,8 @@ void ESP32RMTLEDStripLightOutput::setup() {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  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;
 | 
			
		||||
  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,
 | 
			
		||||
                                                 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;
 | 
			
		||||
 | 
			
		||||
  // 0-bit
 | 
			
		||||
@@ -79,6 +80,11 @@ void ESP32RMTLEDStripLightOutput::set_led_params(uint32_t bit0_high, uint32_t bi
 | 
			
		||||
  this->bit1_.level0 = 1;
 | 
			
		||||
  this->bit1_.duration1 = (uint32_t) (ratio * bit1_low);
 | 
			
		||||
  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) {
 | 
			
		||||
@@ -118,6 +124,12 @@ void ESP32RMTLEDStripLightOutput::write_state(light::LightState *state) {
 | 
			
		||||
    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) {
 | 
			
		||||
    ESP_LOGE(TAG, "RMT TX error");
 | 
			
		||||
    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.
 | 
			
		||||
  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_rmt_channel(rmt_channel_t channel) { this->channel_ = channel; }
 | 
			
		||||
@@ -75,7 +76,7 @@ class ESP32RMTLEDStripLightOutput : public light::AddressableLight {
 | 
			
		||||
  bool is_rgbw_;
 | 
			
		||||
  bool is_wrgb_;
 | 
			
		||||
 | 
			
		||||
  rmt_item32_t bit0_, bit1_;
 | 
			
		||||
  rmt_item32_t bit0_, bit1_, reset_;
 | 
			
		||||
  RGBOrder rgb_order_;
 | 
			
		||||
  rmt_channel_t channel_;
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -43,13 +43,16 @@ class LEDStripTimings:
 | 
			
		||||
    bit0_low: int
 | 
			
		||||
    bit1_high: int
 | 
			
		||||
    bit1_low: int
 | 
			
		||||
    reset_high: int
 | 
			
		||||
    reset_low: int
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
CHIPSETS = {
 | 
			
		||||
    "WS2812": LEDStripTimings(400, 1000, 1000, 400),
 | 
			
		||||
    "SK6812": LEDStripTimings(300, 900, 600, 600),
 | 
			
		||||
    "APA106": LEDStripTimings(350, 1360, 1360, 350),
 | 
			
		||||
    "SM16703": LEDStripTimings(300, 900, 900, 300),
 | 
			
		||||
    "WS2811": LEDStripTimings(300, 1090, 1090, 320, 0, 300000),
 | 
			
		||||
    "WS2812": LEDStripTimings(400, 1000, 1000, 400, 0, 0),
 | 
			
		||||
    "SK6812": LEDStripTimings(300, 900, 600, 600, 0, 0),
 | 
			
		||||
    "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_BIT1_HIGH = "bit1_high"
 | 
			
		||||
CONF_BIT1_LOW = "bit1_low"
 | 
			
		||||
CONF_RESET_HIGH = "reset_high"
 | 
			
		||||
CONF_RESET_LOW = "reset_low"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = cv.All(
 | 
			
		||||
@@ -88,6 +93,14 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
                CONF_BIT1_LOW,
 | 
			
		||||
                "custom",
 | 
			
		||||
            ): 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),
 | 
			
		||||
@@ -113,6 +126,8 @@ async def to_code(config):
 | 
			
		||||
                chipset.bit0_low,
 | 
			
		||||
                chipset.bit1_high,
 | 
			
		||||
                chipset.bit1_low,
 | 
			
		||||
                chipset.reset_high,
 | 
			
		||||
                chipset.reset_low,
 | 
			
		||||
            )
 | 
			
		||||
        )
 | 
			
		||||
    else:
 | 
			
		||||
@@ -122,6 +137,8 @@ async def to_code(config):
 | 
			
		||||
                config[CONF_BIT0_LOW],
 | 
			
		||||
                config[CONF_BIT1_HIGH],
 | 
			
		||||
                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) {
 | 
			
		||||
      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->status_set_warning();
 | 
			
		||||
 
 | 
			
		||||
@@ -1,43 +1,35 @@
 | 
			
		||||
import functools
 | 
			
		||||
import hashlib
 | 
			
		||||
import logging
 | 
			
		||||
 | 
			
		||||
import functools
 | 
			
		||||
from pathlib import Path
 | 
			
		||||
import os
 | 
			
		||||
from pathlib import Path
 | 
			
		||||
import re
 | 
			
		||||
 | 
			
		||||
from packaging import version
 | 
			
		||||
import requests
 | 
			
		||||
 | 
			
		||||
from esphome import core
 | 
			
		||||
from esphome import external_files
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome import core, external_files
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
from esphome.helpers import (
 | 
			
		||||
    copy_file_if_changed,
 | 
			
		||||
    cpp_string_escape,
 | 
			
		||||
)
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_FAMILY,
 | 
			
		||||
    CONF_FILE,
 | 
			
		||||
    CONF_GLYPHS,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_PATH,
 | 
			
		||||
    CONF_RAW_DATA_ID,
 | 
			
		||||
    CONF_TYPE,
 | 
			
		||||
    CONF_REFRESH,
 | 
			
		||||
    CONF_SIZE,
 | 
			
		||||
    CONF_PATH,
 | 
			
		||||
    CONF_WEIGHT,
 | 
			
		||||
    CONF_TYPE,
 | 
			
		||||
    CONF_URL,
 | 
			
		||||
    CONF_WEIGHT,
 | 
			
		||||
)
 | 
			
		||||
from esphome.core import (
 | 
			
		||||
    CORE,
 | 
			
		||||
    HexInt,
 | 
			
		||||
)
 | 
			
		||||
from esphome.core import CORE, HexInt
 | 
			
		||||
from esphome.helpers import copy_file_if_changed, cpp_string_escape
 | 
			
		||||
 | 
			
		||||
_LOGGER = logging.getLogger(__name__)
 | 
			
		||||
 | 
			
		||||
DOMAIN = "font"
 | 
			
		||||
DEPENDENCIES = ["display"]
 | 
			
		||||
MULTI_CONF = True
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@esphome/core", "@clydebarrow"]
 | 
			
		||||
@@ -400,10 +392,7 @@ class EFont:
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def convert_bitmap_to_pillow_font(filepath):
 | 
			
		||||
    from PIL import (
 | 
			
		||||
        PcfFontFile,
 | 
			
		||||
        BdfFontFile,
 | 
			
		||||
    )
 | 
			
		||||
    from PIL import BdfFontFile, PcfFontFile
 | 
			
		||||
 | 
			
		||||
    local_bitmap_font_file = external_files.compute_local_file_dir(
 | 
			
		||||
        DOMAIN,
 | 
			
		||||
 
 | 
			
		||||
@@ -1,9 +1,8 @@
 | 
			
		||||
#include "font.h"
 | 
			
		||||
 | 
			
		||||
#include "esphome/core/color.h"
 | 
			
		||||
#include "esphome/core/hal.h"
 | 
			
		||||
#include "esphome/core/log.h"
 | 
			
		||||
#include "esphome/core/color.h"
 | 
			
		||||
#include "esphome/components/display/display_buffer.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace font {
 | 
			
		||||
@@ -68,6 +67,7 @@ int Font::match_next_glyph(const uint8_t *str, int *match_length) {
 | 
			
		||||
    return -1;
 | 
			
		||||
  return lo;
 | 
			
		||||
}
 | 
			
		||||
#ifdef USE_DISPLAY
 | 
			
		||||
void Font::measure(const char *str, int *width, int *x_offset, int *baseline, int *height) {
 | 
			
		||||
  *baseline = this->baseline_;
 | 
			
		||||
  *height = this->height_;
 | 
			
		||||
@@ -164,6 +164,7 @@ void Font::print(int x_start, int y_start, display::Display *display, Color colo
 | 
			
		||||
    i += match_length;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
}  // namespace font
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
 
 | 
			
		||||
@@ -1,8 +1,11 @@
 | 
			
		||||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "esphome/core/datatypes.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 font {
 | 
			
		||||
@@ -38,7 +41,11 @@ class Glyph {
 | 
			
		||||
  const GlyphData *glyph_data_;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class Font : public display::BaseFont {
 | 
			
		||||
class Font
 | 
			
		||||
#ifdef USE_DISPLAY
 | 
			
		||||
    : public display::BaseFont
 | 
			
		||||
#endif
 | 
			
		||||
{
 | 
			
		||||
 public:
 | 
			
		||||
  /** 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);
 | 
			
		||||
 | 
			
		||||
#ifdef USE_DISPLAY
 | 
			
		||||
  void print(int x_start, int y_start, display::Display *display, Color color, const char *text,
 | 
			
		||||
             Color background) 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_height() { return this->height_; }
 | 
			
		||||
  inline int get_bpp() { return this->bpp_; }
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.components import climate_ir
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import CONF_ID, CONF_MODEL
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@orestismers"]
 | 
			
		||||
@@ -17,6 +17,7 @@ MODELS = {
 | 
			
		||||
    "yaa": Model.GREE_YAA,
 | 
			
		||||
    "yac": Model.GREE_YAC,
 | 
			
		||||
    "yac1fb9": Model.GREE_YAC1FB9,
 | 
			
		||||
    "yx1ff": Model.GREE_YX1FF,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = climate_ir.CLIMATE_IR_WITH_RECEIVER_SCHEMA.extend(
 | 
			
		||||
 
 | 
			
		||||
@@ -6,7 +6,15 @@ namespace gree {
 | 
			
		||||
 | 
			
		||||
static const char *const TAG = "gree.climate";
 | 
			
		||||
 | 
			
		||||
void GreeClimate::set_model(Model model) { this->model_ = model; }
 | 
			
		||||
void GreeClimate::set_model(Model model) {
 | 
			
		||||
  if (model == GREE_YX1FF) {
 | 
			
		||||
    this->fan_modes_.insert(climate::CLIMATE_FAN_QUIET);   // YX1FF 4 speed
 | 
			
		||||
    this->presets_.insert(climate::CLIMATE_PRESET_NONE);   // YX1FF sleep mode
 | 
			
		||||
    this->presets_.insert(climate::CLIMATE_PRESET_SLEEP);  // YX1FF sleep mode
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  this->model_ = model;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void GreeClimate::transmit_state() {
 | 
			
		||||
  uint8_t remote_state[8] = {0x00, 0x00, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00};
 | 
			
		||||
@@ -14,7 +22,7 @@ void GreeClimate::transmit_state() {
 | 
			
		||||
  remote_state[0] = this->fan_speed_() | this->operation_mode_();
 | 
			
		||||
  remote_state[1] = this->temperature_();
 | 
			
		||||
 | 
			
		||||
  if (this->model_ == GREE_YAN) {
 | 
			
		||||
  if (this->model_ == GREE_YAN || this->model_ == GREE_YX1FF) {
 | 
			
		||||
    remote_state[2] = 0x60;
 | 
			
		||||
    remote_state[3] = 0x50;
 | 
			
		||||
    remote_state[4] = this->vertical_swing_();
 | 
			
		||||
@@ -36,8 +44,18 @@ void GreeClimate::transmit_state() {
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  if (this->model_ == GREE_YX1FF) {
 | 
			
		||||
    if (this->fan_speed_() == GREE_FAN_TURBO) {
 | 
			
		||||
      remote_state[2] |= GREE_FAN_TURBO_BIT;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (this->preset_() == GREE_PRESET_SLEEP) {
 | 
			
		||||
      remote_state[0] |= GREE_PRESET_SLEEP_BIT;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  // Calculate the checksum
 | 
			
		||||
  if (this->model_ == GREE_YAN) {
 | 
			
		||||
  if (this->model_ == GREE_YAN || this->model_ == GREE_YX1FF) {
 | 
			
		||||
    remote_state[7] = ((remote_state[0] << 4) + (remote_state[1] << 4) + 0xC0);
 | 
			
		||||
  } else {
 | 
			
		||||
    remote_state[7] =
 | 
			
		||||
@@ -124,6 +142,23 @@ uint8_t GreeClimate::operation_mode_() {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
uint8_t GreeClimate::fan_speed_() {
 | 
			
		||||
  // YX1FF has 4 fan speeds -- we treat low as quiet and turbo as high
 | 
			
		||||
  if (this->model_ == GREE_YX1FF) {
 | 
			
		||||
    switch (this->fan_mode.value()) {
 | 
			
		||||
      case climate::CLIMATE_FAN_QUIET:
 | 
			
		||||
        return GREE_FAN_1;
 | 
			
		||||
      case climate::CLIMATE_FAN_LOW:
 | 
			
		||||
        return GREE_FAN_2;
 | 
			
		||||
      case climate::CLIMATE_FAN_MEDIUM:
 | 
			
		||||
        return GREE_FAN_3;
 | 
			
		||||
      case climate::CLIMATE_FAN_HIGH:
 | 
			
		||||
        return GREE_FAN_TURBO;
 | 
			
		||||
      case climate::CLIMATE_FAN_AUTO:
 | 
			
		||||
      default:
 | 
			
		||||
        return GREE_FAN_AUTO;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  switch (this->fan_mode.value()) {
 | 
			
		||||
    case climate::CLIMATE_FAN_LOW:
 | 
			
		||||
      return GREE_FAN_1;
 | 
			
		||||
@@ -161,5 +196,21 @@ uint8_t GreeClimate::temperature_() {
 | 
			
		||||
  return (uint8_t) roundf(clamp<float>(this->target_temperature, GREE_TEMP_MIN, GREE_TEMP_MAX));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
uint8_t GreeClimate::preset_() {
 | 
			
		||||
  // YX1FF has sleep preset
 | 
			
		||||
  if (this->model_ == GREE_YX1FF) {
 | 
			
		||||
    switch (this->preset.value()) {
 | 
			
		||||
      case climate::CLIMATE_PRESET_NONE:
 | 
			
		||||
        return GREE_PRESET_NONE;
 | 
			
		||||
      case climate::CLIMATE_PRESET_SLEEP:
 | 
			
		||||
        return GREE_PRESET_SLEEP;
 | 
			
		||||
      default:
 | 
			
		||||
        return GREE_PRESET_NONE;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  return GREE_PRESET_NONE;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}  // namespace gree
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
 
 | 
			
		||||
@@ -25,7 +25,6 @@ const uint8_t GREE_FAN_AUTO = 0x00;
 | 
			
		||||
const uint8_t GREE_FAN_1 = 0x10;
 | 
			
		||||
const uint8_t GREE_FAN_2 = 0x20;
 | 
			
		||||
const uint8_t GREE_FAN_3 = 0x30;
 | 
			
		||||
const uint8_t GREE_FAN_TURBO = 0x80;
 | 
			
		||||
 | 
			
		||||
// IR Transmission
 | 
			
		||||
const uint32_t GREE_IR_FREQUENCY = 38000;
 | 
			
		||||
@@ -70,8 +69,16 @@ const uint8_t GREE_HDIR_MIDDLE = 0x04;
 | 
			
		||||
const uint8_t GREE_HDIR_MRIGHT = 0x05;
 | 
			
		||||
const uint8_t GREE_HDIR_RIGHT = 0x06;
 | 
			
		||||
 | 
			
		||||
// Only available on YX1FF
 | 
			
		||||
// Turbo (high) fan mode + sleep preset mode
 | 
			
		||||
const uint8_t GREE_FAN_TURBO = 0x80;
 | 
			
		||||
const uint8_t GREE_FAN_TURBO_BIT = 0x10;
 | 
			
		||||
const uint8_t GREE_PRESET_NONE = 0x00;
 | 
			
		||||
const uint8_t GREE_PRESET_SLEEP = 0x01;
 | 
			
		||||
const uint8_t GREE_PRESET_SLEEP_BIT = 0x80;
 | 
			
		||||
 | 
			
		||||
// Model codes
 | 
			
		||||
enum Model { GREE_GENERIC, GREE_YAN, GREE_YAA, GREE_YAC, GREE_YAC1FB9 };
 | 
			
		||||
enum Model { GREE_GENERIC, GREE_YAN, GREE_YAA, GREE_YAC, GREE_YAC1FB9, GREE_YX1FF };
 | 
			
		||||
 | 
			
		||||
class GreeClimate : public climate_ir::ClimateIR {
 | 
			
		||||
 public:
 | 
			
		||||
@@ -93,6 +100,7 @@ class GreeClimate : public climate_ir::ClimateIR {
 | 
			
		||||
  uint8_t horizontal_swing_();
 | 
			
		||||
  uint8_t vertical_swing_();
 | 
			
		||||
  uint8_t temperature_();
 | 
			
		||||
  uint8_t preset_();
 | 
			
		||||
 | 
			
		||||
  Model model_{};
 | 
			
		||||
};
 | 
			
		||||
 
 | 
			
		||||
@@ -1,11 +1,10 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
 | 
			
		||||
from esphome import pins
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
from esphome.components import i2c, touchscreen
 | 
			
		||||
from esphome.const import CONF_INTERRUPT_PIN, CONF_ID
 | 
			
		||||
from .. import gt911_ns
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import CONF_ID, CONF_INTERRUPT_PIN, CONF_RESET_PIN
 | 
			
		||||
 | 
			
		||||
from .. import gt911_ns
 | 
			
		||||
 | 
			
		||||
GT911ButtonListener = gt911_ns.class_("GT911ButtonListener")
 | 
			
		||||
GT911Touchscreen = gt911_ns.class_(
 | 
			
		||||
@@ -18,6 +17,7 @@ CONFIG_SCHEMA = touchscreen.TOUCHSCREEN_SCHEMA.extend(
 | 
			
		||||
    {
 | 
			
		||||
        cv.GenerateID(): cv.declare_id(GT911Touchscreen),
 | 
			
		||||
        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))
 | 
			
		||||
 | 
			
		||||
@@ -29,3 +29,5 @@ async def to_code(config):
 | 
			
		||||
 | 
			
		||||
    if interrupt_pin := config.get(CONF_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() {
 | 
			
		||||
  i2c::ErrorCode err;
 | 
			
		||||
  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.
 | 
			
		||||
  uint8_t data[4];
 | 
			
		||||
 
 | 
			
		||||
@@ -19,12 +19,14 @@ class GT911Touchscreen : public touchscreen::Touchscreen, public i2c::I2CDevice
 | 
			
		||||
  void dump_config() override;
 | 
			
		||||
 | 
			
		||||
  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); }
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  void update_touches() override;
 | 
			
		||||
 | 
			
		||||
  InternalGPIOPin *interrupt_pin_{};
 | 
			
		||||
  GPIOPin *reset_pin_{};
 | 
			
		||||
  std::vector<GT911ButtonListener *> button_listeners_;
 | 
			
		||||
  uint8_t button_state_{0xFF};  // last button state. Initial FF guarantees first update.
 | 
			
		||||
};
 | 
			
		||||
 
 | 
			
		||||
@@ -5,6 +5,19 @@ from esphome.const import CONF_ATTRIBUTE, CONF_ENTITY_ID, CONF_INTERNAL
 | 
			
		||||
CODEOWNERS = ["@OttoWinter", "@esphome/core"]
 | 
			
		||||
homeassistant_ns = cg.esphome_ns.namespace("homeassistant")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def validate_entity_domain(platform, supported_domains):
 | 
			
		||||
    def validator(config):
 | 
			
		||||
        domain = config[CONF_ENTITY_ID].split(".", 1)[0]
 | 
			
		||||
        if domain not in supported_domains:
 | 
			
		||||
            raise cv.Invalid(
 | 
			
		||||
                f"Entity ID {config[CONF_ENTITY_ID]} is not supported by the {platform} platform."
 | 
			
		||||
            )
 | 
			
		||||
        return config
 | 
			
		||||
 | 
			
		||||
    return validator
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HOME_ASSISTANT_IMPORT_SCHEMA = cv.Schema(
 | 
			
		||||
    {
 | 
			
		||||
        cv.Required(CONF_ENTITY_ID): cv.entity_id,
 | 
			
		||||
 
 | 
			
		||||
@@ -7,19 +7,32 @@ from .. import (
 | 
			
		||||
    HOME_ASSISTANT_IMPORT_CONTROL_SCHEMA,
 | 
			
		||||
    homeassistant_ns,
 | 
			
		||||
    setup_home_assistant_entity,
 | 
			
		||||
    validate_entity_domain,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@Links2004"]
 | 
			
		||||
DEPENDENCIES = ["api"]
 | 
			
		||||
 | 
			
		||||
SUPPORTED_DOMAINS = [
 | 
			
		||||
    "automation",
 | 
			
		||||
    "fan",
 | 
			
		||||
    "humidifier",
 | 
			
		||||
    "input_boolean",
 | 
			
		||||
    "light",
 | 
			
		||||
    "remote",
 | 
			
		||||
    "siren",
 | 
			
		||||
    "switch",
 | 
			
		||||
]
 | 
			
		||||
 | 
			
		||||
HomeassistantSwitch = homeassistant_ns.class_(
 | 
			
		||||
    "HomeassistantSwitch", switch.Switch, cg.Component
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = (
 | 
			
		||||
CONFIG_SCHEMA = cv.All(
 | 
			
		||||
    switch.switch_schema(HomeassistantSwitch)
 | 
			
		||||
    .extend(cv.COMPONENT_SCHEMA)
 | 
			
		||||
    .extend(HOME_ASSISTANT_IMPORT_CONTROL_SCHEMA)
 | 
			
		||||
    .extend(cv.COMPONENT_SCHEMA),
 | 
			
		||||
    validate_entity_domain("switch", SUPPORTED_DOMAINS),
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -42,9 +42,9 @@ void HomeassistantSwitch::write_state(bool state) {
 | 
			
		||||
 | 
			
		||||
  api::HomeassistantServiceResponse resp;
 | 
			
		||||
  if (state) {
 | 
			
		||||
    resp.service = "switch.turn_on";
 | 
			
		||||
    resp.service = "homeassistant.turn_on";
 | 
			
		||||
  } else {
 | 
			
		||||
    resp.service = "switch.turn_off";
 | 
			
		||||
    resp.service = "homeassistant.turn_off";
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  api::HomeassistantServiceMap entity_id_kv;
 | 
			
		||||
 
 | 
			
		||||
@@ -1,16 +1,16 @@
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
import esphome.final_validate as fv
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
 | 
			
		||||
from esphome import pins
 | 
			
		||||
from esphome.const import CONF_ID
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
from esphome.components.esp32 import get_esp32_variant
 | 
			
		||||
from esphome.components.esp32.const import (
 | 
			
		||||
    VARIANT_ESP32,
 | 
			
		||||
    VARIANT_ESP32C3,
 | 
			
		||||
    VARIANT_ESP32S2,
 | 
			
		||||
    VARIANT_ESP32S3,
 | 
			
		||||
    VARIANT_ESP32C3,
 | 
			
		||||
)
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import CONF_CHANNEL, CONF_ID, CONF_SAMPLE_RATE
 | 
			
		||||
from esphome.cpp_generator import MockObjClass
 | 
			
		||||
import esphome.final_validate as fv
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@jesserockz"]
 | 
			
		||||
DEPENDENCIES = ["esp32"]
 | 
			
		||||
@@ -25,16 +25,26 @@ CONF_I2S_LRCLK_PIN = "i2s_lrclk_pin"
 | 
			
		||||
CONF_I2S_AUDIO = "i2s_audio"
 | 
			
		||||
CONF_I2S_AUDIO_ID = "i2s_audio_id"
 | 
			
		||||
 | 
			
		||||
CONF_BITS_PER_SAMPLE = "bits_per_sample"
 | 
			
		||||
CONF_I2S_MODE = "i2s_mode"
 | 
			
		||||
CONF_PRIMARY = "primary"
 | 
			
		||||
CONF_SECONDARY = "secondary"
 | 
			
		||||
 | 
			
		||||
CONF_USE_APLL = "use_apll"
 | 
			
		||||
CONF_BITS_PER_SAMPLE = "bits_per_sample"
 | 
			
		||||
CONF_BITS_PER_CHANNEL = "bits_per_channel"
 | 
			
		||||
CONF_MONO = "mono"
 | 
			
		||||
CONF_LEFT = "left"
 | 
			
		||||
CONF_RIGHT = "right"
 | 
			
		||||
CONF_STEREO = "stereo"
 | 
			
		||||
 | 
			
		||||
i2s_audio_ns = cg.esphome_ns.namespace("i2s_audio")
 | 
			
		||||
I2SAudioComponent = i2s_audio_ns.class_("I2SAudioComponent", cg.Component)
 | 
			
		||||
I2SAudioIn = i2s_audio_ns.class_("I2SAudioIn", cg.Parented.template(I2SAudioComponent))
 | 
			
		||||
I2SAudioOut = i2s_audio_ns.class_(
 | 
			
		||||
    "I2SAudioOut", cg.Parented.template(I2SAudioComponent)
 | 
			
		||||
I2SAudioBase = i2s_audio_ns.class_(
 | 
			
		||||
    "I2SAudioBase", cg.Parented.template(I2SAudioComponent)
 | 
			
		||||
)
 | 
			
		||||
I2SAudioIn = i2s_audio_ns.class_("I2SAudioIn", I2SAudioBase)
 | 
			
		||||
I2SAudioOut = i2s_audio_ns.class_("I2SAudioOut", I2SAudioBase)
 | 
			
		||||
 | 
			
		||||
i2s_mode_t = cg.global_ns.enum("i2s_mode_t")
 | 
			
		||||
I2S_MODE_OPTIONS = {
 | 
			
		||||
@@ -50,6 +60,75 @@ I2S_PORTS = {
 | 
			
		||||
    VARIANT_ESP32C3: 1,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
i2s_channel_fmt_t = cg.global_ns.enum("i2s_channel_fmt_t")
 | 
			
		||||
I2S_CHANNELS = {
 | 
			
		||||
    CONF_MONO: i2s_channel_fmt_t.I2S_CHANNEL_FMT_ALL_LEFT,
 | 
			
		||||
    CONF_LEFT: i2s_channel_fmt_t.I2S_CHANNEL_FMT_ONLY_LEFT,
 | 
			
		||||
    CONF_RIGHT: i2s_channel_fmt_t.I2S_CHANNEL_FMT_ONLY_RIGHT,
 | 
			
		||||
    CONF_STEREO: i2s_channel_fmt_t.I2S_CHANNEL_FMT_RIGHT_LEFT,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
i2s_bits_per_sample_t = cg.global_ns.enum("i2s_bits_per_sample_t")
 | 
			
		||||
I2S_BITS_PER_SAMPLE = {
 | 
			
		||||
    8: i2s_bits_per_sample_t.I2S_BITS_PER_SAMPLE_8BIT,
 | 
			
		||||
    16: i2s_bits_per_sample_t.I2S_BITS_PER_SAMPLE_16BIT,
 | 
			
		||||
    24: i2s_bits_per_sample_t.I2S_BITS_PER_SAMPLE_24BIT,
 | 
			
		||||
    32: i2s_bits_per_sample_t.I2S_BITS_PER_SAMPLE_32BIT,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
i2s_bits_per_chan_t = cg.global_ns.enum("i2s_bits_per_chan_t")
 | 
			
		||||
I2S_BITS_PER_CHANNEL = {
 | 
			
		||||
    "default": i2s_bits_per_chan_t.I2S_BITS_PER_CHAN_DEFAULT,
 | 
			
		||||
    8: i2s_bits_per_chan_t.I2S_BITS_PER_CHAN_8BIT,
 | 
			
		||||
    16: i2s_bits_per_chan_t.I2S_BITS_PER_CHAN_16BIT,
 | 
			
		||||
    24: i2s_bits_per_chan_t.I2S_BITS_PER_CHAN_24BIT,
 | 
			
		||||
    32: i2s_bits_per_chan_t.I2S_BITS_PER_CHAN_32BIT,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
_validate_bits = cv.float_with_unit("bits", "bit")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def i2s_audio_component_schema(
 | 
			
		||||
    class_: MockObjClass,
 | 
			
		||||
    *,
 | 
			
		||||
    default_sample_rate: int,
 | 
			
		||||
    default_channel: str,
 | 
			
		||||
    default_bits_per_sample: str,
 | 
			
		||||
):
 | 
			
		||||
    return cv.Schema(
 | 
			
		||||
        {
 | 
			
		||||
            cv.GenerateID(): cv.declare_id(class_),
 | 
			
		||||
            cv.GenerateID(CONF_I2S_AUDIO_ID): cv.use_id(I2SAudioComponent),
 | 
			
		||||
            cv.Optional(CONF_CHANNEL, default=default_channel): cv.enum(I2S_CHANNELS),
 | 
			
		||||
            cv.Optional(CONF_SAMPLE_RATE, default=default_sample_rate): cv.int_range(
 | 
			
		||||
                min=1
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_BITS_PER_SAMPLE, default=default_bits_per_sample): cv.All(
 | 
			
		||||
                _validate_bits, cv.enum(I2S_BITS_PER_SAMPLE)
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_I2S_MODE, default=CONF_PRIMARY): cv.enum(
 | 
			
		||||
                I2S_MODE_OPTIONS, lower=True
 | 
			
		||||
            ),
 | 
			
		||||
            cv.Optional(CONF_USE_APLL, default=False): cv.boolean,
 | 
			
		||||
            cv.Optional(CONF_BITS_PER_CHANNEL, default="default"): cv.All(
 | 
			
		||||
                cv.Any(cv.float_with_unit("bits", "bit"), "default"),
 | 
			
		||||
                cv.enum(I2S_BITS_PER_CHANNEL),
 | 
			
		||||
            ),
 | 
			
		||||
        }
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
async def register_i2s_audio_component(var, config):
 | 
			
		||||
    await cg.register_parented(var, config[CONF_I2S_AUDIO_ID])
 | 
			
		||||
 | 
			
		||||
    cg.add(var.set_i2s_mode(config[CONF_I2S_MODE]))
 | 
			
		||||
    cg.add(var.set_channel(config[CONF_CHANNEL]))
 | 
			
		||||
    cg.add(var.set_sample_rate(config[CONF_SAMPLE_RATE]))
 | 
			
		||||
    cg.add(var.set_bits_per_sample(config[CONF_BITS_PER_SAMPLE]))
 | 
			
		||||
    cg.add(var.set_bits_per_channel(config[CONF_BITS_PER_CHANNEL]))
 | 
			
		||||
    cg.add(var.set_use_apll(config[CONF_USE_APLL]))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = cv.Schema(
 | 
			
		||||
    {
 | 
			
		||||
        cv.GenerateID(): cv.declare_id(I2SAudioComponent),
 | 
			
		||||
 
 | 
			
		||||
@@ -11,9 +11,27 @@ namespace i2s_audio {
 | 
			
		||||
 | 
			
		||||
class I2SAudioComponent;
 | 
			
		||||
 | 
			
		||||
class I2SAudioIn : public Parented<I2SAudioComponent> {};
 | 
			
		||||
class I2SAudioBase : public Parented<I2SAudioComponent> {
 | 
			
		||||
 public:
 | 
			
		||||
  void set_i2s_mode(i2s_mode_t mode) { this->i2s_mode_ = mode; }
 | 
			
		||||
  void set_channel(i2s_channel_fmt_t channel) { this->channel_ = channel; }
 | 
			
		||||
  void set_sample_rate(uint32_t sample_rate) { this->sample_rate_ = sample_rate; }
 | 
			
		||||
  void set_bits_per_sample(i2s_bits_per_sample_t bits_per_sample) { this->bits_per_sample_ = bits_per_sample; }
 | 
			
		||||
  void set_bits_per_channel(i2s_bits_per_chan_t bits_per_channel) { this->bits_per_channel_ = bits_per_channel; }
 | 
			
		||||
  void set_use_apll(uint32_t use_apll) { this->use_apll_ = use_apll; }
 | 
			
		||||
 | 
			
		||||
class I2SAudioOut : public Parented<I2SAudioComponent> {};
 | 
			
		||||
 protected:
 | 
			
		||||
  i2s_mode_t i2s_mode_{};
 | 
			
		||||
  i2s_channel_fmt_t channel_;
 | 
			
		||||
  uint32_t sample_rate_;
 | 
			
		||||
  i2s_bits_per_sample_t bits_per_sample_;
 | 
			
		||||
  i2s_bits_per_chan_t bits_per_channel_;
 | 
			
		||||
  bool use_apll_;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class I2SAudioIn : public I2SAudioBase {};
 | 
			
		||||
 | 
			
		||||
class I2SAudioOut : public I2SAudioBase {};
 | 
			
		||||
 | 
			
		||||
class I2SAudioComponent : public Component {
 | 
			
		||||
 public:
 | 
			
		||||
 
 | 
			
		||||
@@ -12,6 +12,10 @@ from .. import (
 | 
			
		||||
    I2SAudioOut,
 | 
			
		||||
    CONF_I2S_AUDIO_ID,
 | 
			
		||||
    CONF_I2S_DOUT_PIN,
 | 
			
		||||
    CONF_LEFT,
 | 
			
		||||
    CONF_RIGHT,
 | 
			
		||||
    CONF_MONO,
 | 
			
		||||
    CONF_STEREO,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@jesserockz"]
 | 
			
		||||
@@ -30,12 +34,12 @@ CONF_DAC_TYPE = "dac_type"
 | 
			
		||||
CONF_I2S_COMM_FMT = "i2s_comm_fmt"
 | 
			
		||||
 | 
			
		||||
INTERNAL_DAC_OPTIONS = {
 | 
			
		||||
    "left": i2s_dac_mode_t.I2S_DAC_CHANNEL_LEFT_EN,
 | 
			
		||||
    "right": i2s_dac_mode_t.I2S_DAC_CHANNEL_RIGHT_EN,
 | 
			
		||||
    "stereo": i2s_dac_mode_t.I2S_DAC_CHANNEL_BOTH_EN,
 | 
			
		||||
    CONF_LEFT: i2s_dac_mode_t.I2S_DAC_CHANNEL_LEFT_EN,
 | 
			
		||||
    CONF_RIGHT: i2s_dac_mode_t.I2S_DAC_CHANNEL_RIGHT_EN,
 | 
			
		||||
    CONF_STEREO: i2s_dac_mode_t.I2S_DAC_CHANNEL_BOTH_EN,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
EXTERNAL_DAC_OPTIONS = ["mono", "stereo"]
 | 
			
		||||
EXTERNAL_DAC_OPTIONS = [CONF_MONO, CONF_STEREO]
 | 
			
		||||
 | 
			
		||||
NO_INTERNAL_DAC_VARIANTS = [esp32.const.VARIANT_ESP32S2]
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -23,7 +23,7 @@ enum I2SState : uint8_t {
 | 
			
		||||
  I2S_STATE_STOPPING,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class I2SAudioMediaPlayer : public Component, public media_player::MediaPlayer, public I2SAudioOut {
 | 
			
		||||
class I2SAudioMediaPlayer : public Component, public Parented<I2SAudioComponent>, public media_player::MediaPlayer {
 | 
			
		||||
 public:
 | 
			
		||||
  void setup() override;
 | 
			
		||||
  float get_setup_priority() const override { return esphome::setup_priority::LATE; }
 | 
			
		||||
 
 | 
			
		||||
@@ -1,20 +1,17 @@
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
 | 
			
		||||
from esphome import pins
 | 
			
		||||
from esphome.const import CONF_CHANNEL, CONF_ID, CONF_NUMBER, CONF_SAMPLE_RATE
 | 
			
		||||
from esphome.components import microphone, esp32
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
from esphome.components import esp32, microphone
 | 
			
		||||
from esphome.components.adc import ESP32_VARIANT_ADC1_PIN_TO_CHANNEL, validate_adc_pin
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import CONF_ID, CONF_NUMBER
 | 
			
		||||
 | 
			
		||||
from .. import (
 | 
			
		||||
    CONF_I2S_MODE,
 | 
			
		||||
    CONF_PRIMARY,
 | 
			
		||||
    I2S_MODE_OPTIONS,
 | 
			
		||||
    i2s_audio_ns,
 | 
			
		||||
    I2SAudioComponent,
 | 
			
		||||
    I2SAudioIn,
 | 
			
		||||
    CONF_I2S_AUDIO_ID,
 | 
			
		||||
    CONF_I2S_DIN_PIN,
 | 
			
		||||
    CONF_RIGHT,
 | 
			
		||||
    I2SAudioIn,
 | 
			
		||||
    i2s_audio_component_schema,
 | 
			
		||||
    i2s_audio_ns,
 | 
			
		||||
    register_i2s_audio_component,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@jesserockz"]
 | 
			
		||||
@@ -23,29 +20,14 @@ DEPENDENCIES = ["i2s_audio"]
 | 
			
		||||
CONF_ADC_PIN = "adc_pin"
 | 
			
		||||
CONF_ADC_TYPE = "adc_type"
 | 
			
		||||
CONF_PDM = "pdm"
 | 
			
		||||
CONF_BITS_PER_SAMPLE = "bits_per_sample"
 | 
			
		||||
CONF_USE_APLL = "use_apll"
 | 
			
		||||
 | 
			
		||||
I2SAudioMicrophone = i2s_audio_ns.class_(
 | 
			
		||||
    "I2SAudioMicrophone", I2SAudioIn, microphone.Microphone, cg.Component
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
i2s_channel_fmt_t = cg.global_ns.enum("i2s_channel_fmt_t")
 | 
			
		||||
CHANNELS = {
 | 
			
		||||
    "left": i2s_channel_fmt_t.I2S_CHANNEL_FMT_ONLY_LEFT,
 | 
			
		||||
    "right": i2s_channel_fmt_t.I2S_CHANNEL_FMT_ONLY_RIGHT,
 | 
			
		||||
}
 | 
			
		||||
i2s_bits_per_sample_t = cg.global_ns.enum("i2s_bits_per_sample_t")
 | 
			
		||||
BITS_PER_SAMPLE = {
 | 
			
		||||
    16: i2s_bits_per_sample_t.I2S_BITS_PER_SAMPLE_16BIT,
 | 
			
		||||
    32: i2s_bits_per_sample_t.I2S_BITS_PER_SAMPLE_32BIT,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
INTERNAL_ADC_VARIANTS = [esp32.const.VARIANT_ESP32]
 | 
			
		||||
PDM_VARIANTS = [esp32.const.VARIANT_ESP32, esp32.const.VARIANT_ESP32S3]
 | 
			
		||||
 | 
			
		||||
_validate_bits = cv.float_with_unit("bits", "bit")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def validate_esp32_variant(config):
 | 
			
		||||
    variant = esp32.get_esp32_variant()
 | 
			
		||||
@@ -62,21 +44,15 @@ def validate_esp32_variant(config):
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
BASE_SCHEMA = microphone.MICROPHONE_SCHEMA.extend(
 | 
			
		||||
    {
 | 
			
		||||
        cv.GenerateID(): cv.declare_id(I2SAudioMicrophone),
 | 
			
		||||
        cv.GenerateID(CONF_I2S_AUDIO_ID): cv.use_id(I2SAudioComponent),
 | 
			
		||||
        cv.Optional(CONF_CHANNEL, default="right"): cv.enum(CHANNELS),
 | 
			
		||||
        cv.Optional(CONF_SAMPLE_RATE, default=16000): cv.int_range(min=1),
 | 
			
		||||
        cv.Optional(CONF_BITS_PER_SAMPLE, default="32bit"): cv.All(
 | 
			
		||||
            _validate_bits, cv.enum(BITS_PER_SAMPLE)
 | 
			
		||||
        ),
 | 
			
		||||
        cv.Optional(CONF_USE_APLL, default=False): cv.boolean,
 | 
			
		||||
        cv.Optional(CONF_I2S_MODE, default=CONF_PRIMARY): cv.enum(
 | 
			
		||||
            I2S_MODE_OPTIONS, lower=True
 | 
			
		||||
        ),
 | 
			
		||||
    }
 | 
			
		||||
    i2s_audio_component_schema(
 | 
			
		||||
        I2SAudioMicrophone,
 | 
			
		||||
        default_sample_rate=16000,
 | 
			
		||||
        default_channel=CONF_RIGHT,
 | 
			
		||||
        default_bits_per_sample="32bit",
 | 
			
		||||
    )
 | 
			
		||||
).extend(cv.COMPONENT_SCHEMA)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = cv.All(
 | 
			
		||||
    cv.typed_schema(
 | 
			
		||||
        {
 | 
			
		||||
@@ -88,7 +64,7 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
            "external": BASE_SCHEMA.extend(
 | 
			
		||||
                {
 | 
			
		||||
                    cv.Required(CONF_I2S_DIN_PIN): pins.internal_gpio_input_pin_number,
 | 
			
		||||
                    cv.Required(CONF_PDM): cv.boolean,
 | 
			
		||||
                    cv.Optional(CONF_PDM, default=False): cv.boolean,
 | 
			
		||||
                }
 | 
			
		||||
            ),
 | 
			
		||||
        },
 | 
			
		||||
@@ -101,8 +77,8 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    var = cg.new_Pvariable(config[CONF_ID])
 | 
			
		||||
    await cg.register_component(var, config)
 | 
			
		||||
 | 
			
		||||
    await cg.register_parented(var, config[CONF_I2S_AUDIO_ID])
 | 
			
		||||
    await register_i2s_audio_component(var, config)
 | 
			
		||||
    await microphone.register_microphone(var, config)
 | 
			
		||||
 | 
			
		||||
    if config[CONF_ADC_TYPE] == "internal":
 | 
			
		||||
        variant = esp32.get_esp32_variant()
 | 
			
		||||
@@ -112,11 +88,3 @@ async def to_code(config):
 | 
			
		||||
    else:
 | 
			
		||||
        cg.add(var.set_din_pin(config[CONF_I2S_DIN_PIN]))
 | 
			
		||||
        cg.add(var.set_pdm(config[CONF_PDM]))
 | 
			
		||||
 | 
			
		||||
    cg.add(var.set_i2s_mode(config[CONF_I2S_MODE]))
 | 
			
		||||
    cg.add(var.set_channel(config[CONF_CHANNEL]))
 | 
			
		||||
    cg.add(var.set_sample_rate(config[CONF_SAMPLE_RATE]))
 | 
			
		||||
    cg.add(var.set_bits_per_sample(config[CONF_BITS_PER_SAMPLE]))
 | 
			
		||||
    cg.add(var.set_use_apll(config[CONF_USE_APLL]))
 | 
			
		||||
 | 
			
		||||
    await microphone.register_microphone(var, config)
 | 
			
		||||
 
 | 
			
		||||
@@ -58,7 +58,7 @@ void I2SAudioMicrophone::start_() {
 | 
			
		||||
      .tx_desc_auto_clear = false,
 | 
			
		||||
      .fixed_mclk = 0,
 | 
			
		||||
      .mclk_multiple = I2S_MCLK_MULTIPLE_256,
 | 
			
		||||
      .bits_per_chan = I2S_BITS_PER_CHAN_DEFAULT,
 | 
			
		||||
      .bits_per_chan = this->bits_per_channel_,
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  esp_err_t err;
 | 
			
		||||
@@ -167,21 +167,24 @@ size_t I2SAudioMicrophone::read(int16_t *buf, size_t len) {
 | 
			
		||||
    return 0;
 | 
			
		||||
  }
 | 
			
		||||
  this->status_clear_warning();
 | 
			
		||||
  if (this->bits_per_sample_ == I2S_BITS_PER_SAMPLE_16BIT) {
 | 
			
		||||
    return bytes_read;
 | 
			
		||||
  } else if (this->bits_per_sample_ == I2S_BITS_PER_SAMPLE_32BIT) {
 | 
			
		||||
    std::vector<int16_t> samples;
 | 
			
		||||
    size_t samples_read = bytes_read / sizeof(int32_t);
 | 
			
		||||
    samples.resize(samples_read);
 | 
			
		||||
    for (size_t i = 0; i < samples_read; i++) {
 | 
			
		||||
      int32_t temp = reinterpret_cast<int32_t *>(buf)[i] >> 14;
 | 
			
		||||
      samples[i] = clamp<int16_t>(temp, INT16_MIN, INT16_MAX);
 | 
			
		||||
  // ESP-IDF I2S implementation right-extends 8-bit data to 16 bits,
 | 
			
		||||
  // and 24-bit data to 32 bits.
 | 
			
		||||
  switch (this->bits_per_sample_) {
 | 
			
		||||
    case I2S_BITS_PER_SAMPLE_8BIT:
 | 
			
		||||
    case I2S_BITS_PER_SAMPLE_16BIT:
 | 
			
		||||
      return bytes_read;
 | 
			
		||||
    case I2S_BITS_PER_SAMPLE_24BIT:
 | 
			
		||||
    case I2S_BITS_PER_SAMPLE_32BIT: {
 | 
			
		||||
      size_t samples_read = bytes_read / sizeof(int32_t);
 | 
			
		||||
      for (size_t i = 0; i < samples_read; i++) {
 | 
			
		||||
        int32_t temp = reinterpret_cast<int32_t *>(buf)[i] >> 14;
 | 
			
		||||
        buf[i] = clamp<int16_t>(temp, INT16_MIN, INT16_MAX);
 | 
			
		||||
      }
 | 
			
		||||
      return samples_read * sizeof(int16_t);
 | 
			
		||||
    }
 | 
			
		||||
    memcpy(buf, samples.data(), samples_read * sizeof(int16_t));
 | 
			
		||||
    return samples_read * sizeof(int16_t);
 | 
			
		||||
  } else {
 | 
			
		||||
    ESP_LOGE(TAG, "Unsupported bits per sample: %d", this->bits_per_sample_);
 | 
			
		||||
    return 0;
 | 
			
		||||
    default:
 | 
			
		||||
      ESP_LOGE(TAG, "Unsupported bits per sample: %d", this->bits_per_sample_);
 | 
			
		||||
      return 0;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -30,13 +30,6 @@ class I2SAudioMicrophone : public I2SAudioIn, public microphone::Microphone, pub
 | 
			
		||||
  }
 | 
			
		||||
#endif
 | 
			
		||||
 | 
			
		||||
  void set_i2s_mode(i2s_mode_t mode) { this->i2s_mode_ = mode; }
 | 
			
		||||
 | 
			
		||||
  void set_channel(i2s_channel_fmt_t channel) { this->channel_ = channel; }
 | 
			
		||||
  void set_sample_rate(uint32_t sample_rate) { this->sample_rate_ = sample_rate; }
 | 
			
		||||
  void set_bits_per_sample(i2s_bits_per_sample_t bits_per_sample) { this->bits_per_sample_ = bits_per_sample; }
 | 
			
		||||
  void set_use_apll(uint32_t use_apll) { this->use_apll_ = use_apll; }
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  void start_();
 | 
			
		||||
  void stop_();
 | 
			
		||||
@@ -48,11 +41,6 @@ class I2SAudioMicrophone : public I2SAudioIn, public microphone::Microphone, pub
 | 
			
		||||
  bool adc_{false};
 | 
			
		||||
#endif
 | 
			
		||||
  bool pdm_{false};
 | 
			
		||||
  i2s_mode_t i2s_mode_{};
 | 
			
		||||
  i2s_channel_fmt_t channel_;
 | 
			
		||||
  uint32_t sample_rate_;
 | 
			
		||||
  i2s_bits_per_sample_t bits_per_sample_;
 | 
			
		||||
  bool use_apll_;
 | 
			
		||||
 | 
			
		||||
  HighFrequencyLoopRequester high_freq_;
 | 
			
		||||
};
 | 
			
		||||
 
 | 
			
		||||
@@ -1,15 +1,19 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome import pins
 | 
			
		||||
from esphome.const import CONF_ID, CONF_MODE
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
from esphome.components import esp32, speaker
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import CONF_CHANNEL, CONF_ID, CONF_MODE, CONF_TIMEOUT
 | 
			
		||||
 | 
			
		||||
from .. import (
 | 
			
		||||
    CONF_I2S_AUDIO_ID,
 | 
			
		||||
    CONF_I2S_DOUT_PIN,
 | 
			
		||||
    I2SAudioComponent,
 | 
			
		||||
    CONF_LEFT,
 | 
			
		||||
    CONF_MONO,
 | 
			
		||||
    CONF_RIGHT,
 | 
			
		||||
    CONF_STEREO,
 | 
			
		||||
    I2SAudioOut,
 | 
			
		||||
    i2s_audio_component_schema,
 | 
			
		||||
    i2s_audio_ns,
 | 
			
		||||
    register_i2s_audio_component,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CODEOWNERS = ["@jesserockz"]
 | 
			
		||||
@@ -19,19 +23,16 @@ I2SAudioSpeaker = i2s_audio_ns.class_(
 | 
			
		||||
    "I2SAudioSpeaker", cg.Component, speaker.Speaker, I2SAudioOut
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
i2s_dac_mode_t = cg.global_ns.enum("i2s_dac_mode_t")
 | 
			
		||||
 | 
			
		||||
CONF_MUTE_PIN = "mute_pin"
 | 
			
		||||
CONF_DAC_TYPE = "dac_type"
 | 
			
		||||
 | 
			
		||||
i2s_dac_mode_t = cg.global_ns.enum("i2s_dac_mode_t")
 | 
			
		||||
INTERNAL_DAC_OPTIONS = {
 | 
			
		||||
    "left": i2s_dac_mode_t.I2S_DAC_CHANNEL_LEFT_EN,
 | 
			
		||||
    "right": i2s_dac_mode_t.I2S_DAC_CHANNEL_RIGHT_EN,
 | 
			
		||||
    "stereo": i2s_dac_mode_t.I2S_DAC_CHANNEL_BOTH_EN,
 | 
			
		||||
    CONF_LEFT: i2s_dac_mode_t.I2S_DAC_CHANNEL_LEFT_EN,
 | 
			
		||||
    CONF_RIGHT: i2s_dac_mode_t.I2S_DAC_CHANNEL_RIGHT_EN,
 | 
			
		||||
    CONF_STEREO: i2s_dac_mode_t.I2S_DAC_CHANNEL_BOTH_EN,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
EXTERNAL_DAC_OPTIONS = ["mono", "stereo"]
 | 
			
		||||
 | 
			
		||||
NO_INTERNAL_DAC_VARIANTS = [esp32.const.VARIANT_ESP32S2]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@@ -44,28 +45,40 @@ def validate_esp32_variant(config):
 | 
			
		||||
    return config
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
BASE_SCHEMA = (
 | 
			
		||||
    speaker.SPEAKER_SCHEMA.extend(
 | 
			
		||||
        i2s_audio_component_schema(
 | 
			
		||||
            I2SAudioSpeaker,
 | 
			
		||||
            default_sample_rate=16000,
 | 
			
		||||
            default_channel=CONF_MONO,
 | 
			
		||||
            default_bits_per_sample="16bit",
 | 
			
		||||
        )
 | 
			
		||||
    )
 | 
			
		||||
    .extend(
 | 
			
		||||
        {
 | 
			
		||||
            cv.Optional(
 | 
			
		||||
                CONF_TIMEOUT, default="100ms"
 | 
			
		||||
            ): cv.positive_time_period_milliseconds,
 | 
			
		||||
        }
 | 
			
		||||
    )
 | 
			
		||||
    .extend(cv.COMPONENT_SCHEMA)
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
CONFIG_SCHEMA = cv.All(
 | 
			
		||||
    cv.typed_schema(
 | 
			
		||||
        {
 | 
			
		||||
            "internal": speaker.SPEAKER_SCHEMA.extend(
 | 
			
		||||
            "internal": BASE_SCHEMA.extend(
 | 
			
		||||
                {
 | 
			
		||||
                    cv.GenerateID(): cv.declare_id(I2SAudioSpeaker),
 | 
			
		||||
                    cv.GenerateID(CONF_I2S_AUDIO_ID): cv.use_id(I2SAudioComponent),
 | 
			
		||||
                    cv.Required(CONF_MODE): cv.enum(INTERNAL_DAC_OPTIONS, lower=True),
 | 
			
		||||
                }
 | 
			
		||||
            ).extend(cv.COMPONENT_SCHEMA),
 | 
			
		||||
            "external": speaker.SPEAKER_SCHEMA.extend(
 | 
			
		||||
            ),
 | 
			
		||||
            "external": BASE_SCHEMA.extend(
 | 
			
		||||
                {
 | 
			
		||||
                    cv.GenerateID(): cv.declare_id(I2SAudioSpeaker),
 | 
			
		||||
                    cv.GenerateID(CONF_I2S_AUDIO_ID): cv.use_id(I2SAudioComponent),
 | 
			
		||||
                    cv.Required(
 | 
			
		||||
                        CONF_I2S_DOUT_PIN
 | 
			
		||||
                    ): pins.internal_gpio_output_pin_number,
 | 
			
		||||
                    cv.Optional(CONF_MODE, default="mono"): cv.one_of(
 | 
			
		||||
                        *EXTERNAL_DAC_OPTIONS, lower=True
 | 
			
		||||
                    ),
 | 
			
		||||
                }
 | 
			
		||||
            ).extend(cv.COMPONENT_SCHEMA),
 | 
			
		||||
            ),
 | 
			
		||||
        },
 | 
			
		||||
        key=CONF_DAC_TYPE,
 | 
			
		||||
    ),
 | 
			
		||||
@@ -76,12 +89,11 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
async def to_code(config):
 | 
			
		||||
    var = cg.new_Pvariable(config[CONF_ID])
 | 
			
		||||
    await cg.register_component(var, config)
 | 
			
		||||
    await register_i2s_audio_component(var, config)
 | 
			
		||||
    await speaker.register_speaker(var, config)
 | 
			
		||||
 | 
			
		||||
    await cg.register_parented(var, config[CONF_I2S_AUDIO_ID])
 | 
			
		||||
 | 
			
		||||
    if config[CONF_DAC_TYPE] == "internal":
 | 
			
		||||
        cg.add(var.set_internal_dac_mode(config[CONF_MODE]))
 | 
			
		||||
        cg.add(var.set_internal_dac_mode(config[CONF_CHANNEL]))
 | 
			
		||||
    else:
 | 
			
		||||
        cg.add(var.set_dout_pin(config[CONF_I2S_DOUT_PIN]))
 | 
			
		||||
        cg.add(var.set_external_dac_channels(2 if config[CONF_MODE] == "stereo" else 1))
 | 
			
		||||
    cg.add(var.set_timeout(config[CONF_TIMEOUT]))
 | 
			
		||||
 
 | 
			
		||||
@@ -56,6 +56,21 @@ void I2SAudioSpeaker::start_() {
 | 
			
		||||
  this->task_created_ = true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<typename a, typename b> const uint8_t *convert_data_format(const a *from, b *to, size_t &bytes, bool repeat) {
 | 
			
		||||
  if (sizeof(a) == sizeof(b) && !repeat) {
 | 
			
		||||
    return reinterpret_cast<const uint8_t *>(from);
 | 
			
		||||
  }
 | 
			
		||||
  const b *result = to;
 | 
			
		||||
  for (size_t i = 0; i < bytes; i += sizeof(a)) {
 | 
			
		||||
    b value = static_cast<b>(*from++) << (sizeof(b) - sizeof(a)) * 8;
 | 
			
		||||
    *to++ = value;
 | 
			
		||||
    if (repeat)
 | 
			
		||||
      *to++ = value;
 | 
			
		||||
  }
 | 
			
		||||
  bytes *= (sizeof(b) / sizeof(a)) * (repeat ? 2 : 1);  // NOLINT
 | 
			
		||||
  return reinterpret_cast<const uint8_t *>(result);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void I2SAudioSpeaker::player_task(void *params) {
 | 
			
		||||
  I2SAudioSpeaker *this_speaker = (I2SAudioSpeaker *) params;
 | 
			
		||||
 | 
			
		||||
@@ -64,19 +79,19 @@ void I2SAudioSpeaker::player_task(void *params) {
 | 
			
		||||
  xQueueSend(this_speaker->event_queue_, &event, portMAX_DELAY);
 | 
			
		||||
 | 
			
		||||
  i2s_driver_config_t config = {
 | 
			
		||||
      .mode = (i2s_mode_t) (I2S_MODE_MASTER | I2S_MODE_TX),
 | 
			
		||||
      .sample_rate = 16000,
 | 
			
		||||
      .bits_per_sample = I2S_BITS_PER_SAMPLE_16BIT,
 | 
			
		||||
      .channel_format = I2S_CHANNEL_FMT_RIGHT_LEFT,
 | 
			
		||||
      .mode = (i2s_mode_t) (this_speaker->i2s_mode_ | I2S_MODE_TX),
 | 
			
		||||
      .sample_rate = this_speaker->sample_rate_,
 | 
			
		||||
      .bits_per_sample = this_speaker->bits_per_sample_,
 | 
			
		||||
      .channel_format = this_speaker->channel_,
 | 
			
		||||
      .communication_format = I2S_COMM_FORMAT_STAND_I2S,
 | 
			
		||||
      .intr_alloc_flags = ESP_INTR_FLAG_LEVEL1,
 | 
			
		||||
      .dma_buf_count = 8,
 | 
			
		||||
      .dma_buf_len = 128,
 | 
			
		||||
      .use_apll = false,
 | 
			
		||||
      .dma_buf_len = 256,
 | 
			
		||||
      .use_apll = this_speaker->use_apll_,
 | 
			
		||||
      .tx_desc_auto_clear = true,
 | 
			
		||||
      .fixed_mclk = I2S_PIN_NO_CHANGE,
 | 
			
		||||
      .fixed_mclk = 0,
 | 
			
		||||
      .mclk_multiple = I2S_MCLK_MULTIPLE_256,
 | 
			
		||||
      .bits_per_chan = I2S_BITS_PER_CHAN_DEFAULT,
 | 
			
		||||
      .bits_per_chan = this_speaker->bits_per_channel_,
 | 
			
		||||
  };
 | 
			
		||||
#if SOC_I2S_SUPPORTS_DAC
 | 
			
		||||
  if (this_speaker->internal_dac_mode_ != I2S_DAC_CHANNEL_DISABLE) {
 | 
			
		||||
@@ -114,10 +129,11 @@ void I2SAudioSpeaker::player_task(void *params) {
 | 
			
		||||
  event.type = TaskEventType::STARTED;
 | 
			
		||||
  xQueueSend(this_speaker->event_queue_, &event, portMAX_DELAY);
 | 
			
		||||
 | 
			
		||||
  int16_t buffer[BUFFER_SIZE / 2];
 | 
			
		||||
  int32_t buffer[BUFFER_SIZE];
 | 
			
		||||
 | 
			
		||||
  while (true) {
 | 
			
		||||
    if (xQueueReceive(this_speaker->buffer_queue_, &data_event, 100 / portTICK_PERIOD_MS) != pdTRUE) {
 | 
			
		||||
    if (xQueueReceive(this_speaker->buffer_queue_, &data_event, this_speaker->timeout_ / portTICK_PERIOD_MS) !=
 | 
			
		||||
        pdTRUE) {
 | 
			
		||||
      break;  // End of audio from main thread
 | 
			
		||||
    }
 | 
			
		||||
    if (data_event.stop) {
 | 
			
		||||
@@ -125,17 +141,28 @@ void I2SAudioSpeaker::player_task(void *params) {
 | 
			
		||||
      xQueueReset(this_speaker->buffer_queue_);  // Flush queue
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
    size_t bytes_written;
 | 
			
		||||
 | 
			
		||||
    memmove(buffer, data_event.data, data_event.len);
 | 
			
		||||
    size_t remaining = data_event.len / 2;
 | 
			
		||||
    size_t current = 0;
 | 
			
		||||
    const uint8_t *data = data_event.data;
 | 
			
		||||
    size_t remaining = data_event.len;
 | 
			
		||||
    switch (this_speaker->bits_per_sample_) {
 | 
			
		||||
      case I2S_BITS_PER_SAMPLE_8BIT:
 | 
			
		||||
      case I2S_BITS_PER_SAMPLE_16BIT: {
 | 
			
		||||
        data = convert_data_format(reinterpret_cast<const int16_t *>(data), reinterpret_cast<int16_t *>(buffer),
 | 
			
		||||
                                   remaining, this_speaker->channel_ == I2S_CHANNEL_FMT_ALL_LEFT);
 | 
			
		||||
        break;
 | 
			
		||||
      }
 | 
			
		||||
      case I2S_BITS_PER_SAMPLE_24BIT:
 | 
			
		||||
      case I2S_BITS_PER_SAMPLE_32BIT: {
 | 
			
		||||
        data = convert_data_format(reinterpret_cast<const int16_t *>(data), reinterpret_cast<int32_t *>(buffer),
 | 
			
		||||
                                   remaining, this_speaker->channel_ == I2S_CHANNEL_FMT_ALL_LEFT);
 | 
			
		||||
        break;
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    while (remaining > 0) {
 | 
			
		||||
      uint32_t sample = (buffer[current] << 16) | (buffer[current] & 0xFFFF);
 | 
			
		||||
 | 
			
		||||
      esp_err_t err = i2s_write(this_speaker->parent_->get_port(), &sample, sizeof(sample), &bytes_written,
 | 
			
		||||
                                (10 / portTICK_PERIOD_MS));
 | 
			
		||||
    while (remaining != 0) {
 | 
			
		||||
      size_t bytes_written;
 | 
			
		||||
      esp_err_t err =
 | 
			
		||||
          i2s_write(this_speaker->parent_->get_port(), data, remaining, &bytes_written, (32 / portTICK_PERIOD_MS));
 | 
			
		||||
      if (err != ESP_OK) {
 | 
			
		||||
        event = {.type = TaskEventType::WARNING, .err = err};
 | 
			
		||||
        if (xQueueSend(this_speaker->event_queue_, &event, 10 / portTICK_PERIOD_MS) != pdTRUE) {
 | 
			
		||||
@@ -143,21 +170,8 @@ void I2SAudioSpeaker::player_task(void *params) {
 | 
			
		||||
        }
 | 
			
		||||
        continue;
 | 
			
		||||
      }
 | 
			
		||||
      if (bytes_written != sizeof(sample)) {
 | 
			
		||||
        event = {.type = TaskEventType::WARNING, .err = ESP_FAIL};
 | 
			
		||||
        if (xQueueSend(this_speaker->event_queue_, &event, 10 / portTICK_PERIOD_MS) != pdTRUE) {
 | 
			
		||||
          ESP_LOGW(TAG, "Failed to send WARNING event");
 | 
			
		||||
        }
 | 
			
		||||
        continue;
 | 
			
		||||
      }
 | 
			
		||||
      remaining--;
 | 
			
		||||
      current++;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    event.type = TaskEventType::PLAYING;
 | 
			
		||||
    event.err = current;
 | 
			
		||||
    if (xQueueSend(this_speaker->event_queue_, &event, 10 / portTICK_PERIOD_MS) != pdTRUE) {
 | 
			
		||||
      ESP_LOGW(TAG, "Failed to send PLAYING event");
 | 
			
		||||
      data += bytes_written;
 | 
			
		||||
      remaining -= bytes_written;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
@@ -213,13 +227,11 @@ void I2SAudioSpeaker::watch_() {
 | 
			
		||||
      case TaskEventType::STARTED:
 | 
			
		||||
        ESP_LOGD(TAG, "Started I2S Audio Speaker");
 | 
			
		||||
        this->state_ = speaker::STATE_RUNNING;
 | 
			
		||||
        this->status_clear_warning();
 | 
			
		||||
        break;
 | 
			
		||||
      case TaskEventType::STOPPING:
 | 
			
		||||
        ESP_LOGD(TAG, "Stopping I2S Audio Speaker");
 | 
			
		||||
        break;
 | 
			
		||||
      case TaskEventType::PLAYING:
 | 
			
		||||
        this->status_clear_warning();
 | 
			
		||||
        break;
 | 
			
		||||
      case TaskEventType::STOPPED:
 | 
			
		||||
        this->state_ = speaker::STATE_STOPPED;
 | 
			
		||||
        vTaskDelete(this->player_task_handle_);
 | 
			
		||||
 
 | 
			
		||||
@@ -21,7 +21,6 @@ static const size_t BUFFER_SIZE = 1024;
 | 
			
		||||
enum class TaskEventType : uint8_t {
 | 
			
		||||
  STARTING = 0,
 | 
			
		||||
  STARTED,
 | 
			
		||||
  PLAYING,
 | 
			
		||||
  STOPPING,
 | 
			
		||||
  STOPPED,
 | 
			
		||||
  WARNING = 255,
 | 
			
		||||
@@ -38,18 +37,18 @@ struct DataEvent {
 | 
			
		||||
  uint8_t data[BUFFER_SIZE];
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class I2SAudioSpeaker : public Component, public speaker::Speaker, public I2SAudioOut {
 | 
			
		||||
class I2SAudioSpeaker : public I2SAudioOut, public speaker::Speaker, public Component {
 | 
			
		||||
 public:
 | 
			
		||||
  float get_setup_priority() const override { return esphome::setup_priority::LATE; }
 | 
			
		||||
 | 
			
		||||
  void setup() override;
 | 
			
		||||
  void loop() override;
 | 
			
		||||
 | 
			
		||||
  void set_timeout(uint32_t ms) { this->timeout_ = ms; }
 | 
			
		||||
  void set_dout_pin(uint8_t pin) { this->dout_pin_ = pin; }
 | 
			
		||||
#if SOC_I2S_SUPPORTS_DAC
 | 
			
		||||
  void set_internal_dac_mode(i2s_dac_mode_t mode) { this->internal_dac_mode_ = mode; }
 | 
			
		||||
#endif
 | 
			
		||||
  void set_external_dac_channels(uint8_t channels) { this->external_dac_channels_ = channels; }
 | 
			
		||||
 | 
			
		||||
  void start() override;
 | 
			
		||||
  void stop() override;
 | 
			
		||||
@@ -70,13 +69,13 @@ class I2SAudioSpeaker : public Component, public speaker::Speaker, public I2SAud
 | 
			
		||||
  QueueHandle_t buffer_queue_;
 | 
			
		||||
  QueueHandle_t event_queue_;
 | 
			
		||||
 | 
			
		||||
  uint32_t timeout_{0};
 | 
			
		||||
  uint8_t dout_pin_{0};
 | 
			
		||||
  bool task_created_{false};
 | 
			
		||||
 | 
			
		||||
#if SOC_I2S_SUPPORTS_DAC
 | 
			
		||||
  i2s_dac_mode_t internal_dac_mode_{I2S_DAC_CHANNEL_DISABLE};
 | 
			
		||||
#endif
 | 
			
		||||
  uint8_t external_dac_channels_;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace i2s_audio
 | 
			
		||||
 
 | 
			
		||||
@@ -1,31 +1,31 @@
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
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.core import CORE, HexInt
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
from esphome.const import (
 | 
			
		||||
    CONF_COLOR_ORDER,
 | 
			
		||||
    CONF_COLOR_PALETTE,
 | 
			
		||||
    CONF_DC_PIN,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_LAMBDA,
 | 
			
		||||
    CONF_MODEL,
 | 
			
		||||
    CONF_RAW_DATA_ID,
 | 
			
		||||
    CONF_PAGES,
 | 
			
		||||
    CONF_RESET_PIN,
 | 
			
		||||
    CONF_DIMENSIONS,
 | 
			
		||||
    CONF_WIDTH,
 | 
			
		||||
    CONF_HEIGHT,
 | 
			
		||||
    CONF_ROTATION,
 | 
			
		||||
    CONF_ID,
 | 
			
		||||
    CONF_INVERT_COLORS,
 | 
			
		||||
    CONF_LAMBDA,
 | 
			
		||||
    CONF_MIRROR_X,
 | 
			
		||||
    CONF_MIRROR_Y,
 | 
			
		||||
    CONF_SWAP_XY,
 | 
			
		||||
    CONF_COLOR_ORDER,
 | 
			
		||||
    CONF_MODEL,
 | 
			
		||||
    CONF_OFFSET_HEIGHT,
 | 
			
		||||
    CONF_OFFSET_WIDTH,
 | 
			
		||||
    CONF_PAGES,
 | 
			
		||||
    CONF_RAW_DATA_ID,
 | 
			
		||||
    CONF_RESET_PIN,
 | 
			
		||||
    CONF_ROTATION,
 | 
			
		||||
    CONF_SWAP_XY,
 | 
			
		||||
    CONF_TRANSFORM,
 | 
			
		||||
    CONF_INVERT_COLORS,
 | 
			
		||||
    CONF_WIDTH,
 | 
			
		||||
)
 | 
			
		||||
from esphome.core import CORE, HexInt
 | 
			
		||||
 | 
			
		||||
DEPENDENCIES = ["spi"]
 | 
			
		||||
 | 
			
		||||
@@ -177,7 +177,7 @@ CONFIG_SCHEMA = cv.All(
 | 
			
		||||
            cv.Optional(CONF_INVERT_DISPLAY): cv.invalid(
 | 
			
		||||
                "'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.Exclusive(CONF_ROTATION, CONF_ROTATION): validate_rotation,
 | 
			
		||||
            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)
 | 
			
		||||
        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, "  Mirror_x: %s", YESNO(this->mirror_x_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Mirror_y: %s", YESNO(this->mirror_y_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Invert colors: %s", YESNO(this->pre_invertcolors_));
 | 
			
		||||
 | 
			
		||||
  if (this->is_failed()) {
 | 
			
		||||
    ESP_LOGCONFIG(TAG, "  => Failed to init Memory: YES!");
 | 
			
		||||
@@ -154,7 +155,6 @@ void ILI9XXXDisplay::fill(Color color) {
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
      return;
 | 
			
		||||
      break;
 | 
			
		||||
    default:
 | 
			
		||||
      new_color = display::ColorUtil::color_to_332(color, display::ColorOrder::COLOR_ORDER_RGB);
 | 
			
		||||
      break;
 | 
			
		||||
 
 | 
			
		||||
@@ -28,8 +28,8 @@ class ILI9XXXDisplay : public display::DisplayBuffer,
 | 
			
		||||
                                             spi::CLOCK_PHASE_LEADING, spi::DATA_RATE_40MHZ> {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXDisplay() = default;
 | 
			
		||||
  ILI9XXXDisplay(uint8_t const *init_sequence, int16_t width, int16_t height, bool invert_colors)
 | 
			
		||||
      : init_sequence_{init_sequence}, width_{width}, height_{height}, pre_invertcolors_{invert_colors} {
 | 
			
		||||
  ILI9XXXDisplay(uint8_t const *init_sequence, int16_t width, int16_t height)
 | 
			
		||||
      : init_sequence_{init_sequence}, width_{width}, height_{height} {
 | 
			
		||||
    uint8_t cmd, num_args, bits;
 | 
			
		||||
    const uint8_t *addr = init_sequence;
 | 
			
		||||
    while ((cmd = *addr++) != 0) {
 | 
			
		||||
@@ -144,7 +144,7 @@ class ILI9XXXDisplay : public display::DisplayBuffer,
 | 
			
		||||
  bool need_update_ = false;
 | 
			
		||||
  bool is_18bitdisplay_ = false;
 | 
			
		||||
  PixelMode pixel_mode_{};
 | 
			
		||||
  bool pre_invertcolors_ = false;
 | 
			
		||||
  bool pre_invertcolors_{};
 | 
			
		||||
  display::ColorOrder color_order_{display::COLOR_ORDER_BGR};
 | 
			
		||||
  bool swap_xy_{};
 | 
			
		||||
  bool mirror_x_{};
 | 
			
		||||
@@ -154,54 +154,54 @@ class ILI9XXXDisplay : public display::DisplayBuffer,
 | 
			
		||||
//-----------   M5Stack display --------------
 | 
			
		||||
class ILI9XXXM5Stack : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXM5Stack() : ILI9XXXDisplay(INITCMD_M5STACK, 320, 240, true) {}
 | 
			
		||||
  ILI9XXXM5Stack() : ILI9XXXDisplay(INITCMD_M5STACK, 320, 240) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   M5Stack display --------------
 | 
			
		||||
class ILI9XXXM5CORE : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXM5CORE() : ILI9XXXDisplay(INITCMD_M5CORE, 320, 240, true) {}
 | 
			
		||||
  ILI9XXXM5CORE() : ILI9XXXDisplay(INITCMD_M5CORE, 320, 240) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ST7789V display --------------
 | 
			
		||||
class ILI9XXXST7789V : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXST7789V() : ILI9XXXDisplay(INITCMD_ST7789V, 240, 320, false) {}
 | 
			
		||||
  ILI9XXXST7789V() : ILI9XXXDisplay(INITCMD_ST7789V, 240, 320) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9XXX_24_TFT display --------------
 | 
			
		||||
class ILI9XXXILI9341 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXILI9341() : ILI9XXXDisplay(INITCMD_ILI9341, 240, 320, false) {}
 | 
			
		||||
  ILI9XXXILI9341() : ILI9XXXDisplay(INITCMD_ILI9341, 240, 320) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9XXX_24_TFT rotated display --------------
 | 
			
		||||
class ILI9XXXILI9342 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXILI9342() : ILI9XXXDisplay(INITCMD_ILI9341, 320, 240, false) {}
 | 
			
		||||
  ILI9XXXILI9342() : ILI9XXXDisplay(INITCMD_ILI9341, 320, 240) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9XXX_??_TFT rotated display --------------
 | 
			
		||||
class ILI9XXXILI9481 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXILI9481() : ILI9XXXDisplay(INITCMD_ILI9481, 480, 320, false) {}
 | 
			
		||||
  ILI9XXXILI9481() : ILI9XXXDisplay(INITCMD_ILI9481, 480, 320) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9481 in 18 bit mode --------------
 | 
			
		||||
class ILI9XXXILI948118 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXILI948118() : ILI9XXXDisplay(INITCMD_ILI9481_18, 320, 480, true) {}
 | 
			
		||||
  ILI9XXXILI948118() : ILI9XXXDisplay(INITCMD_ILI9481_18, 320, 480) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9XXX_35_TFT rotated display --------------
 | 
			
		||||
class ILI9XXXILI9486 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXILI9486() : ILI9XXXDisplay(INITCMD_ILI9486, 480, 320, false) {}
 | 
			
		||||
  ILI9XXXILI9486() : ILI9XXXDisplay(INITCMD_ILI9486, 480, 320) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class ILI9XXXILI9488 : public ILI9XXXDisplay {
 | 
			
		||||
 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:
 | 
			
		||||
  void set_madctl() override {
 | 
			
		||||
@@ -246,34 +246,34 @@ class WAVESHARERES35 : public ILI9XXXILI9488 {
 | 
			
		||||
//-----------   ILI9XXX_35_TFT origin colors rotated display --------------
 | 
			
		||||
class ILI9XXXILI9488A : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXILI9488A() : ILI9XXXDisplay(INITCMD_ILI9488_A, 480, 320, true) {}
 | 
			
		||||
  ILI9XXXILI9488A() : ILI9XXXDisplay(INITCMD_ILI9488_A, 480, 320) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9XXX_35_TFT rotated display --------------
 | 
			
		||||
class ILI9XXXST7796 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXST7796() : ILI9XXXDisplay(INITCMD_ST7796, 320, 480, false) {}
 | 
			
		||||
  ILI9XXXST7796() : ILI9XXXDisplay(INITCMD_ST7796, 320, 480) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class ILI9XXXS3Box : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXS3Box() : ILI9XXXDisplay(INITCMD_S3BOX, 320, 240, false) {}
 | 
			
		||||
  ILI9XXXS3Box() : ILI9XXXDisplay(INITCMD_S3BOX, 320, 240) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class ILI9XXXS3BoxLite : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXS3BoxLite() : ILI9XXXDisplay(INITCMD_S3BOXLITE, 320, 240, true) {}
 | 
			
		||||
  ILI9XXXS3BoxLite() : ILI9XXXDisplay(INITCMD_S3BOXLITE, 320, 240) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class ILI9XXXGC9A01A : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXGC9A01A() : ILI9XXXDisplay(INITCMD_GC9A01A, 240, 240, true) {}
 | 
			
		||||
  ILI9XXXGC9A01A() : ILI9XXXDisplay(INITCMD_GC9A01A, 240, 240) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//-----------   ILI9XXX_24_TFT display --------------
 | 
			
		||||
class ILI9XXXST7735 : public ILI9XXXDisplay {
 | 
			
		||||
 public:
 | 
			
		||||
  ILI9XXXST7735() : ILI9XXXDisplay(INITCMD_ST7735, 128, 160, false) {}
 | 
			
		||||
  ILI9XXXST7735() : ILI9XXXDisplay(INITCMD_ST7735, 128, 160) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace ili9xxx
 | 
			
		||||
 
 | 
			
		||||
@@ -101,7 +101,6 @@ static const uint8_t PROGMEM INITCMD_ILI9481[] = {
 | 
			
		||||
  ILI9XXX_MADCTL  , 1, MADCTL_MV | MADCTL_BGR,       // Memory Access Control
 | 
			
		||||
  ILI9XXX_CSCON , 1, 0x01,
 | 
			
		||||
  ILI9XXX_PIXFMT, 1, 0x55,  // 16 bit mode
 | 
			
		||||
  ILI9XXX_INVON, 0,
 | 
			
		||||
  ILI9XXX_DISPON, 0x80,     // Set display on
 | 
			
		||||
  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_CSCON , 1, 0x01,
 | 
			
		||||
    ILI9XXX_PIXFMT, 1, 0x66,  // 18 bit mode
 | 
			
		||||
    ILI9XXX_INVON, 0,
 | 
			
		||||
    ILI9XXX_DISPON, 0x80,     // Set display on
 | 
			
		||||
    0x00 // end
 | 
			
		||||
};
 | 
			
		||||
@@ -204,7 +202,6 @@ static const uint8_t PROGMEM INITCMD_ILI9488_A[] = {
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
  ILI9XXX_SLPOUT,  0x80,    // Exit sleep mode
 | 
			
		||||
  //ILI9XXX_INVON  , 0,
 | 
			
		||||
  ILI9XXX_DISPON,  0x80,    // Set display on
 | 
			
		||||
  0x00 // end
 | 
			
		||||
};
 | 
			
		||||
 
 | 
			
		||||
@@ -8,6 +8,8 @@
 | 
			
		||||
#endif
 | 
			
		||||
#include <driver/ledc.h>
 | 
			
		||||
 | 
			
		||||
#include <cinttypes>
 | 
			
		||||
 | 
			
		||||
#define CLOCK_FREQUENCY 80e6f
 | 
			
		||||
 | 
			
		||||
#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 float duty_rounded = roundf(state * max_duty);
 | 
			
		||||
  auto duty = static_cast<uint32_t>(duty_rounded);
 | 
			
		||||
  ESP_LOGV(TAG, "Setting duty: %" PRIu32 " on channel %u", duty, this->channel_);
 | 
			
		||||
#ifdef USE_ARDUINO
 | 
			
		||||
  ESP_LOGV(TAG, "Setting duty: %u on channel %u", duty, this->channel_);
 | 
			
		||||
  ledcWrite(this->channel_, duty);
 | 
			
		||||
#endif
 | 
			
		||||
#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 chan_num = static_cast<ledc_channel_t>(channel_ % 8);
 | 
			
		||||
  int hpoint = ledc_angle_to_htop(this->phase_angle_, this->bit_depth_);
 | 
			
		||||
  ledc_set_duty_with_hpoint(speed_mode, chan_num, duty, hpoint);
 | 
			
		||||
  ledc_update_duty(speed_mode, chan_num);
 | 
			
		||||
  if (duty == max_duty) {
 | 
			
		||||
    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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -1,10 +1,6 @@
 | 
			
		||||
import json
 | 
			
		||||
import logging
 | 
			
		||||
from os.path import (
 | 
			
		||||
    dirname,
 | 
			
		||||
    isfile,
 | 
			
		||||
    join,
 | 
			
		||||
)
 | 
			
		||||
from os.path import dirname, isfile, join
 | 
			
		||||
 | 
			
		||||
import esphome.codegen as cg
 | 
			
		||||
import esphome.config_validation as cv
 | 
			
		||||
@@ -174,12 +170,11 @@ def _notify_old_style(config):
 | 
			
		||||
    return config
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# NOTE: Keep this in mind when updating the recommended version:
 | 
			
		||||
#  * For all constants below, update platformio.ini (in this repo)
 | 
			
		||||
# The dev and latest branches will be at *least* this version, which is what matters.
 | 
			
		||||
ARDUINO_VERSIONS = {
 | 
			
		||||
    "dev": (cv.Version(0, 0, 0), "https://github.com/libretiny-eu/libretiny.git"),
 | 
			
		||||
    "latest": (cv.Version(0, 0, 0), None),
 | 
			
		||||
    "recommended": (cv.Version(1, 5, 1), None),
 | 
			
		||||
    "dev": (cv.Version(1, 7, 0), "https://github.com/libretiny-eu/libretiny.git"),
 | 
			
		||||
    "latest": (cv.Version(1, 7, 0), "libretiny"),
 | 
			
		||||
    "recommended": (cv.Version(1, 7, 0), None),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@@ -282,10 +277,10 @@ async def component_to_code(config):
 | 
			
		||||
    # if platform version is a valid version constraint, prefix the default package
 | 
			
		||||
    framework = config[CONF_FRAMEWORK]
 | 
			
		||||
    cv.platformio_version_constraint(framework[CONF_VERSION])
 | 
			
		||||
    if str(framework[CONF_VERSION]) != "0.0.0":
 | 
			
		||||
        cg.add_platformio_option("platform", f"libretiny @ {framework[CONF_VERSION]}")
 | 
			
		||||
    elif framework[CONF_SOURCE]:
 | 
			
		||||
    if framework[CONF_SOURCE]:
 | 
			
		||||
        cg.add_platformio_option("platform", framework[CONF_SOURCE])
 | 
			
		||||
    elif str(framework[CONF_VERSION]) != "0.0.0":
 | 
			
		||||
        cg.add_platformio_option("platform", f"libretiny @ {framework[CONF_VERSION]}")
 | 
			
		||||
    else:
 | 
			
		||||
        cg.add_platformio_option("platform", "libretiny")
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										1
									
								
								esphome/components/ltr501/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								esphome/components/ltr501/__init__.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1 @@
 | 
			
		||||
CODEOWNERS = ["@latonita"]
 | 
			
		||||
							
								
								
									
										542
									
								
								esphome/components/ltr501/ltr501.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										542
									
								
								esphome/components/ltr501/ltr501.cpp
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,542 @@
 | 
			
		||||
#include "ltr501.h"
 | 
			
		||||
#include "esphome/core/application.h"
 | 
			
		||||
#include "esphome/core/log.h"
 | 
			
		||||
#include "esphome/core/helpers.h"
 | 
			
		||||
 | 
			
		||||
using esphome::i2c::ErrorCode;
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace ltr501 {
 | 
			
		||||
 | 
			
		||||
static const char *const TAG = "ltr501";
 | 
			
		||||
 | 
			
		||||
static const uint8_t MAX_TRIES = 5;
 | 
			
		||||
static const uint8_t MAX_SENSITIVITY_ADJUSTMENTS = 10;
 | 
			
		||||
 | 
			
		||||
struct GainTimePair {
 | 
			
		||||
  AlsGain501 gain;
 | 
			
		||||
  IntegrationTime501 time;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
bool operator==(const GainTimePair &lhs, const GainTimePair &rhs) {
 | 
			
		||||
  return lhs.gain == rhs.gain && lhs.time == rhs.time;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool operator!=(const GainTimePair &lhs, const GainTimePair &rhs) {
 | 
			
		||||
  return !(lhs.gain == rhs.gain && lhs.time == rhs.time);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<typename T, size_t size> T get_next(const T (&array)[size], const T val) {
 | 
			
		||||
  size_t i = 0;
 | 
			
		||||
  size_t idx = -1;
 | 
			
		||||
  while (idx == -1 && i < size) {
 | 
			
		||||
    if (array[i] == val) {
 | 
			
		||||
      idx = i;
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
    i++;
 | 
			
		||||
  }
 | 
			
		||||
  if (idx == -1 || i + 1 >= size)
 | 
			
		||||
    return val;
 | 
			
		||||
  return array[i + 1];
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
template<typename T, size_t size> T get_prev(const T (&array)[size], const T val) {
 | 
			
		||||
  size_t i = size - 1;
 | 
			
		||||
  size_t idx = -1;
 | 
			
		||||
  while (idx == -1 && i > 0) {
 | 
			
		||||
    if (array[i] == val) {
 | 
			
		||||
      idx = i;
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
    i--;
 | 
			
		||||
  }
 | 
			
		||||
  if (idx == -1 || i == 0)
 | 
			
		||||
    return val;
 | 
			
		||||
  return array[i - 1];
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
static uint16_t get_itime_ms(IntegrationTime501 time) {
 | 
			
		||||
  static const uint16_t ALS_INT_TIME[4] = {100, 50, 200, 400};
 | 
			
		||||
  return ALS_INT_TIME[time & 0b11];
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
static uint16_t get_meas_time_ms(MeasurementRepeatRate rate) {
 | 
			
		||||
  static const uint16_t ALS_MEAS_RATE[8] = {50, 100, 200, 500, 1000, 2000, 2000, 2000};
 | 
			
		||||
  return ALS_MEAS_RATE[rate & 0b111];
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
static float get_gain_coeff(AlsGain501 gain) { return gain == AlsGain501::GAIN_1 ? 1.0f : 150.0f; }
 | 
			
		||||
 | 
			
		||||
static float get_ps_gain_coeff(PsGain501 gain) {
 | 
			
		||||
  static const float PS_GAIN[4] = {1, 4, 8, 16};
 | 
			
		||||
  return PS_GAIN[gain & 0b11];
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::setup() {
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "Setting up LTR-501/301/558");
 | 
			
		||||
  // As per datasheet we need to wait at least 100ms after power on to get ALS chip responsive
 | 
			
		||||
  this->set_timeout(100, [this]() { this->state_ = State::DELAYED_SETUP; });
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::dump_config() {
 | 
			
		||||
  auto get_device_type = [](LtrType typ) {
 | 
			
		||||
    switch (typ) {
 | 
			
		||||
      case LtrType::LTR_TYPE_ALS_ONLY:
 | 
			
		||||
        return "ALS only";
 | 
			
		||||
      case LtrType::LTR_TYPE_PS_ONLY:
 | 
			
		||||
        return "PS only";
 | 
			
		||||
      case LtrType::LTR_TYPE_ALS_AND_PS:
 | 
			
		||||
        return "Als + PS";
 | 
			
		||||
      default:
 | 
			
		||||
        return "Unknown";
 | 
			
		||||
    }
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  LOG_I2C_DEVICE(this);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Device type: %s", get_device_type(this->ltr_type_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Automatic mode: %s", ONOFF(this->automatic_mode_enabled_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Gain: %.0fx", get_gain_coeff(this->gain_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Integration time: %d ms", get_itime_ms(this->integration_time_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Measurement repeat rate: %d ms", get_meas_time_ms(this->repeat_rate_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Glass attenuation factor: %f", this->glass_attenuation_factor_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Proximity gain: %.0fx", get_ps_gain_coeff(this->ps_gain_));
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Proximity cooldown time: %d s", this->ps_cooldown_time_s_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Proximity high threshold: %d", this->ps_threshold_high_);
 | 
			
		||||
  ESP_LOGCONFIG(TAG, "  Proximity low threshold: %d", this->ps_threshold_low_);
 | 
			
		||||
 | 
			
		||||
  LOG_UPDATE_INTERVAL(this);
 | 
			
		||||
 | 
			
		||||
  LOG_SENSOR("  ", "ALS calculated lux", this->ambient_light_sensor_);
 | 
			
		||||
  LOG_SENSOR("  ", "CH1 Infrared counts", this->infrared_counts_sensor_);
 | 
			
		||||
  LOG_SENSOR("  ", "CH0 Visible+IR counts", this->full_spectrum_counts_sensor_);
 | 
			
		||||
  LOG_SENSOR("  ", "Actual gain", this->actual_gain_sensor_);
 | 
			
		||||
 | 
			
		||||
  if (this->is_failed()) {
 | 
			
		||||
    ESP_LOGE(TAG, "Communication with I2C LTR-501/301/558 failed!");
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::update() {
 | 
			
		||||
  if (!this->is_als_()) {
 | 
			
		||||
    ESP_LOGW(TAG, "Update. ALS data not available. Change configuration to ALS or ALS_PS.");
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
  if (this->is_ready() && this->is_als_() && this->state_ == State::IDLE) {
 | 
			
		||||
    ESP_LOGV(TAG, "Update. Initiating new ALS data collection.");
 | 
			
		||||
 | 
			
		||||
    this->state_ = this->automatic_mode_enabled_ ? State::COLLECTING_DATA_AUTO : State::WAITING_FOR_DATA;
 | 
			
		||||
 | 
			
		||||
    this->als_readings_.ch0 = 0;
 | 
			
		||||
    this->als_readings_.ch1 = 0;
 | 
			
		||||
    this->als_readings_.gain = this->gain_;
 | 
			
		||||
    this->als_readings_.integration_time = this->integration_time_;
 | 
			
		||||
    this->als_readings_.lux = 0;
 | 
			
		||||
    this->als_readings_.number_of_adjustments = 0;
 | 
			
		||||
 | 
			
		||||
  } else {
 | 
			
		||||
    ESP_LOGV(TAG, "Update. Component not ready yet.");
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::loop() {
 | 
			
		||||
  ErrorCode err = i2c::ERROR_OK;
 | 
			
		||||
  static uint8_t tries{0};
 | 
			
		||||
 | 
			
		||||
  switch (this->state_) {
 | 
			
		||||
    case State::DELAYED_SETUP:
 | 
			
		||||
      err = this->write(nullptr, 0);
 | 
			
		||||
      if (err != i2c::ERROR_OK) {
 | 
			
		||||
        ESP_LOGW(TAG, "i2c connection failed");
 | 
			
		||||
        this->mark_failed();
 | 
			
		||||
      }
 | 
			
		||||
      this->configure_reset_();
 | 
			
		||||
      if (this->is_als_()) {
 | 
			
		||||
        this->configure_als_();
 | 
			
		||||
        this->configure_integration_time_(this->integration_time_);
 | 
			
		||||
      }
 | 
			
		||||
      if (this->is_ps_()) {
 | 
			
		||||
        this->configure_ps_();
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      this->state_ = State::IDLE;
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    case State::IDLE:
 | 
			
		||||
      if (this->is_ps_()) {
 | 
			
		||||
        this->check_and_trigger_ps_();
 | 
			
		||||
      }
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    case State::WAITING_FOR_DATA:
 | 
			
		||||
      if (this->is_als_data_ready_(this->als_readings_) == DataAvail::DATA_OK) {
 | 
			
		||||
        tries = 0;
 | 
			
		||||
        ESP_LOGV(TAG, "Reading sensor data assuming gain = %.0fx, time = %d ms",
 | 
			
		||||
                 get_gain_coeff(this->als_readings_.gain), get_itime_ms(this->als_readings_.integration_time));
 | 
			
		||||
        this->read_sensor_data_(this->als_readings_);
 | 
			
		||||
        this->apply_lux_calculation_(this->als_readings_);
 | 
			
		||||
        this->state_ = State::DATA_COLLECTED;
 | 
			
		||||
      } else if (tries >= MAX_TRIES) {
 | 
			
		||||
        ESP_LOGW(TAG, "Can't get data after several tries. Aborting.");
 | 
			
		||||
        tries = 0;
 | 
			
		||||
        this->status_set_warning();
 | 
			
		||||
        this->state_ = State::IDLE;
 | 
			
		||||
        return;
 | 
			
		||||
      } else {
 | 
			
		||||
        tries++;
 | 
			
		||||
      }
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    case State::COLLECTING_DATA_AUTO:
 | 
			
		||||
    case State::DATA_COLLECTED:
 | 
			
		||||
      // first measurement in auto mode (COLLECTING_DATA_AUTO state) require device reconfiguration
 | 
			
		||||
      if (this->state_ == State::COLLECTING_DATA_AUTO || this->are_adjustments_required_(this->als_readings_)) {
 | 
			
		||||
        this->state_ = State::ADJUSTMENT_IN_PROGRESS;
 | 
			
		||||
        ESP_LOGD(TAG, "Reconfiguring sensitivity: gain = %.0fx, time = %d ms", get_gain_coeff(this->als_readings_.gain),
 | 
			
		||||
                 get_itime_ms(this->als_readings_.integration_time));
 | 
			
		||||
        this->configure_integration_time_(this->als_readings_.integration_time);
 | 
			
		||||
        this->configure_gain_(this->als_readings_.gain);
 | 
			
		||||
        // if sensitivity adjustment needed - need to wait for first data samples after setting new parameters
 | 
			
		||||
        this->set_timeout(2 * get_meas_time_ms(this->repeat_rate_),
 | 
			
		||||
                          [this]() { this->state_ = State::WAITING_FOR_DATA; });
 | 
			
		||||
      } else {
 | 
			
		||||
        this->state_ = State::READY_TO_PUBLISH;
 | 
			
		||||
      }
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    case State::ADJUSTMENT_IN_PROGRESS:
 | 
			
		||||
      // nothing to be done, just waiting for the timeout
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    case State::READY_TO_PUBLISH:
 | 
			
		||||
      this->publish_data_part_1_(this->als_readings_);
 | 
			
		||||
      this->state_ = State::KEEP_PUBLISHING;
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    case State::KEEP_PUBLISHING:
 | 
			
		||||
      this->publish_data_part_2_(this->als_readings_);
 | 
			
		||||
      this->status_clear_warning();
 | 
			
		||||
      this->state_ = State::IDLE;
 | 
			
		||||
      break;
 | 
			
		||||
 | 
			
		||||
    default:
 | 
			
		||||
      break;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::check_and_trigger_ps_() {
 | 
			
		||||
  static uint32_t last_high_trigger_time{0};
 | 
			
		||||
  static uint32_t last_low_trigger_time{0};
 | 
			
		||||
  uint16_t ps_data = this->read_ps_data_();
 | 
			
		||||
  uint32_t now = millis();
 | 
			
		||||
 | 
			
		||||
  if (ps_data != this->ps_readings_) {
 | 
			
		||||
    this->ps_readings_ = ps_data;
 | 
			
		||||
    // Higher values - object is closer to sensor
 | 
			
		||||
    if (ps_data > this->ps_threshold_high_ && now - last_high_trigger_time >= this->ps_cooldown_time_s_ * 1000) {
 | 
			
		||||
      last_high_trigger_time = now;
 | 
			
		||||
      ESP_LOGD(TAG, "Proximity high threshold triggered. Value = %d, Trigger level = %d", ps_data,
 | 
			
		||||
               this->ps_threshold_high_);
 | 
			
		||||
      this->on_ps_high_trigger_callback_.call();
 | 
			
		||||
    } else if (ps_data < this->ps_threshold_low_ && now - last_low_trigger_time >= this->ps_cooldown_time_s_ * 1000) {
 | 
			
		||||
      last_low_trigger_time = now;
 | 
			
		||||
      ESP_LOGD(TAG, "Proximity low threshold triggered. Value = %d, Trigger level = %d", ps_data,
 | 
			
		||||
               this->ps_threshold_low_);
 | 
			
		||||
      this->on_ps_low_trigger_callback_.call();
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool LTRAlsPs501Component::check_part_number_() {
 | 
			
		||||
  uint8_t manuf_id = this->reg((uint8_t) CommandRegisters::MANUFAC_ID).get();
 | 
			
		||||
  if (manuf_id != 0x05) {  // 0x05 is Lite-On Semiconductor Corp. ID
 | 
			
		||||
    ESP_LOGW(TAG, "Unknown manufacturer ID: 0x%02X", manuf_id);
 | 
			
		||||
    this->mark_failed();
 | 
			
		||||
    return false;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  // Things getting not really funny here, we can't identify device type by part number ID
 | 
			
		||||
  // ======================== ========= ===== =================
 | 
			
		||||
  // Device                    Part ID   Rev   Capabilities
 | 
			
		||||
  // ======================== ========= ===== =================
 | 
			
		||||
  // ltr-558als                 0x08      0    als + ps
 | 
			
		||||
  // ltr-501als                 0x08      0    als + ps
 | 
			
		||||
  // ltr-301als -               0x08      0    als only
 | 
			
		||||
 | 
			
		||||
  PartIdRegister part_id{0};
 | 
			
		||||
  part_id.raw = this->reg((uint8_t) CommandRegisters::PART_ID).get();
 | 
			
		||||
  if (part_id.part_number_id != 0x08) {
 | 
			
		||||
    ESP_LOGW(TAG, "Unknown part number ID: 0x%02X. LTR-501/301 shall have 0x08. It might not work properly.",
 | 
			
		||||
             part_id.part_number_id);
 | 
			
		||||
    this->status_set_warning();
 | 
			
		||||
    return true;
 | 
			
		||||
  }
 | 
			
		||||
  return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::configure_reset_() {
 | 
			
		||||
  ESP_LOGV(TAG, "Resetting");
 | 
			
		||||
 | 
			
		||||
  AlsControlRegister501 als_ctrl{0};
 | 
			
		||||
  als_ctrl.sw_reset = true;
 | 
			
		||||
  this->reg((uint8_t) CommandRegisters::ALS_CONTR) = als_ctrl.raw;
 | 
			
		||||
  delay(2);
 | 
			
		||||
 | 
			
		||||
  uint8_t tries = MAX_TRIES;
 | 
			
		||||
  do {
 | 
			
		||||
    ESP_LOGV(TAG, "Waiting chip to reset");
 | 
			
		||||
    delay(2);
 | 
			
		||||
    als_ctrl.raw = this->reg((uint8_t) CommandRegisters::ALS_CONTR).get();
 | 
			
		||||
  } while (als_ctrl.sw_reset && tries--);  // while sw reset bit is on - keep waiting
 | 
			
		||||
 | 
			
		||||
  if (als_ctrl.sw_reset) {
 | 
			
		||||
    ESP_LOGW(TAG, "Reset failed");
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::configure_als_() {
 | 
			
		||||
  AlsControlRegister501 als_ctrl{0};
 | 
			
		||||
  als_ctrl.sw_reset = false;
 | 
			
		||||
  als_ctrl.als_mode_active = true;
 | 
			
		||||
  als_ctrl.gain = this->gain_;
 | 
			
		||||
 | 
			
		||||
  ESP_LOGV(TAG, "Setting active mode and gain reg 0x%02X", als_ctrl.raw);
 | 
			
		||||
  this->reg((uint8_t) CommandRegisters::ALS_CONTR) = als_ctrl.raw;
 | 
			
		||||
  delay(5);
 | 
			
		||||
 | 
			
		||||
  uint8_t tries = MAX_TRIES;
 | 
			
		||||
  do {
 | 
			
		||||
    ESP_LOGV(TAG, "Waiting for ALS device to become active...");
 | 
			
		||||
    delay(2);
 | 
			
		||||
    als_ctrl.raw = this->reg((uint8_t) CommandRegisters::ALS_CONTR).get();
 | 
			
		||||
  } while (!als_ctrl.als_mode_active && tries--);  // while active mode is not set - keep waiting
 | 
			
		||||
 | 
			
		||||
  if (!als_ctrl.als_mode_active) {
 | 
			
		||||
    ESP_LOGW(TAG, "Failed to activate ALS device");
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::configure_ps_() {
 | 
			
		||||
  PsMeasurementRateRegister ps_meas{0};
 | 
			
		||||
  ps_meas.ps_measurement_rate = PsMeasurementRate::PS_MEAS_RATE_50MS;
 | 
			
		||||
  this->reg((uint8_t) CommandRegisters::PS_MEAS_RATE) = ps_meas.raw;
 | 
			
		||||
 | 
			
		||||
  PsControlRegister501 ps_ctrl{0};
 | 
			
		||||
  ps_ctrl.ps_mode_active = true;
 | 
			
		||||
  ps_ctrl.ps_mode_xxx = true;
 | 
			
		||||
  this->reg((uint8_t) CommandRegisters::PS_CONTR) = ps_ctrl.raw;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
uint16_t LTRAlsPs501Component::read_ps_data_() {
 | 
			
		||||
  AlsPsStatusRegister als_status{0};
 | 
			
		||||
  als_status.raw = this->reg((uint8_t) CommandRegisters::ALS_PS_STATUS).get();
 | 
			
		||||
  if (!als_status.ps_new_data) {
 | 
			
		||||
    return this->ps_readings_;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  uint8_t ps_low = this->reg((uint8_t) CommandRegisters::PS_DATA_0).get();
 | 
			
		||||
  PsData1Register ps_high;
 | 
			
		||||
  ps_high.raw = this->reg((uint8_t) CommandRegisters::PS_DATA_1).get();
 | 
			
		||||
 | 
			
		||||
  uint16_t val = encode_uint16(ps_high.ps_data_high, ps_low);
 | 
			
		||||
  return val;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::configure_gain_(AlsGain501 gain) {
 | 
			
		||||
  AlsControlRegister501 als_ctrl{0};
 | 
			
		||||
  als_ctrl.als_mode_active = true;
 | 
			
		||||
  als_ctrl.gain = gain;
 | 
			
		||||
  this->reg((uint8_t) CommandRegisters::ALS_CONTR) = als_ctrl.raw;
 | 
			
		||||
  delay(2);
 | 
			
		||||
 | 
			
		||||
  AlsControlRegister501 read_als_ctrl{0};
 | 
			
		||||
  read_als_ctrl.raw = this->reg((uint8_t) CommandRegisters::ALS_CONTR).get();
 | 
			
		||||
  if (read_als_ctrl.gain != gain) {
 | 
			
		||||
    ESP_LOGW(TAG, "Failed to set gain. We will try one more time.");
 | 
			
		||||
    this->reg((uint8_t) CommandRegisters::ALS_CONTR) = als_ctrl.raw;
 | 
			
		||||
    delay(2);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::configure_integration_time_(IntegrationTime501 time) {
 | 
			
		||||
  MeasurementRateRegister501 meas{0};
 | 
			
		||||
  meas.measurement_repeat_rate = this->repeat_rate_;
 | 
			
		||||
  meas.integration_time = time;
 | 
			
		||||
  this->reg((uint8_t) CommandRegisters::MEAS_RATE) = meas.raw;
 | 
			
		||||
  delay(2);
 | 
			
		||||
 | 
			
		||||
  MeasurementRateRegister501 read_meas{0};
 | 
			
		||||
  read_meas.raw = this->reg((uint8_t) CommandRegisters::MEAS_RATE).get();
 | 
			
		||||
  if (read_meas.integration_time != time) {
 | 
			
		||||
    ESP_LOGW(TAG, "Failed to set integration time. We will try one more time.");
 | 
			
		||||
    this->reg((uint8_t) CommandRegisters::MEAS_RATE) = meas.raw;
 | 
			
		||||
    delay(2);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
DataAvail LTRAlsPs501Component::is_als_data_ready_(AlsReadings &data) {
 | 
			
		||||
  AlsPsStatusRegister als_status{0};
 | 
			
		||||
  als_status.raw = this->reg((uint8_t) CommandRegisters::ALS_PS_STATUS).get();
 | 
			
		||||
  if (!als_status.als_new_data)
 | 
			
		||||
    return DataAvail::NO_DATA;
 | 
			
		||||
  ESP_LOGV(TAG, "Data ready, reported gain is %.0fx", get_gain_coeff(als_status.gain));
 | 
			
		||||
  if (data.gain != als_status.gain) {
 | 
			
		||||
    ESP_LOGW(TAG, "Actual gain differs from requested (%.0f)", get_gain_coeff(data.gain));
 | 
			
		||||
    return DataAvail::BAD_DATA;
 | 
			
		||||
  }
 | 
			
		||||
  data.gain = als_status.gain;
 | 
			
		||||
  return DataAvail::DATA_OK;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::read_sensor_data_(AlsReadings &data) {
 | 
			
		||||
  data.ch1 = 0;
 | 
			
		||||
  data.ch0 = 0;
 | 
			
		||||
  uint8_t ch1_0 = this->reg((uint8_t) CommandRegisters::ALS_DATA_CH1_0).get();
 | 
			
		||||
  uint8_t ch1_1 = this->reg((uint8_t) CommandRegisters::ALS_DATA_CH1_1).get();
 | 
			
		||||
  uint8_t ch0_0 = this->reg((uint8_t) CommandRegisters::ALS_DATA_CH0_0).get();
 | 
			
		||||
  uint8_t ch0_1 = this->reg((uint8_t) CommandRegisters::ALS_DATA_CH0_1).get();
 | 
			
		||||
  data.ch1 = encode_uint16(ch1_1, ch1_0);
 | 
			
		||||
  data.ch0 = encode_uint16(ch0_1, ch0_0);
 | 
			
		||||
 | 
			
		||||
  ESP_LOGD(TAG, "Got sensor data: CH1 = %d, CH0 = %d", data.ch1, data.ch0);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool LTRAlsPs501Component::are_adjustments_required_(AlsReadings &data) {
 | 
			
		||||
  if (!this->automatic_mode_enabled_)
 | 
			
		||||
    return false;
 | 
			
		||||
 | 
			
		||||
  // sometimes sensors fail to change sensitivity. this prevents us from infinite loop
 | 
			
		||||
  if (data.number_of_adjustments++ > MAX_SENSITIVITY_ADJUSTMENTS) {
 | 
			
		||||
    ESP_LOGW(TAG, "Too many sensitivity adjustments done. Something wrong with the sensor. Stopping.");
 | 
			
		||||
    return false;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  ESP_LOGV(TAG, "Adjusting sensitivity, run #%d", data.number_of_adjustments);
 | 
			
		||||
 | 
			
		||||
  // available combinations of gain and integration times:
 | 
			
		||||
  static const GainTimePair GAIN_TIME_PAIRS[] = {
 | 
			
		||||
      {AlsGain501::GAIN_1, INTEGRATION_TIME_50MS},    {AlsGain501::GAIN_1, INTEGRATION_TIME_100MS},
 | 
			
		||||
      {AlsGain501::GAIN_150, INTEGRATION_TIME_100MS}, {AlsGain501::GAIN_150, INTEGRATION_TIME_200MS},
 | 
			
		||||
      {AlsGain501::GAIN_150, INTEGRATION_TIME_400MS},
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  GainTimePair current_pair = {data.gain, data.integration_time};
 | 
			
		||||
 | 
			
		||||
  // Here comes funky business with this sensor. it has no internal error checking mechanism
 | 
			
		||||
  // as in later versions (LTR-303/329/559/..) and sensor gets overwhelmed when saturated
 | 
			
		||||
  // and readings are strange. We only check high sensitivity mode for now.
 | 
			
		||||
  // Nothing is documented and it is a result of real-world testing.
 | 
			
		||||
  if (data.gain == AlsGain501::GAIN_150) {
 | 
			
		||||
    // when sensor is saturated it returns various crazy numbers
 | 
			
		||||
    // CH1 = 1, CH0 = 0
 | 
			
		||||
    if (data.ch1 == 1 && data.ch0 == 0) {
 | 
			
		||||
      ESP_LOGV(TAG, "Looks like sensor got saturated (?) CH1 = 1, CH0 = 0, Gain 150x");
 | 
			
		||||
      // fake saturation
 | 
			
		||||
      data.ch0 = 0xffff;
 | 
			
		||||
      data.ch1 = 0xffff;
 | 
			
		||||
    } else if (data.ch1 == 65535 && data.ch0 == 0) {
 | 
			
		||||
      ESP_LOGV(TAG, "Looks like sensor got saturated (?) CH1 = 65535, CH0 = 0, Gain 150x");
 | 
			
		||||
      data.ch0 = 0xffff;
 | 
			
		||||
    } else if (data.ch1 > 1000 && data.ch0 == 0) {
 | 
			
		||||
      ESP_LOGV(TAG, "Looks like sensor got saturated (?) CH1 = %d, CH0 = 0, Gain 150x", data.ch1);
 | 
			
		||||
      data.ch0 = 0xffff;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  static const uint16_t LOW_INTENSITY_THRESHOLD_1 = 100;
 | 
			
		||||
  static const uint16_t LOW_INTENSITY_THRESHOLD_200 = 2000;
 | 
			
		||||
  static const uint16_t HIGH_INTENSITY_THRESHOLD = 25000;
 | 
			
		||||
 | 
			
		||||
  if (data.ch0 <= (data.gain == AlsGain501::GAIN_1 ? LOW_INTENSITY_THRESHOLD_1 : LOW_INTENSITY_THRESHOLD_200) ||
 | 
			
		||||
      (data.gain == AlsGain501::GAIN_1 && data.lux < 320)) {
 | 
			
		||||
    GainTimePair next_pair = get_next(GAIN_TIME_PAIRS, current_pair);
 | 
			
		||||
    if (next_pair != current_pair) {
 | 
			
		||||
      data.gain = next_pair.gain;
 | 
			
		||||
      data.integration_time = next_pair.time;
 | 
			
		||||
      ESP_LOGV(TAG, "Low illuminance. Increasing sensitivity.");
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
  } else if (data.ch0 >= HIGH_INTENSITY_THRESHOLD || data.ch1 >= HIGH_INTENSITY_THRESHOLD) {
 | 
			
		||||
    GainTimePair prev_pair = get_prev(GAIN_TIME_PAIRS, current_pair);
 | 
			
		||||
    if (prev_pair != current_pair) {
 | 
			
		||||
      data.gain = prev_pair.gain;
 | 
			
		||||
      data.integration_time = prev_pair.time;
 | 
			
		||||
      ESP_LOGV(TAG, "High illuminance. Decreasing sensitivity.");
 | 
			
		||||
      return true;
 | 
			
		||||
    }
 | 
			
		||||
  } else {
 | 
			
		||||
    ESP_LOGD(TAG, "Illuminance is good enough.");
 | 
			
		||||
    return false;
 | 
			
		||||
  }
 | 
			
		||||
  ESP_LOGD(TAG, "Can't adjust sensitivity anymore.");
 | 
			
		||||
  return false;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::apply_lux_calculation_(AlsReadings &data) {
 | 
			
		||||
  if ((data.ch0 == 0xFFFF) || (data.ch1 == 0xFFFF)) {
 | 
			
		||||
    ESP_LOGW(TAG, "Sensors got saturated");
 | 
			
		||||
    data.lux = 0.0f;
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  if ((data.ch0 == 0x0000) && (data.ch1 == 0x0000)) {
 | 
			
		||||
    ESP_LOGW(TAG, "Sensors blacked out");
 | 
			
		||||
    data.lux = 0.0f;
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  float ch0 = data.ch0;
 | 
			
		||||
  float ch1 = data.ch1;
 | 
			
		||||
  float ratio = ch1 / (ch0 + ch1);
 | 
			
		||||
  float als_gain = get_gain_coeff(data.gain);
 | 
			
		||||
  float als_time = ((float) get_itime_ms(data.integration_time)) / 100.0f;
 | 
			
		||||
  float inv_pfactor = this->glass_attenuation_factor_;
 | 
			
		||||
  float lux = 0.0f;
 | 
			
		||||
 | 
			
		||||
  // method from
 | 
			
		||||
  // https://github.com/fards/Ainol_fire_kernel/blob/83832cf8a3082fd8e963230f4b1984479d1f1a84/customer/drivers/lightsensor/ltr501als.c#L295
 | 
			
		||||
 | 
			
		||||
  if (ratio < 0.45) {
 | 
			
		||||
    lux = 1.7743 * ch0 + 1.1059 * ch1;
 | 
			
		||||
  } else if (ratio < 0.64) {
 | 
			
		||||
    lux = 3.7725 * ch0 - 1.3363 * ch1;
 | 
			
		||||
  } else if (ratio < 0.85) {
 | 
			
		||||
    lux = 1.6903 * ch0 - 0.1693 * ch1;
 | 
			
		||||
  } else {
 | 
			
		||||
    ESP_LOGW(TAG, "Impossible ch1/(ch0 + ch1) ratio");
 | 
			
		||||
    lux = 0.0f;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  lux = inv_pfactor * lux / als_gain / als_time;
 | 
			
		||||
  data.lux = lux;
 | 
			
		||||
 | 
			
		||||
  ESP_LOGD(TAG, "Lux calculation: ratio %.3f, gain %.0fx, int time %.1f, inv_pfactor %.3f, lux %.3f", ratio, als_gain,
 | 
			
		||||
           als_time, inv_pfactor, lux);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::publish_data_part_1_(AlsReadings &data) {
 | 
			
		||||
  if (this->proximity_counts_sensor_ != nullptr) {
 | 
			
		||||
    this->proximity_counts_sensor_->publish_state(this->ps_readings_);
 | 
			
		||||
  }
 | 
			
		||||
  if (this->ambient_light_sensor_ != nullptr) {
 | 
			
		||||
    this->ambient_light_sensor_->publish_state(data.lux);
 | 
			
		||||
  }
 | 
			
		||||
  if (this->infrared_counts_sensor_ != nullptr) {
 | 
			
		||||
    this->infrared_counts_sensor_->publish_state(data.ch1);
 | 
			
		||||
  }
 | 
			
		||||
  if (this->full_spectrum_counts_sensor_ != nullptr) {
 | 
			
		||||
    this->full_spectrum_counts_sensor_->publish_state(data.ch0);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void LTRAlsPs501Component::publish_data_part_2_(AlsReadings &data) {
 | 
			
		||||
  if (this->actual_gain_sensor_ != nullptr) {
 | 
			
		||||
    this->actual_gain_sensor_->publish_state(get_gain_coeff(data.gain));
 | 
			
		||||
  }
 | 
			
		||||
  if (this->actual_integration_time_sensor_ != nullptr) {
 | 
			
		||||
    this->actual_integration_time_sensor_->publish_state(get_itime_ms(data.integration_time));
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
}  // namespace ltr501
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										184
									
								
								esphome/components/ltr501/ltr501.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										184
									
								
								esphome/components/ltr501/ltr501.h
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,184 @@
 | 
			
		||||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "esphome/components/i2c/i2c.h"
 | 
			
		||||
#include "esphome/components/sensor/sensor.h"
 | 
			
		||||
#include "esphome/core/component.h"
 | 
			
		||||
#include "esphome/core/optional.h"
 | 
			
		||||
#include "esphome/core/automation.h"
 | 
			
		||||
 | 
			
		||||
#include "ltr_definitions_501.h"
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace ltr501 {
 | 
			
		||||
 | 
			
		||||
enum DataAvail : uint8_t { NO_DATA, BAD_DATA, DATA_OK };
 | 
			
		||||
 | 
			
		||||
enum LtrType : uint8_t {
 | 
			
		||||
  LTR_TYPE_UNKNOWN = 0,
 | 
			
		||||
  LTR_TYPE_ALS_ONLY = 1,
 | 
			
		||||
  LTR_TYPE_PS_ONLY = 2,
 | 
			
		||||
  LTR_TYPE_ALS_AND_PS = 3,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class LTRAlsPs501Component : public PollingComponent, public i2c::I2CDevice {
 | 
			
		||||
 public:
 | 
			
		||||
  //
 | 
			
		||||
  // EspHome framework functions
 | 
			
		||||
  //
 | 
			
		||||
  float get_setup_priority() const override { return setup_priority::DATA; }
 | 
			
		||||
  void setup() override;
 | 
			
		||||
  void dump_config() override;
 | 
			
		||||
  void update() override;
 | 
			
		||||
  void loop() override;
 | 
			
		||||
 | 
			
		||||
  // Configuration setters : General
 | 
			
		||||
  //
 | 
			
		||||
  void set_ltr_type(LtrType type) { this->ltr_type_ = type; }
 | 
			
		||||
 | 
			
		||||
  // Configuration setters : ALS
 | 
			
		||||
  //
 | 
			
		||||
  void set_als_auto_mode(bool enable) { this->automatic_mode_enabled_ = enable; }
 | 
			
		||||
  void set_als_gain(AlsGain501 gain) { this->gain_ = gain; }
 | 
			
		||||
  void set_als_integration_time(IntegrationTime501 time) { this->integration_time_ = time; }
 | 
			
		||||
  void set_als_meas_repeat_rate(MeasurementRepeatRate rate) { this->repeat_rate_ = rate; }
 | 
			
		||||
  void set_als_glass_attenuation_factor(float factor) { this->glass_attenuation_factor_ = factor; }
 | 
			
		||||
 | 
			
		||||
  // Configuration setters : PS
 | 
			
		||||
  //
 | 
			
		||||
  void set_ps_high_threshold(uint16_t threshold) { this->ps_threshold_high_ = threshold; }
 | 
			
		||||
  void set_ps_low_threshold(uint16_t threshold) { this->ps_threshold_low_ = threshold; }
 | 
			
		||||
  void set_ps_cooldown_time_s(uint16_t time) { this->ps_cooldown_time_s_ = time; }
 | 
			
		||||
  void set_ps_gain(PsGain501 gain) { this->ps_gain_ = gain; }
 | 
			
		||||
 | 
			
		||||
  // Sensors setters
 | 
			
		||||
  //
 | 
			
		||||
  void set_ambient_light_sensor(sensor::Sensor *sensor) { this->ambient_light_sensor_ = sensor; }
 | 
			
		||||
  void set_full_spectrum_counts_sensor(sensor::Sensor *sensor) { this->full_spectrum_counts_sensor_ = sensor; }
 | 
			
		||||
  void set_infrared_counts_sensor(sensor::Sensor *sensor) { this->infrared_counts_sensor_ = sensor; }
 | 
			
		||||
  void set_actual_gain_sensor(sensor::Sensor *sensor) { this->actual_gain_sensor_ = sensor; }
 | 
			
		||||
  void set_actual_integration_time_sensor(sensor::Sensor *sensor) { this->actual_integration_time_sensor_ = sensor; }
 | 
			
		||||
  void set_proximity_counts_sensor(sensor::Sensor *sensor) { this->proximity_counts_sensor_ = sensor; }
 | 
			
		||||
 | 
			
		||||
 protected:
 | 
			
		||||
  //
 | 
			
		||||
  // Internal state machine, used to split all the actions into
 | 
			
		||||
  // small steps in loop() to make sure we are not blocking execution
 | 
			
		||||
  //
 | 
			
		||||
  enum class State : uint8_t {
 | 
			
		||||
    NOT_INITIALIZED,
 | 
			
		||||
    DELAYED_SETUP,
 | 
			
		||||
    IDLE,
 | 
			
		||||
    WAITING_FOR_DATA,
 | 
			
		||||
    COLLECTING_DATA_AUTO,
 | 
			
		||||
    DATA_COLLECTED,
 | 
			
		||||
    ADJUSTMENT_IN_PROGRESS,
 | 
			
		||||
    READY_TO_PUBLISH,
 | 
			
		||||
    KEEP_PUBLISHING
 | 
			
		||||
  } state_{State::NOT_INITIALIZED};
 | 
			
		||||
 | 
			
		||||
  LtrType ltr_type_{LtrType::LTR_TYPE_ALS_ONLY};
 | 
			
		||||
 | 
			
		||||
  //
 | 
			
		||||
  // Current measurements data
 | 
			
		||||
  //
 | 
			
		||||
  struct AlsReadings {
 | 
			
		||||
    uint16_t ch0{0};
 | 
			
		||||
    uint16_t ch1{0};
 | 
			
		||||
    AlsGain501 gain{AlsGain501::GAIN_1};
 | 
			
		||||
    IntegrationTime501 integration_time{IntegrationTime501::INTEGRATION_TIME_100MS};
 | 
			
		||||
    float lux{0.0f};
 | 
			
		||||
    uint8_t number_of_adjustments{0};
 | 
			
		||||
  } als_readings_;
 | 
			
		||||
  uint16_t ps_readings_{0xfffe};
 | 
			
		||||
 | 
			
		||||
  inline bool is_als_() const {
 | 
			
		||||
    return this->ltr_type_ == LtrType::LTR_TYPE_ALS_ONLY || this->ltr_type_ == LtrType::LTR_TYPE_ALS_AND_PS;
 | 
			
		||||
  }
 | 
			
		||||
  inline bool is_ps_() const {
 | 
			
		||||
    return this->ltr_type_ == LtrType::LTR_TYPE_PS_ONLY || this->ltr_type_ == LtrType::LTR_TYPE_ALS_AND_PS;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  //
 | 
			
		||||
  // Device interaction and data manipulation
 | 
			
		||||
  //
 | 
			
		||||
  bool check_part_number_();
 | 
			
		||||
 | 
			
		||||
  void configure_reset_();
 | 
			
		||||
  void configure_als_();
 | 
			
		||||
  void configure_integration_time_(IntegrationTime501 time);
 | 
			
		||||
  void configure_gain_(AlsGain501 gain);
 | 
			
		||||
  DataAvail is_als_data_ready_(AlsReadings &data);
 | 
			
		||||
  void read_sensor_data_(AlsReadings &data);
 | 
			
		||||
  bool are_adjustments_required_(AlsReadings &data);
 | 
			
		||||
  void apply_lux_calculation_(AlsReadings &data);
 | 
			
		||||
  void publish_data_part_1_(AlsReadings &data);
 | 
			
		||||
  void publish_data_part_2_(AlsReadings &data);
 | 
			
		||||
 | 
			
		||||
  void configure_ps_();
 | 
			
		||||
  uint16_t read_ps_data_();
 | 
			
		||||
  void check_and_trigger_ps_();
 | 
			
		||||
 | 
			
		||||
  //
 | 
			
		||||
  // Component configuration
 | 
			
		||||
  //
 | 
			
		||||
  bool automatic_mode_enabled_{false};
 | 
			
		||||
  AlsGain501 gain_{AlsGain501::GAIN_1};
 | 
			
		||||
  IntegrationTime501 integration_time_{IntegrationTime501::INTEGRATION_TIME_100MS};
 | 
			
		||||
  MeasurementRepeatRate repeat_rate_{MeasurementRepeatRate::REPEAT_RATE_500MS};
 | 
			
		||||
  float glass_attenuation_factor_{1.0};
 | 
			
		||||
 | 
			
		||||
  uint16_t ps_cooldown_time_s_{5};
 | 
			
		||||
  PsGain501 ps_gain_{PsGain501::PS_GAIN_1};
 | 
			
		||||
  uint16_t ps_threshold_high_{0xffff};
 | 
			
		||||
  uint16_t ps_threshold_low_{0x0000};
 | 
			
		||||
 | 
			
		||||
  //
 | 
			
		||||
  //   Sensors for publishing data
 | 
			
		||||
  //
 | 
			
		||||
  sensor::Sensor *infrared_counts_sensor_{nullptr};          // direct reading CH1, infrared only
 | 
			
		||||
  sensor::Sensor *full_spectrum_counts_sensor_{nullptr};     // direct reading CH0, infrared + visible light
 | 
			
		||||
  sensor::Sensor *ambient_light_sensor_{nullptr};            // calculated lux
 | 
			
		||||
  sensor::Sensor *actual_gain_sensor_{nullptr};              // actual gain of reading
 | 
			
		||||
  sensor::Sensor *actual_integration_time_sensor_{nullptr};  // actual integration time
 | 
			
		||||
  sensor::Sensor *proximity_counts_sensor_{nullptr};         // proximity sensor
 | 
			
		||||
 | 
			
		||||
  bool is_any_als_sensor_enabled_() const {
 | 
			
		||||
    return this->ambient_light_sensor_ != nullptr || this->full_spectrum_counts_sensor_ != nullptr ||
 | 
			
		||||
           this->infrared_counts_sensor_ != nullptr || this->actual_gain_sensor_ != nullptr ||
 | 
			
		||||
           this->actual_integration_time_sensor_ != nullptr;
 | 
			
		||||
  }
 | 
			
		||||
  bool is_any_ps_sensor_enabled_() const { return this->proximity_counts_sensor_ != nullptr; }
 | 
			
		||||
 | 
			
		||||
  //
 | 
			
		||||
  // Trigger section for the automations
 | 
			
		||||
  //
 | 
			
		||||
  friend class LTRPsHighTrigger;
 | 
			
		||||
  friend class LTRPsLowTrigger;
 | 
			
		||||
 | 
			
		||||
  CallbackManager<void()> on_ps_high_trigger_callback_;
 | 
			
		||||
  CallbackManager<void()> on_ps_low_trigger_callback_;
 | 
			
		||||
 | 
			
		||||
  void add_on_ps_high_trigger_callback_(std::function<void()> callback) {
 | 
			
		||||
    this->on_ps_high_trigger_callback_.add(std::move(callback));
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  void add_on_ps_low_trigger_callback_(std::function<void()> callback) {
 | 
			
		||||
    this->on_ps_low_trigger_callback_.add(std::move(callback));
 | 
			
		||||
  }
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class LTRPsHighTrigger : public Trigger<> {
 | 
			
		||||
 public:
 | 
			
		||||
  explicit LTRPsHighTrigger(LTRAlsPs501Component *parent) {
 | 
			
		||||
    parent->add_on_ps_high_trigger_callback_([this]() { this->trigger(); });
 | 
			
		||||
  }
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class LTRPsLowTrigger : public Trigger<> {
 | 
			
		||||
 public:
 | 
			
		||||
  explicit LTRPsLowTrigger(LTRAlsPs501Component *parent) {
 | 
			
		||||
    parent->add_on_ps_low_trigger_callback_([this]() { this->trigger(); });
 | 
			
		||||
  }
 | 
			
		||||
};
 | 
			
		||||
}  // namespace ltr501
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
							
								
								
									
										260
									
								
								esphome/components/ltr501/ltr_definitions_501.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										260
									
								
								esphome/components/ltr501/ltr_definitions_501.h
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,260 @@
 | 
			
		||||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include <cstdint>
 | 
			
		||||
 | 
			
		||||
namespace esphome {
 | 
			
		||||
namespace ltr501 {
 | 
			
		||||
 | 
			
		||||
enum class CommandRegisters : uint8_t {
 | 
			
		||||
  ALS_CONTR = 0x80,         // ALS operation mode control and SW reset
 | 
			
		||||
  PS_CONTR = 0x81,          // PS operation mode control
 | 
			
		||||
  PS_LED = 0x82,            // PS LED pulse frequency control
 | 
			
		||||
  PS_N_PULSES = 0x83,       // PS number of pulses control
 | 
			
		||||
  PS_MEAS_RATE = 0x84,      // PS measurement rate in active mode
 | 
			
		||||
  MEAS_RATE = 0x85,         // ALS measurement rate in active mode
 | 
			
		||||
  PART_ID = 0x86,           // Part Number ID and Revision ID
 | 
			
		||||
  MANUFAC_ID = 0x87,        // Manufacturer ID
 | 
			
		||||
  ALS_DATA_CH1_0 = 0x88,    // ALS measurement CH1 data, lower byte - infrared only
 | 
			
		||||
  ALS_DATA_CH1_1 = 0x89,    // ALS measurement CH1 data, upper byte - infrared only
 | 
			
		||||
  ALS_DATA_CH0_0 = 0x8A,    // ALS measurement CH0 data, lower byte - visible + infrared
 | 
			
		||||
  ALS_DATA_CH0_1 = 0x8B,    // ALS measurement CH0 data, upper byte - visible + infrared
 | 
			
		||||
  ALS_PS_STATUS = 0x8C,     // ALS PS new data status
 | 
			
		||||
  PS_DATA_0 = 0x8D,         // PS measurement data, lower byte
 | 
			
		||||
  PS_DATA_1 = 0x8E,         // PS measurement data, upper byte
 | 
			
		||||
  ALS_PS_INTERRUPT = 0x8F,  // Interrupt status
 | 
			
		||||
  PS_THRES_UP_0 = 0x90,     // PS interrupt upper threshold, lower byte
 | 
			
		||||
  PS_THRES_UP_1 = 0x91,     // PS interrupt upper threshold, upper byte
 | 
			
		||||
  PS_THRES_LOW_0 = 0x92,    // PS interrupt lower threshold, lower byte
 | 
			
		||||
  PS_THRES_LOW_1 = 0x93,    // PS interrupt lower threshold, upper byte
 | 
			
		||||
  PS_OFFSET_1 = 0x94,       // PS offset, upper byte
 | 
			
		||||
  PS_OFFSET_0 = 0x95,       // PS offset, lower byte
 | 
			
		||||
                            // 0x96 - reserved
 | 
			
		||||
  ALS_THRES_UP_0 = 0x97,    // ALS interrupt upper threshold, lower byte
 | 
			
		||||
  ALS_THRES_UP_1 = 0x98,    // ALS interrupt upper threshold, upper byte
 | 
			
		||||
  ALS_THRES_LOW_0 = 0x99,   // ALS interrupt lower threshold, lower byte
 | 
			
		||||
  ALS_THRES_LOW_1 = 0x9A,   // ALS interrupt lower threshold, upper byte
 | 
			
		||||
                            // 0x9B - reserved
 | 
			
		||||
                            // 0x9C - reserved
 | 
			
		||||
                            // 0x9D - reserved
 | 
			
		||||
  INTERRUPT_PERSIST = 0x9E  // Interrupt persistence filter
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// ALS Sensor gain levels
 | 
			
		||||
enum AlsGain501 : uint8_t {
 | 
			
		||||
  GAIN_1 = 0,    // GAIN_RANGE_2 // default
 | 
			
		||||
  GAIN_150 = 1,  // GAIN_RANGE_1
 | 
			
		||||
};
 | 
			
		||||
static const uint8_t GAINS_COUNT = 2;
 | 
			
		||||
 | 
			
		||||
// ALS Sensor integration times
 | 
			
		||||
enum IntegrationTime501 : uint8_t {
 | 
			
		||||
  INTEGRATION_TIME_100MS = 0,  // default
 | 
			
		||||
  INTEGRATION_TIME_50MS = 1,   // only in Dynamic GAIN_RANGE_2
 | 
			
		||||
  INTEGRATION_TIME_200MS = 2,  // only in Dynamic GAIN_RANGE_1
 | 
			
		||||
  INTEGRATION_TIME_400MS = 3,  // only in Dynamic GAIN_RANGE_1
 | 
			
		||||
};
 | 
			
		||||
static const uint8_t TIMES_COUNT = 4;
 | 
			
		||||
 | 
			
		||||
// ALS Sensor measurement repeat rate
 | 
			
		||||
enum MeasurementRepeatRate {
 | 
			
		||||
  REPEAT_RATE_50MS = 0,
 | 
			
		||||
  REPEAT_RATE_100MS = 1,
 | 
			
		||||
  REPEAT_RATE_200MS = 2,
 | 
			
		||||
  REPEAT_RATE_500MS = 3,  // default
 | 
			
		||||
  REPEAT_RATE_1000MS = 4,
 | 
			
		||||
  REPEAT_RATE_2000MS = 5
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// PS Sensor gain levels
 | 
			
		||||
enum PsGain501 : uint8_t {
 | 
			
		||||
  PS_GAIN_1 = 0,  // default
 | 
			
		||||
  PS_GAIN_4 = 1,
 | 
			
		||||
  PS_GAIN_8 = 2,
 | 
			
		||||
  PS_GAIN_16 = 3,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// LED Pulse Modulation Frequency
 | 
			
		||||
enum PsLedFreq : uint8_t {
 | 
			
		||||
  PS_LED_FREQ_30KHZ = 0,
 | 
			
		||||
  PS_LED_FREQ_40KHZ = 1,
 | 
			
		||||
  PS_LED_FREQ_50KHZ = 2,
 | 
			
		||||
  PS_LED_FREQ_60KHZ = 3,  // default
 | 
			
		||||
  PS_LED_FREQ_70KHZ = 4,
 | 
			
		||||
  PS_LED_FREQ_80KHZ = 5,
 | 
			
		||||
  PS_LED_FREQ_90KHZ = 6,
 | 
			
		||||
  PS_LED_FREQ_100KHZ = 7,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// LED current duty
 | 
			
		||||
enum PsLedDuty : uint8_t {
 | 
			
		||||
  PS_LED_DUTY_25 = 0,
 | 
			
		||||
  PS_LED_DUTY_50 = 1,  // default
 | 
			
		||||
  PS_LED_DUTY_75 = 2,
 | 
			
		||||
  PS_LED_DUTY_100 = 3,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// LED pulsed current level
 | 
			
		||||
enum PsLedCurrent : uint8_t {
 | 
			
		||||
  PS_LED_CURRENT_5MA = 0,
 | 
			
		||||
  PS_LED_CURRENT_10MA = 1,
 | 
			
		||||
  PS_LED_CURRENT_20MA = 2,
 | 
			
		||||
  PS_LED_CURRENT_50MA = 3,  // default
 | 
			
		||||
  PS_LED_CURRENT_100MA = 4,
 | 
			
		||||
  PS_LED_CURRENT_100MA1 = 5,
 | 
			
		||||
  PS_LED_CURRENT_100MA2 = 6,
 | 
			
		||||
  PS_LED_CURRENT_100MA3 = 7,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// PS measurement rate
 | 
			
		||||
enum PsMeasurementRate : uint8_t {
 | 
			
		||||
  PS_MEAS_RATE_50MS = 0,
 | 
			
		||||
  PS_MEAS_RATE_70MS = 1,
 | 
			
		||||
  PS_MEAS_RATE_100MS = 2,  // default
 | 
			
		||||
  PS_MEAS_RATE_200MS = 3,
 | 
			
		||||
  PS_MEAS_RATE_500MS = 4,
 | 
			
		||||
  PS_MEAS_RATE_1000MS = 5,
 | 
			
		||||
  PS_MEAS_RATE_2000MS = 6,
 | 
			
		||||
  PS_MEAS_RATE_2000MS1 = 7,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// ALS_CONTR Register (0x80)
 | 
			
		||||
//
 | 
			
		||||
union AlsControlRegister501 {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    bool asl_mode_xxx : 1;
 | 
			
		||||
    bool als_mode_active : 1;
 | 
			
		||||
    bool sw_reset : 1;
 | 
			
		||||
    AlsGain501 gain : 1;
 | 
			
		||||
    uint8_t reserved : 4;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// PS_CONTR Register (0x81)
 | 
			
		||||
//
 | 
			
		||||
union PsControlRegister501 {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    bool ps_mode_xxx : 1;
 | 
			
		||||
    bool ps_mode_active : 1;
 | 
			
		||||
    PsGain501 ps_gain : 2;
 | 
			
		||||
    bool reserved_4 : 1;
 | 
			
		||||
    bool reserved_5 : 1;
 | 
			
		||||
    bool reserved_6 : 1;
 | 
			
		||||
    bool reserved_7 : 1;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// PS_LED Register (0x82)
 | 
			
		||||
//
 | 
			
		||||
union PsLedRegister {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    PsLedCurrent ps_led_current : 3;
 | 
			
		||||
    PsLedDuty ps_led_duty : 2;
 | 
			
		||||
    PsLedFreq ps_led_freq : 3;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// PS_N_PULSES Register (0x83)
 | 
			
		||||
//
 | 
			
		||||
union PsNPulsesRegister501 {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  uint8_t number_of_pulses;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// PS_MEAS_RATE Register (0x84)
 | 
			
		||||
//
 | 
			
		||||
union PsMeasurementRateRegister {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    PsMeasurementRate ps_measurement_rate : 4;
 | 
			
		||||
    uint8_t reserved : 4;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// ALS_MEAS_RATE Register (0x85)
 | 
			
		||||
//
 | 
			
		||||
union MeasurementRateRegister501 {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    MeasurementRepeatRate measurement_repeat_rate : 3;
 | 
			
		||||
    IntegrationTime501 integration_time : 2;
 | 
			
		||||
    bool reserved_5 : 1;
 | 
			
		||||
    bool reserved_6 : 1;
 | 
			
		||||
    bool reserved_7 : 1;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// PART_ID Register (0x86) (Read Only)
 | 
			
		||||
//
 | 
			
		||||
union PartIdRegister {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    uint8_t part_number_id : 4;
 | 
			
		||||
    uint8_t revision_id : 4;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// ALS_PS_STATUS Register (0x8C) (Read Only)
 | 
			
		||||
//
 | 
			
		||||
union AlsPsStatusRegister {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    bool ps_new_data : 1;    // 0 - old data, 1 - new data
 | 
			
		||||
    bool ps_interrupt : 1;   // 0 - interrupt signal not active, 1 - interrupt signal active
 | 
			
		||||
    bool als_new_data : 1;   // 0 - old data, 1 - new data
 | 
			
		||||
    bool als_interrupt : 1;  // 0 - interrupt signal not active, 1 - interrupt signal active
 | 
			
		||||
    AlsGain501 gain : 1;     // current ALS gain
 | 
			
		||||
    bool reserved_5 : 1;
 | 
			
		||||
    bool reserved_6 : 1;
 | 
			
		||||
    bool reserved_7 : 1;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// PS_DATA_1 Register (0x8E) (Read Only)
 | 
			
		||||
//
 | 
			
		||||
union PsData1Register {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    uint8_t ps_data_high : 3;
 | 
			
		||||
    uint8_t reserved : 4;
 | 
			
		||||
    bool ps_saturation_flag : 1;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// INTERRUPT Register (0x8F) (Read Only)
 | 
			
		||||
//
 | 
			
		||||
union InterruptRegister {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    bool ps_interrupt : 1;
 | 
			
		||||
    bool als_interrupt : 1;
 | 
			
		||||
    bool interrupt_polarity : 1;  // 0 - active low (default), 1 - active high
 | 
			
		||||
    uint8_t reserved : 5;
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
//
 | 
			
		||||
// INTERRUPT_PERSIST Register (0x9E)
 | 
			
		||||
//
 | 
			
		||||
union InterruptPersistRegister {
 | 
			
		||||
  uint8_t raw;
 | 
			
		||||
  struct {
 | 
			
		||||
    uint8_t als_persist : 4;  // 0 - every ALS cycle, 1 - every 2 ALS cycles, ... 15 - every 16 ALS cycles
 | 
			
		||||
    uint8_t ps_persist : 4;   // 0 - every PS cycle, 1 - every 2 PS cycles, ... 15 - every 16 PS cycles
 | 
			
		||||
  } __attribute__((packed));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace ltr501
 | 
			
		||||
}  // namespace esphome
 | 
			
		||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user