adc.c 37.1 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) || defined(STM32H5)
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(STM32H5)
#define VBAT_DIV (4)
165 166
#elif defined(STM32H723xx) || defined(STM32H733xx) || \
    defined(STM32H743xx) || defined(STM32H747xx) || \
167 168
    defined(STM32H7A3xx) || defined(STM32H7A3xxQ) || \
    defined(STM32H7B3xx) || defined(STM32H7B3xxQ) || \
169
    defined(STM32H750xx)
170
#define VBAT_DIV (4)
171
#elif defined(STM32L432xx) || \
172 173
    defined(STM32L451xx) || defined(STM32L452xx) || \
    defined(STM32L462xx) || defined(STM32L475xx) || \
174
    defined(STM32L476xx) || defined(STM32L496xx) || \
175
    defined(STM32L4A6xx) || \
176
    defined(STM32WB55xx)
177
#define VBAT_DIV (3)
Y
yn386 已提交
178 179
#elif defined(STM32L152xE)
// STM32L152xE does not have vbat.
180 181
#else
#error Unsupported processor
D
Dave Hylands 已提交
182 183
#endif

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

D
Dave Hylands 已提交
187
/* Core temperature sensor definitions */
188 189
#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 已提交
190

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

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

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

215 216
// convert user-facing channel number into internal channel number
static inline uint32_t adc_get_internal_channel(uint32_t channel) {
217
    #if defined(STM32F4) || defined(STM32F7)
218 219 220 221 222
    // 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;
    }
223 224 225 226 227 228 229 230
    #elif defined(STM32G4)
    if (channel == 16) {
        channel = ADC_CHANNEL_TEMPSENSOR_ADC1;
    } else if (channel == 17) {
        channel = ADC_CHANNEL_VBAT;
    } else if (channel == 18) {
        channel = ADC_CHANNEL_VREFINT;
    }
231 232 233 234 235 236 237 238
    #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;
    }
239 240 241 242
    #endif
    return channel;
}

