adc.c 35.8 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
/* ADC definitions */
D
Dave Hylands 已提交
54
#define ADCx                    (ADC1)
55 56 57
#define PIN_ADC_MASK            PIN_ADC1
#define pin_adc_table           pin_adc1

58 59 60 61 62
#if defined(STM32H7A3xx) || defined(STM32H7A3xxQ) || \
    defined(STM32H7B3xx) || defined(STM32H7B3xxQ)
#define ADCALLx                 (ADC2)
#define pin_adcall_table        pin_adc2
#elif defined(STM32H7)
63 64 65 66 67 68 69 70 71 72
// 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

73
#define ADCx_CLK_ENABLE         __HAL_RCC_ADC1_CLK_ENABLE
74

75 76
#if defined(STM32F0)

77
#define ADC_SCALE_V             (3.3f)
78
#define ADC_CAL_ADDRESS         (0x1ffff7ba)
79 80
#define ADC_CAL1                ((uint16_t *)0x1ffff7b8)
#define ADC_CAL2                ((uint16_t *)0x1ffff7c2)
81
#define ADC_CAL_BITS            (12)
82 83

#elif defined(STM32F4)
84

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
#define ADC_SCALE_V             (3.3f)
94 95 96 97
#if defined(STM32F722xx) || defined(STM32F723xx) || \
    defined(STM32F732xx) || defined(STM32F733xx)
#define ADC_CAL_ADDRESS         (0x1ff07a2a)
#else
98
#define ADC_CAL_ADDRESS         (0x1ff0f44a)
99 100
#endif

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

105
#elif defined(STM32G0) || defined(STM32G4)
106 107 108 109 110

#define ADC_SCALE_V             (((float)VREFINT_CAL_VREF) / 1000.0f)
#define ADC_CAL_ADDRESS         VREFINT_CAL_ADDR
#define ADC_CAL1                TEMPSENSOR_CAL1_ADDR
#define ADC_CAL2                TEMPSENSOR_CAL2_ADDR
111
#define ADC_CAL_BITS            (12) // UM2319/UM2570, __HAL_ADC_CALC_TEMPERATURE: 'corresponds to a resolution of 12 bits'
112

113 114
#elif defined(STM32H7)

115
#define ADC_SCALE_V             (3.3f)
116
#define ADC_CAL_ADDRESS         (0x1FF1E860)
117 118
#define ADC_CAL1                ((uint16_t *)(0x1FF1E820))
#define ADC_CAL2                ((uint16_t *)(0x1FF1E840))
119
#define ADC_CAL_BITS            (16)
120

Y
yn386 已提交
121 122 123 124 125 126 127 128
#elif defined(STM32L1)

#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)
#define ADC_CAL_BITS            (12)

129
#elif defined(STM32L4) || defined(STM32WB)
130

131 132 133 134
#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)
135
#define ADC_CAL_BITS            (12)
136

137
#else
138

139
#error Unsupported processor
140

141
#endif
D
Dave Hylands 已提交
142

143 144 145
#if defined(STM32F091xC)
#define VBAT_DIV (2)
#elif defined(STM32F405xx) || defined(STM32F415xx) || \
146 147
    defined(STM32F407xx) || defined(STM32F417xx) || \
    defined(STM32F401xC) || defined(STM32F401xE)
D
Dave Hylands 已提交
148
#define VBAT_DIV (2)
149 150 151
#elif defined(STM32F411xE) || defined(STM32F412Zx) || \
    defined(STM32F413xx) || defined(STM32F427xx) || \
    defined(STM32F429xx) || defined(STM32F437xx) || \
D
Damien George 已提交
152 153
    defined(STM32F439xx) || defined(STM32F446xx) || \
    defined(STM32F479xx)
154 155
#define VBAT_DIV (4)
#elif defined(STM32F722xx) || defined(STM32F723xx) || \
156
    defined(STM32F732xx) || defined(STM32F733xx) || \
157 158
    defined(STM32F745xx) || defined(STM32F746xx) || \
    defined(STM32F756xx) || defined(STM32F765xx) || \
159
    defined(STM32F767xx) || defined(STM32F769xx)
D
Dave Hylands 已提交
160
#define VBAT_DIV (4)
161
#elif defined(STM32G0) || defined(STM32G4)
162
#define VBAT_DIV (3)
163 164
#elif defined(STM32H723xx) || defined(STM32H733xx) || \
    defined(STM32H743xx) || defined(STM32H747xx) || \
