| """Sterling Baird: wrapper class for AS7341 sensor.""" |
|
|
| from math import log |
|
|
| from as7341 import AS7341, AS7341_MODE_SPM |
| from machine import I2C, Pin |
|
|
|
|
| class ExternalDeviceNotFound(OSError): |
| pass |
|
|
|
|
| class Sensor: |
| def __init__( |
| self, atime=100, astep=999, gain=8, i2c=I2C(1, scl=Pin(27), sda=Pin(26)) |
| ): |
| """Wrapper for Rob Hamerling's AS7341 implementation. |
| |
| Mimics the original CircuitPython class a bit more, specific to the needs of |
| SDL-Demo. |
| |
| Rob Hamerling's implementation: |
| - https://gitlab.com/robhamerling/micropython-as7341 |
| |
| Original Circuit Python repo: |
| - https://github.com/adafruit/Adafruit_CircuitPython_AS7341 |
| |
| Parameters |
| ---------- |
| atime : int, optional |
| The integration time step size in 2.78 microsecond increments, by default 100 |
| astep : int, optional |
| The integration time step count. Total integration time will be (ATIME + 1) |
| * (ASTEP + 1) * 2.78µS, by default 999, meaning 281 ms assuming atime=100 |
| gain : int, optional |
| The ADC gain multiplier, by default 128 |
| i2c : I2C, optional |
| The I2C bus, by default machine.I2C(1, scl=machine.Pin(27), |
| sda=machine.Pin(26)) |
| |
| Raises |
| ------ |
| ExternalDeviceNotFound |
| Couldn't connect to AS7341. |
| |
| Examples |
| -------- |
| >>> sensor = Sensor(atime=29, astep=599, again=4) |
| >>> channel_data = sensor.all_channels |
| """ |
|
|
| |
| self.i2c = i2c |
| addrlist = " ".join(["0x{:02X}".format(x) for x in i2c.scan()]) |
| print("Detected devices at I2C-addresses:", addrlist) |
|
|
| sensor = AS7341(i2c) |
|
|
| if not sensor.isconnected(): |
| raise ExternalDeviceNotFound("Failed to contact AS7341, terminating") |
|
|
| sensor.set_measure_mode(AS7341_MODE_SPM) |
|
|
| sensor.set_atime(atime) |
| sensor.set_astep(astep) |
| sensor.set_again(gain) |
|
|
| self.sensor = sensor |
|
|
| self.__atime = atime |
| self.__astep = astep |
| self.__gain = gain |
|
|
| @property |
| def _atime(self): |
| return self.__atime |
|
|
| @_atime.setter |
| def _atime(self, value): |
| self.__atime = value |
| self.sensor.set_atime(value) |
|
|
| @property |
| def _astep(self): |
| return self.__astep |
|
|
| @_astep.setter |
| def _astep(self, value): |
| self.__atime = value |
| self.sensor.set_astep(value) |
|
|
| @property |
| def _gain(self): |
| return self.__gain |
|
|
| @_gain.setter |
| def _gain(self, gain): |
| """set AGAIN (code in range 0..10 -> gain factor 0.5 .. 512) |
| gain: *0.5 | *1 | *2 | *4 | *8 | *16 | *32 | *64 | *128 | *256 | *512 |
| code 0 1 2 3 4 5 6 7 8 9 10 |
| """ |
| self.__gain = gain |
| |
| code = int(round(1.4427 * log(2 * gain))) |
| self.sensor.set_again(code) |
|
|
| @property |
| def all_channels(self): |
| self.sensor.start_measure("F1F4CN") |
| f1, f2, f3, f4, clr, nir = self.sensor.get_spectral_data() |
|
|
| self.sensor.start_measure("F5F8CN") |
| f5, f6, f7, f8, clr, nir = self.sensor.get_spectral_data() |
|
|
| clr, nir |
|
|
| return [f1, f2, f3, f4, f5, f6, f7, f8] |
|
|
| @property |
| def all_channels_clr_nir(self): |
| self.sensor.start_measure("F1F4CN") |
| f1, f2, f3, f4, clr, nir = self.sensor.get_spectral_data() |
|
|
| self.sensor.start_measure("F5F8CN") |
| f5, f6, f7, f8, clr, nir = self.sensor.get_spectral_data() |
|
|
| clr, nir |
|
|
| return [f1, f2, f3, f4, f5, f6, f7, f8, clr, nir] |
|
|
| def disable(self): |
| self.sensor.disable() |
|
|
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
