OrcaSlicer/src/slic3r/GUI/FilamentPickerDialog.hpp

#ifndef slic3r_GUI_FilamentPickerDialog_hpp_
#define slic3r_GUI_FilamentPickerDialog_hpp_

#include "GUI_App.hpp"
#include "GUI.hpp"
#include "GUI_Utils.hpp"
#include "FilamentBitmapUtils.hpp"
#include "ColorPickerDialog.hpp"
#include <optional>
#include "Widgets/Button.hpp"
#include "EncodedFilament.hpp"
#include <wx/dialog.h>
#include <wx/string.h>
#include <wx/event.h>
#include <wx/gdicmn.h>
#include <wx/wx.h>
#include <wx/scrolwin.h>
#include <wx/bitmap.h>
#include <wx/region.h>
#include <wx/timer.h>
#include <vector>
#include <string>

namespace Slic3r { namespace GUI {

class FilamentPickerDialog : public DPIDialog
{
public:
    FilamentPickerDialog(wxWindow *parent, const wxString &fila_id, const FilamentColor &fila_color, const std::string &fila_type,
                         std::optional<FilamentColorPickerValue> ordered_color = std::nullopt);
    virtual ~FilamentPickerDialog();

    // Public interface methods
    bool HasSelectionChanged() const { return m_selection_changed; }
    const std::optional<ColorSelection>& GetCustomColorSelection() const { return m_custom_selection; }
    bool IsDataLoaded() const { return m_is_data_loaded; }
    wxColour GetSelectedColour() const;
    const FilamentColor& GetSelectedFilamentColor() const { return m_cur_filament_color; }

protected:
    void on_dpi_changed(const wxRect &suggested_rect) override;

    // Event handlers
#ifdef __WXGTK__
    void OnWindowCreate(wxWindowCreateEvent& event);
#endif
    void OnMouseLeftDown(wxMouseEvent& event);
    void OnMouseMove(wxMouseEvent& event);
    void OnMouseLeftUp(wxMouseEvent& event);
    void OnButtonPaint(wxPaintEvent& event);
    void OnTimerCheck(wxTimerEvent& event);
    void OnFlashTimer(wxTimerEvent& event);

    // Platform-independent window detection
    bool IsClickOnTopMostWindow(const wxPoint& mouse_pos);
    void StartClickDetection();
    void StopClickDetection();
    void CleanupTimers();

private:
    // UI creation methods
    wxBoxSizer* CreatePreviewPanel(const FilamentColor& fila_color, const std::string& fila_type);
    wxScrolledWindow* CreateColorGrid();
    wxBoxSizer* CreateSeparatorLine();
    void CreateMoreInfoButton();
    void BindEvents();

    // Preview panel helper methods
    void CreateColorBitmap(const FilamentColor& fila_color);
    wxBoxSizer* CreateInfoSection();
    void SetupLabelsContent(const FilamentColor& fila_color, const std::string& fila_type);

    // UI update methods
    void UpdatePreview(const FilamentColorCode& filament);
    void UpdateCustomColorPreview(const std::vector<wxColour>& colors, bool gradient);
    void UpdateButtonStates(wxBitmapButton* selected_btn);

    // Shaped window methods
    void SetWindowShape();
    void CreateShapedBitmap();

    // Data loading
    bool LoadFilamentData(const wxString& fila_id);

    // Flash effect
    void StartFlashing();

    // UI elements
    wxStaticBitmap* m_color_demo{nullptr};
    wxStaticText* m_label_preview_color{nullptr};
    wxStaticText* m_label_preview_idx{nullptr};
    wxStaticText* m_label_preview_type{nullptr};
    Button* m_more_btn{nullptr};
    Button* m_ok_btn{nullptr};
    Button* m_cancel_btn{nullptr};

    // Data members
    bool m_is_data_loaded{false};
    wxString *m_cur_color_name{nullptr};
    FilamentColorCodeQuery* m_color_query{nullptr};
    FilamentColorCodes* m_cur_color_codes{nullptr};
    wxBitmapButton* m_cur_selected_btn{nullptr};
    FilamentColor m_cur_filament_color;
    std::optional<FilamentColorPickerValue> m_ordered_color;
    std::optional<ColorSelection> m_custom_selection;
    bool m_selection_changed = false;

    // Shaped window members
    wxBitmap m_shape_bmp;
    int m_corner_radius{8};

    // Mouse drag members
    wxPoint m_drag_delta;

    // Click detection timers
    wxTimer* m_click_timer{nullptr};
    bool m_last_mouse_down{false};

    // Flash effect timer
    wxTimer* m_flash_timer{nullptr};
    int m_flash_step{0};
};

}} // namespace Slic3r::GUI

#endif