adc.c 24.5 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 "genhdr/pins.h"
36
#include "timer.h"
D
Dave Hylands 已提交
37

38 39
#if MICROPY_HW_ENABLE_ADC

40 41 42 43 44 45 46 47 48 49 50 51 52
/// \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 已提交
53 54 55

/* ADC defintions */
#define ADCx                    (ADC1)
56
#define ADCx_CLK_ENABLE         __HAL_RCC_ADC1_CLK_ENABLE
D
Dave Hylands 已提交
57
#define ADC_NUM_CHANNELS        (19)
58

59 60 61 62 63
#if defined(MCU_SERIES_F4)

#define ADC_FIRST_GPIO_CHANNEL  (0)
#define ADC_LAST_GPIO_CHANNEL   (15)
#define ADC_CAL_ADDRESS         (0x1fff7a2a)
64 65
#define ADC_CAL1                ((uint16_t*)(ADC_CAL_ADDRESS + 2))
#define ADC_CAL2                ((uint16_t*)(ADC_CAL_ADDRESS + 4))
66 67 68

#elif defined(MCU_SERIES_F7)

69 70
#define ADC_FIRST_GPIO_CHANNEL  (0)
#define ADC_LAST_GPIO_CHANNEL   (15)
71 72 73 74
#if defined(STM32F722xx) || defined(STM32F723xx) || \
    defined(STM32F732xx) || defined(STM32F733xx)
#define ADC_CAL_ADDRESS         (0x1ff07a2a)
#else
75
#define ADC_CAL_ADDRESS         (0x1ff0f44a)
76 77
#endif

78 79
#define ADC_CAL1                ((uint16_t*)(ADC_CAL_ADDRESS + 2))
#define ADC_CAL2                ((uint16_t*)(ADC_CAL_ADDRESS + 4))
80

81
#elif defined(MCU_SERIES_L4)
82

83 84
#define ADC_FIRST_GPIO_CHANNEL  (1)
#define ADC_LAST_GPIO_CHANNEL   (16)
85
#define ADC_CAL_ADDRESS         (0x1fff75aa)
86 87
#define ADC_CAL1                ((uint16_t*)(ADC_CAL_ADDRESS - 2))
#define ADC_CAL2                ((uint16_t*)(ADC_CAL_ADDRESS + 0x20))
88

89
#else
90

91
#error Unsupported processor
92

93
#endif
D
Dave Hylands 已提交
94 95 96

#if defined(STM32F405xx) || defined(STM32F415xx) || \
    defined(STM32F407xx) || defined(STM32F417xx) || \
97 98
    defined(STM32F401xC) || defined(STM32F401xE) || \
    defined(STM32F411xE)
D
Dave Hylands 已提交
99 100
#define VBAT_DIV (2)
#elif defined(STM32F427xx) || defined(STM32F429xx) || \
101
      defined(STM32F437xx) || defined(STM32F439xx) || \
102 103
      defined(STM32F722xx) || defined(STM32F723xx) || \
      defined(STM32F732xx) || defined(STM32F733xx) || \
R
Rami Ali 已提交
104
      defined(STM32F746xx) || defined(STM32F767xx) || \
105
      defined(STM32F769xx) || defined(STM32F446xx)
D
Dave Hylands 已提交
106
#define VBAT_DIV (4)
107
#elif defined(STM32L475xx) || defined(STM32L476xx)
108
#define VBAT_DIV (3)
109 110
#else
#error Unsupported processor
D
Dave Hylands 已提交
111 112 113 114 115 116
#endif

/* 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) */

117 118 119 120
// scale and calibration values for VBAT and VREF
#define ADC_SCALE (3.3f / 4095)
#define VREFIN_CAL ((uint16_t *)ADC_CAL_ADDRESS)

D
Dave Hylands 已提交
121 122 123 124 125 126 127
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;

128 129 130 131 132 133 134 135 136 137 138 139
// convert user-facing channel number into internal channel number
static inline uint32_t adc_get_internal_channel(uint32_t channel) {
    #if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
    // 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;
}

