LoadingDialog.cpp

Go to the documentation of this file.

#include "Dialogs/LoadingDialog.h"
#include <wx/datetime.h>
#include <wx/statline.h>
 
wxBEGIN_EVENT_TABLE(LoadingDialog, EmberUI::ScalableDialog) EVT_BUTTON(wxID_CANCEL, LoadingDialog::OnCancel)
    EVT_CLOSE(LoadingDialog::OnClose) wxEND_EVENT_TABLE()
 
        LoadingDialog::LoadingDialog(wxWindow *parent, const wxString &title)
    : EmberUI::ScalableDialog(parent, wxID_ANY, title, wxSize(1200, 800), wxDEFAULT_DIALOG_STYLE | wxSTAY_ON_TOP),
      progress_bar_(nullptr), log_text_(nullptr), cancel_button_(nullptr), status_label_(nullptr),
      was_cancelled_(false), is_complete_(false), current_progress_(0), max_progress_(100) {
    CreateControls();
    Centre();
}
 
LoadingDialog::~LoadingDialog() {}
 
void LoadingDialog::CreateControls() {
    // Create main sizer
    wxBoxSizer *main_sizer = new wxBoxSizer(wxVERTICAL);
 
    // Status label (shows current operation)
    status_label_ = new wxStaticText(this, wxID_ANY, "Initializing...");
    wxFont status_font = status_label_->GetFont();
    status_font.SetPointSize(10);
    status_font.SetWeight(wxFONTWEIGHT_BOLD);
    status_label_->SetFont(status_font);
    main_sizer->Add(status_label_, 0, wxALL | wxEXPAND, 10);
 
    // Progress bar
    progress_bar_ = new wxGauge(this, wxID_ANY, 100, wxDefaultPosition, wxSize(-1, 25));
    main_sizer->Add(progress_bar_, 0, wxLEFT | wxRIGHT | wxEXPAND, 10);
 
    // Progress text (e.g., "50/100")
    wxStaticText *progress_text = new wxStaticText(this, wxID_ANY, "0%");
    progress_text->SetName("progress_text");
    main_sizer->Add(progress_text, 0, wxALL | wxALIGN_CENTER, 5);
 
    // Separator
    main_sizer->Add(new wxStaticLine(this), 0, wxALL | wxEXPAND, 5);
 
    // Log label
    wxStaticText *log_label = new wxStaticText(this, wxID_ANY, "Parsing Details:");
    wxFont log_label_font = log_label->GetFont();
    log_label_font.SetPointSize(9);
    log_label_font.SetWeight(wxFONTWEIGHT_BOLD);
    log_label->SetFont(log_label_font);
    main_sizer->Add(log_label, 0, wxLEFT | wxRIGHT | wxTOP, 10);
 
    // Log text control (terminal-like, scrollable)
    log_text_ = new wxTextCtrl(this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxDefaultSize,
                               wxTE_MULTILINE | wxTE_READONLY | wxTE_RICH2 | wxTE_WORDWRAP | wxHSCROLL | wxVSCROLL);
 
    // Style the log to look like a terminal
    wxFont log_font(9, wxFONTFAMILY_TELETYPE, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_NORMAL);
    log_text_->SetFont(log_font);
    log_text_->SetBackgroundColour(wxColour(30, 30, 30));    // Dark background
    log_text_->SetForegroundColour(wxColour(200, 200, 200)); // Light gray text
 
    main_sizer->Add(log_text_, 1, wxALL | wxEXPAND, 10);
 
    // Cancel button
    cancel_button_ = new wxButton(this, wxID_CANCEL, "Cancel");
    main_sizer->Add(cancel_button_, 0, wxALL | wxALIGN_CENTER, 10);
 
    SetSizer(main_sizer);
 
    // Initial log message
    AppendLog("Loading started...", "[INIT]");
}
 
