mirror of
https://github.com/esphome/esphome.git
synced 2025-03-20 01:28:16 +00:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import spi, canbus
|
|
from esphome.const import CONF_ID, CONF_MODE
|
|
from esphome.components.canbus import CanbusComponent
|
|
|
|
AUTO_LOAD = ['canbus']
|
|
DEPENDENCIES = ['spi']
|
|
|
|
CONF_MCP_CLOCK = 'clock'
|
|
|
|
mcp2515_ns = cg.esphome_ns.namespace('mcp2515')
|
|
mcp2515 = mcp2515_ns.class_('MCP2515', CanbusComponent, spi.SPIDevice)
|
|
CanClock = mcp2515_ns.enum('CAN_CLOCK')
|
|
McpMode = mcp2515_ns.enum('CANCTRL_REQOP_MODE')
|
|
|
|
CAN_CLOCK = {
|
|
'20MHZ': CanClock.MCP_20MHZ,
|
|
'16MHZ': CanClock.MCP_16MHZ,
|
|
'8MHZ': CanClock.MCP_8MHZ,
|
|
}
|
|
|
|
MCP_MODE = {
|
|
'NORMAL': McpMode.CANCTRL_REQOP_NORMAL,
|
|
'LOOPBACK': McpMode.CANCTRL_REQOP_LOOPBACK,
|
|
'LISTENONLY': McpMode.CANCTRL_REQOP_LISTENONLY,
|
|
}
|
|
|
|
CONFIG_SCHEMA = canbus.CONFIG_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_id(mcp2515),
|
|
cv.Optional(CONF_MCP_CLOCK, default='8MHZ'): cv.enum(CAN_CLOCK, upper=True),
|
|
cv.Optional(CONF_MODE, default='NORMAL'): cv.enum(MCP_MODE, upper=True),
|
|
}).extend(spi.SPI_DEVICE_SCHEMA)
|
|
|
|
|
|
def to_code(config):
|
|
rhs = mcp2515.new()
|
|
var = cg.Pvariable(config[CONF_ID], rhs)
|
|
yield canbus.register_canbus(var, config)
|
|
if CONF_MCP_CLOCK in config:
|
|
canclock = CAN_CLOCK[config[CONF_MCP_CLOCK]]
|
|
cg.add(var.set_mcp_clock(canclock))
|
|
if CONF_MODE in config:
|
|
mode = MCP_MODE[config[CONF_MODE]]
|
|
cg.add(var.set_mcp_mode(mode))
|
|
|
|
yield spi.register_spi_device(var, config)
|