243
STATIC bool is_adcx_channel(int channel) {
244
    #if defined(STM32F411xE)
245 246
    // 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;
247
    #elif defined(STM32F0) || defined(STM32F4) || defined(STM32F7)
248
    return IS_ADC_CHANNEL(channel);
Y
yn386 已提交
249 250 251 252
    #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]);
253
    #elif defined(STM32G0) || defined(STM32H7)
254
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
255
           || IS_ADC_CHANNEL(__HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
256
    #elif defined(STM32G4) || defined(STM32L4) || defined(STM32WB)
257 258 259 260
    ADC_HandleTypeDef handle;
    handle.Instance = ADCx;
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
           || IS_ADC_CHANNEL(&handle, __HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
261 262 263 264
    #elif defined(STM32H5)
    // The first argument to the IS_ADC_CHANNEL macro is unused.
    return __HAL_ADC_IS_CHANNEL_INTERNAL(channel)
           || IS_ADC_CHANNEL(NULL, __HAL_ADC_DECIMAL_NB_TO_CHANNEL(channel));
265
    #else
266
    #error Unsupported processor
267
    #endif
268 269
}

270
STATIC void adc_wait_for_eoc_or_timeout(ADC_HandleTypeDef *adcHandle, int32_t timeout) {
271
    uint32_t tickstart = HAL_GetTick();
Y
yn386 已提交
272
    #if defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
273
    while ((adcHandle->Instance->SR & ADC_FLAG_EOC) != ADC_FLAG_EOC) {
274
    #elif defined(STM32F0) || defined(STM32G0) || defined(STM32G4) || defined(STM32H5) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
275
    while (READ_BIT(adcHandle->Instance->ISR, ADC_FLAG_EOC) != ADC_FLAG_EOC) {
276
    #else
277
    #error Unsupported processor
278
        #endif
279
        if (((HAL_GetTick() - tickstart) > timeout)) {
280 281 282 283 284
            break; // timeout
        }
    }
}

285
STATIC void adcx_clock_enable(ADC_HandleTypeDef *adch) {
Y
yn386 已提交
286
    #if defined(STM32F0) || defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
287
    ADCx_CLK_ENABLE();
288 289 290
    #elif defined(STM32H7A3xx) || defined(STM32H7A3xxQ) || defined(STM32H7B3xx) || defined(STM32H7B3xxQ)
    __HAL_RCC_ADC12_CLK_ENABLE();
    __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_CLKP);
291 292
    #elif defined(STM32G0)
    __HAL_RCC_ADC_CLK_ENABLE();
293 294
    #elif defined(STM32G4)
    __HAL_RCC_ADC12_CLK_ENABLE();
295 296
    #elif defined(STM32H5)
    __HAL_RCC_ADC_CLK_ENABLE();
297
    #elif defined(STM32H7)
298 299 300 301 302
    if (adch->Instance == ADC3) {
        __HAL_RCC_ADC3_CLK_ENABLE();
    } else {
        __HAL_RCC_ADC12_CLK_ENABLE();
    }
303
    __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_CLKP);
304 305 306 307
    #elif defined(STM32L4) || defined(STM32WB)
    if (__HAL_RCC_GET_ADC_SOURCE() == RCC_ADCCLKSOURCE_NONE) {
        __HAL_RCC_ADC_CONFIG(RCC_ADCCLKSOURCE_SYSCLK);
    }
308
    __HAL_RCC_ADC_CLK_ENABLE();
309
    #else
310
    #error Unsupported processor
311
    #endif
312 313
}

314
STATIC void adcx_init_periph(ADC_HandleTypeDef *adch, uint32_t resolution) {
315
    adcx_clock_enable(adch);
316

317 318
    adch->Init.Resolution = resolution;
    adch->Init.ContinuousConvMode = DISABLE;
319
    adch->Init.DiscontinuousConvMode = DISABLE;
320
    #if !defined(STM32F0) && !defined(STM32G0)
321
    adch->Init.NbrOfDiscConversion = 0;
322 323
    #endif
    #if !defined(STM32F0)
324
    adch->Init.NbrOfConversion = 1;
325
    #endif
326 327 328
    adch->Init.EOCSelection = ADC_EOC_SINGLE_CONV;
    adch->Init.ExternalTrigConv = ADC_SOFTWARE_START;
    adch->Init.ExternalTrigConvEdge = ADC_EXTERNALTRIGCONVEDGE_NONE;
329
    #if defined(STM32F0)
330 331 332
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV4;        // 12MHz
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
333
    adch->Init.DMAContinuousRequests = DISABLE;
334
    adch->Init.SamplingTimeCommon = ADC_SAMPLETIME_55CYCLES_5;    // ~4uS
335
    #elif defined(STM32F4) || defined(STM32F7)
336 337 338
    adch->Init.ClockPrescaler = ADC_CLOCK_SYNC_PCLK_DIV2;
    adch->Init.ScanConvMode = DISABLE;
    adch->Init.DataAlign = ADC_DATAALIGN_RIGHT;
339 340
    adch->Init.DMAContinuousRequests = DISABLE;
    #elif defined(STM32H7)
341 342 343 344 345 346
    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;
347
    adch->Init.ConversionDataManagement = ADC_CONVERSIONDATA_DR;
Y
yn386 已提交
348 349 350 351 352 353
    #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;
354
    #elif defined(STM32G0) || defined(STM32G4) || defined(STM32H5) || defined(STM32L4) || defined(STM32WB)
355 356 357
    #if defined(STM32G4)
    adch->Init.ClockPrescaler = ADC_CLOCK_ASYNC_DIV16;
    #else
358
    adch->Init.ClockPrescaler = ADC_CLOCK_ASYNC_DIV1;
359
    #endif
360 361 362 363 364
    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;
365
    adch->Init.DMAContinuousRequests = DISABLE;
366 367 368 369 370
    #else
    #error Unsupported processor
    #endif

    HAL_ADC_Init(adch);
371 372 373 374

    #if defined(STM32H7)
    HAL_ADCEx_Calibration_Start(adch, ADC_CALIB_OFFSET, ADC_SINGLE_ENDED);
    #endif
375 376
    #if defined(STM32G0)
    HAL_ADCEx_Calibration_Start(adch);
377
    #elif defined(STM32G4) || defined(STM32H5) || defined(STM32L4) || defined(STM32WB)
378 379
    HAL_ADCEx_Calibration_Start(adch, ADC_SINGLE_ENDED);
    #endif
380 381
}

382
STATIC void adc_init_single(pyb_obj_adc_t *adc_obj) {
383
    adc_obj->handle.Instance = ADCx;
384
    adcx_init_periph(&adc_obj->handle, ADC_RESOLUTION_12B);
385

386
    #if (defined(STM32G4) || defined(STM32L4)) && defined(ADC_DUALMODE_REGSIMULT_INJECSIMULT)
387 388
    ADC_MultiModeTypeDef multimode;
    multimode.Mode = ADC_MODE_INDEPENDENT;
389
    if (HAL_ADCEx_MultiModeConfigChannel(&adc_obj->handle, &multimode) != HAL_OK) {
390
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("Can not set multimode on ADC1 channel: %d"), adc_obj->channel);
391
    }
392
    #endif
393
}
D
Dave Hylands 已提交
394