bool LoadingDialog::UpdateProgress(int current, int max, const wxString &message) {
    if (was_cancelled_) {
        return false;
    }
 
    current_progress_ = current;
    max_progress_ = max;
 
    // Update progress bar
    if (progress_bar_) {
        progress_bar_->SetRange(max);
        progress_bar_->SetValue(current);
    }
 
    // Update progress percentage text
    wxWindow *progress_text = FindWindowByName("progress_text");
    if (progress_text) {
        int percentage = (max > 0) ? (current * 100 / max) : 0;
        wxStaticText *text = dynamic_cast<wxStaticText *>(progress_text);
        if (text) {
            text->SetLabel(wxString::Format("%d%% (%d/%d)", percentage, current, max));
        }
    }
 
    // Update status label if message provided
    if (!message.IsEmpty()) {
        if (status_label_) {
            status_label_->SetLabel(message);
        }
 
        // Also append to log
        AppendLog(message, "[STEP]");
    }
 
    // Force UI update
    Update();
    wxYield();
 
    return !was_cancelled_;
}
 
void LoadingDialog::AppendLog(const wxString &message, const wxString &prefix) {
    if (!log_text_)
        return;
 
    // Get current timestamp
    wxDateTime now = wxDateTime::Now();
    wxString timestamp = now.Format("%H:%M:%S");
 
    // Format the log message
    wxString formatted_message;
    if (!prefix.IsEmpty()) {
        formatted_message = wxString::Format("[%s] %s %s\n", timestamp.c_str(), prefix.c_str(), message.c_str());
    } else {
        formatted_message = wxString::Format("[%s] %s\n", timestamp.c_str(), message.c_str());
    }
 
    // Append to log with color coding based on prefix
    if (prefix.Contains("ERROR") || prefix.Contains("FAIL")) {
        log_text_->SetDefaultStyle(wxTextAttr(wxColour(255, 100, 100))); // Red
    } else if (prefix.Contains("WARN")) {
        log_text_->SetDefaultStyle(wxTextAttr(wxColour(255, 200, 100))); // Orange
    } else if (prefix.Contains("SUCCESS") || prefix.Contains("COMPLETE")) {
        log_text_->SetDefaultStyle(wxTextAttr(wxColour(100, 255, 100))); // Green
    } else if (prefix.Contains("STEP")) {
        log_text_->SetDefaultStyle(wxTextAttr(wxColour(100, 200, 255))); // Light blue
    } else {
        log_text_->SetDefaultStyle(wxTextAttr(wxColour(200, 200, 200))); // Default light gray
    }
 
    log_text_->AppendText(formatted_message);
 
    // Auto-scroll to bottom
    log_text_->SetInsertionPointEnd();
    log_text_->ShowPosition(log_text_->GetLastPosition());
}
 
void LoadingDialog::ClearLog() {
    if (log_text_) {
        log_text_->Clear();
    }
}
 
void LoadingDialog::SetMaxProgress(int max) {
    max_progress_ = max;
    if (progress_bar_) {
        progress_bar_->SetRange(max);
    }
}
 
void LoadingDialog::MarkComplete() {
    is_complete_ = true;
 
    // Change button to "Close"
    if (cancel_button_) {
        cancel_button_->SetLabel("Close");
    }
 
    // Update status
    if (status_label_) {
        status_label_->SetLabel("Loading Complete - Review logs and click Close");
    }
}
 
void LoadingDialog::OnCancel(wxCommandEvent &event) {
    if (is_complete_) {
        // Loading is complete, just close the dialog
        if (IsModal()) {
            EndModal(wxID_OK);
        } else {
            Hide();
        }
    } else {
        // Loading in progress, cancel it
        was_cancelled_ = true;
        AppendLog("Loading cancelled by user", "[CANCEL]");
 
        // Close the dialog after a brief delay
        wxMilliSleep(500);
        if (IsModal()) {
            EndModal(wxID_CANCEL);
        } else {
            Hide();
        }
    }
}
 
void LoadingDialog::OnClose(wxCloseEvent &event) {
    if (!is_complete_) {
        was_cancelled_ = true;
    }
    if (IsModal()) {
        EndModal(is_complete_ ? wxID_OK : wxID_CANCEL);
    } else {
        event.Skip();
    }
}