adc.c 32.2 KB
Newer Older
1
/*
2
 * This file is part of the MicroPython project, http://micropython.org/
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2013, 2014 Damien P. George
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

D
Dave Hylands 已提交
27 28 29
#include <stdio.h>
#include <string.h>

30 31
#include "py/runtime.h"
#include "py/binary.h"
32
#include "py/mphal.h"
D
Dave Hylands 已提交
33 34
#include "adc.h"
#include "pin.h"
35
#include "timer.h"
D
Dave Hylands 已提交
36

37 38
#if MICROPY_HW_ENABLE_ADC

39 40 41 42 43 44 45 46 47 48 49 50 51
/// \moduleref pyb
/// \class ADC - analog to digital conversion: read analog values on a pin
///
/// Usage:
///
///     adc = pyb.ADC(pin)              # create an analog object from a pin
///     val = adc.read()                # read an analog value
///
///     adc = pyb.ADCAll(resolution)    # creale an ADCAll object
///     val = adc.read_channel(channel) # read the given channel
///     val = adc.read_core_temp()      # read MCU temperature
///     val = adc.read_core_vbat()      # read MCU VBAT
///     val = adc.read_core_vref()      # read MCU VREF
D
Dave Hylands 已提交
52 53 54

/* ADC defintions */
#define ADCx                    (ADC1)
55 56 57
#define PIN_ADC_MASK            PIN_ADC1
#define pin_adc_table           pin_adc1

58 59 60 61 62 63 64 65 66 67 68
#if defined(STM32H7)
// On the H7 ADC3 is used for ADCAll to be able to read internal
// channels. For all other GPIO channels, ADC12 is used instead.
#define ADCALLx                 (ADC3)
#define pin_adcall_table        pin_adc3
#else
// Use ADC1 for ADCAll instance by default for all other MCUs.
#define ADCALLx                 (ADC1)
#define pin_adcall_table        pin_adc1
#endif

69
#define ADCx_CLK_ENABLE         __HAL_RCC_ADC1_CLK_ENABLE
70

71 72 73 74
#if defined(STM32F0)

#define ADC_FIRST_GPIO_CHANNEL  (0)
#define ADC_LAST_GPIO_CHANNEL   (15)
75
#define ADC_SCALE_V             (3.3f)
76
#define ADC_CAL_ADDRESS         (0x1ffff7ba)
77 78
#define ADC_CAL1                ((uint16_t *)0x1ffff7b8)
#define ADC_CAL2                ((uint16_t *)0x1ffff7c2)
79
#define ADC_CAL_BITS            (12)
80 81

#elif defined(STM32F4)
82 83 84

#define ADC_FIRST_GPIO_CHANNEL  (0)
#define ADC_LAST_GPIO_CHANNEL   (15)
85
#define ADC_SCALE_V             (3.3f)
86
#define ADC_CAL_ADDRESS         (0x1fff7a2a)
87 88
#define ADC_CAL1                ((uint16_t *)(ADC_CAL_ADDRESS + 2))
#define ADC_CAL2                ((uint16_t *)(ADC_CAL_ADDRESS + 4))
89
#define ADC_CAL_BITS            (12)
90

91
#elif defined(STM32F7)
92

93 94
#define ADC_FIRST_GPIO_CHANNEL  (0)
#define ADC_LAST_GPIO_CHANNEL   (15)
95
#define ADC_SCALE_V             (3.3f)
96 97 98 99
#if defined(STM32F722xx) || defined(STM32F723xx) || \
    defined(STM32F732xx) || defined(STM32F733xx)
#define ADC_CAL_ADDRESS         (0x1ff07a2a)
#else
100
#define ADC_CAL_ADDRESS         (0x1ff0f44a)
101 102
#endif

103 104
#define ADC_CAL1                ((uint16_t *)(ADC_CAL_ADDRESS + 2))
#define ADC_CAL2                ((uint16_t *)(ADC_CAL_ADDRESS + 4))
105
#define ADC_CAL_BITS            (12)
106

107 108 109 110
#elif defined(STM32H7)

#define ADC_FIRST_GPIO_CHANNEL  (0)
#define ADC_LAST_GPIO_CHANNEL   (16)
111
#define ADC_SCALE_V             (3.3f)
112
#define ADC_CAL_ADDRESS         (0x1FF1E860)
113 114
#define ADC_CAL1                ((uint16_t *)(0x1FF1E820))
#define ADC_CAL2                ((uint16_t *)(0x1FF1E840))
115
#define ADC_CAL_BITS            (16)
116

117
#elif defined(STM32L4) || defined(STM32WB)
118

119 120
#define ADC_FIRST_GPIO_CHANNEL  (1)
#define ADC_LAST_GPIO_CHANNEL   (16)
121 122 123 124
#define ADC_SCALE_V             (VREFINT_CAL_VREF / 1000.0f)
#define ADC_CAL_ADDRESS         (VREFINT_CAL_ADDR)
#define ADC_CAL1                (TEMPSENSOR_CAL1_ADDR)
#define ADC_CAL2                (TEMPSENSOR_CAL2_ADDR)
125
#define ADC_CAL_BITS            (12)
126

127
#else
128

129
#error Unsupported processor
130

131
#endif
D
Dave Hylands 已提交
132

