sebastian
/
gr-satnogs
mirror of https://gitlab.com/LongHairedHacker/gr-satnogs.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Added color conversion and stub for rendering

This commit is contained in:
Sebastian 2019-03-31 17:35:28 +02:00
parent 71be84ac85
commit 09e9ac688c
2 changed files with 87 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
lib/sstv_pd120_sink_impl.cc
View File

@ -29,6 +29,7 @@ namespace gr {
namespace satnogs {
const size_t image_width = 640;
const size_t image_height = 496;
const size_t sync_length = 105;
const size_t sync_thresh = 100;
const size_t porch_length = 10;
@ -59,7 +60,10 @@ namespace gr {
gr::io_signature::make (0, 0, 0)),
d_filename_png (filename_png),
d_split (split),
d_has_sync(false)
d_has_sync(false),
d_initial_sync(true),
d_line_pos(0),
d_image_y (0)
{
set_history(sync_length);
d_line = new float[line_length];
@ -81,6 +85,33 @@ namespace gr {
return freq;
}
int
sstv_pd120_sink_impl::to_color(float sample) {
sample = (sample - color_low) / (color_high - color_low);
int color = int(sample * 255);
color = std::max(color, 0);
color = std::min(color, 255);
return color;
}
void
ycbcr_to_rgb(int ycbcr[3], int rgb[3]) {
int y = ycbcr[0];
int cb = ycbcr[1];
int cr = ycbcr[2];
//https://stackoverflow.com/questions/4041840/function-to-convert-ycbcr-to-rgb/15333019#15333019
cr = cr - 128;
cb = cb - 128;
int r = y + 45 * cr / 32;
int g = y - (11 * cb + 23 * cr) / 32;
int b = y + 113 * cb / 64;
rgb[0] = r;
rgb[1] = g;
rgb[2] = b;
}
bool
sstv_pd120_sink_impl::is_sync(size_t pos, const float *samples) {
size_t count = 0;
@ -91,14 +122,42 @@ namespace gr {
}
}
bool res = count > sync_length && !d_has_sync;
d_has_sync = count > sync_length;
bool res = count > sync_thresh && !d_has_sync;
d_has_sync = count > sync_thresh;
std::cout << "Count: " << count << " " << d_has_sync << std::endl;
d_initial_sync = false;
return res;
}
void
sstv_pd120_sink_impl::render_line() {
int start_pos = d_line_pos - sync_length - 4 * image_width;
if(start_pos < 0) {
return;
}
for(size_t x = 0; x < image_width; x++) {
int y0 = to_color(d_line[start_pos + x]);
int cr = to_color(d_line[start_pos + image_width + x]);
int cb = to_color(d_line[start_pos + 2 * image_width + x]);
int y1 = to_color(d_line[start_pos + 3 * image_width + x]);
int rgb0[3];
int ycrcb0[] = {y0, cr, cb};
ycbcr_to_rgb(ycrcb0, rgb0);
int rgb1[3];
int ycrcb1[] = {y1, cr, cb};
ycbcr_to_rgb(ycrcb1, rgb1);
//Todo: Write pixels
}
d_image_y += 2;
}
int
sstv_pd120_sink_impl::work(int noutput_items,
gr_vector_const_void_star &input_items,
@ -106,12 +165,21 @@ namespace gr {
{
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;
float sample = to_frequency(in[i]);
d_line[d_line_pos] = sample;
d_line_pos++;
if(is_sync(i, in) || d_line_pos >= line_length) {
if(d_initial_sync) {
d_image_y = 0;
}
else if(!d_initial_sync && d_line_pos > line_length - porch_length) {
std::cout << "Rendering after: " << d_line_pos << std::endl;
}
d_line_pos = 0;
}
}

12
lib/sstv_pd120_sink_impl.h
View File

@ -24,6 +24,9 @@
#include <satnogs/sstv_pd120_sink.h>
#define PNG_DEBUG 3
#include <png++/png.hpp>
namespace gr {
namespace satnogs {
@ -33,12 +36,21 @@ namespace gr {
std::string d_filename_png;
bool d_split;
bool d_has_sync;
bool d_initial_sync;
float *d_line;
size_t d_line_pos;
size_t d_image_y;
png::image<png::rgb_pixel> d_image;
float to_frequency(float sample);
int to_color(float sample);
void ycbcr_to_rgb(int ycbcr[3], int rgb[3]);
bool is_sync(size_t pos, const float *samples);
void render_line();
public:
sstv_pd120_sink_impl(const char *filename_png, bool split);
~sstv_pd120_sink_impl();