mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
Add client mode
This commit is contained in:
Aang23
parent
0218a101ba
commit
005f27fb32
7 changed files with 330 additions and 95 deletions
242
src-cli/live.cpp
242
src-cli/live.cpp
|
|
@ -37,119 +37,173 @@ int main_live(int argc, char *argv[])
|
|||
// Parse flags
|
||||
nlohmann::json parameters = parse_common_flags(argc - 4, &argv[4]);
|
||||
|
||||
uint64_t samplerate;
|
||||
uint64_t frequency;
|
||||
uint64_t timeout;
|
||||
std::string handler_id;
|
||||
|
||||
try
|
||||
if (parameters.contains("client"))
|
||||
{
|
||||
samplerate = parameters["samplerate"].get<uint64_t>();
|
||||
frequency = parameters["frequency"].get<uint64_t>();
|
||||
timeout = parameters.contains("timeout") ? parameters["timeout"].get<uint64_t>() : 0;
|
||||
handler_id = parameters["source"].get<std::string>();
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logger->error("Error parsing arguments! {:s}", e.what());
|
||||
return 1;
|
||||
}
|
||||
logger->info("Starting in client mode!");
|
||||
|
||||
// Create output dir
|
||||
if (!std::filesystem::exists(output_file))
|
||||
std::filesystem::create_directories(output_file);
|
||||
// Create output dir
|
||||
if (!std::filesystem::exists(output_file))
|
||||
std::filesystem::create_directories(output_file);
|
||||
|
||||
// Get all sources
|
||||
dsp::registerAllSources();
|
||||
std::vector<dsp::SourceDescriptor> source_tr = dsp::getAllAvailableSources();
|
||||
dsp::SourceDescriptor selected_src;
|
||||
// Get pipeline
|
||||
std::optional<satdump::Pipeline> pipeline = satdump::getPipelineFromName(downlink_pipeline);
|
||||
|
||||
for (dsp::SourceDescriptor src : source_tr)
|
||||
logger->debug("Device " + src.name);
|
||||
|
||||
// Try to find it and check it's usable
|
||||
bool src_found = false;
|
||||
for (dsp::SourceDescriptor src : source_tr)
|
||||
{
|
||||
if (handler_id == src.source_type)
|
||||
if (!pipeline.has_value())
|
||||
{
|
||||
selected_src = src;
|
||||
src_found = true;
|
||||
logger->critical("Pipeline " + downlink_pipeline + " does not exist!");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (!src_found)
|
||||
{
|
||||
logger->error("Could not find a handler for source type : {:s}!", handler_id.c_str());
|
||||
return 1;
|
||||
}
|
||||
// Init pipeline
|
||||
std::unique_ptr<satdump::LivePipeline> live_pipeline = std::make_unique<satdump::LivePipeline>(pipeline.value(), parameters, output_file);
|
||||
|
||||
// Init source
|
||||
std::shared_ptr<dsp::DSPSampleSource> source_ptr = getSourceFromDescriptor(selected_src);
|
||||
source_ptr->open();
|
||||
source_ptr->set_frequency(frequency);
|
||||
source_ptr->set_samplerate(samplerate);
|
||||
source_ptr->set_settings(parameters);
|
||||
ctpl::thread_pool live_thread_pool(8);
|
||||
|
||||
// Get pipeline
|
||||
std::optional<satdump::Pipeline> pipeline = satdump::getPipelineFromName(downlink_pipeline);
|
||||
|
||||
if (!pipeline.has_value())
|
||||
{
|
||||
logger->critical("Pipeline " + downlink_pipeline + " does not exist!");
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Init pipeline
|
||||
parameters["baseband_format"] = "f32";
|
||||
parameters["buffer_size"] = STREAM_BUFFER_SIZE; // This is required, as we WILL go over the (usually) default 8192 size
|
||||
std::unique_ptr<satdump::LivePipeline> live_pipeline = std::make_unique<satdump::LivePipeline>(pipeline.value(), parameters, output_file);
|
||||
|
||||
ctpl::thread_pool live_thread_pool(8);
|
||||
|
||||
bool server_mode = parameters.contains("server_address") || parameters.contains("server_port");
|
||||
|
||||
// Attempt to start the source and pipeline
|
||||
try
|
||||
{
|
||||
source_ptr->start();
|
||||
live_pipeline->start(source_ptr->output_stream, live_thread_pool, server_mode);
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logger->error("Fatal error running pipeline/device : " + std::string(e.what()));
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Attach signal
|
||||
signal(SIGINT, sig_handler_live);
|
||||
|
||||
// Now, we wait
|
||||
uint64_t start_time = time(0);
|
||||
while (1)
|
||||
{
|
||||
if (timeout > 0)
|
||||
// Attempt to start the source and pipeline
|
||||
try
|
||||
{
|
||||
uint64_t elapsed_time = time(0) - start_time;
|
||||
if (elapsed_time >= timeout)
|
||||
live_pipeline->start_client(live_thread_pool);
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logger->error("Fatal error running pipeline/device : " + std::string(e.what()));
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Attach signal
|
||||
signal(SIGINT, sig_handler_live);
|
||||
|
||||
// Now, we wait
|
||||
while (1)
|
||||
{
|
||||
if (live_should_exit)
|
||||
{
|
||||
logger->warn("Timeout is over! ({:d}s >= {:d}s) Stopping.", elapsed_time, timeout);
|
||||
logger->warn("SIGINT Received. Stopping.");
|
||||
break;
|
||||
}
|
||||
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
||||
}
|
||||
|
||||
// Stop cleanly
|
||||
live_pipeline->stop();
|
||||
}
|
||||
else
|
||||
{
|
||||
uint64_t samplerate;
|
||||
uint64_t frequency;
|
||||
uint64_t timeout;
|
||||
std::string handler_id;
|
||||
|
||||
try
|
||||
{
|
||||
samplerate = parameters["samplerate"].get<uint64_t>();
|
||||
frequency = parameters["frequency"].get<uint64_t>();
|
||||
timeout = parameters.contains("timeout") ? parameters["timeout"].get<uint64_t>() : 0;
|
||||
handler_id = parameters["source"].get<std::string>();
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logger->error("Error parsing arguments! {:s}", e.what());
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Create output dir
|
||||
if (!std::filesystem::exists(output_file))
|
||||
std::filesystem::create_directories(output_file);
|
||||
|
||||
// Get all sources
|
||||
dsp::registerAllSources();
|
||||
std::vector<dsp::SourceDescriptor> source_tr = dsp::getAllAvailableSources();
|
||||
dsp::SourceDescriptor selected_src;
|
||||
|
||||
for (dsp::SourceDescriptor src : source_tr)
|
||||
logger->debug("Device " + src.name);
|
||||
|
||||
// Try to find it and check it's usable
|
||||
bool src_found = false;
|
||||
for (dsp::SourceDescriptor src : source_tr)
|
||||
{
|
||||
if (handler_id == src.source_type)
|
||||
{
|
||||
selected_src = src;
|
||||
src_found = true;
|
||||
}
|
||||
}
|
||||
|
||||
if (live_should_exit)
|
||||
if (!src_found)
|
||||
{
|
||||
logger->warn("SIGINT Received. Stopping.");
|
||||
break;
|
||||
logger->error("Could not find a handler for source type : {:s}!", handler_id.c_str());
|
||||
return 1;
|
||||
}
|
||||
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
||||
}
|
||||
// Init source
|
||||
std::shared_ptr<dsp::DSPSampleSource> source_ptr = getSourceFromDescriptor(selected_src);
|
||||
source_ptr->open();
|
||||
source_ptr->set_frequency(frequency);
|
||||
source_ptr->set_samplerate(samplerate);
|
||||
source_ptr->set_settings(parameters);
|
||||
|
||||
// Stop cleanly
|
||||
source_ptr->stop();
|
||||
live_pipeline->stop();
|
||||
// Get pipeline
|
||||
std::optional<satdump::Pipeline> pipeline = satdump::getPipelineFromName(downlink_pipeline);
|
||||
|
||||
if (!pipeline.has_value())
|
||||
{
|
||||
logger->critical("Pipeline " + downlink_pipeline + " does not exist!");
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Init pipeline
|
||||
parameters["baseband_format"] = "f32";
|
||||
parameters["buffer_size"] = STREAM_BUFFER_SIZE; // This is required, as we WILL go over the (usually) default 8192 size
|
||||
std::unique_ptr<satdump::LivePipeline> live_pipeline = std::make_unique<satdump::LivePipeline>(pipeline.value(), parameters, output_file);
|
||||
|
||||
ctpl::thread_pool live_thread_pool(8);
|
||||
|
||||
bool server_mode = parameters.contains("server_address") || parameters.contains("server_port");
|
||||
|
||||
// Attempt to start the source and pipeline
|
||||
try
|
||||
{
|
||||
source_ptr->start();
|
||||
live_pipeline->start(source_ptr->output_stream, live_thread_pool, server_mode);
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logger->error("Fatal error running pipeline/device : " + std::string(e.what()));
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Attach signal
|
||||
signal(SIGINT, sig_handler_live);
|
||||
|
||||
// Now, we wait
|
||||
uint64_t start_time = time(0);
|
||||
while (1)
|
||||
{
|
||||
if (timeout > 0)
|
||||
{
|
||||
uint64_t elapsed_time = time(0) - start_time;
|
||||
if (elapsed_time >= timeout)
|
||||
{
|
||||
logger->warn("Timeout is over! ({:d}s >= {:d}s) Stopping.", elapsed_time, timeout);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (live_should_exit)
|
||||
{
|
||||
logger->warn("SIGINT Received. Stopping.");
|
||||
break;
|
||||
}
|
||||
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
||||
}
|
||||
|
||||
// Stop cleanly
|
||||
source_ptr->stop();
|
||||
live_pipeline->stop();
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue