1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
* Copyright (c) 2018 Vasile Vilvoiu (YO7JBP) <vasi.vilvoiu@gmail.com>
*
* libsstv is free software; you can redistribute it and/or modify
* it under the terms of the MIT license. See LICENSE for details.
*/
#include <iostream>
#include <malloc.h>
#include <sndfile.h>
#include <glog/logging.h>
#include <gflags/gflags.h>
#include "cimg/CImg.h"
extern "C" {
#include <libsstv.h>
}
/*
* Command line flags
*/
DEFINE_bool(logtostderr, false, "Only log to stderr");
DEFINE_string(mode, "", "SSTV mode for encoder");
DEFINE_string(input, "", "input image");
DEFINE_string(output, "", "output WAV file");
DEFINE_uint64(sample_rate, 48000, "output audio sample rate");
int main(int argc, char **argv)
{
/* Parse command line flags */
gflags::ParseCommandLineFlags(&argc, &argv, true);
/* Initialize logging */
google::InitGoogleLogging(argv[0]);
google::InstallFailureSignalHandler();
/* check input */
if (FLAGS_input == "") {
LOG(FATAL) << "Input image filename not provided, use --input";
}
if (FLAGS_output == "") {
LOG(FATAL) << "Output WAV file not provided, use --output";
}
if (FLAGS_mode == "") {
LOG(FATAL) << "Encoding mode not provided, use --mode";
}
/* TODO: parse SSTV mode */
sstv_mode_t mode = SSTV_MODE_PD120;
/* load image from file (TODO: perform normalization from source to desired properties) */
LOG(INFO) << "Loading image from " << FLAGS_input;
cimg_library::CImg<unsigned char> input_image = (cimg_library::CImg<>(FLAGS_input.c_str())).RGBtoYCbCr();
sstv_image_t sstv_image;
if (sstv_pack_image(&sstv_image, input_image.width(), input_image.height(), SSTV_FORMAT_YCBCR, input_image.data()) != SSTV_OK) {
LOG(FATAL) << "sstv_pack_image() failed";
}
/* create a sample buffer for output */
uint8_t samp_buffer[128 * 1024];
sstv_signal_t signal;
if (sstv_pack_signal(&signal, SSTV_SAMPLE_INT16, 128 * 1024, samp_buffer) != SSTV_OK) {
LOG(FATAL) << "sstv_pack_signal() failed";
}
/* initialize library */
LOG(INFO) << "Initializing libsstv";
if (sstv_init(malloc, free) != SSTV_OK) {
LOG(FATAL) << "Failed to initialize libsstv";
}
/* create encoder context */
LOG(INFO) << "Creating encoding context";
void *ctx = nullptr;
if (sstv_create_encoder(&ctx, sstv_image, mode, FLAGS_sample_rate) != SSTV_OK) {
LOG(FATAL) << "Failed to create SSTV encoder";
}
if (!ctx) {
LOG(FATAL) << "NULL encoder received";
}
/* open WAV file */
SF_INFO wavinfo;
wavinfo.samplerate = FLAGS_sample_rate;
wavinfo.channels = 1;
wavinfo.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
SNDFILE *wavfile = sf_open(FLAGS_output.c_str(), SFM_WRITE, &wavinfo);
if (!wavfile) {
LOG(FATAL) << "sf_open() failed: " << sf_strerror(NULL);
}
/* encode */
while (true) {
/* encode block */
sstv_error_t rc = sstv_encode(ctx, &signal);
if (rc != SSTV_ENCODE_SUCCESSFUL && rc != SSTV_ENCODE_END) {
LOG(FATAL) << "sstv_encode() failed with rc " << rc;
}
/* write to sound file */
sf_write_short(wavfile, (int16_t *)signal.buffer, signal.count);
/* exit case */
if (rc == SSTV_ENCODE_END) {
break;
}
}
/* close wav file */
sf_close(wavfile);
/* cleanup */
LOG(INFO) << "Cleaning up";
if (sstv_delete_encoder(ctx) != SSTV_OK) {
LOG(FATAL) << "Failed to delete SSTV encoder";
}
/* all ok */
LOG(INFO) << "Successfuly exited";
return 0;
}
|
