esphome/esphome/components/uln2003/uln2003.cpp

89 lines
2.3 KiB
C++

#include "uln2003.h"
#include "esphome/core/log.h"
namespace esphome {
namespace uln2003 {
static const char *const TAG = "uln2003.stepper";
void ULN2003::setup() {
this->pin_a_->setup();
this->pin_b_->setup();
this->pin_c_->setup();
this->pin_d_->setup();
this->loop();
}
void ULN2003::loop() {
int dir = this->should_step_();
if (dir == 0 && this->has_reached_target()) {
this->high_freq_.stop();
if (this->sleep_when_done_) {
this->pin_a_->digital_write(false);
this->pin_b_->digital_write(false);
this->pin_c_->digital_write(false);
this->pin_d_->digital_write(false);
// do not write pos
return;
}
} else {
this->high_freq_.start();
this->current_uln_pos_ += dir;
}
this->write_step_(this->current_uln_pos_);
}
void ULN2003::dump_config() {
ESP_LOGCONFIG(TAG, "ULN2003:");
LOG_PIN(" Pin A: ", this->pin_a_);
LOG_PIN(" Pin B: ", this->pin_b_);
LOG_PIN(" Pin C: ", this->pin_c_);
LOG_PIN(" Pin D: ", this->pin_d_);
ESP_LOGCONFIG(TAG, " Sleep when done: %s", YESNO(this->sleep_when_done_));
const char *step_mode_s = "";
switch (this->step_mode_) {
case ULN2003_STEP_MODE_FULL_STEP:
step_mode_s = "FULL STEP";
break;
case ULN2003_STEP_MODE_HALF_STEP:
step_mode_s = "HALF STEP";
break;
case ULN2003_STEP_MODE_WAVE_DRIVE:
step_mode_s = "WAVE DRIVE";
break;
}
ESP_LOGCONFIG(TAG, " Step Mode: %s", step_mode_s);
}
void ULN2003::write_step_(int32_t step) {
int32_t n = this->step_mode_ == ULN2003_STEP_MODE_HALF_STEP ? 8 : 4;
auto i = static_cast<uint32_t>((step % n + n) % n);
uint8_t res = 0;
switch (this->step_mode_) {
case ULN2003_STEP_MODE_FULL_STEP: {
// AB, BC, CD, DA
res |= 1 << i;
res |= 1 << ((i + 1) % 4);
break;
}
case ULN2003_STEP_MODE_HALF_STEP: {
// A, AB, B, BC, C, CD, D, DA
res |= 1 << (i >> 1);
res |= 1 << (((i + 1) >> 1) & 0x3);
break;
}
case ULN2003_STEP_MODE_WAVE_DRIVE: {
// A, B, C, D
res |= 1 << i;
break;
}
}
this->pin_a_->digital_write((res >> 0) & 1);
this->pin_b_->digital_write((res >> 1) & 1);
this->pin_c_->digital_write((res >> 2) & 1);
this->pin_d_->digital_write((res >> 3) & 1);
}
} // namespace uln2003
} // namespace esphome