140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
STATIC bool is_adcx_channel(int channel) {
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
    return IS_ADC_CHANNEL(channel);
#elif defined(MCU_SERIES_L4)
    ADC_HandleTypeDef handle;
    handle.Instance = ADCx;
    return IS_ADC_CHANNEL(&handle, channel);
#else
    #error Unsupported processor
#endif
}

STATIC void adc_wait_for_eoc_or_timeout(int32_t timeout) {
    uint32_t tickstart = HAL_GetTick();
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
    while ((ADCx->SR & ADC_FLAG_EOC) != ADC_FLAG_EOC) {
#elif defined(MCU_SERIES_L4)
    while (READ_BIT(ADCx->ISR, ADC_FLAG_EOC) != ADC_FLAG_EOC) {
#else
    #error Unsupported processor
#endif
        if (((HAL_GetTick() - tickstart ) > timeout)) {
            break; // timeout
        }
    }
}

STATIC void adcx_clock_enable(void) {
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
    ADCx_CLK_ENABLE();
#elif defined(MCU_SERIES_L4)
    __HAL_RCC_ADC_CLK_ENABLE();
#else
    #error Unsupported processor
#endif
}

177
STATIC void adc_init_single(pyb_obj_adc_t *adc_obj) {
178
    if (!is_adcx_channel(adc_obj->channel)) {
D
Dave Hylands 已提交
179 180 181
        return;
    }

182
    if (ADC_FIRST_GPIO_CHANNEL <= adc_obj->channel && adc_obj->channel <= ADC_LAST_GPIO_CHANNEL) {
D
Dave Hylands 已提交
183 184 185
      // Channels 0-16 correspond to real pins. Configure the GPIO pin in
      // ADC mode.
      const pin_obj_t *pin = pin_adc1[adc_obj->channel];
186
      mp_hal_gpio_clock_enable(pin->gpio);
D
Dave Hylands 已提交
187 188
      GPIO_InitTypeDef GPIO_InitStructure;
      GPIO_InitStructure.Pin = pin->pin_mask;
189
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
D
Dave Hylands 已提交
190
      GPIO_InitStructure.Mode = GPIO_MODE_ANALOG;
191 192 193 194 195
#elif defined(MCU_SERIES_L4)
      GPIO_InitStructure.Mode = GPIO_MODE_ANALOG_ADC_CONTROL;
#else
    #error Unsupported processor
#endif
D
Dave Hylands 已提交
196 197 198 199
      GPIO_InitStructure.Pull = GPIO_NOPULL;
      HAL_GPIO_Init(pin->gpio, &GPIO_InitStructure);
    }

200
    adcx_clock_enable();
D
Dave Hylands 已提交
201 202 203 204 205 206 207 208 209 210

    ADC_HandleTypeDef *adcHandle = &adc_obj->handle;
    adcHandle->Instance                   = ADCx;
    adcHandle->Init.ContinuousConvMode    = DISABLE;
    adcHandle->Init.DiscontinuousConvMode = DISABLE;
    adcHandle->Init.NbrOfDiscConversion   = 0;
    adcHandle->Init.ExternalTrigConvEdge  = ADC_EXTERNALTRIGCONVEDGE_NONE;
    adcHandle->Init.DataAlign             = ADC_DATAALIGN_RIGHT;
    adcHandle->Init.NbrOfConversion       = 1;
    adcHandle->Init.DMAContinuousRequests = DISABLE;
211
    adcHandle->Init.Resolution            = ADC_RESOLUTION_12B;
212
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
213
    adcHandle->Init.ClockPrescaler        = ADC_CLOCK_SYNC_PCLK_DIV2;
214 215
    adcHandle->Init.ScanConvMode          = DISABLE;
    adcHandle->Init.ExternalTrigConv      = ADC_EXTERNALTRIGCONV_T1_CC1;
216
    adcHandle->Init.EOCSelection          = DISABLE;
217
#elif defined(MCU_SERIES_L4)
218
    adcHandle->Init.ClockPrescaler        = ADC_CLOCK_ASYNC_DIV1;
219
    adcHandle->Init.ScanConvMode          = ADC_SCAN_DISABLE;
220 221 222
    adcHandle->Init.EOCSelection          = ADC_EOC_SINGLE_CONV;
    adcHandle->Init.ExternalTrigConv      = ADC_SOFTWARE_START;
    adcHandle->Init.ExternalTrigConvEdge  = ADC_EXTERNALTRIGCONVEDGE_NONE;
223 224 225 226 227 228
    adcHandle->Init.LowPowerAutoWait      = DISABLE;
    adcHandle->Init.Overrun               = ADC_OVR_DATA_PRESERVED;
    adcHandle->Init.OversamplingMode      = DISABLE;
#else
    #error Unsupported processor
#endif
D
Dave Hylands 已提交
229 230

    HAL_ADC_Init(adcHandle);
231 232 233 234 235 236 237 238 239

#if defined(MCU_SERIES_L4)
    ADC_MultiModeTypeDef multimode;
    multimode.Mode = ADC_MODE_INDEPENDENT;
    if (HAL_ADCEx_MultiModeConfigChannel(adcHandle, &multimode) != HAL_OK)
    {
        nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "Can not set multimode on ADC1 channel: %d", adc_obj->channel));
    }
