satdump/src-cli/offline.cpp
2024-05-15 17:53:20 +02:00

69 lines
No EOL
2.3 KiB
C++

#include "offline.h"
#include "common/detect_header.h"
#include "init.h"
#include "common/cli_utils.h"
#include "core/module.h"
#include "core/pipeline.h"
#include <filesystem>
#include "nlohmann/json.hpp"
#include "logger.h"
int main_offline(int argc, char *argv[])
{
if (argc < 5) // Check overall command
{
logger->error("Usage : " + std::string(argv[0]) + " [pipeline_id] [input_level] [input_file] [output_file_or_directory] [additional options as required]");
logger->error("Extra options (examples. Any parameter used in modules can be used here) :");
logger->error(" --samplerate [baseband_samplerate] --baseband_format [f32/s16/s8/u8] --dc_block --iq_swap");
logger->error("Sample command :");
logger->error("./satdump metop_ahrpt baseband /home/user/metop_baseband.cs16 metop_output_directory --samplerate 6e6 --baseband_format s16");
return 1;
}
std::string downlink_pipeline = argv[1];
std::string input_level = argv[2];
std::string input_file = argv[3];
std::string output_file = argv[4];
// Parse flags
nlohmann::json parameters = parse_common_flags(argc - 5, &argv[5]);
try_get_params_from_input_file(parameters, input_file);
// Init SatDump
satdump::tle_file_override = parameters.contains("tle_override") ? parameters["tle_override"].get<std::string>() : "";
satdump::initSatdump();
completeLoggerInit();
// logger->warn("\n" + parameters.dump(4));
// exit(0);
// Print some useful info
logger->info("Starting processing pipeline " + downlink_pipeline + "...");
logger->debug("Input file (" + input_level + ") : " + input_file);
logger->debug("Output file : " + output_file);
// Create output dir
if (!std::filesystem::exists(output_file))
std::filesystem::create_directories(output_file);
// Get pipeline
std::optional<satdump::Pipeline> pipeline = satdump::getPipelineFromName(downlink_pipeline);
if (pipeline.has_value())
{
try
{
pipeline.value().run(input_file, output_file, parameters, input_level);
}
catch (std::exception &e)
{
logger->error("Fatal error running pipeline : " + std::string(e.what()));
return 1;
}
}
else
logger->critical("Pipeline " + downlink_pipeline + " does not exist!");
return 0;
}