133 134 135
#if defined(STM32F091xC)
#define VBAT_DIV (2)
#elif defined(STM32F405xx) || defined(STM32F415xx) || \
136 137
    defined(STM32F407xx) || defined(STM32F417xx) || \
    defined(STM32F401xC) || defined(STM32F401xE)
D
Dave Hylands 已提交
138
#define VBAT_DIV (2)
139 140 141 142
#elif defined(STM32F411xE) || defined(STM32F412Zx) || \
    defined(STM32F413xx) || defined(STM32F427xx) || \
    defined(STM32F429xx) || defined(STM32F437xx) || \
    defined(STM32F439xx) || defined(STM32F446xx)
143 144
#define VBAT_DIV (4)
#elif defined(STM32F722xx) || defined(STM32F723xx) || \
145 146 147
    defined(STM32F732xx) || defined(STM32F733xx) || \
    defined(STM32F746xx) || defined(STM32F765xx) || \
    defined(STM32F767xx) || defined(STM32F769xx)
D
Dave Hylands 已提交
148
#define VBAT_DIV (4)
149
#elif defined(STM32H743xx) || defined(STM32H747xx)
150
#define VBAT_DIV (4)
151
#elif defined(STM32L432xx) || \
152 153
    defined(STM32L451xx) || defined(STM32L452xx) || \
    defined(STM32L462xx) || defined(STM32L475xx) || \
154 155
    defined(STM32L476xx) || defined(STM32L496xx) || \
    defined(STM32WB55xx)
156
#define VBAT_DIV (3)
157 158
#else
#error Unsupported processor
D
Dave Hylands 已提交
159 160
#endif

161 162 163
// Timeout for waiting for end-of-conversion, in ms
#define EOC_TIMEOUT (10)

D
Dave Hylands 已提交
164 165 166 167
/* Core temperature sensor definitions */
#define CORE_TEMP_V25          (943)  /* (0.76v/3.3v)*(2^ADC resoultion) */
#define CORE_TEMP_AVG_SLOPE    (3)    /* (2.5mv/3.3v)*(2^ADC resoultion) */

168
// scale and calibration values for VBAT and VREF
169
#define ADC_SCALE (ADC_SCALE_V / ((1 << ADC_CAL_BITS) - 1))
170 171
#define VREFIN_CAL ((uint16_t *)ADC_CAL_ADDRESS)

172 173 174
#ifndef __HAL_ADC_IS_CHANNEL_INTERNAL
#define __HAL_ADC_IS_CHANNEL_INTERNAL(channel) \
    (channel == ADC_CHANNEL_VBAT \
175 176
    || channel == ADC_CHANNEL_VREFINT \
    || channel == ADC_CHANNEL_TEMPSENSOR)
177 178
#endif

D
Dave Hylands 已提交
179 180 181 182 183 184 185
typedef struct _pyb_obj_adc_t {
    mp_obj_base_t base;
    mp_obj_t pin_name;
    int channel;
    ADC_HandleTypeDef handle;
} pyb_obj_adc_t;

186 187
// convert user-facing channel number into internal channel number
static inline uint32_t adc_get_internal_channel(uint32_t channel) {
188
    #if defined(STM32F4) || defined(STM32F7)
189 190 191 192 193 194 195 196 197
    // on F4 and F7 MCUs we want channel 16 to always be the TEMPSENSOR
    // (on some MCUs ADC_CHANNEL_TEMPSENSOR=16, on others it doesn't)
    if (channel == 16) {
        channel = ADC_CHANNEL_TEMPSENSOR;
    }
    #endif
    return channel;
}

198
STATIC bool is_adcx_channel(int channel) {
199
    #if defined(STM32F411xE)
200 201
    // The HAL has an incorrect IS_ADC_CHANNEL macro for the F411 so we check for temp
    return IS_ADC_CHANNEL(channel) || channel == ADC_CHANNEL_TEMPSENSOR;
202
    #elif defined(STM32F0) || defined(STM32F4) || defined(STM32F7)
203
    return IS_ADC_CHANNEL(channel);
204
    #elif defined(STM32H7)
205
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
206 207
           || IS_ADC_CHANNEL(__HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
    #elif defined(STM32L4)
208 209 210
    ADC_HandleTypeDef handle;
    handle.Instance = ADCx;
    return IS_ADC_CHANNEL(&handle, channel);
211 212 213 214 215
    #elif defined(STM32WB)
    ADC_HandleTypeDef handle;
    handle.Instance = ADCx;
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
           || IS_ADC_CHANNEL(&handle, __HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
216
    #else
217
    #error Unsupported processor
218
    #endif
219 220
}

221
STATIC void adc_wait_for_eoc_or_timeout(ADC_HandleTypeDef *adcHandle, int32_t timeout) {
222
    uint32_t tickstart = HAL_GetTick();
223
    #if defined(STM32F4) || defined(STM32F7)
224
    while ((adcHandle->Instance->SR & ADC_FLAG_EOC) != ADC_FLAG_EOC) {
225
    #elif defined(STM32F0) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
226
    while (READ_BIT(adcHandle->Instance->ISR, ADC_FLAG_EOC) != ADC_FLAG_EOC) {
227
    #else
228
    #error Unsupported processor
229
        #endif
230
        if (((HAL_GetTick() - tickstart) > timeout)) {
231 232 233 234 235
            break; // timeout
        }
    }
}

236
STATIC void adcx_clock_enable(ADC_HandleTypeDef *adch) {
237
    #if defined(STM32F0) || defined(STM32F4) || defined(STM32F7)
238
    ADCx_CLK_ENABLE();
239
    #elif defined(STM32H7)
240 241 242 243 244
    if (adch->Instance == ADC3) {
        __HAL_RCC_ADC3_CLK_ENABLE();
    } else {
        __HAL_RCC_ADC12_CLK_ENABLE();
    }
245
    __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_CLKP);
246 247 248 249
    #elif defined(STM32L4) || defined(STM32WB)
    if (__HAL_RCC_GET_ADC_SOURCE() == RCC_ADCCLKSOURCE_NONE) {
        __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_SYSCLK);
    }
250
    __HAL_RCC_ADC_CLK_ENABLE();
251
    #else
252
    #error Unsupported processor
253
    #endif
254 255
}

256
STATIC void adcx_init_periph(ADC_HandleTypeDef *adch, uint32_t resolution) {
257
    adcx_clock_enable(adch);
258

259 260
    adch->Init.Resolution = resolution;
    adch->Init.ContinuousConvMode = DISABLE;
261
    adch->Init.DiscontinuousConvMode = DISABLE;
262
    #if !defined(STM32F0)
263 264
    adch->Init.NbrOfDiscConversion = 0;
    adch->Init.NbrOfConversion = 1;
265
    #endif
266 267 268
    adch->Init.EOCSelection = ADC_EOC_SINGLE_CONV;
    adch->Init.ExternalTrigConv = ADC_SOFTWARE_START;
    adch->Init.ExternalTrigConvEdge = ADC_EXTERNALTRIGCONVEDGE_NONE;
269
    #if defined(STM32F0)
270 271 272
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV4;        // 12MHz
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
273
    adch->Init.DMAContinuousRequests = DISABLE;
274
    adch->Init.SamplingTimeCommon = ADC_SAMPLETIME_55CYCLES_5;    // ~4uS
275
    #elif defined(STM32F4) || defined(STM32F7)
276 277 278
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV2;
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
279 280
    adch->Init.DMAContinuousRequests = DISABLE;
    #elif defined(STM32H7)
281 282 283 284 285 286
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV4;
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.LowPowerAutoWait = DISABLE;
    adch->Init.Overrun = ADC_OVR_DATA_OVERWRITTEN;
    adch->Init.OversamplingMode = DISABLE;
    adch->Init.LeftBitShift = ADC_LEFTBITSHIFT_NONE;
287
    adch->Init.ConversionDataManagement = ADC_CONVERSIONDATA_DR;
288
    #elif defined(STM32L4) || defined(STM32WB)
289 290 291 292 293 294
    adch->Init.ClockPrescaler = ADC_CLOCK_ASYNC_DIV1;
    adch->Init.ScanConvMode = ADC_SCAN_DISABLE;
    adch->Init.LowPowerAutoWait = DISABLE;
    adch->Init.Overrun = ADC_OVR_DATA_PRESERVED;
    adch->Init.OversamplingMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
295
    adch->Init.DMAContinuousRequests = DISABLE;
296 297 298 299 300
    #else
    #error Unsupported processor
    #endif

    HAL_ADC_Init(adch);
301 302 303 304

    #if defined(STM32H7)
    HAL_ADCEx_Calibration_Start(adch, ADC_CALIB_OFFSET, ADC_SINGLE_ENDED);
    #endif
305
    #if defined(STM32L4) || defined(STM32WB)
306 307
    HAL_ADCEx_Calibration_Start(adch, ADC_SINGLE_ENDED);
    #endif
308 309
}

310
STATIC void adc_init_single(pyb_obj_adc_t *adc_obj) {
D
Dave Hylands 已提交
311

312
    if (ADC_FIRST_GPIO_CHANNEL <= adc_obj->channel && adc_obj->channel <= ADC_LAST_GPIO_CHANNEL) {
313
        // Channels 0-16 correspond to real pins. Configure the GPIO pin in ADC mode.
314
        const pin_obj_t *pin = pin_adc_table[adc_obj->channel];
315
        mp_hal_pin_config(pin, MP_HAL_PIN_MODE_ADC, MP_HAL_PIN_PULL_NONE, 0);
D
Dave Hylands 已提交
316 317
    }

318
    adc_obj->handle.Instance = ADCx;
319
    adcx_init_periph(&adc_obj->handle, ADC_RESOLUTION_12B);
320

321
    #if defined(STM32L4) && defined(ADC_DUALMODE_REGSIMULT_INJECSIMULT)
322 323
    ADC_MultiModeTypeDef multimode;
    multimode.Mode = ADC_MODE_INDEPENDENT;
324
    if (HAL_ADCEx_MultiModeConfigChannel(&adc_obj->handle, &multimode) != HAL_OK) {
325
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("Can not set multimode on ADC1 channel: %d"), adc_obj->channel);
326
    }
327
    #endif
328
}
D
Dave Hylands 已提交
329