165 166
    defined(STM32H7A3xx) || defined(STM32H7A3xxQ) || \
    defined(STM32H7B3xx) || defined(STM32H7B3xxQ) || \
167
    defined(STM32H750xx)
168
#define VBAT_DIV (4)
169
#elif defined(STM32L432xx) || \
170 171
    defined(STM32L451xx) || defined(STM32L452xx) || \
    defined(STM32L462xx) || defined(STM32L475xx) || \
172
    defined(STM32L476xx) || defined(STM32L496xx) || \
173
    defined(STM32L4A6xx) || \
174
    defined(STM32WB55xx)
175
#define VBAT_DIV (3)
Y
yn386 已提交
176 177
#elif defined(STM32L152xE)
// STM32L152xE does not have vbat.
178 179
#else
#error Unsupported processor
D
Dave Hylands 已提交
180 181
#endif

182 183 184
// Timeout for waiting for end-of-conversion, in ms
#define EOC_TIMEOUT (10)

D
Dave Hylands 已提交
185
/* Core temperature sensor definitions */
186 187
#define CORE_TEMP_V25          (943)  /* (0.76v/3.3v)*(2^ADC resolution) */
#define CORE_TEMP_AVG_SLOPE    (3)    /* (2.5mv/3.3v)*(2^ADC resolution) */
D
Dave Hylands 已提交
188

189
// scale and calibration values for VBAT and VREF
190
#define ADC_SCALE (ADC_SCALE_V / ((1 << ADC_CAL_BITS) - 1))
191 192
#define VREFIN_CAL ((uint16_t *)ADC_CAL_ADDRESS)

193
#ifndef __HAL_ADC_IS_CHANNEL_INTERNAL
Y
yn386 已提交
194 195 196 197 198
#if defined(STM32L1)
#define __HAL_ADC_IS_CHANNEL_INTERNAL(channel) \
    (channel == ADC_CHANNEL_VREFINT \
    || channel == ADC_CHANNEL_TEMPSENSOR)
#else
199 200
#define __HAL_ADC_IS_CHANNEL_INTERNAL(channel) \
    (channel == ADC_CHANNEL_VBAT \
201 202
    || channel == ADC_CHANNEL_VREFINT \
    || channel == ADC_CHANNEL_TEMPSENSOR)
203
#endif
Y
yn386 已提交
204
#endif
205

D
Dave Hylands 已提交
206 207 208
typedef struct _pyb_obj_adc_t {
    mp_obj_base_t base;
    mp_obj_t pin_name;
209
    uint32_t channel;
D
Dave Hylands 已提交
210 211 212
    ADC_HandleTypeDef handle;
} pyb_obj_adc_t;

213 214
// convert user-facing channel number into internal channel number
static inline uint32_t adc_get_internal_channel(uint32_t channel) {
215
    #if defined(STM32F4) || defined(STM32F7)
216 217 218 219 220
    // 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;
    }
221 222 223 224 225 226 227 228
    #elif defined(STM32L4)
    if (channel == 0) {
        channel = ADC_CHANNEL_VREFINT;
    } else if (channel == 17) {
        channel = ADC_CHANNEL_TEMPSENSOR;
    } else if (channel == 18) {
        channel = ADC_CHANNEL_VBAT;
    }
229 230 231 232
    #endif
    return channel;
}

