diff --git a/src/drivers/imu/invensense/icm40609d/ICM40609D.cpp b/src/drivers/imu/invensense/icm40609d/ICM40609D.cpp index b43748f295..6c63872abc 100644 --- a/src/drivers/imu/invensense/icm40609d/ICM40609D.cpp +++ b/src/drivers/imu/invensense/icm40609d/ICM40609D.cpp @@ -40,9 +40,11 @@ static constexpr int16_t combine(uint8_t msb, uint8_t lsb) return (msb << 8u) | lsb; } -ICM40609D::ICM40609D(int bus, uint32_t device, enum Rotation rotation) : - SPI(MODULE_NAME, nullptr, bus, device, SPIDEV_MODE3, SPI_SPEED), - ScheduledWorkItem(MODULE_NAME, px4::device_bus_to_wq(get_device_id())), +ICM40609D::ICM40609D(I2CSPIBusOption bus_option, int bus, uint32_t device, enum Rotation rotation, int bus_frequency, + spi_mode_e spi_mode, spi_drdy_gpio_t drdy_gpio) : + SPI(MODULE_NAME, nullptr, bus, device, spi_mode, bus_frequency), + I2CSPIDriver(MODULE_NAME, px4::device_bus_to_wq(get_device_id()), bus_option, bus), + _drdy_gpio(drdy_gpio), _px4_accel(get_device_id(), ORB_PRIO_VERY_HIGH, rotation), _px4_gyro(get_device_id(), ORB_PRIO_VERY_HIGH, rotation) { @@ -56,8 +58,6 @@ ICM40609D::ICM40609D(int bus, uint32_t device, enum Rotation rotation) : ICM40609D::~ICM40609D() { - Stop(); - perf_free(_transfer_perf); perf_free(_bad_register_perf); perf_free(_bad_transfer_perf); @@ -67,36 +67,35 @@ ICM40609D::~ICM40609D() perf_free(_drdy_interval_perf); } -bool ICM40609D::Init() +int ICM40609D::init() { - if (SPI::init() != PX4_OK) { - PX4_ERR("SPI::init failed"); - return false; + int ret = SPI::init(); + + if (ret != PX4_OK) { + DEVICE_DEBUG("SPI::init failed (%i)", ret); + return ret; } - return Reset(); -} - -void ICM40609D::Stop() -{ - // wait until stopped - while (_state.load() != STATE::STOPPED) { - _state.store(STATE::REQUEST_STOP); - ScheduleNow(); - px4_usleep(10); - } + return Reset() ? 0 : -1; } bool ICM40609D::Reset() { - _state.store(STATE::RESET); + _state = STATE::RESET; ScheduleClear(); ScheduleNow(); return true; } -void ICM40609D::PrintInfo() +void ICM40609D::exit_and_cleanup() { + DataReadyInterruptDisable(); + I2CSPIDriverBase::exit_and_cleanup(); +} + +void ICM40609D::print_status() +{ + I2CSPIDriverBase::print_status(); PX4_INFO("FIFO empty interval: %d us (%.3f Hz)", _fifo_empty_interval_us, static_cast(1000000 / _fifo_empty_interval_us)); @@ -117,35 +116,35 @@ int ICM40609D::probe() const uint8_t whoami = RegisterRead(Register::BANK_0::WHO_AM_I); if (whoami != WHOAMI) { - PX4_WARN("unexpected WHO_AM_I 0x%02x", whoami); + DEVICE_DEBUG("unexpected WHO_AM_I 0x%02x", whoami); return PX4_ERROR; } return PX4_OK; } -void ICM40609D::Run() +void ICM40609D::RunImpl() { - switch (_state.load()) { + switch (_state) { case STATE::RESET: // DEVICE_CONFIG: Software reset RegisterWrite(Register::BANK_0::DEVICE_CONFIG, DEVICE_CONFIG_BIT::SOFT_RESET_CONFIG); _reset_timestamp = hrt_absolute_time(); - _state.store(STATE::WAIT_FOR_RESET); + _state = STATE::WAIT_FOR_RESET; ScheduleDelayed(1_ms); // wait 1 ms for soft reset to be effective break; case STATE::WAIT_FOR_RESET: if ((RegisterRead(Register::BANK_0::WHO_AM_I) == WHOAMI)) { // if reset succeeded then configure - _state.store(STATE::CONFIGURE); + _state = STATE::CONFIGURE; ScheduleNow(); } else { // RESET not complete if (hrt_elapsed_time(&_reset_timestamp) > 10_ms) { PX4_ERR("Reset failed, retrying"); - _state.store(STATE::RESET); + _state = STATE::RESET; ScheduleDelayed(10_ms); } else { @@ -159,7 +158,7 @@ void ICM40609D::Run() case STATE::CONFIGURE: if (Configure()) { // if configure succeeded then start reading from FIFO - _state.store(STATE::FIFO_READ); + _state = STATE::FIFO_READ; if (DataReadyInterruptConfigure()) { _data_ready_interrupt_enabled = true; @@ -237,7 +236,7 @@ void ICM40609D::Run() } else { // register check failed, force reconfigure PX4_DEBUG("Health check failed, reconfiguring"); - _state.store(STATE::CONFIGURE); + _state = STATE::CONFIGURE; ScheduleNow(); } @@ -251,16 +250,6 @@ void ICM40609D::Run() } break; - - case STATE::REQUEST_STOP: - DataReadyInterruptDisable(); - ScheduleClear(); - _state.store(STATE::STOPPED); - break; - - case STATE::STOPPED: - // DO NOTHING - break; } } @@ -391,29 +380,21 @@ void ICM40609D::DataReady() bool ICM40609D::DataReadyInterruptConfigure() { - int ret_setevent = -1; + if (_drdy_gpio == 0) { + return false; + } // Setup data ready on falling edge - // TODO: cleanup horrible DRDY define mess -#if defined(GPIO_SPI1_DRDY1_ICM40609D) - ret_setevent = px4_arch_gpiosetevent(GPIO_SPI1_DRDY1_ICM40609D, false, true, true, - &ICM40609D::DataReadyInterruptCallback, this); -#endif - - return (ret_setevent == 0); + return px4_arch_gpiosetevent(_drdy_gpio, false, true, true, &ICM40609D::DataReadyInterruptCallback, this) == 0; } bool ICM40609D::DataReadyInterruptDisable() { - int ret_setevent = -1; + if (_drdy_gpio == 0) { + return false; + } - // Disable data ready callback - // TODO: cleanup horrible DRDY define mess -#if defined(GPIO_SPI1_DRDY1_ICM40609D) - ret_setevent = px4_arch_gpiosetevent(GPIO_SPI1_DRDY1_ICM40609D, false, false, false, nullptr, nullptr); -#endif - - return (ret_setevent == 0); + return px4_arch_gpiosetevent(_drdy_gpio, false, false, false, nullptr, nullptr) == 0; } bool ICM40609D::RegisterCheck(const register_bank0_config_t ®_cfg, bool notify) diff --git a/src/drivers/imu/invensense/icm40609d/ICM40609D.hpp b/src/drivers/imu/invensense/icm40609d/ICM40609D.hpp index fca4d01e1f..074005dc0b 100644 --- a/src/drivers/imu/invensense/icm40609d/ICM40609D.hpp +++ b/src/drivers/imu/invensense/icm40609d/ICM40609D.hpp @@ -49,24 +49,35 @@ #include #include #include -#include +#include using namespace InvenSense_ICM40609D; -class ICM40609D : public device::SPI, public px4::ScheduledWorkItem +class ICM40609D : public device::SPI, public I2CSPIDriver { public: - ICM40609D(int bus, uint32_t device, enum Rotation rotation = ROTATION_NONE); + ICM40609D(I2CSPIBusOption bus_option, int bus, uint32_t device, enum Rotation rotation, int bus_frequency, + spi_mode_e spi_mode, spi_drdy_gpio_t drdy_gpio); ~ICM40609D() override; - bool Init(); - void Start(); - void Stop(); - bool Reset(); - void PrintInfo(); + static I2CSPIDriverBase *instantiate(const BusCLIArguments &cli, const BusInstanceIterator &iterator, + int runtime_instance); + static void print_usage(); + void RunImpl(); + + int init() override; + void print_status() override; + + void Start(); + bool Reset(); + +protected: + void custom_method(const BusCLIArguments &cli) override; + void exit_and_cleanup() override; private: + // Sensor Configuration static constexpr uint32_t GYRO_RATE{8000}; // 8 kHz gyro static constexpr uint32_t ACCEL_RATE{8000}; // 8 kHz accel static constexpr uint32_t FIFO_MAX_SAMPLES{ math::min(FIFO::SIZE / sizeof(FIFO::DATA) + 1, sizeof(PX4Gyroscope::FIFOSample::x) / sizeof(PX4Gyroscope::FIFOSample::x[0]))}; @@ -90,8 +101,6 @@ private: int probe() override; - void Run() override; - bool Configure(); void ConfigureAccel(); void ConfigureGyro(); @@ -117,6 +126,8 @@ private: void ProcessGyro(const hrt_abstime ×tamp_sample, const FIFOTransferBuffer &buffer, uint8_t samples); void UpdateTemperature(); + const spi_drdy_gpio_t _drdy_gpio; + PX4Accelerometer _px4_accel; PX4Gyroscope _px4_gyro; @@ -141,11 +152,9 @@ private: WAIT_FOR_RESET, CONFIGURE, FIFO_READ, - REQUEST_STOP, - STOPPED, }; - px4::atomic _state{STATE::RESET}; + STATE _state{STATE::RESET}; uint16_t _fifo_empty_interval_us{500}; // default 500 us / 2000 Hz transfer interval uint8_t _fifo_gyro_samples{static_cast(_fifo_empty_interval_us / (1000000 / GYRO_RATE))}; diff --git a/src/drivers/imu/invensense/icm40609d/icm40609d_main.cpp b/src/drivers/imu/invensense/icm40609d/icm40609d_main.cpp index 4d946d0fef..aab912c496 100644 --- a/src/drivers/imu/invensense/icm40609d/icm40609d_main.cpp +++ b/src/drivers/imu/invensense/icm40609d/icm40609d_main.cpp @@ -34,118 +34,83 @@ #include "ICM40609D.hpp" #include +#include -namespace icm40609d +void ICM40609D::print_usage() { -ICM40609D *g_dev{nullptr}; - -static int start(enum Rotation rotation) -{ - if (g_dev != nullptr) { - PX4_WARN("already started"); - return 0; - } - - // create the driver -#if defined(PX4_SPI_BUS_SENSORS) && defined(PX4_SPIDEV_ICM_40609D) - g_dev = new ICM40609D(PX4_SPI_BUS_SENSORS, PX4_SPIDEV_ICM_40609D, rotation); -#endif - - if (g_dev == nullptr) { - PX4_ERR("driver start failed"); - return -1; - } - - if (!g_dev->Init()) { - PX4_ERR("driver init failed"); - delete g_dev; - g_dev = nullptr; - return -1; - } - - return 0; + PRINT_MODULE_USAGE_NAME("icm40609d", "driver"); + PRINT_MODULE_USAGE_SUBCATEGORY("imu"); + PRINT_MODULE_USAGE_COMMAND("start"); + PRINT_MODULE_USAGE_PARAMS_I2C_SPI_DRIVER(false, true); + PRINT_MODULE_USAGE_PARAM_INT('R', 0, 0, 35, "Rotation", true); + PRINT_MODULE_USAGE_COMMAND("reset"); + PRINT_MODULE_USAGE_DEFAULT_COMMANDS(); } -static int stop() +I2CSPIDriverBase *ICM40609D::instantiate(const BusCLIArguments &cli, const BusInstanceIterator &iterator, + int runtime_instance) { - if (g_dev == nullptr) { - PX4_WARN("driver not running"); - return -1; + ICM40609D *instance = new ICM40609D(iterator.configuredBusOption(), iterator.bus(), iterator.devid(), cli.rotation, + cli.bus_frequency, cli.spi_mode, iterator.DRDYGPIO()); + + if (!instance) { + PX4_ERR("alloc failed"); + return nullptr; } - g_dev->Stop(); - delete g_dev; - g_dev = nullptr; - - return 0; -} - -static int reset() -{ - if (g_dev == nullptr) { - PX4_WARN("driver not running"); - return 0; + if (OK != instance->init()) { + delete instance; + return nullptr; } - return g_dev->Reset(); + return instance; } -static int status() +void ICM40609D::custom_method(const BusCLIArguments &cli) { - if (g_dev == nullptr) { - PX4_INFO("driver not running"); - return 0; - } - - g_dev->PrintInfo(); - - return 0; + Reset(); } -static int usage() -{ - PX4_INFO("missing command: try 'start', 'stop', 'reset', 'status'"); - PX4_INFO("options:"); - PX4_INFO(" -R rotation"); - - return 0; -} - -} // namespace icm40609d - extern "C" int icm40609d_main(int argc, char *argv[]) { - enum Rotation rotation = ROTATION_NONE; - int myoptind = 1; - int ch = 0; - const char *myoptarg = nullptr; + int ch; + using ThisDriver = ICM40609D; + BusCLIArguments cli{false, true}; + cli.default_spi_frequency = SPI_SPEED; - // start options - while ((ch = px4_getopt(argc, argv, "R:", &myoptind, &myoptarg)) != EOF) { + while ((ch = cli.getopt(argc, argv, "R:")) != EOF) { switch (ch) { case 'R': - rotation = (enum Rotation)atoi(myoptarg); + cli.rotation = (enum Rotation)atoi(cli.optarg()); break; - - default: - return icm40609d::usage(); } } - const char *verb = argv[myoptind]; + const char *verb = cli.optarg(); - if (!strcmp(verb, "start")) { - return icm40609d::start(rotation); - - } else if (!strcmp(verb, "stop")) { - return icm40609d::stop(); - - } else if (!strcmp(verb, "status")) { - return icm40609d::status(); - - } else if (!strcmp(verb, "reset")) { - return icm40609d::reset(); + if (!verb) { + ThisDriver::print_usage(); + return -1; } - return icm40609d::usage(); + BusInstanceIterator iterator(MODULE_NAME, cli, DRV_IMU_DEVTYPE_ICM40609D); + + if (!strcmp(verb, "start")) { + return ThisDriver::module_start(cli, iterator); + } + + if (!strcmp(verb, "stop")) { + return ThisDriver::module_stop(iterator); + } + + if (!strcmp(verb, "status")) { + return ThisDriver::module_status(iterator); + } + + if (!strcmp(verb, "reset")) { + return ThisDriver::module_custom_method(cli, iterator); + } + + ThisDriver::print_usage(); + return -1; }