330
STATIC void adc_config_channel(ADC_HandleTypeDef *adc_handle, uint32_t channel) {
D
Dave Hylands 已提交
331 332
    ADC_ChannelConfTypeDef sConfig;

333
    #if defined(STM32H7) || defined(STM32WB)
334 335 336 337 338
    sConfig.Rank = ADC_REGULAR_RANK_1;
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel) == 0) {
        channel = __HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel);
    }
    #else
D
Dave Hylands 已提交
339
    sConfig.Rank = 1;
340 341 342
    #endif
    sConfig.Channel = channel;

343
    #if defined(STM32F0)
344
    sConfig.SamplingTime = ADC_SAMPLETIME_55CYCLES_5;
345
    #elif defined(STM32F4) || defined(STM32F7)
346 347 348 349 350
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_480CYCLES;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_15CYCLES;
    }
351
    #elif defined(STM32H7)
352
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
353
        sConfig.SamplingTime = ADC_SAMPLETIME_810CYCLES_5;
354 355 356
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_8CYCLES_5;
    }
357 358 359 360
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
    sConfig.OffsetRightShift = DISABLE;
    sConfig.OffsetSignedSaturation = DISABLE;
361
    #elif defined(STM32L4) || defined(STM32WB)
362
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
363 364 365 366
        sConfig.SamplingTime = ADC_SAMPLETIME_247CYCLES_5;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_12CYCLES_5;
    }
367 368
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
369
    sConfig.Offset = 0;
370
    #else
371
    #error Unsupported processor
372
    #endif
D
Dave Hylands 已提交
373

374 375 376 377 378 379
    #if defined(STM32F0)
    // On the STM32F0 we must select only one channel at a time to sample, so clear all
    // channels before calling HAL_ADC_ConfigChannel, which will select the desired one.
    adc_handle->Instance->CHSELR = 0;
    #endif

380
    HAL_ADC_ConfigChannel(adc_handle, &sConfig);
D
Dave Hylands 已提交
381 382
}

383
STATIC uint32_t adc_read_channel(ADC_HandleTypeDef *adcHandle) {
D
Dave Hylands 已提交
384
    HAL_ADC_Start(adcHandle);
385 386
    adc_wait_for_eoc_or_timeout(adcHandle, EOC_TIMEOUT);
    uint32_t value = adcHandle->Instance->DR;
D
Dave Hylands 已提交
387
    HAL_ADC_Stop(adcHandle);
388 389
    return value;
}
D
Dave Hylands 已提交
390

391 392
STATIC uint32_t adc_config_and_read_channel(ADC_HandleTypeDef *adcHandle, uint32_t channel) {
    adc_config_channel(adcHandle, channel);
393 394 395 396 397 398 399
    uint32_t raw_value = adc_read_channel(adcHandle);

    // ST docs say that (at least on STM32F42x and STM32F43x), VBATE must
    // be disabled when TSVREFE is enabled for TEMPSENSOR and VREFINT
    // conversions to work.  VBATE is enabled by the above call to read
    // the channel, and here we disable VBATE so a subsequent call for
    // TEMPSENSOR or VREFINT works correctly.
400 401 402
    // It's also good to disable the VBAT switch to prevent battery drain,
    // so disable it for all MCUs.
    adc_deselect_vbat(adcHandle->Instance, channel);
403 404

    return raw_value;
D
Dave Hylands 已提交
405 406 407
}

/******************************************************************************/
408
/* MicroPython bindings : adc object (single channel)                         */
D
Dave Hylands 已提交
409

410
STATIC void adc_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
411
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
412 413
    mp_print_str(print, "<ADC on ");
    mp_obj_print_helper(print, self->pin_name, PRINT_STR);
414
    mp_printf(print, " channel=%u>", self->channel);
D
Dave Hylands 已提交
415 416
}

417 418 419
/// \classmethod \constructor(pin)
/// Create an ADC object associated with the given pin.
/// This allows you to then read analog values on that pin.
420
STATIC mp_obj_t adc_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
421
    // check number of arguments
D
Damien George 已提交
422
    mp_arg_check_num(n_args, n_kw, 1, 1, false);
D
Dave Hylands 已提交
423

424 425
    // 1st argument is the pin name
    mp_obj_t pin_obj = args[0];
D
Dave Hylands 已提交
426 427 428

    uint32_t channel;

429
    if (mp_obj_is_int(pin_obj)) {
430
        channel = adc_get_internal_channel(mp_obj_get_int(pin_obj));
D
Dave Hylands 已提交
431
    } else {
432
        const pin_obj_t *pin = pin_find(pin_obj);
433
        if ((pin->adc_num & PIN_ADC_MASK) == 0) {
D
Dave Hylands 已提交
434
            // No ADC1 function on that pin
435
            mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("pin %q does not have ADC capabilities"), pin->name);
D
Dave Hylands 已提交
436 437 438 439
        }
        channel = pin->adc_channel;
    }

440
    if (!is_adcx_channel(channel)) {
441
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("not a valid ADC Channel: %d"), channel);
D
Dave Hylands 已提交
442
    }
443 444 445 446


    if (ADC_FIRST_GPIO_CHANNEL <= channel && channel <= ADC_LAST_GPIO_CHANNEL) {
        // these channels correspond to physical GPIO ports so make sure they exist
447
        if (pin_adc_table[channel] == NULL) {
448
            mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("channel %d not available on this board"), channel);
449
        }