233
STATIC bool is_adcx_channel(int channel) {
234
    #if defined(STM32F411xE)
235 236
    // 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;
237
    #elif defined(STM32F0) || defined(STM32F4) || defined(STM32F7)
238
    return IS_ADC_CHANNEL(channel);
Y
yn386 已提交
239 240 241 242
    #elif defined(STM32L1)
    // The HAL of STM32L1 defines some channels those may not be available on package
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
           || (channel < MP_ARRAY_SIZE(pin_adcall_table) && pin_adcall_table[channel]);
243
    #elif defined(STM32G0) || defined(STM32H7)
244
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
245
           || IS_ADC_CHANNEL(__HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
246
    #elif defined(STM32G4) || defined(STM32L4) || defined(STM32WB)
247 248 249 250
    ADC_HandleTypeDef handle;
    handle.Instance = ADCx;
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
           || IS_ADC_CHANNEL(&handle, __HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
251
    #else
252
    #error Unsupported processor
253
    #endif
254 255
}

256
STATIC void adc_wait_for_eoc_or_timeout(ADC_HandleTypeDef *adcHandle, int32_t timeout) {
257
    uint32_t tickstart = HAL_GetTick();
Y
yn386 已提交
258
    #if defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
259
    while ((adcHandle->Instance->SR & ADC_FLAG_EOC) != ADC_FLAG_EOC) {
260
    #elif defined(STM32F0) || defined(STM32G0) || defined(STM32G4) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
261
    while (READ_BIT(adcHandle->Instance->ISR, ADC_FLAG_EOC) != ADC_FLAG_EOC) {
262
    #else
263
    #error Unsupported processor
264
        #endif
265
        if (((HAL_GetTick() - tickstart) > timeout)) {
266 267 268 269 270
            break; // timeout
        }
    }
}

271
STATIC void adcx_clock_enable(ADC_HandleTypeDef *adch) {
Y
yn386 已提交
272
    #if defined(STM32F0) || defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
273
    ADCx_CLK_ENABLE();
274 275 276
    #elif defined(STM32H7A3xx) || defined(STM32H7A3xxQ) || defined(STM32H7B3xx) || defined(STM32H7B3xxQ)
    __HAL_RCC_ADC12_CLK_ENABLE();
    __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_CLKP);
277 278
    #elif defined(STM32G0)
    __HAL_RCC_ADC_CLK_ENABLE();
279 280
    #elif defined(STM32G4)
    __HAL_RCC_ADC12_CLK_ENABLE();
281
    #elif defined(STM32H7)
282 283 284 285 286
    if (adch->Instance == ADC3) {
        __HAL_RCC_ADC3_CLK_ENABLE();
    } else {
        __HAL_RCC_ADC12_CLK_ENABLE();
    }
287
    __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_CLKP);
288 289 290 291
    #elif defined(STM32L4) || defined(STM32WB)
    if (__HAL_RCC_GET_ADC_SOURCE() == RCC_ADCCLKSOURCE_NONE) {
        __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_SYSCLK);
    }
292
    __HAL_RCC_ADC_CLK_ENABLE();
293
    #else
294
    #error Unsupported processor
295
    #endif
296 297
}

298
STATIC void adcx_init_periph(ADC_HandleTypeDef *adch, uint32_t resolution) {
299
    adcx_clock_enable(adch);
300

301 302
    adch->Init.Resolution = resolution;
    adch->Init.ContinuousConvMode = DISABLE;
303
    adch->Init.DiscontinuousConvMode = DISABLE;
304
    #if !defined(STM32F0) && !defined(STM32G0)
305
    adch->Init.NbrOfDiscConversion = 0;
306 307
    #endif
    #if !defined(STM32F0)
308
    adch->Init.NbrOfConversion = 1;
309
    #endif
310 311 312
    adch->Init.EOCSelection = ADC_EOC_SINGLE_CONV;
    adch->Init.ExternalTrigConv = ADC_SOFTWARE_START;
    adch->Init.ExternalTrigConvEdge = ADC_EXTERNALTRIGCONVEDGE_NONE;
313
    #if defined(STM32F0)
314 315 316
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV4;        // 12MHz
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
317
    adch->Init.DMAContinuousRequests = DISABLE;
318
    adch->Init.SamplingTimeCommon = ADC_SAMPLETIME_55CYCLES_5;    // ~4uS
319
    #elif defined(STM32F4) || defined(STM32F7)
320 321 322
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV2;
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
323 324
    adch->Init.DMAContinuousRequests = DISABLE;
    #elif defined(STM32H7)
325 326 327 328 329 330
    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;
331
    adch->Init.ConversionDataManagement = ADC_CONVERSIONDATA_DR;
Y
yn386 已提交
332 333 334 335 336 337
    #elif defined(STM32L1)
    adch->Init.ClockPrescaler = ADC_CLOCK_ASYNC_DIV1;
    adch->Init.ScanConvMode = ADC_SCAN_DISABLE;
    adch->Init.LowPowerAutoWait = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
    adch->Init.DMAContinuousRequests = DISABLE;
338
    #elif defined(STM32G0) || defined(STM32G4) || defined(STM32L4) || defined(STM32WB)
339 340 341 342 343 344
    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;
345
    adch->Init.DMAContinuousRequests = DISABLE;
346 347 348 349 350
    #else
    #error Unsupported processor
    #endif

    HAL_ADC_Init(adch);
351 352 353 354

    #if defined(STM32H7)
    HAL_ADCEx_Calibration_Start(adch, ADC_CALIB_OFFSET, ADC_SINGLE_ENDED);
    #endif
355 356 357
    #if defined(STM32G0)
    HAL_ADCEx_Calibration_Start(adch);
    #elif defined(STM32G4) || defined(STM32L4) || defined(STM32WB)
358 359
    HAL_ADCEx_Calibration_Start(adch, ADC_SINGLE_ENDED);
    #endif
360 361
}