395
STATIC void adc_config_channel(ADC_HandleTypeDef *adc_handle, uint32_t channel) {
D
Dave Hylands 已提交
396 397
    ADC_ChannelConfTypeDef sConfig;

398
    #if defined(STM32G0) || defined(STM32G4) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
399 400 401 402 403
    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 已提交
404
    sConfig.Rank = 1;
405 406 407
    #endif
    sConfig.Channel = channel;

408
    #if defined(STM32F0)
409
    sConfig.SamplingTime = ADC_SAMPLETIME_55CYCLES_5;
410
    #elif defined(STM32F4) || defined(STM32F7)
411 412 413 414 415
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_480CYCLES;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_15CYCLES;
    }
416
    #elif defined(STM32H7)
417
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
418
        sConfig.SamplingTime = ADC_SAMPLETIME_810CYCLES_5;
419 420 421
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_8CYCLES_5;
    }
422 423 424 425
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
    sConfig.OffsetRightShift = DISABLE;
    sConfig.OffsetSignedSaturation = DISABLE;
Y
yn386 已提交
426 427 428 429 430 431
    #elif defined(STM32L1)
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_384CYCLES;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_384CYCLES;
    }
432 433 434 435 436 437
    #elif defined(STM32G0)
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
        sConfig.SamplingTime = ADC_SAMPLETIME_160CYCLES_5;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_12CYCLES_5;
    }
438
    #elif defined(STM32G4) || defined(STM32H5) || defined(STM32L4) || defined(STM32WB)
439
    if (__HAL_ADC_IS_CHANNEL_INTERNAL(channel)) {
440 441 442 443
        sConfig.SamplingTime = ADC_SAMPLETIME_247CYCLES_5;
    } else {
        sConfig.SamplingTime = ADC_SAMPLETIME_12CYCLES_5;
    }
444 445
    sConfig.SingleDiff = ADC_SINGLE_ENDED;
    sConfig.OffsetNumber = ADC_OFFSET_NONE;
446
    sConfig.Offset = 0;
447
    #else
448
    #error Unsupported processor
449
    #endif
D
Dave Hylands 已提交
450

451 452 453 454 455 456
    #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

457
    HAL_ADC_ConfigChannel(adc_handle, &sConfig);
D
Dave Hylands 已提交
458 459
}

460
STATIC uint32_t adc_read_channel(ADC_HandleTypeDef *adcHandle) {
461 462 463 464 465 466 467 468 469 470 471 472
    uint32_t value;
    #if defined(STM32G4)
    // For STM32G4 there is errata 2.7.7, "Wrong ADC result if conversion done late after
    // calibration or previous conversion".  According to the errata, this can be avoided
    // by performing two consecutive ADC conversions and keeping the second result.
    for (uint8_t i = 0; i < 2; i++)
    #endif
    {
        HAL_ADC_Start(adcHandle);
        adc_wait_for_eoc_or_timeout(adcHandle, EOC_TIMEOUT);
        value = adcHandle->Instance->DR;
    }
D
Dave Hylands 已提交
473
    HAL_ADC_Stop(adcHandle);
474 475
    return value;
}
D
Dave Hylands 已提交
476

477 478
STATIC uint32_t adc_config_and_read_channel(ADC_HandleTypeDef *adcHandle, uint32_t channel) {
    adc_config_channel(adcHandle, channel);
479 480 481 482 483 484 485
    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.
486 487 488
    // 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);
489 490

    return raw_value;
D
Dave Hylands 已提交
491 492 493
}

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

496
STATIC void adc_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
497
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
498 499
    mp_print_str(print, "<ADC on ");
    mp_obj_print_helper(print, self->pin_name, PRINT_STR);