D
Dave Hylands 已提交
450 451 452 453
    }

    pyb_obj_adc_t *o = m_new_obj(pyb_obj_adc_t);
    memset(o, 0, sizeof(*o));
454
    o->base.type = &pyb_adc_type;
D
Dave Hylands 已提交
455 456 457 458
    o->pin_name = pin_obj;
    o->channel = channel;
    adc_init_single(o);

459
    return MP_OBJ_FROM_PTR(o);
D
Dave Hylands 已提交
460 461
}

462 463 464
/// \method read()
/// Read the value on the analog pin and return it.  The returned value
/// will be between 0 and 4095.
465
STATIC mp_obj_t adc_read(mp_obj_t self_in) {
466
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
467
    return mp_obj_new_int(adc_config_and_read_channel(&self->handle, self->channel));
468 469 470
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_read_obj, adc_read);

471
/// \method read_timed(buf, timer)
472
///
473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495
/// Read analog values into `buf` at a rate set by the `timer` object.
///
/// `buf` can be bytearray or array.array for example.  The ADC values have
/// 12-bit resolution and are stored directly into `buf` if its element size is
/// 16 bits or greater.  If `buf` has only 8-bit elements (eg a bytearray) then
/// the sample resolution will be reduced to 8 bits.
///
/// `timer` should be a Timer object, and a sample is read each time the timer
/// triggers.  The timer must already be initialised and running at the desired
/// sampling frequency.
///
/// To support previous behaviour of this function, `timer` can also be an
/// integer which specifies the frequency (in Hz) to sample at.  In this case
/// Timer(6) will be automatically configured to run at the given frequency.
///
/// Example using a Timer object (preferred way):
///
///     adc = pyb.ADC(pyb.Pin.board.X19)    # create an ADC on pin X19
///     tim = pyb.Timer(6, freq=10)         # create a timer running at 10Hz
///     buf = bytearray(100)                # creat a buffer to store the samples
///     adc.read_timed(buf, tim)            # sample 100 values, taking 10s
///
/// Example using an integer for the frequency:
496 497 498 499 500 501 502 503 504
///
///     adc = pyb.ADC(pyb.Pin.board.X19)    # create an ADC on pin X19
///     buf = bytearray(100)                # create a buffer of 100 bytes
///     adc.read_timed(buf, 10)             # read analog values into buf at 10Hz
///                                         #   this will take 10 seconds to finish
///     for val in buf:                     # loop over all values
///         print(val)                      # print the value out
///
/// This function does not allocate any memory.
505
STATIC mp_obj_t adc_read_timed(mp_obj_t self_in, mp_obj_t buf_in, mp_obj_t freq_in) {
506
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
507

508 509
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_WRITE);
510
    size_t typesize = mp_binary_get_size('@', bufinfo.typecode, NULL);
511

512 513 514 515 516 517 518 519 520 521 522 523
    TIM_HandleTypeDef *tim;
    #if defined(TIM6)
    if (mp_obj_is_integer(freq_in)) {
        // freq in Hz given so init TIM6 (legacy behaviour)
        tim = timer_tim6_init(mp_obj_get_int(freq_in));
        HAL_TIM_Base_Start(tim);
    } else
    #endif
    {
        // use the supplied timer object as the sampling time base
        tim = pyb_timer_get_handle(freq_in);
    }
524

525
    // configure the ADC channel
526
    adc_config_channel(&self->handle, self->channel);
527 528 529 530

    // This uses the timer in polling mode to do the sampling
    // TODO use DMA

531 532
    uint nelems = bufinfo.len / typesize;
    for (uint index = 0; index < nelems; index++) {
533
        // Wait for the timer to trigger so we sample at the correct frequency
534
        while (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) == RESET) {
535
        }
536
        __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);
537 538 539 540 541 542

        if (index == 0) {
            // for the first sample we need to turn the ADC on
            HAL_ADC_Start(&self->handle);
        } else {
            // for subsequent samples we can just set the "start sample" bit
543
            #if defined(STM32F4) || defined(STM32F7)
544
            self->handle.Instance->CR2 |= (uint32_t)ADC_CR2_SWSTART;
545
            #elif defined(STM32F0) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
546
            SET_BIT(self->handle.Instance->CR, ADC_CR_ADSTART);
547
            #else
548
            #error Unsupported processor
549
            #endif
550 551 552
        }

        // wait for sample to complete
553
        adc_wait_for_eoc_or_timeout(&self->handle, EOC_TIMEOUT);
554 555

        // read value
556
        uint value = self->handle.Instance->DR;
557 558

        // store value in buffer
559 560 561 562
        if (typesize == 1) {
            value >>= 4;
        }
        mp_binary_set_val_array_from_int(bufinfo.typecode, bufinfo.buf, index, value);
563 564
    }

565 566 567
    // turn the ADC off
    HAL_ADC_Stop(&self->handle);

568 569 570 571 572 573
    #if defined(TIM6)
    if (mp_obj_is_integer(freq_in)) {
        // stop timer if we initialised TIM6 in this function (legacy behaviour)
        HAL_TIM_Base_Stop(tim);
    }
    #endif