362
STATIC void adc_init_single(pyb_obj_adc_t *adc_obj) {
363
    adc_obj->handle.Instance = ADCx;
364
    adcx_init_periph(&adc_obj->handle, ADC_RESOLUTION_12B);
365

366
    #if (defined(STM32G4) || defined(STM32L4)) && defined(ADC_DUALMODE_REGSIMULT_INJECSIMULT)
367 368
    ADC_MultiModeTypeDef multimode;
    multimode.Mode = ADC_MODE_INDEPENDENT;
369
    if (HAL_ADCEx_MultiModeConfigChannel(&adc_obj->handle, &multimode) != HAL_OK) {
370
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("Can not set multimode on ADC1 channel: %d"), adc_obj->channel);
371
    }
372
    #endif
373
}
D
Dave Hylands 已提交
374

375
STATIC void adc_config_channel(ADC_HandleTypeDef *adc_handle, uint32_t channel) {
D
Dave Hylands 已提交
376 377
    ADC_ChannelConfTypeDef sConfig;

378
    #if defined(STM32G0) || defined(STM32G4) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
379 380 381 382 383
    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 已提交
384
    sConfig.Rank = 1;
385 386 387
    #endif
    sConfig.Channel = channel;

388
    #if defined(STM32F0)
389
    sConfig.SamplingTime = ADC_SAMPLETIME_55CYCLES_5;
390
    #elif defined(STM32F4) || defined(STM32F7)
391 392 393 394 395
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_480CYCLES;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_15CYCLES;
    }
396
    #elif defined(STM32H7)
397
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
398
        sConfig.SamplingTime = ADC_SAMPLETIME_810CYCLES_5;
399 400 401
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_8CYCLES_5;
    }
402 403 404 405
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
    sConfig.OffsetRightShift = DISABLE;
    sConfig.OffsetSignedSaturation = DISABLE;
Y
yn386 已提交
406 407 408 409 410 411
    #elif defined(STM32L1)
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_384CYCLES;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_384CYCLES;
    }
412 413 414 415 416 417
    #elif defined(STM32G0)
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_160CYCLES_5;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_12CYCLES_5;
    }
418
    #elif defined(STM32G4) || defined(STM32L4) || defined(STM32WB)
419
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
420 421 422 423
        sConfig.SamplingTime = ADC_SAMPLETIME_247CYCLES_5;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_12CYCLES_5;
    }
424 425
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
426
    sConfig.Offset = 0;
427
    #else
428
    #error Unsupported processor
429
    #endif
D
Dave Hylands 已提交
430

431 432 433 434 435 436
    #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

437
    HAL_ADC_ConfigChannel(adc_handle, &sConfig);
D
Dave Hylands 已提交
438 439
}

440
STATIC uint32_t adc_read_channel(ADC_HandleTypeDef *adcHandle) {
D
Dave Hylands 已提交
441
    HAL_ADC_Start(adcHandle);
442 443
    adc_wait_for_eoc_or_timeout(adcHandle, EOC_TIMEOUT);
    uint32_t value = adcHandle->Instance->DR;
D
Dave Hylands 已提交
444
    HAL_ADC_Stop(adcHandle);
445 446
    return value;
}
D
Dave Hylands 已提交
447

448 449
STATIC uint32_t adc_config_and_read_channel(ADC_HandleTypeDef *adcHandle, uint32_t channel) {
    adc_config_channel(adcHandle, channel);
450 451 452 453 454 455 456
    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.
457 458 459
    // 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);
460 461

    return raw_value;
D
Dave Hylands 已提交
462 463 464
}

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

467
STATIC void adc_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
468
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
469 470
    mp_print_str(print, "<ADC on ");
    mp_obj_print_helper(print, self->pin_name, PRINT_STR);
