125 lines
3.6 KiB
C++
125 lines
3.6 KiB
C++
/* -*- c++ -*- */
|
|
/*
|
|
* Copyright 2020
|
|
* Federico "Larroca" La Rocca <flarroca@fing.edu.uy>
|
|
*
|
|
* Instituto de Ingenieria Electrica, Facultad de Ingenieria,
|
|
* Universidad de la Republica, Uruguay.
|
|
*
|
|
* This 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, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This software 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 software; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, Inc., 51 Franklin Street,
|
|
* Boston, MA 02110-1301, USA.
|
|
*
|
|
*/
|
|
|
|
#ifndef INCLUDED_TEMPEST_SSAMP_CORRECTION_IMPL_H
|
|
#define INCLUDED_TEMPEST_SSAMP_CORRECTION_IMPL_H
|
|
|
|
#include <tempest/ssamp_correction.h>
|
|
#include <gnuradio/filter/mmse_fir_interpolator_cc.h>
|
|
#include <random>
|
|
|
|
namespace gr {
|
|
namespace tempest {
|
|
|
|
class ssamp_correction_impl : public ssamp_correction
|
|
{
|
|
private:
|
|
int d_Htotal;
|
|
int d_Vtotal;
|
|
int d_correct_sampling;
|
|
float d_max_deviation;
|
|
//Counters
|
|
int d_frame_height_counter;
|
|
int d_frames_counter;
|
|
|
|
//Fixed parameter
|
|
int d_discarded_amount_per_frame;
|
|
|
|
|
|
|
|
// to accelerate the process, I'll only update the interpolation
|
|
// once every a random number of iterations with probability d_proba_of_updating
|
|
std::geometric_distribution<int> d_dist;
|
|
std::minstd_rand d_gen;
|
|
|
|
int d_next_update;
|
|
int d_required_for_interpolation;
|
|
|
|
double d_samp_inc_rem;
|
|
double d_new_interpolation_ratio_rem;
|
|
double d_samp_phase;
|
|
|
|
float d_alpha_samp_inc;
|
|
int d_max_deviation_px;
|
|
|
|
// correlation with the last line
|
|
|
|
gr_complex * d_current_line_corr;
|
|
gr_complex * d_historic_line_corr;
|
|
float * d_abs_historic_line_corr;
|
|
|
|
// correlation with the last frame
|
|
|
|
gr_complex * d_current_frame_corr;
|
|
gr_complex * d_historic_frame_corr;
|
|
float * d_abs_historic_frame_corr;
|
|
|
|
bool d_stop_fine_sampling_synch;
|
|
// where is the line correlation peak
|
|
int d_peak_line_index;
|
|
|
|
float d_alpha_corr;
|
|
|
|
// the interpolating filter
|
|
gr::filter::mmse_fir_interpolator_cc d_inter;
|
|
|
|
int interpolate_input(const gr_complex * in, gr_complex * out, int size);
|
|
|
|
void update_interpolation_ratio(const gr_complex * in, int in_size);
|
|
|
|
void estimate_peak_line_index(const gr_complex * in, int in_size);
|
|
|
|
void set_iHsize_msg(pmt::pmt_t msg);
|
|
|
|
void set_Vsize_msg(pmt::pmt_t msg);
|
|
|
|
|
|
void set_ena_msg(pmt::pmt_t msg);
|
|
gr::thread::mutex d_mutex;
|
|
void set_ratio_msg(pmt::pmt_t msg);
|
|
|
|
public:
|
|
ssamp_correction_impl(int Htotal, int Vtotal, int correct_sampling, float max_deviation);
|
|
~ssamp_correction_impl();
|
|
|
|
|
|
void set_Htotal_Vtotal(int Htotal, int Vtotal) override; // Asynch Callback see: tempest_ssamp_correction.block.yml
|
|
|
|
// Where all the action really happens
|
|
void forecast (int noutput_items, gr_vector_int &ninput_items_required);
|
|
|
|
int general_work(int noutput_items,
|
|
gr_vector_int &ninput_items,
|
|
gr_vector_const_void_star &input_items,
|
|
gr_vector_void_star &output_items);
|
|
|
|
};
|
|
|
|
} // namespace tempest
|
|
} // namespace gr
|
|
|
|
#endif /* INCLUDED_TEMPEST_SSAMP_CORRECTION_IMPL_H */
|
|
|