574 575 576 577 578

    return mp_obj_new_int(bufinfo.len);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_3(adc_read_timed_obj, adc_read_timed);

579 580 581 582 583 584 585 586 587 588 589 590 591 592 593
// read_timed_multi((adcx, adcy, ...), (bufx, bufy, ...), timer)
//
// Read analog values from multiple ADC's into buffers at a rate set by the
// timer.  The ADC values have 12-bit resolution and are stored directly into
// the corresponding buffer if its element size is 16 bits or greater, otherwise
// the sample resolution will be reduced to 8 bits.
//
// This function should not allocate any heap memory.
STATIC mp_obj_t adc_read_timed_multi(mp_obj_t adc_array_in, mp_obj_t buf_array_in, mp_obj_t tim_in) {
    size_t nadcs, nbufs;
    mp_obj_t *adc_array, *buf_array;
    mp_obj_get_array(adc_array_in, &nadcs, &adc_array);
    mp_obj_get_array(buf_array_in, &nbufs, &buf_array);

    if (nadcs < 1) {
594
        mp_raise_ValueError(MP_ERROR_TEXT("need at least 1 ADC"));
595 596
    }
    if (nadcs != nbufs) {
597
        mp_raise_ValueError(MP_ERROR_TEXT("length of ADC and buffer lists differ"));
598 599 600 601 602 603
    }

    // Get buf for first ADC, get word size, check other buffers match in type
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(buf_array[0], &bufinfo, MP_BUFFER_WRITE);
    size_t typesize = mp_binary_get_size('@', bufinfo.typecode, NULL);
604
    void *bufptrs[nbufs];
605 606 607 608
    for (uint array_index = 0; array_index < nbufs; array_index++) {
        mp_buffer_info_t bufinfo_curr;
        mp_get_buffer_raise(buf_array[array_index], &bufinfo_curr, MP_BUFFER_WRITE);
        if ((bufinfo.len != bufinfo_curr.len) || (bufinfo.typecode != bufinfo_curr.typecode)) {
609
            mp_raise_ValueError(MP_ERROR_TEXT("size and type of buffers must match"));
610
        }
611
        bufptrs[array_index] = bufinfo_curr.buf;
612 613 614 615 616 617 618
    }

    // Use the supplied timer object as the sampling time base
    TIM_HandleTypeDef *tim;
    tim = pyb_timer_get_handle(tim_in);

    // Start adc; this is slow so wait for it to start
619
    pyb_obj_adc_t *adc0 = MP_OBJ_TO_PTR(adc_array[0]);
620 621 622
    adc_config_channel(&adc0->handle, adc0->channel);
    HAL_ADC_Start(&adc0->handle);
    // Wait for sample to complete and discard
623
    adc_wait_for_eoc_or_timeout(&adc0->handle, EOC_TIMEOUT);
624
    // Read (and discard) value
625
    uint value = adc0->handle.Instance->DR;
626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647

    // Ensure first sample is on a timer tick
    __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);
    while (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) == RESET) {
    }
    __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);

    // Overrun check: assume success
    bool success = true;
    size_t nelems = bufinfo.len / typesize;
    for (size_t elem_index = 0; elem_index < nelems; elem_index++) {
        if (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) != RESET) {
            // Timer has already triggered
            success = false;
        } else {
            // Wait for the timer to trigger so we sample at the correct frequency
            while (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) == RESET) {
            }
        }
        __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);

        for (size_t array_index = 0; array_index < nadcs; array_index++) {
648
            pyb_obj_adc_t *adc = MP_OBJ_TO_PTR(adc_array[array_index]);
649 650 651 652 653
            // configure the ADC channel
            adc_config_channel(&adc->handle, adc->channel);
            // for the first sample we need to turn the ADC on
            // ADC is started: set the "start sample" bit
            #if defined(STM32F4) || defined(STM32F7)
654
            adc->handle.Instance->CR2 |= (uint32_t)ADC_CR2_SWSTART;
655
            #elif defined(STM32F0) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
656
            SET_BIT(adc->handle.Instance->CR, ADC_CR_ADSTART);
657 658 659 660
            #else
            #error Unsupported processor
            #endif
            // wait for sample to complete
661
            adc_wait_for_eoc_or_timeout(&adc->handle, EOC_TIMEOUT);
662 663

            // read value
664
            value = adc->handle.Instance->DR;
665 666 667 668 669

            // store values in buffer
            if (typesize == 1) {
                value >>= 4;
            }
670
            mp_binary_set_val_array_from_int(bufinfo.typecode, bufptrs[array_index], elem_index, value);
671 672 673 674
        }
    }

    // Turn the ADC off
675
    adc0 = MP_OBJ_TO_PTR(adc_array[0]);
676 677 678 679 680 681 682
    HAL_ADC_Stop(&adc0->handle);

    return mp_obj_new_bool(success);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_3(adc_read_timed_multi_fun_obj, adc_read_timed_multi);
STATIC MP_DEFINE_CONST_STATICMETHOD_OBJ(adc_read_timed_multi_obj, MP_ROM_PTR(&adc_read_timed_multi_fun_obj));