471
    mp_printf(print, " channel=%u>", self->channel);
D
Dave Hylands 已提交
472 473
}

474 475 476
/// \classmethod \constructor(pin)
/// Create an ADC object associated with the given pin.
/// This allows you to then read analog values on that pin.
477
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) {
478
    // check number of arguments
D
Damien George 已提交
479
    mp_arg_check_num(n_args, n_kw, 1, 1, false);
D
Dave Hylands 已提交
480

481 482
    // 1st argument is the pin name
    mp_obj_t pin_obj = args[0];
D
Dave Hylands 已提交
483 484 485

    uint32_t channel;

486
    if (mp_obj_is_int(pin_obj)) {
487
        channel = adc_get_internal_channel(mp_obj_get_int(pin_obj));
D
Dave Hylands 已提交
488
    } else {
489
        const pin_obj_t *pin = pin_find(pin_obj);
490
        if ((pin->adc_num & PIN_ADC_MASK) == 0) {
491 492
            // No ADC function on the given pin.
            mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("Pin(%q) doesn't have ADC capabilities"), pin->name);
D
Dave Hylands 已提交
493 494 495 496
        }
        channel = pin->adc_channel;
    }

497
    if (!is_adcx_channel(channel)) {
498
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("not a valid ADC Channel: %d"), channel);
D
Dave Hylands 已提交
499
    }
500

501 502 503 504 505
    // If this channel corresponds to a pin then configure the pin in ADC mode.
    if (channel < MP_ARRAY_SIZE(pin_adc_table)) {
        const pin_obj_t *pin = pin_adc_table[channel];
        if (pin != NULL) {
            mp_hal_pin_config(pin, MP_HAL_PIN_MODE_ADC, MP_HAL_PIN_PULL_NONE, 0);
506
        }
D
Dave Hylands 已提交
507 508 509 510
    }

    pyb_obj_adc_t *o = m_new_obj(pyb_obj_adc_t);
    memset(o, 0, sizeof(*o));
511
    o->base.type = &pyb_adc_type;
D
Dave Hylands 已提交
512 513 514 515
    o->pin_name = pin_obj;
    o->channel = channel;
    adc_init_single(o);

516
    return MP_OBJ_FROM_PTR(o);
D
Dave Hylands 已提交
517 518
}

519 520 521
/// \method read()
/// Read the value on the analog pin and return it.  The returned value
/// will be between 0 and 4095.
522
STATIC mp_obj_t adc_read(mp_obj_t self_in) {
523
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
524
    return mp_obj_new_int(adc_config_and_read_channel(&self->handle, self->channel));
525 526 527
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_read_obj, adc_read);

528
/// \method read_timed(buf, timer)
529
///
530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
/// 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:
553 554 555 556 557 558 559 560 561
///
///     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.
562
STATIC mp_obj_t adc_read_timed(mp_obj_t self_in, mp_obj_t buf_in, mp_obj_t freq_in) {
563
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
564

565 566
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_WRITE);
567
    size_t typesize = mp_binary_get_size('@', bufinfo.typecode, NULL);
568

569 570 571 572 573 574 575 576 577 578 579 580
    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);
    }
581

582
    // configure the ADC channel
583
    adc_config_channel(&self->handle, self->channel);
584 585 586 587

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

588 589
    uint nelems = bufinfo.len / typesize;
    for (uint index = 0; index < nelems; index++) {
590
        // Wait for the timer to trigger so we sample at the correct frequency
591
        while (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) == RESET) {
592
        }
593
        __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);
594 595 596 597 598 599

        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
Y
yn386 已提交
600
            #if defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
601
            self->handle.Instance->CR2 |= (uint32_t)ADC_CR2_SWSTART;
602
            #elif defined(STM32F0) || defined(STM32G0) || defined(STM32G4) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
603
            SET_BIT(self->handle.Instance->CR, ADC_CR_ADSTART);
604
            #else
605
            #error Unsupported processor
606
            #endif
607 608 609
        }

        // wait for sample to complete
610
        adc_wait_for_eoc_or_timeout(&self->handle, EOC_TIMEOUT);
611 612

        // read value
613
        uint value = self->handle.Instance->DR;
614 615

        // store value in buffer
616 617 618 619
        if (typesize == 1) {
            value >>= 4;
        }
        mp_binary_set_val_array_from_int(bufinfo.typecode, bufinfo.buf, index, value);
