main.cpp

Go to the documentation of this file.

#include "CLIOptionsMap.hpp"
#include <CLI/CLI.hpp>
#include <print>
#include <spdlog/spdlog.h>
#include <srs/Application.hpp>
#include <srs/ConfigHandler.hpp>
 
#ifdef HAS_ROOT
#include <TROOT.h>
#endif
 
auto main(int argc, char** argv) -> int
{
    auto cli_args = CLI::App{ "SRS system command line interface" };
    try
    {
        argv = cli_args.ensure_utf8(argv);
 
        auto app_config = srs::Config{};
 
        auto spdlog_level = spdlog::level::info;
        auto print_mode = srs::common::DataPrintMode::print_speed;
        auto output_filenames = std::vector<std::string>{ "output.bin" };
        auto is_version_print = false;
        auto is_root_version_print = false;
        auto is_dump_needed = false;
        const auto home_dir = std::string_view{ getenv("HOME") };
        auto json_filepath = home_dir.empty() ? "" : std::format("{}/.config/srs-control/config.json", getenv("HOME"));
        auto dump_config_callback = [&json_filepath, &is_dump_needed](const std::string& filename)
        {
            is_dump_needed = true;
            json_filepath = filename;
        };
 
        cli_args.add_flag("-v, --version", is_version_print, "Show the current version");
        cli_args.add_flag("--root-version", is_root_version_print, "Show the ROOT version if used");
        cli_args.add_option("-l, --log-level", spdlog_level, "Set log level")
            ->transform(CLI::CheckedTransformer(spd_log_map, CLI::ignore_case))
            ->capture_default_str();
        cli_args.add_option("-p, --print-mode", print_mode, "Set data print mode")
            ->transform(CLI::CheckedTransformer(print_mode_map, CLI::ignore_case))
            ->capture_default_str();
        cli_args.add_option("-c, --config-file", json_filepath, "Set the path of the JSON config file")
            ->capture_default_str();
        cli_args
            .add_option_function<std::string>(
                "--dump-config", dump_config_callback, "Dump default configuration to the file")
            ->default_val(json_filepath)
            ->run_callback_for_default()
            ->expected(0, 1);
 
        cli_args.add_option("-o, --output-files", output_filenames, "Set output file (or socket) names")
            ->capture_default_str()
            ->expected(0, -1);
        cli_args.parse(argc, argv);
 
        if (is_version_print)
        {
            std::println("{}", SRS_PROJECT_VERSION);
            return 0;
        }
 
        if (is_dump_needed)
        {
            srs::config::output_config_to_json(app_config, json_filepath);
            return 0;
        }
 
        if (is_root_version_print)
        {
#ifdef HAS_ROOT
            std::println("{}", gROOT->GetVersion());
#else
            std::println("ROOT is not built");
#endif
            return 0;
        }
 
        spdlog::set_level(spdlog_level);
 
        auto app = srs::App{};
 
        srs::config::set_config_from_json(app_config, json_filepath);
        app.set_options(std::move(app_config));
        // app.set_remote_endpoint(srs::common::DEFAULT_SRS_IP, srs::common::DEFAULT_SRS_CONTROL_PORT);
        app.set_print_mode(print_mode);
        app.set_output_filenames(output_filenames);
 
        app.init();
        app.read_data();
        app.switch_on();
        app.start_workflow();
        // app.wait_for_finish();
    }
    catch (const CLI::ParseError& e)
    {
        cli_args.exit(e);
    }
    catch (std::exception& ex)
    {
        spdlog::critical("Exception occured: {}", ex.what());
    }
 
    return 0;
}