500
    mp_printf(print, " channel=%u>", self->channel);
D
Dave Hylands 已提交
501 502
}

503 504 505
/// \classmethod \constructor(pin)
/// Create an ADC object associated with the given pin.
/// This allows you to then read analog values on that pin.
506
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) {
507
    // check number of arguments
D
Damien George 已提交
508
    mp_arg_check_num(n_args, n_kw, 1, 1, false);
D
Dave Hylands 已提交
509

510 511
    // 1st argument is the pin name
    mp_obj_t pin_obj = args[0];
D
Dave Hylands 已提交
512 513 514

    uint32_t channel;

515
    if (mp_obj_is_int(pin_obj)) {
516
        channel = adc_get_internal_channel(mp_obj_get_int(pin_obj));
D
Dave Hylands 已提交
517
    } else {
518
        const pin_obj_t *pin = pin_find(pin_obj);
519
        if ((pin->adc_num & PIN_ADC_MASK) == 0) {
520 521
            // 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 已提交
522 523 524 525
        }
        channel = pin->adc_channel;
    }

526
    if (!is_adcx_channel(channel)) {
527
        mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("not a valid ADC Channel: %d"), channel);
D
Dave Hylands 已提交
528
    }
529

530 531 532 533 534
    // 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);
535
        }
D
Dave Hylands 已提交
536 537 538 539
    }

    pyb_obj_adc_t *o = m_new_obj(pyb_obj_adc_t);
    memset(o, 0, sizeof(*o));
540
    o->base.type = &pyb_adc_type;
D
Dave Hylands 已提交
541 542 543 544
    o->pin_name = pin_obj;
    o->channel = channel;
    adc_init_single(o);

545
    return MP_OBJ_FROM_PTR(o);
D
Dave Hylands 已提交
546 547
}

548 549 550
/// \method read()
/// Read the value on the analog pin and return it.  The returned value
/// will be between 0 and 4095.
551
STATIC mp_obj_t adc_read(mp_obj_t self_in) {
552
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
553
    return mp_obj_new_int(adc_config_and_read_channel(&self->handle, self->channel));
554 555 556
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_read_obj, adc_read);

557
/// \method read_timed(buf, timer)
558
///
559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581
/// 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:
582 583 584 585 586 587 588 589 590
///
///     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.
591
STATIC mp_obj_t adc_read_timed(mp_obj_t self_in, mp_obj_t buf_in, mp_obj_t freq_in) {
592
    pyb_obj_adc_t *self = MP_OBJ_TO_PTR(self_in);
593

594 595
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_WRITE);
596
    size_t typesize = mp_binary_get_size('@', bufinfo.typecode, NULL);
597

598 599 600 601 602 603 604 605 606 607 608 609
    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);
    }
610

611
    // configure the ADC channel
612
    adc_config_channel(&self->handle, self->channel);
613 614 615 616

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

617 618
    uint nelems = bufinfo.len / typesize;
    for (uint index = 0; index < nelems; index++) {
619
        // Wait for the timer to trigger so we sample at the correct frequency
620
        while (__HAL_TIM_GET_FLAG(tim, TIM_FLAG_UPDATE) == RESET) {
621
        }
622
        __HAL_TIM_CLEAR_FLAG(tim, TIM_FLAG_UPDATE);
623 624 625 626 627 628

        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 已提交
629
            #if defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
630
            self->handle.Instance->CR2 |= (uint32_t)ADC_CR2_SWSTART;
631
            #elif defined(STM32F0) || defined(STM32G0) || defined(STM32G4) || defined(STM32H5) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
632
            SET_BIT(self->handle.Instance->CR, ADC_CR_ADSTART);
633
            #else
634
            #error Unsupported processor
635
            #endif
636 637 638
        }

        // wait for sample to complete
639
        adc_wait_for_eoc_or_timeout(&self->handle, EOC_TIMEOUT);
640 641

        // read value
642
        uint value = self->handle.Instance->DR;
643 644

        // store value in buffer
645 646 647 648
        if (typesize == 1) {
            value >>= 4;
        }
        mp_binary_set_val_array_from_int(bufinfo.typecode, bufinfo.buf, index, value);
649 650
    }

651 652 653
    // turn the ADC off
    HAL_ADC_Stop(&self->handle);

654 655 656 657 658 659
    #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
660 661 662 663 664

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