620 621
    }

622 623 624
    // turn the ADC off
    HAL_ADC_Stop(&self->handle);

625 626 627 628 629 630
    #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
631 632 633 634 635

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

636 637 638 639 640 641 642 643 644 645 646 647 648 649 650
// 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) {
651
        mp_raise_ValueError(MP_ERROR_TEXT("need at least 1 ADC"));
652 653
    }
    if (nadcs != nbufs) {
654
        mp_raise_ValueError(MP_ERROR_TEXT("length of ADC and buffer lists differ"));
655 656 657 658 659 660
    }

    // 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);
661
    void *bufptrs[nbufs];
662 663 664 665
    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)) {
666
            mp_raise_ValueError(MP_ERROR_TEXT("size and type of buffers must match"));
667
        }
668
        bufptrs[array_index] = bufinfo_curr.buf;
669 670 671 672 673 674 675
    }

    // 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
676
    pyb_obj_adc_t *adc0 = MP_OBJ_TO_PTR(adc_array[0]);
677 678 679
    adc_config_channel(&adc0->handle, adc0->channel);
    HAL_ADC_Start(&adc0->handle);
    // Wait for sample to complete and discard
680
    adc_wait_for_eoc_or_timeout(&adc0->handle, EOC_TIMEOUT);
681
    // Read (and discard) value
682
    uint value = adc0->handle.Instance->DR;
683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704

    // 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++) {
705
            pyb_obj_adc_t *adc = MP_OBJ_TO_PTR(adc_array[array_index]);
706 707 708 709
            // 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
Y
yn386 已提交
710
            #if defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
711
            adc->handle.Instance->CR2 |= (uint32_t)ADC_CR2_SWSTART;
712
            #elif defined(STM32F0) || defined(STM32G0) || defined(STM32G4) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
713
            SET_BIT(adc->handle.Instance->CR, ADC_CR_ADSTART);
714 715 716 717
            #else
            #error Unsupported processor
            #endif
            // wait for sample to complete
718
            adc_wait_for_eoc_or_timeout(&adc->handle, EOC_TIMEOUT);
719 720

            // read value
721
            value = adc->handle.Instance->DR;
722 723 724 725 726

            // store values in buffer
            if (typesize == 1) {
                value >>= 4;
            }
727
            mp_binary_set_val_array_from_int(bufinfo.typecode, bufptrs[array_index], elem_index, value);
728 729 730 731
        }
    }

    // Turn the ADC off
732
    adc0 = MP_OBJ_TO_PTR(adc_array[0]);
733 734 735 736 737 738 739
    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));

740 741 742
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) },
743
    { MP_ROM_QSTR(MP_QSTR_read_timed_multi), MP_ROM_PTR(&adc_read_timed_multi_obj) },
744 745
};

746 747
STATIC MP_DEFINE_CONST_DICT(adc_locals_dict, adc_locals_dict_table);

748 749 750 751
MP_DEFINE_CONST_OBJ_TYPE(
    pyb_adc_type,
    MP_QSTR_ADC,
    MP_TYPE_FLAG_NONE,
752
    make_new, adc_make_new,
753
    print, adc_print,
754
    locals_dict, &adc_locals_dict
755
    );
D
Dave Hylands 已提交
756 757 758 759

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

760
typedef struct _pyb_adc_all_obj_t {
D
Dave Hylands 已提交
761 762
    mp_obj_base_t base;
    ADC_HandleTypeDef handle;
763
} pyb_adc_all_obj_t;
D
Dave Hylands 已提交
764

Y
yn386 已提交
765 766
float adc_read_core_vref(ADC_HandleTypeDef *adcHandle);

767
void adc_init_all(pyb_adc_all_obj_t *adc_all, uint32_t resolution, uint32_t en_mask) {
D
Dave Hylands 已提交
768 769

    switch (resolution) {
770
        #if !defined(STM32H7)
771 772 773
        case 6:
            resolution = ADC_RESOLUTION_6B;
            break;
774
        #endif
775 776 777 778 779 780 781 782 783
        case 8:
            resolution = ADC_RESOLUTION_8B;
            break;
        case 10:
            resolution = ADC_RESOLUTION_10B;
            break;
        case 12:
            resolution = ADC_RESOLUTION_12B;
            break;
784
        #if defined(STM32H7)
785 786 787
        case 16:
            resolution = ADC_RESOLUTION_16B;
            break;
788
        #endif
D
Dave Hylands 已提交
789
        default:
790
            mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("resolution %d not supported"), resolution);
D
Dave Hylands 已提交
791 792
    }

