395 lines
10 KiB
C++
395 lines
10 KiB
C++
/*
|
|
* Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
|
|
*
|
|
* This file is part of PortaPack.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, Inc., 51 Franklin Street,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include "ui_spectrum.hpp"
|
|
|
|
#include "spectrum_color_lut.hpp"
|
|
|
|
#include "portapack.hpp"
|
|
using namespace portapack;
|
|
|
|
#include "baseband_api.hpp"
|
|
|
|
#include "string_format.hpp"
|
|
|
|
#include <cmath>
|
|
#include <array>
|
|
|
|
namespace ui {
|
|
namespace spectrum {
|
|
|
|
/* AudioSpectrumView******************************************************/
|
|
|
|
AudioSpectrumView::AudioSpectrumView(
|
|
const Rect parent_rect
|
|
) : View { parent_rect }
|
|
{
|
|
set_focusable(true);
|
|
|
|
add_children({
|
|
&labels,
|
|
&field_frequency,
|
|
&waveform
|
|
});
|
|
|
|
field_frequency.on_change = [this](int32_t) {
|
|
set_dirty();
|
|
};
|
|
field_frequency.set_value(0);
|
|
}
|
|
|
|
void AudioSpectrumView::paint(Painter& painter) {
|
|
const auto r = screen_rect();
|
|
|
|
painter.fill_rectangle(r, Color::black());
|
|
|
|
//if( !spectrum_sampling_rate ) return;
|
|
|
|
// Cursor
|
|
const Rect r_cursor {
|
|
field_frequency.value() / (48000 / 240), r.bottom() - 32 - cursor_band_height,
|
|
1, cursor_band_height
|
|
};
|
|
painter.fill_rectangle(
|
|
r_cursor,
|
|
Color::red()
|
|
);
|
|
}
|
|
|
|
void AudioSpectrumView::on_audio_spectrum(const AudioSpectrum* spectrum) {
|
|
for (size_t i = 0; i < spectrum->db.size(); i++)
|
|
audio_spectrum[i] = ((int16_t)spectrum->db[i] - 127) * 256;
|
|
waveform.set_dirty();
|
|
}
|
|
|
|
/* FrequencyScale ********************************************************/
|
|
|
|
void FrequencyScale::on_show() {
|
|
clear();
|
|
}
|
|
|
|
void FrequencyScale::set_spectrum_sampling_rate(const int new_sampling_rate) {
|
|
if( (spectrum_sampling_rate != new_sampling_rate) ) {
|
|
spectrum_sampling_rate = new_sampling_rate;
|
|
set_dirty();
|
|
}
|
|
}
|
|
|
|
void FrequencyScale::set_channel_filter(
|
|
const int low_frequency,
|
|
const int high_frequency,
|
|
const int transition
|
|
) {
|
|
if( (channel_filter_low_frequency != low_frequency) ||
|
|
(channel_filter_high_frequency != high_frequency) ||
|
|
(channel_filter_transition != transition) ) {
|
|
channel_filter_low_frequency = low_frequency;
|
|
channel_filter_high_frequency = high_frequency;
|
|
channel_filter_transition = transition;
|
|
set_dirty();
|
|
}
|
|
}
|
|
|
|
void FrequencyScale::paint(Painter& painter) {
|
|
const auto r = screen_rect();
|
|
|
|
clear_background(painter, r);
|
|
|
|
if( !spectrum_sampling_rate ) {
|
|
// Can't draw without non-zero scale.
|
|
return;
|
|
}
|
|
|
|
draw_filter_ranges(painter, r);
|
|
draw_frequency_ticks(painter, r);
|
|
|
|
if (_blink) {
|
|
const Rect r_cursor {
|
|
118 + cursor_position, r.bottom() - filter_band_height,
|
|
5, filter_band_height
|
|
};
|
|
painter.fill_rectangle(
|
|
r_cursor,
|
|
Color::red()
|
|
);
|
|
}
|
|
}
|
|
|
|
void FrequencyScale::clear() {
|
|
spectrum_sampling_rate = 0;
|
|
set_dirty();
|
|
}
|
|
|
|
void FrequencyScale::clear_background(Painter& painter, const Rect r) {
|
|
painter.fill_rectangle(r, Color::black());
|
|
}
|
|
|
|
void FrequencyScale::draw_frequency_ticks(Painter& painter, const Rect r) {
|
|
const auto x_center = r.width() / 2;
|
|
|
|
const Rect tick { r.left() + x_center, r.top(), 1, r.height() };
|
|
painter.fill_rectangle(tick, Color::white());
|
|
|
|
constexpr int tick_count_max = 4;
|
|
float rough_tick_interval = float(spectrum_sampling_rate) / tick_count_max;
|
|
int magnitude = 1;
|
|
int magnitude_n = 0;
|
|
while(rough_tick_interval >= 10.0f) {
|
|
rough_tick_interval /= 10;
|
|
magnitude *= 10;
|
|
magnitude_n += 1;
|
|
}
|
|
const int tick_interval = std::ceil(rough_tick_interval);
|
|
|
|
auto tick_offset = tick_interval;
|
|
while((tick_offset * magnitude) < spectrum_sampling_rate / 2) {
|
|
const Dim pixel_offset = tick_offset * magnitude * spectrum_bins / spectrum_sampling_rate;
|
|
|
|
const std::string zero_pad =
|
|
((magnitude_n % 3) == 0) ? "" :
|
|
((magnitude_n % 3) == 1) ? "0" : "00";
|
|
const std::string unit =
|
|
(magnitude_n >= 6) ? "M" :
|
|
(magnitude_n >= 3) ? "k" : "";
|
|
const std::string label = to_string_dec_uint(tick_offset) + zero_pad + unit;
|
|
const auto label_width = style().font.size_of(label).width();
|
|
|
|
const Coord offset_low = r.left() + x_center - pixel_offset;
|
|
const Rect tick_low { offset_low, r.top(), 1, r.height() };
|
|
painter.fill_rectangle(tick_low, Color::white());
|
|
painter.draw_string({ offset_low + 2, r.top() }, style(), label );
|
|
|
|
const Coord offset_high = r.left() + x_center + pixel_offset;
|
|
const Rect tick_high { offset_high, r.top(), 1, r.height() };
|
|
painter.fill_rectangle(tick_high, Color::white());
|
|
painter.draw_string({ offset_high - 2 - label_width, r.top() }, style(), label );
|
|
|
|
tick_offset += tick_interval;
|
|
}
|
|
}
|
|
|
|
void FrequencyScale::draw_filter_ranges(Painter& painter, const Rect r) {
|
|
if( channel_filter_low_frequency != channel_filter_high_frequency ) {
|
|
const auto x_center = r.width() / 2;
|
|
|
|
const auto x_low = x_center + channel_filter_low_frequency * spectrum_bins / spectrum_sampling_rate;
|
|
const auto x_high = x_center + channel_filter_high_frequency * spectrum_bins / spectrum_sampling_rate;
|
|
|
|
if( channel_filter_transition ) {
|
|
const auto trans = channel_filter_transition * spectrum_bins / spectrum_sampling_rate;
|
|
|
|
const Rect r_all {
|
|
r.left() + x_low - trans, r.bottom() - filter_band_height,
|
|
x_high - x_low + trans*2, filter_band_height
|
|
};
|
|
painter.fill_rectangle(
|
|
r_all,
|
|
Color::yellow()
|
|
);
|
|
}
|
|
|
|
const Rect r_pass {
|
|
r.left() + x_low, r.bottom() - filter_band_height,
|
|
x_high - x_low, filter_band_height
|
|
};
|
|
painter.fill_rectangle(
|
|
r_pass,
|
|
Color::green()
|
|
);
|
|
}
|
|
}
|
|
|
|
void FrequencyScale::on_focus() {
|
|
_blink = true;
|
|
on_tick_second();
|
|
signal_token_tick_second = rtc_time::signal_tick_second += [this]() {
|
|
this->on_tick_second();
|
|
};
|
|
}
|
|
|
|
void FrequencyScale::on_blur() {
|
|
rtc_time::signal_tick_second -= signal_token_tick_second;
|
|
_blink = false;
|
|
set_dirty();
|
|
}
|
|
|
|
bool FrequencyScale::on_encoder(const EncoderEvent delta) {
|
|
cursor_position += delta;
|
|
|
|
cursor_position = std::min<int32_t>(cursor_position, 119);
|
|
cursor_position = std::max<int32_t>(cursor_position, -120);
|
|
|
|
set_dirty();
|
|
|
|
return true;
|
|
}
|
|
|
|
bool FrequencyScale::on_key(const KeyEvent key) {
|
|
if( key == KeyEvent::Select ) {
|
|
if( on_select ) {
|
|
on_select((cursor_position * spectrum_sampling_rate) / 240);
|
|
cursor_position = 0;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void FrequencyScale::on_tick_second() {
|
|
set_dirty();
|
|
_blink = !_blink;
|
|
}
|
|
|
|
/* WaterfallView *********************************************************/
|
|
|
|
void WaterfallView::on_show() {
|
|
clear();
|
|
|
|
const auto screen_r = screen_rect();
|
|
display.scroll_set_area(screen_r.top(), screen_r.bottom());
|
|
}
|
|
|
|
void WaterfallView::on_hide() {
|
|
/* TODO: Clear region to eliminate brief flash of content at un-shifted
|
|
* position?
|
|
*/
|
|
display.scroll_disable();
|
|
}
|
|
|
|
void WaterfallView::paint(Painter& painter) {
|
|
// Do nothing.
|
|
(void)painter;
|
|
}
|
|
|
|
void WaterfallView::on_channel_spectrum(
|
|
const ChannelSpectrum& spectrum
|
|
) {
|
|
/* TODO: static_assert that message.spectrum.db.size() >= pixel_row.size() */
|
|
|
|
std::array<Color, 240> pixel_row;
|
|
for(size_t i=0; i<120; i++) {
|
|
const auto pixel_color = spectrum_rgb3_lut[spectrum.db[256 - 120 + i]];
|
|
pixel_row[i] = pixel_color;
|
|
}
|
|
|
|
for(size_t i=120; i<240; i++) {
|
|
const auto pixel_color = spectrum_rgb3_lut[spectrum.db[i - 120]];
|
|
pixel_row[i] = pixel_color;
|
|
}
|
|
|
|
const auto draw_y = display.scroll(1);
|
|
|
|
display.draw_pixels(
|
|
{ { 0, draw_y }, { pixel_row.size(), 1 } },
|
|
pixel_row
|
|
);
|
|
}
|
|
|
|
void WaterfallView::clear() {
|
|
display.fill_rectangle(
|
|
screen_rect(),
|
|
Color::black()
|
|
);
|
|
}
|
|
|
|
/* WaterfallWidget *******************************************************/
|
|
|
|
WaterfallWidget::WaterfallWidget(const bool cursor) {
|
|
add_children({
|
|
&waterfall_view,
|
|
&frequency_scale
|
|
});
|
|
|
|
frequency_scale.set_focusable(cursor);
|
|
|
|
// Making the event climb up all the way up to here kinda sucks
|
|
frequency_scale.on_select = [this](int32_t offset) {
|
|
if (on_select) on_select(offset);
|
|
};
|
|
}
|
|
|
|
void WaterfallWidget::on_show() {
|
|
baseband::spectrum_streaming_start();
|
|
}
|
|
|
|
void WaterfallWidget::on_hide() {
|
|
baseband::spectrum_streaming_stop();
|
|
}
|
|
|
|
void WaterfallWidget::show_audio_spectrum_view(const bool show) {
|
|
if ((audio_spectrum_view && show) || (!audio_spectrum_view && !show)) return;
|
|
|
|
if (show) {
|
|
audio_spectrum_view = std::make_unique<AudioSpectrumView>(audio_spectrum_view_rect);
|
|
add_child(audio_spectrum_view.get());
|
|
update_widgets_rect();
|
|
} else {
|
|
audio_spectrum_update = false;
|
|
remove_child(audio_spectrum_view.get());
|
|
audio_spectrum_view.reset();
|
|
update_widgets_rect();
|
|
}
|
|
}
|
|
|
|
void WaterfallWidget::update_widgets_rect() {
|
|
if (audio_spectrum_view) {
|
|
frequency_scale.set_parent_rect({ 0, audio_spectrum_height, screen_rect().width(), scale_height });
|
|
waterfall_view.set_parent_rect(waterfall_reduced_rect);
|
|
} else {
|
|
frequency_scale.set_parent_rect({ 0, 0, screen_rect().width(), scale_height });
|
|
waterfall_view.set_parent_rect(waterfall_normal_rect);
|
|
}
|
|
waterfall_view.on_show();
|
|
}
|
|
|
|
void WaterfallWidget::set_parent_rect(const Rect new_parent_rect) {
|
|
View::set_parent_rect(new_parent_rect);
|
|
|
|
waterfall_normal_rect = { 0, scale_height, new_parent_rect.width(), new_parent_rect.height() - scale_height};
|
|
waterfall_reduced_rect = { 0, audio_spectrum_height + scale_height, new_parent_rect.width(), new_parent_rect.height() - scale_height - audio_spectrum_height };
|
|
|
|
update_widgets_rect();
|
|
}
|
|
|
|
void WaterfallWidget::paint(Painter& painter) {
|
|
// TODO:
|
|
(void)painter;
|
|
}
|
|
|
|
void WaterfallWidget::on_channel_spectrum(const ChannelSpectrum& spectrum) {
|
|
waterfall_view.on_channel_spectrum(spectrum);
|
|
sampling_rate = spectrum.sampling_rate;
|
|
frequency_scale.set_spectrum_sampling_rate(sampling_rate);
|
|
frequency_scale.set_channel_filter(
|
|
spectrum.channel_filter_low_frequency,
|
|
spectrum.channel_filter_high_frequency,
|
|
spectrum.channel_filter_transition
|
|
);
|
|
}
|
|
|
|
void WaterfallWidget::on_audio_spectrum() {
|
|
audio_spectrum_view->on_audio_spectrum(audio_spectrum_data);
|
|
}
|
|
|
|
} /* namespace spectrum */
|
|
} /* namespace ui */
|