124 lines
3.5 KiB
C++
124 lines
3.5 KiB
C++
|
/* -*- c++ -*- */
|
||
|
/*
|
||
|
* gr-satnogs: SatNOGS GNU Radio Out-Of-Tree Module
|
||
|
*
|
||
|
* Copyright (C) 2018, Libre Space Foundation <http://librespacefoundation.org/>
|
||
|
*
|
||
|
* 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 3 of the License, 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. If not, see <http://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
|
||
|
#ifdef HAVE_CONFIG_H
|
||
|
#include "config.h"
|
||
|
#endif
|
||
|
|
||
|
#include <gnuradio/io_signature.h>
|
||
|
#include "sstv_pd120_sink_impl.h"
|
||
|
|
||
|
namespace gr {
|
||
|
namespace satnogs {
|
||
|
|
||
|
const size_t image_width = 640;
|
||
|
const size_t sync_length = 105;
|
||
|
const size_t sync_thresh = 100;
|
||
|
const size_t porch_length = 10;
|
||
|
const size_t line_length = sync_length + porch_length + 4 * image_width;
|
||
|
|
||
|
const float max_dev = 600;
|
||
|
const float center = 1750;
|
||
|
const float min_freq = 1200;
|
||
|
const float max_freq = 2300;
|
||
|
|
||
|
const float color_low = 1500;
|
||
|
const float color_high = max_freq;
|
||
|
|
||
|
|
||
|
sstv_pd120_sink::sptr
|
||
|
sstv_pd120_sink::make(const char *filename_png, bool split)
|
||
|
{
|
||
|
return gnuradio::get_initial_sptr
|
||
|
(new sstv_pd120_sink_impl(filename_png, split));
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* The private constructor
|
||
|
*/
|
||
|
sstv_pd120_sink_impl::sstv_pd120_sink_impl(const char *filename_png, bool split)
|
||
|
: gr::sync_block("sstv_pd120_sink",
|
||
|
gr::io_signature::make (1, 1, sizeof(float)),
|
||
|
gr::io_signature::make (0, 0, 0)),
|
||
|
d_filename_png (filename_png),
|
||
|
d_split (split),
|
||
|
d_has_sync(false)
|
||
|
{
|
||
|
set_history(sync_length);
|
||
|
d_line = new float[line_length];
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Our virtual destructor.
|
||
|
*/
|
||
|
sstv_pd120_sink_impl::~sstv_pd120_sink_impl()
|
||
|
{
|
||
|
delete[] d_line;
|
||
|
}
|
||
|
|
||
|
float
|
||
|
sstv_pd120_sink_impl::to_frequency(float sample) {
|
||
|
float freq = center + sample * max_dev;
|
||
|
freq = std::max(min_freq, freq);
|
||
|
freq = std::min(max_freq, freq);
|
||
|
return freq;
|
||
|
}
|
||
|
|
||
|
bool
|
||
|
sstv_pd120_sink_impl::is_sync(size_t pos, const float *samples) {
|
||
|
size_t count = 0;
|
||
|
for(size_t i = 0; i < sync_length; i++) {
|
||
|
float sample = to_frequency(samples[pos - (sync_length - 1) + i]);
|
||
|
if(sample < color_low) {
|
||
|
count += 1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool res = count > sync_length && !d_has_sync;
|
||
|
d_has_sync = count > sync_length;
|
||
|
|
||
|
std::cout << "Count: " << count << " " << d_has_sync << std::endl;
|
||
|
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
int
|
||
|
sstv_pd120_sink_impl::work(int noutput_items,
|
||
|
gr_vector_const_void_star &input_items,
|
||
|
gr_vector_void_star &output_items)
|
||
|
{
|
||
|
const float *in = (const float *) input_items[0];
|
||
|
|
||
|
std::cout << "foo" << std::endl;
|
||
|
for (size_t i = sync_length - 1;
|
||
|
i < noutput_items + sync_length - 1; i++) {
|
||
|
|
||
|
if(is_sync(i, in)) {
|
||
|
std::cout << "Sync: " << i << std::endl;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Tell runtime system how many output items we produced.
|
||
|
return noutput_items;
|
||
|
}
|
||
|
|
||
|
} /* namespace satnogs */
|
||
|
} /* namespace gr */
|