satdump/src-interface/recorder/tracking/tracking_widget.cpp

#include "tracking_widget.h"
#include "common/imgui_utils.h"
#include "common/tracking/rotator/rotcl_handler.h"
#include "core/config.h"
#include "core/style.h"
#include "imgui/imgui.h"
#include "logger.h"
#include "main_ui.h"
#include "utils/time.h"

namespace satdump
{
    TrackingWidget::TrackingWidget()
    {
        try
        {
            qth_lon = satdump_cfg.getValueFromSatDumpGeneral<double>("qth_lon");
            qth_lat = satdump_cfg.getValueFromSatDumpGeneral<double>("qth_lat");
            qth_alt = satdump_cfg.getValueFromSatDumpGeneral<double>("qth_alt");
        }
        catch (std::exception &e)
        {
            logger->error("Could not get QTH lon/lat! %s", e.what());
        }

        logger->trace("Using QTH %f %f Alt %f", qth_lon, qth_lat, qth_alt);

        rotator_options = rotator::getRotatorHandlerOptions();
        for (auto &st : rotator_options)
            rotator_options_str += st.name + '\0';

        rotator_handler = rotator_options[selected_rotator_handler].construct();

        if (rotator_handler)
        {
            try
            {
                nlohmann::json cfg = db->get_user_json("recorder_tracking");
                rotator_handler->set_settings(cfg["rotator_config"][rotator_handler->get_id()]);
            }
            catch (std::exception &)
            {
            }
        }

        // Init Obj Tracker
        object_tracker.setQTH(qth_lon, qth_lat, qth_alt);
        object_tracker.setRotator(rotator_handler);
        object_tracker.setObject(object_tracker.TRACKING_SATELLITE, 57166); // Phoenix replaced King for default sat
        object_tracker.rotator_target_pos_updated_callback = [this](double az, double el) { sat_finder.setTarget(az, el); };

        // Init Sat finder
        sat_finder.setQTH(qth_lon, qth_lat, qth_alt);

        // Init scheduler
        auto_scheduler.eng_callback = [this](AutoTrackCfg, SatellitePass, TrackedObject obj)
        {
            object_tracker.setObject(object_tracker.TRACKING_SATELLITE, obj.norad);
            saveConfig();
        };
        auto_scheduler.aos_callback = [this](AutoTrackCfg autotrack_cfg, SatellitePass pass, TrackedObject obj)
        {
            this->aos_callback(autotrack_cfg, pass, obj);
            object_tracker.setObject(object_tracker.TRACKING_SATELLITE, obj.norad);
        };
        auto_scheduler.los_callback = [this](AutoTrackCfg autotrack_cfg, SatellitePass pass, TrackedObject obj) { this->los_callback(autotrack_cfg, pass, obj); };

        auto_scheduler.setQTH(qth_lon, qth_lat, qth_alt);

        // Restore config
        loadConfig();

        // Start scheduler
        auto_scheduler.start();

        // Attempt to apply provided CLI settings
        if (satdump::satdump_cfg.main_cfg.contains("cli"))
        {
            auto &cli_settings = satdump::satdump_cfg.main_cfg["cli"];

            if (cli_settings.contains("engage_autotrack") && cli_settings["engage_autotrack"].get<bool>())
            {
                auto_scheduler.setEngaged(true, getTime());
            }
        }
    }

    TrackingWidget::~TrackingWidget() { saveConfig(); }