665 666 667 668 669 670 671 672 673 674 675 676 677 678 679
// 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) {
680
        mp_raise_ValueError(MP_ERROR_TEXT("need at least 1 ADC"));
681 682
    }
    if (nadcs != nbufs) {
683
        mp_raise_ValueError(MP_ERROR_TEXT("length of ADC and buffer lists differ"));
684 685 686 687 688 689
    }

    // 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);
690
    void *bufptrs[nbufs];
691 692 693 694
    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)) {
695
            mp_raise_ValueError(MP_ERROR_TEXT("size and type of buffers must match"));
696
        }
697
        bufptrs[array_index] = bufinfo_curr.buf;
698 699 700 701 702 703 704
    }

    // 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
705
    pyb_obj_adc_t *adc0 = MP_OBJ_TO_PTR(adc_array[0]);
706 707 708
    adc_config_channel(&adc0->handle, adc0->channel);
    HAL_ADC_Start(&adc0->handle);
    // Wait for sample to complete and discard
709
    adc_wait_for_eoc_or_timeout(&adc0->handle, EOC_TIMEOUT);
710
    // Read (and discard) value
711
    uint value = adc0->handle.Instance->DR;
712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733

    // 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++) {
734
            pyb_obj_adc_t *adc = MP_OBJ_TO_PTR(adc_array[array_index]);
735 736 737 738
            // 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 已提交
739
            #if defined(STM32F4) || defined(STM32F7) || defined(STM32L1)
740
            adc->handle.Instance->CR2 |= (uint32_t)ADC_CR2_SWSTART;
741
            #elif defined(STM32F0) || defined(STM32G0) || defined(STM32G4) || defined(STM32H5) || defined(STM32H7) || defined(STM32L4) || defined(STM32WB)
742
            SET_BIT(adc->handle.Instance->CR, ADC_CR_ADSTART);
743 744 745 746
            #else
            #error Unsupported processor
            #endif
            // wait for sample to complete
747
            adc_wait_for_eoc_or_timeout(&adc->handle, EOC_TIMEOUT);
748 749

            // read value
750
            value = adc->handle.Instance->DR;
751 752 753 754 755

            // store values in buffer
            if (typesize == 1) {
                value >>= 4;
            }
756
            mp_binary_set_val_array_from_int(bufinfo.typecode, bufptrs[array_index], elem_index, value);
757 758 759 760
        }
    }

    // Turn the ADC off
761
    adc0 = MP_OBJ_TO_PTR(adc_array[0]);
762 763 764 765 766 767 768
    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));

769 770 771
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) },
772
    { MP_ROM_QSTR(MP_QSTR_read_timed_multi), MP_ROM_PTR(&adc_read_timed_multi_obj) },
773 774
};

775 776
STATIC MP_DEFINE_CONST_DICT(adc_locals_dict, adc_locals_dict_table);

777 778 779 780
MP_DEFINE_CONST_OBJ_TYPE(
    pyb_adc_type,
    MP_QSTR_ADC,
    MP_TYPE_FLAG_NONE,
781
    make_new, adc_make_new,
782
    print, adc_print,
783
    locals_dict, &adc_locals_dict
784
    );
D
Dave Hylands 已提交
785 786 787 788

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

789
typedef struct _pyb_adc_all_obj_t {
D
Dave Hylands 已提交
790 791
    mp_obj_base_t base;
    ADC_HandleTypeDef handle;
792
} pyb_adc_all_obj_t;
D
Dave Hylands 已提交
793

Y
yn386 已提交
794 795
float adc_read_core_vref(ADC_HandleTypeDef *adcHandle);

