NodeWidget.h

Go to the documentation of this file.

#pragma once
 
#include "Core/Node.h"
#include <memory>
#include <wx/panel.h>
#include <wx/wx.h>
 
namespace EmberForge {

class NodeWidget : public wxPanel {
  public:
    enum class RenderStyle {
        Compact,  // Minimal representation
        Normal,   // Standard representation with icon and text
        Detailed, // Full representation with all node information
        Debug     // Debug view with internal state
    };

    enum class InteractionState {
        Normal,   // Default state
        Hovered,  // Mouse hovering over node
        Selected, // Node is selected
        Dragging, // Node is being dragged
        Editing   // Node is being edited
    };

    NodeWidget(wxWindow *parent, EmberCore::Node *node, RenderStyle style = RenderStyle::Normal);

    virtual ~NodeWidget();
 
    // Node management
    void SetNode(EmberCore::Node *node);
    EmberCore::Node *GetNode() const { return node_; }
    bool HasNode() const { return node_ != nullptr; }
 
    // Visual properties
    RenderStyle GetRenderStyle() const { return render_style_; }
    void SetRenderStyle(RenderStyle style);
 
    InteractionState GetInteractionState() const { return interaction_state_; }
    void SetInteractionState(InteractionState state);
 
    // Selection
    bool IsSelected() const { return interaction_state_ == InteractionState::Selected; }
    void SetSelected(bool selected);
 
    // Editing
    bool IsEditing() const { return interaction_state_ == InteractionState::Editing; }
    void StartEditing();
    void StopEditing(bool save_changes = true);
 
    // Visual state synchronization
    void UpdateFromNode();
    void SyncVisualState();
 
    // Size and layout
    wxSize GetPreferredSize() const;
    wxSize CalculateMinSize() const;
    void SetCustomSize(const wxSize &size);
 
    // Appearance customization
    void SetNodeColors(const wxColour &background, const wxColour &border, const wxColour &text);
    void SetNodeFont(const wxFont &font);
    void ResetToDefaultAppearance();
 
    // Animation support
    void StartPulseAnimation();
    void StopPulseAnimation();
    void SetHighlighted(bool highlighted);
 
    // Drag & Drop support
    bool IsDraggable() const { return is_draggable_; }
    void SetDraggable(bool draggable) { is_draggable_ = draggable; }
 
    // Event callbacks
    using NodeClickCallback = std::function<void(NodeWidget *, EmberCore::Node *)>;
    using NodeDoubleClickCallback = std::function<void(NodeWidget *, EmberCore::Node *)>;
    using NodeDragCallback = std::function<void(NodeWidget *, const wxPoint &delta)>;
    using NodeEditCallback = std::function<void(NodeWidget *, const wxString &new_name)>;
 
    void SetNodeClickCallback(NodeClickCallback callback) { click_callback_ = callback; }
    void SetNodeDoubleClickCallback(NodeDoubleClickCallback callback) { double_click_callback_ = callback; }
    void SetNodeDragCallback(NodeDragCallback callback) { drag_callback_ = callback; }
    void SetNodeEditCallback(NodeEditCallback callback) { edit_callback_ = callback; }
 
  protected:
    // Rendering methods
    virtual void OnPaint(wxPaintEvent &event);
    virtual void OnSize(wxSizeEvent &event);
    virtual void OnEraseBackground(wxEraseEvent &event);
 
    // Mouse events
    virtual void OnMouseEnter(wxMouseEvent &event);
    virtual void OnMouseLeave(wxMouseEvent &event);
    virtual void OnMouseDown(wxMouseEvent &event);
    virtual void OnMouseUp(wxMouseEvent &event);
    virtual void OnMouseMove(wxMouseEvent &event);
    virtual void OnMouseDoubleClick(wxMouseEvent &event);
    virtual void OnRightClick(wxContextMenuEvent &event);
 
    // Keyboard events
    virtual void OnKeyDown(wxKeyEvent &event);
    virtual void OnChar(wxKeyEvent &event);
 
    // Focus events
    virtual void OnSetFocus(wxFocusEvent &event);
    virtual void OnKillFocus(wxFocusEvent &event);
 
    // Drawing helpers
    void DrawNodeBackground(wxDC &dc, const wxRect &rect);
    void DrawNodeBorder(wxDC &dc, const wxRect &rect);
    void DrawNodeIcon(wxDC &dc, const wxRect &icon_rect);
    void DrawNodeText(wxDC &dc, const wxRect &text_rect);
    void DrawNodeStatus(wxDC &dc, const wxRect &status_rect);
    void DrawConnectionPoints(wxDC &dc);
 
    // Layout helpers
    wxRect GetContentRect() const;
    wxRect GetIconRect() const;
    wxRect GetTextRect() const;
    wxRect GetStatusRect() const;
    std::vector<wxPoint> GetConnectionPoints() const;
 
    // State helpers
    wxColour GetBackgroundColor() const;
    wxColour GetBorderColor() const;
    wxColour GetTextColor() const;
    wxBitmap GetNodeIcon() const;
    wxString GetDisplayText() const;
 
    // Animation helpers
    void OnAnimationTimer(wxTimerEvent &event);
    void UpdateAnimation();
 
  private:
    // Core data
    EmberCore::Node *node_;
    RenderStyle render_style_;
    InteractionState interaction_state_;
 
    // Visual properties
    wxSize custom_size_;
    wxColour background_color_;
    wxColour border_color_;
    wxColour text_color_;
    wxFont node_font_;
 
    // Interaction state
    bool is_draggable_;
    bool is_highlighted_;
    wxPoint drag_start_pos_;
    wxPoint last_mouse_pos_;
 
    // Animation
    wxTimer *animation_timer_;
    double animation_phase_;
    bool is_animating_;
 
    // Text editing
    wxTextCtrl *edit_control_;
    wxString original_name_;
 
    // Event callbacks
    NodeClickCallback click_callback_;
    NodeDoubleClickCallback double_click_callback_;
    NodeDragCallback drag_callback_;
    NodeEditCallback edit_callback_;
 
    // Helper methods
    void InitializeWidget();
    void UpdateColors();
    void CreateEditControl();
    void DestroyEditControl();
    void ShowContextMenu(const wxPoint &position);
 
    // Constants
    static const int BORDER_WIDTH = 2;
    static const int PADDING = 6;
    static const int ICON_SIZE = 16;
    static const int MIN_WIDTH = 80;
    static const int MIN_HEIGHT = 32;
 
    DECLARE_EVENT_TABLE()
};
 
} // namespace EmberForge