793
    for (uint32_t channel = 0; channel < MP_ARRAY_SIZE(pin_adcall_table); ++channel) {
794 795
        // only initialise those channels that are selected with the en_mask
        if (en_mask & (1 << channel)) {
796
            // If this channel corresponds to a pin then configure the pin in ADC mode.
797
            const pin_obj_t *pin = pin_adcall_table[channel];
798
            if (pin) {
799
                mp_hal_pin_config(pin, MP_HAL_PIN_MODE_ADC, MP_HAL_PIN_PULL_NONE, 0);
800 801
            }
        }
D
Dave Hylands 已提交
802 803
    }

804
    adc_all->handle.Instance = ADCALLx;
805
    adcx_init_periph(&adc_all->handle, resolution);
D
Dave Hylands 已提交
806 807
}

D
Dave Hylands 已提交
808
int adc_get_resolution(ADC_HandleTypeDef *adcHandle) {
Y
yn386 已提交
809 810 811
    #if defined(STM32L1)
    uint32_t res_reg = adcHandle->Instance->CR1 & ADC_CR1_RES_Msk;
    #else
812
    uint32_t res_reg = ADC_GET_RESOLUTION(adcHandle);
Y
yn386 已提交
813
    #endif
D
Dave Hylands 已提交
814 815

    switch (res_reg) {
816
        #if !defined(STM32H7)
817 818
        case ADC_RESOLUTION_6B:
            return 6;
819
        #endif
820 821 822 823
        case ADC_RESOLUTION_8B:
            return 8;
        case ADC_RESOLUTION_10B:
            return 10;
824
        #if defined(STM32H7)
825 826
        case ADC_RESOLUTION_16B:
            return 16;
827
        #endif
D
Dave Hylands 已提交
828 829 830 831
    }
    return 12;
}

832 833 834 835 836
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 已提交
837

838
int adc_read_core_temp(ADC_HandleTypeDef *adcHandle) {
839 840 841 842 843 844 845 846
    #if defined(STM32G4)
    int32_t raw_value = 0;
    if (adcHandle->Instance == ADC1) {
        raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR_ADC1);
    } else {
        return 0;
    }
    #else
847
    int32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR);
848
    #endif
D
Dave Hylands 已提交
849 850 851
    return ((raw_value - CORE_TEMP_V25) / CORE_TEMP_AVG_SLOPE) + 25;
}

D
Dave Hylands 已提交
852
#if MICROPY_PY_BUILTINS_FLOAT
853 854 855
// correction factor for reference value
STATIC volatile float adc_refcor = 1.0f;

856
float adc_read_core_temp_float(ADC_HandleTypeDef *adcHandle) {
857 858 859 860 861 862 863 864
    #if defined(STM32G4)
    int32_t raw_value = 0;
    if (adcHandle->Instance == ADC1) {
        raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR_ADC1);
    } else {
        return 0;
    }
    #else
865
    #if defined(STM32L1) || defined(STM32L4)
Y
yn386 已提交
866
    // Update the reference correction factor before reading tempsensor
867
    // because TS_CAL1 and TS_CAL2 of STM32L1/L4 are at VDDA=3.0V
Y
yn386 已提交
868 869
    adc_read_core_vref(adcHandle);
    #endif
870
    int32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR);
871
    #endif
872
    float core_temp_avg_slope = (*ADC_CAL2 - *ADC_CAL1) / 80.0f;
873 874 875
    return (((float)raw_value * adc_refcor - *ADC_CAL1) / core_temp_avg_slope) + 30.0f;
}

876
float adc_read_core_vbat(ADC_HandleTypeDef *adcHandle) {
Y
yn386 已提交
877 878 879
    #if defined(STM32L152xE)
    mp_raise_NotImplementedError(MP_ERROR_TEXT("read_core_vbat not supported"));
    #else
880
    uint32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_VBAT);
881
    return raw_value * VBAT_DIV * ADC_SCALE * adc_refcor;
Y
yn386 已提交
882
    #endif
D
Dave Hylands 已提交
883 884
}

885
float adc_read_core_vref(ADC_HandleTypeDef *adcHandle) {
886
    uint32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_VREFINT);
