add some code
This commit is contained in:
@@ -0,0 +1,6 @@
|
||||
# The following lines of boilerplate have to be in your project's
|
||||
# CMakeLists in this exact order for cmake to work correctly
|
||||
cmake_minimum_required(VERSION 3.5)
|
||||
|
||||
include($ENV{IDF_PATH}/tools/cmake/project.cmake)
|
||||
project(fft_window)
|
||||
@@ -0,0 +1,134 @@
|
||||
# FFT Window Example
|
||||
|
||||
(See the README.md file in the upper level 'examples' directory for more information about examples.)
|
||||
|
||||
This example demonstrates how to use Window and FFT functionality from esp-dsp library. Example does the following steps:
|
||||
|
||||
1. Initialize the library
|
||||
2. Initialize input signals with 1024 samples
|
||||
3. Apply window to input signal.
|
||||
4. Calculate FFT for 1024 complex samples
|
||||
5. Apply bit reverse operation for output complex vector
|
||||
6. Split one complex FFT output spectrum to two real signal spectrums
|
||||
7. Show results on the plots
|
||||
|
||||
## How to use example
|
||||
|
||||
### Hardware required
|
||||
|
||||
This example does not require any special hardware, and can be run on any common development board.
|
||||
|
||||
### Configure the project
|
||||
|
||||
Under Component Config ---> DSP Library ---> DSP Optimization, it's possible to choose either the optimized or ANSI implementation, to compare them.
|
||||
|
||||
### Build and flash
|
||||
|
||||
Build the project and flash it to the board, then run monitor tool to view serial output (replace PORT with serial port name):
|
||||
|
||||
```
|
||||
idf.py -p PORT flash monitor
|
||||
```
|
||||
|
||||
(To exit the serial monitor, type ``Ctrl-]``.)
|
||||
|
||||
See the Getting Started Guide for full steps to configure and use ESP-IDF to build projects.
|
||||
|
||||
## Example output
|
||||
|
||||
Here is an typical example console output.
|
||||
|
||||
```
|
||||
I (128) main: Start Example.
|
||||
W (128) main: Hann Window
|
||||
I (128) view: Data min[256] = -inf, Data max[1] = 24.086628
|
||||
________________________________________________________________
|
||||
0| |
|
||||
1| |
|
||||
2| |
|
||||
3| |
|
||||
4| |
|
||||
5 | |
|
||||
6 | |
|
||||
7 ||||| |
|
||||
8 ||||||||||||||| |
|
||||
9 ||||||||||||||||||||||||||||||||||||||||||
|
||||
0123456789012345678901234567890123456789012345678901234567890123
|
||||
I (138) view: Plot: Length=512, min=-120.000000, max=40.000000
|
||||
W (138) main: Blackman Window
|
||||
I (148) view: Data min[355] = -165.295654, Data max[1] = 24.083012
|
||||
________________________________________________________________
|
||||
0| |
|
||||
1| |
|
||||
2| |
|
||||
3| |
|
||||
4| |
|
||||
5| |
|
||||
6 | |
|
||||
7 ||| |
|
||||
8 ||||||||| |
|
||||
9 |||||||||||||||||||||||||||||||||||||||||||||||||||
|
||||
0123456789012345678901234567890123456789012345678901234567890123
|
||||
I (158) view: Plot: Length=512, min=-120.000000, max=40.000000
|
||||
W (158) main: Blackman-Harris Window
|
||||
I (168) view: Data min[128] = -inf, Data max[1] = 23.874702
|
||||
________________________________________________________________
|
||||
0| |
|
||||
1| |
|
||||
2| |
|
||||
3| |
|
||||
4| |
|
||||
5| |
|
||||
6| |
|
||||
7|| |
|
||||
8| |||| |
|
||||
9 |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
||||
0123456789012345678901234567890123456789012345678901234567890123
|
||||
I (178) view: Plot: Length=512, min=-120.000000, max=40.000000
|
||||
W (178) main: Blackman-Nuttall Window
|
||||
I (188) view: Data min[128] = -inf, Data max[1] = 23.890663
|
||||
________________________________________________________________
|
||||
0| |
|
||||
1| |
|
||||
2| |
|
||||
3| |
|
||||
4| |
|
||||
5| |
|
||||
6| |
|
||||
7 || |
|
||||
8 |||| | |
|
||||
9 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
||||
0123456789012345678901234567890123456789012345678901234567890123
|
||||
I (198) view: Plot: Length=512, min=-120.000000, max=40.000000
|
||||
W (198) main: Nuttall Window
|
||||
I (208) view: Data min[203] = -175.147400, Data max[1] = 23.858671
|
||||
________________________________________________________________
|
||||
0| |
|
||||
1| |
|
||||
2| |
|
||||
3| |
|
||||
4| |
|
||||
5| |
|
||||
6| |
|
||||
7|| |
|
||||
8 ||| |
|
||||
9 ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
||||
0123456789012345678901234567890123456789012345678901234567890123
|
||||
I (218) view: Plot: Length=512, min=-120.000000, max=40.000000
|
||||
W (218) main: Flat-Top Window
|
||||
I (228) view: Data min[256] = -inf, Data max[1] = 22.490753
|
||||
________________________________________________________________
|
||||
0| |
|
||||
1| |
|
||||
2| |
|
||||
3| |
|
||||
4| |
|
||||
5| |
|
||||
6| |
|
||||
7 || |
|
||||
8 ||||| |
|
||||
9 |||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
||||
0123456789012345678901234567890123456789012345678901234567890123
|
||||
I (238) view: Plot: Length=512, min=-120.000000, max=40.000000
|
||||
I (238) main: End Example.
|
||||
```
|
||||
@@ -0,0 +1 @@
|
||||
idf_component_register(SRCS "dsps_window_main.c")
|
||||
@@ -0,0 +1,138 @@
|
||||
// Copyright 2018-2019 Espressif Systems (Shanghai) PTE LTD
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include "freertos/FreeRTOS.h"
|
||||
#include "freertos/task.h"
|
||||
#include "esp_system.h"
|
||||
#include "driver/spi_master.h"
|
||||
#include "soc/gpio_struct.h"
|
||||
#include "driver/gpio.h"
|
||||
#include "driver/uart.h"
|
||||
#include "soc/uart_struct.h"
|
||||
#include <math.h>
|
||||
|
||||
#include "esp_dsp.h"
|
||||
|
||||
static const char *TAG = "main";
|
||||
|
||||
// This example shows how to use FFT from esp-dsp library
|
||||
|
||||
#define N_SAMPLES 1024
|
||||
int N = N_SAMPLES;
|
||||
// Input test array
|
||||
__attribute__((aligned(16)))
|
||||
float x1[N_SAMPLES];
|
||||
// Window coefficients
|
||||
__attribute__((aligned(16)))
|
||||
float wind[N_SAMPLES];
|
||||
// working complex array
|
||||
__attribute__((aligned(16)))
|
||||
float y_cf[N_SAMPLES * 2];
|
||||
// Pointers to result arrays
|
||||
__attribute__((aligned(16)))
|
||||
float *y1_cf = &y_cf[0];
|
||||
|
||||
void process_and_show(float *data, int length)
|
||||
{
|
||||
dsps_fft2r_fc32(data, length);
|
||||
// Bit reverse
|
||||
dsps_bit_rev_fc32(data, length);
|
||||
// Convert one complex vector to two complex vectors
|
||||
dsps_cplx2reC_fc32(data, length);
|
||||
|
||||
for (int i = 0 ; i < length / 2 ; i++) {
|
||||
data[i] = 10 * log10f((data[i * 2 + 0] * data[i * 2 + 0] + data[i * 2 + 1] * data[i * 2 + 1]) / N);
|
||||
}
|
||||
|
||||
// Show power spectrum in 64x10 window from -100 to 0 dB from 0..N/4 samples
|
||||
dsps_view(data, length / 2, 64, 10, -120, 40, '|');
|
||||
|
||||
}
|
||||
|
||||
void app_main()
|
||||
{
|
||||
esp_err_t ret;
|
||||
ESP_LOGI(TAG, "Start Example.");
|
||||
ret = dsps_fft2r_init_fc32(NULL, CONFIG_DSP_MAX_FFT_SIZE);
|
||||
if (ret != ESP_OK) {
|
||||
ESP_LOGE(TAG, "Not possible to initialize FFT. Error = %i", ret);
|
||||
return;
|
||||
}
|
||||
|
||||
ESP_LOGW(TAG, "Hann Window");
|
||||
// Generate Hann window
|
||||
dsps_wind_hann_f32(wind, N);
|
||||
// Convert two input vectors to one complex vector
|
||||
for (int i = 0 ; i < N ; i++) {
|
||||
y_cf[i * 2 + 0] = wind[i];
|
||||
y_cf[i * 2 + 1] = 0;
|
||||
}
|
||||
process_and_show(y_cf, N);
|
||||
|
||||
ESP_LOGW(TAG, "Blackman Window");
|
||||
// Generate Blackman window
|
||||
dsps_wind_blackman_f32(wind, N);
|
||||
// Convert two input vectors to one complex vector
|
||||
for (int i = 0 ; i < N ; i++) {
|
||||
y_cf[i * 2 + 0] = wind[i];
|
||||
y_cf[i * 2 + 1] = 0;
|
||||
}
|
||||
process_and_show(y_cf, N);
|
||||
|
||||
ESP_LOGW(TAG, "Blackman-Harris Window");
|
||||
// Generate Blackman-Harris window
|
||||
dsps_wind_blackman_harris_f32(wind, N);
|
||||
// Convert two input vectors to one complex vector
|
||||
for (int i = 0 ; i < N ; i++) {
|
||||
y_cf[i * 2 + 0] = wind[i];
|
||||
y_cf[i * 2 + 1] = 0;
|
||||
}
|
||||
process_and_show(y_cf, N);
|
||||
|
||||
ESP_LOGW(TAG, "Blackman-Nuttall Window");
|
||||
// Generate Blackman-Nuttall window
|
||||
dsps_wind_blackman_nuttall_f32(wind, N);
|
||||
// Convert two input vectors to one complex vector
|
||||
for (int i = 0 ; i < N ; i++) {
|
||||
y_cf[i * 2 + 0] = wind[i];
|
||||
y_cf[i * 2 + 1] = 0;
|
||||
}
|
||||
process_and_show(y_cf, N);
|
||||
|
||||
ESP_LOGW(TAG, "Nuttall Window");
|
||||
// Generate Nuttall window
|
||||
dsps_wind_nuttall_f32(wind, N);
|
||||
// Convert two input vectors to one complex vector
|
||||
for (int i = 0 ; i < N ; i++) {
|
||||
y_cf[i * 2 + 0] = wind[i];
|
||||
y_cf[i * 2 + 1] = 0;
|
||||
}
|
||||
process_and_show(y_cf, N);
|
||||
|
||||
ESP_LOGW(TAG, "Flat-Top Window");
|
||||
// Generate Flat-Top window
|
||||
dsps_wind_flat_top_f32(wind, N);
|
||||
// Convert two input vectors to one complex vector
|
||||
for (int i = 0 ; i < N ; i++) {
|
||||
y_cf[i * 2 + 0] = wind[i];
|
||||
y_cf[i * 2 + 1] = 0;
|
||||
}
|
||||
process_and_show(y_cf, N);
|
||||
|
||||
ESP_LOGI(TAG, "End Example.");
|
||||
}
|
||||
@@ -0,0 +1,4 @@
|
||||
dependencies:
|
||||
espressif/esp-dsp:
|
||||
override_path: "../../../"
|
||||
version: "*"
|
||||
@@ -0,0 +1 @@
|
||||
CONFIG_PARTITION_TABLE_OFFSET=0x9000
|
||||
Reference in New Issue
Block a user