796
void adc_init_all(pyb_adc_all_obj_t *adc_all, uint32_t resolution, uint32_t en_mask) {
D
Dave Hylands 已提交
797 798

    switch (resolution) {
799
        #if !defined(STM32H7)
800 801 802
        case 6:
            resolution = ADC_RESOLUTION_6B;
            break;
803
        #endif
804 805 806 807 808 809 810 811 812
        case 8:
            resolution = ADC_RESOLUTION_8B;
            break;
        case 10:
            resolution = ADC_RESOLUTION_10B;
            break;
        case 12:
            resolution = ADC_RESOLUTION_12B;
            break;
813
        #if defined(STM32H7)
814 815 816
        case 16:
            resolution = ADC_RESOLUTION_16B;
            break;
817
        #endif
D
Dave Hylands 已提交
818
        default:
819
            mp_raise_msg_varg(&mp_type_ValueError, MP_ERROR_TEXT("resolution %d not supported"), resolution);
D
Dave Hylands 已提交
820 821
    }

822
    for (uint32_t channel = 0; channel < MP_ARRAY_SIZE(pin_adcall_table); ++channel) {
823 824
        // only initialise those channels that are selected with the en_mask
        if (en_mask & (1 << channel)) {
825
            // If this channel corresponds to a pin then configure the pin in ADC mode.
826
            const pin_obj_t *pin = pin_adcall_table[channel];
827
            if (pin) {
828
                mp_hal_pin_config(pin, MP_HAL_PIN_MODE_ADC, MP_HAL_PIN_PULL_NONE, 0);
829 830
            }
        }
D
Dave Hylands 已提交
831 832
    }

833
    adc_all->handle.Instance = ADCALLx;
834
    adcx_init_periph(&adc_all->handle, resolution);
D
Dave Hylands 已提交
835 836
}

D
Dave Hylands 已提交
837
int adc_get_resolution(ADC_HandleTypeDef *adcHandle) {
Y
yn386 已提交
838 839 840
    #if defined(STM32L1)
    uint32_t res_reg = adcHandle->Instance->CR1 & ADC_CR1_RES_Msk;
    #else
841
    uint32_t res_reg = ADC_GET_RESOLUTION(adcHandle);
Y
yn386 已提交
842
    #endif
D
Dave Hylands 已提交
843 844

    switch (res_reg) {
845
        #if !defined(STM32H7)
846 847
        case ADC_RESOLUTION_6B:
            return 6;
848
        #endif
849 850 851 852
        case ADC_RESOLUTION_8B:
            return 8;
        case ADC_RESOLUTION_10B:
            return 10;
853
        #if defined(STM32H7)
854 855
        case ADC_RESOLUTION_16B:
            return 16;
856
        #endif
D
Dave Hylands 已提交
857 858 859 860
    }
    return 12;
}

861 862 863 864 865
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 已提交
866

867
int adc_read_core_temp(ADC_HandleTypeDef *adcHandle) {
868 869 870 871 872 873 874 875
    #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
876
    int32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR);
877
    #endif
D
Dave Hylands 已提交
878 879 880
    return ((raw_value - CORE_TEMP_V25) / CORE_TEMP_AVG_SLOPE) + 25;
}

D
Dave Hylands 已提交
881
#if MICROPY_PY_BUILTINS_FLOAT
882 883 884
// correction factor for reference value
STATIC volatile float adc_refcor = 1.0f;

885
float adc_read_core_temp_float(ADC_HandleTypeDef *adcHandle) {
886 887 888 889 890 891
    #if defined(STM32G4) || defined(STM32L1) || defined(STM32L4)
    // Update the reference correction factor before reading tempsensor
    // because TS_CAL1 and TS_CAL2 of STM32G4,L1/L4 are at VDDA=3.0V
    adc_read_core_vref(adcHandle);
    #endif

892 893 894 895 896 897 898
    #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;
    }
899
    float core_temp_avg_slope = (*ADC_CAL2 - *ADC_CAL1) / 100.0f;
900
    #else
901
    int32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_TEMPSENSOR);
902
    float core_temp_avg_slope = (*ADC_CAL2 - *ADC_CAL1) / 80.0f;
903
    #endif
904 905 906
    return (((float)raw_value * adc_refcor - *ADC_CAL1) / core_temp_avg_slope) + 30.0f;
}

907
float adc_read_core_vbat(ADC_HandleTypeDef *adcHandle) {
Y
yn386 已提交
908 909 910
    #if defined(STM32L152xE)
    mp_raise_NotImplementedError(MP_ERROR_TEXT("read_core_vbat not supported"));
    #else
911
    uint32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_VBAT);
912
    return raw_value * VBAT_DIV * ADC_SCALE * adc_refcor;
Y
yn386 已提交
913
    #endif
D
Dave Hylands 已提交
914 915
}