#endif
240
}
D
Dave Hylands 已提交
241

242
STATIC void adc_config_channel(ADC_HandleTypeDef *adc_handle, uint32_t channel) {
D
Dave Hylands 已提交
243 244
    ADC_ChannelConfTypeDef sConfig;

245
    sConfig.Channel = channel;
D
Dave Hylands 已提交
246
    sConfig.Rank = 1;
247
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
D
Dave Hylands 已提交
248
    sConfig.SamplingTime = ADC_SAMPLETIME_15CYCLES;
249 250
#elif defined(MCU_SERIES_L4)
    sConfig.SamplingTime = ADC_SAMPLETIME_12CYCLES_5;
251 252
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
253 254 255
#else
    #error Unsupported processor
#endif
D
Dave Hylands 已提交
256 257
    sConfig.Offset = 0;

258
    HAL_ADC_ConfigChannel(adc_handle, &sConfig);
D
Dave Hylands 已提交
259 260
}

261
STATIC uint32_t adc_read_channel(ADC_HandleTypeDef *adcHandle) {
D
Dave Hylands 已提交
262 263 264
    uint32_t rawValue = 0;

    HAL_ADC_Start(adcHandle);
265
    if (HAL_ADC_PollForConversion(adcHandle, 10) == HAL_OK
266
        && (HAL_ADC_GetState(adcHandle) & HAL_ADC_STATE_REG_EOC) == HAL_ADC_STATE_REG_EOC) {
D
Dave Hylands 已提交
267 268 269 270 271 272 273 274
        rawValue = HAL_ADC_GetValue(adcHandle);
    }
    HAL_ADC_Stop(adcHandle);

    return rawValue;
}

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

277
STATIC void adc_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
D
Dave Hylands 已提交
278
    pyb_obj_adc_t *self = self_in;
279 280 281
    mp_print_str(print, "<ADC on ");
    mp_obj_print_helper(print, self->pin_name, PRINT_STR);
    mp_printf(print, " channel=%lu>", self->channel);
D
Dave Hylands 已提交
282 283
}

284 285 286
/// \classmethod \constructor(pin)
/// Create an ADC object associated with the given pin.
/// This allows you to then read analog values on that pin.
287
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) {
288
    // check number of arguments
D
Damien George 已提交
289
    mp_arg_check_num(n_args, n_kw, 1, 1, false);
D
Dave Hylands 已提交
290

291 292
    // 1st argument is the pin name
    mp_obj_t pin_obj = args[0];
D
Dave Hylands 已提交
293 294 295 296

    uint32_t channel;

    if (MP_OBJ_IS_INT(pin_obj)) {
297
        channel = adc_get_internal_channel(mp_obj_get_int(pin_obj));
D
Dave Hylands 已提交
298
    } else {
299
        const pin_obj_t *pin = pin_find(pin_obj);
D
Dave Hylands 已提交
300 301
        if ((pin->adc_num & PIN_ADC1) == 0) {
            // No ADC1 function on that pin
302
            nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "pin %q does not have ADC capabilities", pin->name));
D
Dave Hylands 已提交
303 304 305 306
        }
        channel = pin->adc_channel;
    }

