Adolfo Reyna
145 lines
4.1 KiB
Python
145 lines
4.1 KiB
Python
"""CircuitPython FT6336U touch controller driver.
|
|
|
|
Ports the MicroPython ft6336u.py driver using busio.I2C and digitalio.DigitalInOut.
|
|
"""
|
|
|
|
import time
|
|
|
|
class FT6336U:
|
|
ADDR = 0x38
|
|
|
|
# Registers
|
|
REG_DEV_MODE = 0x00
|
|
REG_TD_STATUS = 0x02
|
|
REG_P1_XH = 0x03
|
|
REG_P1_XL = 0x04
|
|
REG_P1_YH = 0x05
|
|
REG_P1_YL = 0x06
|
|
REG_CTRL = 0x86
|
|
REG_CHIPID = 0xA3
|
|
REG_G_MODE = 0xA4
|
|
|
|
# Modes
|
|
CTRL_KEEP_ACTIVE = 0x00
|
|
G_MODE_TRIGGER = 0x01
|
|
|
|
def __init__(self, i2c, rst_pin, int_pin, width=480, height=320, swap_xy=True, invert_x=True, invert_y=False):
|
|
self.i2c = i2c
|
|
self.rst = rst_pin
|
|
self.int = int_pin
|
|
self.width = width
|
|
self.height = height
|
|
self.swap_xy = swap_xy
|
|
self.invert_x = invert_x
|
|
self.invert_y = invert_y
|
|
self.initialized = False
|
|
|
|
# Configure reset and interrupt pins
|
|
self.rst.switch_to_output(value=True)
|
|
self.int.switch_to_input(pull=None) # Typically pulled up externally or on-board
|
|
|
|
self.reset()
|
|
self.init_chip()
|
|
|
|
def reset(self):
|
|
self.rst.value = False
|
|
time.sleep(0.010)
|
|
self.rst.value = True
|
|
time.sleep(0.300) # Wait for chip to wake up
|
|
|
|
def read_reg(self, reg, n=1):
|
|
while not self.i2c.try_lock():
|
|
pass
|
|
try:
|
|
self.i2c.writeto(self.ADDR, bytes([reg]))
|
|
buf = bytearray(n)
|
|
self.i2c.readfrom_into(self.ADDR, buf)
|
|
return buf
|
|
except Exception as e:
|
|
print(f"I2C read failed at reg 0x{reg:02X}: {e}")
|
|
return None
|
|
finally:
|
|
self.i2c.unlock()
|
|
|
|
def write_reg(self, reg, val):
|
|
while not self.i2c.try_lock():
|
|
pass
|
|
try:
|
|
self.i2c.writeto(self.ADDR, bytes([reg, val]))
|
|
return True
|
|
except Exception as e:
|
|
print(f"I2C write failed at reg 0x{reg:02X}: {e}")
|
|
return False
|
|
finally:
|
|
self.i2c.unlock()
|
|
|
|
def init_chip(self):
|
|
# 1. Read Chip ID
|
|
chip_id = self.read_reg(self.REG_CHIPID)
|
|
if chip_id is None or chip_id[0] != 0x64:
|
|
time.sleep(0.100)
|
|
chip_id = self.read_reg(self.REG_CHIPID)
|
|
if chip_id is None or chip_id[0] != 0x64:
|
|
print(f"FT6336U error: Invalid Chip ID (got {chip_id[0] if chip_id else None}, expected 0x64)")
|
|
self.initialized = False
|
|
return False
|
|
|
|
print(f"FT6336U touch controller detected (Chip ID: 0x{chip_id[0]:02X})")
|
|
|
|
# 2. Configure operating mode (0 = Normal Mode)
|
|
self.write_reg(self.REG_DEV_MODE, 0x00)
|
|
|
|
# 3. Configure CTRL mode (0 = Keep Active)
|
|
self.write_reg(self.REG_CTRL, self.CTRL_KEEP_ACTIVE)
|
|
|
|
self.initialized = True
|
|
return True
|
|
|
|
def is_touched(self):
|
|
"""Returns True if screen is touched by checking the TD_STATUS register."""
|
|
if not self.initialized:
|
|
return False
|
|
td_status = self.read_reg(self.REG_TD_STATUS)
|
|
if td_status is None:
|
|
return False
|
|
touch_count = td_status[0] & 0x0F
|
|
if 0 < touch_count < 3:
|
|
# Read P1 coordinates to clear register/interrupt state on the chip
|
|
self.read_reg(self.REG_P1_XH, 6)
|
|
return True
|
|
return False
|
|
|
|
def read_touch(self):
|
|
"""Reads touch point coordinates."""
|
|
if not self.initialized:
|
|
return None
|
|
|
|
td_status = self.read_reg(self.REG_TD_STATUS)
|
|
if td_status is None:
|
|
return None
|
|
|
|
touch_count = td_status[0] & 0x0F
|
|
if touch_count == 0:
|
|
return None
|
|
|
|
buf = self.read_reg(self.REG_P1_XH, 6)
|
|
if buf is None or len(buf) < 6:
|
|
return None
|
|
|
|
raw_x = ((buf[0] & 0x0F) << 8) | buf[1]
|
|
raw_y = ((buf[2] & 0x0F) << 8) | buf[3]
|
|
|
|
if self.swap_xy:
|
|
raw_x, raw_y = raw_y, raw_x
|
|
|
|
if self.invert_x:
|
|
raw_x = self.width - 1 - raw_x
|
|
|
|
if self.invert_y:
|
|
raw_y = self.height - 1 - raw_y
|
|
|
|
x = max(0, min(self.width - 1, raw_x))
|
|
y = max(0, min(self.height - 1, raw_y))
|
|
|
|
return (x, y)
|