diff --git a/ports/rp2/machine_pwm.c b/ports/rp2/machine_pwm.c index 1b70746a73..0c840eca94 100644 --- a/ports/rp2/machine_pwm.c +++ b/ports/rp2/machine_pwm.c @@ -100,11 +100,33 @@ STATIC void mp_machine_pwm_deinit(machine_pwm_obj_t *self) { pwm_set_enabled(self->slice, false); } -STATIC mp_obj_t mp_machine_pwm_freq_get(machine_pwm_obj_t *self) { +// Returns: floor((16*F + offset) / div16) +// Avoids overflow in the numerator that would occur if +// 16*F + offset > 2**32 +// F + offset/16 > 2**28 = 268435456 (approximately, due to flooring) +uint32_t get_slice_hz(uint32_t offset, uint32_t div16) { uint32_t source_hz = clock_get_hz(clk_sys); + if (source_hz + offset / 16 > 268000000) { + return (16 * (uint64_t)source_hz + offset) / div16; + } else { + return (16 * source_hz + offset) / div16; + } +} + +// Returns 16*F / denom, rounded. +uint32_t get_slice_hz_round(uint32_t div16) { + return get_slice_hz(div16 / 2, div16); +} + +// Returns ceil(16*F / denom). +uint32_t get_slice_hz_ceil(uint32_t div16) { + return get_slice_hz(div16 - 1, div16); +} + +STATIC mp_obj_t mp_machine_pwm_freq_get(machine_pwm_obj_t *self) { uint32_t div16 = pwm_hw->slice[self->slice].div; uint32_t top = pwm_hw->slice[self->slice].top; - uint32_t pwm_freq = 16 * source_hz / div16 / (top + 1); + uint32_t pwm_freq = get_slice_hz_round(div16 * (top + 1)); return MP_OBJ_NEW_SMALL_INT(pwm_freq); } @@ -114,22 +136,27 @@ STATIC void mp_machine_pwm_freq_set(machine_pwm_obj_t *self, mp_int_t freq) { #define TOP_MAX 65534 uint32_t source_hz = clock_get_hz(clk_sys); uint32_t div16; + uint32_t top; if ((source_hz + freq / 2) / freq < TOP_MAX) { // If possible (based on the formula for TOP below), use a DIV of 1. // This also prevents overflow in the DIV calculation. div16 = 16; + + // Same as get_slice_hz_round() below but canceling the 16s + // to avoid overflow for high freq. + top = (source_hz + freq / 2) / freq - 1; } else { // Otherwise, choose the smallest possible DIV for maximum // duty cycle resolution. // Constraint: 16*F/(div16*freq) < TOP_MAX - // So: div16 = ceil(16*F/(TOP_MAX*freq)) + // So: + div16 = get_slice_hz_ceil(TOP_MAX * freq); - div16 = (16 * source_hz + TOP_MAX * freq - 1) / (TOP_MAX * freq); + // Set TOP as accurately as possible using rounding. + top = get_slice_hz_round(div16 * freq) - 1; } - // Set TOP as accurately as possible using rounding. - uint32_t top = (16 * source_hz + div16 * freq / 2) / (div16 * freq) - 1; if (div16 < 16) { mp_raise_ValueError(MP_ERROR_TEXT("freq too large")); @@ -167,17 +194,15 @@ STATIC void mp_machine_pwm_duty_set_u16(machine_pwm_obj_t *self, mp_int_t duty_u } STATIC mp_obj_t mp_machine_pwm_duty_get_ns(machine_pwm_obj_t *self) { - uint32_t source_hz = clock_get_hz(clk_sys); - uint32_t slice_hz = 16 * source_hz / pwm_hw->slice[self->slice].div; + uint32_t slice_hz = get_slice_hz_round(pwm_hw->slice[self->slice].div); uint32_t cc = pwm_hw->slice[self->slice].cc; cc = (cc >> (self->channel ? PWM_CH0_CC_B_LSB : PWM_CH0_CC_A_LSB)) & 0xffff; - return MP_OBJ_NEW_SMALL_INT((uint64_t)cc * 1000000000ULL / slice_hz); + return MP_OBJ_NEW_SMALL_INT(((uint64_t)cc * 1000000000ULL + slice_hz / 2) / slice_hz); } STATIC void mp_machine_pwm_duty_set_ns(machine_pwm_obj_t *self, mp_int_t duty_ns) { - uint32_t source_hz = clock_get_hz(clk_sys); - uint32_t slice_hz = 16 * source_hz / pwm_hw->slice[self->slice].div; - uint32_t cc = (uint64_t)duty_ns * slice_hz / 1000000000ULL; + uint32_t slice_hz = get_slice_hz_round(pwm_hw->slice[self->slice].div); + uint32_t cc = ((uint64_t)duty_ns * slice_hz + 500000000ULL) / 1000000000ULL; if (cc > 65535) { mp_raise_ValueError(MP_ERROR_TEXT("duty larger than period")); }