307
    if (!is_adcx_channel(channel)) {
308
        nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "not a valid ADC Channel: %d", channel));
D
Dave Hylands 已提交
309
    }
310 311 312 313 314 315 316 317


    if (ADC_FIRST_GPIO_CHANNEL <= channel && channel <= ADC_LAST_GPIO_CHANNEL) {
        // these channels correspond to physical GPIO ports so make sure they exist
        if (pin_adc1[channel] == NULL) {
            nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError,
                "channel %d not available on this board", channel));
        }
D
Dave Hylands 已提交
318 319 320 321
    }

    pyb_obj_adc_t *o = m_new_obj(pyb_obj_adc_t);
    memset(o, 0, sizeof(*o));
322
    o->base.type = &pyb_adc_type;
D
Dave Hylands 已提交
323 324 325 326 327 328 329
    o->pin_name = pin_obj;
    o->channel = channel;
    adc_init_single(o);

    return o;
}

330 331 332
/// \method read()
/// Read the value on the analog pin and return it.  The returned value
/// will be between 0 and 4095.
333 334 335
STATIC mp_obj_t adc_read(mp_obj_t self_in) {
    pyb_obj_adc_t *self = self_in;

336
    adc_config_channel(&self->handle, self->channel);
337 338 339 340 341
    uint32_t data = adc_read_channel(&self->handle);
    return mp_obj_new_int(data);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_read_obj, adc_read);

342
/// \method read_timed(buf, timer)
343
///
344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366
/// 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:
367 368 369 370 371 372 373 374 375
///
///     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.
376 377 378
STATIC mp_obj_t adc_read_timed(mp_obj_t self_in, mp_obj_t buf_in, mp_obj_t freq_in) {
    pyb_obj_adc_t *self = self_in;

379 380
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_WRITE);
381
    size_t typesize = mp_binary_get_size('@', bufinfo.typecode, NULL);
382

383 384 385 386 387 388 389 390 391 392 393 394
    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);
    }
395

396
    // configure the ADC channel
397
    adc_config_channel(&self->handle, self->channel);
398 399 400 401

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

402 403
    uint nelems = bufinfo.len / typesize;
    for (uint index = 0; index < nelems; index++) {
404
        // Wait for the timer to trigger so we sample at the correct frequency
405
        while (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) == RESET) {
406
        }
407
        __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);
408 409 410 411 412 413

        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
414
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
415
            ADCx->CR2 |= (uint32_t)ADC_CR2_SWSTART;
416 417 418 419 420
#elif defined(MCU_SERIES_L4)
            SET_BIT(ADCx->CR, ADC_CR_ADSTART);
#else
            #error Unsupported processor
#endif
421 422 423
        }

        // wait for sample to complete
424 425
        #define READ_TIMED_TIMEOUT (10) // in ms
        adc_wait_for_eoc_or_timeout(READ_TIMED_TIMEOUT);
426 427 428 429 430

        // read value
        uint value = ADCx->DR;

        // store value in buffer
431 432 433 434
        if (typesize == 1) {
            value >>= 4;
        }
        mp_binary_set_val_array_from_int(bufinfo.typecode, bufinfo.buf, index, value);
435 436
    }

437 438 439
    // turn the ADC off
    HAL_ADC_Stop(&self->handle);

440 441 442 443 444 445
    #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
446 447 448 449 450

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

451 452 453
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) },
454 455
};

456 457
STATIC MP_DEFINE_CONST_DICT(adc_locals_dict, adc_locals_dict_table);