683 684 685
STATIC const mp_rom_map_elem_t adc_locals_dict_table[] = {
    { MP_ROM_QSTR(MP_QSTR_read), MP_ROM_PTR(&adc_read_obj) },
    { MP_ROM_QSTR(MP_QSTR_read_timed), MP_ROM_PTR(&adc_read_timed_obj) },
686
    { MP_ROM_QSTR(MP_QSTR_read_timed_multi), MP_ROM_PTR(&adc_read_timed_multi_obj) },
687 688
};

689 690
STATIC MP_DEFINE_CONST_DICT(adc_locals_dict, adc_locals_dict_table);

691 692 693 694 695
const mp_obj_type_t pyb_adc_type = {
    { &mp_type_type },
    .name = MP_QSTR_ADC,
    .print = adc_print,
    .make_new = adc_make_new,
696
    .locals_dict = (mp_obj_dict_t *)&adc_locals_dict,
697
};
D
Dave Hylands 已提交
698 699 700 701

/******************************************************************************/
/* adc all object                                                             */

702
typedef struct _pyb_adc_all_obj_t {
D
Dave Hylands 已提交
703 704
    mp_obj_base_t base;
    ADC_HandleTypeDef handle;
705
} pyb_adc_all_obj_t;
D
Dave Hylands 已提交
706

707
void adc_init_all(pyb_adc_all_obj_t *adc_all, uint32_t resolution, uint32_t en_mask) {
D
Dave Hylands 已提交
708 709

    switch (resolution) {
710
        #if !defined(STM32H7)
711 712 713
        case 6:
            resolution = ADC_RESOLUTION_6B;
            break;
714
        #endif
715 716 717 718 719 720 721 722 723
        case 8:
            resolution = ADC_RESOLUTION_8B;
            break;
        case 10:
            resolution = ADC_RESOLUTION_10B;
            break;
        case 12:
            resolution = ADC_RESOLUTION_12B;
            break;
724
        #if defined(STM32H7)
725 726 727
        case 16:
            resolution = ADC_RESOLUTION_16B;
            break;
728
        #endif
D
Dave Hylands 已提交
729
        default:
730
            mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("resolution %d not supported"), resolution);
D
Dave Hylands 已提交
731 732
    }

733
    for (uint32_t channel = ADC_FIRST_GPIO_CHANNEL; channel <= ADC_LAST_GPIO_CHANNEL; ++channel) {
734 735 736 737
        // only initialise those channels that are selected with the en_mask
        if (en_mask & (1 << channel)) {
            // Channels 0-16 correspond to real pins. Configure the GPIO pin in
            // ADC mode.
738
            const pin_obj_t *pin = pin_adcall_table[channel];
739
            if (pin) {
740
                mp_hal_pin_config(pin, MP_HAL_PIN_MODE_ADC, MP_HAL_PIN_PULL_NONE, 0);
741 742
            }
        }
D
Dave Hylands 已提交
743 744
    }

745
    adc_all->handle.Instance = ADCALLx;
746
    adcx_init_periph(&adc_all->handle, resolution);
D
Dave Hylands 已提交
747 748
}

D
Dave Hylands 已提交
749
int adc_get_resolution(ADC_HandleTypeDef *adcHandle) {
750
    uint32_t res_reg = ADC_GET_RESOLUTION(adcHandle);
D
Dave Hylands 已提交
751 752

    switch (res_reg) {
753
        #if !defined(STM32H7)
754 755
        case ADC_RESOLUTION_6B:
            return 6;
756
        #endif
757 758 759 760
        case ADC_RESOLUTION_8B:
            return 8;
        case ADC_RESOLUTION_10B:
            return 10;
761
        #if defined(STM32H7)
762 763
        case ADC_RESOLUTION_16B:
            return 16;
764
        #endif
D
Dave Hylands 已提交
765 766 767 768
    }
    return 12;
}

769 770 771 772 773
STATIC uint32_t adc_config_and_read_ref(ADC_HandleTypeDef *adcHandle, uint32_t channel) {
    uint32_t raw_value = adc_config_and_read_channel(adcHandle, channel);
    // Scale raw reading to the number of bits used by the calibration constants
    return raw_value << (ADC_CAL_BITS - adc_get_resolution(adcHandle));
}
D
Dave Hylands 已提交
774

775 776
int adc_read_core_temp(ADC_HandleTypeDef *adcHandle) {
    int32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR);
D
Dave Hylands 已提交
777 778 779
    return ((raw_value - CORE_TEMP_V25) / CORE_TEMP_AVG_SLOPE) + 25;
}

D
Dave Hylands 已提交
780
#if MICROPY_PY_BUILTINS_FLOAT
781 782 783
// correction factor for reference value
STATIC volatile float adc_refcor = 1.0f;

784
float adc_read_core_temp_float(ADC_HandleTypeDef *adcHandle) {
785
    int32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR);
786
    float core_temp_avg_slope = (*ADC_CAL2 - *ADC_CAL1) / 80.0f;
787 788 789
    return (((float)raw_value * adc_refcor - *ADC_CAL1) / core_temp_avg_slope) + 30.0f;
}

790
float adc_read_core_vbat(ADC_HandleTypeDef *adcHandle) {
791
    uint32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_VBAT);
792
    return raw_value * VBAT_DIV * ADC_SCALE * adc_refcor;
