140 lines
3.7 KiB
C++
140 lines
3.7 KiB
C++
/* -*- c++ -*- */
|
|
/*
|
|
* gr-satnogs: SatNOGS GNU Radio Out-Of-Tree Module
|
|
*
|
|
* Copyright (C) 2017, 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 "ogg_encoder_impl.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <math.h>
|
|
|
|
namespace gr {
|
|
namespace satnogs {
|
|
|
|
ogg_encoder::sptr
|
|
ogg_encoder::make(char *filename, double samp_rate, float quality)
|
|
{
|
|
return gnuradio::get_initial_sptr(
|
|
new ogg_encoder_impl(filename, samp_rate, quality));
|
|
}
|
|
|
|
/*
|
|
* The private constructor
|
|
*/
|
|
ogg_encoder_impl::ogg_encoder_impl(char *filename, double samp_rate,
|
|
float quality) :
|
|
gr::sync_block("ogg_encoder",
|
|
gr::io_signature::make(1, 1, sizeof(float)),
|
|
gr::io_signature::make(0, 0, 0))
|
|
{
|
|
d_quality = quality;
|
|
d_out = fopen(filename, "wb");
|
|
d_samp_rate = samp_rate;
|
|
vorbis_info_init(&d_vi);
|
|
int ret = vorbis_encode_init_vbr(&d_vi, 1, d_samp_rate, d_quality);
|
|
if (ret) {
|
|
exit(1);
|
|
}
|
|
|
|
vorbis_comment_init(&d_vc);
|
|
vorbis_comment_add_tag(&d_vc, "ENCODER", "satnogs ogg encoder");
|
|
|
|
vorbis_analysis_init(&d_vd, &d_vi);
|
|
vorbis_block_init(&d_vd, &d_vb);
|
|
|
|
srand(time(NULL));
|
|
ogg_stream_init(&d_os, rand());
|
|
|
|
ogg_packet header;
|
|
ogg_packet header_comm;
|
|
ogg_packet header_code;
|
|
|
|
vorbis_analysis_headerout(&d_vd, &d_vc, &header, &header_comm,
|
|
&header_code);
|
|
ogg_stream_packetin(&d_os, &header);
|
|
ogg_stream_packetin(&d_os, &header_comm);
|
|
ogg_stream_packetin(&d_os, &header_code);
|
|
int result = 1;
|
|
while (result) {
|
|
result = ogg_stream_flush(&d_os, &d_og);
|
|
if (result == 0) {
|
|
break;
|
|
}
|
|
fwrite(d_og.header, 1, d_og.header_len, d_out);
|
|
fwrite(d_og.body, 1, d_og.body_len, d_out);
|
|
}
|
|
}
|
|
|
|
ogg_encoder_impl::~ogg_encoder_impl()
|
|
{
|
|
vorbis_analysis_wrote(&d_vd, 0);
|
|
ogg_stream_clear(&d_os);
|
|
vorbis_block_clear(&d_vb);
|
|
vorbis_dsp_clear(&d_vd);
|
|
vorbis_comment_clear(&d_vc);
|
|
vorbis_info_clear(&d_vi);
|
|
fclose(d_out);
|
|
}
|
|
|
|
int
|
|
ogg_encoder_impl::work(int noutput_items,
|
|
gr_vector_const_void_star &input_items,
|
|
gr_vector_void_star &output_items)
|
|
{
|
|
const char *in = (const char *) input_items[0];
|
|
int i;
|
|
float **buffer = vorbis_analysis_buffer(&d_vd, noutput_items);
|
|
memcpy(buffer[0], in, noutput_items * sizeof(float));
|
|
|
|
vorbis_analysis_wrote(&d_vd, noutput_items);
|
|
|
|
while (vorbis_analysis_blockout(&d_vd, &d_vb) == 1) {
|
|
vorbis_analysis(&d_vb, NULL);
|
|
vorbis_bitrate_addblock(&d_vb);
|
|
|
|
while (vorbis_bitrate_flushpacket(&d_vd, &d_op)) {
|
|
|
|
ogg_stream_packetin(&d_os, &d_op);
|
|
int result = 1;
|
|
while (result) {
|
|
int result = ogg_stream_pageout(&d_os, &d_og);
|
|
if (result == 0) {
|
|
break;
|
|
}
|
|
fwrite(d_og.header, 1, d_og.header_len, d_out);
|
|
fwrite(d_og.body, 1, d_og.body_len, d_out);
|
|
if (ogg_page_eos(&d_og)) {
|
|
result = 1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return noutput_items;
|
|
}
|
|
|
|
} /* namespace satnogs */
|
|
} /* namespace gr */
|