916
float adc_read_core_vref(ADC_HandleTypeDef *adcHandle) {
917
    uint32_t raw_value = adc_config_and_read_ref(adcHandle, ADC_CHANNEL_VREFINT);
D
Dave Hylands 已提交
918

919 920 921 922
    // update the reference correction factor
    adc_refcor = ((float)(*VREFIN_CAL)) / ((float)raw_value);

    return (*VREFIN_CAL) * ADC_SCALE;
D
Dave Hylands 已提交
923
}
D
Dave Hylands 已提交
924
#endif
D
Dave Hylands 已提交
925 926

/******************************************************************************/
927
/* MicroPython bindings : adc_all object                                      */
D
Dave Hylands 已提交
928

929
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) {
930
    // check number of arguments
931
    mp_arg_check_num(n_args, n_kw, 1, 2, false);
932 933

    // make ADCAll object
934
    pyb_adc_all_obj_t *o = mp_obj_malloc(pyb_adc_all_obj_t, &pyb_adc_all_type);
935 936 937
    mp_int_t res = mp_obj_get_int(args[0]);
    uint32_t en_mask = 0xffffffff;
    if (n_args > 1) {
938
        en_mask = mp_obj_get_int(args[1]);
939 940
    }
    adc_init_all(o, res, en_mask);
941

942
    return MP_OBJ_FROM_PTR(o);
D
Dave Hylands 已提交
943 944
}

945
STATIC mp_obj_t adc_all_read_channel(mp_obj_t self_in, mp_obj_t channel) {
946
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
947
    uint32_t chan = adc_get_internal_channel(mp_obj_get_int(channel));
948 949 950
    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 已提交
951 952 953
    uint32_t data = adc_config_and_read_channel(&self->handle, chan);
    return mp_obj_new_int(data);
}
954
STATIC MP_DEFINE_CONST_FUN_OBJ_2(adc_all_read_channel_obj, adc_all_read_channel);
D
Dave Hylands 已提交
955

956
STATIC mp_obj_t adc_all_read_core_temp(mp_obj_t self_in) {
957
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
958 959 960 961
    #if MICROPY_PY_BUILTINS_FLOAT
    float data = adc_read_core_temp_float(&self->handle);
    return mp_obj_new_float(data);
    #else
962
    int data = adc_read_core_temp(&self->handle);
D
Dave Hylands 已提交
963
    return mp_obj_new_int(data);
964
    #endif
D
Dave Hylands 已提交
965
}
966
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_temp_obj, adc_all_read_core_temp);
D
Dave Hylands 已提交
967

D
Dave Hylands 已提交
968
#if MICROPY_PY_BUILTINS_FLOAT
969
STATIC mp_obj_t adc_all_read_core_vbat(mp_obj_t self_in) {
970
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
D
Dave Hylands 已提交
971 972 973
    float data = adc_read_core_vbat(&self->handle);
    return mp_obj_new_float(data);
}
974
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vbat_obj, adc_all_read_core_vbat);
D
Dave Hylands 已提交
975

976
STATIC mp_obj_t adc_all_read_core_vref(mp_obj_t self_in) {
977
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
978
    float data = adc_read_core_vref(&self->handle);
D
Dave Hylands 已提交
979 980
    return mp_obj_new_float(data);
}
981
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_core_vref_obj, adc_all_read_core_vref);
982 983

STATIC mp_obj_t adc_all_read_vref(mp_obj_t self_in) {
984
    pyb_adc_all_obj_t *self = MP_OBJ_TO_PTR(self_in);
985
    adc_read_core_vref(&self->handle);
986
    return mp_obj_new_float(ADC_SCALE_V * adc_refcor);
987 988
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(adc_all_read_vref_obj, adc_all_read_vref);
D
Dave Hylands 已提交
989
#endif
D
Dave Hylands 已提交
990

991 992 993
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) },
994
    #if MICROPY_PY_BUILTINS_FLOAT
995 996 997
    { 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) },
998
    #endif
D
Dave Hylands 已提交
999 1000
};

1001 1002
STATIC MP_DEFINE_CONST_DICT(adc_all_locals_dict, adc_all_locals_dict_table);

1003 1004 1005 1006
MP_DEFINE_CONST_OBJ_TYPE(
    pyb_adc_all_type,
    MP_QSTR_ADCAll,
    MP_TYPE_FLAG_NONE,
1007
    make_new, adc_all_make_new,
1008
    locals_dict, &adc_all_locals_dict
1009
    );
1010 1011

#endif // MICROPY_HW_ENABLE_ADC