D
Dave Hylands 已提交
887

888 889 890 891
    // update the reference correction factor
    adc_refcor = ((float)(*VREFIN_CAL)) / ((float)raw_value);

    return (*VREFIN_CAL) * ADC_SCALE;
D
Dave Hylands 已提交
892
}
D
Dave Hylands 已提交
893
#endif
D
Dave Hylands 已提交
894 895

/******************************************************************************/
896
/* MicroPython bindings : adc_all object                                      */
D
Dave Hylands 已提交
897

898
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) {
899
    // check number of arguments
900
    mp_arg_check_num(n_args, n_kw, 1, 2, false);
901 902

    // make ADCAll object
903
    pyb_adc_all_obj_t *o = mp_obj_malloc(pyb_adc_all_obj_t, &pyb_adc_all_type);
904 905 906
    mp_int_t res = mp_obj_get_int(args[0]);
    uint32_t en_mask = 0xffffffff;
    if (n_args > 1) {
907
        en_mask = mp_obj_get_int(args[1]);
908 909
    }
    adc_init_all(o, res, en_mask);
910

911
    return MP_OBJ_FROM_PTR(o);
D
Dave Hylands 已提交
912 913
}

914
STATIC mp_obj_t adc_all_read_channel(mp_obj_t self_in, mp_obj_t channel) {
915
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
916
    uint32_t chan = adc_get_internal_channel(mp_obj_get_int(channel));
917 918 919
    if (!is_adcx_channel(chan)) {
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("not a valid ADC Channel: %d"), chan);
    }
D
Dave Hylands 已提交
920 921 922
    uint32_t data = adc_config_and_read_channel(&self->handle, chan);
    return mp_obj_new_int(data);
}
923
STATIC MP_DEFINE_CONST_FUN_OBJ_2(adc_all_read_channel_obj, adc_all_read_channel);
D
Dave Hylands 已提交
924

925
STATIC mp_obj_t adc_all_read_core_temp(mp_obj_t self_in) {
926
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
927 928 929 930
    #if MICROPY_PY_BUILTINS_FLOAT
    float data = adc_read_core_temp_float(&self->handle);
    return mp_obj_new_float(data);
    #else
931
    int data = adc_read_core_temp(&self->handle);
D
Dave Hylands 已提交
932
    return mp_obj_new_int(data);
933
    #endif
D
Dave Hylands 已提交
934
}
935
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_temp_obj, adc_all_read_core_temp);
D
Dave Hylands 已提交
936

D
Dave Hylands 已提交
937
#if MICROPY_PY_BUILTINS_FLOAT
938
STATIC mp_obj_t adc_all_read_core_vbat(mp_obj_t self_in) {
939
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
D
Dave Hylands 已提交
940 941 942
    float data = adc_read_core_vbat(&self->handle);
    return mp_obj_new_float(data);
}
943
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vbat_obj, adc_all_read_core_vbat);
D
Dave Hylands 已提交
944

945
STATIC mp_obj_t adc_all_read_core_vref(mp_obj_t self_in) {
946
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
947
    float data = adc_read_core_vref(&self->handle);
D
Dave Hylands 已提交
948 949
    return mp_obj_new_float(data);
}
950
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vref_obj, adc_all_read_core_vref);
951 952

STATIC mp_obj_t adc_all_read_vref(mp_obj_t self_in) {
953
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
954
    adc_read_core_vref(&self->handle);
955
    return mp_obj_new_float(ADC_SCALE_V * adc_refcor);
956 957
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_vref_obj, adc_all_read_vref);
D
Dave Hylands 已提交
958
#endif
D
Dave Hylands 已提交
959

960 961 962
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) },
963
    #if MICROPY_PY_BUILTINS_FLOAT
964 965 966
    { 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) },
967
    #endif
D
Dave Hylands 已提交
968 969
};

970 971
STATIC MP_DEFINE_CONST_DICT(adc_all_locals_dict, adc_all_locals_dict_table);

972 973 974 975
MP_DEFINE_CONST_OBJ_TYPE(
    pyb_adc_all_type,
    MP_QSTR_ADCAll,
    MP_TYPE_FLAG_NONE,
976
    make_new, adc_all_make_new,
977
    locals_dict, &adc_all_locals_dict
978
    );
979 980

#endif // MICROPY_HW_ENABLE_ADC