#pragma once
// SurfaceAnalyticsScreen — 35 financial surface visualizations
// Equity derivatives, fixed income, FX, credit, commodities, risk, macro

#include "SurfaceCapabilities.h"
#include "SurfaceDemoData.h"
#include "SurfaceTypes.h"
#include "core/symbol/IGroupLinked.h"
#include "screens/common/IStatefulScreen.h"
#include "services/databento/DatabentoService.h"

#include <QEvent>
#include <QHash>
#include <QLabel>
#include <QPushButton>
#include <QScrollArea>
#include <QSet>
#include <QSplitter>
#include <QStackedWidget>
#include <QString>
#include <QVBoxLayout>
#include <QWidget>

namespace fincept::surface {

class Surface3DWidget;
class SurfaceTableWidget;
class SurfaceLineWidget;
class SurfaceControlPanel;
class SurfaceDataInspector;

class SurfaceAnalyticsScreen : public QWidget, public fincept::screens::IStatefulScreen, public fincept::IGroupLinked {
    Q_OBJECT
    Q_INTERFACES(fincept::IGroupLinked)
  public:
    explicit SurfaceAnalyticsScreen(QWidget* parent = nullptr);

    void restore_state(const QVariantMap& state) override;
    QVariantMap save_state() const override;
    QString state_key() const override { return "surface_analytics"; }
    int state_version() const override { return 1; }

    // IGroupLinked — subscribe-only. Equity group changes are forwarded to
    // the SurfaceControlPanel which becomes the new symbol of record; demo data
    // and chart re-render via on_control_symbol_changed.
    void set_group(fincept::SymbolGroup g) override { link_group_ = g; }
    fincept::SymbolGroup group() const override { return link_group_; }
    void on_group_symbol_changed(const fincept::SymbolRef& ref) override;
    fincept::SymbolRef current_symbol() const override;

  private slots:
    void on_category_clicked(int index);
    void on_surface_clicked(int cat, int surf_index);
    void on_view_3d();
    void on_view_table();
    void on_view_line();
    void on_import_csv();
    void on_refresh();
    void on_controls_changed();
    void on_control_symbol_changed(const QString& sym);
    void on_fetch_requested();
    // Databento data slots
    void on_vol_surface_received(const fincept::DatabentoVolSurfaceResult& r);
    void on_ohlcv_received(const fincept::DatabentoOhlcvResult& r);
    void on_futures_received(const fincept::DatabentoFuturesResult& r);
    void on_surface_received(const fincept::DatabentoSurfaceResult& r);
    void on_db_fetch_started(const QString& desc);
    void on_db_fetch_failed(const QString& err);
    void on_db_connection_tested(bool ok, const QString& msg);
    void on_db_raw_response(const QString& cmd, const QString& raw_stdout);

  protected:
    void showEvent(QShowEvent* e) override;
    void hideEvent(QHideEvent* e) override;
    void changeEvent(QEvent* e) override;

  private:
    enum class ViewMode { Surface3D = 0, Table = 1, Line = 2 };

    void setup_ui();
    /// Re-apply tr() lookups to the fixed-label toolbar buttons.
    void retranslateUi();
    QWidget* build_category_bar();
    QWidget* build_surface_bar();
    void refresh_surface_bar();
    void load_demo_data();
    void update_chart();
    void update_metrics();
    void update_inspector_lineage();
    void dispatch_csv(const QString& path);
    void apply_view_mode_buttons();
    void update_line_view();
    void refresh_provider_status();
    void load_dataset_range_for_active_capability();

    /// Mark `type`'s z-grid as holding genuine (fetched or imported) data and
    /// refresh the tier badge. Called only from a provider/CSV path that
    /// actually wrote the grid.
    void mark_chart_real(ChartType type);
    /// Re-apply the control panel's DEMO badge for whatever surface is on
    /// screen. Every surface starts synthetic; the badge is per-surface, not
    /// per-screen, so switching from a fetched surface to an un-fetched one has
    /// to put DEMO back.
    void sync_synthetic_badge();

    QString current_symbol_or_default() const;
    float spot_for(const QString& sym) const;
    const std::vector<std::vector<float>>* active_z_grid() const;