D
Dave Hylands 已提交
793 794
}

795
float adc_read_core_vref(ADC_HandleTypeDef *adcHandle) {
796
    uint32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_VREFINT);
D
Dave Hylands 已提交
797

798 799 800 801
    // update the reference correction factor
    adc_refcor = ((float)(*VREFIN_CAL)) / ((float)raw_value);

    return (*VREFIN_CAL) * ADC_SCALE;
D
Dave Hylands 已提交
802
}
D
Dave Hylands 已提交
803
#endif
D
Dave Hylands 已提交
804 805

/******************************************************************************/
806
/* MicroPython bindings : adc_all object                                      */
D
Dave Hylands 已提交
807

808
STATIC mp_obj_t adc_all_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
809
    // check number of arguments
810
    mp_arg_check_num(n_args, n_kw, 1, 2, false);
811 812 813 814

    // make ADCAll object
    pyb_adc_all_obj_t *o = m_new_obj(pyb_adc_all_obj_t);
    o->base.type = &pyb_adc_all_type;
815 816 817
    mp_int_t res = mp_obj_get_int(args[0]);
    uint32_t en_mask = 0xffffffff;
    if (n_args > 1) {
818
        en_mask = mp_obj_get_int(args[1]);
819 820
    }
    adc_init_all(o, res, en_mask);
821

822
    return MP_OBJ_FROM_PTR(o);
D
Dave Hylands 已提交
823 824
}

825
STATIC mp_obj_t adc_all_read_channel(mp_obj_t self_in, mp_obj_t channel) {
826
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
827
    uint32_t chan = adc_get_internal_channel(mp_obj_get_int(channel));
D
Dave Hylands 已提交
828 829 830
    uint32_t data = adc_config_and_read_channel(&self->handle, chan);
    return mp_obj_new_int(data);
}
831
STATIC MP_DEFINE_CONST_FUN_OBJ_2(adc_all_read_channel_obj, adc_all_read_channel);
D
Dave Hylands 已提交
832

833
STATIC mp_obj_t adc_all_read_core_temp(mp_obj_t self_in) {
834
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
835 836 837 838
    #if MICROPY_PY_BUILTINS_FLOAT
    float data = adc_read_core_temp_float(&self->handle);
    return mp_obj_new_float(data);
    #else
839
    int data = adc_read_core_temp(&self->handle);
D
Dave Hylands 已提交
840
    return mp_obj_new_int(data);
841
    #endif
D
Dave Hylands 已提交
842
}
843
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_temp_obj, adc_all_read_core_temp);
D
Dave Hylands 已提交
844

D
Dave Hylands 已提交
845
#if MICROPY_PY_BUILTINS_FLOAT
846
STATIC mp_obj_t adc_all_read_core_vbat(mp_obj_t self_in) {
847
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
D
Dave Hylands 已提交
848 849 850
    float data = adc_read_core_vbat(&self->handle);
    return mp_obj_new_float(data);
}
851
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vbat_obj, adc_all_read_core_vbat);
D
Dave Hylands 已提交
852

853
STATIC mp_obj_t adc_all_read_core_vref(mp_obj_t self_in) {
854
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
855
    float data = adc_read_core_vref(&self->handle);
D
Dave Hylands 已提交
856 857
    return mp_obj_new_float(data);
}
858
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vref_obj, adc_all_read_core_vref);
859 860

STATIC mp_obj_t adc_all_read_vref(mp_obj_t self_in) {
861
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
862
    adc_read_core_vref(&self->handle);
863
    return mp_obj_new_float(ADC_SCALE_V * adc_refcor);
864 865
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_vref_obj, adc_all_read_vref);
D
Dave Hylands 已提交
866
#endif
D
Dave Hylands 已提交
867

868 869 870
STATIC const mp_rom_map_elem_t adc_all_locals_dict_table[] = {
    { MP_ROM_QSTR(MP_QSTR_read_channel), MP_ROM_PTR(&adc_all_read_channel_obj) },
    { MP_ROM_QSTR(MP_QSTR_read_core_temp), MP_ROM_PTR(&adc_all_read_core_temp_obj) },
871
    #if MICROPY_PY_BUILTINS_FLOAT
872 873 874
    { MP_ROM_QSTR(MP_QSTR_read_core_vbat), MP_ROM_PTR(&adc_all_read_core_vbat_obj) },
    { MP_ROM_QSTR(MP_QSTR_read_core_vref), MP_ROM_PTR(&adc_all_read_core_vref_obj) },
    { MP_ROM_QSTR(MP_QSTR_read_vref), MP_ROM_PTR(&adc_all_read_vref_obj) },
875
    #endif
D
Dave Hylands 已提交
876 877
};

878 879
STATIC MP_DEFINE_CONST_DICT(adc_all_locals_dict, adc_all_locals_dict_table);

880
const mp_obj_type_t pyb_adc_all_type = {
D
Dave Hylands 已提交
881
    { &mp_type_type },
882 883
    .name = MP_QSTR_ADCAll,
    .make_new = adc_all_make_new,
884
    .locals_dict = (mp_obj_dict_t *)&adc_all_locals_dict,
D
Dave Hylands 已提交
885
};
886 887

#endif // MICROPY_HW_ENABLE_ADC