458 459 460 461 462
const mp_obj_type_t pyb_adc_type = {
    { &mp_type_type },
    .name = MP_QSTR_ADC,
    .print = adc_print,
    .make_new = adc_make_new,
463
    .locals_dict = (mp_obj_dict_t*)&adc_locals_dict,
464
};
D
Dave Hylands 已提交
465 466 467 468

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

469
typedef struct _pyb_adc_all_obj_t {
D
Dave Hylands 已提交
470 471
    mp_obj_base_t base;
    ADC_HandleTypeDef handle;
472
} pyb_adc_all_obj_t;
D
Dave Hylands 已提交
473

474
void adc_init_all(pyb_adc_all_obj_t *adc_all, uint32_t resolution, uint32_t en_mask) {
D
Dave Hylands 已提交
475 476

    switch (resolution) {
477 478 479 480
        case 6:  resolution = ADC_RESOLUTION_6B;  break;
        case 8:  resolution = ADC_RESOLUTION_8B;  break;
        case 10: resolution = ADC_RESOLUTION_10B; break;
        case 12: resolution = ADC_RESOLUTION_12B; break;
D
Dave Hylands 已提交
481
        default:
482
            nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError,
D
Dave Hylands 已提交
483 484 485
                "resolution %d not supported", resolution));
    }

486
    for (uint32_t channel = ADC_FIRST_GPIO_CHANNEL; channel <= ADC_LAST_GPIO_CHANNEL; ++channel) {
487 488 489 490 491 492 493 494 495 496 497 498 499 500
        // 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.
            const pin_obj_t *pin = pin_adc1[channel];
            if (pin) {
                mp_hal_gpio_clock_enable(pin->gpio);
                GPIO_InitTypeDef GPIO_InitStructure;
                GPIO_InitStructure.Pin = pin->pin_mask;
                GPIO_InitStructure.Mode = GPIO_MODE_ANALOG;
                GPIO_InitStructure.Pull = GPIO_NOPULL;
                HAL_GPIO_Init(pin->gpio, &GPIO_InitStructure);
            }
        }
D
Dave Hylands 已提交
501 502
    }

503
    adcx_clock_enable();
D
Dave Hylands 已提交
504 505 506 507 508 509 510 511 512 513 514 515

    ADC_HandleTypeDef *adcHandle = &adc_all->handle;
    adcHandle->Instance = ADCx;
    adcHandle->Init.Resolution            = resolution;
    adcHandle->Init.ContinuousConvMode    = DISABLE;
    adcHandle->Init.DiscontinuousConvMode = DISABLE;
    adcHandle->Init.NbrOfDiscConversion   = 0;
    adcHandle->Init.ExternalTrigConvEdge  = ADC_EXTERNALTRIGCONVEDGE_NONE;
    adcHandle->Init.DataAlign             = ADC_DATAALIGN_RIGHT;
    adcHandle->Init.NbrOfConversion       = 1;
    adcHandle->Init.DMAContinuousRequests = DISABLE;
    adcHandle->Init.EOCSelection          = DISABLE;
516
#if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
517
    adcHandle->Init.ClockPrescaler        = ADC_CLOCK_SYNC_PCLK_DIV2;
518 519 520 521 522 523 524 525 526 527 528 529
    adcHandle->Init.ScanConvMode          = DISABLE;
    adcHandle->Init.ExternalTrigConv      = ADC_EXTERNALTRIGCONV_T1_CC1;
#elif defined(MCU_SERIES_L4)
    adcHandle->Init.ClockPrescaler        = ADC_CLOCK_ASYNC_DIV2;
    adcHandle->Init.ScanConvMode          = ADC_SCAN_DISABLE;
    adcHandle->Init.ExternalTrigConv      = ADC_EXTERNALTRIG_T1_CC1;
    adcHandle->Init.LowPowerAutoWait      = DISABLE;
    adcHandle->Init.Overrun               = ADC_OVR_DATA_PRESERVED;
    adcHandle->Init.OversamplingMode      = DISABLE;
#else
    #error Unsupported processor
#endif
D
Dave Hylands 已提交
530 531 532 533

    HAL_ADC_Init(adcHandle);
}