    void TrackingWidget::render()
    {
        object_tracker.renderPolarPlot();

        ImGui::Separator();

        object_tracker.renderSelectionMenu();

        ImGui::Spacing();

        if (ImGui::CollapsingHeader("Object Information"))
            object_tracker.renderObjectStatus();

        if (ImGui::CollapsingHeader("Rotator Configuration"))
        {
            object_tracker.renderRotatorStatus();
            ImGui::SameLine();

            if (rotator_handler->is_connected())
                style::beginDisabled();
            ImGui::SetNextItemWidth(ImGui::GetContentRegionAvail().x);
            if (ImGui::Combo("Type##rotatortype", &selected_rotator_handler, rotator_options_str.c_str()))
            {
                rotator_handler = rotator_options[selected_rotator_handler].construct();
                object_tracker.setRotator(rotator_handler);

                try
                {
                    nlohmann::json cfg = db->get_user_json("recorder_tracking");
                    rotator_handler->set_settings(cfg["rotator_config"][rotator_handler->get_id()]);
                }
                catch (std::exception &)
                {
                }
            }
            if (rotator_handler->is_connected())
                style::endDisabled();

            rotator_handler->render();
        }

        ImGui::Spacing();
        ImGui::Separator();
        ImGui::Spacing();

        float width_available = ImGui::GetContentRegionAvail().x;
        std::string is_engaged = auto_scheduler.getEngaged() ? "YES" : "NO";
        float centered_pos = width_available / 2.0f - ImGui::CalcTextSize(std::string("Autotrack Engaged: " + is_engaged).c_str()).x / 2.0f;
        if (centered_pos > 0)
            ImGui::SetCursorPosX(centered_pos);
        ImGui::TextUnformatted("Autotrack Engaged:");
        ImGui::SameLine();
        ImGui::TextColored(auto_scheduler.getEngaged() ? style::theme.green : style::theme.red, "%s", is_engaged.c_str());
        if (ImGui::Button("Schedule and Config", ImVec2(width_available, 0.0f)))
            config_window_was_asked = show_window_config = true;
        if (ImGui::Button("Satellite finder", ImVec2(width_available, 0.0f)))
            show_satellite_finder_window = satellite_finder_was_asked = true;
        ImGui::Spacing();
        renderConfig();
        renderSatfinder();
    }

    void TrackingWidget::renderConfig()
    {
        if (show_window_config)
        {
            ImGui::SetNextWindowSizeConstraints(ImVec2(800 * ui_scale, 300 * ui_scale), ImVec2(INFINITY, INFINITY));
            ImGui::Begin("Tracking Configuration", &show_window_config);
            ImGui::SetWindowSize(ImVec2(800 * ui_scale, 550 * ui_scale), ImGuiCond_FirstUseEver);

            if (ImGui::BeginTabBar("##trackingtabbar"))
            {
                if (ImGui::BeginTabItem("Scheduling"))
                {
                    ImGui::BeginChild("##trackingbarschedule", ImVec2(0, 0), false, ImGuiWindowFlags_NoResize);
                    auto_scheduler.renderAutotrackConfig(getTime());
                    ImGui::EndChild();
                    ImGui::EndTabItem();
                }
                if (ImGui::BeginTabItem("Rotator Config"))
                {
                    object_tracker.renderRotatorConfig();
                    ImGui::EndTabItem();
                }
                if (ImGui::BeginTabItem("Export/Import"))
                {
                    ImGui::BeginChild("##trackingimportexport", ImVec2(0, 0), false, ImGuiWindowFlags_NoResize);
                    if (config_import_export.draw_export())
                        config_import_export.do_export(auto_scheduler, object_tracker, rotator_handler);
                    ImGui::Spacing();
                    bool disable_import = auto_scheduler.getEngaged();
                    if (disable_import)
                        style::beginDisabled();
                    if (config_import_export.draw_import())
                        config_import_export.do_import(auto_scheduler, object_tracker, rotator_handler);
                    if (disable_import)
                        style::endDisabled();

                    ImGui::EndChild();
                    ImGui::EndTabItem();
                }
                ImGui::EndTabBar();
            }

            if (config_window_was_asked)
                ImGuiUtils_BringCurrentWindowToFront();
            config_window_was_asked = false;

            ImGui::End();
        }
    }

    void TrackingWidget::renderSatfinder()
    {
        if (show_satellite_finder_window)
        {
            ImGui::Begin("Satellite finder", &show_satellite_finder_window);
            ImGui::SetWindowSize(ImVec2(800, 550), ImGuiCond_FirstUseEver);
            if (satellite_finder_was_asked)
                ImGuiUtils_BringCurrentWindowToFront();
            satellite_finder_was_asked = false;

            sat_finder.render();

            ImGui::End();
        }
    }
} // namespace satdump