    // Control bar widgets (kept for dynamic rebuild)
    QWidget* category_bar_ = nullptr;
    QWidget* surface_bar_ = nullptr;
    QPushButton* btn_3d_ = nullptr;
    QPushButton* btn_table_ = nullptr;
    QPushButton* btn_line_ = nullptr;
    QPushButton* import_btn_ = nullptr;  // fixed-label, cached for retranslateUi
    QPushButton* refresh_btn_ = nullptr; // fixed-label, cached for retranslateUi

    // Main panels
    SurfaceControlPanel* control_panel_ = nullptr;
    SurfaceDataInspector* data_inspector_ = nullptr;
    Surface3DWidget* surface_3d_ = nullptr;
    SurfaceTableWidget* surface_table_ = nullptr;
    SurfaceLineWidget* surface_line_ = nullptr;
    QStackedWidget* view_stack_ = nullptr;
    /// Always-visible banner directly above the chart, shown whenever the
    /// surface on screen is SurfaceDemoData's rand() output. The control
    /// panel's DEMO badge lives inside the ASSET group box, which
    /// apply_capability_visibility() *hides* for the 12 surfaces that need no
    /// underlying — precisely the always-synthetic ones. Those users saw
    /// fabricated curves with no marking at all.
    QLabel* demo_banner_ = nullptr;

    // State
    int active_category_ = 0;
    ChartType active_chart_ = ChartType::Volatility;
    ViewMode view_mode_ = ViewMode::Surface3D;

    // ---- Data for all 35 surfaces ----
    VolatilitySurfaceData vol_data_;
    GreeksSurfaceData delta_data_, gamma_data_, vega_data_, theta_data_;
    SkewSurfaceData skew_data_;
    LocalVolData local_vol_data_;
    YieldCurveData yield_data_;
    SwaptionVolData swaption_data_;
    CapFloorVolData capfloor_data_;
    BondSpreadData bond_spread_data_;
    OISBasisData ois_data_;
    RealYieldData real_yield_data_;
    ForwardRateData fwd_rate_data_;
    FXVolData fx_vol_data_;
    FXForwardPointsData fx_fwd_data_;
    CrossCurrencyBasisData xccy_data_;
    CDSSpreadData cds_data_;
    CreditTransitionData credit_trans_data_;
    RecoveryRateData recovery_data_;
    CommodityForwardData cmdty_fwd_data_;
    CommodityVolData cmdty_vol_data_;
    CrackSpreadData crack_data_;
    ContangoData contango_data_;
    CorrelationMatrixData corr_data_;
    PCAData pca_data_;
    VaRData var_data_;
    StressTestData stress_data_;
    FactorExposureData factor_data_;
    LiquidityData liquidity_data_;
    DrawdownData drawdown_data_;
    BetaData beta_data_;
    ImpliedDividendData impl_div_data_;
    InflationExpData inflation_data_;
    MonetaryPolicyData monetary_data_;

    /// ChartTypes (as int) whose z-grid currently holds real fetched/imported
    /// data rather than SurfaceDemoData's rand() output. Everything not in here
    /// is synthetic and must be badged DEMO. Cleared by load_demo_data().
    QSet<int> real_data_charts_;

    // Symbol group link — SymbolGroup::None when unlinked.
    fincept::SymbolGroup link_group_ = fincept::SymbolGroup::None;

    // Spot cache, populated from DataHub::peek("market:quote:<sym>") on demand.
    // Falls back to a constant if the hub has no quote for the symbol.
    QHash<QString, float> spot_cache_;

    // Category accent colors (R,G,B)
    static constexpr int CAT_COLORS[][3] = {
        {51, 153, 255},  // EQUITY DERIV — blue
        {102, 217, 140}, // FIXED INCOME — green
        {255, 191, 51},  // FX           — gold
        {230, 89, 89},   // CREDIT       — red
        {191, 140, 255}, // COMMODITIES  — purple
        {255, 140, 38},  // RISK         — orange
        {89, 217, 242},  // MACRO        — cyan
    };
};

} // namespace fincept::surface