534
uint32_t adc_config_and_read_channel(ADC_HandleTypeDef *adcHandle, uint32_t channel) {
535
    adc_config_channel(adcHandle, channel);
D
Dave Hylands 已提交
536 537 538
    return adc_read_channel(adcHandle);
}

D
Dave Hylands 已提交
539
int adc_get_resolution(ADC_HandleTypeDef *adcHandle) {
540
    uint32_t res_reg = ADC_GET_RESOLUTION(adcHandle);
D
Dave Hylands 已提交
541 542

    switch (res_reg) {
543 544 545
        case ADC_RESOLUTION_6B:  return 6;
        case ADC_RESOLUTION_8B:  return 8;
        case ADC_RESOLUTION_10B: return 10;
D
Dave Hylands 已提交
546 547 548 549
    }
    return 12;
}

550
int adc_read_core_temp(ADC_HandleTypeDef *adcHandle) {
D
Dave Hylands 已提交
551
    int32_t raw_value = adc_config_and_read_channel(adcHandle, ADC_CHANNEL_TEMPSENSOR);
D
Dave Hylands 已提交
552 553 554 555 556

    // Note: constants assume 12-bit resolution, so we scale the raw value to
    //       be 12-bits.
    raw_value <<= (12 - adc_get_resolution(adcHandle));

D
Dave Hylands 已提交
557 558 559
    return ((raw_value - CORE_TEMP_V25) / CORE_TEMP_AVG_SLOPE) + 25;
}

D
Dave Hylands 已提交
560
#if MICROPY_PY_BUILTINS_FLOAT
561 562 563
// correction factor for reference value
STATIC volatile float adc_refcor = 1.0f;

564 565 566 567 568 569 570 571 572 573
float adc_read_core_temp_float(ADC_HandleTypeDef *adcHandle) {
    int32_t raw_value = adc_config_and_read_channel(adcHandle, ADC_CHANNEL_TEMPSENSOR);

    // constants assume 12-bit resolution so we scale the raw value to 12-bits
    raw_value <<= (12 - adc_get_resolution(adcHandle));

    float core_temp_avg_slope = (*ADC_CAL2 - *ADC_CAL1) / 80.0;
    return (((float)raw_value * adc_refcor - *ADC_CAL1) / core_temp_avg_slope) + 30.0f;
}

574
float adc_read_core_vbat(ADC_HandleTypeDef *adcHandle) {
D
Dave Hylands 已提交
575
    uint32_t raw_value = adc_config_and_read_channel(adcHandle, ADC_CHANNEL_VBAT);
D
Dave Hylands 已提交
576 577 578 579 580

    // Note: constants assume 12-bit resolution, so we scale the raw value to
    //       be 12-bits.
    raw_value <<= (12 - adc_get_resolution(adcHandle));

581 582 583 584 585 586 587 588 589 590
    #if defined(MCU_SERIES_F4) || defined(MCU_SERIES_F7)
    // 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.
    ADC->CCR &= ~ADC_CCR_VBATE;
    #endif

    return raw_value * VBAT_DIV * ADC_SCALE * adc_refcor;
D
Dave Hylands 已提交
591 592
}

593
float adc_read_core_vref(ADC_HandleTypeDef *adcHandle) {
D
Dave Hylands 已提交
594
    uint32_t raw_value = adc_config_and_read_channel(adcHandle, ADC_CHANNEL_VREFINT);
D
Dave Hylands 已提交
595 596 597 598 599

    // Note: constants assume 12-bit resolution, so we scale the raw value to
    //       be 12-bits.
    raw_value <<= (12 - adc_get_resolution(adcHandle));

600 601 602 603
    // update the reference correction factor
    adc_refcor = ((float)(*VREFIN_CAL)) / ((float)raw_value);

    return (*VREFIN_CAL) * ADC_SCALE;
D
Dave Hylands 已提交
604
}
D
Dave Hylands 已提交
605
#endif
D
Dave Hylands 已提交
606 607

/******************************************************************************/
608
/* MicroPython bindings : adc_all object                                      */
D
Dave Hylands 已提交
609

