diff options
| author | Yuval Adam <_@yuv.al> | 2024-09-23 15:53:46 +0200 |
|---|---|---|
| committer | Yuval Adam <_@yuv.al> | 2024-09-23 15:53:46 +0200 |
| commit | 021f1a48f71b16449ed499099e894d23b7f45a99 (patch) | |
| tree | 4b80ffb6338c0e211b020ef404190e716c8e8d9f /SI1132.py | |
| parent | 356fee40992fbb47bd2813a241033cd8f1b3ee48 (diff) | |
Formatting and cleanup lints
Diffstat (limited to 'SI1132.py')
| -rw-r--r-- | SI1132.py | 199 |
1 files changed, 108 insertions, 91 deletions
@@ -5,78 +5,75 @@ SI1132_I2C_ADDR = 0x60 SI1132_PARTID = 0x32 cmds = { - 'Si1132_PARAM_QUERY' : 0x80, - 'Si1132_PARAM_SET' : 0xA0, - 'Si1132_NOP' : 0x00, - 'Si1132_RESET' : 0x01, - 'Si1132_BUSADDR' : 0x02, - 'Si1132_GET_CAL' : 0x12, - 'Si1132_ALS_FORCE' : 0x06, - 'Si1132_ALS_PAUSE' : 0x0A, - 'Si1132_ALS_AUTO' : 0x0E, + "Si1132_PARAM_QUERY": 0x80, + "Si1132_PARAM_SET": 0xA0, + "Si1132_NOP": 0x00, + "Si1132_RESET": 0x01, + "Si1132_BUSADDR": 0x02, + "Si1132_GET_CAL": 0x12, + "Si1132_ALS_FORCE": 0x06, + "Si1132_ALS_PAUSE": 0x0A, + "Si1132_ALS_AUTO": 0x0E, } params = { - 'Si1132_PARAM_I2CADDR' : 0x00, - 'Si1132_PARAM_CHLIST' : 0x01, - 'Si1132_PARAM_CHLIST_ENUV' : 0x80, - 'Si1132_PARAM_CHLIST_ENAUX' : 0x40, - 'Si1132_PARAM_CHLIST_ENALSIR' : 0x20, - 'Si1132_PARAM_CHLIST_ENALSVIS' : 0x10, - 'Si1132_PARAM_ALSENCODING' : 0x06, - 'Si1132_PARAM_ALSIRADCMUX' : 0x0E, - 'Si1132_PARAM_AUXADCMUX' : 0x0F, - 'Si1132_PARAM_ALSVISADCCOUNTER' : 0x10, - 'Si1132_PARAM_ALSENCODING' : 0x06, - 'Si1132_PARAM_ALSIRADCMUX' : 0x0E, - 'Si1132_PARAM_AUXADCMUX' : 0x0F, - 'Si1132_PARAM_ALSVISADCCOUNTER' : 0x10, - 'Si1132_PARAM_ALSVISADCGAIN' : 0x11, - 'Si1132_PARAM_ALSVISADCMISC' : 0x12, - 'Si1132_PARAM_ALSVISADCMISC_VISRANGE' : 0x20, - 'Si1132_PARAM_ALSIRADCCOUNTER' : 0x1D, - 'Si1132_PARAM_ALSIRADCGAIN' : 0x1E, - 'Si1132_PARAM_ALSIRADCMISC' : 0x1F, - 'Si1132_PARAM_ALSIRADCMISC_RANGE' : 0x20, - 'Si1132_PARAM_ADCCOUNTER_511CLK' : 0x70, - 'Si1132_PARAM_ADCMUX_SMALLIR' : 0x00, - 'Si1132_PARAM_ADCMUX_LARGEIR' : 0x03, + "Si1132_PARAM_I2CADDR": 0x00, + "Si1132_PARAM_CHLIST": 0x01, + "Si1132_PARAM_CHLIST_ENUV": 0x80, + "Si1132_PARAM_CHLIST_ENAUX": 0x40, + "Si1132_PARAM_CHLIST_ENALSIR": 0x20, + "Si1132_PARAM_CHLIST_ENALSVIS": 0x10, + "Si1132_PARAM_ALSENCODING": 0x06, + "Si1132_PARAM_ALSIRADCMUX": 0x0E, + "Si1132_PARAM_AUXADCMUX": 0x0F, + "Si1132_PARAM_ALSVISADCCOUNTER": 0x10, + "Si1132_PARAM_ALSVISADCGAIN": 0x11, + "Si1132_PARAM_ALSVISADCMISC": 0x12, + "Si1132_PARAM_ALSVISADCMISC_VISRANGE": 0x20, + "Si1132_PARAM_ALSIRADCCOUNTER": 0x1D, + "Si1132_PARAM_ALSIRADCGAIN": 0x1E, + "Si1132_PARAM_ALSIRADCMISC": 0x1F, + "Si1132_PARAM_ALSIRADCMISC_RANGE": 0x20, + "Si1132_PARAM_ADCCOUNTER_511CLK": 0x70, + "Si1132_PARAM_ADCMUX_SMALLIR": 0x00, + "Si1132_PARAM_ADCMUX_LARGEIR": 0x03, } regs = { - 'Si1132_REG_PARTID' : 0x00, - 'Si1132_REG_REVID' : 0x01, - 'Si1132_REG_SEQID' : 0x02, - 'Si1132_REG_INTCFG' : 0x03, - 'Si1132_REG_INTCFG_INTOE' : 0x01, - 'Si1132_REG_IRQEN' : 0x04, - 'Si1132_REG_IRQEN_ALSEVERYSAMPLE' : 0x01, - 'Si1132_REG_IRQMODE1' : 0x05, - 'Si1132_REG_IRQMODE2' : 0x06, - 'Si1132_REG_HWKEY' : 0x07, - 'Si1132_REG_MEASRATE0' : 0x08, - 'Si1132_REG_MEASRATE1' : 0x09, - 'Si1132_REG_UCOEF0' : 0x13, - 'Si1132_REG_UCOEF1' : 0x14, - 'Si1132_REG_UCOEF2' : 0x15, - 'Si1132_REG_UCOEF3' : 0x16, - 'Si1132_REG_PARAMWR' : 0x17, - 'Si1132_REG_COMMAND' : 0x18, - 'Si1132_REG_RESPONSE' : 0x20, - 'Si1132_REG_IRQSTAT' : 0x21, - 'Si1132_REG_ALSVISDATA0' : 0x22, - 'Si1132_REG_ALSVISDATA1' : 0x23, - 'Si1132_REG_ALSIRDATA0' : 0x24, - 'Si1132_REG_ALSIRDATA1' : 0x25, - 'Si1132_REG_UVINDEX0' : 0x2C, - 'Si1132_REG_UVINDEX1' : 0x2D, - 'Si1132_REG_PARAMRD' : 0x2E, - 'Si1132_REG_CHIPSTAT' : 0x30, + "Si1132_REG_PARTID": 0x00, + "Si1132_REG_REVID": 0x01, + "Si1132_REG_SEQID": 0x02, + "Si1132_REG_INTCFG": 0x03, + "Si1132_REG_INTCFG_INTOE": 0x01, + "Si1132_REG_IRQEN": 0x04, + "Si1132_REG_IRQEN_ALSEVERYSAMPLE": 0x01, + "Si1132_REG_IRQMODE1": 0x05, + "Si1132_REG_IRQMODE2": 0x06, + "Si1132_REG_HWKEY": 0x07, + "Si1132_REG_MEASRATE0": 0x08, + "Si1132_REG_MEASRATE1": 0x09, + "Si1132_REG_UCOEF0": 0x13, + "Si1132_REG_UCOEF1": 0x14, + "Si1132_REG_UCOEF2": 0x15, + "Si1132_REG_UCOEF3": 0x16, + "Si1132_REG_PARAMWR": 0x17, + "Si1132_REG_COMMAND": 0x18, + "Si1132_REG_RESPONSE": 0x20, + "Si1132_REG_IRQSTAT": 0x21, + "Si1132_REG_ALSVISDATA0": 0x22, + "Si1132_REG_ALSVISDATA1": 0x23, + "Si1132_REG_ALSIRDATA0": 0x24, + "Si1132_REG_ALSIRDATA1": 0x25, + "Si1132_REG_UVINDEX0": 0x2C, + "Si1132_REG_UVINDEX1": 0x2D, + "Si1132_REG_PARAMRD": 0x2E, + "Si1132_REG_CHIPSTAT": 0x30, } + class SI1132: def __init__(self, i2c_dev): - self.bus = smbus.SMBus(int(i2c_dev.split('-')[-1])) + self.bus = smbus.SMBus(int(i2c_dev.split("-")[-1])) self.id = self.read_reg8(0x00) if self.id != SI1132_PARTID: print("Wrong Device ID [", hex(self.id), "]") @@ -84,43 +81,63 @@ class SI1132: self.reset() - self.write_reg8(regs['Si1132_REG_UCOEF0'], 0x7B) - self.write_reg8(regs['Si1132_REG_UCOEF1'], 0x6B) - self.write_reg8(regs['Si1132_REG_UCOEF2'], 0x01) - self.write_reg8(regs['Si1132_REG_UCOEF3'], 0x00) + self.write_reg8(regs["Si1132_REG_UCOEF0"], 0x7B) + self.write_reg8(regs["Si1132_REG_UCOEF1"], 0x6B) + self.write_reg8(regs["Si1132_REG_UCOEF2"], 0x01) + self.write_reg8(regs["Si1132_REG_UCOEF3"], 0x00) - param = params['Si1132_PARAM_CHLIST_ENUV'] | params['Si1132_PARAM_CHLIST_ENALSIR'] | params['Si1132_PARAM_CHLIST_ENALSVIS'] - self.writeParam(params['Si1132_PARAM_CHLIST'], param) + param = ( + params["Si1132_PARAM_CHLIST_ENUV"] + | params["Si1132_PARAM_CHLIST_ENALSIR"] + | params["Si1132_PARAM_CHLIST_ENALSVIS"] + ) + self.writeParam(params["Si1132_PARAM_CHLIST"], param) - self.write_reg8(regs['Si1132_REG_INTCFG'], regs['Si1132_REG_INTCFG_INTOE']) - self.write_reg8(regs['Si1132_REG_IRQEN'], regs['Si1132_REG_IRQEN_ALSEVERYSAMPLE']) + self.write_reg8(regs["Si1132_REG_INTCFG"], regs["Si1132_REG_INTCFG_INTOE"]) + self.write_reg8( + regs["Si1132_REG_IRQEN"], regs["Si1132_REG_IRQEN_ALSEVERYSAMPLE"] + ) - self.writeParam(params['Si1132_PARAM_ALSIRADCMUX'], params['Si1132_PARAM_ADCMUX_SMALLIR']) + self.writeParam( + params["Si1132_PARAM_ALSIRADCMUX"], params["Si1132_PARAM_ADCMUX_SMALLIR"] + ) time.sleep(0.01) - self.writeParam(params['Si1132_PARAM_ALSIRADCGAIN'], 0) + self.writeParam(params["Si1132_PARAM_ALSIRADCGAIN"], 0) time.sleep(0.01) - self.writeParam(params['Si1132_PARAM_ALSIRADCCOUNTER'], params['Si1132_PARAM_ADCCOUNTER_511CLK']) - self.writeParam(params['Si1132_PARAM_ALSIRADCMISC'], params['Si1132_PARAM_ALSIRADCMISC_RANGE']) + self.writeParam( + params["Si1132_PARAM_ALSIRADCCOUNTER"], + params["Si1132_PARAM_ADCCOUNTER_511CLK"], + ) + self.writeParam( + params["Si1132_PARAM_ALSIRADCMISC"], + params["Si1132_PARAM_ALSIRADCMISC_RANGE"], + ) time.sleep(0.01) - self.writeParam(params['Si1132_PARAM_ALSIRADCGAIN'], 0) + self.writeParam(params["Si1132_PARAM_ALSIRADCGAIN"], 0) time.sleep(0.01) - self.writeParam(params['Si1132_PARAM_ALSIRADCCOUNTER'], params['Si1132_PARAM_ADCCOUNTER_511CLK']) - self.writeParam(params['Si1132_PARAM_ALSVISADCMISC'], params['Si1132_PARAM_ALSVISADCMISC_VISRANGE']) + self.writeParam( + params["Si1132_PARAM_ALSIRADCCOUNTER"], + params["Si1132_PARAM_ADCCOUNTER_511CLK"], + ) + self.writeParam( + params["Si1132_PARAM_ALSVISADCMISC"], + params["Si1132_PARAM_ALSVISADCMISC_VISRANGE"], + ) time.sleep(0.01) - self.write_reg8(regs['Si1132_REG_MEASRATE0'], 0xFF) - self.write_reg8(regs['Si1132_REG_COMMAND'], cmds['Si1132_ALS_AUTO']) + self.write_reg8(regs["Si1132_REG_MEASRATE0"], 0xFF) + self.write_reg8(regs["Si1132_REG_COMMAND"], cmds["Si1132_ALS_AUTO"]) def reset(self): - self.write_reg8(regs['Si1132_REG_MEASRATE0'], 0) - self.write_reg8(regs['Si1132_REG_MEASRATE1'], 0) - self.write_reg8(regs['Si1132_REG_IRQEN'], 0) - self.write_reg8(regs['Si1132_REG_IRQMODE1'], 0) - self.write_reg8(regs['Si1132_REG_IRQMODE2'], 0) - self.write_reg8(regs['Si1132_REG_INTCFG'], 0) - self.write_reg8(regs['Si1132_REG_IRQSTAT'], 0xFF) - self.write_reg8(regs['Si1132_REG_COMMAND'], cmds['Si1132_RESET']) + self.write_reg8(regs["Si1132_REG_MEASRATE0"], 0) + self.write_reg8(regs["Si1132_REG_MEASRATE1"], 0) + self.write_reg8(regs["Si1132_REG_IRQEN"], 0) + self.write_reg8(regs["Si1132_REG_IRQMODE1"], 0) + self.write_reg8(regs["Si1132_REG_IRQMODE2"], 0) + self.write_reg8(regs["Si1132_REG_INTCFG"], 0) + self.write_reg8(regs["Si1132_REG_IRQSTAT"], 0xFF) + self.write_reg8(regs["Si1132_REG_COMMAND"], cmds["Si1132_RESET"]) time.sleep(0.01) - self.write_reg8(regs['Si1132_REG_HWKEY'], 0x17) + self.write_reg8(regs["Si1132_REG_HWKEY"], 0x17) time.sleep(0.01) def read_reg8(self, reg): @@ -136,8 +153,8 @@ class SI1132: self.bus.write_word_data(SI1132_I2C_ADDR, reg, val) def writeParam(self, param, val): - self.write_reg8(regs['Si1132_REG_PARAMWR'], val) - self.write_reg8(regs['Si1132_REG_COMMAND'], param | cmds['Si1132_PARAM_SET']) + self.write_reg8(regs["Si1132_REG_PARAMWR"], val) + self.write_reg8(regs["Si1132_REG_COMMAND"], param | cmds["Si1132_PARAM_SET"]) def readVisible(self): time.sleep(0.01) @@ -149,4 +166,4 @@ class SI1132: def readUV(self): time.sleep(0.01) - return self.read_reg16(0x2c) + return self.read_reg16(0x2C) |