610
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) {
611
    // check number of arguments
612
    mp_arg_check_num(n_args, n_kw, 1, 2, false);
613 614 615 616

    // make ADCAll object
    pyb_adc_all_obj_t *o = m_new_obj(pyb_adc_all_obj_t);
    o->base.type = &pyb_adc_all_type;
617 618 619 620 621 622
    mp_int_t res = mp_obj_get_int(args[0]);
    uint32_t en_mask = 0xffffffff;
    if (n_args > 1) {
        en_mask =  mp_obj_get_int(args[1]);
    }
    adc_init_all(o, res, en_mask);
623 624

    return o;
D
Dave Hylands 已提交
625 626
}

627
STATIC mp_obj_t adc_all_read_channel(mp_obj_t self_in, mp_obj_t channel) {
628
    pyb_adc_all_obj_t *self = self_in;
629
    uint32_t chan = adc_get_internal_channel(mp_obj_get_int(channel));
D
Dave Hylands 已提交
630 631 632
    uint32_t data = adc_config_and_read_channel(&self->handle, chan);
    return mp_obj_new_int(data);
}
633
STATIC MP_DEFINE_CONST_FUN_OBJ_2(adc_all_read_channel_obj, adc_all_read_channel);
D
Dave Hylands 已提交
634

635
STATIC mp_obj_t adc_all_read_core_temp(mp_obj_t self_in) {
636
    pyb_adc_all_obj_t *self = self_in;
637 638 639 640
    #if MICROPY_PY_BUILTINS_FLOAT
    float data = adc_read_core_temp_float(&self->handle);
    return mp_obj_new_float(data);
    #else
D
Dave Hylands 已提交
641 642
    int data  = adc_read_core_temp(&self->handle);
    return mp_obj_new_int(data);
643
    #endif
D
Dave Hylands 已提交
644
}
645
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_temp_obj, adc_all_read_core_temp);
D
Dave Hylands 已提交
646

D
Dave Hylands 已提交
647
#if MICROPY_PY_BUILTINS_FLOAT
648
STATIC mp_obj_t adc_all_read_core_vbat(mp_obj_t self_in) {
649
    pyb_adc_all_obj_t *self = self_in;
D
Dave Hylands 已提交
650 651 652
    float data = adc_read_core_vbat(&self->handle);
    return mp_obj_new_float(data);
}
653
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vbat_obj, adc_all_read_core_vbat);
D
Dave Hylands 已提交
654

655
STATIC mp_obj_t adc_all_read_core_vref(mp_obj_t self_in) {
656
    pyb_adc_all_obj_t *self = self_in;
D
Dave Hylands 已提交
657 658 659
    float data  = adc_read_core_vref(&self->handle);
    return mp_obj_new_float(data);
}
660
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vref_obj, adc_all_read_core_vref);
661 662 663 664 665 666 667

STATIC mp_obj_t adc_all_read_vref(mp_obj_t self_in) {
    pyb_adc_all_obj_t *self = self_in;
    adc_read_core_vref(&self->handle);
    return mp_obj_new_float(3.3 * adc_refcor);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_vref_obj, adc_all_read_vref);
D
Dave Hylands 已提交
668
#endif
D
Dave Hylands 已提交
669

670 671 672
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) },
D
Dave Hylands 已提交
673
#if MICROPY_PY_BUILTINS_FLOAT
674 675 676
    { 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) },
D
Dave Hylands 已提交
677
#endif
D
Dave Hylands 已提交
678 679
};

680 681
STATIC MP_DEFINE_CONST_DICT(adc_all_locals_dict, adc_all_locals_dict_table);

682
const mp_obj_type_t pyb_adc_all_type = {
D
Dave Hylands 已提交
683
    { &mp_type_type },
684 685
    .name = MP_QSTR_ADCAll,
    .make_new = adc_all_make_new,
686
    .locals_dict = (mp_obj_dict_t*)&adc_all_locals_dict,
D
Dave Hylands 已提交
687
};
688 689

#endif // MICROPY_HW_ENABLE_ADC